From 365c1e6c022cdb05931011cbe4c422455adedca4 Mon Sep 17 00:00:00 2001
From: Andrew Jiang <andrew@buildwithfern.com>
Date: Tue, 8 Oct 2024 11:22:04 -0400
Subject: [PATCH] feat: launchdarkly context + chore: refactor edge functions
 (#1604)

---
 packages/commons/fdr-utils/tsconfig.json      |   2 +-
 .../src/jsonld/components/Breadcrumb.tsx      |   1 +
 packages/commons/search-utils/package.json    |   3 +-
 .../commons/search-utils/src/SearchConfig.ts  |  29 +--
 packages/commons/search-utils/tsconfig.json   |   2 +-
 packages/fdr-sdk/tsconfig.build.json          |   2 +-
 packages/healthchecks/tsconfig.json           |   2 +-
 packages/template-resolver/tsconfig.json      |   2 +-
 packages/ui/app/package.json                  |   3 +-
 packages/ui/app/src/atoms/auth.ts             |   2 +-
 packages/ui/app/src/atoms/docs.ts             |  36 +---
 packages/ui/app/src/atoms/flags.ts            |   2 +-
 packages/ui/app/src/atoms/launchdarkly.ts     |  94 +++++++++
 packages/ui/app/src/atoms/types.ts            |  36 +---
 .../ui/app/src/components/LinkPreload.tsx     |  22 ++
 packages/ui/app/src/docs/DocsPage.tsx         |   3 +
 packages/ui/app/src/index.ts                  |   3 +-
 .../components/launchdarkly/LaunchDarkly.tsx  |  77 +------
 .../app/src/resolver/ApiDefinitionResolver.ts |   2 +-
 .../app/src/resolver/ApiEndpointResolver.ts   |   2 +-
 .../__test__/ApiDefinitionResolver.test.ts    |   2 +-
 .../src/resolver/resolveApiEndpointPage.ts    |   2 +-
 .../src/resolver/resolveApiReferencePage.ts   |   2 +-
 .../ui/app/src/resolver/resolveDocsContent.ts |   2 +-
 .../app/src/resolver/resolveMarkdownPage.ts   |   2 +-
 packages/ui/app/src/search/SearchHits.tsx     |   1 -
 .../src/services/useApiKeyInjectionConfig.ts  |   2 +-
 packages/ui/app/tsconfig.json                 |   8 +-
 packages/ui/docs-bundle/next.config.mjs       |   3 +
 packages/ui/docs-bundle/package.json          |   3 +-
 packages/ui/docs-bundle/src/middleware.ts     |   6 +-
 .../[api]/endpoint/[endpoint].ts              |   2 +-
 .../api-definition/[api]/webhook/[webhook].ts |   2 +-
 .../[api]/websocket/[websocket].ts            |   2 +-
 .../api/fern-docs/auth/api-key-injection.ts   |   6 +-
 .../src/pages/api/fern-docs/auth/callback.ts  |   6 +-
 .../pages/api/fern-docs/auth/jwt/callback.ts  |   4 +-
 .../src/pages/api/fern-docs/auth/logout.ts    |   2 +-
 .../src/pages/api/fern-docs/changelog.ts      |   2 +-
 .../src/pages/api/fern-docs/feature-flags.ts  | 166 +--------------
 .../fern-docs/integrations/launchdarkly.ts    | 110 ++++++++--
 .../integrations/launchdarkly/identify.ts     |  16 ++
 .../pages/api/fern-docs/oauth/ory/callback.ts |   6 +-
 .../api/fern-docs/oauth/webflow/callback.ts   |   2 +-
 .../src/pages/api/fern-docs/preview.ts        |   2 +-
 .../src/pages/api/fern-docs/resolve-api.ts    |   2 +-
 .../src/pages/api/fern-docs/robots.txt.ts     |   2 +-
 .../src/pages/api/fern-docs/search.ts         |   5 +-
 .../src/pages/api/fern-docs/sitemap.xml.ts    |   2 +-
 .../src/server/ApiDefinitionLoader.ts         |   4 +-
 .../ui/docs-bundle/src/server/DocsLoader.ts   |   4 +-
 .../src/server/LoadDocsPerformanceTracker.ts  |   2 +-
 .../ui/docs-bundle/src/server/auth/FernJWT.ts |   2 +-
 .../src/server/auth/OAuth2Client.ts           |   6 +-
 .../src/server/auth/checkViewerAllowed.ts     |   4 +-
 .../server/auth/getApiKeyInjectionConfig.ts   |   4 +-
 .../ui/docs-bundle/src/server/authProps.ts    |   4 +-
 .../src/server/getDocsPageProps.ts            |   8 +-
 .../src/server/getDynamicDocsPageProps.ts     |   4 +-
 .../ui/docs-bundle/src/server/loadWithUrl.ts  |   2 +-
 .../src/server/withBasicTokenPublic.ts        |   2 +-
 .../src/server/withInitialProps.ts            |  25 +--
 .../ui/docs-bundle/src/server/withSSGProps.ts |   2 +-
 .../docs-bundle/src/server/xfernhost/edge.ts  |   2 +-
 .../docs-bundle/src/server/xfernhost/node.ts  |   2 +-
 packages/ui/docs-bundle/tsconfig.json         |  11 +-
 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 +
 .../ui/fern-docs-edge-config/.depcheckrc.json |   1 +
 .../ui/fern-docs-edge-config/.prettierrc.cjs  |   1 +
 .../fern-docs-edge-config/.stylelintrc.json   |   1 +
 .../ui/fern-docs-edge-config/package.json     |  55 +++++
 .../src}/getAuthEdgeConfig.ts                 |   4 +-
 .../src}/getCustomerAnalytics.ts              |  11 +-
 .../src/getFeatureFlags.ts                    | 165 +++++++++++++++
 .../src/getInkeepSettings.ts                  |   8 +
 .../src/getLaunchDarklySettings.ts            |  15 ++
 .../src/getSeoDisabled.ts}                    |   2 +-
 .../ui/fern-docs-edge-config/src/index.ts     |   6 +
 .../ui/fern-docs-edge-config/tsconfig.json    |  19 ++
 .../ui/fern-docs-edge-config/vitest.config.ts |   7 +
 .../src}/constants.ts                         |   1 +
 packages/ui/fern-docs-utils/src/flags.ts      |  65 ++++++
 packages/ui/fern-docs-utils/src/index.ts      |   5 +
 .../src}/isCustomDomain.ts                    |   0
 .../src}/isDevelopment.ts                     |   0
 .../src}/withoutStaging.ts                    |   0
 packages/ui/fern-docs-utils/tsconfig.json     |   2 +-
 .../src/utils/getDocsPageProps.ts             |   4 +-
 .../ui/local-preview-bundle/tsconfig.json     |   2 +-
 pnpm-lock.yaml                                | 195 +++++++++---------
 98 files changed, 887 insertions(+), 535 deletions(-)
 create mode 100644 packages/ui/app/src/atoms/launchdarkly.ts
 create mode 100644 packages/ui/app/src/components/LinkPreload.tsx
 create mode 100644 packages/ui/docs-bundle/src/pages/api/fern-docs/integrations/launchdarkly/identify.ts
 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-config/.depcheckrc.json
 create mode 100644 packages/ui/fern-docs-edge-config/.prettierrc.cjs
 create mode 100644 packages/ui/fern-docs-edge-config/.stylelintrc.json
 create mode 100644 packages/ui/fern-docs-edge-config/package.json
 rename packages/ui/{docs-bundle/src/server/auth => fern-docs-edge-config/src}/getAuthEdgeConfig.ts (86%)
 rename packages/ui/{docs-bundle/src/server => fern-docs-edge-config/src}/getCustomerAnalytics.ts (71%)
 create mode 100644 packages/ui/fern-docs-edge-config/src/getFeatureFlags.ts
 create mode 100644 packages/ui/fern-docs-edge-config/src/getInkeepSettings.ts
 create mode 100644 packages/ui/fern-docs-edge-config/src/getLaunchDarklySettings.ts
 rename packages/ui/{docs-bundle/src/server/disabledSeo.ts => fern-docs-edge-config/src/getSeoDisabled.ts} (82%)
 create mode 100644 packages/ui/fern-docs-edge-config/src/index.ts
 create mode 100644 packages/ui/fern-docs-edge-config/tsconfig.json
 create mode 100644 packages/ui/fern-docs-edge-config/vitest.config.ts
 rename packages/ui/{docs-bundle/src/server => fern-docs-utils/src}/constants.ts (94%)
 create mode 100644 packages/ui/fern-docs-utils/src/flags.ts
 rename packages/ui/{docs-bundle/src/server => fern-docs-utils/src}/isCustomDomain.ts (100%)
 rename packages/ui/{docs-bundle/src/server => fern-docs-utils/src}/isDevelopment.ts (100%)
 rename packages/ui/{docs-bundle/src/server => fern-docs-utils/src}/withoutStaging.ts (100%)

diff --git a/packages/commons/fdr-utils/tsconfig.json b/packages/commons/fdr-utils/tsconfig.json
index 54cc57ae2a..dc07764da5 100644
--- a/packages/commons/fdr-utils/tsconfig.json
+++ b/packages/commons/fdr-utils/tsconfig.json
@@ -7,7 +7,7 @@
   "include": ["./src/**/*"],
   "references": [
     {
-      "path": "../../fdr-sdk"
+      "path": "../../fdr-sdk/tsconfig.build.json"
     }
   ]
 }
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/commons/search-utils/tsconfig.json b/packages/commons/search-utils/tsconfig.json
index a865844d83..1d1791590f 100644
--- a/packages/commons/search-utils/tsconfig.json
+++ b/packages/commons/search-utils/tsconfig.json
@@ -8,7 +8,7 @@
   "include": ["./src/**/*"],
   "references": [
     {
-      "path": "../../fdr-sdk"
+      "path": "../../fdr-sdk/tsconfig.build.json"
     },
     {
       "path": "../core-utils"
diff --git a/packages/fdr-sdk/tsconfig.build.json b/packages/fdr-sdk/tsconfig.build.json
index c1a3106bd1..c700db4a70 100644
--- a/packages/fdr-sdk/tsconfig.build.json
+++ b/packages/fdr-sdk/tsconfig.build.json
@@ -1,5 +1,5 @@
 {
   "$schema": "https://json.schemastore.org/tsconfig",
   "extends": "./tsconfig.json",
-  "exclude": ["**/__test__/**"]
+  "exclude": ["node_modules", "**/__test__/**", "**/*.json"]
 }
diff --git a/packages/healthchecks/tsconfig.json b/packages/healthchecks/tsconfig.json
index 616842c23b..2cab52d4c1 100644
--- a/packages/healthchecks/tsconfig.json
+++ b/packages/healthchecks/tsconfig.json
@@ -11,7 +11,7 @@
       "path": "../commons/core-utils"
     },
     {
-      "path": "../fdr-sdk"
+      "path": "../../fdr-sdk/tsconfig.build.json"
     }
   ]
 }
diff --git a/packages/template-resolver/tsconfig.json b/packages/template-resolver/tsconfig.json
index abb355044b..4f8e286f60 100644
--- a/packages/template-resolver/tsconfig.json
+++ b/packages/template-resolver/tsconfig.json
@@ -7,7 +7,7 @@
   "include": ["./src/**/*"],
   "references": [
     {
-      "path": "../fdr-sdk"
+      "path": "../fdr-sdk/tsconfig.build.json"
     }
   ]
 }
diff --git a/packages/ui/app/package.json b/packages/ui/app/package.json
index f3cb327d05..db19bc0e5b 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": [
@@ -40,6 +39,7 @@
     "@fern-ui/components": "workspace:*",
     "@fern-ui/core-utils": "workspace:*",
     "@fern-ui/fdr-utils": "workspace:*",
+    "@fern-ui/fern-docs-utils": "workspace:*",
     "@fern-ui/loadable": "workspace:*",
     "@fern-ui/next-seo": "workspace:*",
     "@fern-ui/react-commons": "workspace:*",
@@ -120,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/docs.ts b/packages/ui/app/src/atoms/docs.ts
index 6f1155476c..b035b93832 100644
--- a/packages/ui/app/src/atoms/docs.ts
+++ b/packages/ui/app/src/atoms/docs.ts
@@ -1,41 +1,9 @@
 import { DocsV1Read } from "@fern-api/fdr-sdk";
 import * as FernNavigation from "@fern-api/fdr-sdk/navigation";
+import { DEFAULT_FEATURE_FLAGS } from "@fern-ui/fern-docs-utils";
 import { atomWithReducer, useHydrateAtoms } from "jotai/utils";
 import type { PropsWithChildren, ReactNode } from "react";
