@stainless-api/docs 0.1.0-beta.5 → 0.1.0-beta.50

package/plugin/vendor/templates/go.md

@@ -31,7 +31,7 @@ SDK_ReleasePleaseMarkdownBlockEnd
 
 ## Requirements
 
-This library requires Go 1.18+.
+This library requires Go 1.22+.
 
 ## Usage
 

package/plugin/vendor/templates/python.md

@@ -242,7 +242,7 @@ print(SDK_PythonPackageImportName.__version__)
 
 ## Requirements
 
-Python 3.8 or higher.
+Python 3.9 or higher.
 
 ## Contributing
 

package/resolveSrcFile.ts

@@ -0,0 +1,10 @@
+import { join } from 'path';
+const dirName = import.meta.dirname;
+/**
+ * Resolve a source file path relative to the stl-starlight package.
+ * @param projectPath - The relative path to the source file.
+ * @returns The absolute path to the source file.
+ */
+export function resolveSrcFile(...projectPath: string[]) {
+  return join(dirName, ...projectPath);
+}

package/scripts/vendor_deps.ts

@@ -3,7 +3,7 @@ import fs from 'fs';
 import path from 'path';
 import { z } from 'zod';
 
-import { Languages } from '@stainless-api/docs-ui/src/routing';
+import { Languages } from '@stainless-api/docs-ui/routing';
 
 const DEPS_DIR = './plugin/vendor';
 

package/shared/getSharedLogger.ts

@@ -0,0 +1,15 @@
+import { AstroIntegrationLogger } from 'astro';
+
+let sharedLogger: AstroIntegrationLogger | null = null;
+
+// This is probably temporary, but it's a quick way to share a logger between our many integrations
+// we want to share a logger so they have the same "stainless" label
+
+export function setSharedLogger(logger: AstroIntegrationLogger) {
+  sharedLogger = logger;
+}
+
+// a fallback is probably not required, but it's a good safeguard in case we somehow call a logger before the shared logger is set
+export function getSharedLogger({ fallback }: { fallback: AstroIntegrationLogger }) {
+  return sharedLogger ?? fallback;
+}

package/shared/terminalUtils.ts

@@ -0,0 +1,3 @@
+export function bold(text: string) {
+  return `\x1b[1m${text}\x1b[0m`;
+}

package/src/content.config.ts

@@ -0,0 +1,9 @@
+// Stub here so that `astro sync` generates the correct types for the `docs` content collection
+
+import { defineCollection } from 'astro:content';
+import { docsLoader } from '@astrojs/starlight/loaders';
+import { docsSchema } from '@astrojs/starlight/schema';
+
+export const collections = {
+  docs: defineCollection({ loader: docsLoader(), schema: docsSchema() }),
+};

package/stl-docs/components/AIDropdown.tsx

