Skip to content

Commit 788d453

Browse files
committed
couple of fixes
1 parent 1113eec commit 788d453

File tree

2 files changed

+11
-3
lines changed

2 files changed

+11
-3
lines changed

packages/core/src/v3/runMetadata/manager.ts

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -303,7 +303,15 @@ export class StandardMetadataManager implements RunMetadataManager {
303303
}
304304

305305
public async fetchStream<T>(key: string, signal?: AbortSignal): Promise<AsyncIterableStream<T>> {
306-
throw new Error("This needs to use the new realtime streams API");
306+
if (!this.runId) {
307+
throw new Error("Run ID is not set. fetchStream() can only be used inside a task.");
308+
}
309+
310+
return await this.apiClient.fetchStream(this.runId, key, {
311+
signal,
312+
timeoutInSeconds: 60,
313+
lastEventId: undefined,
314+
});
307315
}
308316

309317
private async doStream<T>(
@@ -319,7 +327,7 @@ export class StandardMetadataManager implements RunMetadataManager {
319327
return $value;
320328
}
321329

322-
const streamInstance = await realtimeStreams.pipe(key, value, {
330+
const streamInstance = realtimeStreams.pipe(key, value, {
323331
signal,
324332
target,
325333
});

packages/core/test/standardMetadataManager.test.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ describe("StandardMetadataManager", () => {
3232

3333
const apiClient = new ApiClient(server.http.url().origin, "tr-123");
3434

35-
manager = new StandardMetadataManager(apiClient, server.http.url().origin);
35+
manager = new StandardMetadataManager(apiClient);
3636
manager.runId = runId;
3737
});
3838

0 commit comments

Comments
 (0)