-import type { DocsProps, FeatureFlags } from "./types";
-
-export const DEFAULT_FEATURE_FLAGS: FeatureFlags = {
-    isApiPlaygroundEnabled: false,
-    isApiScrollingDisabled: false,
-    isWhitelabeled: false,
-    isSeoDisabled: false,
-    isTocDefaultEnabled: false,
-    isSnippetTemplatesEnabled: false,
-    isHttpSnippetsEnabled: false,
-    isInlineFeedbackEnabled: false,
-    isDarkCodeEnabled: false,
-    proxyShouldUseAppBuildwithfernCom: false,
-    isImageZoomDisabled: false,
-    useJavaScriptAsTypeScript: false,
-    alwaysEnableJavaScriptFetch: false,
-    scrollInContainerEnabled: false,
-    useMdxBundler: false,
-    isBatchStreamToggleDisabled: false,
-    isAuthEnabledInDocs: false,
-    isAiChatbotEnabledInPreview: false,
-    isAudioFileDownloadSpanSummary: false,
-    isDocsLogoTextEnabled: false,
-    isAudioExampleInternal: false,
-    usesApplicationJsonInFormDataValue: false,
-    isBinaryOctetStreamAudioPlayer: false,
-    hasVoiceIdPlaygroundForm: false,
-    isCohereTheme: false,
-    isFileForgeHackEnabled: false,
-    is404PageHidden: false,
-    isNewSearchExperienceEnabled: false,
-    // TODO: remove this after pinecone demo, this is a temporary flag
-    grpcEndpoints: [],
-};
+import type { DocsProps } from "./types";
 
 export const EMPTY_ANALYTICS_CONFIG: DocsV1Read.AnalyticsConfig = {
     segment: undefined,
diff --git a/packages/ui/app/src/atoms/flags.ts b/packages/ui/app/src/atoms/flags.ts
index 07d27c2eb0..a5e1d17924 100644
--- a/packages/ui/app/src/atoms/flags.ts
+++ b/packages/ui/app/src/atoms/flags.ts
@@ -1,8 +1,8 @@
+import type { FeatureFlags } from "@fern-ui/fern-docs-utils";
 import { useAtomValue } from "jotai";
 import { selectAtom } from "jotai/utils";
 import { isEqual } from "lodash-es";
 import { DOCS_ATOM } from "./docs";
-import { FeatureFlags } from "./types";
 
 export const FEATURE_FLAGS_ATOM = selectAtom(DOCS_ATOM, (docs) => docs.featureFlags, isEqual);
 FEATURE_FLAGS_ATOM.debugLabel = "FEATURE_FLAGS_ATOM";
diff --git a/packages/ui/app/src/atoms/launchdarkly.ts b/packages/ui/app/src/atoms/launchdarkly.ts
new file mode 100644
index 0000000000..34233e4852
--- /dev/null
+++ b/packages/ui/app/src/atoms/launchdarkly.ts
@@ -0,0 +1,94 @@
+import { atom, useAtomValue, useSetAtom } from "jotai";
+import * as LDClient from "launchdarkly-js-client-sdk";
+import { useCallback, useEffect, useState } from "react";
+import useSWR from "swr";
+import { useApiRoute } from "../hooks/useApiRoute";
+
+// NOTE do not export this file in any index.ts file so that it can be properly tree-shaken
+// otherwise we risk importing launchdarkly-js-client-sdk in all of our bundles
+
+// TODO: consolidate the types with the edge-config package
+interface LaunchDarklyInfo {
+    clientSideId: string;
+    kind: "multi";
+    user:
+        | { anonymous: true }
+        | {
+              key: string;
+              email?: string;
+              name?: string;
+          };
+    device: {
+        key: string;
+        [key: string]: unknown;
+    };
+}
+
+// this is a singleton atom that initializes the LaunchDarkly client-side SDK
+const LD_CLIENT_ATOM = atom<LDClient.LDClient | undefined>(undefined);
+
+const SET_LAUNCH_DARKLY_INFO_ATOM = atom(undefined, async (get, set, info: LaunchDarklyInfo) => {
+    const client = get(LD_CLIENT_ATOM);
+    const { clientSideId, ...context } = info;
+
+    if (client) {
+        await client.identify(context);
+        return;
+    } else {
+        const client = LDClient.initialize(clientSideId, context);
+        await client.waitForInitialization();
+        set(LD_CLIENT_ATOM, client);
+    }
+});
+
+// TODO: support non-boolean flags
+export const useLaunchDarklyFlag = (flag: string): boolean => {
+    useInitLaunchDarklyClient();
+
+    const client = useAtomValue(LD_CLIENT_ATOM);
+
+    // TODO: bootstrap the flag value from the server, and/or local storage
+    const getFlagEnabled = useCallback(() => {
+        if (!client) {
+            return false;
+        }
+        // force the flag to be a boolean:
+        return !!client.variation(flag, false);
+    }, [client, flag]);
+
+    const [enabled, setEnabled] = useState(getFlagEnabled);
+
+    // listen for changes to the flag
+    useEffect(() => {
+        setEnabled(getFlagEnabled());
+
+        if (!client) {
+            return;
+        }
+
+        const listener = () => {
+            setEnabled(getFlagEnabled());
+        };
+
+        client.on("ready", listener);
+        client.on("change", listener);
+
+        return () => {
+            client.off("ready", listener);
+            client.off("change", listener);
+        };
+    }, [client, flag, getFlagEnabled]);
+
+    return enabled;
+};
+
+// since useSWR is cached globally, we can use this hook in multiple components without worrying about multiple requests
+function useInitLaunchDarklyClient() {
+    const route = useApiRoute("/api/fern-docs/integrations/launchdarkly");
+    const setInfo = useSetAtom(SET_LAUNCH_DARKLY_INFO_ATOM);
+    useSWR(route, (key): Promise<LaunchDarklyInfo> => fetch(key).then((res) => res.json()), {
+        onSuccess(data) {
+            void setInfo(data);
+        },
+    });
+}
diff --git a/packages/ui/app/src/atoms/types.ts b/packages/ui/app/src/atoms/types.ts
index bda1d88d1d..9cae6e4c92 100644
--- a/packages/ui/app/src/atoms/types.ts
+++ b/packages/ui/app/src/atoms/types.ts
@@ -2,45 +2,13 @@ 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 type { FeatureFlags } from "@fern-ui/fern-docs-utils";
 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";
 
-export interface FeatureFlags {
-    isApiPlaygroundEnabled: boolean;
-    isApiScrollingDisabled: boolean;
-    isWhitelabeled: boolean;
-    isSeoDisabled: boolean;
-    isTocDefaultEnabled: boolean;
-    isSnippetTemplatesEnabled: boolean;
-    isHttpSnippetsEnabled: boolean;
-    isInlineFeedbackEnabled: boolean;
-    isDarkCodeEnabled: boolean;
-    proxyShouldUseAppBuildwithfernCom: boolean;
-    isImageZoomDisabled: boolean;
-    useJavaScriptAsTypeScript: boolean;
-    alwaysEnableJavaScriptFetch: boolean;
-    scrollInContainerEnabled: boolean;
-    useMdxBundler: boolean;
-    isBatchStreamToggleDisabled: boolean;
-    isAuthEnabledInDocs: boolean;
-    isAiChatbotEnabledInPreview: boolean;
-    isAudioFileDownloadSpanSummary: boolean;
-    isDocsLogoTextEnabled: boolean;
-    isAudioExampleInternal: boolean;
-    usesApplicationJsonInFormDataValue: boolean;
-    isBinaryOctetStreamAudioPlayer: boolean;
-    hasVoiceIdPlaygroundForm: boolean;
-    isCohereTheme: boolean;
-    isFileForgeHackEnabled: boolean;
-    is404PageHidden: boolean;
-    isNewSearchExperienceEnabled: boolean;
-    // TODO: remove this after pinecone demo, this is a temporary flag
-    grpcEndpoints: readonly string[];
-}
-
 export interface NavigationProps {
     currentTabIndex: number | undefined;
     tabs: SidebarTab[];
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/index.ts b/packages/ui/app/src/index.ts
index 1026449615..6c43ad07e8 100644
--- a/packages/ui/app/src/index.ts
+++ b/packages/ui/app/src/index.ts
@@ -1,6 +1,5 @@
 export { type CustomerAnalytics } from "./analytics/types";
-export { DEFAULT_FEATURE_FLAGS } from "./atoms";
-export type { DocsProps, FeatureFlags } from "./atoms";
+export type { DocsProps } from "./atoms";
 export * from "./docs/DocsPage";
 export * from "./docs/NextApp";
 export { getApiRouteSupplier } from "./hooks/useApiRoute";
diff --git a/packages/ui/app/src/mdx/components/launchdarkly/LaunchDarkly.tsx b/packages/ui/app/src/mdx/components/launchdarkly/LaunchDarkly.tsx
index 72f4418156..a23a02fd48 100644
--- a/packages/ui/app/src/mdx/components/launchdarkly/LaunchDarkly.tsx
+++ b/packages/ui/app/src/mdx/components/launchdarkly/LaunchDarkly.tsx
@@ -1,70 +1,6 @@
-import { atom, useAtomValue } from "jotai";
-import { loadable } 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"];
-}
-
-// this is a singleton atom that initializes the LaunchDarkly client-side SDK
-const ldClientAtom = 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) {
-        return undefined;
-    }
-
-    const client = LDClient.initialize(clientSideId, {
-        kind: "user",
-        anonymous: true,
-    });
-
-    await client.waitForInitialization();
-
-    return client;
-});
-
-const useLaunchDarklyFlag = (flag: string): boolean => {
-    const loadableClient = useAtomValue(loadable(ldClientAtom));
-    const client = loadableClient.state === "hasData" ? loadableClient.data : undefined;
-    if (loadableClient.state === "hasError") {
-        // TODO: sentry
-    }
-
-    const getFlagEnabled = useCallback(() => {
-        return client?.variation(flag, false) ?? false;
-    }, [client, flag]);
-
-    const [enabled, setEnabled] = useState(getFlagEnabled);
-
-    useEffect(() => {
-        setEnabled(getFlagEnabled());
-
-        if (!client) {
-            return;
-        }
-
-        const listener = () => {
-            setEnabled(getFlagEnabled());
-        };
-
-        client.on("ready", listener);
-        client.on("change", listener);
-
-        return () => {
-            client.off("ready", listener);
-            client.off("change", listener);
-        };
-    }, [client, flag, getFlagEnabled]);
-
-    return enabled;
-};
+import { PropsWithChildren, ReactNode } from "react";
+import { useLaunchDarklyFlag } from "../../../atoms/launchdarkly";
+import { LinkPreloadApiRoute } from "../../../components/LinkPreload";
 
 export interface LaunchDarklyProps {
     flag: string;
@@ -77,5 +13,10 @@ export function LaunchDarkly({ flag, children }: PropsWithChildren<LaunchDarklyP
         return null;
     }
 
-    return children;
+    return (
+        <>
+            <LinkPreloadApiRoute href="/api/fern-docs/integrations/launchdarkly" />
+            {children}
+        </>
+    );
 }
diff --git a/packages/ui/app/src/resolver/ApiDefinitionResolver.ts b/packages/ui/app/src/resolver/ApiDefinitionResolver.ts
index 3c4aa53a11..639c910edb 100644
--- a/packages/ui/app/src/resolver/ApiDefinitionResolver.ts
+++ b/packages/ui/app/src/resolver/ApiDefinitionResolver.ts
@@ -1,8 +1,8 @@
 import type { DocsV1Read } from "@fern-api/fdr-sdk";
 import type * as FernNavigation from "@fern-api/fdr-sdk/navigation";
 import { isNonNullish, visitDiscriminatedUnion } from "@fern-ui/core-utils";
+import type { FeatureFlags } from "@fern-ui/fern-docs-utils";
 import { FernRegistry } from "../../../../fdr-sdk/src/client/generated";
-import type { FeatureFlags } from "../atoms";
 import { type MDX_SERIALIZER } from "../mdx/bundler";
 import type { FernSerializeMdxOptions } from "../mdx/types";
 import { ApiEndpointResolver } from "./ApiEndpointResolver";
diff --git a/packages/ui/app/src/resolver/ApiEndpointResolver.ts b/packages/ui/app/src/resolver/ApiEndpointResolver.ts
index dc7ee123e9..fb594b19f5 100644
--- a/packages/ui/app/src/resolver/ApiEndpointResolver.ts
+++ b/packages/ui/app/src/resolver/ApiEndpointResolver.ts
@@ -1,9 +1,9 @@
 import { APIV1Read } from "@fern-api/fdr-sdk/client/types";
 import * as FernNavigation from "@fern-api/fdr-sdk/navigation";
 import { visitDiscriminatedUnion } from "@fern-ui/core-utils";
+import type { FeatureFlags } from "@fern-ui/fern-docs-utils";
 import { compact, mapValues } from "lodash-es";
 import { sortKeysByShape } from "../api-reference/examples/sortKeysByShape";
-import type { FeatureFlags } from "../atoms";
 import { MDX_SERIALIZER } from "../mdx/bundler";
 import type { FernSerializeMdxOptions } from "../mdx/types";
 import { ApiTypeResolver } from "./ApiTypeResolver";
diff --git a/packages/ui/app/src/resolver/__test__/ApiDefinitionResolver.test.ts b/packages/ui/app/src/resolver/__test__/ApiDefinitionResolver.test.ts
index 2475916998..5d7eb4f8c4 100644
--- a/packages/ui/app/src/resolver/__test__/ApiDefinitionResolver.test.ts
+++ b/packages/ui/app/src/resolver/__test__/ApiDefinitionResolver.test.ts
@@ -1,8 +1,8 @@
 import { APIV1Read } from "@fern-api/fdr-sdk/client/types";
 import * as FernNavigation from "@fern-api/fdr-sdk/navigation";
+import { DEFAULT_FEATURE_FLAGS } from "@fern-ui/fern-docs-utils";
 import fs from "fs";
 import path from "path";
-import { DEFAULT_FEATURE_FLAGS } from "../../atoms";
 import { serializeMdx } from "../../mdx/bundlers/next-mdx-remote";
 import { ApiDefinitionResolver } from "../ApiDefinitionResolver";
 import { ResolvedEndpointDefinition } from "../types";
diff --git a/packages/ui/app/src/resolver/resolveApiEndpointPage.ts b/packages/ui/app/src/resolver/resolveApiEndpointPage.ts
index 47c9fee8e7..0c5f2530e7 100644
--- a/packages/ui/app/src/resolver/resolveApiEndpointPage.ts
+++ b/packages/ui/app/src/resolver/resolveApiEndpointPage.ts
@@ -1,7 +1,7 @@
 import type { APIV1Read } from "@fern-api/fdr-sdk";
 import * as FernNavigation from "@fern-api/fdr-sdk/navigation";
 import { visitDiscriminatedUnion } from "@fern-ui/core-utils";
