@ai-sdk/anthropic 0.0.51 → 0.0.54

package/dist/index.mjs

@@ -3,7 +3,7 @@ import { loadApiKey, withoutTrailingSlash } from "@ai-sdk/provider-utils";
 
 // src/anthropic-messages-language-model.ts
 import {
-  UnsupportedFunctionalityError as UnsupportedFunctionalityError2
+  UnsupportedFunctionalityError as UnsupportedFunctionalityError3
 } from "@ai-sdk/provider";
 import {
   combineHeaders,
@@ -38,6 +38,7 @@ function convertToAnthropicMessagesPrompt({
   cacheControl: isCacheControlEnabled
 }) {
   var _a, _b, _c, _d;
+  const betas = /* @__PURE__ */ new Set();
   const blocks = groupIntoBlocks(prompt);
   let system = void 0;
   const messages = [];
@@ -103,6 +104,29 @@ function convertToAnthropicMessagesPrompt({
                     });
                     break;
                   }
+                  case "file": {
+                    if (part.data instanceof URL) {
+                      throw new UnsupportedFunctionalityError({
+                        functionality: "Image URLs in user messages"
+                      });
+                    }
+                    if (part.mimeType !== "application/pdf") {
+                      throw new UnsupportedFunctionalityError({
+                        functionality: "Non-PDF files in user messages"
+                      });
+                    }
+                    betas.add("pdfs-2024-09-25");
+                    anthropicContent.push({
+                      type: "document",
+                      source: {
+                        type: "base64",
+                        media_type: "application/pdf",
+                        data: part.data
+                      },
+                      cache_control: cacheControl
+                    });
+                    break;
+                  }
                 }
               }
               break;
