Merge

langchain-ai · Aug 22, 2024 · 81ebfe8 · 81ebfe8
2 parents 9e53262 + 8c8d0a9
commit 81ebfe8
Show file tree

Hide file tree

Showing 8 changed files with 45 additions and 14 deletions.
diff --git a/libs/langgraph/langchain.config.js b/libs/langgraph/langchain.config.js
@@ -16,7 +16,7 @@ export const config = {
     web: "web",
     pregel: "pregel/index",
     prebuilt: "prebuilt/index",
-    "checkpoint/sqlite": "checkpoint/sqlite",
+    "checkpoint/sqlite": "checkpoint/sqlite"
   },
   tsConfigPath: resolve("./tsconfig.json"),
   cjsSource: "./dist-cjs",

diff --git a/libs/langgraph/package.json b/libs/langgraph/package.json
@@ -50,7 +50,7 @@
     "@langchain/anthropic": "^0.2.12",
     "@langchain/community": "^0.2.25",
     "@langchain/openai": "^0.2.4",
-    "@langchain/scripts": "^0.0.22",
+    "@langchain/scripts": "^0.0.23",
     "@swc/core": "^1.3.90",
     "@swc/jest": "^0.2.29",
     "@tsconfig/recommended": "^1.0.3",

diff --git a/libs/langgraph/src/graph/message.ts b/libs/langgraph/src/graph/message.ts
@@ -75,7 +75,3 @@ export class MessageGraph extends StateGraph<
     });
   }
 }
-
-export interface MessagesState {
-  messages: BaseMessage[];
-}
diff --git a/libs/langgraph/src/graph/messages_annotation.ts b/libs/langgraph/src/graph/messages_annotation.ts
@@ -0,0 +1,12 @@
+/* __LC_ALLOW_ENTRYPOINT_SIDE_EFFECTS__ */
+
+import { BaseMessage } from "@langchain/core/messages";
+import { Annotation } from "./annotation.js";
+import { messagesStateReducer } from "./message.js";
+
+export const MessagesAnnotation = Annotation.Root({
+  messages: Annotation<BaseMessage[]>({
+    reducer: messagesStateReducer,
+    default: () => [],
+  }),
+});
diff --git a/libs/langgraph/src/index.ts b/libs/langgraph/src/index.ts
@@ -6,3 +6,4 @@ import { initializeAsyncLocalStorageSingleton } from "./setup/async_local_storag
 initializeAsyncLocalStorageSingleton();
 
 export * from "./web.js";
+export { MessagesAnnotation } from "./graph/messages_annotation.js";
diff --git a/libs/langgraph/src/prebuilt/react_agent_executor.ts b/libs/langgraph/src/prebuilt/react_agent_executor.ts
@@ -21,7 +21,7 @@ import {
 import { ChatPromptTemplate } from "@langchain/core/prompts";
 import { BaseCheckpointSaver } from "@langchain/langgraph-checkpoint";
 import { END, START, StateGraph } from "../graph/index.js";
-import { MessagesState } from "../graph/message.js";
+import { MessagesAnnotation } from "../graph/messages_annotation.js";
 import { CompiledStateGraph, StateGraphArgs } from "../graph/state.js";
 import { All } from "../pregel/types.js";
 import { ToolNode } from "./tool_node.js";
@@ -40,7 +40,7 @@ export type N = typeof START | "agent" | "tools";
 export type CreateReactAgentParams = {
   llm: BaseChatModel;
   tools:
-    | ToolNode<MessagesState>
+    | ToolNode<typeof MessagesAnnotation.State>
     | (StructuredToolInterface | RunnableToolLike)[];
   messageModifier?:
     | SystemMessage

diff --git a/libs/langgraph/src/prebuilt/tool_node.ts b/libs/langgraph/src/prebuilt/tool_node.ts
@@ -8,7 +8,7 @@ import { RunnableConfig, RunnableToolLike } from "@langchain/core/runnables";
 import { StructuredToolInterface } from "@langchain/core/tools";
 import { RunnableCallable } from "../utils.js";
 import { END } from "../graph/graph.js";
-import { MessagesState } from "../graph/message.js";
+import { MessagesAnnotation } from "../graph/messages_annotation.js";
 
 export type ToolNodeOptions = {
   name?: string;
@@ -17,7 +17,7 @@ export type ToolNodeOptions = {
 };
 
 export class ToolNode<
-  T extends BaseMessage[] | MessagesState
+  T extends BaseMessage[] | typeof MessagesAnnotation.State
 > extends RunnableCallable<T, T> {
   /**
   A node that runs the tools requested in the last AIMessage. It can be used
@@ -41,9 +41,9 @@ export class ToolNode<
   }
 
   private async run(
-    input: BaseMessage[] | MessagesState,
+    input: BaseMessage[] | typeof MessagesAnnotation.State,
     config: RunnableConfig
-  ): Promise<BaseMessage[] | MessagesState> {
+  ): Promise<BaseMessage[] | typeof MessagesAnnotation.State> {
     const message = Array.isArray(input)
       ? input[input.length - 1]
       : input.messages[input.messages.length - 1];
@@ -92,7 +92,7 @@ export class ToolNode<
 }
 
 export function toolsCondition(
-  state: BaseMessage[] | MessagesState
+  state: BaseMessage[] | typeof MessagesAnnotation.State
 ): "tools" | typeof END {
   const message = Array.isArray(state)
     ? state[state.length - 1]

diff --git a/yarn.lock b/yarn.lock
@@ -1959,7 +1959,7 @@ __metadata:
     "@langchain/langgraph-checkpoint": 0.0.0
     "@langchain/langgraph-checkpoint-sqlite": 0.0.0
     "@langchain/openai": ^0.2.4
-    "@langchain/scripts": ^0.0.22
+    "@langchain/scripts": ^0.0.23
     "@swc/core": ^1.3.90
     "@swc/jest": ^0.2.29
     "@tsconfig/recommended": ^1.0.3
@@ -2060,6 +2060,28 @@ __metadata:
   languageName: node
   linkType: hard
 
+"@langchain/scripts@npm:^0.0.23":
+  version: 0.0.23
+  resolution: "@langchain/scripts@npm:0.0.23"
+  dependencies:
+    "@rollup/wasm-node": ^4.19.0
+    axios: ^1.6.7
+    commander: ^11.1.0
+    glob: ^10.3.10
+    lodash: ^4.17.21
+    readline: ^1.3.0
+    rimraf: ^5.0.1
+    rollup: ^4.5.2
+    ts-morph: ^21.0.1
+    typescript: ^5.4.5
+  bin:
+    lc-build: bin/build.js
+    lc_build_v2: bin/build_v2.js
+    notebook_validate: bin/validate_notebook.js
+  checksum: 975ac6fea3832353bc15b48ae5e66087dcda9e3f8be02c3f2f5c93b2545cacb803b4d160a04b33ab14da75808909de6077d775fb51d2960d9a769fc968342e54
+  languageName: node
+  linkType: hard
+
 "@langchain/textsplitters@npm:~0.0.0":
   version: 0.0.2
   resolution: "@langchain/textsplitters@npm:0.0.2"