ai-sdk-provider-claude-code 3.3.1 → 3.3.3

package/dist/index.cjs

@@ -568,7 +568,11 @@ var claudeCodeSettingsSchema = import_zod.z.object({
     message: "spawnClaudeCodeProcess must be a function"
   }).optional(),
   sdkOptions: import_zod.z.record(import_zod.z.string(), import_zod.z.any()).optional(),
-  maxToolResultSize: import_zod.z.number().int().min(100).max(1e6).optional()
+  maxToolResultSize: import_zod.z.number().int().min(100).max(1e6).optional(),
+  // Callback invoked when Query object is created - for mid-stream injection via streamInput()
+  onQueryCreated: import_zod.z.any().refine((val) => val === void 0 || typeof val === "function", {
+    message: "onQueryCreated must be a function"
+  }).optional()
 }).strict();
 function validateModelId(modelId) {
   const knownModels = ["opus", "sonnet", "haiku"];
@@ -1289,6 +1293,7 @@ var ClaudeCodeLanguageModel = class _ClaudeCodeLanguageModel {
     let wasTruncated = false;
     let costUsd;
     let durationMs;
+    let modelUsage;
     const warnings = this.generateAllWarnings(options, messagesPrompt);
     if (messageWarnings) {
       messageWarnings.forEach((warning) => {
@@ -1332,6 +1337,7 @@ var ClaudeCodeLanguageModel = class _ClaudeCodeLanguageModel {
         prompt: sdkPrompt,
         options: queryOptions
       });
+      this.settings.onQueryCreated?.(response);
       for await (const message of response) {
         this.logger.debug(`[claude-code] Received message type: ${message.type}`);
         if (message.type === "assistant") {
@@ -1341,6 +1347,7 @@ var ClaudeCodeLanguageModel = class _ClaudeCodeLanguageModel {
           this.setSessionId(message.session_id);
           costUsd = message.total_cost_usd;
           durationMs = message.duration_ms;
+          modelUsage = message.modelUsage;
           if ("is_error" in message && message.is_error === true) {
             const errorMessage = "result" in message && typeof message.result === "string" ? message.result : "Claude Code CLI returned an error";
             throw Object.assign(new Error(errorMessage), { exitCode: 1 });
@@ -1416,6 +1423,7 @@ var ClaudeCodeLanguageModel = class _ClaudeCodeLanguageModel {
           ...this.sessionId !== void 0 && { sessionId: this.sessionId },
           ...costUsd !== void 0 && { costUsd },
           ...durationMs !== void 0 && { durationMs },
+          ...modelUsage !== void 0 && { modelUsage },
           ...wasTruncated && { truncated: true }
         }
       }
@@ -1564,6 +1572,7 @@ var ClaudeCodeLanguageModel = class _ClaudeCodeLanguageModel {
             prompt: sdkPrompt,
             options: queryOptions
           });
+          this.settings.onQueryCreated?.(response);
           for await (const message of response) {
             this.logger.debug(`[claude-code] Stream received message type: ${message.type}`);
             if (message.type === "stream_event") {
@@ -2149,6 +2158,9 @@ var ClaudeCodeLanguageModel = class _ClaudeCodeLanguageModel {
                       costUsd: message.total_cost_usd
                     },
                     ...message.duration_ms !== void 0 && { durationMs: message.duration_ms },
+                    ...message.modelUsage !== void 0 && {
+                      modelUsage: message.modelUsage
+                    },
                     // JSON validation warnings are collected during streaming and included
                     // in providerMetadata since the AI SDK's finish event doesn't support
                     // a top-level warnings field (unlike stream-start which was already emitted)