@@ -112,10 +136,31 @@ function convertToAnthropicMessagesPrompt({
                 const part = content[i2];
                 const isLastPart = i2 === content.length - 1;
                 const cacheControl = (_c = getCacheControl(part.providerMetadata)) != null ? _c : isLastPart ? getCacheControl(message.providerMetadata) : void 0;
+                const toolResultContent = part.content != null ? part.content.map((part2) => {
+                  var _a2;
+                  switch (part2.type) {
+                    case "text":
+                      return {
+                        type: "text",
+                        text: part2.text,
+                        cache_control: void 0
+                      };
+                    case "image":
+                      return {
+                        type: "image",
+                        source: {
+                          type: "base64",
+                          media_type: (_a2 = part2.mimeType) != null ? _a2 : "image/jpeg",
+                          data: part2.data
+                        },
+                        cache_control: void 0
+                      };
+                  }
+                }) : JSON.stringify(part.result);
                 anthropicContent.push({
                   type: "tool_result",
                   tool_use_id: part.toolCallId,
-                  content: JSON.stringify(part.result),
+                  content: toolResultContent,
                   is_error: part.isError,
                   cache_control: cacheControl
                 });
@@ -176,8 +221,8 @@ function convertToAnthropicMessagesPrompt({
     }
   }
   return {
-    system,
-    messages
+    prompt: { system, messages },
+    betas
   };
 }
 function groupIntoBlocks(prompt) {
@@ -242,6 +287,105 @@ function mapAnthropicStopReason(finishReason) {
   }
 }
 
+// src/anthropic-prepare-tools.ts
+import {
+  UnsupportedFunctionalityError as UnsupportedFunctionalityError2
+} from "@ai-sdk/provider";
+function prepareTools(mode) {
+  var _a;
+  const tools = ((_a = mode.tools) == null ? void 0 : _a.length) ? mode.tools : void 0;
+  const toolWarnings = [];
+  const betas = /* @__PURE__ */ new Set();
+  if (tools == null) {
+    return { tools: void 0, tool_choice: void 0, toolWarnings, betas };
+  }
+  const anthropicTools2 = [];
+  for (const tool of tools) {
+    switch (tool.type) {
+      case "function":
+        anthropicTools2.push({
+          name: tool.name,
+          description: tool.description,
+          input_schema: tool.parameters
+        });
+        break;
+      case "provider-defined":
+        betas.add("computer-use-2024-10-22");
+        switch (tool.id) {
+          case "anthropic.computer_20241022":
+            anthropicTools2.push({
+              name: tool.name,
+              type: "computer_20241022",
+              display_width_px: tool.args.displayWidthPx,
+              display_height_px: tool.args.displayHeightPx,
+              display_number: tool.args.displayNumber
+            });
+            break;
+          case "anthropic.text_editor_20241022":
+            anthropicTools2.push({
+              name: tool.name,
+              type: "text_editor_20241022"
+            });
+            break;
+          case "anthropic.bash_20241022":
+            anthropicTools2.push({
+              name: tool.name,
+              type: "bash_20241022"
+            });
+            break;
+          default:
+            toolWarnings.push({ type: "unsupported-tool", tool });
+            break;
+        }
+        break;
+      default:
+        toolWarnings.push({ type: "unsupported-tool", tool });
+        break;
+    }
+  }
+  const toolChoice = mode.toolChoice;
+  if (toolChoice == null) {
+    return {
+      tools: anthropicTools2,
+      tool_choice: void 0,
+      toolWarnings,
+      betas
+    };
+  }
+  const type = toolChoice.type;
+  switch (type) {
+    case "auto":
+      return {
+        tools: anthropicTools2,
+        tool_choice: { type: "auto" },
+        toolWarnings,
+        betas
+      };
+    case "required":
+      return {
+        tools: anthropicTools2,
+        tool_choice: { type: "any" },
+        toolWarnings,
+        betas
+      };
+    case "none":
+      return { tools: void 0, tool_choice: void 0, toolWarnings, betas };
+    case "tool":
+      return {
+        tools: anthropicTools2,
+        tool_choice: { type: "tool", name: toolChoice.toolName },
+        toolWarnings,
+        betas
+      };
+    default: {
+      const _exhaustiveCheck = type;
+      throw new UnsupportedFunctionalityError2({
+        functionality: `Unsupported tool choice type: ${_exhaustiveCheck}`
+      });
+    }
+  }
+}
+
 // src/anthropic-messages-language-model.ts
 var AnthropicMessagesLanguageModel = class {
   constructor(modelId, settings, config) {
@@ -296,7 +440,7 @@ var AnthropicMessagesLanguageModel = class {
         details: "JSON response format is not supported."
       });
     }
-    const messagesPrompt = convertToAnthropicMessagesPrompt({
+    const { prompt: messagesPrompt, betas: messagesBetas } = convertToAnthropicMessagesPrompt({
       prompt,
       cacheControl: (_a = this.settings.cacheControl) != null ? _a : false
     });
@@ -317,13 +461,20 @@ var AnthropicMessagesLanguageModel = class {
     };
     switch (type) {
       case "regular": {
+        const {
+          tools,
+          tool_choice,
+          toolWarnings,
+          betas: toolsBetas
+        } = prepareTools(mode);
         return {
-          args: { ...baseArgs, ...prepareToolsAndToolChoice(mode) },
-          warnings
+          args: { ...baseArgs, tools, tool_choice },
+          warnings: [...warnings, ...toolWarnings],
+          betas: /* @__PURE__ */ new Set([...messagesBetas, ...toolsBetas])
         };
       }
       case "object-json": {
-        throw new UnsupportedFunctionalityError2({
+        throw new UnsupportedFunctionalityError3({
           functionality: "json-mode object generation"
         });
       }
@@ -335,7 +486,8 @@ var AnthropicMessagesLanguageModel = class {
             tools: [{ name, description, input_schema: parameters }],
             tool_choice: { type: "tool", name }
           },
-          warnings
+          warnings,
+          betas: messagesBetas
         };
       }
       default: {
@@ -344,19 +496,25 @@ var AnthropicMessagesLanguageModel = class {
       }
     }
   }
-  getHeaders(optionHeaders) {
+  getHeaders({
+    betas,
+    headers
+  }) {
+    if (this.settings.cacheControl) {
+      betas.add("prompt-caching-2024-07-31");
+    }
     return combineHeaders(
       this.config.headers(),
-      this.settings.cacheControl ? { "anthropic-beta": "prompt-caching-2024-07-31" } : {},
-      optionHeaders
+      betas.size > 0 ? { "anthropic-beta": Array.from(betas).join(",") } : {},
+      headers
     );
   }
   async doGenerate(options) {
     var _a, _b, _c, _d;
-    const { args, warnings } = await this.getArgs(options);
+    const { args, warnings, betas } = await this.getArgs(options);
     const { responseHeaders, value: response } = await postJsonToApi({
       url: `${this.config.baseURL}/messages`,
-      headers: this.getHeaders(options.headers),
+      headers: this.getHeaders({ betas, headers: options.headers }),
       body: args,
       failedResponseHandler: anthropicFailedResponseHandler,
       successfulResponseHandler: createJsonResponseHandler(
@@ -406,15 +564,17 @@ var AnthropicMessagesLanguageModel = class {
           cacheCreationInputTokens: (_c = response.usage.cache_creation_input_tokens) != null ? _c : null,
           cacheReadInputTokens: (_d = response.usage.cache_read_input_tokens) != null ? _d : null
         }
-      } : void 0
+      } : void 0,
+      request: { body: JSON.stringify(args) }
     };
   }
   async doStream(options) {
-    const { args, warnings } = await this.getArgs(options);
+    const { args, warnings, betas } = await this.getArgs(options);
+    const body = { ...args, stream: true };
     const { responseHeaders, value: response } = await postJsonToApi({
       url: `${this.config.baseURL}/messages`,
-      headers: this.getHeaders(options.headers),
-      body: { ...args, stream: true },
+      headers: this.getHeaders({ betas, headers: options.headers }),
+      body,
       failedResponseHandler: anthropicFailedResponseHandler,
       successfulResponseHandler: createEventSourceResponseHandler(
         anthropicMessagesChunkSchema
@@ -557,7 +717,8 @@ var AnthropicMessagesLanguageModel = class {
       ),
       rawCall: { rawPrompt, rawSettings },
       rawResponse: { headers: responseHeaders },
-      warnings
+      warnings,
+      request: { body: JSON.stringify(body) }
     };
   }
 };
@@ -653,40 +814,6 @@ var anthropicMessagesChunkSchema = z2.discriminatedUnion("type", [
     type: z2.literal("ping")
   })
 ]);
-function prepareToolsAndToolChoice(mode) {
-  var _a;
-  const tools = ((_a = mode.tools) == null ? void 0 : _a.length) ? mode.tools : void 0;
-  if (tools == null) {
-    return { tools: void 0, tool_choice: void 0 };
-  }
-  const mappedTools = tools.map((tool) => ({
-    name: tool.name,
-    description: tool.description,
-    input_schema: tool.parameters
-  }));
-  const toolChoice = mode.toolChoice;
-  if (toolChoice == null) {
-    return { tools: mappedTools, tool_choice: void 0 };
-  }
-  const type = toolChoice.type;
-  switch (type) {
-    case "auto":
-      return { tools: mappedTools, tool_choice: { type: "auto" } };
-    case "required":
-      return { tools: mappedTools, tool_choice: { type: "any" } };
-    case "none":
-      return { tools: void 0, tool_choice: void 0 };
-    case "tool":
-      return {
-        tools: mappedTools,
-        tool_choice: { type: "tool", name: toolChoice.toolName }
-      };
-    default: {
-      const _exhaustiveCheck = type;
-      throw new Error(`Unsupported tool choice type: ${_exhaustiveCheck}`);
-    }
-  }
-}
 
 // src/anthropic-facade.ts
 var Anthropic = class {
@@ -735,6 +862,79 @@ import {
   loadApiKey as loadApiKey2,
   withoutTrailingSlash as withoutTrailingSlash2
 } from "@ai-sdk/provider-utils";
+
+// src/anthropic-tools.ts
+import { z as z3 } from "zod";
+var Bash20241022Parameters = z3.object({
+  command: z3.string(),
+  restart: z3.boolean().optional()
+});
+function bashTool_20241022(options = {}) {
+  return {
+    type: "provider-defined",
+    id: "anthropic.bash_20241022",
+    args: {},
+    parameters: Bash20241022Parameters,
+    execute: options.execute,
+    experimental_toToolResultContent: options.experimental_toToolResultContent
+  };
+}
+var TextEditor20241022Parameters = z3.object({
+  command: z3.enum(["view", "create", "str_replace", "insert", "undo_edit"]),
+  path: z3.string(),
+  file_text: z3.string().optional(),
+  insert_line: z3.number().int().optional(),
+  new_str: z3.string().optional(),
+  old_str: z3.string().optional(),
+  view_range: z3.array(z3.number().int()).optional()
+});
+function textEditorTool_20241022(options = {}) {
+  return {
+    type: "provider-defined",
+    id: "anthropic.text_editor_20241022",
+    args: {},
+    parameters: TextEditor20241022Parameters,
+    execute: options.execute,
+    experimental_toToolResultContent: options.experimental_toToolResultContent
+  };
+}
+var Computer20241022Parameters = z3.object({
+  action: z3.enum([
+    "key",
+    "type",
+    "mouse_move",
+    "left_click",
+    "left_click_drag",
+    "right_click",
+    "middle_click",
+    "double_click",
+    "screenshot",
+    "cursor_position"
+  ]),
+  coordinate: z3.array(z3.number().int()).optional(),
+  text: z3.string().optional()
+});
+function computerTool_20241022(options) {
+  return {
+    type: "provider-defined",
+    id: "anthropic.computer_20241022",
+    args: {
+      displayWidthPx: options.displayWidthPx,
+      displayHeightPx: options.displayHeightPx,
+      displayNumber: options.displayNumber
+    },
+    parameters: Computer20241022Parameters,
+    execute: options.execute,
+    experimental_toToolResultContent: options.experimental_toToolResultContent
+  };
+}
+var anthropicTools = {
+  bash_20241022: bashTool_20241022,
+  textEditor_20241022: textEditorTool_20241022,
+  computer_20241022: computerTool_20241022
+};
+
+// src/anthropic-provider.ts
 function createAnthropic(options = {}) {
   var _a, _b;
   const baseURL = (_b = withoutTrailingSlash2((_a = options.baseURL) != null ? _a : options.baseUrl)) != null ? _b : "https://api.anthropic.com/v1";
@@ -767,6 +967,7 @@ function createAnthropic(options = {}) {
   provider.textEmbeddingModel = (modelId) => {
     throw new NoSuchModelError({ modelId, modelType: "textEmbeddingModel" });
   };
+  provider.tools = anthropicTools;
   return provider;
 }
 var anthropic = createAnthropic();