@langchain/langgraph 0.0.26 → 0.0.27

package/dist/tests/prebuilt.test.js

@@ -4,7 +4,6 @@ import { PromptTemplate } from "@langchain/core/prompts";
 import { StructuredTool, Tool } from "@langchain/core/tools";
 import { FakeStreamingLLM } from "@langchain/core/utils/testing";
 import { AIMessage, HumanMessage, SystemMessage, ToolMessage, } from "@langchain/core/messages";
-import { RunnableLambda } from "@langchain/core/runnables";
 import { z } from "zod";
 import { FakeToolCallingChatModel } from "./utils.js";
 import { createAgentExecutor, createReactAgent } from "../prebuilt/index.js";
@@ -227,6 +226,38 @@ describe("createReactAgent", () => {
             return `result for ${input?.query}`;
         }
     }
+    class SearchAPIWithArtifact extends StructuredTool {
+        constructor() {
+            super(...arguments);
+            Object.defineProperty(this, "name", {
+                enumerable: true,
+                configurable: true,
+                writable: true,
+                value: "search_api"
+            });
+            Object.defineProperty(this, "description", {
+                enumerable: true,
+                configurable: true,
+                writable: true,
+                value: "A simple API that returns the input string."
+            });
+            Object.defineProperty(this, "schema", {
+                enumerable: true,
+                configurable: true,
+                writable: true,
+                value: searchSchema
+            });
+            Object.defineProperty(this, "responseFormat", {
+                enumerable: true,
+                configurable: true,
+                writable: true,
+                value: "content_and_artifact"
+            });
+        }
+        async _call(_) {
+            return ["some response format", Buffer.from("123")];
+        }
+    }
     const tools = [new SearchAPI()];
     it("Can use string message modifier", async () => {
         const llm = new FakeToolCallingChatModel({
@@ -299,100 +330,41 @@ describe("createReactAgent", () => {
             new AIMessage("result2"),
         ]);
     });
-    it("Can use custom function message modifier", async () => {
-        const aiM1 = new AIMessage({
-            content: "result1",
-            tool_calls: [
-                { name: "search_api", id: "tool_abcd123", args: { query: "foo" } },
-            ],
-        });
-        const aiM2 = new AIMessage("result2");
+    it("Works with tools that return content_and_artifact response format", async () => {
         const llm = new FakeToolCallingChatModel({
-            responses: [aiM1, aiM2],
-        });
-        const messageModifier = (messages) => [
-            new SystemMessage("You are a helpful assistant"),
-            ...messages,
-        ];
-        const agent = createReactAgent({ llm, tools, messageModifier });
-        const result = await agent.invoke({
-            messages: [new HumanMessage("Hello Input!")],
-        });
-        expect(result.messages).toEqual([
-            new HumanMessage("Hello Input!"),
-            aiM1,
-            new ToolMessage({
-                name: "search_api",
-                content: "result for foo",
-                tool_call_id: "tool_abcd123",
-            }),
-            aiM2,
-        ]);
-    });
-    it("Can use async custom function message modifier", async () => {
-        const aiM1 = new AIMessage({
-            content: "result1",
-            tool_calls: [
-                { name: "search_api", id: "tool_abcd123", args: { query: "foo" } },
+            responses: [
+                new AIMessage({
+                    content: "result1",
+                    tool_calls: [
+                        { name: "search_api", id: "tool_abcd123", args: { query: "foo" } },
+                    ],
+                }),
+                new AIMessage("result2"),
             ],
         });
-        const aiM2 = new AIMessage("result2");
-        const llm = new FakeToolCallingChatModel({
-            responses: [aiM1, aiM2],
+        const agent = createReactAgent({
+            llm,
+            tools: [new SearchAPIWithArtifact()],
+            messageModifier: "You are a helpful assistant",
         });
-        const messageModifier = async (messages) => [
-            new SystemMessage("You are a helpful assistant"),
-            ...messages,
-        ];
-        const agent = createReactAgent({ llm, tools, messageModifier });
         const result = await agent.invoke({
             messages: [new HumanMessage("Hello Input!")],
         });
         expect(result.messages).toEqual([
             new HumanMessage("Hello Input!"),
-            aiM1,
-            new ToolMessage({
-                name: "search_api",
-                content: "result for foo",
-                tool_call_id: "tool_abcd123",
+            new AIMessage({
+                content: "result1",
+                tool_calls: [
+                    { name: "search_api", id: "tool_abcd123", args: { query: "foo" } },
+                ],
             }),
-            aiM2,
-        ]);
-    });
-    it("Can use RunnableLambda message modifier", async () => {
-        const aiM1 = new AIMessage({
-            content: "result1",
-            tool_calls: [
-                { name: "search_api", id: "tool_abcd123", args: { query: "foo" } },
-            ],
-        });
-        const aiM2 = new AIMessage("result2");
-        const llm = new FakeToolCallingChatModel({
-            responses: [aiM1, aiM2],
-        });
-        const messageModifier = new RunnableLambda({
-            func: (messages) => [
-                new SystemMessage("You are a helpful assistant"),
-                ...messages,
-            ],
-        });
-        const agent = createReactAgent({ llm, tools, messageModifier });
-        const result = await agent.invoke({
-            messages: [
-                new HumanMessage("Hello Input!"),
-                new HumanMessage("Another Input!"),
-            ],
-        });
-        expect(result.messages).toEqual([
-            new HumanMessage("Hello Input!"),
-            new HumanMessage("Another Input!"),
-            aiM1,
             new ToolMessage({
                 name: "search_api",
-                content: "result for foo",
+                content: "some response format",
                 tool_call_id: "tool_abcd123",
+                artifact: Buffer.from("123"),
             }),
-            aiM2,
+            new AIMessage("result2"),
         ]);
     });
 });

package/dist/tests/tracing.test.js

@@ -53,7 +53,12 @@ it("should pass config through if importing from the primary entrypoint", async
             name: "__start__",
             tags: ["graph:step:0", "langsmith:hidden"],
             run_id: expect.any(String),
-            metadata: {},
+            metadata: {
+                langgraph_node: "__start__",
+                langgraph_step: 0,
+                langgraph_task_idx: 0,
+                langgraph_triggers: ["__pregel_tasks"],
+            },
         },
         {
             event: "on_chain_end",
@@ -66,7 +71,12 @@ it("should pass config through if importing from the primary entrypoint", async
             run_id: expect.any(String),
             name: "__start__",
             tags: ["graph:step:0", "langsmith:hidden"],
-            metadata: {},
+            metadata: {
+                langgraph_node: "__start__",
+                langgraph_step: 0,
+                langgraph_task_idx: 0,
+                langgraph_triggers: ["__pregel_tasks"],
+            },
         },
         {
             event: "on_chain_start",
@@ -78,7 +88,12 @@ it("should pass config through if importing from the primary entrypoint", async
             name: "testnode",
             tags: ["graph:step:1"],
             run_id: expect.any(String),
-            metadata: {},
+            metadata: {
+                langgraph_node: "testnode",
+                langgraph_step: 1,
+                langgraph_task_idx: 0,
+                langgraph_triggers: ["__pregel_tasks"],
+            },
         },
         {
             event: "on_chain_start",
@@ -90,7 +105,12 @@ it("should pass config through if importing from the primary entrypoint", async
             name: "RunnableLambda",
             tags: ["seq:step:1"],
             run_id: expect.any(String),
-            metadata: {},
+            metadata: {
+                langgraph_node: "testnode",
+                langgraph_step: 1,
+                langgraph_task_idx: 0,
+                langgraph_triggers: ["__pregel_tasks"],
+            },
         },
         {
             event: "on_chat_model_start",
@@ -103,6 +123,10 @@ it("should pass config through if importing from the primary entrypoint", async
             tags: [],
             run_id: expect.any(String),
             metadata: {
+                langgraph_node: "testnode",
+                langgraph_step: 1,
+                langgraph_task_idx: 0,
+                langgraph_triggers: ["__pregel_tasks"],
                 ls_model_type: "chat",
                 ls_stop: undefined,
             },
@@ -119,6 +143,10 @@ it("should pass config through if importing from the primary entrypoint", async
             name: "model_call",
             tags: [],
             metadata: {
+                langgraph_node: "testnode",
+                langgraph_step: 1,
+                langgraph_task_idx: 0,
+                langgraph_triggers: ["__pregel_tasks"],
                 ls_model_type: "chat",
                 ls_stop: undefined,
             },
@@ -136,7 +164,12 @@ it("should pass config through if importing from the primary entrypoint", async
             run_id: expect.any(String),
             name: "RunnableLambda",
             tags: ["seq:step:1"],
-            metadata: {},
+            metadata: {
+                langgraph_node: "testnode",
+                langgraph_step: 1,
+                langgraph_task_idx: 0,
+                langgraph_triggers: ["__pregel_tasks"],
+            },
         },
         {
             event: "on_chain_start",
@@ -148,7 +181,12 @@ it("should pass config through if importing from the primary entrypoint", async
             name: "ChannelWrite<messages,testnode>",
             tags: ["seq:step:2", "langsmith:hidden"],
             run_id: expect.any(String),
-            metadata: {},
+            metadata: {
+                langgraph_node: "testnode",
+                langgraph_step: 1,
+                langgraph_task_idx: 0,
+                langgraph_triggers: ["__pregel_tasks"],
+            },
         },
         {
             event: "on_chain_end",
@@ -161,7 +199,12 @@ it("should pass config through if importing from the primary entrypoint", async
             run_id: expect.any(String),
             name: "ChannelWrite<messages,testnode>",
             tags: ["seq:step:2", "langsmith:hidden"],
-            metadata: {},
+            metadata: {
+                langgraph_node: "testnode",
+                langgraph_step: 1,
+                langgraph_task_idx: 0,
+                langgraph_triggers: ["__pregel_tasks"],
+            },
         },
         {
             event: "on_chain_start",
@@ -173,7 +216,12 @@ it("should pass config through if importing from the primary entrypoint", async
             name: "func",
             tags: ["seq:step:3"],
             run_id: expect.any(String),
-            metadata: {},
+            metadata: {
+                langgraph_node: "testnode",
+                langgraph_step: 1,
+                langgraph_task_idx: 0,
+                langgraph_triggers: ["__pregel_tasks"],
+            },
         },
         {
             event: "on_chat_model_start",
@@ -186,6 +234,10 @@ it("should pass config through if importing from the primary entrypoint", async
             tags: [],
             run_id: expect.any(String),
             metadata: {
+                langgraph_node: "testnode",
+                langgraph_step: 1,
+                langgraph_task_idx: 0,
+                langgraph_triggers: ["__pregel_tasks"],
                 ls_model_type: "chat",
                 ls_stop: undefined,
             },
@@ -202,6 +254,10 @@ it("should pass config through if importing from the primary entrypoint", async
             name: "conditional_edge_call",
             tags: [],
             metadata: {
+                langgraph_node: "testnode",
+                langgraph_step: 1,
+                langgraph_task_idx: 0,
+                langgraph_triggers: ["__pregel_tasks"],
                 ls_model_type: "chat",
                 ls_stop: undefined,
             },
@@ -219,7 +275,12 @@ it("should pass config through if importing from the primary entrypoint", async
             run_id: expect.any(String),
             name: "func",
             tags: ["seq:step:3"],
-            metadata: {},
+            metadata: {
+                langgraph_node: "testnode",
+                langgraph_step: 1,
+                langgraph_task_idx: 0,
+                langgraph_triggers: ["__pregel_tasks"],
+            },
         },
         {
             event: "on_chain_end",
@@ -232,7 +293,12 @@ it("should pass config through if importing from the primary entrypoint", async
             run_id: expect.any(String),
             name: "testnode",
             tags: ["graph:step:1"],
-            metadata: {},
+            metadata: {
+                langgraph_node: "testnode",
+                langgraph_step: 1,
+                langgraph_task_idx: 0,
+                langgraph_triggers: ["__pregel_tasks"],
+            },
         },
         {
             event: "on_chain_stream",

package/dist/tests/utils.d.ts

@@ -4,6 +4,7 @@ import { BaseMessage } from "@langchain/core/messages";
 import { ChatResult } from "@langchain/core/outputs";
 import { RunnableConfig } from "@langchain/core/runnables";
 import { Tool } from "@langchain/core/tools";
+import { z } from "zod";
 import { MemorySaver } from "../checkpoint/memory.js";
 import { Checkpoint, CheckpointMetadata } from "../checkpoint/base.js";
 export interface FakeChatModelArgs extends BaseChatModelParams {
@@ -35,3 +36,18 @@ export declare class MemorySaverAssertImmutable extends MemorySaver {
     constructor();
     put(config: RunnableConfig, checkpoint: Checkpoint, metadata: CheckpointMetadata): Promise<RunnableConfig>;
 }
+export declare class FakeSearchTool extends Tool {
+    name: string;
+    description: string;
+    schema: z.ZodEffects<z.ZodObject<{
+        input: z.ZodOptional<z.ZodString>;
+    }, "strip", z.ZodTypeAny, {
+        input?: string | undefined;
+    }, {
+        input?: string | undefined;
+    }>, string | undefined, {
+        input?: string | undefined;
+    }>;
+    constructor();
+    _call(query: string): Promise<string>;
+}

package/dist/tests/utils.js

@@ -1,6 +1,8 @@
 import assert from "node:assert";
 import { BaseChatModel, } from "@langchain/core/language_models/chat_models";
 import { AIMessage } from "@langchain/core/messages";
+import { Tool } from "@langchain/core/tools";
+import { z } from "zod";
 import { MemorySaver } from "../checkpoint/memory.js";
 export class FakeChatModel extends BaseChatModel {
     constructor(fields) {
@@ -133,3 +135,33 @@ export class MemorySaverAssertImmutable extends MemorySaver {
         return super.put(config, checkpoint, metadata);
     }
 }
+export class FakeSearchTool extends Tool {
+    constructor() {
+        super();
+        Object.defineProperty(this, "name", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "search_api"
+        });
+        Object.defineProperty(this, "description", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "A simple API that returns the input string."
+        });
+        Object.defineProperty(this, "schema", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: z
+                .object({
+                input: z.string().optional(),
+            })
+                .transform((data) => data.input)
+        });
+    }
+    async _call(query) {
+        return `result for ${query}`;
+    }
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langchain/langgraph",
-  "version": "0.0.26",
+  "version": "0.0.27",
   "description": "LangGraph",
   "type": "module",
   "engines": {
@@ -37,12 +37,13 @@
   "author": "LangChain",
   "license": "MIT",
   "dependencies": {
-    "@langchain/core": ">0.1.61 <0.3.0",
-    "uuid": "^10.0.0"
+    "@langchain/core": ">=0.2.16 <0.3.0",
+    "uuid": "^10.0.0",
+    "zod": "^3.23.8"
   },
   "devDependencies": {
     "@jest/globals": "^29.5.0",
-    "@langchain/anthropic": "^0.1.21",
+    "@langchain/anthropic": "^0.2.3",
     "@langchain/community": "^0.2.12",
     "@langchain/openai": "^0.1.3",
     "@langchain/scripts": "^0.0.13",
@@ -73,7 +74,6 @@
     "ts-jest": "^29.1.0",
     "tsx": "^4.7.0",
     "typescript": "^4.9.5 || ^5.4.5",
-    "zod": "^3.22.4",
     "zod-to-json-schema": "^3.22.4"
   },
   "peerDependencies": {