ai-sdk-provider-claude-code 2.0.2 → 2.0.3

package/README.md

@@ -79,7 +79,7 @@ import { streamText } from 'ai';
 import { claudeCode } from 'ai-sdk-provider-claude-code';
 
 const result = streamText({
-  model: claudeCode('sonnet'),
+  model: claudeCode('haiku'),
   prompt: 'Hello, Claude!',
 });
 
@@ -94,7 +94,7 @@ import { generateText } from 'ai';
 import { claudeCode } from 'ai-sdk-provider-claude-code';
 
 const { text } = await generateText({
-  model: claudeCode('sonnet'),
+  model: claudeCode('haiku'),
   prompt: 'Hello, Claude!',
 });
 
@@ -124,8 +124,9 @@ Key changes:
 
 ## Models
 
-- **`opus`** - Claude 4 Opus (most capable)
-- **`sonnet`** - Claude 4 Sonnet (balanced performance)
+- **`opus`** - Claude 4.1 Opus (most capable)
+- **`sonnet`** - Claude 4.5 Sonnet (balanced performance)
+- **`haiku`** - Claude 4.5 Haiku (fastest, most cost-effective)
 
 ## Documentation
 

package/dist/index.cjs

@@ -613,7 +613,7 @@ var claudeCodeSettingsSchema = import_zod.z.object({
   extraArgs: import_zod.z.record(import_zod.z.string(), import_zod.z.union([import_zod.z.string(), import_zod.z.null()])).optional()
 }).strict();
 function validateModelId(modelId) {
-  const knownModels = ["opus", "sonnet"];
+  const knownModels = ["opus", "sonnet", "haiku"];
   if (!modelId || modelId.trim() === "") {
     throw new Error("Model ID cannot be empty");
   }
@@ -708,6 +708,81 @@ function getLogger(logger) {
 
 // src/claude-code-language-model.ts
 var import_claude_agent_sdk = require("@anthropic-ai/claude-agent-sdk");
+var CLAUDE_CODE_TRUNCATION_WARNING = "Claude Code CLI output ended unexpectedly; returning truncated response from buffered text. Await upstream fix to avoid data loss.";
+var MIN_TRUNCATION_LENGTH = 1024;
+var POSITION_PATTERN = /position\s+(\d+)/i;
+function hasUnclosedJsonStructure(text) {
+  let depth = 0;
+  let inString = false;
+  let escapeNext = false;
+  for (let i = 0; i < text.length; i++) {
+    const char = text[i];
+    if (escapeNext) {
+      escapeNext = false;
+      continue;
+    }
+    if (inString) {
+      if (char === "\\") {
+        escapeNext = true;
+        continue;
+      }
+      if (char === '"') {
+        inString = false;
+      }
+      continue;
+    }
+    if (char === '"') {
+      inString = true;
+      continue;
+    }
+    if (char === "{" || char === "[") {
+      depth++;
+      continue;
+    }
+    if (char === "}" || char === "]") {
+      if (depth > 0) {
+        depth--;
+      }
+    }
+  }
+  return depth > 0 || inString;
+}
+function isClaudeCodeTruncationError(error, bufferedText) {
+  if (!(error instanceof SyntaxError)) {
+    return false;
+  }
+  if (!bufferedText) {
+    return false;
+  }
+  const rawMessage = typeof error.message === "string" ? error.message : "";
+  const message = rawMessage.toLowerCase();
+  const truncationIndicators = [
+    "unexpected end of json input",
+    "unexpected end of input",
+    "unexpected end of string",
+    "unterminated string"
+  ];
+  if (!truncationIndicators.some((indicator) => message.includes(indicator))) {
+    return false;
+  }
+  const positionMatch = rawMessage.match(POSITION_PATTERN);
+  if (positionMatch) {
+    const position = Number.parseInt(positionMatch[1], 10);
+    if (Number.isFinite(position)) {
+      const isNearBufferEnd = Math.abs(position - bufferedText.length) <= 16;
+      if (isNearBufferEnd && position >= MIN_TRUNCATION_LENGTH) {
+        return true;
+      }
+      if (!isNearBufferEnd) {
+        return false;
+      }
+    }
+  }
+  if (bufferedText.length < MIN_TRUNCATION_LENGTH) {
+    return false;
+  }
+  return hasUnclosedJsonStructure(bufferedText);
+}
 function isAbortError(err) {
   if (err && typeof err === "object") {
     const e = err;
@@ -737,7 +812,8 @@ function toAsyncIterablePrompt(messagesPrompt, outputStreamEnded, sessionId, con
 }
 var modelMap = {
   opus: "opus",
-  sonnet: "sonnet"
+  sonnet: "sonnet",
+  haiku: "haiku"
 };
 var ClaudeCodeLanguageModel = class _ClaudeCodeLanguageModel {
   specificationVersion = "v2";
@@ -1087,6 +1163,7 @@ var ClaudeCodeLanguageModel = class _ClaudeCodeLanguageModel {
     let text = "";
     let usage = { inputTokens: 0, outputTokens: 0, totalTokens: 0 };
     let finishReason = "stop";
+    let wasTruncated = false;
     let costUsd;
     let durationMs;
     let rawUsage;
@@ -1157,7 +1234,16 @@ var ClaudeCodeLanguageModel = class _ClaudeCodeLanguageModel {
       if (isAbortError(error)) {
         throw options.abortSignal?.aborted ? options.abortSignal.reason : error;
       }
-      throw this.handleClaudeCodeError(error, messagesPrompt);
+      if (isClaudeCodeTruncationError(error, text)) {
+        wasTruncated = true;
+        finishReason = "length";
+        warnings.push({
+          type: "other",
+          message: CLAUDE_CODE_TRUNCATION_WARNING
+        });
+      } else {
+        throw this.handleClaudeCodeError(error, messagesPrompt);
+      }
     } finally {
       if (options.abortSignal && abortListener) {
         options.abortSignal.removeEventListener("abort", abortListener);
@@ -1184,7 +1270,8 @@ var ClaudeCodeLanguageModel = class _ClaudeCodeLanguageModel {
           ...this.sessionId !== void 0 && { sessionId: this.sessionId },
           ...costUsd !== void 0 && { costUsd },
           ...durationMs !== void 0 && { durationMs },
-          ...rawUsage !== void 0 && { rawUsage }
+          ...rawUsage !== void 0 && { rawUsage },
+          ...wasTruncated && { truncated: true }
         }
       }
     };
@@ -1276,6 +1363,9 @@ var ClaudeCodeLanguageModel = class _ClaudeCodeLanguageModel {
           }
           toolStates.clear();
         };
+        let usage = { inputTokens: 0, outputTokens: 0, totalTokens: 0 };
+        let accumulatedText = "";
+        let textPartId;
         try {
           controller.enqueue({ type: "stream-start", warnings });
           if (this.settings.canUseTool && this.settings.permissionPromptToolName) {
@@ -1293,9 +1383,6 @@ var ClaudeCodeLanguageModel = class _ClaudeCodeLanguageModel {
             prompt: sdkPrompt,
             options: queryOptions
           });
-          let usage = { inputTokens: 0, outputTokens: 0, totalTokens: 0 };
-          let accumulatedText = "";
-          let textPartId;
           for await (const message of response) {
             if (message.type === "assistant") {
               if (!message.message?.content) {
@@ -1545,6 +1632,71 @@ var ClaudeCodeLanguageModel = class _ClaudeCodeLanguageModel {
           controller.close();
         } catch (error) {
           done();
+          if (isClaudeCodeTruncationError(error, accumulatedText)) {
+            const truncationWarning = {
+              type: "other",
+              message: CLAUDE_CODE_TRUNCATION_WARNING
+            };
+            streamWarnings.push(truncationWarning);
+            const emitJsonText = () => {
+              const extractedJson = this.handleJsonExtraction(accumulatedText, streamWarnings);
+              const jsonTextId = (0, import_provider_utils.generateId)();
+              controller.enqueue({
+                type: "text-start",
+                id: jsonTextId
+              });
+              controller.enqueue({
+                type: "text-delta",
+                id: jsonTextId,
+                delta: extractedJson
+              });
+              controller.enqueue({
+                type: "text-end",
+                id: jsonTextId
+              });
+            };
+            if (options.responseFormat?.type === "json") {
+              emitJsonText();
+            } else if (textPartId) {
+              controller.enqueue({
+                type: "text-end",
+                id: textPartId
+              });
+            } else if (accumulatedText) {
+              const fallbackTextId = (0, import_provider_utils.generateId)();
+              controller.enqueue({
+                type: "text-start",
+                id: fallbackTextId
+              });
+              controller.enqueue({
+                type: "text-delta",
+                id: fallbackTextId,
+                delta: accumulatedText
+              });
+              controller.enqueue({
+                type: "text-end",
+                id: fallbackTextId
+              });
+            }
+            finalizeToolCalls();
+            const warningsJson = this.serializeWarningsForMetadata(streamWarnings);
+            controller.enqueue({
+              type: "finish",
+              finishReason: "length",
+              usage,
+              providerMetadata: {
+                "claude-code": {
+                  ...this.sessionId !== void 0 && { sessionId: this.sessionId },
+                  truncated: true,
+                  ...streamWarnings.length > 0 && {
+                    warnings: warningsJson
+                  }
+                }
+              }
+            });
+            controller.close();
+            return;
+          }
           finalizeToolCalls();
           let errorToEmit;
           if (isAbortError(error)) {