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

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,36 @@ 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 anthropicFilePartProviderOptions = z2.object({
+  /**
+   * Citation configuration for this document.
+   * When enabled, this document will generate citations in the response.
+   */
+  citations: z2.object({
+    /**
+     * Enable citations for this document
+     */
+    enabled: z2.boolean()
+  }).optional(),
+  /**
+   * Custom title for the document.
+   * If not provided, the filename will be used.
+   */
+  title: z2.string().optional(),
+  /**
+   * Context about the document that will be passed to the model
+   * but not used towards cited content.
+   * Useful for storing document metadata as text or stringified JSON.
+   */
+  context: z2.string().optional()
+});
 var anthropicProviderOptions = z2.object({
   /**
   Include reasoning content in requests sent to the model. Defaults to `true`.
@@ -39,6 +71,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 +103,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 +118,10 @@ function prepareTools({
   }
   const anthropicTools2 = [];
   for (const tool of tools) {
+    if (isWebSearchTool(tool)) {
+      anthropicTools2.push(tool);
+      continue;
+    }
     switch (tool.type) {
       case "function":
         anthropicTools2.push({
@@ -178,7 +242,7 @@ async function convertToAnthropicMessagesPrompt({
   sendReasoning,
   warnings
 }) {
-  var _a, _b, _c;
+  var _a, _b, _c, _d;
   const betas = /* @__PURE__ */ new Set();
   const blocks = groupIntoBlocks(prompt);
   let system = void 0;
@@ -189,6 +253,26 @@ async function convertToAnthropicMessagesPrompt({
     const cacheControlValue = (_a2 = anthropic == null ? void 0 : anthropic.cacheControl) != null ? _a2 : anthropic == null ? void 0 : anthropic.cache_control;
     return cacheControlValue;
   }
+  async function shouldEnableCitations(providerMetadata) {
+    var _a2, _b2;
+    const anthropicOptions = await parseProviderOptions({
+      provider: "anthropic",
+      providerOptions: providerMetadata,
+      schema: anthropicFilePartProviderOptions
+    });
+    return (_b2 = (_a2 = anthropicOptions == null ? void 0 : anthropicOptions.citations) == null ? void 0 : _a2.enabled) != null ? _b2 : false;
+  }
+  async function getDocumentMetadata(providerMetadata) {
+    const anthropicOptions = await parseProviderOptions({
+      provider: "anthropic",
+      providerOptions: providerMetadata,
+      schema: anthropicFilePartProviderOptions
+    });
+    return {
+      title: anthropicOptions == null ? void 0 : anthropicOptions.title,
+      context: anthropicOptions == null ? void 0 : anthropicOptions.context
+    };
+  }
   for (let i = 0; i < blocks.length; i++) {
     const block = blocks[i];
     const isLastBlock = i === blocks.length - 1;
@@ -242,6 +326,12 @@ async function convertToAnthropicMessagesPrompt({
                       });
                     } else if (part.mediaType === "application/pdf") {
                       betas.add("pdfs-2024-09-25");
+                      const enableCitations = await shouldEnableCitations(
+                        part.providerOptions
+                      );
+                      const metadata = await getDocumentMetadata(
+                        part.providerOptions
+                      );
                       anthropicContent.push({
                         type: "document",
                         source: part.data instanceof URL ? {
@@ -252,6 +342,11 @@ async function convertToAnthropicMessagesPrompt({
                           media_type: "application/pdf",
                           data: convertToBase64(part.data)
                         },
+                        title: (_b = metadata.title) != null ? _b : part.filename,
+                        ...metadata.context && { context: metadata.context },
+                        ...enableCitations && {
+                          citations: { enabled: true }
+                        },
                         cache_control: cacheControl
                       });
                     } else {
@@ -269,7 +364,7 @@ async function convertToAnthropicMessagesPrompt({
               for (let i2 = 0; i2 < content.length; i2++) {
                 const part = content[i2];
                 const isLastPart = i2 === content.length - 1;
-                const cacheControl = (_b = getCacheControl(part.providerOptions)) != null ? _b : isLastPart ? getCacheControl(message.providerOptions) : void 0;
+                const cacheControl = (_c = getCacheControl(part.providerOptions)) != null ? _c : isLastPart ? getCacheControl(message.providerOptions) : void 0;
                 const toolResultContent = part.content != null ? part.content.map((part2) => {
                   var _a2;
                   switch (part2.type) {
@@ -319,7 +414,7 @@ async function convertToAnthropicMessagesPrompt({
           for (let k = 0; k < content.length; k++) {
             const part = content[k];
             const isLastContentPart = k === content.length - 1;
-            const cacheControl = (_c = getCacheControl(part.providerOptions)) != null ? _c : isLastContentPart ? getCacheControl(message.providerOptions) : void 0;
+            const cacheControl = (_d = getCacheControl(part.providerOptions)) != null ? _d : isLastContentPart ? getCacheControl(message.providerOptions) : void 0;
             switch (part.type) {
               case "text": {
                 anthropicContent.push({
@@ -450,13 +545,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:
@@ -465,11 +563,47 @@ function mapAnthropicStopReason(finishReason) {
 }
 
 // src/anthropic-messages-language-model.ts
+function processPageLocationCitation(citation, citationDocuments, generateId2, onSource) {
+  if (citation.type === "page_location") {
+    const source = createCitationSource(
+      citation,
+      citationDocuments,
+      generateId2
+    );
+    if (source) {
+      onSource(source);
+    }
+  }
+}
+function createCitationSource(citation, citationDocuments, generateId2) {
+  var _a;
+  const documentInfo = citationDocuments[citation.document_index];
+  if (!documentInfo) {
+    return null;
+  }
+  return {
+    type: "source",
+    sourceType: "document",
+    id: generateId2(),
+    mediaType: documentInfo.mediaType,
+    title: (_a = citation.document_title) != null ? _a : documentInfo.title,
+    filename: documentInfo.filename,
+    providerMetadata: {
+      anthropic: {
+        citedText: citation.cited_text,
+        startPageNumber: citation.start_page_number,
+        endPageNumber: citation.end_page_number
+      }
+    }
+  };
+}
 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 +651,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 +733,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 +772,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({
@@ -619,9 +794,29 @@ var AnthropicMessagesLanguageModel = class {
     var _a, _b, _c;
     return (_c = (_b = (_a = this.config).transformRequestBody) == null ? void 0 : _b.call(_a, args)) != null ? _c : args;
   }
+  extractCitationDocuments(prompt) {
+    const isCitationEnabled = (part) => {
+      var _a, _b;
+      const anthropic = (_a = part.providerOptions) == null ? void 0 : _a.anthropic;
+      const citationsConfig = anthropic == null ? void 0 : anthropic.citations;
+      return (_b = citationsConfig == null ? void 0 : citationsConfig.enabled) != null ? _b : false;
+    };
+    return prompt.filter((message) => message.role === "user").flatMap((message) => message.content).filter(
+      (part) => part.type === "file" && part.mediaType === "application/pdf" && isCitationEnabled(part)
+    ).map((part) => {
+      var _a;
+      const filePart = part;
+      return {
+        title: (_a = filePart.filename) != null ? _a : "Untitled Document",
+        filename: filePart.filename,
+        mediaType: filePart.mediaType
+      };
+    });
+  }
   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 citationDocuments = this.extractCitationDocuments(options.prompt);
     const {
       responseHeaders,
       value: response,
@@ -641,7 +836,19 @@ 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 });
+            if (part.citations) {
+              for (const citation of part.citations) {
+                processPageLocationCitation(
+                  citation,
+                  citationDocuments,
+                  this.generateId,
+                  (source) => content.push(source)
+                );
+              }
+            }
+          }
           break;
         }
         case "thinking": {
@@ -669,43 +876,85 @@ 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 citationDocuments = this.extractCitationDocuments(options.prompt);
     const body = { ...args, stream: true };
     const { responseHeaders, value: response } = await postJsonToApi({
       url: this.buildRequestUrl(true),
@@ -727,6 +976,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 +985,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 +1024,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 +1069,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 +1087,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 +1120,34 @@ 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": {
+                    const citation = value.delta.citation;
+                    processPageLocationCitation(
+                      citation,
+                      citationDocuments,
+                      generateId2,
+                      (source) => controller.enqueue(source)
+                    );
+                    return;
+                  }
                   default: {
                     const _exhaustiveCheck = deltaType;
                     throw new Error(
@@ -850,23 +1158,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": {
@@ -903,7 +1214,26 @@ var anthropicMessagesResponseSchema = z3.object({
     z3.discriminatedUnion("type", [
       z3.object({
         type: z3.literal("text"),
-        text: z3.string()
+        text: z3.string(),
+        citations: z3.array(
+          z3.discriminatedUnion("type", [
+            z3.object({
+              type: z3.literal("web_search_result_location"),
+              cited_text: z3.string(),
+              url: z3.string(),
+              title: z3.string(),
+              encrypted_index: z3.string()
+            }),
+            z3.object({
+              type: z3.literal("page_location"),
+              cited_text: z3.string(),
+              document_index: z3.number(),
+              document_title: z3.string().nullable(),
+              start_page_number: z3.number(),
+              end_page_number: z3.number()
+            })
+          ])
+        ).optional()
       }),
       z3.object({
         type: z3.literal("thinking"),
@@ -919,6 +1249,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 +1282,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 +1322,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 +1369,26 @@ var anthropicMessagesChunkSchema = z3.discriminatedUnion("type", [
       z3.object({
         type: z3.literal("signature_delta"),
         signature: z3.string()
+      }),
+      z3.object({
+        type: z3.literal("citations_delta"),
+        citation: z3.discriminatedUnion("type", [
+          z3.object({
+            type: z3.literal("web_search_result_location"),
+            cited_text: z3.string(),
+            url: z3.string(),
+            title: z3.string(),
+            encrypted_index: z3.string()
+          }),
+          z3.object({
+            type: z3.literal("page_location"),
+            cited_text: z3.string(),
+            document_index: z3.number(),
+            document_title: z3.string().nullable(),
+            start_page_number: z3.number(),
+            end_page_number: z3.number()
+          })
+        ])
       })
     ])
   }),