@langchain/anthropic 0.2.4 → 0.2.5

package/dist/chat_models.cjs

@@ -12,6 +12,7 @@ const runnables_1 = require("@langchain/core/runnables");
 const types_1 = require("@langchain/core/utils/types");
 const stream_1 = require("@langchain/core/utils/stream");
 const output_parsers_js_1 = require("./output_parsers.cjs");
+const utils_js_1 = require("./utils.cjs");
 function _toolsInParams(params) {
     return !!(params.tools && params.tools.length > 0);
 }
@@ -49,6 +50,7 @@ function anthropicResponseToChatMessages(messages, additionalKwargs) {
                     additional_kwargs: additionalKwargs,
                     usage_metadata: usageMetadata,
                     response_metadata: additionalKwargs,
+                    id: additionalKwargs.id,
                 }),
             },
         ];
@@ -65,6 +67,7 @@ function anthropicResponseToChatMessages(messages, additionalKwargs) {
                     tool_calls: toolCalls,
                     usage_metadata: usageMetadata,
                     response_metadata: additionalKwargs,
+                    id: additionalKwargs.id,
                 }),
             },
         ];
@@ -96,11 +99,13 @@ function _makeMessageChunkFromAnthropicEvent(data, fields) {
                 total_tokens: usage.input_tokens + usage.output_tokens,
             };
         }
+        console.log("data.message", data.message);
         return {
             chunk: new messages_1.AIMessageChunk({
                 content: fields.coerceContentToString ? "" : [],
                 additional_kwargs: filteredAdditionalKwargs,
                 usage_metadata: usageMetadata,
+                id: data.message.id,
             }),
             usageData: usageDataCopy,
         };
@@ -667,22 +672,7 @@ class ChatAnthropicMessages extends chat_models_1.BaseChatModel {
      * Get the parameters used to invoke the model
      */
     invocationParams(options) {
-        let tool_choice;
-        if (options?.tool_choice) {
-            if (options?.tool_choice === "any") {
-                tool_choice = {
-                    type: "any",
-                };
-            }
-            else if (options?.tool_choice === "auto") {
-                tool_choice = {
-                    type: "auto",
-                };
-            }
-            else {
-                tool_choice = options?.tool_choice;
-            }
-        }
+        const tool_choice = (0, utils_js_1.handleToolChoice)(options?.tool_choice);
         return {
             model: this.model,
             temperature: this.temperature,
@@ -768,6 +758,7 @@ class ChatAnthropicMessages extends chat_models_1.BaseChatModel {
                     tool_call_chunks: newToolCallChunk ? [newToolCallChunk] : undefined,
                     usage_metadata: chunk.usage_metadata,
                     response_metadata: chunk.response_metadata,
+                    id: chunk.id,
                 }),
                 text: token ?? "",
             });

package/dist/chat_models.d.ts

@@ -3,24 +3,21 @@ import type { Stream } from "@anthropic-ai/sdk/streaming";
 import { CallbackManagerForLLMRun } from "@langchain/core/callbacks/manager";
 import { AIMessageChunk, type BaseMessage } from "@langchain/core/messages";
 import { ChatGeneration, ChatGenerationChunk, type ChatResult } from "@langchain/core/outputs";
-import { BaseChatModel, LangSmithParams, type BaseChatModelParams } from "@langchain/core/language_models/chat_models";
-import { type StructuredOutputMethodOptions, type BaseLanguageModelCallOptions, type BaseLanguageModelInput, type ToolDefinition } from "@langchain/core/language_models/base";
+import { BaseChatModel, BaseChatModelCallOptions, LangSmithParams, type BaseChatModelParams } from "@langchain/core/language_models/chat_models";
+import { type StructuredOutputMethodOptions, type BaseLanguageModelInput, type ToolDefinition } from "@langchain/core/language_models/base";
 import { StructuredToolInterface } from "@langchain/core/tools";
 import { Runnable, RunnableToolLike } from "@langchain/core/runnables";
 import { ToolCall } from "@langchain/core/messages/tool";
 import { z } from "zod";
 import type { Tool as AnthropicTool } from "@anthropic-ai/sdk/resources/index.mjs";
 import { AnthropicToolResponse } from "./types.js";
+import { AnthropicToolChoice, AnthropicToolTypes } from "./utils.js";
 type AnthropicMessageCreateParams = Anthropic.MessageCreateParamsNonStreaming;
 type AnthropicStreamingMessageCreateParams = Anthropic.MessageCreateParamsStreaming;
 type AnthropicMessageStreamEvent = Anthropic.MessageStreamEvent;
 type AnthropicRequestOptions = Anthropic.RequestOptions;
