Include generated docs in local circuit prompt

Author: Fire-Fairy84

lib/prompt-templates/create-local-circuit-prompt.ts

@@ -1,7 +1,7 @@
 import {
+  fp,
   getFootprintNamesByType,
   getFootprintSizes,
-  fp,
 } from "@tscircuit/footprinter"
 
 async function fetchFileContent(url: string): Promise<string> {
@@ -19,6 +19,24 @@ async function fetchFileContent(url: string): Promise<string> {
   }
 }
 
+const generatedDocsUrl = "https://docs.tscircuit.com/ai.txt"
+let generatedDocsPromise: Promise<string> | undefined
+
+export function clearGeneratedDocsCacheForTests() {
+  generatedDocsPromise = undefined
+}
+
+async function fetchGeneratedDocs(): Promise<string> {
+  generatedDocsPromise ??= fetch(generatedDocsUrl)
+    .then((response) => {
+      if (!response.ok) return ""
+      return response.text()
+    })
+    .catch(() => "")
+
+  return generatedDocsPromise
+}
+
 export const createLocalCircuitPrompt = async () => {
   const footprintNamesByType = getFootprintNamesByType()
   const footprintSizes = getFootprintSizes()
@@ -33,10 +51,12 @@ export const createLocalCircuitPrompt = async () => {
     "",
   )
 
-  const propsDoc =
-    (await fetchFileContent(
+  const [propsDoc, generatedDocs] = await Promise.all([
+    fetchFileContent(
       "https://raw.githubusercontent.com/tscircuit/props/main/generated/COMPONENT_TYPES.md",
-    )) || ""
+    ),
+    fetchGeneratedDocs(),
+  ])
 
   const cleanedPropsDoc = propsDoc
     .split("\n")
@@ -51,6 +71,8 @@ YOU MUST ABIDE BY THE RULES IN THE RULES SECTION
 
 ## tscircuit API overview
 
+${generatedDocs ? `Generated tscircuit documentation:\n\n${generatedDocs}\n\n` : ""}
+
 Here's an overview of the tscircuit API:
 
 <board width="10mm" height="10mm" /> // usually the root component

tests/prompt-templates/create-local-circuit-prompt.test.ts

@@ -0,0 +1,59 @@
+import { afterEach, describe, expect, it, mock } from "bun:test"
+import {
+  clearGeneratedDocsCacheForTests,
+  createLocalCircuitPrompt,
+} from "../../lib/prompt-templates/create-local-circuit-prompt"
+
+const originalFetch = globalThis.fetch
+
+afterEach(() => {
+  globalThis.fetch = originalFetch
+  clearGeneratedDocsCacheForTests()
+})
+
+function mockDocsFetch(generatedDocsResponse: Response | Error) {
+  const fetchMock = mock(async (url: string | URL | Request) => {
+    const requestUrl = String(url)
+
+    if (requestUrl === "https://docs.tscircuit.com/ai.txt") {
+      if (generatedDocsResponse instanceof Error) {
+        throw generatedDocsResponse
+      }
+
+      return generatedDocsResponse
+    }
+
+    if (
+      requestUrl ===
+      "https://raw.githubusercontent.com/tscircuit/props/main/generated/COMPONENT_TYPES.md"
+    ) {
+      return new Response("# Components\n\n<resistor />")
+    }
+
+    throw new Error(`Unexpected fetch URL: ${requestUrl}`)
+  })
+
+  globalThis.fetch = fetchMock as unknown as typeof fetch
+
+  return fetchMock
+}
+
+describe("createLocalCircuitPrompt", () => {
+  it("includes the generated tscircuit docs when they are available", async () => {
+    mockDocsFetch(new Response("Generated docs content"))
+
+    const prompt = await createLocalCircuitPrompt()
+
+    expect(prompt).toContain("Generated tscircuit documentation:")
+    expect(prompt).toContain("Generated docs content")
+  })
+
+  it("keeps creating the prompt when generated docs cannot be fetched", async () => {
+    mockDocsFetch(new Error("docs unavailable"))
+
+    const prompt = await createLocalCircuitPrompt()
+
+    expect(prompt).toContain("Here's an overview of the tscircuit API:")
+    expect(prompt).not.toContain("Generated tscircuit documentation:")
+  })
+})