@ai-sdk/anthropic 0.0.51 → 0.0.54

package/dist/index.js

@@ -30,7 +30,7 @@ module.exports = __toCommonJS(src_exports);
 var import_provider_utils4 = require("@ai-sdk/provider-utils");
 
 // src/anthropic-messages-language-model.ts
-var import_provider2 = require("@ai-sdk/provider");
+var import_provider3 = require("@ai-sdk/provider");
 var import_provider_utils3 = require("@ai-sdk/provider-utils");
 var import_zod2 = require("zod");
 
@@ -57,6 +57,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 = [];
@@ -122,6 +123,29 @@ function convertToAnthropicMessagesPrompt({
                     });
                     break;
                   }
+                  case "file": {
+                    if (part.data instanceof URL) {
+                      throw new import_provider.UnsupportedFunctionalityError({
+                        functionality: "Image URLs in user messages"
+                      });
+                    }
+                    if (part.mimeType !== "application/pdf") {
+                      throw new import_provider.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;
@@ -131,10 +155,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
                 });
@@ -195,8 +240,8 @@ function convertToAnthropicMessagesPrompt({
     }
   }
   return {
-    system,
-    messages
+    prompt: { system, messages },
+    betas
   };
 }
 function groupIntoBlocks(prompt) {
@@ -261,6 +306,103 @@ function mapAnthropicStopReason(finishReason) {
   }
 }
 
+// src/anthropic-prepare-tools.ts
+var import_provider2 = require("@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 import_provider2.UnsupportedFunctionalityError({
+        functionality: `Unsupported tool choice type: ${_exhaustiveCheck}`
+      });
+    }
+  }
+}
+
 // src/anthropic-messages-language-model.ts
 var AnthropicMessagesLanguageModel = class {
   constructor(modelId, settings, config) {
@@ -315,7 +457,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
     });
@@ -336,13 +478,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 import_provider2.UnsupportedFunctionalityError({
+        throw new import_provider3.UnsupportedFunctionalityError({
           functionality: "json-mode object generation"
         });
       }
@@ -354,7 +503,8 @@ var AnthropicMessagesLanguageModel = class {
             tools: [{ name, description, input_schema: parameters }],
             tool_choice: { type: "tool", name }
           },
-          warnings
+          warnings,
+          betas: messagesBetas
         };
       }
       default: {
@@ -363,19 +513,25 @@ var AnthropicMessagesLanguageModel = class {
       }
     }
   }
-  getHeaders(optionHeaders) {
+  getHeaders({
+    betas,
+    headers
+  }) {
+    if (this.settings.cacheControl) {
+      betas.add("prompt-caching-2024-07-31");
+    }
     return (0, import_provider_utils3.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 (0, import_provider_utils3.postJsonToApi)({
       url: `${this.config.baseURL}/messages`,
-      headers: this.getHeaders(options.headers),
+      headers: this.getHeaders({ betas, headers: options.headers }),
       body: args,
       failedResponseHandler: anthropicFailedResponseHandler,
       successfulResponseHandler: (0, import_provider_utils3.createJsonResponseHandler)(
@@ -425,15 +581,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 (0, import_provider_utils3.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: (0, import_provider_utils3.createEventSourceResponseHandler)(
         anthropicMessagesChunkSchema
@@ -576,7 +734,8 @@ var AnthropicMessagesLanguageModel = class {
       ),
       rawCall: { rawPrompt, rawSettings },
       rawResponse: { headers: responseHeaders },
-      warnings
+      warnings,
+      request: { body: JSON.stringify(body) }
     };
   }
 };
@@ -672,40 +831,6 @@ var anthropicMessagesChunkSchema = import_zod2.z.discriminatedUnion("type", [
     type: import_zod2.z.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 {
@@ -747,8 +872,81 @@ var Anthropic = class {
 };
 
 // src/anthropic-provider.ts
-var import_provider3 = require("@ai-sdk/provider");
+var import_provider4 = require("@ai-sdk/provider");
 var import_provider_utils5 = require("@ai-sdk/provider-utils");
+
+// src/anthropic-tools.ts
+var import_zod3 = require("zod");
+var Bash20241022Parameters = import_zod3.z.object({
+  command: import_zod3.z.string(),
+  restart: import_zod3.z.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 = import_zod3.z.object({
+  command: import_zod3.z.enum(["view", "create", "str_replace", "insert", "undo_edit"]),
+  path: import_zod3.z.string(),
+  file_text: import_zod3.z.string().optional(),
+  insert_line: import_zod3.z.number().int().optional(),
+  new_str: import_zod3.z.string().optional(),
+  old_str: import_zod3.z.string().optional(),
+  view_range: import_zod3.z.array(import_zod3.z.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 = import_zod3.z.object({
+  action: import_zod3.z.enum([
+    "key",
+    "type",
+    "mouse_move",
+    "left_click",
+    "left_click_drag",
+    "right_click",
+    "middle_click",
+    "double_click",
+    "screenshot",
+    "cursor_position"
+  ]),
+  coordinate: import_zod3.z.array(import_zod3.z.number().int()).optional(),
+  text: import_zod3.z.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 = (0, import_provider_utils5.withoutTrailingSlash)((_a = options.baseURL) != null ? _a : options.baseUrl)) != null ? _b : "https://api.anthropic.com/v1";
@@ -779,8 +977,9 @@ function createAnthropic(options = {}) {
   provider.chat = createChatModel;
   provider.messages = createChatModel;
   provider.textEmbeddingModel = (modelId) => {
-    throw new import_provider3.NoSuchModelError({ modelId, modelType: "textEmbeddingModel" });
+    throw new import_provider4.NoSuchModelError({ modelId, modelType: "textEmbeddingModel" });
   };
+  provider.tools = anthropicTools;
   return provider;
 }
 var anthropic = createAnthropic();