-import type { FeatureFlags } from "../atoms";
+import type { FeatureFlags } from "@fern-ui/fern-docs-utils";
 import type { MDX_SERIALIZER } from "../mdx/bundler";
 import type { FernSerializeMdxOptions } from "../mdx/types";
 import { ApiEndpointResolver } from "./ApiEndpointResolver";
diff --git a/packages/ui/app/src/resolver/resolveApiReferencePage.ts b/packages/ui/app/src/resolver/resolveApiReferencePage.ts
index a8009474d9..6b1fdea0fd 100644
--- a/packages/ui/app/src/resolver/resolveApiReferencePage.ts
+++ b/packages/ui/app/src/resolver/resolveApiReferencePage.ts
@@ -1,6 +1,6 @@
 import type { APIV1Read, DocsV1Read } from "@fern-api/fdr-sdk";
 import * as FernNavigation from "@fern-api/fdr-sdk/navigation";
-import { FeatureFlags } from "../atoms";
+import type { FeatureFlags } from "@fern-ui/fern-docs-utils";
 import type { MDX_SERIALIZER } from "../mdx/bundler";
 import type { FernSerializeMdxOptions } from "../mdx/types";
 import { ApiDefinitionResolver } from "./ApiDefinitionResolver";
diff --git a/packages/ui/app/src/resolver/resolveDocsContent.ts b/packages/ui/app/src/resolver/resolveDocsContent.ts
index d55f4135a4..7dfa871887 100644
--- a/packages/ui/app/src/resolver/resolveDocsContent.ts
+++ b/packages/ui/app/src/resolver/resolveDocsContent.ts
@@ -1,6 +1,6 @@
 import type { APIV1Read, DocsV1Read } from "@fern-api/fdr-sdk/client/types";
 import * as FernNavigation from "@fern-api/fdr-sdk/navigation";
-import type { FeatureFlags } from "../atoms";
+import type { FeatureFlags } from "@fern-ui/fern-docs-utils";
 import type { MDX_SERIALIZER } from "../mdx/bundler";
 import type { FernSerializeMdxOptions } from "../mdx/types";
 import type { DocsContent } from "./DocsContent";
diff --git a/packages/ui/app/src/resolver/resolveMarkdownPage.ts b/packages/ui/app/src/resolver/resolveMarkdownPage.ts
index 795c1e197a..6fd4e2cf42 100644
--- a/packages/ui/app/src/resolver/resolveMarkdownPage.ts
+++ b/packages/ui/app/src/resolver/resolveMarkdownPage.ts
@@ -1,7 +1,7 @@
 import type { APIV1Read, DocsV1Read } from "@fern-api/fdr-sdk";
 import type * as FernDocs from "@fern-api/fdr-sdk/docs";
 import * as FernNavigation from "@fern-api/fdr-sdk/navigation";
-import type { FeatureFlags } from "../atoms";
+import type { FeatureFlags } from "@fern-ui/fern-docs-utils";
 import type { MDX_SERIALIZER } from "../mdx/bundler";
 import type { FernSerializeMdxOptions } from "../mdx/types";
 import { ApiDefinitionResolver } from "./ApiDefinitionResolver";
diff --git a/packages/ui/app/src/search/SearchHits.tsx b/packages/ui/app/src/search/SearchHits.tsx
index 4121270b56..09d6c4b1d7 100644
--- a/packages/ui/app/src/search/SearchHits.tsx
+++ b/packages/ui/app/src/search/SearchHits.tsx
@@ -2,7 +2,6 @@ import * as FernNavigation from "@fern-api/fdr-sdk/navigation";
 import { FernButton, FernScrollArea } from "@fern-ui/components";
 import { useKeyboardPress } from "@fern-ui/react-commons";
 import { getSlugForSearchRecord, type SearchRecord } from "@fern-ui/search-utils";
