split up

agent/types.ts

@@ -0,0 +1,11 @@
+import { MessagesAnnotation, Annotation } from "@langchain/langgraph";
+import { uiMessageReducer } from "@langchain/langgraph-sdk/react-ui/types";
+
+export const GenerativeUIAnnotation = Annotation.Root({
+  messages: MessagesAnnotation.spec["messages"],
+  ui: Annotation({ default: () => [], reducer: uiMessageReducer }),
+  timestamp: Annotation<number>,
+  next: Annotation<"stockbroker" | "weather" | "generalInput">(),
+});
+
+export type GenerativeUIState = typeof GenerativeUIAnnotation.State;