fix issues

langchain-ai · Aug 20, 2024 · 39c1310 · 39c1310
1 parent 292c09b
commit 39c1310
Show file tree

Hide file tree

Showing 5 changed files with 12 additions and 8 deletions.
diff --git a/langgraph/src/graph/index.ts b/langgraph/src/graph/index.ts
@@ -8,5 +8,6 @@ export {
 export {
   MessageGraph,
   messagesStateReducer,
-  MessagesState,
+  type MessagesState,
+  createMessagesState,
 } from "./message.js";
diff --git a/langgraph/src/graph/message.ts b/langgraph/src/graph/message.ts
@@ -77,9 +77,11 @@ export class MessageGraph extends StateGraph<
   }
 }
 
-export const MessagesState = Annotation.Root({
+export const createMessagesState = () => Annotation.Root({
   messages: Annotation<BaseMessage[]>({
     reducer: messagesStateReducer,
     default: () => [],
   }),
 });
+
+export type MessagesState = ReturnType<typeof createMessagesState>['State'];
diff --git a/langgraph/src/prebuilt/react_agent_executor.ts b/langgraph/src/prebuilt/react_agent_executor.ts
@@ -40,7 +40,7 @@ export type N = typeof START | "agent" | "tools";
 export type CreateReactAgentParams = {
   llm: BaseChatModel;
   tools:
-    | ToolNode<typeof MessagesState.State>
+    | ToolNode<MessagesState>
     | (StructuredToolInterface | RunnableToolLike)[];
   messageModifier?:
     | SystemMessage

diff --git a/langgraph/src/prebuilt/tool_node.ts b/langgraph/src/prebuilt/tool_node.ts
@@ -17,7 +17,7 @@ export type ToolNodeOptions = {
 };
 
 export class ToolNode<
-  T extends BaseMessage[] | typeof MessagesState.State
+  T extends BaseMessage[] | MessagesState
 > 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[] | typeof MessagesState.State,
+    input: BaseMessage[] | MessagesState,
     config: RunnableConfig
-  ): Promise<BaseMessage[] | typeof MessagesState.State> {
+  ): Promise<BaseMessage[] | MessagesState> {
     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[] | typeof MessagesState.State
+  state: BaseMessage[] | MessagesState
 ): "tools" | typeof END {
   const message = Array.isArray(state)
     ? state[state.length - 1]

diff --git a/langgraph/src/web.ts b/langgraph/src/web.ts
@@ -8,7 +8,8 @@ export {
   MessageGraph,
   messagesStateReducer,
   Annotation,
-  MessagesState,
+  createMessagesState,
+  type MessagesState,
   type StateType,
   type UpdateType,
   type CompiledGraph,