fix

Signed-off-by: dbczumar <[email protected]>

Author: dbczumar

website/src/components/GetStartedWithMLflow/GetStartedWithMLflow.tsx

@@ -5,7 +5,7 @@ import useBaseUrl from "@docusaurus/useBaseUrl";
 import useBrokenLinks from "@docusaurus/useBrokenLinks";
 
 import { GetStartedButton } from "../GetStartedButton/GetStartedButton";
-import { cn } from "../../utils";
+import { cn, isClassicalMLPage } from "../../utils";
 import { Heading } from "../Typography/Heading";
 import { Body } from "../Typography/Body";
 import { useLayoutVariant } from "../Layout/Layout";
@@ -19,11 +19,11 @@ export const GetStartedWithMLflow = () => {
   const variant = useLayoutVariant();
   const location = useLocation();
   const classicalMLPath = useBaseUrl("/classical-ml");
-  const isClassicalMLPage = location.pathname.startsWith(classicalMLPath);
-  const databricksUrl = isClassicalMLPage
+  const isClassicalML = isClassicalMLPage(location.pathname, classicalMLPath);
+  const databricksUrl = isClassicalML
     ? MLFLOW_DBX_INSTALL_URL
     : MLFLOW_DBX_TRIAL_URL;
-  const databricksButtonText = isClassicalMLPage
+  const databricksButtonText = isClassicalML
     ? "Get started"
     : "Get started for free";
 

website/src/components/Header/Header.tsx

@@ -5,7 +5,7 @@ import useBaseUrl from "@docusaurus/useBaseUrl";
 import Logo from "@site/static/img/mlflow-logo-white.svg";
 import DownIcon from "@site/static/img/chevron-down-small.svg";
 
-import { cn } from "../../utils";
+import { cn, getStartedLinkForPage } from "../../utils";
 
 import { Button } from "../Button/Button";
 import { HeaderMenuItem } from "../HeaderMenuItem/HeaderMenuItem";
@@ -17,19 +17,6 @@ import { cva } from "class-variance-authority";
 
 const MD_BREAKPOINT = 640;
 
-// Helper function to determine the appropriate "Get started" link based on current page
-function getStartedLinkForPage(pathname: string): string {
-  const classicalMLPath = useBaseUrl("/classical-ml");
-  const genAIPath = useBaseUrl("/genai");
-
-  if (pathname.startsWith(classicalMLPath)) {
-    return "/classical-ml#get-started";
-  } else if (pathname.startsWith(genAIPath)) {
-    return "/genai#get-started";
-  } else {
-    return "/#get-started";
-  }
-}
 
 const navStyles = cva(
   "fixed w-full z-20 top-0 start-0 bg-black/20 border-b border-[#F7F8F8]/8 backdrop-blur-[20px] overflow-y-auto",
@@ -48,8 +35,10 @@ export const Header = () => {
   const [isProductItemHovered, setIsProductItemHovered] = useState(false);
   const [isProductSubmenuOpen, setIsProductSubmenuOpen] = useState(false);
   const location = useLocation();
+  const classicalMLPath = useBaseUrl("/classical-ml");
+  const genAIPath = useBaseUrl("/genai");
 
-  const getStartedHref = getStartedLinkForPage(location.pathname);
+  const getStartedHref = getStartedLinkForPage(location.pathname, classicalMLPath, genAIPath);
 
   const handleProductItemHover = () => {
     setIsProductItemHovered(true);

website/src/utils.ts

@@ -4,3 +4,28 @@ import { twMerge } from "tailwind-merge";
 export function cn(...inputs: ClassValue[]) {
   return twMerge(clsx(inputs));
 }
+
+// Helper function to determine the appropriate "Get started" link based on current page
+export function getStartedLinkForPage(
+  pathname: string,
+  classicalMLPath: string,
+  genAIPath: string
+): string {
+  if (pathname.startsWith(classicalMLPath)) {
+    return "/classical-ml#get-started";
+  }
+  if (pathname.startsWith(genAIPath)) {
+    return "/genai#get-started";
+  }
+  return "/#get-started";
+}
+
+// Helper function to determine if current page is classical ML
+export function isClassicalMLPage(pathname: string, classicalMLPath: string): boolean {
+  return pathname.startsWith(classicalMLPath);
+}
+
+// Helper function to determine if current page is GenAI
+export function isGenAIPage(pathname: string, genAIPath: string): boolean {
+  return pathname.startsWith(genAIPath);
+}