taozuqin/ST-Bionic-Memory-Ecology
1
0
Fork 0
mirror of https://github.com/Youzini-afk/ST-Bionic-Memory-Ecology.git synced 2026-05-15 22:30:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

Reuse main timeout for Ena planner requests

Browse Source
This commit is contained in:
Hao19911125
2026-04-09 22:14:08 +08:00
parent 29524aa328
commit d475b121da
1 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
ena-planner/ena-planner.js
View File

@@ -20,6 +20,13 @@ function getPlannerRecallTimeoutMs() {
: VECTOR_RECALL_TIMEOUT_MS; : VECTOR_RECALL_TIMEOUT_MS;
} }
function getPlannerRequestTimeoutMs() {
const timeoutMs = Number(_bmeRuntime?.getPlannerRecallTimeoutMs?.());
return Number.isFinite(timeoutMs) && timeoutMs > 0
? timeoutMs
: PLANNER_REQUEST_TIMEOUT_MS;
}
function getTrustedOrigin() { return window.location.origin; } function getTrustedOrigin() { return window.location.origin; }
function postToIframe(iframe, payload) { function postToIframe(iframe, payload) {
@@ -988,7 +995,8 @@ async function callPlanner(messages, options = {}) {
if (mt != null && !Number.isNaN(mt) && mt > 0) body.max_tokens = mt; if (mt != null && !Number.isNaN(mt) && mt > 0) body.max_tokens = mt;
const controller = new AbortController(); const controller = new AbortController();
const timeoutId = setTimeout(() => controller.abort(), PLANNER_REQUEST_TIMEOUT_MS); const plannerRequestTimeoutMs = getPlannerRequestTimeoutMs();
const timeoutId = setTimeout(() => controller.abort(), plannerRequestTimeoutMs);
try { try {
const res = await fetch(url, { const res = await fetch(url, {
method: 'POST', method: 'POST',
@@ -1047,7 +1055,7 @@ async function callPlanner(messages, options = {}) {
return full; return full;
} catch (err) { } catch (err) {
if (controller.signal.aborted || err?.name === 'AbortError') { if (controller.signal.aborted || err?.name === 'AbortError') {
throw new Error(`规划请求超时（>${Math.floor(PLANNER_REQUEST_TIMEOUT_MS / 1000)}s）`); throw new Error(`规划请求超时（>${Math.floor(plannerRequestTimeoutMs / 1000)}s）`);
} }
throw err; throw err;
} finally { } finally {