Export Annotation etc from root (#299)

* Export Annotation etc from root * Fix
langchain-ai · Aug 6, 2024 · 3ea5202 · 3ea5202
1 parent d3080bd
commit 3ea5202
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 4 deletions.
diff --git a/langgraph/src/graph/index.ts b/langgraph/src/graph/index.ts
@@ -3,5 +3,8 @@ export {
   type StateGraphArgs,
   StateGraph,
   type CompiledStateGraph,
+  Annotation,
+  type StateType,
+  type UpdateType,
 } from "./state.js";
 export { MessageGraph, messagesStateReducer } from "./message.js";
diff --git a/langgraph/src/graph/state.ts b/langgraph/src/graph/state.ts
@@ -58,11 +58,11 @@ type ExtractUpdateType<C> = C extends BaseChannel
   ? ReturnType<C>["UpdateType"]
   : never;
 
-export type StateInterface<S extends StateDefinition> = {
+export type StateType<S extends StateDefinition> = {
   [key in keyof S]: ExtractValueType<S[key]>;
 };
 
-export type UpdateInterface<S extends StateDefinition> = {
+export type UpdateType<S extends StateDefinition> = {
   [key in keyof S]?: ExtractUpdateType<S[key]>;
 };
 
@@ -95,8 +95,8 @@ export interface StateGraphArgs<Channels extends object | unknown> {
 
 export class StateGraph<
   SD extends StateDefinition | unknown,
-  S = SD extends StateDefinition ? StateInterface<SD> : SD,
-  U = SD extends StateDefinition ? UpdateInterface<SD> : Partial<S>,
+  S = SD extends StateDefinition ? StateType<SD> : SD,
+  U = SD extends StateDefinition ? UpdateType<SD> : Partial<S>,
   N extends string = typeof START
 > extends Graph<N, S, U> {
   channels: Record<string, BaseChannel>;

diff --git a/langgraph/src/web.ts b/langgraph/src/web.ts
@@ -7,6 +7,9 @@ export {
   type CompiledStateGraph,
   MessageGraph,
   messagesStateReducer,
+  Annotation,
+  type StateType,
+  type UpdateType,
 } from "./graph/index.js";
 export { MemorySaver } from "./checkpoint/memory.js";
 export {