@ai-sdk/anthropic 2.0.0-alpha.1 → 2.0.0-alpha.10

package/dist/internal/index.mjs

@@ -1,11 +1,13 @@
 // src/anthropic-messages-language-model.ts
 import {
+  APICallError,
   UnsupportedFunctionalityError as UnsupportedFunctionalityError3
 } from "@ai-sdk/provider";
 import {
   combineHeaders,
   createEventSourceResponseHandler,
   createJsonResponseHandler,
+  generateId,
   parseProviderOptions as parseProviderOptions2,
   postJsonToApi,
   resolve
@@ -29,6 +31,13 @@ var anthropicFailedResponseHandler = createJsonErrorResponseHandler({
 
 // src/anthropic-messages-options.ts
 import { z as z2 } from "zod";
+var webSearchLocationSchema = z2.object({
+  type: z2.literal("approximate"),
+  city: z2.string().optional(),
+  region: z2.string().optional(),
+  country: z2.string(),
+  timezone: z2.string().optional()
+});
 var anthropicProviderOptions = z2.object({
   /**
   Include reasoning content in requests sent to the model. Defaults to `true`.
@@ -39,6 +48,31 @@ var anthropicProviderOptions = z2.object({
   thinking: z2.object({
     type: z2.union([z2.literal("enabled"), z2.literal("disabled")]),
     budgetTokens: z2.number().optional()
+  }).optional(),
+  /**
+   * Web search tool configuration for Claude models that support it.
+   * When provided, automatically adds the web search tool to the request.
+   */
+  webSearch: z2.object({
+    /**
+     * Limit the number of searches per request (optional)
+     * Defaults to 5 if not specified
+     */
+    maxUses: z2.number().min(1).max(20).optional(),
+    /**
+     * Only include results from these domains (optional)
+     * Cannot be used with blockedDomains
+     */
+    allowedDomains: z2.array(z2.string()).optional(),
+    /**
+     * Never include results from these domains (optional)
+     * Cannot be used with allowedDomains
+     */
+    blockedDomains: z2.array(z2.string()).optional(),
+    /**
+     * Localize search results based on user location (optional)
+     */
+    userLocation: webSearchLocationSchema.optional()
   }).optional()
 });
 
@@ -46,6 +80,9 @@ var anthropicProviderOptions = z2.object({
 import {
   UnsupportedFunctionalityError
 } from "@ai-sdk/provider";
+function isWebSearchTool(tool) {
+  return typeof tool === "object" && tool !== null && "type" in tool && tool.type === "web_search_20250305";
+}
 function prepareTools({
   tools,
   toolChoice
@@ -58,6 +95,10 @@ function prepareTools({
   }
   const anthropicTools2 = [];
   for (const tool of tools) {
+    if (isWebSearchTool(tool)) {
+      anthropicTools2.push(tool);
+      continue;
+    }
     switch (tool.type) {
       case "function":
         anthropicTools2.push({
@@ -450,13 +491,16 @@ function groupIntoBlocks(prompt) {
 }
 
 // src/map-anthropic-stop-reason.ts
-function mapAnthropicStopReason(finishReason) {
+function mapAnthropicStopReason({
+  finishReason,
+  isJsonResponseFromTool
+}) {
   switch (finishReason) {
     case "end_turn":
     case "stop_sequence":
       return "stop";
     case "tool_use":
-      return "tool-calls";
+      return isJsonResponseFromTool ? "stop" : "tool-calls";
     case "max_tokens":
       return "length";
     default:
@@ -468,8 +512,10 @@ function mapAnthropicStopReason(finishReason) {
 var AnthropicMessagesLanguageModel = class {
   constructor(modelId, config) {
     this.specificationVersion = "v2";
+    var _a;
     this.modelId = modelId;
     this.config = config;
+    this.generateId = (_a = config.generateId) != null ? _a : generateId;
   }
   supportsUrl(url) {
     return url.protocol === "https:";
@@ -517,13 +563,27 @@ var AnthropicMessagesLanguageModel = class {
         setting: "seed"
       });
     }
-    if (responseFormat != null && responseFormat.type !== "text") {
-      warnings.push({
-        type: "unsupported-setting",
-        setting: "responseFormat",
-        details: "JSON response format is not supported."
-      });
+    if ((responseFormat == null ? void 0 : responseFormat.type) === "json") {
+      if (responseFormat.schema == null) {
+        warnings.push({
+          type: "unsupported-setting",
+          setting: "responseFormat",
+          details: "JSON response format requires a schema. The response format is ignored."
+        });
+      } else if (tools != null) {
+        warnings.push({
+          type: "unsupported-setting",
+          setting: "tools",
+          details: "JSON response format does not support tools. The provided tools are ignored."
+        });
+      }
     }
+    const jsonResponseTool = (responseFormat == null ? void 0 : responseFormat.type) === "json" && responseFormat.schema != null ? {
+      type: "function",
+      name: "json",
+      description: "Respond with a JSON object.",
+      parameters: responseFormat.schema
+    } : void 0;
     const anthropicOptions = await parseProviderOptions2({
       provider: "anthropic",
       providerOptions,
@@ -585,12 +645,38 @@ var AnthropicMessagesLanguageModel = class {
       }
       baseArgs.max_tokens = maxOutputTokens + thinkingBudget;
     }
+    let modifiedTools = tools;
+    let modifiedToolChoice = toolChoice;
+    if (anthropicOptions == null ? void 0 : anthropicOptions.webSearch) {
+      const webSearchTool = {
+        type: "web_search_20250305",
+        name: "web_search",
+        max_uses: anthropicOptions.webSearch.maxUses,
+        allowed_domains: anthropicOptions.webSearch.allowedDomains,
+        blocked_domains: anthropicOptions.webSearch.blockedDomains,
+        ...anthropicOptions.webSearch.userLocation && {
+          user_location: {
+            type: anthropicOptions.webSearch.userLocation.type,
+            country: anthropicOptions.webSearch.userLocation.country,
+            city: anthropicOptions.webSearch.userLocation.city,
+            region: anthropicOptions.webSearch.userLocation.region,
+            timezone: anthropicOptions.webSearch.userLocation.timezone
+          }
+        }
+      };
+      modifiedTools = tools ? [...tools, webSearchTool] : [webSearchTool];
+    }
     const {
       tools: anthropicTools2,
       toolChoice: anthropicToolChoice,
       toolWarnings,
       betas: toolsBetas
-    } = prepareTools({ tools, toolChoice });
+    } = prepareTools(
+      jsonResponseTool != null ? {
+        tools: [jsonResponseTool],
+        toolChoice: { type: "tool", toolName: jsonResponseTool.name }
+      } : { tools: modifiedTools, toolChoice: modifiedToolChoice }
+    );
     return {
       args: {
         ...baseArgs,
@@ -598,7 +684,8 @@ var AnthropicMessagesLanguageModel = class {
         tool_choice: anthropicToolChoice
       },
       warnings: [...warnings, ...toolWarnings],
-      betas: /* @__PURE__ */ new Set([...messagesBetas, ...toolsBetas])
+      betas: /* @__PURE__ */ new Set([...messagesBetas, ...toolsBetas]),
+      jsonResponseTool
     };
   }
   async getHeaders({
@@ -620,8 +707,8 @@ var AnthropicMessagesLanguageModel = class {
     return (_c = (_b = (_a = this.config).transformRequestBody) == null ? void 0 : _b.call(_a, args)) != null ? _c : args;
   }
   async doGenerate(options) {
-    var _a, _b, _c, _d;
-    const { args, warnings, betas } = await this.getArgs(options);
+    var _a, _b, _c, _d, _e;
+    const { args, warnings, betas, jsonResponseTool } = await this.getArgs(options);
     const {
       responseHeaders,
       value: response,
@@ -641,7 +728,9 @@ var AnthropicMessagesLanguageModel = class {
     for (const part of response.content) {
       switch (part.type) {
         case "text": {
-          content.push({ type: "text", text: part.text });
+          if (jsonResponseTool == null) {
+            content.push({ type: "text", text: part.text });
+          }
           break;
         }
         case "thinking": {
@@ -669,43 +758,84 @@ var AnthropicMessagesLanguageModel = class {
           break;
         }
         case "tool_use": {
-          content.push({
-            type: "tool-call",
-            toolCallType: "function",
-            toolCallId: part.id,
-            toolName: part.name,
-            args: JSON.stringify(part.input)
-          });
+          content.push(
+            // when a json response tool is used, the tool call becomes the text:
+            jsonResponseTool != null ? {
+              type: "text",
+              text: JSON.stringify(part.input)
+            } : {
+              type: "tool-call",
+              toolCallType: "function",
+              toolCallId: part.id,
+              toolName: part.name,
+              args: JSON.stringify(part.input)
+            }
+          );
+          break;
+        }
+        case "server_tool_use": {
+          continue;
+        }
+        case "web_search_tool_result": {
+          if (Array.isArray(part.content)) {
+            for (const result of part.content) {
+              if (result.type === "web_search_result") {
+                content.push({
+                  type: "source",
+                  sourceType: "url",
+                  id: this.generateId(),
+                  url: result.url,
+                  title: result.title,
+                  providerMetadata: {
+                    anthropic: {
+                      encryptedContent: result.encrypted_content,
+                      pageAge: (_a = result.page_age) != null ? _a : null
+                    }
+                  }
+                });
+              }
+            }
+          } else if (part.content.type === "web_search_tool_result_error") {
+            throw new APICallError({
+              message: `Web search failed: ${part.content.error_code}`,
+              url: "web_search_api",
+              requestBodyValues: { tool_use_id: part.tool_use_id },
+              data: { error_code: part.content.error_code }
+            });
+          }
           break;
         }
       }
     }
     return {
       content,
-      finishReason: mapAnthropicStopReason(response.stop_reason),
+      finishReason: mapAnthropicStopReason({
+        finishReason: response.stop_reason,
+        isJsonResponseFromTool: jsonResponseTool != null
+      }),
       usage: {
         inputTokens: response.usage.input_tokens,
         outputTokens: response.usage.output_tokens,
         totalTokens: response.usage.input_tokens + response.usage.output_tokens,
-        cachedInputTokens: (_a = response.usage.cache_read_input_tokens) != null ? _a : void 0
+        cachedInputTokens: (_b = response.usage.cache_read_input_tokens) != null ? _b : void 0
       },
       request: { body: args },
       response: {
-        id: (_b = response.id) != null ? _b : void 0,
-        modelId: (_c = response.model) != null ? _c : void 0,
+        id: (_c = response.id) != null ? _c : void 0,
+        modelId: (_d = response.model) != null ? _d : void 0,
         headers: responseHeaders,
         body: rawResponse
       },
       warnings,
       providerMetadata: {
         anthropic: {
-          cacheCreationInputTokens: (_d = response.usage.cache_creation_input_tokens) != null ? _d : null
+          cacheCreationInputTokens: (_e = response.usage.cache_creation_input_tokens) != null ? _e : null
         }
       }
     };
   }
   async doStream(options) {
-    const { args, warnings, betas } = await this.getArgs(options);
+    const { args, warnings, betas, jsonResponseTool } = await this.getArgs(options);
     const body = { ...args, stream: true };
     const { responseHeaders, value: response } = await postJsonToApi({
       url: this.buildRequestUrl(true),
@@ -727,6 +857,8 @@ var AnthropicMessagesLanguageModel = class {
     const toolCallContentBlocks = {};
     let providerMetadata = void 0;
     let blockType = void 0;
+    const config = this.config;
+    const generateId2 = this.generateId;
     return {
       stream: response.pipeThrough(
         new TransformStream({
@@ -734,7 +866,7 @@ var AnthropicMessagesLanguageModel = class {
             controller.enqueue({ type: "stream-start", warnings });
           },
           transform(chunk, controller) {
-            var _a, _b, _c, _d, _e, _f;
+            var _a, _b, _c, _d, _e, _f, _g;
             if (!chunk.success) {
               controller.enqueue({ type: "error", error: chunk.error });
               return;
@@ -773,6 +905,40 @@ var AnthropicMessagesLanguageModel = class {
                     };
                     return;
                   }
+                  case "server_tool_use": {
+                    return;
+                  }
+                  case "web_search_tool_result": {
+                    if (Array.isArray(value.content_block.content)) {
+                      for (const result of value.content_block.content) {
+                        if (result.type === "web_search_result") {
+                          controller.enqueue({
+                            type: "source",
+                            sourceType: "url",
+                            id: generateId2(),
+                            url: result.url,
+                            title: result.title,
+                            providerMetadata: {
+                              anthropic: {
+                                encryptedContent: result.encrypted_content,
+                                pageAge: (_a = result.page_age) != null ? _a : null
+                              }
+                            }
+                          });
+                        }
+                      }
+                    } else if (value.content_block.content.type === "web_search_tool_result_error") {
+                      controller.enqueue({
+                        type: "error",
+                        error: {
+                          type: "web-search-error",
+                          message: `Web search failed: ${value.content_block.content.error_code}`,
+                          code: value.content_block.content.error_code
+                        }
+                      });
+                    }
+                    return;
+                  }
                   default: {
                     const _exhaustiveCheck = contentBlockType;
                     throw new Error(
@@ -784,13 +950,15 @@ var AnthropicMessagesLanguageModel = class {
               case "content_block_stop": {
                 if (toolCallContentBlocks[value.index] != null) {
                   const contentBlock = toolCallContentBlocks[value.index];
-                  controller.enqueue({
-                    type: "tool-call",
-                    toolCallType: "function",
-                    toolCallId: contentBlock.toolCallId,
-                    toolName: contentBlock.toolName,
-                    args: contentBlock.jsonText
-                  });
+                  if (jsonResponseTool == null) {
+                    controller.enqueue({
+                      type: "tool-call",
+                      toolCallType: "function",
+                      toolCallId: contentBlock.toolCallId,
+                      toolName: contentBlock.toolName,
+                      args: contentBlock.jsonText
+                    });
+                  }
                   delete toolCallContentBlocks[value.index];
                 }
                 blockType = void 0;
@@ -800,6 +968,9 @@ var AnthropicMessagesLanguageModel = class {
                 const deltaType = value.delta.type;
                 switch (deltaType) {
                   case "text_delta": {
+                    if (jsonResponseTool != null) {
+                      return;
+                    }
                     controller.enqueue({
                       type: "text",
                       text: value.delta.text
@@ -830,16 +1001,27 @@ var AnthropicMessagesLanguageModel = class {
                   }
                   case "input_json_delta": {
                     const contentBlock = toolCallContentBlocks[value.index];
-                    controller.enqueue({
-                      type: "tool-call-delta",
-                      toolCallType: "function",
-                      toolCallId: contentBlock.toolCallId,
-                      toolName: contentBlock.toolName,
-                      argsTextDelta: value.delta.partial_json
-                    });
+                    if (!contentBlock) {
+                      return;
+                    }
+                    controller.enqueue(
+                      jsonResponseTool != null ? {
+                        type: "text",
+                        text: value.delta.partial_json
+                      } : {
+                        type: "tool-call-delta",
+                        toolCallType: "function",
+                        toolCallId: contentBlock.toolCallId,
+                        toolName: contentBlock.toolName,
+                        argsTextDelta: value.delta.partial_json
+                      }
+                    );
                     contentBlock.jsonText += value.delta.partial_json;
                     return;
                   }
+                  case "citations_delta": {
+                    return;
+                  }
                   default: {
                     const _exhaustiveCheck = deltaType;
                     throw new Error(
@@ -850,23 +1032,26 @@ var AnthropicMessagesLanguageModel = class {
               }
               case "message_start": {
                 usage.inputTokens = value.message.usage.input_tokens;
-                usage.cachedInputTokens = (_a = value.message.usage.cache_read_input_tokens) != null ? _a : void 0;
+                usage.cachedInputTokens = (_b = value.message.usage.cache_read_input_tokens) != null ? _b : void 0;
                 providerMetadata = {
                   anthropic: {
-                    cacheCreationInputTokens: (_b = value.message.usage.cache_creation_input_tokens) != null ? _b : null
+                    cacheCreationInputTokens: (_c = value.message.usage.cache_creation_input_tokens) != null ? _c : null
                   }
                 };
                 controller.enqueue({
                   type: "response-metadata",
-                  id: (_c = value.message.id) != null ? _c : void 0,
-                  modelId: (_d = value.message.model) != null ? _d : void 0
+                  id: (_d = value.message.id) != null ? _d : void 0,
+                  modelId: (_e = value.message.model) != null ? _e : void 0
                 });
                 return;
               }
               case "message_delta": {
                 usage.outputTokens = value.usage.output_tokens;
-                usage.totalTokens = ((_e = usage.inputTokens) != null ? _e : 0) + ((_f = value.usage.output_tokens) != null ? _f : 0);
-                finishReason = mapAnthropicStopReason(value.delta.stop_reason);
+                usage.totalTokens = ((_f = usage.inputTokens) != null ? _f : 0) + ((_g = value.usage.output_tokens) != null ? _g : 0);
+                finishReason = mapAnthropicStopReason({
+                  finishReason: value.delta.stop_reason,
+                  isJsonResponseFromTool: jsonResponseTool != null
+                });
                 return;
               }
               case "message_stop": {
@@ -919,6 +1104,31 @@ var anthropicMessagesResponseSchema = z3.object({
         id: z3.string(),
         name: z3.string(),
         input: z3.unknown()
+      }),
+      z3.object({
+        type: z3.literal("server_tool_use"),
+        id: z3.string(),
+        name: z3.string(),
+        input: z3.record(z3.unknown()).nullish()
+      }),
+      z3.object({
+        type: z3.literal("web_search_tool_result"),
+        tool_use_id: z3.string(),
+        content: z3.union([
+          z3.array(
+            z3.object({
+              type: z3.literal("web_search_result"),
+              url: z3.string(),
+              title: z3.string(),
+              encrypted_content: z3.string(),
+              page_age: z3.string().nullish()
+            })
+          ),
+          z3.object({
+            type: z3.literal("web_search_tool_result_error"),
+            error_code: z3.string()
+          })
+        ])
       })
     ])
   ),
@@ -927,7 +1137,10 @@ var anthropicMessagesResponseSchema = z3.object({
     input_tokens: z3.number(),
     output_tokens: z3.number(),
     cache_creation_input_tokens: z3.number().nullish(),
-    cache_read_input_tokens: z3.number().nullish()
+    cache_read_input_tokens: z3.number().nullish(),
+    server_tool_use: z3.object({
+      web_search_requests: z3.number()
+    }).nullish()
   })
 });
 var anthropicMessagesChunkSchema = z3.discriminatedUnion("type", [
@@ -964,6 +1177,31 @@ var anthropicMessagesChunkSchema = z3.discriminatedUnion("type", [
       z3.object({
         type: z3.literal("redacted_thinking"),
         data: z3.string()
+      }),
+      z3.object({
+        type: z3.literal("server_tool_use"),
+        id: z3.string(),
+        name: z3.string(),
+        input: z3.record(z3.unknown()).nullish()
+      }),
+      z3.object({
+        type: z3.literal("web_search_tool_result"),
+        tool_use_id: z3.string(),
+        content: z3.union([
+          z3.array(
+            z3.object({
+              type: z3.literal("web_search_result"),
+              url: z3.string(),
+              title: z3.string(),
+              encrypted_content: z3.string(),
+              page_age: z3.string().nullish()
+            })
+          ),
+          z3.object({
+            type: z3.literal("web_search_tool_result_error"),
+            error_code: z3.string()
+          })
+        ])
       })
     ])
   }),
@@ -986,6 +1224,16 @@ var anthropicMessagesChunkSchema = z3.discriminatedUnion("type", [
       z3.object({
         type: z3.literal("signature_delta"),
         signature: z3.string()
+      }),
+      z3.object({
+        type: z3.literal("citations_delta"),
+        citation: z3.object({
+          type: z3.literal("web_search_result_location"),
+          cited_text: z3.string(),
+          url: z3.string(),
+          title: z3.string(),
+          encrypted_index: z3.string()
+        })
       })
     ])
   }),