WEBDIENSTE/langgraph
1
0
Fork 0
mirror of https://github.com/langchain-ai/langgraph.git synced 2026-09-10 19:57:52 +02:00
Code Issues Packages Projects Releases Wiki Activity

Make sure we clear when switching threads

Browse Source
This commit is contained in:
Tat Dat Duong
2025-02-11 09:01:53 -08:00
parent 2c5ddceeb1
commit f3fe30380c
1 changed files with 14 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
libs/sdk-js/src/react/stream.tsx
+14 -1
View File
@@ -145,6 +145,7 @@ function fetchHistory<StateType extends Record<string, unknown>>(
function useThreadHistory<StateType extends Record<string, unknown>>(
threadId: string | undefined | null,
client: Client,
clearCallbackRef: MutableRefObject<(() => void) | undefined>,
submittingRef: MutableRefObject<boolean>,
) {
const [history, setHistory] = useState<ThreadState<StateType>[]>([]);
@@ -161,6 +162,7 @@ function useThreadHistory<StateType extends Record<string, unknown>>(
}
setHistory([]);
clearCallbackRef.current?.();
return Promise.resolve([]);
},
[],
@@ -238,9 +240,20 @@ export function useStream<
[],
);
const clearCallbackRef = useRef<() => void>(null!);
clearCallbackRef.current = () => {
setStreamError(undefined);
setStreamValues(null);
};
// TODO: this should be done on the server to avoid pagination
// TODO: should we permit adapter? SWR / React Query?
const history = useThreadHistory<StateType>(threadId, client, submittingRef);
const history = useThreadHistory<StateType>(
threadId,
client,
clearCallbackRef,
submittingRef,
);
const getMessages = useMemo(() => {
if (withMessages == null) return undefined;