From 54bef9f63be279d351e57b735d24e32de23adf5c Mon Sep 17 00:00:00 2001
From: Andrew Jiang <andrew@buildwithfern.com>
Date: Mon, 7 Oct 2024 19:20:06 -0400
Subject: [PATCH] code split

---
 .../src/jsonld/components/Breadcrumb.tsx      |   1 +
 packages/commons/search-utils/package.json    |   3 +-
 .../commons/search-utils/src/SearchConfig.ts  |  29 ++---
 packages/ui/app/package.json                  |   2 +-
 packages/ui/app/src/atoms/auth.ts             |   2 +-
 packages/ui/app/src/atoms/types.ts            |   2 +-
 .../ui/app/src/components/LinkPreload.tsx     |  22 ++++
 packages/ui/app/src/docs/DocsPage.tsx         |   3 +
 .../components/launchdarkly/LaunchDarkly.tsx  |  48 +++++---
 .../src/services/useApiKeyInjectionConfig.ts  |   2 +-
 packages/ui/app/tsconfig.json                 |   3 +
 packages/ui/docs-bundle/next.config.mjs       |   3 +-
 packages/ui/docs-bundle/package.json          |   3 +-
 packages/ui/docs-bundle/src/middleware.ts     |   2 +-
 .../api/fern-docs/auth/api-key-injection.ts   |   2 +-
 .../src/pages/api/fern-docs/auth/callback.ts  |   2 +-
 .../fern-docs/integrations/launchdarkly.ts    | 106 ++++++++++++++----
 .../pages/api/fern-docs/oauth/ory/callback.ts |   2 +-
 .../src/pages/api/fern-docs/search.ts         |   5 +-
 .../ui/docs-bundle/src/server/DocsLoader.ts   |   2 +-
 .../ui/docs-bundle/src/server/auth/FernJWT.ts |   2 +-
 .../src/server/auth/OAuth2Client.ts           |   6 +-
 .../src/server/auth/checkViewerAllowed.ts     |   2 +-
 .../server/auth/getApiKeyInjectionConfig.ts   |   2 +-
 .../ui/docs-bundle/src/server/authProps.ts    |   2 +-
 .../src/server/getDocsPageProps.ts            |   2 +-
 .../src/server/withBasicTokenPublic.ts        |   2 +-
 .../src/server/withInitialProps.ts            |  15 ---
 packages/ui/docs-bundle/tsconfig.json         |   5 +-
 packages/ui/fern-docs-auth/.depcheckrc.json   |   1 +
 packages/ui/fern-docs-auth/.prettierrc.cjs    |   1 +
 packages/ui/fern-docs-auth/.stylelintrc.json  |   1 +
 packages/ui/fern-docs-auth/package.json       |  50 +++++++++
 .../src/auth => fern-docs-auth/src}/index.ts  |   0
 .../auth => fern-docs-auth/src}/injection.ts  |   0
 .../src/auth => fern-docs-auth/src}/types.ts  |   0
 packages/ui/fern-docs-auth/tsconfig.json      |   8 ++
 packages/ui/fern-docs-auth/vitest.config.ts   |   7 ++
 packages/ui/fern-docs-edge/package.json       |   7 +-
 .../fern-docs-edge/src/getAuthEdgeConfig.ts   |   4 +-
 .../src/getCustomerAnalytics.ts               |  11 +-
 .../ui/fern-docs-edge/src/getFeatureFlags.ts  |   3 +-
 .../fern-docs-edge/src/getInkeepSettings.ts   |   8 ++
 .../src/getLaunchDarklySettings.ts            |  15 +++
 packages/ui/fern-docs-edge/src/index.ts       |   2 +
 packages/ui/fern-docs-edge/tsconfig.json      |   6 +
 pnpm-lock.yaml                                |  52 ++++++++-
 47 files changed, 351 insertions(+), 107 deletions(-)
 create mode 100644 packages/ui/app/src/components/LinkPreload.tsx
 create mode 100644 packages/ui/fern-docs-auth/.depcheckrc.json
 create mode 100644 packages/ui/fern-docs-auth/.prettierrc.cjs
 create mode 100644 packages/ui/fern-docs-auth/.stylelintrc.json
 create mode 100644 packages/ui/fern-docs-auth/package.json
 rename packages/ui/{app/src/auth => fern-docs-auth/src}/index.ts (100%)
 rename packages/ui/{app/src/auth => fern-docs-auth/src}/injection.ts (100%)
 rename packages/ui/{app/src/auth => fern-docs-auth/src}/types.ts (100%)
 create mode 100644 packages/ui/fern-docs-auth/tsconfig.json
 create mode 100644 packages/ui/fern-docs-auth/vitest.config.ts
 create mode 100644 packages/ui/fern-docs-edge/src/getInkeepSettings.ts
 create mode 100644 packages/ui/fern-docs-edge/src/getLaunchDarklySettings.ts

