fix: Allow configuring stream mode in useStream.joinStream()

This commit is contained in:
bracesproul
2025-06-19 11:14:59 -07:00
parent eba18c3213
commit 7bd364c457
+3 -2
View File
@@ -616,7 +616,7 @@ export interface UseStream<
/**
* Join an active stream.
*/
joinStream: (runId: string) => Promise<void>;
joinStream: (runId: string, lastEventId?: string, options?: { streamMode?: StreamMode | StreamMode[] }) => Promise<void>;
}
type ConfigWithConfigurable<ConfigurableType extends Record<string, unknown>> =
@@ -951,13 +951,14 @@ export function useStream<
}
}
const joinStream = async (runId: string, lastEventId?: string) => {
const joinStream = async (runId: string, lastEventId?: string, options?: { streamMode?: StreamMode | StreamMode[]; }) => {
lastEventId ??= "-1";
if (!threadId) return;
await consumeStream(async (signal: AbortSignal) => {
const stream = client.runs.joinStream(threadId, runId, {
signal,
lastEventId,
streamMode: options?.streamMode,
}) as AsyncGenerator<EventStreamEvent>;
return {