@@ -0,0 +1,63 @@
+import { DropdownButton } from '@stainless-api/ui-primitives';
+import { CopyIcon } from 'lucide-react';
+import { ChatGPTIcon } from './icons/chat-gpt';
+import { ClaudeIcon } from './icons/claude';
+import { GeminiIcon } from './icons/gemini';
+import { MarkdownIcon } from './icons/markdown';
+import { CursorIcon } from './icons/cursor';
+import React from 'react';
+
+import { getAIDropdownOptions, type DropdownIcon } from '../../plugin/globalJs/ai-dropdown-options';
+
+const iconMap: { [key in DropdownIcon]: React.ReactNode } = {
+  markdown: <MarkdownIcon />,
+  copy: <CopyIcon size={16} />,
+  claude: <ClaudeIcon />,
+  chatgpt: <ChatGPTIcon />,
+  gemini: <GeminiIcon />,
+  cursor: <CursorIcon />,
+};
+
+const { primaryAction, groups } = getAIDropdownOptions();
+
+function ItemLabel({ label }: { label: string[] }) {
+  if (label.length > 1) {
+    return (
+      <DropdownButton.MenuItemText subtle>
+        {label[0]}
+        <strong>{label[1]}</strong>
+      </DropdownButton.MenuItemText>
+    );
+  }
+
+  return (
+    <DropdownButton.MenuItemText>
+      <strong>{label[0]}</strong>
+    </DropdownButton.MenuItemText>
+  );
+}
+
+export function AIDropdown() {
+  return (
+    <DropdownButton data-dropdown-id="ai-dropdown-button">
+      <DropdownButton.PrimaryAction>
+        {iconMap[primaryAction.icon]}
+        <DropdownButton.PrimaryActionText>{primaryAction.label}</DropdownButton.PrimaryActionText>
+      </DropdownButton.PrimaryAction>
+      <DropdownButton.Trigger />
+      <DropdownButton.Menu>
+        {groups.map((group) => (
+          <React.Fragment key={group.reactKey}>
+            {group.options.map((option) => (
+              <DropdownButton.MenuItem value={option.id} isExternalLink={option.external} key={option.id}>
+                <DropdownButton.Icon>{iconMap[option.icon]}</DropdownButton.Icon>
+                <ItemLabel label={option.label} />
+              </DropdownButton.MenuItem>
+            ))}
+            {group.isLast ? null : <hr />}
+          </React.Fragment>
+        ))}
+      </DropdownButton.Menu>
+    </DropdownButton>
+  );
+}

package/stl-docs/components/{content-panel/ContentBreadcrumbs.tsx → ContentBreadcrumbs.tsx}

@@ -1,6 +1,6 @@
 import { ChevronRight } from 'lucide-react';
-import { Join } from '@stainless-api/docs-ui/src/components';
-import style from '@stainless-api/docs-ui/src/style';
+import { Join } from '@stainless-api/docs-ui/components';
+import style from '@stainless-api/docs-ui/style';
 import type { StarlightRouteData } from '@astrojs/starlight/route-data';
 
 type Breadcrumb = { title: string; href: string };

package/stl-docs/components/Head.astro

@@ -0,0 +1,16 @@
+---
+import Default from '@astrojs/starlight/components/Head.astro';
+
+// TODO: for users who are overriding the font stack in their own styles, how can we know that and
+// preload their font instead of ours?
+import geistPath from '../../assets/fonts/geist/geist-latin.woff2';
+---
+
+<Default />
+
+<link rel="preload" as="font" type="font/woff2" crossorigin="anonymous" href={geistPath} />
+
+<script>
+  import { wireAIDropdown } from '../../plugin/globalJs/ai-dropdown-options.ts';
+  wireAIDropdown();
+</script>

package/stl-docs/components/Header.astro

@@ -1,7 +1,7 @@
 ---
 import config from 'virtual:starlight/user-config';
 
-import { HEADER_LAYOUT } from 'virtual:stl-stl-starlight-virtual-module';
+import { HEADER_LAYOUT } from 'virtual:stl-docs-virtual-module';
 import DefaultHeader from './headers/DefaultHeader.astro';
 import StackedHeader from './headers/StackedHeader.astro';
 
