fixed lib path, now can use custom anthropic client

README.md

@@ -36,6 +36,8 @@ export interface AiCoder {
 }
 ```
 
+*Note: The `createAiCoder` function now accepts an optional `anthropicClient` parameter to override the default Anthropic client. This allows you to provide a custom client (for example, with different API key configuration) when using AiCoder in environments like the browser.*
+
 The AI Coder supports streaming of AI responses and notifying you when the virtual file system (VFS) is updated. To achieve this, you can pass two callback functions when creating an AiCoder instance:
 - **onStreamedChunk**: A callback function that receives streamed chunks from the AI. This is useful for logging or updating a UI with gradual progress.
 - **onVfsChanged**: A callback function that is invoked whenever the VFS is updated with new content. This is useful for refreshing a file view or triggering further processing.

lib/ai/aiCoder.ts

@@ -17,13 +17,16 @@ export class AiCoderImpl implements AiCoder {
   onVfsChanged: () => void
   vfs: { [filepath: string]: string } = {}
   availableOptions = [{ name: "microController", options: ["pico", "esp32"] }]
+  anthropicClient: import("@anthropic-ai/sdk").Anthropic | undefined
 
   constructor(
     onStreamedChunk: (chunk: string) => void,
     onVfsChanged: () => void,
+    anthropicClient?: import("@anthropic-ai/sdk").Anthropic,
   ) {
     this.onStreamedChunk = onStreamedChunk
     this.onVfsChanged = onVfsChanged
+    this.anthropicClient = anthropicClient
   }
 
   async submitPrompt(
@@ -62,6 +65,7 @@ export class AiCoderImpl implements AiCoder {
 export const createAiCoder = (
   onStreamedChunk: (chunk: string) => void,
   onVfsChanged: () => void,
+  anthropicClient?: import("@anthropic-ai/sdk").Anthropic,
 ): AiCoder => {
-  return new AiCoderImpl(onStreamedChunk, onVfsChanged)
+  return new AiCoderImpl(onStreamedChunk, onVfsChanged, anthropicClient)
 }

lib/ai/anthropic.ts

@@ -1,7 +1,16 @@
 import Anthropic from "@anthropic-ai/sdk"
-import dotenv from "dotenv"
-dotenv.config()
+
+let apiKey = ""
+if (
+  typeof process !== "undefined" &&
+  process.env &&
+  process.env.ANTHROPIC_API_KEY
+) {
+  import("dotenv").then((dotenv) => dotenv.config())
+  apiKey = process.env.ANTHROPIC_API_KEY
+}
 
 export const anthropic = new Anthropic({
-  apiKey: process.env.ANTHROPIC_API_KEY,
+  apiKey,
+  dangerouslyAllowBrowser: true,
 })

lib/ai/ask-ai-with-previous-attempts.ts

@@ -10,12 +10,15 @@ export const askAiWithPreviousAttempts = async ({
   systemPrompt,
   previousAttempts,
   onStream,
+  anthropicClient,
 }: {
   prompt: string
   systemPrompt: string
   previousAttempts?: AttemptHistory[]
   onStream?: (chunk: string) => void
+  anthropicClient?: typeof anthropic
 }): Promise<string> => {
+  const client = anthropicClient || anthropic
   const messages: { role: "assistant" | "user"; content: string }[] = [
     { role: "user", content: prompt },
   ]
@@ -50,7 +53,7 @@ export const askAiWithPreviousAttempts = async ({
       onStream(
         `Start streaming AI response, attempt: ${(previousAttempts?.length || 0) + 1}`,
       )
-    const completionStream = await anthropic.messages.create({
+    const completionStream = await client.messages.create({
       model: "claude-3-5-haiku-20241022",
       max_tokens: 2048,
       system: systemPrompt,

lib/ai/run-ai-with-error-correction.ts

@@ -45,6 +45,7 @@ export const runAiWithErrorCorrection = async ({
   onStream,
   onVfsChanged,
   vfs,
+  anthropicClient,
 }: {
   attempt?: number
   logsDir?: string
@@ -56,6 +57,7 @@ export const runAiWithErrorCorrection = async ({
   onStream?: (chunk: string) => void
   onVfsChanged?: () => void
   vfs?: Record<string, string>
+  anthropicClient?: import("@anthropic-ai/sdk").Anthropic
 }): Promise<{
   code: string
   codeBlock: string
@@ -66,6 +68,7 @@ export const runAiWithErrorCorrection = async ({
     systemPrompt,
     previousAttempts,
     onStream,
+    anthropicClient,
   })
   const codeMatch = aiResponse.match(/```tsx\s*([\s\S]*?)\s*```/)
   const code = codeMatch ? codeMatch[1].trim() : ""

tsconfig.json

@@ -18,7 +18,8 @@
     "paths": {
       "src/*": ["src/*"],
       "tests/*": ["tests/*"],
-      "prompt-templates/*": ["lib/prompt-templates/*"]
+      "prompt-templates/*": ["lib/prompt-templates/*"],
+      "lib/*": ["lib/*"]
     },
 
     // Best practices