@ai-sdk/anthropic 2.0.0-alpha.8 → 2.0.0-alpha.9

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({
@@ -471,8 +512,10 @@ function mapAnthropicStopReason({
 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:";
@@ -602,6 +645,27 @@ 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,
@@ -611,7 +675,7 @@ var AnthropicMessagesLanguageModel = class {
       jsonResponseTool != null ? {
         tools: [jsonResponseTool],
         toolChoice: { type: "tool", toolName: jsonResponseTool.name }
-      } : { tools, toolChoice }
+      } : { tools: modifiedTools, toolChoice: modifiedToolChoice }
     );
     return {
       args: {
@@ -643,7 +707,7 @@ 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;
+    var _a, _b, _c, _d, _e;
     const { args, warnings, betas, jsonResponseTool } = await this.getArgs(options);
     const {
       responseHeaders,
@@ -709,6 +773,38 @@ var AnthropicMessagesLanguageModel = class {
           );
           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 {
@@ -721,19 +817,19 @@ var AnthropicMessagesLanguageModel = class {
         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
         }
       }
     };
@@ -761,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({
@@ -768,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;
@@ -807,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(
@@ -869,6 +1001,9 @@ var AnthropicMessagesLanguageModel = class {
                   }
                   case "input_json_delta": {
                     const contentBlock = toolCallContentBlocks[value.index];
+                    if (!contentBlock) {
+                      return;
+                    }
                     controller.enqueue(
                       jsonResponseTool != null ? {
                         type: "text",
@@ -884,6 +1019,9 @@ var AnthropicMessagesLanguageModel = class {
                     contentBlock.jsonText += value.delta.partial_json;
                     return;
                   }
+                  case "citations_delta": {
+                    return;
+                  }
                   default: {
                     const _exhaustiveCheck = deltaType;
                     throw new Error(
@@ -894,22 +1032,22 @@ 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);
+                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
@@ -966,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()
+          })
+        ])
       })
     ])
   ),
@@ -974,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", [
@@ -1011,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()
+          })
+        ])
       })
     ])
   }),
@@ -1033,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()
+        })
       })
     ])
   }),