sune-org/ORP
1
0
Fork 0
mirror of https://github.com/sune-org/ORP.git synced 2026-01-13 16:17:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

Refactor: Make stream termination more robust

Browse Source
This commit is contained in:
multipleof4
2025-09-27 03:58:44 -07:00
parent 9b6d2a4c48
commit 02eaf94c49
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
index.js
View File

@@ -172,15 +172,15 @@ export class MyDurableObject {
async stream({ apiKey, body, provider }) {
try {
if (provider === 'openai') await this.streamOpenAI({ apiKey, body });
else if (provider === 'google') await this.streamGoogle({ apiKey, body });
else await this.streamOpenRouter({ apiKey, body });
if (this.phase === 'running') this.stop();
const providerMap = { openai: this.streamOpenAI, google: this.streamGoogle };
await (providerMap[provider] || this.streamOpenRouter).call(this, { apiKey, body });
} catch (e) {
if (this.phase === 'running') {
const msg = String(e?.message || 'stream_failed');
if (!((e && e.name === 'AbortError') || /abort/i.test(msg))) this.fail(msg);
}
} finally {
if (this.phase === 'running') this.stop();
}
}
@@ -261,7 +261,7 @@ export class MyDurableObject {
}
fail(message) {
if (this.phase === 'error') return;
if (this.phase !== 'running') return;
this.flush(true);
this.phase = 'error';
this.error = String(message || 'stream_failed');
@@ -340,4 +340,4 @@ export class MyDurableObject {
if (contents.at(-1)?.role !== 'user') contents.pop();
return contents;
}
}
}