@yeaft/webchat-agent 1.0.61 → 1.0.63

package/connection/heartbeat-policy.js

@@ -0,0 +1,15 @@
+const DEFAULT_AGENT_HEARTBEAT_TIMEOUT_MS = 180000;
+
+function parsePositiveInt(value, fallback) {
+  const n = Number(value);
+  return Number.isFinite(n) && n > 0 ? n : fallback;
+}
+
+export const AGENT_HEARTBEAT_TIMEOUT_MS = parsePositiveInt(
+  process.env.YEAFT_AGENT_HEARTBEAT_TIMEOUT_MS || process.env.AGENT_HEARTBEAT_TIMEOUT_MS,
+  DEFAULT_AGENT_HEARTBEAT_TIMEOUT_MS,
+);
+
+export function shouldReconnectForHeartbeat(lastPongAt, now = Date.now(), timeoutMs = AGENT_HEARTBEAT_TIMEOUT_MS) {
+  return Number.isFinite(lastPongAt) && lastPongAt > 0 && (now - lastPongAt) > timeoutMs;
+}

package/connection/heartbeat.js

@@ -1,5 +1,6 @@
 import WebSocket from 'ws';
 import ctx from '../context.js';
+import { shouldReconnectForHeartbeat } from './heartbeat-policy.js';
 
 export function startAgentHeartbeat() {
   stopAgentHeartbeat();
@@ -16,8 +17,9 @@ export function startAgentHeartbeat() {
     if (!ctx.ws || ctx.ws.readyState !== WebSocket.OPEN) return;
 
     // 检查上次 pong 是否超时
-    const sincePong = Date.now() - ctx.lastPongAt;
-    if (sincePong > 45000) {
+    const now = Date.now();
+    const sincePong = now - ctx.lastPongAt;
+    if (shouldReconnectForHeartbeat(ctx.lastPongAt, now)) {
       console.warn(`[Heartbeat] No pong for ${Math.round(sincePong / 1000)}s, reconnecting...`);
       ctx.ws.terminate();
       return;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@yeaft/webchat-agent",
-  "version": "1.0.61",
+  "version": "1.0.63",
   "description": "Remote worker agent for Yeaft Web Code Agent — connects the native Yeaft engine, CLI providers, and workbench tools",
   "main": "index.js",
   "type": "module",

package/yeaft/sessions/pre-flow.js

@@ -266,15 +266,18 @@ export function buildRelevantScopes({ sessionId, chatId, vpId, extra } = {}) {
     scopes.push(`chat/${chatId}`);
     if (vpId) scopes.push(`chat/${chatId}/vp/${vpId}`);
   } else if (sessionId) {
-    // Read both legacy `group/<id>` and new `session/<id>` scopes so recall
-    // works regardless of which path the writers used for a given session.
-    // (Writers still emit `group/<id>` for back-compat; migration may have
-    // rewritten on-disk data to `session/<id>`.)
+    // Read every persisted Session spelling that can exist on disk. Current
+    // Dream writers use `sessions/<id>`; older readers/writers used singular
+    // `session/<id>` or legacy `group/<id>`. SQLite FTS is scope-filtered, so
+    // omitting any spelling makes valid indexed memory invisible.
+    scopes.push(`sessions/${sessionId}`);
+    scopes.push(`sessions/${sessionId}/user`);
     scopes.push(`session/${sessionId}`);
     scopes.push(`session/${sessionId}/user`);
     scopes.push(`group/${sessionId}`);
     scopes.push(`group/${sessionId}/user`);
     if (vpId) {
+      scopes.push(`sessions/${sessionId}/vp/${vpId}`);
       scopes.push(`session/${sessionId}/vp/${vpId}`);
       scopes.push(`group/${sessionId}/vp/${vpId}`);
     }

package/yeaft/web-bridge.js

@@ -4821,6 +4821,12 @@ export function __testAppendTurnToSessionHistory(...args) {
  * Backwards-compat: when neither field is set, defaults to `vpId='default'`
  * which matches the pre-v0.1.754 behavior.
  */
+function resolveDreamTriggerSessionId(msg = {}) {
+  return typeof msg.sessionId === 'string' && msg.sessionId
+    ? msg.sessionId
+    : (typeof msg.groupId === 'string' && msg.groupId ? msg.groupId : null);
+}
+
 export function normalizeDreamResult(result) {
   const sessions = Array.isArray(result?.sessions) ? result.sessions : [];
   const targets = Array.isArray(result?.targets) ? result.targets : [];
@@ -4869,7 +4875,7 @@ export async function handleYeaftDreamTrigger(msg = {}) {
   // route the error event back to the right row and the per-group
   // "Run dream now" button would stay stuck on "Running…" forever
   // (review feedback from PR #757).
-  const sessionId = typeof msg.sessionId === 'string' && msg.sessionId ? msg.sessionId : null;
+  const sessionId = resolveDreamTriggerSessionId(msg);
   const vpId = !sessionId ? (msg.vpId || 'default') : null;
   const tag = sessionId ? { sessionId } : { vpId };
 
@@ -5924,6 +5930,7 @@ export const __testHooks = {
     return getVpStatusBroker().transition(status);
   },
   decorateSessionsWithRuntimeState,
+  resolveDreamTriggerSessionId,
   async loadProjectRuntime(workDir) {
     return loadProjectRuntime(workDir);
   },