diff --git a/packages/commons/next-seo/src/jsonld/components/Breadcrumb.tsx b/packages/commons/next-seo/src/jsonld/components/Breadcrumb.tsx
index 5eab60bfc1..3eb680a5ff 100644
--- a/packages/commons/next-seo/src/jsonld/components/Breadcrumb.tsx
+++ b/packages/commons/next-seo/src/jsonld/components/Breadcrumb.tsx
@@ -7,6 +7,7 @@ export const Breadcrumb = memo(
         return (
             <Head>
                 <script
+                    key="jsonld-breadcrumb"
                     type="application/ld+json"
                     id="jsonld-breadcrumb"
                     dangerouslySetInnerHTML={{
diff --git a/packages/commons/search-utils/package.json b/packages/commons/search-utils/package.json
index 352a4feb66..4f2e237412 100644
--- a/packages/commons/search-utils/package.json
+++ b/packages/commons/search-utils/package.json
@@ -31,8 +31,7 @@
   },
   "dependencies": {
     "@fern-api/fdr-sdk": "workspace:*",
-    "@fern-ui/core-utils": "workspace:*",
-    "@vercel/edge-config": "^1.1.0"
+    "@fern-ui/core-utils": "workspace:*"
   },
   "devDependencies": {
     "@fern-platform/configs": "workspace:*",
diff --git a/packages/commons/search-utils/src/SearchConfig.ts b/packages/commons/search-utils/src/SearchConfig.ts
index c9872df531..095fc3c5bb 100644
--- a/packages/commons/search-utils/src/SearchConfig.ts
+++ b/packages/commons/search-utils/src/SearchConfig.ts
@@ -7,11 +7,8 @@ import type {
     InkeepSearchSettings,
     InkeepWidgetBaseSettings,
 } from "@inkeep/widgets";
-import { getAll } from "@vercel/edge-config";
 import type { DeepReadonly } from "ts-essentials";
 
-const FEATURE_FLAGS = ["inkeep-enabled" as const];
-
 export type InkeepSharedSettings = DeepReadonly<{
     replaceSearch: boolean;
     baseSettings: InkeepWidgetBaseSettings;
@@ -20,10 +17,6 @@ export type InkeepSharedSettings = DeepReadonly<{
     modalSettings?: Partial<InkeepModalSettings>;
 }>;
 
-interface EdgeConfigResponse {
-    "inkeep-enabled"?: Record<string, InkeepSharedSettings>;
-}
-
 export declare namespace SearchConfig {
     interface Unversioned {
         type: "unversioned";
@@ -128,8 +121,8 @@ async function getAlgoliaSearchConfig(
 
 export async function getSearchConfig(
     client: FdrClient,
-    domain: string,
     searchInfo: Algolia.SearchInfo,
+    inkeep: InkeepSharedSettings | undefined,
 ): Promise<SearchConfig> {
     const algolia = await getAlgoliaSearchConfig(client, searchInfo);
 
@@ -138,20 +131,12 @@ export async function getSearchConfig(
         return { isAvailable: false };
     }
 
-    try {
-        const config = await getAll<EdgeConfigResponse>(FEATURE_FLAGS);
-        const maybeInkeep = config["inkeep-enabled"]?.[domain];
-
-        if (maybeInkeep?.baseSettings.integrationId != null) {
-            return {
-                isAvailable: true,
-                algolia,
-                inkeep: maybeInkeep,
-            };
-        }
-    } catch (e) {
-        // eslint-disable-next-line no-console
-        console.error("Error fetching edge config", e);
+    if (inkeep?.baseSettings.integrationId != null) {
+        return {
+            isAvailable: true,
+            algolia,
+            inkeep,
+        };
     }
 
     return { isAvailable: true, algolia };
diff --git a/packages/ui/app/package.json b/packages/ui/app/package.json
index 69c75b4735..7df90263ae 100644
--- a/packages/ui/app/package.json
+++ b/packages/ui/app/package.json
@@ -11,7 +11,6 @@
   "exports": {
     "./bundlers": "./src/mdx/bundlers/index.ts",
     "./bundlers/next-mdx-remote": "./src/mdx/bundlers/next-mdx-remote.ts",
-    "./auth": "./src/auth/index.ts",
     ".": "./src/index.ts"
   },
   "sideEffects": [
@@ -121,6 +120,7 @@
     "zod": "^3.23.8"
   },
   "devDependencies": {
+    "@fern-ui/fern-docs-auth": "workspace:*",
     "@chromatic-com/storybook": "^1.3.5",
     "@fern-platform/configs": "workspace:*",
     "@mdx-js/esbuild": "^3.0.1",
diff --git a/packages/ui/app/src/atoms/auth.ts b/packages/ui/app/src/atoms/auth.ts
index 84ee7a4f6a..ad6fb7226f 100644
--- a/packages/ui/app/src/atoms/auth.ts
+++ b/packages/ui/app/src/atoms/auth.ts
@@ -1,7 +1,7 @@
+import type { FernUser } from "@fern-ui/fern-docs-auth";
 import { useAtomValue } from "jotai";
 import { selectAtom } from "jotai/utils";
 import { isEqual } from "lodash-es";
-import type { FernUser } from "../auth";
 import { DOCS_ATOM } from "./docs";
 
 export const FERN_USER_ATOM = selectAtom(DOCS_ATOM, (docs) => docs.user, isEqual);
diff --git a/packages/ui/app/src/atoms/types.ts b/packages/ui/app/src/atoms/types.ts
index bda1d88d1d..22f0d579b3 100644
--- a/packages/ui/app/src/atoms/types.ts
+++ b/packages/ui/app/src/atoms/types.ts
@@ -2,9 +2,9 @@ import type { DocsV1Read, DocsV2Read, FdrAPI } from "@fern-api/fdr-sdk/client/ty
 import type * as FernDocs from "@fern-api/fdr-sdk/docs";
 import type * as FernNavigation from "@fern-api/fdr-sdk/navigation";
 import { ColorsConfig, SidebarTab, VersionSwitcherInfo } from "@fern-ui/fdr-utils";
+import type { FernUser } from "@fern-ui/fern-docs-auth";
 import { NextSeoProps } from "@fern-ui/next-seo";
 import { CustomerAnalytics } from "../analytics/types";
-import { FernUser } from "../auth";
 import { DocsContent } from "../resolver/DocsContent";
 import { FernTheme } from "../themes/ThemedDocs";
 
diff --git a/packages/ui/app/src/components/LinkPreload.tsx b/packages/ui/app/src/components/LinkPreload.tsx
new file mode 100644
index 0000000000..07374cafc8
--- /dev/null
+++ b/packages/ui/app/src/components/LinkPreload.tsx
@@ -0,0 +1,22 @@
+import Head from "next/head";
+import { ReactElement } from "react";
+import { useIsLocalPreview } from "../contexts/local-preview";
+import { useApiRoute } from "../hooks/useApiRoute";
+
+export function LinkPreload({ href }: { href: string }): ReactElement {
+    return (
+        <Head>
+            <link key={href} rel="preload" href={href} as="fetch" crossOrigin="anonymous" />
+        </Head>
+    );
+}
+
+type FernDocsApiRoute = `/api/fern-docs/${string}`;
+export function LinkPreloadApiRoute({ href }: { href: FernDocsApiRoute }): ReactElement | null {
+    const isLocalPreview = useIsLocalPreview();
+    const key = useApiRoute(href);
+    if (isLocalPreview) {
+        return null;
+    }
+    return <LinkPreload href={key} />;
+}
diff --git a/packages/ui/app/src/docs/DocsPage.tsx b/packages/ui/app/src/docs/DocsPage.tsx
index 8c0ac94aa0..80e206c7a5 100644
--- a/packages/ui/app/src/docs/DocsPage.tsx
+++ b/packages/ui/app/src/docs/DocsPage.tsx
@@ -3,6 +3,7 @@ import { ReactElement } from "react";
 import { HydrateAtoms, useMessageHandler, useSetJustNavigated, type DocsProps } from "../atoms";
 import { BgImageGradient } from "../components/BgImageGradient";
 import { JavascriptProvider } from "../components/JavascriptProvider";
+import { LinkPreloadApiRoute } from "../components/LinkPreload";
 import { useBeforePopState } from "../hooks/useBeforePopState";
 import { useConsoleMessage } from "../hooks/useConsoleMessage";
 import { useRouteChangeComplete, useRouteChangeStart } from "../hooks/useRouteChanged";
@@ -45,6 +46,8 @@ export function DocsPage(pageProps: DocsProps): ReactElement | null {
 
     return (
         <>
+            <LinkPreloadApiRoute href="/api/fern-docs/search" />
+            <LinkPreloadApiRoute href="/api/fern-docs/auth/api-key-injection" />
             <NextSeo />
             <InitializeTheme />
             <SearchDialog />
diff --git a/packages/ui/app/src/mdx/components/launchdarkly/LaunchDarkly.tsx b/packages/ui/app/src/mdx/components/launchdarkly/LaunchDarkly.tsx
index d6654ff3ec..a5903faaef 100644
--- a/packages/ui/app/src/mdx/components/launchdarkly/LaunchDarkly.tsx
+++ b/packages/ui/app/src/mdx/components/launchdarkly/LaunchDarkly.tsx
@@ -1,30 +1,43 @@
 import { captureException } from "@sentry/nextjs";
 import { atom, useAtomValue } from "jotai";
-import { loadable } from "jotai/utils";
+import { loadable, useHydrateAtoms } from "jotai/utils";
 import * as LDClient from "launchdarkly-js-client-sdk";
 import { PropsWithChildren, ReactNode, useCallback, useEffect, useState } from "react";
-import { selectApiRoute } from "../../../hooks/useApiRoute";
-
-async function fetchClientSideId(route: string): Promise<string | undefined> {
-    const json = await fetch(route).then((res) => res.json());
-    return json?.["client-side-id"];
+import useSWR from "swr";
+import { useApiRoute } from "../../../hooks/useApiRoute";
+
+interface LaunchDarklyInfo {
+    clientSideId: string;
+    kind: "multi";
+    user:
+        | { anonymous: true }
+        | {
+              key: string;
+              email?: string;
+              name?: string;
+          };
+    device: {
+        key: string;
+        [key: string]: unknown;
+    };
 }
 
+const LAUNCH_DARKLY_INFO_ATOM = atom<LaunchDarklyInfo | undefined>(undefined);
+
 // this is a singleton atom that initializes the LaunchDarkly client-side SDK
-const ldClientAtom = atom<Promise<LDClient.LDClient | undefined>>(async (get) => {
+const LD_CLIENT_ATOM = atom<Promise<LDClient.LDClient | undefined>>(async (get) => {
     if (typeof window === "undefined") {
         return undefined;
     }
-    const route = selectApiRoute(get, "/api/fern-docs/integrations/launchdarkly");
-    const clientSideId = await fetchClientSideId(route);
-    if (!clientSideId) {
+
+    const info = get(LAUNCH_DARKLY_INFO_ATOM);
+    if (!info) {
         return undefined;
     }
 
-    const client = LDClient.initialize(clientSideId, {
-        kind: "user",
-        anonymous: true,
-    });
+    const { clientSideId, ...context } = info;
+
+    const client = LDClient.initialize(clientSideId, context);
 
     await client.waitForInitialization();
 
@@ -32,8 +45,9 @@ const ldClientAtom = atom<Promise<LDClient.LDClient | undefined>>(async (get) =>
 });
 
 const useLaunchDarklyFlag = (flag: string): boolean => {
-    const loadableClient = useAtomValue(loadable(ldClientAtom));
+    const loadableClient = useAtomValue(loadable(LD_CLIENT_ATOM));
     const client = loadableClient.state === "hasData" ? loadableClient.data : undefined;
+
     if (loadableClient.state === "hasError") {
         captureException(loadableClient.error);
     }
@@ -72,6 +86,10 @@ export interface LaunchDarklyProps {
 }
 
 export function LaunchDarkly({ flag, children }: PropsWithChildren<LaunchDarklyProps>): ReactNode {
+    const route = useApiRoute("/api/fern-docs/integrations/launchdarkly");
+    const { data } = useSWR(route, (key): Promise<LaunchDarklyInfo> => fetch(key).then((res) => res.json()));
+    useHydrateAtoms([[LAUNCH_DARKLY_INFO_ATOM, data]], { dangerouslyForceHydrate: true });
+
     const ldClient = useLaunchDarklyFlag(flag);
 
     if (!ldClient) {
diff --git a/packages/ui/app/src/services/useApiKeyInjectionConfig.ts b/packages/ui/app/src/services/useApiKeyInjectionConfig.ts
index 3ce5d34740..feac8ea98e 100644
--- a/packages/ui/app/src/services/useApiKeyInjectionConfig.ts
+++ b/packages/ui/app/src/services/useApiKeyInjectionConfig.ts
@@ -1,5 +1,5 @@
+import type { APIKeyInjectionConfig } from "@fern-ui/fern-docs-auth";
 import useSWR from "swr";
-import { APIKeyInjectionConfig } from "../auth";
 import { useApiRoute } from "../hooks/useApiRoute";
 
 const DEFAULT = { enabled: false as const };
diff --git a/packages/ui/app/tsconfig.json b/packages/ui/app/tsconfig.json
index 21f290192a..e387640442 100644
--- a/packages/ui/app/tsconfig.json
+++ b/packages/ui/app/tsconfig.json
@@ -35,6 +35,9 @@
     {
       "path": "../chatbot"
     },
+    {
+      "path": "../fern-docs-auth"
+    },
     {
       "path": "../components"
     }
diff --git a/packages/ui/docs-bundle/next.config.mjs b/packages/ui/docs-bundle/next.config.mjs
index 2982f7d1a5..a5c822593d 100644
--- a/packages/ui/docs-bundle/next.config.mjs
+++ b/packages/ui/docs-bundle/next.config.mjs
@@ -54,8 +54,9 @@ const nextConfig = {
         "@fern-ui/components",
         "@fern-ui/core-utils",
         "@fern-ui/fdr-utils",
-        "@fern-ui/fern-docs-utils",
+        "@fern-ui/fern-docs-auth",
         "@fern-ui/fern-docs-edge",
+        "@fern-ui/fern-docs-utils",
         "@fern-ui/loadable",
         "@fern-ui/next-seo",
         "@fern-ui/react-commons",
diff --git a/packages/ui/docs-bundle/package.json b/packages/ui/docs-bundle/package.json
index a3df940032..844da601f5 100644
--- a/packages/ui/docs-bundle/package.json
+++ b/packages/ui/docs-bundle/package.json
@@ -35,8 +35,9 @@
     "@fern-ui/components": "workspace:*",
     "@fern-ui/core-utils": "workspace:*",
     "@fern-ui/fdr-utils": "workspace:*",
-    "@fern-ui/fern-docs-utils": "workspace:*",
+    "@fern-ui/fern-docs-auth": "workspace:*",
     "@fern-ui/fern-docs-edge": "workspace:*",
+    "@fern-ui/fern-docs-utils": "workspace:*",
     "@fern-ui/search-utils": "workspace:*",
     "@fern-ui/ui": "workspace:*",
     "@sentry/nextjs": "^8.30.0",
diff --git a/packages/ui/docs-bundle/src/middleware.ts b/packages/ui/docs-bundle/src/middleware.ts
index f0d4236cb1..c3d19e96f5 100644
--- a/packages/ui/docs-bundle/src/middleware.ts
+++ b/packages/ui/docs-bundle/src/middleware.ts
@@ -2,9 +2,9 @@ import { extractBuildId, extractNextDataPathname } from "@/server/extractNextDat
 import { getPageRoute, getPageRouteMatch, getPageRoutePath } from "@/server/pageRoutes";
 import { rewritePosthog } from "@/server/rewritePosthog";
 import { getXFernHostEdge } from "@/server/xfernhost/edge";
+import type { FernUser } from "@fern-ui/fern-docs-auth";
 import { getAuthEdgeConfig } from "@fern-ui/fern-docs-edge";
 import { COOKIE_FERN_TOKEN, HEADER_X_FERN_HOST } from "@fern-ui/fern-docs-utils";
-import type { FernUser } from "@fern-ui/ui/auth";
 import { removeTrailingSlash } from "next/dist/shared/lib/router/utils/remove-trailing-slash";
 import { NextRequest, NextResponse, type NextMiddleware } from "next/server";
 import urlJoin from "url-join";
diff --git a/packages/ui/docs-bundle/src/pages/api/fern-docs/auth/api-key-injection.ts b/packages/ui/docs-bundle/src/pages/api/fern-docs/auth/api-key-injection.ts
index d016221b8c..324e802541 100644
--- a/packages/ui/docs-bundle/src/pages/api/fern-docs/auth/api-key-injection.ts
+++ b/packages/ui/docs-bundle/src/pages/api/fern-docs/auth/api-key-injection.ts
@@ -2,9 +2,9 @@ import { OAuth2Client } from "@/server/auth/OAuth2Client";
 import { getAPIKeyInjectionConfig } from "@/server/auth/getApiKeyInjectionConfig";
 import { withSecureCookie } from "@/server/auth/withSecure";
 import { getXFernHostEdge } from "@/server/xfernhost/edge";
+import { APIKeyInjectionConfig, OryAccessTokenSchema } from "@fern-ui/fern-docs-auth";
 import { getAuthEdgeConfig } from "@fern-ui/fern-docs-edge";
 import { COOKIE_FERN_TOKEN } from "@fern-ui/fern-docs-utils";
-import { APIKeyInjectionConfig, OryAccessTokenSchema } from "@fern-ui/ui/auth";
 import { NextRequest, NextResponse } from "next/server";
 import { WebflowClient } from "webflow-api";
 import type { OauthScope } from "webflow-api/api/types/OAuthScope";
diff --git a/packages/ui/docs-bundle/src/pages/api/fern-docs/auth/callback.ts b/packages/ui/docs-bundle/src/pages/api/fern-docs/auth/callback.ts
index c61cdaa7f5..826017b6f8 100644
--- a/packages/ui/docs-bundle/src/pages/api/fern-docs/auth/callback.ts
+++ b/packages/ui/docs-bundle/src/pages/api/fern-docs/auth/callback.ts
@@ -2,9 +2,9 @@ import { signFernJWT } from "@/server/auth/FernJWT";
 import { withSecureCookie } from "@/server/auth/withSecure";
 import { getWorkOS, getWorkOSClientId } from "@/server/workos";
 import { getXFernHostEdge } from "@/server/xfernhost/edge";
+import { FernUser } from "@fern-ui/fern-docs-auth";
 import { getAuthEdgeConfig } from "@fern-ui/fern-docs-edge";
 import { COOKIE_FERN_TOKEN } from "@fern-ui/fern-docs-utils";
-import { FernUser } from "@fern-ui/ui/auth";
 import { NextRequest, NextResponse } from "next/server";
 
 export const runtime = "edge";
diff --git a/packages/ui/docs-bundle/src/pages/api/fern-docs/integrations/launchdarkly.ts b/packages/ui/docs-bundle/src/pages/api/fern-docs/integrations/launchdarkly.ts
index 068783eacd..5939854807 100644
--- a/packages/ui/docs-bundle/src/pages/api/fern-docs/integrations/launchdarkly.ts
+++ b/packages/ui/docs-bundle/src/pages/api/fern-docs/integrations/launchdarkly.ts
@@ -1,46 +1,108 @@
 import { verifyFernJWTConfig } from "@/server/auth/FernJWT";
 import { getXFernHostEdge } from "@/server/xfernhost/edge";
-import { getAuthEdgeConfig } from "@fern-ui/fern-docs-edge";
+import { LaunchDarklyEdgeConfig, getAuthEdgeConfig, getLaunchDarklySettings } from "@fern-ui/fern-docs-edge";
 import { COOKIE_EMAIL, COOKIE_FERN_TOKEN } from "@fern-ui/fern-docs-utils";
-import { get } from "@vercel/edge-config";
+import { randomUUID } from "crypto";
 import { cookies } from "next/headers";
-import { NextRequest, NextResponse } from "next/server";
-import { z } from "zod";
+import { NextRequest, NextResponse, userAgent } from "next/server";
 
 export const runtime = "edge";
 
-const LaunchDarklyEdgeConfigSchema = z.object({
-    // NOTE: this is client-side visible, so we should be careful about what we expose here if we add more fields
-    "client-side-id": z.string().optional(),
-});
-
-type LaunchDarklyEdgeConfigSchema = z.infer<typeof LaunchDarklyEdgeConfigSchema>;
+interface LaunchDarklyInfo {
+    clientSideId: string;
+    kind: "multi";
+    user:
+        | { anonymous: true }
+        | {
+              key: string;
+              email?: string;
+              name?: string;
+          };
+    device: {
+        key: string;
+        [key: string]: unknown;
+    };
+}
 
-export default async function handler(req: NextRequest): Promise<NextResponse<LaunchDarklyEdgeConfigSchema>> {
+export default async function handler(req: NextRequest): Promise<NextResponse<LaunchDarklyInfo | undefined>> {
     const domain = getXFernHostEdge(req);
+
+    const config = await safeGetLaunchDarklySettings(domain);
+    const clientSideId = config?.["client-side-id"];
+
+    if (clientSideId == null) {
+        return NextResponse.json(undefined, { status: 404 });
+    }
+
+    return NextResponse.json({
+        clientSideId,
+        kind: "multi" as const,
+        user: await getUserContext(req),
+        device: await getDeviceContext(req),
+    });
+}
+
+async function hashString(input: string | undefined): Promise<string | undefined> {
+    if (!input) {
+        return undefined;
+    }
+
+    // Encode the input string as a Uint8Array
+    const encoder = new TextEncoder();
+    const data = encoder.encode(input);
+
+    // Hash the data using SHA-256
+    const hashBuffer = await crypto.subtle.digest("SHA-256", data);
+
+    // Convert the hash to a hexadecimal string
+    const hashArray = Array.from(new Uint8Array(hashBuffer));
+    const hashHex = hashArray.map((byte) => byte.toString(16).padStart(2, "0")).join("");
+
+    return hashHex;
+}
+
+async function getUserContext(req: NextRequest): Promise<LaunchDarklyInfo["user"]> {
     const jar = cookies();
-    let email = jar.get(COOKIE_EMAIL)?.value;
+
     const fernToken = jar.get(COOKIE_FERN_TOKEN)?.value;
+    const email = jar.get(COOKIE_EMAIL)?.value;
 
-    if (fernToken && !email) {
+    if (fernToken) {
         try {
-            const user = await verifyFernJWTConfig(fernToken, await getAuthEdgeConfig(domain));
-            email = user.email;
+            const user = await verifyFernJWTConfig(fernToken, await getAuthEdgeConfig(getXFernHostEdge(req)));
+            const key = (await hashString(user.email)) ?? randomUUID();
+            return { key: `fern-docs-user-${key}`, email: user.email, name: user.name };
         } catch (e) {
             // do nothing
         }
     }
 
-    try {
-        const config = (await get<Record<string, LaunchDarklyEdgeConfigSchema>>("launchdarkly"))?.[domain];
-        if (config == null) {
-            return NextResponse.json({}, { status: 404 });
-        }
+    if (email) {
+        const key = (await hashString(email)) ?? randomUUID();
+        return { key: `fern-docs-user-${key}`, email };
+    }
 
-        return NextResponse.json(config);
+    return { anonymous: true };
+}
+
+async function getDeviceContext(req: NextRequest): Promise<LaunchDarklyInfo["device"]> {
+    const agent = userAgent(req);
+
+    const hash = (await hashString(agent.ua)) ?? randomUUID();
+
+    return {
+        ...agent,
+        key: `fern-docs-device-${hash}`,
+        locale: req.nextUrl.locale,
+    };
+}
+
+async function safeGetLaunchDarklySettings(domain: string): Promise<LaunchDarklyEdgeConfig | undefined> {
+    try {
+        return await getLaunchDarklySettings(domain);
     } catch (e) {
         // eslint-disable-next-line no-console
         console.error(e);
-        return NextResponse.json({}, { status: 500 });
+        return undefined;
     }
 }
diff --git a/packages/ui/docs-bundle/src/pages/api/fern-docs/oauth/ory/callback.ts b/packages/ui/docs-bundle/src/pages/api/fern-docs/oauth/ory/callback.ts
index 1134c20201..6244f8a3b3 100644
--- a/packages/ui/docs-bundle/src/pages/api/fern-docs/oauth/ory/callback.ts
+++ b/packages/ui/docs-bundle/src/pages/api/fern-docs/oauth/ory/callback.ts
@@ -2,9 +2,9 @@ import { signFernJWT } from "@/server/auth/FernJWT";
 import { OAuth2Client } from "@/server/auth/OAuth2Client";
 import { withSecureCookie } from "@/server/auth/withSecure";
 import { getXFernHostEdge } from "@/server/xfernhost/edge";
+import { FernUser, OryAccessTokenSchema } from "@fern-ui/fern-docs-auth";
 import { getAuthEdgeConfig } from "@fern-ui/fern-docs-edge";
 import { COOKIE_ACCESS_TOKEN, COOKIE_FERN_TOKEN, COOKIE_REFRESH_TOKEN } from "@fern-ui/fern-docs-utils";
-import { FernUser, OryAccessTokenSchema } from "@fern-ui/ui/auth";
 import { NextRequest, NextResponse } from "next/server";
 
 export const runtime = "edge";
diff --git a/packages/ui/docs-bundle/src/pages/api/fern-docs/search.ts b/packages/ui/docs-bundle/src/pages/api/fern-docs/search.ts
index 30fee119c8..1b7e4b2cb4 100644
--- a/packages/ui/docs-bundle/src/pages/api/fern-docs/search.ts
+++ b/packages/ui/docs-bundle/src/pages/api/fern-docs/search.ts
@@ -1,7 +1,7 @@
 import { checkViewerAllowedEdge } from "@/server/auth/checkViewerAllowed";
 import { loadWithUrl } from "@/server/loadWithUrl";
 import { getXFernHostEdge } from "@/server/xfernhost/edge";
-import { getAuthEdgeConfig } from "@fern-ui/fern-docs-edge";
+import { getAuthEdgeConfig, getInkeepSettings } from "@fern-ui/fern-docs-edge";
 import { SearchConfig, getSearchConfig } from "@fern-ui/search-utils";
 import { provideRegistryService } from "@fern-ui/ui";
 import { captureException } from "@sentry/nextjs";
@@ -31,8 +31,9 @@ export default async function handler(req: NextRequest): Promise<NextResponse<Se
             return NextResponse.json({ isAvailable: false }, { status: 503 });
         }
 
+        const inkeepSettings = await getInkeepSettings(domain);
         const searchInfo = docs.body.definition.search;
-        const config = await getSearchConfig(provideRegistryService(), domain, searchInfo);
+        const config = await getSearchConfig(provideRegistryService(), searchInfo, inkeepSettings);
         return NextResponse.json(config, { status: config.isAvailable ? 200 : 503 });
     } catch (e) {
         const id = captureException(e, { level: "fatal" });
diff --git a/packages/ui/docs-bundle/src/server/DocsLoader.ts b/packages/ui/docs-bundle/src/server/DocsLoader.ts
index aa26d1d90d..80265ad8a9 100644
--- a/packages/ui/docs-bundle/src/server/DocsLoader.ts
+++ b/packages/ui/docs-bundle/src/server/DocsLoader.ts
@@ -1,7 +1,7 @@
 import type { DocsV1Read, DocsV2Read } from "@fern-api/fdr-sdk";
 import * as FernNavigation from "@fern-api/fdr-sdk/navigation";
+import type { AuthEdgeConfig, FernUser } from "@fern-ui/fern-docs-auth";
 import { getAuthEdgeConfig } from "@fern-ui/fern-docs-edge";
-import type { AuthEdgeConfig, FernUser } from "@fern-ui/ui/auth";
 import { verifyFernJWTConfig } from "./auth/FernJWT";
 import { AuthProps } from "./authProps";
 import { loadWithUrl } from "./loadWithUrl";
diff --git a/packages/ui/docs-bundle/src/server/auth/FernJWT.ts b/packages/ui/docs-bundle/src/server/auth/FernJWT.ts
index 34e1bfbdc5..08bc41f4e7 100644
--- a/packages/ui/docs-bundle/src/server/auth/FernJWT.ts
+++ b/packages/ui/docs-bundle/src/server/auth/FernJWT.ts
@@ -1,4 +1,4 @@
-import { FernUserSchema, type AuthEdgeConfig, type FernUser } from "@fern-ui/ui/auth";
+import { FernUserSchema, type AuthEdgeConfig, type FernUser } from "@fern-ui/fern-docs-auth";
 import { SignJWT, jwtVerify } from "jose";
 
 // "user" is reserved for workos
diff --git a/packages/ui/docs-bundle/src/server/auth/OAuth2Client.ts b/packages/ui/docs-bundle/src/server/auth/OAuth2Client.ts
index ee36c44841..002fd243c5 100644
--- a/packages/ui/docs-bundle/src/server/auth/OAuth2Client.ts
+++ b/packages/ui/docs-bundle/src/server/auth/OAuth2Client.ts
@@ -1,4 +1,8 @@
-import { OAuthTokenResponseSchema, type AuthEdgeConfigOAuth2Ory, type OAuthTokenResponse } from "@fern-ui/ui/auth";
+import {
+    OAuthTokenResponseSchema,
+    type AuthEdgeConfigOAuth2Ory,
+    type OAuthTokenResponse,
+} from "@fern-ui/fern-docs-auth";
 import { JWTPayload, createRemoteJWKSet, decodeJwt, jwtVerify } from "jose";
 import { NextApiRequestCookies } from "next/dist/server/api-utils";
 import type { NextRequest } from "next/server";
diff --git a/packages/ui/docs-bundle/src/server/auth/checkViewerAllowed.ts b/packages/ui/docs-bundle/src/server/auth/checkViewerAllowed.ts
index 277b0b0cb7..14d43cf42b 100644
--- a/packages/ui/docs-bundle/src/server/auth/checkViewerAllowed.ts
+++ b/packages/ui/docs-bundle/src/server/auth/checkViewerAllowed.ts
@@ -1,5 +1,5 @@
+import { AuthEdgeConfig } from "@fern-ui/fern-docs-auth";
 import { COOKIE_FERN_TOKEN } from "@fern-ui/fern-docs-utils";
-import { AuthEdgeConfig } from "@fern-ui/ui/auth";
 import type { NextRequest } from "next/server";
 import { withBasicTokenPublic } from "../withBasicTokenPublic";
 import { verifyFernJWT } from "./FernJWT";
diff --git a/packages/ui/docs-bundle/src/server/auth/getApiKeyInjectionConfig.ts b/packages/ui/docs-bundle/src/server/auth/getApiKeyInjectionConfig.ts
index 21315e1e7d..0ee19181bf 100644
--- a/packages/ui/docs-bundle/src/server/auth/getApiKeyInjectionConfig.ts
+++ b/packages/ui/docs-bundle/src/server/auth/getApiKeyInjectionConfig.ts
@@ -1,5 +1,5 @@
+import type { APIKeyInjectionConfig } from "@fern-ui/fern-docs-auth";
 import { getAuthEdgeConfig } from "@fern-ui/fern-docs-edge";
-import type { APIKeyInjectionConfig } from "@fern-ui/ui/auth";
 import type { NextApiRequestCookies } from "next/dist/server/api-utils";
 import type { NextRequest } from "next/server";
 import { OAuth2Client } from "./OAuth2Client";
diff --git a/packages/ui/docs-bundle/src/server/authProps.ts b/packages/ui/docs-bundle/src/server/authProps.ts
index c4ef223c3a..91902355ff 100644
--- a/packages/ui/docs-bundle/src/server/authProps.ts
+++ b/packages/ui/docs-bundle/src/server/authProps.ts
@@ -1,5 +1,5 @@
+import type { FernUser } from "@fern-ui/fern-docs-auth";
 import { getAuthEdgeConfig } from "@fern-ui/fern-docs-edge";
-import type { FernUser } from "@fern-ui/ui/auth";
 import { verifyFernJWTConfig } from "./auth/FernJWT";
 
 export interface AuthProps {
diff --git a/packages/ui/docs-bundle/src/server/getDocsPageProps.ts b/packages/ui/docs-bundle/src/server/getDocsPageProps.ts
index cde6679c14..3342c30b77 100644
--- a/packages/ui/docs-bundle/src/server/getDocsPageProps.ts
+++ b/packages/ui/docs-bundle/src/server/getDocsPageProps.ts
@@ -1,5 +1,5 @@
+import type { FernUser } from "@fern-ui/fern-docs-auth";
 import type { DocsPage } from "@fern-ui/ui";
-import type { FernUser } from "@fern-ui/ui/auth";
 import type { GetServerSidePropsResult } from "next";
 import type { NextApiRequestCookies } from "next/dist/server/api-utils";
 import type { ComponentProps } from "react";
diff --git a/packages/ui/docs-bundle/src/server/withBasicTokenPublic.ts b/packages/ui/docs-bundle/src/server/withBasicTokenPublic.ts
index b51d2bb5ed..b774d4f50a 100644
--- a/packages/ui/docs-bundle/src/server/withBasicTokenPublic.ts
+++ b/packages/ui/docs-bundle/src/server/withBasicTokenPublic.ts
@@ -1,6 +1,6 @@
 import { getChildren, isLeaf, isPage, utils, type NavigationNode, type RootNode } from "@fern-api/fdr-sdk/navigation";
+import type { AuthEdgeConfigBasicTokenVerification } from "@fern-ui/fern-docs-auth";
 import { matchPath } from "@fern-ui/fern-docs-utils";
-import type { AuthEdgeConfigBasicTokenVerification } from "@fern-ui/ui/auth";
 import { captureMessage } from "@sentry/nextjs";
 
 /**
diff --git a/packages/ui/docs-bundle/src/server/withInitialProps.ts b/packages/ui/docs-bundle/src/server/withInitialProps.ts
index 6613be9624..f5933b8d99 100644
--- a/packages/ui/docs-bundle/src/server/withInitialProps.ts
+++ b/packages/ui/docs-bundle/src/server/withInitialProps.ts
@@ -4,24 +4,20 @@ import visitDiscriminatedUnion from "@fern-ui/core-utils/visitDiscriminatedUnion
 import { SidebarTab } from "@fern-ui/fdr-utils";
 import { getAuthEdgeConfig, getCustomerAnalytics, getFeatureFlags, getSeoDisabled } from "@fern-ui/fern-docs-edge";
 import { getRedirectForPath } from "@fern-ui/fern-docs-utils";
-import { getSearchConfig } from "@fern-ui/search-utils";
 import {
     DocsPage,
     getApiRouteSupplier,
     getGitHubInfo,
     getGitHubRepo,
     getSeoProps,
-    provideRegistryService,
     renderThemeStylesheet,
     resolveDocsContent,
 } from "@fern-ui/ui";
 import { getMdxBundler } from "@fern-ui/ui/bundlers";
 import { GetServerSidePropsResult } from "next";
-import type { NextApiRequestCookies } from "next/dist/server/api-utils";
 import { ComponentProps } from "react";
 import urlJoin from "url-join";
 import { DocsLoader } from "./DocsLoader";
-import { getAPIKeyInjectionConfigNode } from "./auth/getApiKeyInjectionConfig";
 import type { AuthProps } from "./authProps";
 import { handleLoadDocsError } from "./handleLoadDocsError";
 import type { LoadWithUrlResponse } from "./loadWithUrl";
@@ -33,7 +29,6 @@ interface WithInitialProps {
     slug: string[];
     xFernHost: string;
     auth?: AuthProps;
-    cookies?: NextApiRequestCookies;
 }
 
 export async function withInitialProps({
@@ -41,7 +36,6 @@ export async function withInitialProps({
     slug: slugArray,
     xFernHost,
     auth,
-    cookies,
 }: WithInitialProps): Promise<GetServerSidePropsResult<ComponentProps<typeof DocsPage>>> {
     if (!docsResponse.ok) {
         return handleLoadDocsError(xFernHost, slugArray, docsResponse.error);
@@ -284,12 +278,6 @@ export async function withInitialProps({
         ),
     };
 
-    props.fallback[getApiRoute("/api/fern-docs/search")] = await getSearchConfig(
-        provideRegistryService(),
-        xFernHost,
-        docs.definition.search,
-    );
-
     // if the user specifies a github navbar link, grab the repo info from it and save it as an SWR fallback
     const githubNavbarLink = docsConfig.navbarLinks?.find((link) => link.type === "github");
     if (githubNavbarLink) {
@@ -302,9 +290,6 @@ export async function withInitialProps({
         }
     }
 
-    const apiKeyInjectionConfig = await getAPIKeyInjectionConfigNode(xFernHost, cookies);
-    props.fallback[getApiRoute("/api/fern-docs/auth/api-key-injection")] = apiKeyInjectionConfig;
-
     return {
         props: JSON.parse(JSON.stringify(props)), // remove all undefineds
     };
diff --git a/packages/ui/docs-bundle/tsconfig.json b/packages/ui/docs-bundle/tsconfig.json
index 7978e0c5a3..c378c6d2d6 100644
--- a/packages/ui/docs-bundle/tsconfig.json
+++ b/packages/ui/docs-bundle/tsconfig.json
@@ -20,11 +20,14 @@
       "path": "../../commons/search-utils"
     },
     {
-      "path": "../fern-docs-utils"
+      "path": "../fern-docs-auth"
     },
     {
       "path": "../fern-docs-edge"
     },
+    {
+      "path": "../fern-docs-utils"
+    },
     {
       "path": "../../fdr-sdk"
     },
diff --git a/packages/ui/fern-docs-auth/.depcheckrc.json b/packages/ui/fern-docs-auth/.depcheckrc.json
new file mode 100644
index 0000000000..cf8a6564f2
--- /dev/null
+++ b/packages/ui/fern-docs-auth/.depcheckrc.json
@@ -0,0 +1 @@
+{ "ignores": ["@fern-platform/configs", "@types/node", "vite", "@types/react"], "ignore-patterns": ["dist"] }
diff --git a/packages/ui/fern-docs-auth/.prettierrc.cjs b/packages/ui/fern-docs-auth/.prettierrc.cjs
new file mode 100644
index 0000000000..39cf0d0b8c
--- /dev/null
+++ b/packages/ui/fern-docs-auth/.prettierrc.cjs
@@ -0,0 +1 @@
+module.exports = require("../../../.prettierrc.json");
diff --git a/packages/ui/fern-docs-auth/.stylelintrc.json b/packages/ui/fern-docs-auth/.stylelintrc.json
new file mode 100644
index 0000000000..0d2e3ff61d
--- /dev/null
+++ b/packages/ui/fern-docs-auth/.stylelintrc.json
@@ -0,0 +1 @@
+{ "extends": ["../../../shared/stylelintrc.shared.json"] }
diff --git a/packages/ui/fern-docs-auth/package.json b/packages/ui/fern-docs-auth/package.json
new file mode 100644
index 0000000000..884c87f3ab
--- /dev/null
+++ b/packages/ui/fern-docs-auth/package.json
@@ -0,0 +1,50 @@
+{
+  "name": "@fern-ui/fern-docs-auth",
+  "version": "0.0.0",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/fern-api/fern-platform.git",
+    "directory": "packages/ui/fern-docs-auth"
+  },
+  "private": true,
+  "files": [
+    "dist"
+  ],
+  "type": "module",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "default": "./dist/index.js"
+    }
+  },
+  "sideEffects": false,
+  "scripts": {
+    "clean": "rm -rf ./lib && tsc --build --clean",
+    "compile": "tsc --build",
+    "test": "vitest --run --passWithNoTests --globals",
+    "lint:eslint": "eslint --max-warnings 0 . --ignore-path=../../../.eslintignore",
+    "lint:eslint:fix": "pnpm lint:eslint --fix",
+    "lint:style": "stylelint 'src/**/*.scss' --allow-empty-input --max-warnings 0",
+    "lint:style:fix": "pnpm lint:style --fix",
+    "format": "prettier --write --ignore-unknown --ignore-path ../../../shared/.prettierignore \"**\"",
+    "format:check": "prettier --check --ignore-unknown --ignore-path ../../../shared/.prettierignore \"**\"",
+    "organize-imports": "organize-imports-cli tsconfig.json",
+    "depcheck": "depcheck",
+    "dev": "tsc --watch",
+    "docs:dev": "pnpm dev"
+  },
+  "dependencies": {
+    "zod": "^3.23.8"
+  },
+  "devDependencies": {
+    "@fern-platform/configs": "workspace:*",
+    "@types/node": "^18.7.18",
+    "depcheck": "^1.4.3",
+    "eslint": "^8.56.0",
+    "organize-imports-cli": "^0.10.0",
+    "prettier": "^3.3.2",
+    "stylelint": "^16.1.0",
+    "typescript": "5.4.3",
+    "vitest": "^1.5.0"
+  }
+}
diff --git a/packages/ui/app/src/auth/index.ts b/packages/ui/fern-docs-auth/src/index.ts
similarity index 100%
rename from packages/ui/app/src/auth/index.ts
rename to packages/ui/fern-docs-auth/src/index.ts
diff --git a/packages/ui/app/src/auth/injection.ts b/packages/ui/fern-docs-auth/src/injection.ts
similarity index 100%
rename from packages/ui/app/src/auth/injection.ts
rename to packages/ui/fern-docs-auth/src/injection.ts
diff --git a/packages/ui/app/src/auth/types.ts b/packages/ui/fern-docs-auth/src/types.ts
similarity index 100%
rename from packages/ui/app/src/auth/types.ts
rename to packages/ui/fern-docs-auth/src/types.ts
diff --git a/packages/ui/fern-docs-auth/tsconfig.json b/packages/ui/fern-docs-auth/tsconfig.json
new file mode 100644
index 0000000000..44763e4893
--- /dev/null
+++ b/packages/ui/fern-docs-auth/tsconfig.json
@@ -0,0 +1,8 @@
+{
+  "extends": "@fern-platform/configs/tsconfig/library.json",
+  "compilerOptions": {
+    "outDir": "./dist",
+    "rootDir": "./src"
+  },
+  "include": ["./src/**/*"]
+}
diff --git a/packages/ui/fern-docs-auth/vitest.config.ts b/packages/ui/fern-docs-auth/vitest.config.ts
new file mode 100644
index 0000000000..4f772037cd
--- /dev/null
+++ b/packages/ui/fern-docs-auth/vitest.config.ts
@@ -0,0 +1,7 @@
+import { defineConfig } from "vitest/config";
+
+export default defineConfig({
+    test: {
+        globals: true,
+    },
+});
diff --git a/packages/ui/fern-docs-edge/package.json b/packages/ui/fern-docs-edge/package.json
index e99272aaaf..932f4f7d9f 100644
--- a/packages/ui/fern-docs-edge/package.json
+++ b/packages/ui/fern-docs-edge/package.json
@@ -34,13 +34,16 @@
     "docs:dev": "pnpm dev"
   },
   "dependencies": {
+    "@fern-ui/fern-docs-auth": "workspace:*",
     "@fern-ui/fern-docs-utils": "workspace:*",
+    "@sentry/nextjs": "^8.30.0",
     "@vercel/edge-config": "^1.1.0",
-    "@sentry/nextjs": "^8.30.0"
+    "zod": "^3.23.8"
   },
   "devDependencies": {
-    "@fern-ui/ui": "workspace:*",
     "@fern-platform/configs": "workspace:*",
+    "@fern-ui/search-utils": "workspace:*",
+    "@fern-ui/ui": "workspace:*",
     "@types/node": "^18.7.18",
     "depcheck": "^1.4.3",
     "eslint": "^8.56.0",
diff --git a/packages/ui/fern-docs-edge/src/getAuthEdgeConfig.ts b/packages/ui/fern-docs-edge/src/getAuthEdgeConfig.ts
index c6e8d5bd8a..7b2a9a1f7f 100644
--- a/packages/ui/fern-docs-edge/src/getAuthEdgeConfig.ts
+++ b/packages/ui/fern-docs-edge/src/getAuthEdgeConfig.ts
@@ -1,7 +1,7 @@
-import { AuthEdgeConfig, AuthEdgeConfigSchema } from "@fern-ui/ui/auth";
+import { AuthEdgeConfigSchema, type AuthEdgeConfig } from "@fern-ui/fern-docs-auth";
+import { withoutStaging } from "@fern-ui/fern-docs-utils";
 import { captureMessage } from "@sentry/nextjs";
 import { get } from "@vercel/edge-config";
-import { withoutStaging } from "../../fern-docs-utils/src/withoutStaging";
 
 const KEY = "authentication";
 
diff --git a/packages/ui/fern-docs-edge/src/getCustomerAnalytics.ts b/packages/ui/fern-docs-edge/src/getCustomerAnalytics.ts
index 72f2a2548d..725d397838 100644
--- a/packages/ui/fern-docs-edge/src/getCustomerAnalytics.ts
+++ b/packages/ui/fern-docs-edge/src/getCustomerAnalytics.ts
@@ -1,7 +1,16 @@
-import type { CustomerAnalytics } from "@fern-ui/ui";
 import { get } from "@vercel/edge-config";
 import urlJoin from "url-join";
 
+interface GTMParams {
+    tagId: string;
+}
+interface CustomerAnalytics {
+    ga4?: {
+        measurementId: string;
+    };
+    gtm?: GTMParams;
+}
+
 /**
  * @deprecated please migrate to docs.yml
  */
diff --git a/packages/ui/fern-docs-edge/src/getFeatureFlags.ts b/packages/ui/fern-docs-edge/src/getFeatureFlags.ts
index d5626cd7e9..6beb53fbff 100644
--- a/packages/ui/fern-docs-edge/src/getFeatureFlags.ts
+++ b/packages/ui/fern-docs-edge/src/getFeatureFlags.ts
@@ -1,7 +1,6 @@
+import { isCustomDomain, isDevelopment, isFern } from "@fern-ui/fern-docs-utils";
 import type { FeatureFlags } from "@fern-ui/ui";
 import { getAll } from "@vercel/edge-config";
-import { isCustomDomain, isFern } from "../../fern-docs-utils/src/isCustomDomain";
-import { isDevelopment } from "../../fern-docs-utils/src/isDevelopment";
 
 export const runtime = "edge";
 
diff --git a/packages/ui/fern-docs-edge/src/getInkeepSettings.ts b/packages/ui/fern-docs-edge/src/getInkeepSettings.ts
new file mode 100644
index 0000000000..b40e4c03b3
--- /dev/null
+++ b/packages/ui/fern-docs-edge/src/getInkeepSettings.ts
@@ -0,0 +1,8 @@
+import { withoutStaging } from "@fern-ui/fern-docs-utils";
+import type { InkeepSharedSettings } from "@fern-ui/search-utils";
+import { get } from "@vercel/edge-config";
+
+export async function getInkeepSettings(domain: string): Promise<InkeepSharedSettings | undefined> {
+    const config = await get<Record<string, InkeepSharedSettings>>("inkeep-enabled");
+    return config?.[domain] ?? config?.[withoutStaging(domain)];
+}
diff --git a/packages/ui/fern-docs-edge/src/getLaunchDarklySettings.ts b/packages/ui/fern-docs-edge/src/getLaunchDarklySettings.ts
new file mode 100644
index 0000000000..ec3dbffca9
--- /dev/null
+++ b/packages/ui/fern-docs-edge/src/getLaunchDarklySettings.ts
@@ -0,0 +1,15 @@
+import { withoutStaging } from "@fern-ui/fern-docs-utils";
+import { get } from "@vercel/edge-config";
+import { z } from "zod";
+
+const LaunchDarklyEdgeConfigSchema = z.object({
+    // NOTE: this is client-side visible, so we should be careful about what we expose here if we add more fields
+    "client-side-id": z.string().optional(),
+});
+
+export type LaunchDarklyEdgeConfig = z.infer<typeof LaunchDarklyEdgeConfigSchema>;
+
+export async function getLaunchDarklySettings(domain: string): Promise<LaunchDarklyEdgeConfig | undefined> {
+    const config = await get<Record<string, LaunchDarklyEdgeConfig>>("launchdarkly");
+    return config?.[domain] ?? config?.[withoutStaging(domain)];
+}
diff --git a/packages/ui/fern-docs-edge/src/index.ts b/packages/ui/fern-docs-edge/src/index.ts
index 181e5f2a05..8e281180a4 100644
--- a/packages/ui/fern-docs-edge/src/index.ts
+++ b/packages/ui/fern-docs-edge/src/index.ts
@@ -1,4 +1,6 @@
 export * from "./getAuthEdgeConfig";
 export * from "./getCustomerAnalytics";
 export * from "./getFeatureFlags";
+export * from "./getInkeepSettings";
+export * from "./getLaunchDarklySettings";
 export * from "./getSeoDisabled";
diff --git a/packages/ui/fern-docs-edge/tsconfig.json b/packages/ui/fern-docs-edge/tsconfig.json
index e0544bfc82..be81ccb27b 100644
--- a/packages/ui/fern-docs-edge/tsconfig.json
+++ b/packages/ui/fern-docs-edge/tsconfig.json
@@ -6,11 +6,17 @@
   },
   "include": ["./src/**/*"],
   "references": [
+    {
+      "path": "../fern-docs-auth"
+    },
     {
       "path": "../fern-docs-utils"
     },
     {
       "path": "../app"
+    },
+    {
+      "path": "../../commons/search-utils"
     }
   ]
 }
diff --git a/pnpm-lock.yaml b/pnpm-lock.yaml
index 5b6d42c676..4921122359 100644
--- a/pnpm-lock.yaml
+++ b/pnpm-lock.yaml
@@ -726,9 +726,6 @@ importers:
       '@fern-ui/core-utils':
         specifier: workspace:*
         version: link:../core-utils
-      '@vercel/edge-config':
-        specifier: ^1.1.0
-        version: 1.1.0(@opentelemetry/api@1.9.0)(typescript@5.4.3)
     devDependencies:
       '@fern-platform/configs':
         specifier: workspace:*
@@ -1314,6 +1311,9 @@ importers:
       '@fern-platform/configs':
         specifier: workspace:*
         version: link:../../configs
+      '@fern-ui/fern-docs-auth':
+        specifier: workspace:*
+        version: link:../fern-docs-auth
       '@mdx-js/esbuild':
         specifier: ^3.0.1
         version: 3.0.1(esbuild@0.20.2)
@@ -1775,6 +1775,9 @@ importers:
       '@fern-ui/fdr-utils':
         specifier: workspace:*
         version: link:../../commons/fdr-utils
+      '@fern-ui/fern-docs-auth':
+        specifier: workspace:*
+        version: link:../fern-docs-auth
       '@fern-ui/fern-docs-edge':
         specifier: workspace:*
         version: link:../fern-docs-edge
@@ -2094,8 +2097,45 @@ importers:
         specifier: ^5.2.0
         version: 5.2.11(@types/node@20.12.12)(less@4.2.0)(sass@1.77.0)(stylus@0.62.0)(terser@5.31.0)
 
+  packages/ui/fern-docs-auth:
+    dependencies:
+      zod:
+        specifier: ^3.23.8
+        version: 3.23.8
+    devDependencies:
+      '@fern-platform/configs':
+        specifier: workspace:*
+        version: link:../../configs
+      '@types/node':
+        specifier: ^18.7.18
+        version: 18.19.33
+      depcheck:
+        specifier: ^1.4.3
+        version: 1.4.7
+      eslint:
+        specifier: ^8.56.0
+        version: 8.57.0
+      organize-imports-cli:
+        specifier: ^0.10.0
+        version: 0.10.0
+      prettier:
+        specifier: ^3.3.2
+        version: 3.3.2
+      stylelint:
+        specifier: ^16.1.0
+        version: 16.5.0(typescript@5.4.3)
+      typescript:
+        specifier: 5.4.3
+        version: 5.4.3
+      vitest:
+        specifier: ^1.5.0
+        version: 1.6.0(@edge-runtime/vm@3.2.0)(@types/node@18.19.33)(jsdom@24.0.0)(less@4.2.0)(sass@1.77.0)(stylus@0.62.0)(terser@5.31.0)
+
   packages/ui/fern-docs-edge:
     dependencies:
+      '@fern-ui/fern-docs-auth':
+        specifier: workspace:*
+        version: link:../fern-docs-auth
       '@fern-ui/fern-docs-utils':
         specifier: workspace:*
         version: link:../fern-docs-utils
@@ -2105,10 +2145,16 @@ importers:
       '@vercel/edge-config':
         specifier: ^1.1.0
         version: 1.1.0(@opentelemetry/api@1.9.0)(typescript@5.4.3)
+      zod:
+        specifier: ^3.23.8
+        version: 3.23.8
     devDependencies:
       '@fern-platform/configs':
         specifier: workspace:*
         version: link:../../configs
+      '@fern-ui/search-utils':
+        specifier: workspace:*
+        version: link:../../commons/search-utils
       '@fern-ui/ui':
         specifier: workspace:*
         version: link:../app