fix: router issues

This commit is contained in:
bracesproul
2025-03-03 13:51:41 -08:00
parent adcb0e09ec
commit 97c1f4ca5d
6 changed files with 3983 additions and 2086 deletions

View File

@@ -1,11 +1,11 @@
import React, { createContext, useContext, ReactNode } from "react";
import { useStream } from "@langchain/langgraph-sdk/react";
import { Client, type Message } from "@langchain/langgraph-sdk";
import { type Message } from "@langchain/langgraph-sdk";
import type {
UIMessage,
RemoveUIMessage,
} from "@langchain/langgraph-sdk/react-ui/types";
import { useQueryParam, StringParam } from 'use-query-params';
import { useQueryParam, StringParam } from "use-query-params";
const useTypedStream = useStream<
{ messages: Message[]; ui: UIMessage[] },
@@ -24,15 +24,18 @@ const StreamContext = createContext<StreamContextType | undefined>(undefined);
export const StreamProvider: React.FC<{ children: ReactNode }> = ({
children,
}) => {
const [threadId, setThreadId] = useQueryParam('threadId', StringParam);
const [threadId, setThreadId] = useQueryParam("threadId", StringParam);
const streamValue = useTypedStream({
apiUrl: "http://localhost:2024",
assistantId: "agent",
threadId: threadId || "",
threadId,
onThreadId: setThreadId,
});
console.log("threadId", threadId);
console.log("streamValue", streamValue.values);
return (
<StreamContext.Provider value={streamValue}>
{children}