docs: update docs usage of useLangGraphRuntime (#820)

Yonom · Sep 11, 2024 · 60ad0fa · 60ad0fa
1 parent e9f60ae
commit 60ad0fa
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 13 deletions.
diff --git a/apps/docs/content/docs/runtimes/langgraph.mdx b/apps/docs/content/docs/runtimes/langgraph.mdx
@@ -81,15 +81,15 @@ export const createThread = async () => {
 
 export const sendMessage = async (params: {
   threadId: string;
-  message: LangChainMessage;
+  messages: LangChainMessage;
 }) => {
   const client = createClient();
   return client.runs.stream(
     params.threadId,
     process.env["NEXT_PUBLIC_LANGGRAPH_ASSISTANT_ID"]!,
     {
       input: {
-        messages: [params.message],
+        messages: params.messages,
       },
       streamMode: "messages",
     },
@@ -122,15 +122,15 @@ export function MyRuntimeProvider({
   const threadIdRef = useRef<string | undefined>();
   const runtime = useLangGraphRuntime({
     threadId: threadIdRef.current,
-    stream: async (message) => {
+    stream: async (messages) => {
       if (!threadIdRef.current) {
         const { thread_id } = await createThread();
         threadIdRef.current = thread_id;
       }
       const threadId = threadIdRef.current;
       return sendMessage({
         threadId,
-        message,
+        messages,
       });
     },
   });

diff --git a/examples/with-langgraph/app/MyRuntimeProvider.tsx b/examples/with-langgraph/app/MyRuntimeProvider.tsx
@@ -13,15 +13,15 @@ export function MyRuntimeProvider({
   const threadIdRef = useRef<string | undefined>();
   const runtime = useLangGraphRuntime({
     threadId: threadIdRef.current,
-    stream: async (message) => {
+    stream: async (messages) => {
       if (!threadIdRef.current) {
         const { thread_id } = await createThread();
         threadIdRef.current = thread_id;
       }
       const threadId = threadIdRef.current;
       return sendMessage({
         threadId,
-        message,
+        messages,
       });
     },
   });

diff --git a/examples/with-langgraph/lib/chatApi.ts b/examples/with-langgraph/lib/chatApi.ts
@@ -41,16 +41,13 @@ export const updateState = async (
 
 export const sendMessage = async (params: {
   threadId: string;
-  message: LangChainMessage | null;
+  messages: LangChainMessage[];
 }) => {
   const client = createClient();
 
-  let input: Record<string, any> | null = null;
-  if (params.message !== null) {
-    input = {
-      messages: [params.message],
-    };
-  }
+  let input: Record<string, any> | null = {
+    messages: params.messages,
+  };
   const config = {
     configurable: {
       model_name: "openai",