@langchain/langgraph 0.0.1 → 0.0.3-rc.0

package/README.md

@@ -198,10 +198,26 @@ const _getAction = (state: { messages: Array<BaseMessage> }): AgentAction => {
   };
 };
 
-const callTool = async (state: { messages: Array<BaseMessage> }) => {
+// Define the function that calls the model
+const callModel = async (
+  state: { messages: Array<BaseMessage> },
+  config?: RunnableConfig
+) => {
+  const { messages } = state;
+  const response = await newModel.invoke(messages, config);
+  // We return a list, because this will get added to the existing list
+  return {
+    messages: [response],
+  };
+};
+
+const callTool = async (
+  state: { messages: Array<BaseMessage> },
+  config?: RunnableConfig
+) => {
   const action = _getAction(state);
   // We call the tool_executor and get back a response
-  const response = await toolExecutor.invoke(action);
+  const response = await toolExecutor.invoke(action, config);
   // We use the response to create a FunctionMessage
   const functionMessage = new FunctionMessage({
     content: response,
@@ -315,7 +331,6 @@ Langchain Expression Language allows you to easily define chains (DAGs) but does
 
 ## Examples
 
-
 ### ChatAgentExecutor: with function calling
 
 This agent executor takes a list of messages as input and outputs a list of messages. 
@@ -331,6 +346,12 @@ This agent executor uses existing LangChain agents.
 
 - [Getting Started Notebook](https://github.com/langchain-ai/langgraphjs/blob/main/examples/agent_executor/base.ipynb): Walks through creating this type of executor from scratch
 
+### Multi-agent Examples
+
+- [Multi-agent collaboration](https://github.com/langchain-ai/langgraphjs/blob/main/examples/multi_agent/multi-agent-collaboration.ipynb): how to create two agents that work together to accomplish a task
+- [Multi-agent with supervisor](https://github.com/langchain-ai/langgraphjs/blob/main/examples/multi_agent/agent_supervisor.ipynb): how to orchestrate individual agents by using an LLM as a "supervisor" to distribute work
+- [Hierarchical agent teams](https://github.com/langchain-ai/langgraphjs/blob/main/examples/multi_agent/hierarchical_agent_teams.ipynb): how to orchestrate "teams" of agents as nested graphs that can collaborate to solve a problem
+
 ## Documentation
 
 There are only a few new APIs to use.

package/dist/checkpoint/base.cjs

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.BaseCheckpointSaver = exports.CheckpointAt = exports.emptyCheckpoint = void 0;
+exports.BaseCheckpointSaver = exports.emptyCheckpoint = void 0;
 function emptyCheckpoint() {
     return {
         v: 1,
@@ -11,18 +11,13 @@ function emptyCheckpoint() {
     };
 }
 exports.emptyCheckpoint = emptyCheckpoint;
-var CheckpointAt;
-(function (CheckpointAt) {
-    CheckpointAt["END_OF_STEP"] = "end_of_step";
-    CheckpointAt["END_OF_RUN"] = "end_of_run";
-})(CheckpointAt || (exports.CheckpointAt = CheckpointAt = {}));
 class BaseCheckpointSaver {
     constructor() {
         Object.defineProperty(this, "at", {
             enumerable: true,
             configurable: true,
             writable: true,
-            value: CheckpointAt.END_OF_RUN
+            value: "end_of_run" /* CheckpointAt.END_OF_RUN */
         });
     }
     get configSpecs() {

package/dist/checkpoint/base.d.ts

@@ -35,7 +35,7 @@ export interface Checkpoint {
     versionsSeen: Record<string, Record<string, number>>;
 }
 export declare function emptyCheckpoint(): Checkpoint;
-export declare enum CheckpointAt {
+export declare const enum CheckpointAt {
     END_OF_STEP = "end_of_step",
     END_OF_RUN = "end_of_run"
 }

package/dist/checkpoint/base.js

@@ -7,18 +7,13 @@ export function emptyCheckpoint() {
         versionsSeen: {},
     };
 }
-export var CheckpointAt;
-(function (CheckpointAt) {
-    CheckpointAt["END_OF_STEP"] = "end_of_step";
-    CheckpointAt["END_OF_RUN"] = "end_of_run";
-})(CheckpointAt || (CheckpointAt = {}));
 export class BaseCheckpointSaver {
     constructor() {
         Object.defineProperty(this, "at", {
             enumerable: true,
             configurable: true,
             writable: true,
-            value: CheckpointAt.END_OF_RUN
+            value: "end_of_run" /* CheckpointAt.END_OF_RUN */
         });
     }
     get configSpecs() {

package/dist/graph/state.d.ts

@@ -4,14 +4,16 @@ import { Graph } from "./graph.js";
 import { BaseCheckpointSaver } from "../checkpoint/base.js";
 import { Pregel } from "../pregel/index.js";
 export declare const START = "__start__";
-export interface StateGraphArgs<T = any> {
-    channels: Record<string, {
-        value: BinaryOperator<T> | null;
-        default?: () => T;
-    }>;
+export interface StateGraphArgs<Channels extends Record<string, any>> {
+    channels: {
+        [K in keyof Channels]: {
+            value: BinaryOperator<Channels[K]> | null;
+            default?: () => Channels[K];
+        };
+    };
 }
-export declare class StateGraph<T> extends Graph {
+export declare class StateGraph<Channels extends Record<string, any>> extends Graph<Channels> {
     channels: Record<string, BaseChannel>;
-    constructor(fields: StateGraphArgs<T>);
+    constructor(fields: StateGraphArgs<Channels>);
     compile(checkpointer?: BaseCheckpointSaver): Pregel;
 }

package/dist/prebuilt/agent_executor.cjs

@@ -43,18 +43,18 @@ function createAgentExecutor({ agentRunnable, tools, inputSchema, }) {
         }
         return "continue";
     };
-    const runAgent = async (data) => {
-        const agentOutcome = await agentRunnable.invoke(data);
+    const runAgent = async (data, config) => {
+        const agentOutcome = await agentRunnable.invoke(data, config);
         return {
             agentOutcome,
         };
     };
-    const executeTools = async (data) => {
+    const executeTools = async (data, config) => {
         const agentAction = data.agentOutcome;
         if (!agentAction || "returnValues" in agentAction) {
             throw new Error("Agent has not been run yet");
         }
-        const output = await toolExecutor.invoke(agentAction);
+        const output = await toolExecutor.invoke(agentAction, config);
         return {
             steps: [[agentAction, output]],
         };

package/dist/prebuilt/agent_executor.d.ts

@@ -1,10 +1,21 @@
+import { AgentAction, AgentFinish } from "@langchain/core/agents";
+import { BaseMessage } from "@langchain/core/messages";
 import { Runnable } from "@langchain/core/runnables";
 import { Tool } from "@langchain/core/tools";
 import { ToolExecutor } from "./tool_executor.js";
 import { StateGraphArgs } from "../graph/state.js";
 import { Pregel } from "../pregel/index.js";
-type AgentChannels<T> = StateGraphArgs<Array<any> | T>["channels"];
-export declare function createAgentExecutor<T extends Array<any> = Array<any>>({ agentRunnable, tools, inputSchema, }: {
+type Step = [AgentAction | AgentFinish, string];
+interface AgentStateBase {
+    agentOutcome?: AgentAction | AgentFinish;
+    steps: Array<Step>;
+}
+export interface AgentExecutorState extends AgentStateBase {
+    input: string;
+    chatHistory?: BaseMessage[];
+}
+type AgentChannels<T extends AgentExecutorState> = StateGraphArgs<AgentExecutorState | T>["channels"];
+export declare function createAgentExecutor<T extends AgentExecutorState>({ agentRunnable, tools, inputSchema, }: {
     agentRunnable: Runnable;
     tools: Array<Tool> | ToolExecutor;
     inputSchema?: AgentChannels<T>;

package/dist/prebuilt/agent_executor.js

@@ -1,4 +1,4 @@
-import { RunnableLambda } from "@langchain/core/runnables";
+import { RunnableLambda, } from "@langchain/core/runnables";
 import { ToolExecutor } from "./tool_executor.js";
 import { StateGraph } from "../graph/state.js";
 import { END } from "../index.js";
@@ -40,18 +40,18 @@ export function createAgentExecutor({ agentRunnable, tools, inputSchema, }) {
         }
         return "continue";
     };
-    const runAgent = async (data) => {
-        const agentOutcome = await agentRunnable.invoke(data);
+    const runAgent = async (data, config) => {
+        const agentOutcome = await agentRunnable.invoke(data, config);
         return {
             agentOutcome,
         };
     };
-    const executeTools = async (data) => {
+    const executeTools = async (data, config) => {
         const agentAction = data.agentOutcome;
         if (!agentAction || "returnValues" in agentAction) {
             throw new Error("Agent has not been run yet");
         }
-        const output = await toolExecutor.invoke(agentAction);
+        const output = await toolExecutor.invoke(agentAction, config);
         return {
             steps: [[agentAction, output]],
         };

package/dist/prebuilt/chat_agent_executor.cjs

@@ -41,9 +41,9 @@ function createFunctionCallingExecutor({ model, tools, }) {
         return "continue";
     };
     // Define the function that calls the model
-    const callModel = async (state) => {
+    const callModel = async (state, config) => {
         const { messages } = state;
-        const response = await newModel.invoke(messages);
+        const response = await newModel.invoke(messages, config);
         // We return a list, because this will get added to the existing list
         return {
             messages: [response],
@@ -68,10 +68,10 @@ function createFunctionCallingExecutor({ model, tools, }) {
             log: "",
         };
     };
-    const callTool = async (state) => {
+    const callTool = async (state, config) => {
         const action = _getAction(state);
         // We call the tool_executor and get back a response
-        const response = await toolExecutor.invoke(action);
+        const response = await toolExecutor.invoke(action, config);
         // We use the response to create a FunctionMessage
         const functionMessage = new messages_1.FunctionMessage({
             content: response,

package/dist/prebuilt/chat_agent_executor.d.ts

@@ -1,5 +1,9 @@
 import { Tool } from "@langchain/core/tools";
+import { BaseMessage } from "@langchain/core/messages";
 import { ToolExecutor } from "./tool_executor.js";
+export type FunctionCallingExecutorState = {
+    messages: Array<BaseMessage>;
+};
 export declare function createFunctionCallingExecutor<Model extends object>({ model, tools, }: {
     model: Model;
     tools: Array<Tool> | ToolExecutor;

package/dist/prebuilt/chat_agent_executor.js

@@ -38,9 +38,9 @@ export function createFunctionCallingExecutor({ model, tools, }) {
         return "continue";
     };
     // Define the function that calls the model
-    const callModel = async (state) => {
+    const callModel = async (state, config) => {
         const { messages } = state;
-        const response = await newModel.invoke(messages);
+        const response = await newModel.invoke(messages, config);
         // We return a list, because this will get added to the existing list
         return {
             messages: [response],
@@ -65,10 +65,10 @@ export function createFunctionCallingExecutor({ model, tools, }) {
             log: "",
         };
     };
-    const callTool = async (state) => {
+    const callTool = async (state, config) => {
         const action = _getAction(state);
         // We call the tool_executor and get back a response
-        const response = await toolExecutor.invoke(action);
+        const response = await toolExecutor.invoke(action, config);
         // We use the response to create a FunctionMessage
         const functionMessage = new FunctionMessage({
             content: response,

package/dist/prebuilt/index.d.ts

@@ -1,3 +1,3 @@
-export { createAgentExecutor } from "./agent_executor.js";
-export { createFunctionCallingExecutor } from "./chat_agent_executor.js";
+export { type AgentExecutorState, createAgentExecutor, } from "./agent_executor.js";
+export { type FunctionCallingExecutorState, createFunctionCallingExecutor, } from "./chat_agent_executor.js";
 export { type ToolExecutorArgs, type ToolInvocationInterface, ToolExecutor, } from "./tool_executor.js";

package/dist/prebuilt/index.js

@@ -1,3 +1,3 @@
-export { createAgentExecutor } from "./agent_executor.js";
-export { createFunctionCallingExecutor } from "./chat_agent_executor.js";
+export { createAgentExecutor, } from "./agent_executor.js";
+export { createFunctionCallingExecutor, } from "./chat_agent_executor.js";
 export { ToolExecutor, } from "./tool_executor.js";

package/dist/prebuilt/tool_executor.cjs

@@ -47,7 +47,7 @@ class ToolExecutor extends runnables_1.RunnableBinding {
             return acc;
         }, {});
     }
-    async _execute(toolInvocation, _config) {
+    async _execute(toolInvocation, config) {
         if (!(toolInvocation.tool in this.toolMap)) {
             return this.invalidToolMsgTemplate
                 .replace("{requestedToolName}", toolInvocation.tool)
@@ -55,7 +55,7 @@ class ToolExecutor extends runnables_1.RunnableBinding {
         }
         else {
             const tool = this.toolMap[toolInvocation.tool];
-            const output = await tool.invoke(toolInvocation.toolInput);
+            const output = await tool.invoke(toolInvocation.toolInput, config);
             return output;
         }
     }

package/dist/prebuilt/tool_executor.d.ts

@@ -22,6 +22,6 @@ export declare class ToolExecutor extends RunnableBinding<ToolExecutorInputType,
     toolMap: Record<string, Tool>;
     invalidToolMsgTemplate: string;
     constructor(fields: ToolExecutorArgs);
-    _execute(toolInvocation: ToolInvocationInterface, _config?: RunnableConfig): Promise<string>;
+    _execute(toolInvocation: ToolInvocationInterface, config?: RunnableConfig): Promise<string>;
 }
 export {};

package/dist/prebuilt/tool_executor.js

@@ -44,7 +44,7 @@ export class ToolExecutor extends RunnableBinding {
             return acc;
         }, {});
     }
-    async _execute(toolInvocation, _config) {
+    async _execute(toolInvocation, config) {
         if (!(toolInvocation.tool in this.toolMap)) {
             return this.invalidToolMsgTemplate
                 .replace("{requestedToolName}", toolInvocation.tool)
@@ -52,7 +52,7 @@ export class ToolExecutor extends RunnableBinding {
         }
         else {
             const tool = this.toolMap[toolInvocation.tool];
-            const output = await tool.invoke(toolInvocation.toolInput);
+            const output = await tool.invoke(toolInvocation.toolInput, config);
             return output;
         }
     }

package/dist/pregel/index.cjs

@@ -256,7 +256,7 @@ class Pregel extends runnables_1.Runnable {
             }
             // save end of step checkpoint
             if (this.checkpointer &&
-                this.checkpointer.at === base_js_2.CheckpointAt.END_OF_STEP) {
+                this.checkpointer.at === "end_of_step" /* CheckpointAt.END_OF_STEP */) {
                 checkpoint = await (0, base_js_1.createCheckpoint)(checkpoint, channels);
                 this.checkpointer.put(config, checkpoint);
             }
@@ -266,7 +266,7 @@ class Pregel extends runnables_1.Runnable {
             }
         }
         // save end of run checkpoint
-        if (this.checkpointer && this.checkpointer.at === base_js_2.CheckpointAt.END_OF_RUN) {
+        if (this.checkpointer && this.checkpointer.at === "end_of_run" /* CheckpointAt.END_OF_RUN */) {
             checkpoint = await (0, base_js_1.createCheckpoint)(checkpoint, channels);
             this.checkpointer.put(config, checkpoint);
         }
@@ -342,7 +342,7 @@ pendingWrites, config, forStep) {
     const pendingWritesByChannel = {};
     // Group writes by channel
     for (const [chan, val] of pendingWrites) {
-        for (const c in reserved_js_1.ReservedChannels) {
+        for (const c in reserved_js_1.ReservedChannelsMap) {
             if (chan === c) {
                 throw new Error(`Can't write to reserved channel ${chan}`);
             }
@@ -355,7 +355,7 @@ pendingWrites, config, forStep) {
         }
     }
     // Update reserved channels
-    pendingWritesByChannel[reserved_js_1.ReservedChannels.isLastStep] = [
+    pendingWritesByChannel[reserved_js_1.ReservedChannelsMap.isLastStep] = [
         forStep + 1 === config.recursionLimit,
     ];
     const updatedChannels = new Set();

package/dist/pregel/index.js

@@ -2,10 +2,10 @@
 import { Runnable, _coerceToRunnable, patchConfig, } from "@langchain/core/runnables";
 import { IterableReadableStream } from "@langchain/core/utils/stream";
 import { EmptyChannelError, createCheckpoint, emptyChannels, } from "../channels/base.js";
-import { CheckpointAt, emptyCheckpoint, } from "../checkpoint/base.js";
+import { emptyCheckpoint, } from "../checkpoint/base.js";
 import { ChannelBatch, ChannelInvoke } from "./read.js";
 import { validateGraph } from "./validate.js";
-import { ReservedChannels } from "./reserved.js";
+import { ReservedChannelsMap } from "./reserved.js";
 import { mapInput, mapOutput } from "./io.js";
 import { ChannelWrite } from "./write.js";
 import { CONFIG_KEY_READ, CONFIG_KEY_SEND } from "../constants.js";
@@ -251,7 +251,7 @@ export class Pregel extends Runnable {
             }
             // save end of step checkpoint
             if (this.checkpointer &&
-                this.checkpointer.at === CheckpointAt.END_OF_STEP) {
+                this.checkpointer.at === "end_of_step" /* CheckpointAt.END_OF_STEP */) {
                 checkpoint = await createCheckpoint(checkpoint, channels);
                 this.checkpointer.put(config, checkpoint);
             }
@@ -261,7 +261,7 @@ export class Pregel extends Runnable {
             }
         }
         // save end of run checkpoint
-        if (this.checkpointer && this.checkpointer.at === CheckpointAt.END_OF_RUN) {
+        if (this.checkpointer && this.checkpointer.at === "end_of_run" /* CheckpointAt.END_OF_RUN */) {
             checkpoint = await createCheckpoint(checkpoint, channels);
             this.checkpointer.put(config, checkpoint);
         }
@@ -336,7 +336,7 @@ pendingWrites, config, forStep) {
     const pendingWritesByChannel = {};
     // Group writes by channel
     for (const [chan, val] of pendingWrites) {
-        for (const c in ReservedChannels) {
+        for (const c in ReservedChannelsMap) {
             if (chan === c) {
                 throw new Error(`Can't write to reserved channel ${chan}`);
             }
@@ -349,7 +349,7 @@ pendingWrites, config, forStep) {
         }
     }
     // Update reserved channels
-    pendingWritesByChannel[ReservedChannels.isLastStep] = [
+    pendingWritesByChannel[ReservedChannelsMap.isLastStep] = [
         forStep + 1 === config.recursionLimit,
     ];
     const updatedChannels = new Set();

package/dist/pregel/read.cjs

@@ -58,7 +58,7 @@ class ChannelRead extends runnables_1.RunnableLambda {
     }
 }
 exports.ChannelRead = ChannelRead;
-const defaultRunnableBound = new runnables_1.RunnablePassthrough();
+const defaultRunnableBound = /* #__PURE__ */ new runnables_1.RunnablePassthrough();
 class ChannelInvoke extends runnables_1.RunnableBinding {
     constructor(fields) {
         const { channels, triggers, when } = fields;

package/dist/pregel/read.js

@@ -54,7 +54,7 @@ export class ChannelRead extends RunnableLambda {
         return read(this.channel);
     }
 }
-const defaultRunnableBound = new RunnablePassthrough();
+const defaultRunnableBound = /* #__PURE__ */ new RunnablePassthrough();
 export class ChannelInvoke extends RunnableBinding {
     constructor(fields) {
         const { channels, triggers, when } = fields;

package/dist/pregel/reserved.cjs

@@ -1,7 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.ReservedChannels = void 0;
-var ReservedChannels;
-(function (ReservedChannels) {
-    ReservedChannels["isLastStep"] = "isLastStep";
-})(ReservedChannels || (exports.ReservedChannels = ReservedChannels = {}));
+exports.ReservedChannelsMap = void 0;
+exports.ReservedChannelsMap = {
+    isLastStep: "isLastStep",
+};

package/dist/pregel/reserved.d.ts

@@ -1,3 +1,3 @@
-export declare enum ReservedChannels {
-    isLastStep = "isLastStep"
-}
+export declare const ReservedChannelsMap: {
+    isLastStep: string;
+};

package/dist/pregel/reserved.js

@@ -1,4 +1,3 @@
-export var ReservedChannels;
-(function (ReservedChannels) {
-    ReservedChannels["isLastStep"] = "isLastStep";
-})(ReservedChannels || (ReservedChannels = {}));
+export const ReservedChannelsMap = {
+    isLastStep: "isLastStep",
+};

package/dist/pregel/validate.cjs

@@ -63,7 +63,7 @@ function validateGraph({ nodes, channels, input, output, hidden, interrupt, }) {
             }
         }
     }
-    for (const chan in reserved_js_1.ReservedChannels) {
+    for (const chan in reserved_js_1.ReservedChannelsMap) {
         if (!(chan in newChannels)) {
             // eslint-disable-next-line @typescript-eslint/no-explicit-any
             newChannels[chan] = new last_value_js_1.LastValue();

package/dist/pregel/validate.js

@@ -1,5 +1,5 @@
 import { LastValue } from "../channels/last_value.js";
-import { ReservedChannels } from "./reserved.js";
+import { ReservedChannelsMap } from "./reserved.js";
 export function validateGraph({ nodes, channels, input, output, hidden, interrupt, }) {
     const newChannels = channels;
     const subscribedChannels = new Set();
@@ -60,7 +60,7 @@ export function validateGraph({ nodes, channels, input, output, hidden, interrup
             }
         }
     }
-    for (const chan in ReservedChannels) {
+    for (const chan in ReservedChannelsMap) {
         if (!(chan in newChannels)) {
             // eslint-disable-next-line @typescript-eslint/no-explicit-any
             newChannels[chan] = new LastValue();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langchain/langgraph",
-  "version": "0.0.1",
+  "version": "0.0.3-rc.0",
   "description": "LangGraph",
   "type": "module",
   "engines": {
@@ -32,14 +32,11 @@
   "author": "LangChain",
   "license": "MIT",
   "dependencies": {
-    "@langchain/community": "^0.0.17",
-    "@langchain/core": "^0.1.15",
-    "@langchain/openai": "^0.0.12",
-    "langchain": "^0.1.3",
-    "zod": "^3.22.4"
+    "@langchain/core": "^0.1.22"
   },
   "devDependencies": {
     "@jest/globals": "^29.5.0",
+    "@langchain/community": "^0.0.17",
     "@langchain/openai": "^0.0.12",
     "@swc/core": "^1.3.90",
     "@swc/jest": "^0.2.29",
@@ -56,12 +53,14 @@
     "eslint-plugin-prettier": "^4.2.1",
     "jest": "^29.5.0",
     "jest-environment-node": "^29.6.4",
+    "langchain": "^0.1.3",
     "prettier": "^2.8.3",
     "release-it": "^15.10.1",
     "rollup": "^4.5.2",
     "ts-jest": "^29.1.0",
     "tsx": "^4.7.0",
-    "typescript": "<5.2.0"
+    "typescript": "<5.2.0",
+    "zod": "^3.22.4"
   },
   "publishConfig": {
     "access": "public",