@@ -23,12 +23,9 @@ const shouldRenderSearch = !!(
 
 <style is:global>
   @layer starlight.core {
-    header {
-      border-color: var(--sl-color-hairline);
-
-      &.header {
-        border-bottom-color: var(--sl-color-hairline);
-      }
+    header,
+    header.header {
+      border-color: var(--stl-color-border-faint);
     }
 
     .header {
@@ -44,9 +41,10 @@ const shouldRenderSearch = !!(
       overflow: clip;
       /* Avoid clipping focus ring around link inside title wrapper. */
       padding: 0.25rem;
-      padding-left: 0;
       margin: -0.25rem;
       min-width: 0;
+      /* Max height should match button height */
+      max-height: 32px;
     }
 
     .default-tabs-container {

package/stl-docs/components/PageTitle.astro

@@ -0,0 +1,82 @@
+---
+import {
+  ENABLE_PROSE_MARKDOWN_RENDERING,
+  ENABLE_CONTEXT_MENU,
+  RENDER_PAGE_DESCRIPTIONS,
+} from 'virtual:stl-docs-virtual-module';
+import type { StarlightRouteData } from '@astrojs/starlight/route-data';
+import Default from '@astrojs/starlight/components/PageTitle.astro';
+import { ContentBreadcrumbs } from './ContentBreadcrumbs';
+import { AIDropdown } from './AIDropdown';
+
+function sidebarHasEntry(sidebarEntry: StarlightRouteData['sidebar'], currentPath?: string) {
+  if (!currentPath) return false;
+  const normalizePath = (path: string) => {
+    return path.endsWith('/') ? path : path + '/';
+  };
+
+  const normalizedCurrent = normalizePath(currentPath);
+
+  for (const entry of sidebarEntry) {
+    if (entry.type === 'link') {
+      const normalizedHref = normalizePath(entry.href);
+      if (normalizedHref === normalizedCurrent) {
+        return true;
+      }
+    } else if (entry.type === 'group') {
+      const hasInGroup = sidebarHasEntry(entry.entries, currentPath);
+      if (hasInGroup) return true;
+    }
+  }
+
+  return false;
+}
+
+const skipRenderingStarlightTitle = Astro.locals._stlStarlightPage?.skipRenderingStarlightTitle ?? false;
+
+const hasMarkdownRoute = Astro.locals._stlStarlightPage?.hasMarkdownRoute ?? true;
+// ⬆️ hopefully this is temporary. But for now, we don't have a markdown representation of the API reference overview page
+
+const currentPath = Astro.url.pathname;
+
+function shouldShowAIDropdown() {
+  // Hide if there is no associated sidebar entry. This applies to pages like the 404 page.
+  const hasSidebarEntry = sidebarHasEntry(Astro.locals.starlightRoute.sidebar, currentPath);
+  if (!hasSidebarEntry) return false;
+
+  if (!hasMarkdownRoute) return false;
+
+  if (!ENABLE_PROSE_MARKDOWN_RENDERING) return false;
+  if (!ENABLE_CONTEXT_MENU) return false;
+
+  return true;
+}
+
+const showAIDropdown = shouldShowAIDropdown();
+
+const description = Astro.locals.starlightRoute.entry.data.description;
+const shouldRenderDescription = RENDER_PAGE_DESCRIPTIONS;
+---
+
+{
+  skipRenderingStarlightTitle ? null : (
+    <>
+      <div class="stl-ui-not-prose stl-page-nav-container stl-prose-page-nav-container">
+        <ContentBreadcrumbs currentPath={currentPath} sidebarEntry={Astro.locals.starlightRoute.sidebar} />
+        {showAIDropdown && <AIDropdown />}
+      </div>
+      <Default />
+      {!!description && shouldRenderDescription && <p class="description">{description}</p>}
+    </>
+  )
+}
+
+<style>
+  p.description {
+    font-size: var(--stl-typography-scale-xl);
+    font-weight: 400;
+    line-height: 1.4;
+    letter-spacing: -0.02em;
+    margin-top: 0.4em;
+  }
+</style>

package/stl-docs/components/TableOfContents.astro

@@ -0,0 +1,34 @@
+---
+import Default from '@astrojs/starlight/components/TableOfContents.astro';
+---
+
+<Default {...Astro.props} />
+
+<script>
+  /**
+   * Starlight will mark TOC links as active as they are scrolled to. However, it does not mark
+   * a link as active if it cannot be scrolled to on the page. This happens for small sections at
+   * the bottom of the page which are not large enough to scroll to the top of the viewport.
+   *
+   * This script will mark a section as active if its corresponding hash is in the URL, even if
+   * it cannot be scrolled to.
+   */
+  document.addEventListener('DOMContentLoaded', () => {
+    const tocLinks = document.querySelectorAll('starlight-toc a') as NodeListOf<HTMLAnchorElement>;
+
+    function updateAriaCurrent() {
+      const current = decodeURIComponent(window.location.hash || '');
+      tocLinks.forEach((link) => {
+        const linkHash = decodeURIComponent(link.hash || link.getAttribute('href') || '');
+        if (linkHash === current) {
+          link.setAttribute('aria-current', 'true');
+        } else {
+          link.removeAttribute('aria-current');
+        }
+      });
+    }
+
+    window.addEventListener('hashchange', updateAriaCurrent);
+    updateAriaCurrent();
+  });
+</script>

package/stl-docs/components/ThemeSelect.astro

@@ -1,94 +1,133 @@
 ---
-import {
-  Dropdown,
-  DropdownTrigger,
-  DropdownMenu,
-  DropdownItem,
-} from '@stainless-api/docs-ui/src/components/dropdown';
+import { Dropdown } from '@stainless-api/docs/components';
+import { ChevronsUpDownIcon } from 'lucide-react';
 
 const options = [
   {
     value: 'auto',
     label: 'Auto',
-    icon: `<svg
-  width="24"
-  height="24"
-  viewBox="0 0 24 24"
-  fill="none"
-  xmlns="http://www.w3.org/2000/svg"
->
-  <path
+    icon: `
+      <svg
+        width="24"
+        height="24"
+        viewBox="0 0 24 24"
+        fill="none"
+        xmlns="http://www.w3.org/2000/svg"
+      >
+        <path
           d="M21 5C21 4.44771 20.5523 4 20 4H4C3.44772 4 3 4.44772 3 5V15C3 15.5523 3.44772 16 4 16H20C20.5523 16 21 15.5523 21 15V5ZM23 15C23 16.6569 21.6569 18 20 18H13V20H16C16.5523 20 17 20.4477 17 21C17 21.5523 16.5523 22 16 22H8C7.44772 22 7 21.5523 7 21C7 20.4477 7.44772 20 8 20H11V18H4C2.34315 18 1 16.6569 1 15V5C1 3.34315 2.34315 2 4 2H20C21.6569 2 23 3.34315 23 5V15Z"
-          fill="#262626"></path></svg
->`,
+          fill="var(--stl-color-foreground)"
+        ></path>
+      </svg>
+    `,
   },
   {
     value: 'light',
     label: 'Light',
-    icon: `<svg
-  width="24"
-  height="24"
-  viewBox="0 0 24 24"
-  fill="none"
-  xmlns="http://www.w3.org/2000/svg"
->
-  <path
+    icon: `
+      <svg
+        width="24"
+        height="24"
+        viewBox="0 0 24 24"
+        fill="none"
+        xmlns="http://www.w3.org/2000/svg"
+      >
+        <path
           d="M11 22V20C11 19.4477 11.4477 19 12 19C12.5523 19 13 19.4477 13 20V22C13 22.5523 12.5523 23 12 23C11.4477 23 11 22.5523 11 22ZM5.63281 16.9531C6.02332 16.5626 6.65635 16.5626 7.04688 16.9531C7.4374 17.3436 7.4374 17.9767 7.04688 18.3672L5.6377 19.7773C5.24717 20.1679 4.61318 20.1679 4.22266 19.7773C3.83232 19.387 3.83259 18.7538 4.22266 18.3633L5.63281 16.9531ZM16.9531 16.9531C17.3191 16.5871 17.8982 16.5637 18.291 16.8838L18.3672 16.9531L19.7773 18.3633L19.8457 18.4385C20.1663 18.8313 20.1435 19.4111 19.7773 19.7773C19.4113 20.1433 18.8322 20.1658 18.4395 19.8457L18.3633 19.7773L16.9531 18.3672L16.8848 18.291C16.5643 17.8983 16.5871 17.3193 16.9531 16.9531ZM15 12C15 10.3431 13.6569 9 12 9C10.3431 9 9 10.3431 9 12C9 13.6569 10.3431 15 12 15C13.6569 15 15 13.6569 15 12ZM4 11C4.55228 11 5 11.4477 5 12C5 12.5523 4.55228 13 4 13H2C1.44772 13 1 12.5523 1 12C1 11.4477 1.44772 11 2 11H4ZM22 11C22.5523 11 23 11.4477 23 12C23 12.5523 22.5523 13 22 13H20C19.4477 13 19 12.5523 19 12C19 11.4477 19.4477 11 20 11H22ZM7.04688 5.63281C7.4374 6.02334 7.4374 6.65635 7.04688 7.04688C6.65634 7.43728 6.0233 7.43736 5.63281 7.04688L4.22266 5.63672L5.6377 4.22266L7.04688 5.63281ZM18.3633 4.22266C18.7538 3.83235 19.3869 3.83224 19.7773 4.22266C20.1678 4.61311 20.1677 5.24618 19.7773 5.63672L18.3672 7.04688C17.9767 7.4374 17.3436 7.4374 16.9531 7.04688C16.5626 6.65635 16.5626 6.02332 16.9531 5.63281L18.3633 4.22266ZM4.22266 4.22266C4.61318 3.83213 5.24717 3.83213 5.6377 4.22266L4.22266 5.63672C3.83254 5.24621 3.83238 4.61307 4.22266 4.22266ZM11 4V2C11 1.44772 11.4477 1 12 1C12.5523 1 13 1.44772 13 2V4C13 4.55228 12.5523 5 12 5C11.4477 5 11 4.55228 11 4ZM17 12C17 14.7614 14.7614 17 12 17C9.23858 17 7 14.7614 7 12C7 9.23858 9.23858 7 12 7C14.7614 7 17 9.23858 17 12Z"
-          fill="#262626"></path></svg
->`,
+          fill="var(--stl-color-foreground)"
+        ></path>
+      </svg>
+    `,
   },
   {
     value: 'dark',
     label: 'Dark',
-    icon: `<svg
-  width="24"
-  height="24"
-  viewBox="0 0 24 24"
-  fill="none"
-  xmlns="http://www.w3.org/2000/svg"
->
-  <path
-    d="M9.13569 7.5C9.13569 6.37669 9.39314 5.28007 9.87397 4.28809C9.05381 4.51419 8.27063 4.87087 7.55561 5.34863C6.24009 6.22764 5.21483 7.47676 4.60932 8.93848C4.00386 10.4002 3.84465 12.0088 4.15327 13.5605C4.46195 15.1124 5.2239 16.5384 6.34272 17.6572C7.46154 18.776 8.88756 19.538 10.4394 19.8467C11.9912 20.1553 13.5997 19.9961 15.0615 19.3906C16.5232 18.7851 17.7723 17.7599 18.6513 16.4443C19.1292 15.7291 19.4848 14.9454 19.7109 14.125C18.719 14.6057 17.6231 14.8643 16.4999 14.8643C14.5469 14.8643 12.6739 14.088 11.2929 12.707C9.91191 11.326 9.13569 9.45304 9.13569 7.5ZM11.1357 7.5C11.1357 8.92261 11.701 10.287 12.707 11.293C13.7129 12.2989 15.0773 12.8643 16.4999 12.8643C17.9226 12.8643 19.287 12.2989 20.2929 11.293C20.5789 11.007 21.0091 10.9214 21.3828 11.0762C21.7564 11.231 22 11.5956 22 12C22 13.9778 21.4132 15.9112 20.3144 17.5557C19.2156 19.2001 17.6543 20.4824 15.8271 21.2393C13.9998 21.9961 11.9886 22.1935 10.0488 21.8076C8.10909 21.4217 6.32711 20.4697 4.92866 19.0713C3.53021 17.6728 2.57822 15.8909 2.19233 13.9512C1.80648 12.0114 2.00382 10.0001 2.76069 8.17285C3.51756 6.34567 4.79987 4.78434 6.44429 3.68555C8.08878 2.58673 10.0221 2 11.9999 2C12.4044 2 12.769 2.24359 12.9238 2.61719C13.0786 2.99086 12.993 3.42103 12.707 3.70703C11.701 4.71297 11.1357 6.07739 11.1357 7.5Z"
-    fill="#262626"></path></svg
->`,
+    icon: `
+      <svg
+        width="24"
+        height="24"
+        viewBox="0 0 24 24"
+        fill="none"
+        xmlns="http://www.w3.org/2000/svg"
+      >
+        <path
+          d="M9.13569 7.5C9.13569 6.37669 9.39314 5.28007 9.87397 4.28809C9.05381 4.51419 8.27063 4.87087 7.55561 5.34863C6.24009 6.22764 5.21483 7.47676 4.60932 8.93848C4.00386 10.4002 3.84465 12.0088 4.15327 13.5605C4.46195 15.1124 5.2239 16.5384 6.34272 17.6572C7.46154 18.776 8.88756 19.538 10.4394 19.8467C11.9912 20.1553 13.5997 19.9961 15.0615 19.3906C16.5232 18.7851 17.7723 17.7599 18.6513 16.4443C19.1292 15.7291 19.4848 14.9454 19.7109 14.125C18.719 14.6057 17.6231 14.8643 16.4999 14.8643C14.5469 14.8643 12.6739 14.088 11.2929 12.707C9.91191 11.326 9.13569 9.45304 9.13569 7.5ZM11.1357 7.5C11.1357 8.92261 11.701 10.287 12.707 11.293C13.7129 12.2989 15.0773 12.8643 16.4999 12.8643C17.9226 12.8643 19.287 12.2989 20.2929 11.293C20.5789 11.007 21.0091 10.9214 21.3828 11.0762C21.7564 11.231 22 11.5956 22 12C22 13.9778 21.4132 15.9112 20.3144 17.5557C19.2156 19.2001 17.6543 20.4824 15.8271 21.2393C13.9998 21.9961 11.9886 22.1935 10.0488 21.8076C8.10909 21.4217 6.32711 20.4697 4.92866 19.0713C3.53021 17.6728 2.57822 15.8909 2.19233 13.9512C1.80648 12.0114 2.00382 10.0001 2.76069 8.17285C3.51756 6.34567 4.79987 4.78434 6.44429 3.68555C8.08878 2.58673 10.0221 2 11.9999 2C12.4044 2 12.769 2.24359 12.9238 2.61719C13.0786 2.99086 12.993 3.42103 12.707 3.70703C11.701 4.71297 11.1357 6.07739 11.1357 7.5Z"
+          fill="var(--stl-color-foreground)"
+        ></path>
+      </svg>
+    `,
   },
 ];
 ---
 
-<div class="stldocs-root theme-select-root">
-  <Dropdown className="theme-select-container" data-theme-select>
-    <DropdownTrigger withChevron>
-      <span class="stl-dropdown-icon">
-        {options.map(({ value, label, icon }) => <Fragment set:html={icon} />)}
-      </span>
-    </DropdownTrigger>
-    <DropdownMenu role="listbox">
-      {
-        options.map(({ value, label, icon }) => (
-          <DropdownItem role="option" value={value}>
-            <div>
-              <span class="stl-dropdown-icon">
-                <Fragment set:html={icon} />
-              </span>
-              <span class="stl-dropdown-label">{label}</span>
-            </div>
-          </DropdownItem>
-        ))
-      }
-    </DropdownMenu>
-  </Dropdown>
-</div>
+<Dropdown data-theme-select>
+  <Dropdown.Trigger>
+    <Dropdown.TriggerSelectedItem>
+      <span class="stl-dropdown-icon"></span>
+    </Dropdown.TriggerSelectedItem>
+    <Dropdown.TriggerIcon>
+      <ChevronsUpDownIcon size={16} />
+    </Dropdown.TriggerIcon>
+  </Dropdown.Trigger>
+  <Dropdown.Menu>
+    {
+      options.map(({ value, label, icon }) => (
+        <Dropdown.MenuItem value={value}>
+          <Dropdown.Icon>
+            <Fragment set:html={icon} />
+          </Dropdown.Icon>
+          <Dropdown.MenuItemText>{label}</Dropdown.MenuItemText>
+          <Dropdown.MenuItemTemplate>
+            <span class="stl-dropdown-icon">
+              <Fragment set:html={icon} />
+            </span>
+          </Dropdown.MenuItemTemplate>
+        </Dropdown.MenuItem>
+      ))
+    }
+  </Dropdown.Menu>
+</Dropdown>
 
-{/* Inlined to avoid FOUC. Uses global scope from `ThemeProvider.astro` */}
+{/* Inlined to avoid FOUC. All theme selects are initialized here */}
 <script is:inline>
-  document.addEventListener('astro:page-load', () => {
-    StarlightThemeProvider.updatePickers();
-  });
-  StarlightThemeProvider.updatePickers();
+  window.didInitThemePickers = window.didInitThemePickers ?? false;
+
+  // Only run once.
+  if (!window.didInitThemePickers) {
+    window.didInitThemePickers = true;
+
+    // The stored theme will be either 'auto', 'light', 'dark' or null.
+    const storedTheme = typeof localStorage !== 'undefined' && localStorage.getItem('starlight-theme');
+
+    // This theme is either 'light' or 'dark'. It's used for setting the data-theme attribute.
+    const theme =
+      !storedTheme || storedTheme === 'auto'
+        ? window.matchMedia('(prefers-color-scheme: light)').matches
+          ? 'light'
+          : 'dark'
+        : storedTheme;
+
+    document.documentElement.dataset.theme = theme === 'light' ? 'light' : 'dark';
+
+    const themeSelects = document.querySelectorAll('[data-theme-select]');
+    const selectedThemeValue = storedTheme || 'auto';
+
+    themeSelects.forEach((select) => {
+      const tmpl = select.querySelector(`[data-value="${selectedThemeValue}"] template`);
+      const selectedSlot = select.querySelector('[data-part="trigger-selected"]');
+
+      selectedSlot.innerHTML = '';
+      if (tmpl) {
+        selectedSlot.appendChild(tmpl.content.cloneNode(true));
+        return;
+      }
+    });
+  }
 </script>
 
 <script>
   import { getPageLoadEvent } from '../../plugin/helpers/getPageLoadEvent';
+  import { initDropdown } from '@stainless-api/docs/components/scripts';
 
   const storageKey = 'starlight-theme';
 
@@ -100,115 +139,54 @@ const options = [
   const loadTheme = (): Theme => parseTheme(localStorage.getItem(storageKey));
 
   const storeTheme = (theme: Theme) => {
-    localStorage.setItem(storageKey, theme === 'light' || theme === 'dark' ? theme : '');
+    localStorage.setItem(storageKey, parseTheme(theme));
   };
 
   const getPreferredColorScheme = (): Theme =>
     window.matchMedia('(prefers-color-scheme: light)').matches ? 'light' : 'dark';
 
-  const highlightSelected = (doc: Document, theme: Theme) => {
-    doc.querySelectorAll('.theme-select-container.stldocs-dropdown').forEach((container) => {
-      const icons = container.querySelectorAll('.stl-dropdown-icon svg');
-      icons.forEach((icon, index) => {
-        const themeForIcon = ['auto', 'light', 'dark'][index];
-        icon.classList.toggle('selected', theme === themeForIcon);
-      });
-
-      container.querySelectorAll('.stldocs-dropdown-item').forEach((option) => {
-        const label = option.querySelector('.stl-dropdown-label')?.textContent?.toLowerCase();
-        option.classList.toggle('selected', label === theme);
-      });
-
-      const menu = container.querySelector('.stldocs-dropdown-menu');
-      menu?.classList.remove('open');
-    });
-  };
-
   const applyTheme = (doc: Document, theme: Theme) => {
     const effective = theme === 'auto' ? getPreferredColorScheme() : theme;
     doc.documentElement.dataset.theme = effective;
     storeTheme(theme);
-    StarlightThemeProvider?.updatePickers?.(theme);
-    highlightSelected(doc, theme);
   };
 
   function setupThemeMode(doc: Document) {
-    // Attach to all instances of .theme-select-container.stldocs-dropdown. We need this because
-    // the theme component is used in the mobile dropdown and the desktop sidebar.
-    doc.querySelectorAll('.theme-select-container.stldocs-dropdown').forEach((container) => {
-      const trigger = container.querySelector('.stldocs-dropdown-trigger');
-      const menu = container.querySelector('.stldocs-dropdown-menu');
-
-      if (trigger && menu) {
-        trigger.addEventListener('click', (e) => {
-          e.stopPropagation();
-
-          doc.querySelectorAll('.theme-select-container .stldocs-dropdown-menu.open').forEach((otherMenu) => {
-            if (otherMenu !== menu) otherMenu.classList.remove('open');
-          });
-
-          menu.classList.toggle('open');
-        });
-
-        container.querySelectorAll('.stldocs-dropdown-item').forEach((option) => {
-          option.addEventListener('click', () => {
-            const label = option.querySelector('.stl-dropdown-label')?.textContent?.toLowerCase() ?? '';
-            if (['auto', 'light', 'dark'].includes(label)) {
-              applyTheme(doc, label as Theme);
-            }
-          });
-        });
-      }
-    });
-
-    // Close dropdown if you click outside
-    doc.addEventListener('click', (e) => {
-      doc.querySelectorAll('.theme-select-container .stldocs-dropdown-menu.open').forEach((menu) => {
-        if (e.target instanceof HTMLElement && !menu.contains(e.target)) {
-          menu.classList.remove('open');
-        }
-      });
-    });
-
-    // React to system preference changes
     matchMedia('(prefers-color-scheme: light)').addEventListener('change', () => {
       if (loadTheme() === 'auto') applyTheme(doc, 'auto');
     });
 
     applyTheme(doc, loadTheme());
+    return loadTheme();
   }
 
   document.addEventListener('astro:before-swap', (event) => {
     applyTheme(event.newDocument, loadTheme());
   });
 
-  document.addEventListener(getPageLoadEvent(), (e) => {
-    setupThemeMode(document);
+  document.addEventListener(getPageLoadEvent(), () => {
+    const initialTheme = setupThemeMode(document);
+    const themeSelects = document.querySelectorAll<HTMLElement>('[data-theme-select]');
+    themeSelects.forEach((select) => {
+      initDropdown({
+        root: select,
+        onSelect: (value: string) => applyTheme(document, value as Theme),
+        initialValue: initialTheme as string,
+      });
+    });
   });
 </script>
 
 <style is:global>
-  .theme-select-root {
-    background-color: unset;
-    display: flex;
-    align-items: center;
-  }
   .stl-dropdown-icon svg {
     width: 16px;
     height: 16px;
 
     path {
-      fill: var(--sl-color-gray-2);
+      fill: var(--stl-color-foreground-reduced);
     }
   }
 
-  .stldocs-dropdown-trigger .stl-dropdown-icon svg {
-    display: none;
-
-    &.selected {
-      display: flex;
-    }
-  }
   .stl-dropdown-icon {
     display: flex;
     align-items: center;
@@ -217,8 +195,7 @@ const options = [
     height: 16px;
   }
 
-  span.stl-dropdown-icon,
-  span.stl-dropdown-chevron {
+  span.stl-dropdown-icon {
     padding-left: 0;
     padding-right: 0%;
   }