-type AnthropicToolChoice = {
-    type: "tool";
-    name: string;
-} | "any" | "auto";
-export interface ChatAnthropicCallOptions extends BaseLanguageModelCallOptions, Pick<AnthropicInput, "streamUsage"> {
-    tools?: (StructuredToolInterface | AnthropicTool | Record<string, unknown> | ToolDefinition | RunnableToolLike)[];
+export interface ChatAnthropicCallOptions extends BaseChatModelCallOptions, Pick<AnthropicInput, "streamUsage"> {
+    tools?: AnthropicToolTypes[];
     /**
      * Whether or not to specify what tool the model should use
      * @default "auto"

package/dist/chat_models.js

@@ -9,6 +9,7 @@ import { RunnablePassthrough, RunnableSequence, } from "@langchain/core/runnable
 import { isZodSchema } from "@langchain/core/utils/types";
 import { concat } from "@langchain/core/utils/stream";
 import { AnthropicToolsOutputParser, extractToolCalls, } from "./output_parsers.js";
+import { handleToolChoice, } from "./utils.js";
 function _toolsInParams(params) {
     return !!(params.tools && params.tools.length > 0);
 }
@@ -46,6 +47,7 @@ function anthropicResponseToChatMessages(messages, additionalKwargs) {
                     additional_kwargs: additionalKwargs,
                     usage_metadata: usageMetadata,
                     response_metadata: additionalKwargs,
+                    id: additionalKwargs.id,
                 }),
             },
         ];
@@ -62,6 +64,7 @@ function anthropicResponseToChatMessages(messages, additionalKwargs) {
                     tool_calls: toolCalls,
                     usage_metadata: usageMetadata,
                     response_metadata: additionalKwargs,
+                    id: additionalKwargs.id,
                 }),
             },
         ];
@@ -93,11 +96,13 @@ function _makeMessageChunkFromAnthropicEvent(data, fields) {
                 total_tokens: usage.input_tokens + usage.output_tokens,
             };
         }
+        console.log("data.message", data.message);
         return {
             chunk: new AIMessageChunk({
                 content: fields.coerceContentToString ? "" : [],
                 additional_kwargs: filteredAdditionalKwargs,
                 usage_metadata: usageMetadata,
+                id: data.message.id,
             }),
             usageData: usageDataCopy,
         };
@@ -663,22 +668,7 @@ export class ChatAnthropicMessages extends BaseChatModel {
      * Get the parameters used to invoke the model
      */
     invocationParams(options) {
-        let tool_choice;
-        if (options?.tool_choice) {
-            if (options?.tool_choice === "any") {
-                tool_choice = {
-                    type: "any",
-                };
-            }
-            else if (options?.tool_choice === "auto") {
-                tool_choice = {
-                    type: "auto",
-                };
-            }
-            else {
-                tool_choice = options?.tool_choice;
-            }
-        }
+        const tool_choice = handleToolChoice(options?.tool_choice);
         return {
             model: this.model,
             temperature: this.temperature,
@@ -764,6 +754,7 @@ export class ChatAnthropicMessages extends BaseChatModel {
                     tool_call_chunks: newToolCallChunk ? [newToolCallChunk] : undefined,
                     usage_metadata: chunk.usage_metadata,
                     response_metadata: chunk.response_metadata,
+                    id: chunk.id,
                 }),
                 text: token ?? "",
             });

package/dist/utils.cjs

@@ -0,0 +1,28 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.handleToolChoice = void 0;
+function handleToolChoice(toolChoice) {
+    if (!toolChoice) {
+        return undefined;
+    }
+    else if (toolChoice === "any") {
+        return {
+            type: "any",
+        };
+    }
+    else if (toolChoice === "auto") {
+        return {
+            type: "auto",
+        };
+    }
+    else if (typeof toolChoice === "string") {
+        return {
+            type: "tool",
+            name: toolChoice,
+        };
+    }
+    else {
+        return toolChoice;
+    }
+}
+exports.handleToolChoice = handleToolChoice;

package/dist/utils.d.ts

@@ -0,0 +1,10 @@
+import type { MessageCreateParams, Tool as AnthropicTool } from "@anthropic-ai/sdk/resources/index.mjs";
+import { ToolDefinition } from "@langchain/core/language_models/base";
+import { RunnableToolLike } from "@langchain/core/runnables";
+import { StructuredToolInterface } from "@langchain/core/tools";
+export type AnthropicToolChoice = {
+    type: "tool";
+    name: string;
+} | "any" | "auto" | "none" | string;
+export type AnthropicToolTypes = StructuredToolInterface | AnthropicTool | Record<string, unknown> | ToolDefinition | RunnableToolLike;
+export declare function handleToolChoice(toolChoice?: AnthropicToolChoice): MessageCreateParams.ToolChoiceAuto | MessageCreateParams.ToolChoiceAny | MessageCreateParams.ToolChoiceTool | undefined;

package/dist/utils.js

@@ -0,0 +1,24 @@
+export function handleToolChoice(toolChoice) {
+    if (!toolChoice) {
+        return undefined;
+    }
+    else if (toolChoice === "any") {
+        return {
+            type: "any",
+        };
+    }
+    else if (toolChoice === "auto") {
+        return {
+            type: "auto",
+        };
+    }
+    else if (typeof toolChoice === "string") {
+        return {
+            type: "tool",
+            name: toolChoice,
+        };
+    }
+    else {
+        return toolChoice;
+    }
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langchain/anthropic",
-  "version": "0.2.4",
+  "version": "0.2.5",
   "description": "Anthropic integrations for LangChain.js",
   "type": "module",
   "engines": {