-
 import { Minus, Xmark } from "iconoir-react";
 import { useSetAtom } from "jotai";
 import { useRouter } from "next/router";
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..7bf64b8f5f 100644
--- a/packages/ui/app/tsconfig.json
+++ b/packages/ui/app/tsconfig.json
@@ -27,7 +27,7 @@
       "path": "../../commons/search-utils"
     },
     {
-      "path": "../../fdr-sdk"
+      "path": "../../fdr-sdk/tsconfig.build.json"
     },
     {
       "path": "../../template-resolver"
@@ -35,6 +35,12 @@
     {
       "path": "../chatbot"
     },
+    {
+      "path": "../fern-docs-auth"
+    },
+    {
+      "path": "../fern-docs-utils"
+    },
     {
       "path": "../components"
     }
diff --git a/packages/ui/docs-bundle/next.config.mjs b/packages/ui/docs-bundle/next.config.mjs
index d24deb34fb..b2dee496f4 100644
--- a/packages/ui/docs-bundle/next.config.mjs
+++ b/packages/ui/docs-bundle/next.config.mjs
@@ -50,6 +50,9 @@ const nextConfig = {
         "@fern-ui/components",
         "@fern-ui/core-utils",
         "@fern-ui/fdr-utils",
+        "@fern-ui/fern-docs-auth",
+        "@fern-ui/fern-docs-edge-config",
+        "@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 8c5652af24..f03fe33878 100644
--- a/packages/ui/docs-bundle/package.json
+++ b/packages/ui/docs-bundle/package.json
@@ -35,11 +35,12 @@
     "@fern-ui/components": "workspace:*",
     "@fern-ui/core-utils": "workspace:*",
     "@fern-ui/fdr-utils": "workspace:*",
+    "@fern-ui/fern-docs-auth": "workspace:*",
+    "@fern-ui/fern-docs-edge-config": "workspace:*",
     "@fern-ui/fern-docs-utils": "workspace:*",
     "@fern-ui/search-utils": "workspace:*",
     "@fern-ui/ui": "workspace:*",
     "@vercel/analytics": "^1.3.1",
-    "@vercel/edge-config": "^1.1.0",
     "@vercel/kv": "^2.0.0",
     "@workos-inc/node": "^6.1.0",
     "algoliasearch": "^4.24.0",
diff --git a/packages/ui/docs-bundle/src/middleware.ts b/packages/ui/docs-bundle/src/middleware.ts
index 5059521149..ea1a01e00f 100644
--- a/packages/ui/docs-bundle/src/middleware.ts
+++ b/packages/ui/docs-bundle/src/middleware.ts
@@ -2,13 +2,13 @@ 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/ui/auth";
+import type { FernUser } from "@fern-ui/fern-docs-auth";
+import { getAuthEdgeConfig } from "@fern-ui/fern-docs-edge-config";
+import { COOKIE_FERN_TOKEN, HEADER_X_FERN_HOST } from "@fern-ui/fern-docs-utils";
 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";
 import { verifyFernJWTConfig } from "./server/auth/FernJWT";
-import { getAuthEdgeConfig } from "./server/auth/getAuthEdgeConfig";
-import { COOKIE_FERN_TOKEN, HEADER_X_FERN_HOST } from "./server/constants";
 import { withBasicTokenPublic } from "./server/withBasicTokenPublic";
 
 const API_FERN_DOCS_PATTERN = /^(?!\/api\/fern-docs\/).*(\/api\/fern-docs\/)/;
diff --git a/packages/ui/docs-bundle/src/pages/api/fern-docs/api-definition/[api]/endpoint/[endpoint].ts b/packages/ui/docs-bundle/src/pages/api/fern-docs/api-definition/[api]/endpoint/[endpoint].ts
index b5906aba7d..ed7d152a20 100644
--- a/packages/ui/docs-bundle/src/pages/api/fern-docs/api-definition/[api]/endpoint/[endpoint].ts
+++ b/packages/ui/docs-bundle/src/pages/api/fern-docs/api-definition/[api]/endpoint/[endpoint].ts
@@ -1,8 +1,8 @@
 import { ApiDefinitionLoader } from "@/server/ApiDefinitionLoader";
 import { getXFernHostNode } from "@/server/xfernhost/node";
 import * as ApiDefinition from "@fern-api/fdr-sdk/api-definition";
+import { getFeatureFlags } from "@fern-ui/fern-docs-edge-config";
 import { NextApiHandler, NextApiResponse } from "next";
-import { getFeatureFlags } from "../../../feature-flags";
 
 const resolveApiHandler: NextApiHandler = async (req, res: NextApiResponse<ApiDefinition.ApiDefinition>) => {
     const xFernHost = getXFernHostNode(req);
diff --git a/packages/ui/docs-bundle/src/pages/api/fern-docs/api-definition/[api]/webhook/[webhook].ts b/packages/ui/docs-bundle/src/pages/api/fern-docs/api-definition/[api]/webhook/[webhook].ts
index a66f4cd504..e2e0d1af5b 100644
--- a/packages/ui/docs-bundle/src/pages/api/fern-docs/api-definition/[api]/webhook/[webhook].ts
+++ b/packages/ui/docs-bundle/src/pages/api/fern-docs/api-definition/[api]/webhook/[webhook].ts
@@ -1,8 +1,8 @@
 import { ApiDefinitionLoader } from "@/server/ApiDefinitionLoader";
 import { getXFernHostNode } from "@/server/xfernhost/node";
 import * as ApiDefinition from "@fern-api/fdr-sdk/api-definition";
+import { getFeatureFlags } from "@fern-ui/fern-docs-edge-config";
 import { NextApiHandler, NextApiResponse } from "next";
-import { getFeatureFlags } from "../../../feature-flags";
 
 const resolveApiHandler: NextApiHandler = async (req, res: NextApiResponse<ApiDefinition.ApiDefinition>) => {
     const xFernHost = getXFernHostNode(req);
diff --git a/packages/ui/docs-bundle/src/pages/api/fern-docs/api-definition/[api]/websocket/[websocket].ts b/packages/ui/docs-bundle/src/pages/api/fern-docs/api-definition/[api]/websocket/[websocket].ts
index 7f8acbaa55..c4dd9cea4b 100644
--- a/packages/ui/docs-bundle/src/pages/api/fern-docs/api-definition/[api]/websocket/[websocket].ts
+++ b/packages/ui/docs-bundle/src/pages/api/fern-docs/api-definition/[api]/websocket/[websocket].ts
@@ -1,8 +1,8 @@
 import { ApiDefinitionLoader } from "@/server/ApiDefinitionLoader";
 import { getXFernHostNode } from "@/server/xfernhost/node";
 import * as ApiDefinition from "@fern-api/fdr-sdk/api-definition";
+import { getFeatureFlags } from "@fern-ui/fern-docs-edge-config";
 import { NextApiHandler, NextApiResponse } from "next";
-import { getFeatureFlags } from "../../../feature-flags";
 
 const resolveApiHandler: NextApiHandler = async (req, res: NextApiResponse<ApiDefinition.ApiDefinition>) => {
     const xFernHost = getXFernHostNode(req);
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 68dc42e37a..2657db009b 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
@@ -1,10 +1,10 @@
 import { OAuth2Client } from "@/server/auth/OAuth2Client";
 import { getAPIKeyInjectionConfig } from "@/server/auth/getApiKeyInjectionConfig";
-import { getAuthEdgeConfig } from "@/server/auth/getAuthEdgeConfig";
 import { withSecureCookie } from "@/server/auth/withSecure";
-import { COOKIE_FERN_TOKEN } from "@/server/constants";
 import { getXFernHostEdge } from "@/server/xfernhost/edge";
-import { APIKeyInjectionConfig, OryAccessTokenSchema } from "@fern-ui/ui/auth";
+import { APIKeyInjectionConfig, OryAccessTokenSchema } from "@fern-ui/fern-docs-auth";
+import { getAuthEdgeConfig } from "@fern-ui/fern-docs-edge-config";
+import { COOKIE_FERN_TOKEN } from "@fern-ui/fern-docs-utils";
 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 2f1f6a14e6..e5eda1a1a0 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
@@ -1,10 +1,10 @@
 import { signFernJWT } from "@/server/auth/FernJWT";
-import { getAuthEdgeConfig } from "@/server/auth/getAuthEdgeConfig";
 import { withSecureCookie } from "@/server/auth/withSecure";
-import { COOKIE_FERN_TOKEN } from "@/server/constants";
 import { getWorkOS, getWorkOSClientId } from "@/server/workos";
 import { getXFernHostEdge } from "@/server/xfernhost/edge";
-import { FernUser } from "@fern-ui/ui/auth";
+import { FernUser } from "@fern-ui/fern-docs-auth";
+import { getAuthEdgeConfig } from "@fern-ui/fern-docs-edge-config";
+import { COOKIE_FERN_TOKEN } from "@fern-ui/fern-docs-utils";
 import { NextRequest, NextResponse } from "next/server";
 
 export const runtime = "edge";
diff --git a/packages/ui/docs-bundle/src/pages/api/fern-docs/auth/jwt/callback.ts b/packages/ui/docs-bundle/src/pages/api/fern-docs/auth/jwt/callback.ts
index 108841a2cd..9e9205b7ca 100644
--- a/packages/ui/docs-bundle/src/pages/api/fern-docs/auth/jwt/callback.ts
+++ b/packages/ui/docs-bundle/src/pages/api/fern-docs/auth/jwt/callback.ts
@@ -1,8 +1,8 @@
 import { verifyFernJWTConfig } from "@/server/auth/FernJWT";
-import { getAuthEdgeConfig } from "@/server/auth/getAuthEdgeConfig";
 import { withSecureCookie } from "@/server/auth/withSecure";
-import { COOKIE_FERN_TOKEN } from "@/server/constants";
 import { getXFernHostEdge } from "@/server/xfernhost/edge";
+import { getAuthEdgeConfig } from "@fern-ui/fern-docs-edge-config";
+import { COOKIE_FERN_TOKEN } from "@fern-ui/fern-docs-utils";
 import { NextRequest, NextResponse } from "next/server";
 
 export const runtime = "edge";
diff --git a/packages/ui/docs-bundle/src/pages/api/fern-docs/auth/logout.ts b/packages/ui/docs-bundle/src/pages/api/fern-docs/auth/logout.ts
index 5bfefbee77..3929a08d57 100644
--- a/packages/ui/docs-bundle/src/pages/api/fern-docs/auth/logout.ts
+++ b/packages/ui/docs-bundle/src/pages/api/fern-docs/auth/logout.ts
@@ -1,5 +1,5 @@
-import { COOKIE_ACCESS_TOKEN, COOKIE_FERN_TOKEN, COOKIE_REFRESH_TOKEN } from "@/server/constants";
 import { getXFernHostEdge } from "@/server/xfernhost/edge";
+import { COOKIE_ACCESS_TOKEN, COOKIE_FERN_TOKEN, COOKIE_REFRESH_TOKEN } from "@fern-ui/fern-docs-utils";
 import { NextRequest, NextResponse } from "next/server";
 
 export const runtime = "edge";
diff --git a/packages/ui/docs-bundle/src/pages/api/fern-docs/changelog.ts b/packages/ui/docs-bundle/src/pages/api/fern-docs/changelog.ts
index bd834dbf01..59bbbc0c47 100644
--- a/packages/ui/docs-bundle/src/pages/api/fern-docs/changelog.ts
+++ b/packages/ui/docs-bundle/src/pages/api/fern-docs/changelog.ts
@@ -1,10 +1,10 @@
 import { DocsLoader } from "@/server/DocsLoader";
-import { COOKIE_FERN_TOKEN } from "@/server/constants";
 import { getXFernHostNode } from "@/server/xfernhost/node";
 import type { DocsV1Read } from "@fern-api/fdr-sdk/client/types";
 import * as FernNavigation from "@fern-api/fdr-sdk/navigation";
 import { NodeCollector } from "@fern-api/fdr-sdk/navigation";
 import { assertNever } from "@fern-ui/core-utils";
+import { COOKIE_FERN_TOKEN } from "@fern-ui/fern-docs-utils";
 import { getFrontmatter } from "@fern-ui/ui";
 import { Feed, Item } from "feed";
 import { NextApiRequest, NextApiResponse } from "next";
diff --git a/packages/ui/docs-bundle/src/pages/api/fern-docs/feature-flags.ts b/packages/ui/docs-bundle/src/pages/api/fern-docs/feature-flags.ts
index 142a365833..35164afaad 100644
--- a/packages/ui/docs-bundle/src/pages/api/fern-docs/feature-flags.ts
+++ b/packages/ui/docs-bundle/src/pages/api/fern-docs/feature-flags.ts
@@ -1,173 +1,11 @@
-import { isCustomDomain, isFern } from "@/server/isCustomDomain";
-import { isDevelopment } from "@/server/isDevelopment";
 import { getXFernHostEdge } from "@/server/xfernhost/edge";
-import { FeatureFlags } from "@fern-ui/ui";
-import { getAll } from "@vercel/edge-config";
+import { getFeatureFlags } from "@fern-ui/fern-docs-edge-config";
+import { FeatureFlags } from "@fern-ui/fern-docs-utils";
 import { NextRequest, NextResponse } from "next/server";
 
 export const runtime = "edge";
 
-const FEATURE_FLAGS = [
-    "api-playground-enabled" as const,
-    "api-scrolling-disabled" as const,
-    "whitelabeled" as const,
-    "seo-disabled" as const,
-    "toc-default-enabled" as const,
-    "snippet-template-enabled" as const,
-    "http-snippets-enabled" as const,
-    "inline-feedback-enabled" as const,
-    "dark-code-enabled" as const,
-    "proxy-uses-app-buildwithfern" as const,
-    "image-zoom-disabled" as const,
-    "use-javascript-as-typescript" as const,
-    "always-enable-javascript-fetch" as const,
-    "scroll-in-container-enabled" as const,
-    "use-mdx-bundler" as const,
-    "batch-stream-toggle-disabled" as const,
-    "enabled-auth-in-generated-docs" as const,
-    "ai-chat-preview" as const,
-    "audio-file-download-span-summary" as const,
-    "docs-logo-text-enabled" as const,
-    "audio-example-internal" as const,
-    "uses-application-json-in-form-data-value" as const,
-    "binary-octet-stream-audio-player" as const,
-    "voice-id-playground-form" as const,
-    "cohere-theme" as const,
-    "file-forge-hack-enabled" as const,
-    "hide-404-page" as const,
-    "new-search-experience" as const,
-    "grpc-endpoints" as const,
-];
-
-type FeatureFlag = (typeof FEATURE_FLAGS)[number];
-
-type EdgeConfigResponse = Record<FeatureFlag, string[]>;
-
 export default async function handler(req: NextRequest): Promise<NextResponse<FeatureFlags>> {
     const domain = getXFernHostEdge(req);
     return NextResponse.json(await getFeatureFlags(domain));
 }
-
-export async function getFeatureFlags(domain: string): Promise<FeatureFlags> {
-    try {
-        const config = await getAll<EdgeConfigResponse>(FEATURE_FLAGS);
-
-        const isApiPlaygroundEnabled = checkDomainMatchesCustomers(domain, config["api-playground-enabled"]);
-        const isApiScrollingDisabled = checkDomainMatchesCustomers(domain, config["api-scrolling-disabled"]);
-        const isWhitelabeled = checkDomainMatchesCustomers(domain, config.whitelabeled);
-        const isSeoDisabled = checkDomainMatchesCustomers(domain, config["seo-disabled"]);
-        const isTocDefaultEnabled = checkDomainMatchesCustomers(domain, config["toc-default-enabled"]);
-        const isSnippetTemplatesEnabled = checkDomainMatchesCustomers(domain, config["snippet-template-enabled"]);
-        const isHttpSnippetsEnabled = checkDomainMatchesCustomers(domain, config["http-snippets-enabled"]);
-        const isInlineFeedbackEnabled = checkDomainMatchesCustomers(domain, config["inline-feedback-enabled"]);
-        const isDarkCodeEnabled = checkDomainMatchesCustomers(domain, config["dark-code-enabled"]);
-        const proxyShouldUseAppBuildwithfernCom = checkDomainMatchesCustomers(
-            domain,
-            config["proxy-uses-app-buildwithfern"],
-        );
-        const isImageZoomDisabled = checkDomainMatchesCustomers(domain, config["image-zoom-disabled"]);
-        const useJavaScriptAsTypeScript = checkDomainMatchesCustomers(domain, config["use-javascript-as-typescript"]);
-        const alwaysEnableJavaScriptFetch = checkDomainMatchesCustomers(
-            domain,
-            config["always-enable-javascript-fetch"],
-        );
-        const scrollInContainerEnabled = checkDomainMatchesCustomers(domain, config["scroll-in-container-enabled"]);
-        const useMdxBundler = checkDomainMatchesCustomers(domain, config["use-mdx-bundler"]);
-        const isBatchStreamToggleDisabled = checkDomainMatchesCustomers(domain, config["batch-stream-toggle-disabled"]);
-        const isAuthEnabledInDocs = checkDomainMatchesCustomers(domain, config["enabled-auth-in-generated-docs"]);
-        const isAiChatbotEnabledInPreview = checkDomainMatchesCustomers(domain, config["ai-chat-preview"]);
-        const isAudioFileDownloadSpanSummary = checkDomainMatchesCustomers(
-            domain,
-            config["audio-file-download-span-summary"],
-        );
-        const isDocsLogoTextEnabled = checkDomainMatchesCustomers(domain, config["docs-logo-text-enabled"]);
-        const isAudioExampleInternal = checkDomainMatchesCustomers(domain, config["audio-example-internal"]);
-        const usesApplicationJsonInFormDataValue = checkDomainMatchesCustomers(
-            domain,
-            config["uses-application-json-in-form-data-value"],
-        );
-        const isBinaryOctetStreamAudioPlayer = checkDomainMatchesCustomers(
-            domain,
-            config["binary-octet-stream-audio-player"],
-        );
-        const hasVoiceIdPlaygroundForm = checkDomainMatchesCustomers(domain, config["voice-id-playground-form"]);
-        const isCohereTheme = checkDomainMatchesCustomers(domain, config["cohere-theme"]);
-        const isFileForgeHackEnabled = checkDomainMatchesCustomers(domain, config["file-forge-hack-enabled"]);
-        const is404PageHidden = checkDomainMatchesCustomers(domain, config["hide-404-page"]);
-        const isNewSearchExperienceEnabled = checkDomainMatchesCustomers(domain, config["new-search-experience"]);
-        const grpcEndpoints = config["grpc-endpoints"];
-
-        return {
-            isApiPlaygroundEnabled: isDevelopment(domain) || isApiPlaygroundEnabled,
-            isApiScrollingDisabled,
-            isWhitelabeled,
-            isSeoDisabled: !isCustomDomain(domain) || isSeoDisabled,
-            isTocDefaultEnabled,
-            isSnippetTemplatesEnabled: isSnippetTemplatesEnabled || isDevelopment(domain),
-            isHttpSnippetsEnabled,
-            isInlineFeedbackEnabled,
-            isDarkCodeEnabled,
-            proxyShouldUseAppBuildwithfernCom,
-            isImageZoomDisabled,
-            useJavaScriptAsTypeScript,
-            alwaysEnableJavaScriptFetch,
-            scrollInContainerEnabled,
-            useMdxBundler,
-            isBatchStreamToggleDisabled,
-            isAuthEnabledInDocs,
-            isAiChatbotEnabledInPreview,
-            isAudioFileDownloadSpanSummary,
-            isDocsLogoTextEnabled,
-            isAudioExampleInternal,
-            usesApplicationJsonInFormDataValue,
-            isBinaryOctetStreamAudioPlayer,
-            hasVoiceIdPlaygroundForm,
-            isCohereTheme,
-            isFileForgeHackEnabled,
-            is404PageHidden,
-            isNewSearchExperienceEnabled,
-            grpcEndpoints,
-        };
-    } catch (e) {
-        // eslint-disable-next-line no-console
-        console.error(e);
-        return {
-            isApiPlaygroundEnabled: isDevelopment(domain),
-            isApiScrollingDisabled: false,
-            isWhitelabeled: false,
-            isSeoDisabled: !isCustomDomain(domain),
-            isTocDefaultEnabled: false,
-            isSnippetTemplatesEnabled: isDevelopment(domain),
-            isHttpSnippetsEnabled: false,
-            isInlineFeedbackEnabled: isFern(domain),
-            isDarkCodeEnabled: false,
-            proxyShouldUseAppBuildwithfernCom: false,
-            isImageZoomDisabled: false,
-            useJavaScriptAsTypeScript: false,
-            alwaysEnableJavaScriptFetch: false,
-            scrollInContainerEnabled: false,
-            useMdxBundler: false,
-            isBatchStreamToggleDisabled: false,
-            isAuthEnabledInDocs: false,
-            isAiChatbotEnabledInPreview: false,
-            isAudioFileDownloadSpanSummary: false,
-            isDocsLogoTextEnabled: false,
-            isAudioExampleInternal: false,
-            usesApplicationJsonInFormDataValue: false,
-            isBinaryOctetStreamAudioPlayer: false,
-            hasVoiceIdPlaygroundForm: false,
-            isCohereTheme: false,
-            isFileForgeHackEnabled: false,
-            is404PageHidden: false,
-            isNewSearchExperienceEnabled: false,
-            grpcEndpoints: [],
-        };
-    }
-}
-
-function checkDomainMatchesCustomers(domain: string, customers: readonly string[]): boolean {
-    if (customers == null) {
-        return false;
-    }
-    return customers.some((customer) => domain.toLowerCase().includes(customer.toLowerCase()));
-}
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 272da12f0a..53d0891703 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,29 +1,107 @@
+import { verifyFernJWTConfig } from "@/server/auth/FernJWT";
 import { getXFernHostEdge } from "@/server/xfernhost/edge";
-import { get } from "@vercel/edge-config";
-import { NextRequest, NextResponse } from "next/server";
-import { z } from "zod";
+import { LaunchDarklyEdgeConfig, getAuthEdgeConfig, getLaunchDarklySettings } from "@fern-ui/fern-docs-edge-config";
+import { COOKIE_EMAIL, COOKIE_FERN_TOKEN } from "@fern-ui/fern-docs-utils";
+import { randomUUID } from "crypto";
+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);
-    try {
-        const config = (await get<Record<string, LaunchDarklyEdgeConfigSchema>>("launchdarkly"))?.[domain];
-        if (config == null) {
-            return NextResponse.json({}, { status: 404 });
+
+    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 = req.cookies;
+
+    const fernToken = jar.get(COOKIE_FERN_TOKEN)?.value;
+    const email = jar.get(COOKIE_EMAIL)?.value;
+
+    if (fernToken) {
+        try {
+            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
         }
+    }
+
+    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/integrations/launchdarkly/identify.ts b/packages/ui/docs-bundle/src/pages/api/fern-docs/integrations/launchdarkly/identify.ts
new file mode 100644
index 0000000000..75f4bef80f
--- /dev/null
+++ b/packages/ui/docs-bundle/src/pages/api/fern-docs/integrations/launchdarkly/identify.ts
@@ -0,0 +1,16 @@
+import { COOKIE_EMAIL } from "@fern-ui/fern-docs-utils";
+import { NextRequest, NextResponse } from "next/server";
+
+export const runtime = "edge";
+
+export default async function handler(req: NextRequest): Promise<NextResponse> {
+    const email = req.nextUrl.searchParams.get(COOKIE_EMAIL);
+
+    const res = NextResponse.redirect(new URL("/", req.url));
+
+    if (email) {
+        res.cookies.set({ name: COOKIE_EMAIL, value: email });
+    }
+
+    return res;
+}
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 671982e920..fda45f531e 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
@@ -1,10 +1,10 @@
 import { signFernJWT } from "@/server/auth/FernJWT";
 import { OAuth2Client } from "@/server/auth/OAuth2Client";
-import { getAuthEdgeConfig } from "@/server/auth/getAuthEdgeConfig";
 import { withSecureCookie } from "@/server/auth/withSecure";
-import { COOKIE_ACCESS_TOKEN, COOKIE_FERN_TOKEN, COOKIE_REFRESH_TOKEN } from "@/server/constants";
 import { getXFernHostEdge } from "@/server/xfernhost/edge";
-import { FernUser, OryAccessTokenSchema } from "@fern-ui/ui/auth";
+import { FernUser, OryAccessTokenSchema } from "@fern-ui/fern-docs-auth";
+import { getAuthEdgeConfig } from "@fern-ui/fern-docs-edge-config";
+import { COOKIE_ACCESS_TOKEN, COOKIE_FERN_TOKEN, COOKIE_REFRESH_TOKEN } from "@fern-ui/fern-docs-utils";
 import { NextRequest, NextResponse } from "next/server";
 
 export const runtime = "edge";
diff --git a/packages/ui/docs-bundle/src/pages/api/fern-docs/oauth/webflow/callback.ts b/packages/ui/docs-bundle/src/pages/api/fern-docs/oauth/webflow/callback.ts
index 1591ab64b3..90548246fa 100644
--- a/packages/ui/docs-bundle/src/pages/api/fern-docs/oauth/webflow/callback.ts
+++ b/packages/ui/docs-bundle/src/pages/api/fern-docs/oauth/webflow/callback.ts
@@ -1,6 +1,6 @@
-import { getAuthEdgeConfig } from "@/server/auth/getAuthEdgeConfig";
 import { withSecureCookie } from "@/server/auth/withSecure";
 import { getXFernHostEdge } from "@/server/xfernhost/edge";
+import { getAuthEdgeConfig } from "@fern-ui/fern-docs-edge-config";
 import { NextRequest, NextResponse } from "next/server";
 import { WebflowClient } from "webflow-api";
 
diff --git a/packages/ui/docs-bundle/src/pages/api/fern-docs/preview.ts b/packages/ui/docs-bundle/src/pages/api/fern-docs/preview.ts
index 488ba64d6f..93bb8619bd 100644
--- a/packages/ui/docs-bundle/src/pages/api/fern-docs/preview.ts
+++ b/packages/ui/docs-bundle/src/pages/api/fern-docs/preview.ts
@@ -1,5 +1,5 @@
-import { COOKIE_FERN_DOCS_PREVIEW } from "@/server/constants";
 import { notFoundResponse, redirectResponse } from "@/server/serverResponse";
+import { COOKIE_FERN_DOCS_PREVIEW } from "@fern-ui/fern-docs-utils";
 import { NextRequest, NextResponse } from "next/server";
 
 export const runtime = "edge";
diff --git a/packages/ui/docs-bundle/src/pages/api/fern-docs/resolve-api.ts b/packages/ui/docs-bundle/src/pages/api/fern-docs/resolve-api.ts
index 78d6882011..cbb1d37250 100644
--- a/packages/ui/docs-bundle/src/pages/api/fern-docs/resolve-api.ts
+++ b/packages/ui/docs-bundle/src/pages/api/fern-docs/resolve-api.ts
@@ -3,10 +3,10 @@ import { getXFernHostNode } from "@/server/xfernhost/node";
 import { FdrAPI } from "@fern-api/fdr-sdk";
 import * as FernNavigation from "@fern-api/fdr-sdk/navigation";
 import { ApiDefinitionHolder } from "@fern-api/fdr-sdk/navigation";
+import { getFeatureFlags } from "@fern-ui/fern-docs-edge-config";
 import { ApiDefinitionResolver, provideRegistryService, type ResolvedRootPackage } from "@fern-ui/ui";
 import { getMdxBundler } from "@fern-ui/ui/bundlers";
 import { NextApiHandler, NextApiResponse } from "next";
-import { getFeatureFlags } from "./feature-flags";
 
 export const dynamic = "force-dynamic";
 
diff --git a/packages/ui/docs-bundle/src/pages/api/fern-docs/robots.txt.ts b/packages/ui/docs-bundle/src/pages/api/fern-docs/robots.txt.ts
index a087142b33..8bc7dde788 100644
--- a/packages/ui/docs-bundle/src/pages/api/fern-docs/robots.txt.ts
+++ b/packages/ui/docs-bundle/src/pages/api/fern-docs/robots.txt.ts
@@ -1,5 +1,5 @@
-import { getSeoDisabled } from "@/server/disabledSeo";
 import { getXFernHostEdge } from "@/server/xfernhost/edge";
+import { getSeoDisabled } from "@fern-ui/fern-docs-edge-config";
 import { NextRequest, NextResponse } from "next/server";
 import urlJoin from "url-join";
 
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 6bee714209..8dffdb2e69 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 { getAuthEdgeConfig } from "@/server/auth/getAuthEdgeConfig";
 import { loadWithUrl } from "@/server/loadWithUrl";
 import { getXFernHostEdge } from "@/server/xfernhost/edge";
+import { getAuthEdgeConfig, getInkeepSettings } from "@fern-ui/fern-docs-edge-config";
 import { SearchConfig, getSearchConfig } from "@fern-ui/search-utils";
 import { provideRegistryService } from "@fern-ui/ui";
 import { NextRequest, NextResponse } from "next/server";
@@ -30,8 +30,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) {
         // TODO: sentry
diff --git a/packages/ui/docs-bundle/src/pages/api/fern-docs/sitemap.xml.ts b/packages/ui/docs-bundle/src/pages/api/fern-docs/sitemap.xml.ts
index 56022ff25f..dc8513a67b 100644
--- a/packages/ui/docs-bundle/src/pages/api/fern-docs/sitemap.xml.ts
+++ b/packages/ui/docs-bundle/src/pages/api/fern-docs/sitemap.xml.ts
@@ -1,9 +1,9 @@
 import { DocsLoader } from "@/server/DocsLoader";
-import { COOKIE_FERN_TOKEN } from "@/server/constants";
 import { conformTrailingSlash } from "@/server/trailingSlash";
 import { getXFernHostEdge } from "@/server/xfernhost/edge";
 import { NodeCollector } from "@fern-api/fdr-sdk/navigation";
 import { withDefaultProtocol } from "@fern-ui/core-utils";
+import { COOKIE_FERN_TOKEN } from "@fern-ui/fern-docs-utils";
 import { NextRequest, NextResponse } from "next/server";
 import urljoin from "url-join";
 
diff --git a/packages/ui/docs-bundle/src/server/ApiDefinitionLoader.ts b/packages/ui/docs-bundle/src/server/ApiDefinitionLoader.ts
index f611e15c97..a9369a9177 100644
--- a/packages/ui/docs-bundle/src/server/ApiDefinitionLoader.ts
+++ b/packages/ui/docs-bundle/src/server/ApiDefinitionLoader.ts
@@ -11,7 +11,9 @@ import {
     type PruningNodeType,
 } from "@fern-api/fdr-sdk/api-definition";
 import type * as FernDocs from "@fern-api/fdr-sdk/docs";
-import { DEFAULT_FEATURE_FLAGS, provideRegistryService, type FeatureFlags } from "@fern-ui/ui";
+import type { FeatureFlags } from "@fern-ui/fern-docs-utils";
+import { DEFAULT_FEATURE_FLAGS } from "@fern-ui/fern-docs-utils";
+import { provideRegistryService } from "@fern-ui/ui";
 import { getMdxBundler } from "@fern-ui/ui/bundlers";
 import { HTTPSnippet, type TargetId } from "httpsnippet-lite";
 import { UnreachableCaseError } from "ts-essentials";
diff --git a/packages/ui/docs-bundle/src/server/DocsLoader.ts b/packages/ui/docs-bundle/src/server/DocsLoader.ts
index 0b58660d9e..505660495e 100644
--- a/packages/ui/docs-bundle/src/server/DocsLoader.ts
+++ b/packages/ui/docs-bundle/src/server/DocsLoader.ts
@@ -1,8 +1,8 @@
 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/ui/auth";
+import type { AuthEdgeConfig, FernUser } from "@fern-ui/fern-docs-auth";
+import { getAuthEdgeConfig } from "@fern-ui/fern-docs-edge-config";
 import { verifyFernJWTConfig } from "./auth/FernJWT";
-import { getAuthEdgeConfig } from "./auth/getAuthEdgeConfig";
 import { AuthProps } from "./authProps";
 import { loadWithUrl } from "./loadWithUrl";
 import { pruneWithBasicTokenPublic } from "./withBasicTokenPublic";
diff --git a/packages/ui/docs-bundle/src/server/LoadDocsPerformanceTracker.ts b/packages/ui/docs-bundle/src/server/LoadDocsPerformanceTracker.ts
index 3f3a7bec1a..d1eca768ca 100644
--- a/packages/ui/docs-bundle/src/server/LoadDocsPerformanceTracker.ts
+++ b/packages/ui/docs-bundle/src/server/LoadDocsPerformanceTracker.ts
@@ -1,8 +1,8 @@
+import { TRACK_LOAD_DOCS_PERFORMANCE } from "@fern-ui/fern-docs-utils";
 import { DocsPage } from "@fern-ui/ui";
 import { track } from "@vercel/analytics/server";
 import { GetServerSidePropsResult } from "next/types";
 import { ComponentProps } from "react";
-import { TRACK_LOAD_DOCS_PERFORMANCE } from "./constants";
 import type { LoadWithUrlResponse } from "./loadWithUrl";
 
 export class LoadDocsPerformanceTracker {
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 ad93cab727..14d43cf42b 100644
--- a/packages/ui/docs-bundle/src/server/auth/checkViewerAllowed.ts
+++ b/packages/ui/docs-bundle/src/server/auth/checkViewerAllowed.ts
@@ -1,6 +1,6 @@
-import { AuthEdgeConfig } from "@fern-ui/ui/auth";
+import { AuthEdgeConfig } from "@fern-ui/fern-docs-auth";
+import { COOKIE_FERN_TOKEN } from "@fern-ui/fern-docs-utils";
 import type { NextRequest } from "next/server";
-import { COOKIE_FERN_TOKEN } from "../constants";
 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 2bf07a9b5e..3887ddd511 100644
--- a/packages/ui/docs-bundle/src/server/auth/getApiKeyInjectionConfig.ts
+++ b/packages/ui/docs-bundle/src/server/auth/getApiKeyInjectionConfig.ts
@@ -1,8 +1,8 @@
-import type { APIKeyInjectionConfig } from "@fern-ui/ui/auth";
+import type { APIKeyInjectionConfig } from "@fern-ui/fern-docs-auth";
+import { getAuthEdgeConfig } from "@fern-ui/fern-docs-edge-config";
 import type { NextApiRequestCookies } from "next/dist/server/api-utils";
 import type { NextRequest } from "next/server";
 import { OAuth2Client } from "./OAuth2Client";
-import { getAuthEdgeConfig } from "./getAuthEdgeConfig";
 
 // TODO: since this is for ORY (rightbrain) only, lets refactor
 export async function getAPIKeyInjectionConfig(
diff --git a/packages/ui/docs-bundle/src/server/authProps.ts b/packages/ui/docs-bundle/src/server/authProps.ts
index 2a0ace4aaf..6d93fb6401 100644
--- a/packages/ui/docs-bundle/src/server/authProps.ts
+++ b/packages/ui/docs-bundle/src/server/authProps.ts
@@ -1,6 +1,6 @@
-import type { FernUser } from "@fern-ui/ui/auth";
+import type { FernUser } from "@fern-ui/fern-docs-auth";
+import { getAuthEdgeConfig } from "@fern-ui/fern-docs-edge-config";
 import { verifyFernJWTConfig } from "./auth/FernJWT";
-import { getAuthEdgeConfig } from "./auth/getAuthEdgeConfig";
 
 export interface AuthProps {
     token: string;
diff --git a/packages/ui/docs-bundle/src/server/getDocsPageProps.ts b/packages/ui/docs-bundle/src/server/getDocsPageProps.ts
index cde6679c14..a8f79657e6 100644
--- a/packages/ui/docs-bundle/src/server/getDocsPageProps.ts
+++ b/packages/ui/docs-bundle/src/server/getDocsPageProps.ts
@@ -1,7 +1,6 @@
+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";
 import { LoadDocsPerformanceTracker } from "./LoadDocsPerformanceTracker";
 import type { AuthProps } from "./authProps";
@@ -19,7 +18,6 @@ export async function getDocsPageProps(
     xFernHost: string | undefined,
     slug: string[],
     auth?: AuthProps,
-    cookies?: NextApiRequestCookies,
 ): Promise<SSGDocsPageProps> {
     if (xFernHost == null || Array.isArray(xFernHost)) {
         return { notFound: true };
@@ -35,9 +33,7 @@ export async function getDocsPageProps(
     /**
      * Convert the docs into initial props for the page.
      */
-    const initialProps = await performance.trackInitialPropsPromise(
-        withInitialProps({ docs, slug, xFernHost, auth, cookies }),
-    );
+    const initialProps = await performance.trackInitialPropsPromise(withInitialProps({ docs, slug, xFernHost, auth }));
 
     /**
      * Send performance data to Vercel Analytics.
diff --git a/packages/ui/docs-bundle/src/server/getDynamicDocsPageProps.ts b/packages/ui/docs-bundle/src/server/getDynamicDocsPageProps.ts
index 3d3ac79a84..6fb6f6ef9c 100644
--- a/packages/ui/docs-bundle/src/server/getDynamicDocsPageProps.ts
+++ b/packages/ui/docs-bundle/src/server/getDynamicDocsPageProps.ts
@@ -1,9 +1,9 @@
+import { COOKIE_FERN_TOKEN } from "@fern-ui/fern-docs-utils";
 import { type DocsPage } from "@fern-ui/ui";
 import type { NextApiRequestCookies } from "next/dist/server/api-utils";
 import type { GetServerSidePropsResult } from "next/types";
 import type { ComponentProps } from "react";
 import { withAuthProps } from "./authProps";
-import { COOKIE_FERN_TOKEN } from "./constants";
 import { getDocsPageProps } from "./getDocsPageProps";
 
 type GetServerSideDocsPagePropsResult = GetServerSidePropsResult<ComponentProps<typeof DocsPage>>;
@@ -26,5 +26,5 @@ export async function getDynamicDocsPageProps(
      * would have redirected them to the login page
      */
     const authProps = await withAuthProps(xFernHost, cookies[COOKIE_FERN_TOKEN]);
-    return getDocsPageProps(xFernHost, slug, authProps, cookies);
+    return getDocsPageProps(xFernHost, slug, authProps);
 }
diff --git a/packages/ui/docs-bundle/src/server/loadWithUrl.ts b/packages/ui/docs-bundle/src/server/loadWithUrl.ts
index 6ba0d1450f..59517c375c 100644
--- a/packages/ui/docs-bundle/src/server/loadWithUrl.ts
+++ b/packages/ui/docs-bundle/src/server/loadWithUrl.ts
@@ -1,7 +1,7 @@
 import { APIResponse, FdrAPI } from "@fern-api/fdr-sdk/client/types";
+import { withoutStaging } from "@fern-ui/fern-docs-utils";
 import { getRegistryServiceWithToken, provideRegistryService } from "@fern-ui/ui";
 import type { AuthProps } from "./authProps";
-import { withoutStaging } from "./withoutStaging";
 
 export type LoadWithUrlResponse = APIResponse<
     FdrAPI.docs.v2.read.LoadDocsForUrlResponse,
diff --git a/packages/ui/docs-bundle/src/server/withBasicTokenPublic.ts b/packages/ui/docs-bundle/src/server/withBasicTokenPublic.ts
index 6dfc64ddab..47ec230b92 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";
 
 /**
  * @param auth Basic token verification configuration
diff --git a/packages/ui/docs-bundle/src/server/withInitialProps.ts b/packages/ui/docs-bundle/src/server/withInitialProps.ts
index 6c97346d90..a3528e24cb 100644
--- a/packages/ui/docs-bundle/src/server/withInitialProps.ts
+++ b/packages/ui/docs-bundle/src/server/withInitialProps.ts
@@ -1,31 +1,29 @@
-import { getFeatureFlags } from "@/pages/api/fern-docs/feature-flags";
 import * as FernNavigation from "@fern-api/fdr-sdk/navigation";
 import { withDefaultProtocol } from "@fern-ui/core-utils";
 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-config";
 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 { getAuthEdgeConfig } from "./auth/getAuthEdgeConfig";
 import type { AuthProps } from "./authProps";
-import { getSeoDisabled } from "./disabledSeo";
-import { getCustomerAnalytics } from "./getCustomerAnalytics";
 import { handleLoadDocsError } from "./handleLoadDocsError";
 import type { LoadWithUrlResponse } from "./loadWithUrl";
 import { isTrailingSlashEnabled } from "./trailingSlash";
@@ -36,7 +34,6 @@ interface WithInitialProps {
     slug: string[];
     xFernHost: string;
     auth?: AuthProps;
-    cookies?: NextApiRequestCookies;
 }
 
 export async function withInitialProps({
@@ -44,7 +41,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);
@@ -287,12 +283,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) {
@@ -305,9 +295,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/src/server/withSSGProps.ts b/packages/ui/docs-bundle/src/server/withSSGProps.ts
index 036c7b2b51..927ca05477 100644
--- a/packages/ui/docs-bundle/src/server/withSSGProps.ts
+++ b/packages/ui/docs-bundle/src/server/withSSGProps.ts
@@ -1,6 +1,6 @@
+import { REVALIDATE_SECONDS } from "@fern-ui/fern-docs-utils";
 import { GetServerSidePropsResult, GetStaticPropsResult } from "next";
 import { UnreachableCaseError } from "ts-essentials";
-import { REVALIDATE_SECONDS } from "./constants";
 
 /**
  * Always revalidate redirects and notFound pages.
diff --git a/packages/ui/docs-bundle/src/server/xfernhost/edge.ts b/packages/ui/docs-bundle/src/server/xfernhost/edge.ts
index 7f96dd1938..56177fbad5 100644
--- a/packages/ui/docs-bundle/src/server/xfernhost/edge.ts
+++ b/packages/ui/docs-bundle/src/server/xfernhost/edge.ts
@@ -1,4 +1,4 @@
-import { COOKIE_FERN_DOCS_PREVIEW, HEADER_X_FERN_HOST } from "@/server/constants";
+import { COOKIE_FERN_DOCS_PREVIEW, HEADER_X_FERN_HOST } from "@fern-ui/fern-docs-utils";
 import type { NextRequest } from "next/server";
 import { getNextPublicDocsDomain } from "./dev";
 import { cleanHost } from "./util";
diff --git a/packages/ui/docs-bundle/src/server/xfernhost/node.ts b/packages/ui/docs-bundle/src/server/xfernhost/node.ts
index e22682dffc..ed24223684 100644
--- a/packages/ui/docs-bundle/src/server/xfernhost/node.ts
+++ b/packages/ui/docs-bundle/src/server/xfernhost/node.ts
@@ -1,4 +1,4 @@
-import { COOKIE_FERN_DOCS_PREVIEW, HEADER_X_FERN_HOST } from "@/server/constants";
+import { COOKIE_FERN_DOCS_PREVIEW, HEADER_X_FERN_HOST } from "@fern-ui/fern-docs-utils";
 import type { NextApiRequest } from "next";
 import { getNextPublicDocsDomain } from "./dev";
 import { cleanHost } from "./util";
diff --git a/packages/ui/docs-bundle/tsconfig.json b/packages/ui/docs-bundle/tsconfig.json
index 9ee90994e8..99f67216da 100644
--- a/packages/ui/docs-bundle/tsconfig.json
+++ b/packages/ui/docs-bundle/tsconfig.json
@@ -20,7 +20,16 @@
       "path": "../../commons/search-utils"
     },
     {
-      "path": "../../fdr-sdk"
+      "path": "../fern-docs-auth"
+    },
+    {
+      "path": "../fern-docs-edge-config"
+    },
+    {
+      "path": "../fern-docs-utils"
+    },
+    {
+      "path": "../../fdr-sdk/tsconfig.build.json"
     },
     {
       "path": "../app/tsconfig.build.json"
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-config/.depcheckrc.json b/packages/ui/fern-docs-edge-config/.depcheckrc.json
new file mode 100644
index 0000000000..cf8a6564f2
--- /dev/null
+++ b/packages/ui/fern-docs-edge-config/.depcheckrc.json
@@ -0,0 +1 @@
+{ "ignores": ["@fern-platform/configs", "@types/node", "vite", "@types/react"], "ignore-patterns": ["dist"] }
diff --git a/packages/ui/fern-docs-edge-config/.prettierrc.cjs b/packages/ui/fern-docs-edge-config/.prettierrc.cjs
new file mode 100644
index 0000000000..39cf0d0b8c
--- /dev/null
+++ b/packages/ui/fern-docs-edge-config/.prettierrc.cjs
@@ -0,0 +1 @@
+module.exports = require("../../../.prettierrc.json");
diff --git a/packages/ui/fern-docs-edge-config/.stylelintrc.json b/packages/ui/fern-docs-edge-config/.stylelintrc.json
new file mode 100644
index 0000000000..0d2e3ff61d
--- /dev/null
+++ b/packages/ui/fern-docs-edge-config/.stylelintrc.json
@@ -0,0 +1 @@
+{ "extends": ["../../../shared/stylelintrc.shared.json"] }
diff --git a/packages/ui/fern-docs-edge-config/package.json b/packages/ui/fern-docs-edge-config/package.json
new file mode 100644
index 0000000000..f150a7942d
--- /dev/null
+++ b/packages/ui/fern-docs-edge-config/package.json
@@ -0,0 +1,55 @@
+{
+  "name": "@fern-ui/fern-docs-edge-config",
+  "version": "0.0.0",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/fern-api/fern-platform.git",
+    "directory": "packages/commons/fdr-utils"
+  },
+  "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": {
+    "@fern-ui/fern-docs-auth": "workspace:*",
+    "@fern-ui/fern-docs-utils": "workspace:*",
+    "@vercel/edge-config": "^1.1.0",
+    "url-join": "5.0.0",
+    "zod": "^3.23.8"
+  },
+  "devDependencies": {
+    "@fern-platform/configs": "workspace:*",
+    "@fern-ui/search-utils": "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/docs-bundle/src/server/auth/getAuthEdgeConfig.ts b/packages/ui/fern-docs-edge-config/src/getAuthEdgeConfig.ts
similarity index 86%
rename from packages/ui/docs-bundle/src/server/auth/getAuthEdgeConfig.ts
rename to packages/ui/fern-docs-edge-config/src/getAuthEdgeConfig.ts
index 71d5a80509..10d321e0a2 100644
--- a/packages/ui/docs-bundle/src/server/auth/getAuthEdgeConfig.ts
+++ b/packages/ui/fern-docs-edge-config/src/getAuthEdgeConfig.ts
@@ -1,6 +1,6 @@
-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 { get } from "@vercel/edge-config";
-import { withoutStaging } from "../withoutStaging";
 
 const KEY = "authentication";
 
diff --git a/packages/ui/docs-bundle/src/server/getCustomerAnalytics.ts b/packages/ui/fern-docs-edge-config/src/getCustomerAnalytics.ts
similarity index 71%
rename from packages/ui/docs-bundle/src/server/getCustomerAnalytics.ts
rename to packages/ui/fern-docs-edge-config/src/getCustomerAnalytics.ts
index 891abad3e8..725d397838 100644
--- a/packages/ui/docs-bundle/src/server/getCustomerAnalytics.ts
+++ b/packages/ui/fern-docs-edge-config/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-config/src/getFeatureFlags.ts b/packages/ui/fern-docs-edge-config/src/getFeatureFlags.ts
new file mode 100644
index 0000000000..c4c0886ffc
--- /dev/null
+++ b/packages/ui/fern-docs-edge-config/src/getFeatureFlags.ts
@@ -0,0 +1,165 @@
+import type { FeatureFlags } from "@fern-ui/fern-docs-utils";
+import { isCustomDomain, isDevelopment, isFern } from "@fern-ui/fern-docs-utils";
+import { getAll } from "@vercel/edge-config";
+
+export const runtime = "edge";
+
+const FEATURE_FLAGS = [
+    "api-playground-enabled" as const,
+    "api-scrolling-disabled" as const,
+    "whitelabeled" as const,
+    "seo-disabled" as const,
+    "toc-default-enabled" as const,
+    "snippet-template-enabled" as const,
+    "http-snippets-enabled" as const,
+    "inline-feedback-enabled" as const,
+    "dark-code-enabled" as const,
+    "proxy-uses-app-buildwithfern" as const,
+    "image-zoom-disabled" as const,
+    "use-javascript-as-typescript" as const,
+    "always-enable-javascript-fetch" as const,
+    "scroll-in-container-enabled" as const,
+    "use-mdx-bundler" as const,
+    "batch-stream-toggle-disabled" as const,
+    "enabled-auth-in-generated-docs" as const,
+    "ai-chat-preview" as const,
+    "audio-file-download-span-summary" as const,
+    "docs-logo-text-enabled" as const,
+    "audio-example-internal" as const,
+    "uses-application-json-in-form-data-value" as const,
+    "binary-octet-stream-audio-player" as const,
+    "voice-id-playground-form" as const,
+    "cohere-theme" as const,
+    "file-forge-hack-enabled" as const,
+    "hide-404-page" as const,
+    "new-search-experience" as const,
+    "grpc-endpoints" as const,
+];
+
+type FeatureFlag = (typeof FEATURE_FLAGS)[number];
+
+type EdgeConfigResponse = Record<FeatureFlag, string[]>;
+
+export async function getFeatureFlags(domain: string): Promise<FeatureFlags> {
+    try {
+        const config = await getAll<EdgeConfigResponse>(FEATURE_FLAGS);
+
+        const isApiPlaygroundEnabled = checkDomainMatchesCustomers(domain, config["api-playground-enabled"]);
+        const isApiScrollingDisabled = checkDomainMatchesCustomers(domain, config["api-scrolling-disabled"]);
+        const isWhitelabeled = checkDomainMatchesCustomers(domain, config.whitelabeled);
+        const isSeoDisabled = checkDomainMatchesCustomers(domain, config["seo-disabled"]);
+        const isTocDefaultEnabled = checkDomainMatchesCustomers(domain, config["toc-default-enabled"]);
+        const isSnippetTemplatesEnabled = checkDomainMatchesCustomers(domain, config["snippet-template-enabled"]);
+        const isHttpSnippetsEnabled = checkDomainMatchesCustomers(domain, config["http-snippets-enabled"]);
+        const isInlineFeedbackEnabled = checkDomainMatchesCustomers(domain, config["inline-feedback-enabled"]);
+        const isDarkCodeEnabled = checkDomainMatchesCustomers(domain, config["dark-code-enabled"]);
+        const proxyShouldUseAppBuildwithfernCom = checkDomainMatchesCustomers(
+            domain,
+            config["proxy-uses-app-buildwithfern"],
+        );
+        const isImageZoomDisabled = checkDomainMatchesCustomers(domain, config["image-zoom-disabled"]);
+        const useJavaScriptAsTypeScript = checkDomainMatchesCustomers(domain, config["use-javascript-as-typescript"]);
+        const alwaysEnableJavaScriptFetch = checkDomainMatchesCustomers(
+            domain,
+            config["always-enable-javascript-fetch"],
+        );
+        const scrollInContainerEnabled = checkDomainMatchesCustomers(domain, config["scroll-in-container-enabled"]);
+        const useMdxBundler = checkDomainMatchesCustomers(domain, config["use-mdx-bundler"]);
+        const isBatchStreamToggleDisabled = checkDomainMatchesCustomers(domain, config["batch-stream-toggle-disabled"]);
+        const isAuthEnabledInDocs = checkDomainMatchesCustomers(domain, config["enabled-auth-in-generated-docs"]);
+        const isAiChatbotEnabledInPreview = checkDomainMatchesCustomers(domain, config["ai-chat-preview"]);
+        const isAudioFileDownloadSpanSummary = checkDomainMatchesCustomers(
+            domain,
+            config["audio-file-download-span-summary"],
+        );
+        const isDocsLogoTextEnabled = checkDomainMatchesCustomers(domain, config["docs-logo-text-enabled"]);
+        const isAudioExampleInternal = checkDomainMatchesCustomers(domain, config["audio-example-internal"]);
+        const usesApplicationJsonInFormDataValue = checkDomainMatchesCustomers(
+            domain,
+            config["uses-application-json-in-form-data-value"],
+        );
+        const isBinaryOctetStreamAudioPlayer = checkDomainMatchesCustomers(
+            domain,
+            config["binary-octet-stream-audio-player"],
+        );
+        const hasVoiceIdPlaygroundForm = checkDomainMatchesCustomers(domain, config["voice-id-playground-form"]);
+        const isCohereTheme = checkDomainMatchesCustomers(domain, config["cohere-theme"]);
+        const isFileForgeHackEnabled = checkDomainMatchesCustomers(domain, config["file-forge-hack-enabled"]);
+        const is404PageHidden = checkDomainMatchesCustomers(domain, config["hide-404-page"]);
+        const isNewSearchExperienceEnabled = checkDomainMatchesCustomers(domain, config["new-search-experience"]);
+        const grpcEndpoints = config["grpc-endpoints"];
+
+        return {
+            isApiPlaygroundEnabled: isDevelopment(domain) || isApiPlaygroundEnabled,
+            isApiScrollingDisabled,
+            isWhitelabeled,
+            isSeoDisabled: !isCustomDomain(domain) || isSeoDisabled,
+            isTocDefaultEnabled,
+            isSnippetTemplatesEnabled: isSnippetTemplatesEnabled || isDevelopment(domain),
+            isHttpSnippetsEnabled,
+            isInlineFeedbackEnabled,
+            isDarkCodeEnabled,
+            proxyShouldUseAppBuildwithfernCom,
+            isImageZoomDisabled,
+            useJavaScriptAsTypeScript,
+            alwaysEnableJavaScriptFetch,
+            scrollInContainerEnabled,
+            useMdxBundler,
+            isBatchStreamToggleDisabled,
+            isAuthEnabledInDocs,
+            isAiChatbotEnabledInPreview,
+            isAudioFileDownloadSpanSummary,
+            isDocsLogoTextEnabled,
+            isAudioExampleInternal,
+            usesApplicationJsonInFormDataValue,
+            isBinaryOctetStreamAudioPlayer,
+            hasVoiceIdPlaygroundForm,
+            isCohereTheme,
+            isFileForgeHackEnabled,
+            is404PageHidden,
+            isNewSearchExperienceEnabled,
+            grpcEndpoints,
+        };
+    } catch (e) {
+        // eslint-disable-next-line no-console
+        console.error(e);
+        return {
+            isApiPlaygroundEnabled: isDevelopment(domain),
+            isApiScrollingDisabled: false,
+            isWhitelabeled: false,
+            isSeoDisabled: !isCustomDomain(domain),
+            isTocDefaultEnabled: false,
+            isSnippetTemplatesEnabled: isDevelopment(domain),
+            isHttpSnippetsEnabled: false,
+            isInlineFeedbackEnabled: isFern(domain),
+            isDarkCodeEnabled: false,
+            proxyShouldUseAppBuildwithfernCom: false,
+            isImageZoomDisabled: false,
+            useJavaScriptAsTypeScript: false,
+            alwaysEnableJavaScriptFetch: false,
+            scrollInContainerEnabled: false,
+            useMdxBundler: false,
+            isBatchStreamToggleDisabled: false,
+            isAuthEnabledInDocs: false,
+            isAiChatbotEnabledInPreview: false,
+            isAudioFileDownloadSpanSummary: false,
+            isDocsLogoTextEnabled: false,
+            isAudioExampleInternal: false,
+            usesApplicationJsonInFormDataValue: false,
+            isBinaryOctetStreamAudioPlayer: false,
+            hasVoiceIdPlaygroundForm: false,
+            isCohereTheme: false,
+            isFileForgeHackEnabled: false,
+            is404PageHidden: false,
+            isNewSearchExperienceEnabled: false,
+            grpcEndpoints: [],
+        };
+    }
+}
+
+function checkDomainMatchesCustomers(domain: string, customers: readonly string[]): boolean {
+    if (customers == null) {
+        return false;
+    }
+    return customers.some((customer) => domain.toLowerCase().includes(customer.toLowerCase()));
+}
diff --git a/packages/ui/fern-docs-edge-config/src/getInkeepSettings.ts b/packages/ui/fern-docs-edge-config/src/getInkeepSettings.ts
new file mode 100644
index 0000000000..b40e4c03b3
--- /dev/null
+++ b/packages/ui/fern-docs-edge-config/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-config/src/getLaunchDarklySettings.ts b/packages/ui/fern-docs-edge-config/src/getLaunchDarklySettings.ts
new file mode 100644
index 0000000000..ec3dbffca9
--- /dev/null
+++ b/packages/ui/fern-docs-edge-config/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/docs-bundle/src/server/disabledSeo.ts b/packages/ui/fern-docs-edge-config/src/getSeoDisabled.ts
similarity index 82%
rename from packages/ui/docs-bundle/src/server/disabledSeo.ts
rename to packages/ui/fern-docs-edge-config/src/getSeoDisabled.ts
index 0825c19f24..ffea760ec7 100644
--- a/packages/ui/docs-bundle/src/server/disabledSeo.ts
+++ b/packages/ui/fern-docs-edge-config/src/getSeoDisabled.ts
@@ -1,5 +1,5 @@
+import { isCustomDomain } from "@fern-ui/fern-docs-utils";
 import { get } from "@vercel/edge-config";
-import { isCustomDomain } from "./isCustomDomain";
 
 export async function getSeoDisabled(host: string): Promise<boolean> {
     if (!isCustomDomain(host)) {
diff --git a/packages/ui/fern-docs-edge-config/src/index.ts b/packages/ui/fern-docs-edge-config/src/index.ts
new file mode 100644
index 0000000000..8e281180a4
--- /dev/null
+++ b/packages/ui/fern-docs-edge-config/src/index.ts
@@ -0,0 +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-config/tsconfig.json b/packages/ui/fern-docs-edge-config/tsconfig.json
new file mode 100644
index 0000000000..145c1a3bd9
--- /dev/null
+++ b/packages/ui/fern-docs-edge-config/tsconfig.json
@@ -0,0 +1,19 @@
+{
+  "extends": "@fern-platform/configs/tsconfig/library.json",
+  "compilerOptions": {
+    "outDir": "./dist",
+    "rootDir": "./src"
+  },
+  "include": ["./src/**/*"],
+  "references": [
+    {
+      "path": "../fern-docs-auth"
+    },
+    {
+      "path": "../fern-docs-utils"
+    },
+    {
+      "path": "../../commons/search-utils"
+    }
+  ]
+}
diff --git a/packages/ui/fern-docs-edge-config/vitest.config.ts b/packages/ui/fern-docs-edge-config/vitest.config.ts
new file mode 100644
index 0000000000..4f772037cd
--- /dev/null
+++ b/packages/ui/fern-docs-edge-config/vitest.config.ts
@@ -0,0 +1,7 @@
+import { defineConfig } from "vitest/config";
+
+export default defineConfig({
+    test: {
+        globals: true,
+    },
+});
diff --git a/packages/ui/docs-bundle/src/server/constants.ts b/packages/ui/fern-docs-utils/src/constants.ts
similarity index 94%
rename from packages/ui/docs-bundle/src/server/constants.ts
rename to packages/ui/fern-docs-utils/src/constants.ts
index aff2a4a349..0692d71ccc 100644
--- a/packages/ui/docs-bundle/src/server/constants.ts
+++ b/packages/ui/fern-docs-utils/src/constants.ts
@@ -4,6 +4,7 @@ export const COOKIE_FERN_DOCS_PREVIEW = "_fern_docs_preview" as const;
 export const COOKIE_FERN_TOKEN = "fern_token" as const;
 export const COOKIE_ACCESS_TOKEN = "access_token" as const; // for api key injection
 export const COOKIE_REFRESH_TOKEN = "refresh_token" as const; // for api key injection
+export const COOKIE_EMAIL = "email" as const; // for api key injection
 export const HEADER_X_FERN_HOST = "x-fern-host" as const;
 
 /**
diff --git a/packages/ui/fern-docs-utils/src/flags.ts b/packages/ui/fern-docs-utils/src/flags.ts
new file mode 100644
index 0000000000..8cf47cde66
--- /dev/null
+++ b/packages/ui/fern-docs-utils/src/flags.ts
@@ -0,0 +1,65 @@
+export interface FeatureFlags {
+    isApiPlaygroundEnabled: boolean;
+    isApiScrollingDisabled: boolean;
+    isWhitelabeled: boolean;
+    isSeoDisabled: boolean;
+    isTocDefaultEnabled: boolean;
+    isSnippetTemplatesEnabled: boolean;
+    isHttpSnippetsEnabled: boolean;
+    isInlineFeedbackEnabled: boolean;
+    isDarkCodeEnabled: boolean;
+    proxyShouldUseAppBuildwithfernCom: boolean;
+    isImageZoomDisabled: boolean;
+    useJavaScriptAsTypeScript: boolean;
+    alwaysEnableJavaScriptFetch: boolean;
+    scrollInContainerEnabled: boolean;
+    useMdxBundler: boolean;
+    isBatchStreamToggleDisabled: boolean;
+    isAuthEnabledInDocs: boolean;
+    isAiChatbotEnabledInPreview: boolean;
+    isAudioFileDownloadSpanSummary: boolean;
+    isDocsLogoTextEnabled: boolean;
+    isAudioExampleInternal: boolean;
+    usesApplicationJsonInFormDataValue: boolean;
+    isBinaryOctetStreamAudioPlayer: boolean;
+    hasVoiceIdPlaygroundForm: boolean;
+    isCohereTheme: boolean;
+    isFileForgeHackEnabled: boolean;
+    is404PageHidden: boolean;
+    isNewSearchExperienceEnabled: boolean;
+    // TODO: remove this after pinecone demo, this is a temporary flag
+    grpcEndpoints: readonly string[];
+}
+
+export const DEFAULT_FEATURE_FLAGS: FeatureFlags = {
+    isApiPlaygroundEnabled: false,
+    isApiScrollingDisabled: false,
+    isWhitelabeled: false,
+    isSeoDisabled: false,
+    isTocDefaultEnabled: false,
+    isSnippetTemplatesEnabled: false,
+    isHttpSnippetsEnabled: false,
+    isInlineFeedbackEnabled: false,
+    isDarkCodeEnabled: false,
+    proxyShouldUseAppBuildwithfernCom: false,
+    isImageZoomDisabled: false,
+    useJavaScriptAsTypeScript: false,
+    alwaysEnableJavaScriptFetch: false,
+    scrollInContainerEnabled: false,
+    useMdxBundler: false,
+    isBatchStreamToggleDisabled: false,
+    isAuthEnabledInDocs: false,
+    isAiChatbotEnabledInPreview: false,
+    isAudioFileDownloadSpanSummary: false,
+    isDocsLogoTextEnabled: false,
+    isAudioExampleInternal: false,
+    usesApplicationJsonInFormDataValue: false,
+    isBinaryOctetStreamAudioPlayer: false,
+    hasVoiceIdPlaygroundForm: false,
+    isCohereTheme: false,
+    isFileForgeHackEnabled: false,
+    is404PageHidden: false,
+    isNewSearchExperienceEnabled: false,
+    // TODO: remove this after pinecone demo, this is a temporary flag
+    grpcEndpoints: [],
+};
diff --git a/packages/ui/fern-docs-utils/src/index.ts b/packages/ui/fern-docs-utils/src/index.ts
index 5ccdc4fa47..3eddb856e9 100644
--- a/packages/ui/fern-docs-utils/src/index.ts
+++ b/packages/ui/fern-docs-utils/src/index.ts
@@ -1 +1,6 @@
+export * from "./constants";
+export * from "./flags";
 export * from "./getRedirectForPath";
+export * from "./isCustomDomain";
+export * from "./isDevelopment";
+export * from "./withoutStaging";
diff --git a/packages/ui/docs-bundle/src/server/isCustomDomain.ts b/packages/ui/fern-docs-utils/src/isCustomDomain.ts
similarity index 100%
rename from packages/ui/docs-bundle/src/server/isCustomDomain.ts
rename to packages/ui/fern-docs-utils/src/isCustomDomain.ts
diff --git a/packages/ui/docs-bundle/src/server/isDevelopment.ts b/packages/ui/fern-docs-utils/src/isDevelopment.ts
similarity index 100%
rename from packages/ui/docs-bundle/src/server/isDevelopment.ts
rename to packages/ui/fern-docs-utils/src/isDevelopment.ts
diff --git a/packages/ui/docs-bundle/src/server/withoutStaging.ts b/packages/ui/fern-docs-utils/src/withoutStaging.ts
similarity index 100%
rename from packages/ui/docs-bundle/src/server/withoutStaging.ts
rename to packages/ui/fern-docs-utils/src/withoutStaging.ts
diff --git a/packages/ui/fern-docs-utils/tsconfig.json b/packages/ui/fern-docs-utils/tsconfig.json
index 54cc57ae2a..dc07764da5 100644
--- a/packages/ui/fern-docs-utils/tsconfig.json
+++ b/packages/ui/fern-docs-utils/tsconfig.json
@@ -7,7 +7,7 @@
   "include": ["./src/**/*"],
   "references": [
     {
-      "path": "../../fdr-sdk"
+      "path": "../../fdr-sdk/tsconfig.build.json"
     }
   ]
 }
diff --git a/packages/ui/local-preview-bundle/src/utils/getDocsPageProps.ts b/packages/ui/local-preview-bundle/src/utils/getDocsPageProps.ts
index 46c77ecec5..f3c42b1b70 100644
--- a/packages/ui/local-preview-bundle/src/utils/getDocsPageProps.ts
+++ b/packages/ui/local-preview-bundle/src/utils/getDocsPageProps.ts
@@ -2,11 +2,9 @@ import { FdrAPI, type DocsV2Read } from "@fern-api/fdr-sdk/client/types";
 import * as FernNavigation from "@fern-api/fdr-sdk/navigation";
 import { visitDiscriminatedUnion } from "@fern-ui/core-utils";
 import { SidebarTab } from "@fern-ui/fdr-utils";
-import { getRedirectForPath } from "@fern-ui/fern-docs-utils";
+import { DEFAULT_FEATURE_FLAGS, getRedirectForPath, type FeatureFlags } from "@fern-ui/fern-docs-utils";
 import {
-    DEFAULT_FEATURE_FLAGS,
     DocsPage,
-    FeatureFlags,
     getGitHubInfo,
     getGitHubRepo,
     getSeoProps,
diff --git a/packages/ui/local-preview-bundle/tsconfig.json b/packages/ui/local-preview-bundle/tsconfig.json
index 10c55d2e59..4ff41a8ba9 100644
--- a/packages/ui/local-preview-bundle/tsconfig.json
+++ b/packages/ui/local-preview-bundle/tsconfig.json
@@ -14,7 +14,7 @@
       "path": "../../commons/fdr-utils"
     },
     {
-      "path": "../../fdr-sdk"
+      "path": "../../fdr-sdk/tsconfig.build.json"
     },
     {
       "path": "../app/tsconfig.build.json"
diff --git a/pnpm-lock.yaml b/pnpm-lock.yaml
index 0815529b83..88d0efc938 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:*
@@ -1070,6 +1067,9 @@ importers:
       '@fern-ui/fdr-utils':
         specifier: workspace:*
         version: link:../../commons/fdr-utils
+      '@fern-ui/fern-docs-utils':
+        specifier: workspace:*
+        version: link:../fern-docs-utils
       '@fern-ui/loadable':
         specifier: workspace:*
         version: link:../../commons/loadable
@@ -1311,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)
@@ -1772,6 +1775,12 @@ 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-config':
+        specifier: workspace:*
+        version: link:../fern-docs-edge-config
       '@fern-ui/fern-docs-utils':
         specifier: workspace:*
         version: link:../fern-docs-utils
@@ -1784,9 +1793,6 @@ importers:
       '@vercel/analytics':
         specifier: ^1.3.1
         version: 1.3.1(@fern-api/next@14.2.9-fork.0(@babel/core@7.24.5)(@opentelemetry/api@1.9.0)(@playwright/test@1.47.1)(react-dom@18.3.1(react@18.3.1))(react@18.3.1)(sass@1.77.0))(react@18.3.1)
-      '@vercel/edge-config':
-        specifier: ^1.1.0
-        version: 1.1.0(@opentelemetry/api@1.9.0)(typescript@5.4.3)
       '@vercel/kv':
         specifier: ^2.0.0
         version: 2.0.0
@@ -2085,6 +2091,89 @@ 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-config:
+    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
+      '@vercel/edge-config':
+        specifier: ^1.1.0
+        version: 1.1.0(@opentelemetry/api@1.9.0)(typescript@5.4.3)
+      url-join:
+        specifier: 5.0.0
+        version: 5.0.0
+      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
+      '@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-utils:
     dependencies:
       '@fern-api/fdr-sdk':
@@ -16583,10 +16672,10 @@ snapshots:
       '@aws-crypto/sha1-browser': 3.0.0
       '@aws-crypto/sha256-browser': 3.0.0
       '@aws-crypto/sha256-js': 3.0.0
-      '@aws-sdk/client-sso-oidc': 3.572.0
-      '@aws-sdk/client-sts': 3.572.0(@aws-sdk/client-sso-oidc@3.572.0)
+      '@aws-sdk/client-sso-oidc': 3.572.0(@aws-sdk/client-sts@3.572.0)
+      '@aws-sdk/client-sts': 3.572.0
       '@aws-sdk/core': 3.572.0
-      '@aws-sdk/credential-provider-node': 3.572.0(@aws-sdk/client-sso-oidc@3.572.0)(@aws-sdk/client-sts@3.572.0(@aws-sdk/client-sso-oidc@3.572.0))
+      '@aws-sdk/credential-provider-node': 3.572.0(@aws-sdk/client-sso-oidc@3.572.0(@aws-sdk/client-sts@3.572.0))(@aws-sdk/client-sts@3.572.0)
       '@aws-sdk/middleware-bucket-endpoint': 3.568.0
       '@aws-sdk/middleware-expect-continue': 3.572.0
       '@aws-sdk/middleware-flexible-checksums': 3.572.0
@@ -16647,7 +16736,7 @@ snapshots:
       '@aws-crypto/sha256-js': 3.0.0
       '@aws-sdk/client-sts': 3.572.0
       '@aws-sdk/core': 3.572.0
-      '@aws-sdk/credential-provider-node': 3.572.0(@aws-sdk/client-sso-oidc@3.572.0)(@aws-sdk/client-sts@3.572.0(@aws-sdk/client-sso-oidc@3.572.0))
+      '@aws-sdk/credential-provider-node': 3.572.0(@aws-sdk/client-sso-oidc@3.572.0)(@aws-sdk/client-sts@3.572.0)
       '@aws-sdk/middleware-host-header': 3.567.0
       '@aws-sdk/middleware-logger': 3.568.0
       '@aws-sdk/middleware-recursion-detection': 3.567.0
@@ -16781,52 +16870,7 @@ snapshots:
       '@aws-crypto/sha256-js': 3.0.0
       '@aws-sdk/client-sso-oidc': 3.572.0
       '@aws-sdk/core': 3.572.0
-      '@aws-sdk/credential-provider-node': 3.572.0(@aws-sdk/client-sso-oidc@3.572.0)(@aws-sdk/client-sts@3.572.0)
-      '@aws-sdk/middleware-host-header': 3.567.0
-      '@aws-sdk/middleware-logger': 3.568.0
-      '@aws-sdk/middleware-recursion-detection': 3.567.0
-      '@aws-sdk/middleware-user-agent': 3.572.0
-      '@aws-sdk/region-config-resolver': 3.572.0
-      '@aws-sdk/types': 3.567.0
-      '@aws-sdk/util-endpoints': 3.572.0
-      '@aws-sdk/util-user-agent-browser': 3.567.0
-      '@aws-sdk/util-user-agent-node': 3.568.0
-      '@smithy/config-resolver': 2.2.0
-      '@smithy/core': 1.4.2
-      '@smithy/fetch-http-handler': 2.5.0
-      '@smithy/hash-node': 2.2.0
-      '@smithy/invalid-dependency': 2.2.0
-      '@smithy/middleware-content-length': 2.2.0
-      '@smithy/middleware-endpoint': 2.5.1
-      '@smithy/middleware-retry': 2.3.1
-      '@smithy/middleware-serde': 2.3.0
-      '@smithy/middleware-stack': 2.2.0
-      '@smithy/node-config-provider': 2.3.0
-      '@smithy/node-http-handler': 2.5.0
-      '@smithy/protocol-http': 3.3.0
-      '@smithy/smithy-client': 2.5.1
-      '@smithy/types': 2.12.0
-      '@smithy/url-parser': 2.2.0
-      '@smithy/util-base64': 2.3.0
-      '@smithy/util-body-length-browser': 2.2.0
-      '@smithy/util-body-length-node': 2.3.0
-      '@smithy/util-defaults-mode-browser': 2.2.1
-      '@smithy/util-defaults-mode-node': 2.3.1
-      '@smithy/util-endpoints': 1.2.0
-      '@smithy/util-middleware': 2.2.0
-      '@smithy/util-retry': 2.2.0
-      '@smithy/util-utf8': 2.3.0
-      tslib: 2.6.2
-    transitivePeerDependencies:
-      - aws-crt
-
-  '@aws-sdk/client-sts@3.572.0(@aws-sdk/client-sso-oidc@3.572.0)':
-    dependencies:
-      '@aws-crypto/sha256-browser': 3.0.0
-      '@aws-crypto/sha256-js': 3.0.0
-      '@aws-sdk/client-sso-oidc': 3.572.0
-      '@aws-sdk/core': 3.572.0
-      '@aws-sdk/credential-provider-node': 3.572.0(@aws-sdk/client-sso-oidc@3.572.0)(@aws-sdk/client-sts@3.572.0(@aws-sdk/client-sso-oidc@3.572.0))
+      '@aws-sdk/credential-provider-node': 3.572.0(@aws-sdk/client-sso-oidc@3.572.0(@aws-sdk/client-sts@3.572.0))(@aws-sdk/client-sts@3.572.0)
       '@aws-sdk/middleware-host-header': 3.567.0
       '@aws-sdk/middleware-logger': 3.568.0
       '@aws-sdk/middleware-recursion-detection': 3.567.0
@@ -16863,7 +16907,6 @@ snapshots:
       '@smithy/util-utf8': 2.3.0
       tslib: 2.6.2
     transitivePeerDependencies:
-      - '@aws-sdk/client-sso-oidc'
       - aws-crt
 
   '@aws-sdk/core@3.572.0':
@@ -16912,23 +16955,6 @@ snapshots:
       - '@aws-sdk/client-sso-oidc'
       - aws-crt
 
-  '@aws-sdk/credential-provider-ini@3.572.0(@aws-sdk/client-sso-oidc@3.572.0)(@aws-sdk/client-sts@3.572.0(@aws-sdk/client-sso-oidc@3.572.0))':
-    dependencies:
-      '@aws-sdk/client-sts': 3.572.0(@aws-sdk/client-sso-oidc@3.572.0)
-      '@aws-sdk/credential-provider-env': 3.568.0
-      '@aws-sdk/credential-provider-process': 3.572.0
-      '@aws-sdk/credential-provider-sso': 3.572.0(@aws-sdk/client-sso-oidc@3.572.0)
-      '@aws-sdk/credential-provider-web-identity': 3.568.0(@aws-sdk/client-sts@3.572.0)
-      '@aws-sdk/types': 3.567.0
-      '@smithy/credential-provider-imds': 2.3.0
-      '@smithy/property-provider': 2.2.0
-      '@smithy/shared-ini-file-loader': 2.4.0
-      '@smithy/types': 2.12.0
-      tslib: 2.6.2
-    transitivePeerDependencies:
-      - '@aws-sdk/client-sso-oidc'
-      - aws-crt
-
   '@aws-sdk/credential-provider-ini@3.572.0(@aws-sdk/client-sso-oidc@3.572.0)(@aws-sdk/client-sts@3.572.0)':
     dependencies:
       '@aws-sdk/client-sts': 3.572.0
@@ -16965,25 +16991,6 @@ snapshots:
       - '@aws-sdk/client-sts'
       - aws-crt
 
-  '@aws-sdk/credential-provider-node@3.572.0(@aws-sdk/client-sso-oidc@3.572.0)(@aws-sdk/client-sts@3.572.0(@aws-sdk/client-sso-oidc@3.572.0))':
-    dependencies:
-      '@aws-sdk/credential-provider-env': 3.568.0
-      '@aws-sdk/credential-provider-http': 3.568.0
-      '@aws-sdk/credential-provider-ini': 3.572.0(@aws-sdk/client-sso-oidc@3.572.0)(@aws-sdk/client-sts@3.572.0(@aws-sdk/client-sso-oidc@3.572.0))
-      '@aws-sdk/credential-provider-process': 3.572.0
-      '@aws-sdk/credential-provider-sso': 3.572.0(@aws-sdk/client-sso-oidc@3.572.0)
-      '@aws-sdk/credential-provider-web-identity': 3.568.0(@aws-sdk/client-sts@3.572.0)
-      '@aws-sdk/types': 3.567.0
-      '@smithy/credential-provider-imds': 2.3.0
-      '@smithy/property-provider': 2.2.0
-      '@smithy/shared-ini-file-loader': 2.4.0
-      '@smithy/types': 2.12.0
-      tslib: 2.6.2
-    transitivePeerDependencies:
-      - '@aws-sdk/client-sso-oidc'
-      - '@aws-sdk/client-sts'
-      - aws-crt
-
   '@aws-sdk/credential-provider-node@3.572.0(@aws-sdk/client-sso-oidc@3.572.0)(@aws-sdk/client-sts@3.572.0)':
     dependencies:
       '@aws-sdk/credential-provider-env': 3.568.0
@@ -17039,7 +17046,7 @@ snapshots:
 
   '@aws-sdk/credential-provider-web-identity@3.568.0(@aws-sdk/client-sts@3.572.0)':
     dependencies:
-      '@aws-sdk/client-sts': 3.572.0(@aws-sdk/client-sso-oidc@3.572.0)
+      '@aws-sdk/client-sts': 3.572.0
       '@aws-sdk/types': 3.567.0
       '@smithy/property-provider': 2.2.0
       '@smithy/types': 2.12.0