@yeaft/webchat-agent 1.0.494 → 1.0.495

package/local-runtime/server/handlers/agent-file-terminal.js

@@ -185,7 +185,8 @@ async function handleOneShotResponse(agentId, agent, msg, routeKey) {
     userId: pending.userId,
     role: pending.role,
   });
-  if (!currentGeneration || currentGeneration !== pending.workspaceGeneration) return;
+  if (currentGeneration === null) return;
+  if (currentGeneration && currentGeneration !== pending.workspaceGeneration) return;
   const projected = msg.type === 'file_content' && msg.binary
     ? cacheBinaryPreview(msg)
     : msg;

package/local-runtime/server/handlers/client-workbench.js

@@ -37,8 +37,20 @@ function isYeaftVirtualConversation(conversationId) {
 }
 
 async function denyWorkbenchRoute(client, msg) {
+  const error = 'Invalid Workbench Session route';
   console.warn(`[Security] Invalid Workbench route for ${msg?.type || 'unknown'}`);
-  await sendToWebClient(client, { type: 'error', message: 'Invalid Workbench Session route' });
+  const response = workbenchFailureResponse({
+    agentId: msg?.agentId || client.currentAgent,
+    msg,
+    resolved: {
+      conversationId: msg?.conversationId,
+      routeKey: msg?.workbenchRouteKey,
+      workspaceGeneration: msg?.workbenchWorkspaceGeneration,
+    },
+    error,
+  });
+  if (response) await sendToWebClient(client, response);
+  await sendToWebClient(client, { type: 'error', message: error });
 }
 
 const AGENT_DIRECTORY_PICKER_CONVERSATION = '_workdir_picker';
@@ -93,10 +105,9 @@ const FILE_OPERATIONS = Object.freeze({
   upload_to_dir: 'upload',
 });
 
-function workbenchTimeoutResponse({ agentId, msg, resolved }) {
+function workbenchFailureResponse({ agentId, msg, resolved, error }) {
   const type = TIMEOUT_RESPONSE_TYPES[msg.type];
   if (!type) return null;
-  const error = 'Workbench request timed out';
   const response = {
     type,
     agentId,
@@ -189,7 +200,12 @@ function correlateWorkbenchRequest({ agentId, clientId, client, msg, resolved, c
     allowLegacyCorrelation: !supportsRequestCorrelation,
     onTimeout: async () => {
       if (!client.authenticated || client.userId !== registration.userId) return;
-      const response = workbenchTimeoutResponse({ agentId, msg, resolved });
+      const response = workbenchFailureResponse({
+        agentId,
+        msg,
+        resolved,
+        error: 'Workbench request timed out',
+      });
       if (response) await sendToWebClient(client, response);
     },
   };

package/local-runtime/server/workbench-route.js

@@ -120,14 +120,19 @@ function resolveChatRow(client, route) {
  *
  * `legacy: true` preserves old clients that predate route-scoped Workbench.
  */
+/**
+ * Return null when the route is no longer valid, an empty string when the
+ * Session is valid but has no Server-owned cwd, or its canonical generation.
+ */
 export function currentWorkbenchWorkspaceGeneration({ route, userId, role }) {
-  if (!route || !userId) return '';
+  if (!route || !userId) return null;
   const client = { userId, role };
   const row = route.runtimeProvider === 'yeaft'
     ? resolveYeaftRow(client, route)
     : resolveChatRow(client, route);
-  if (!row || row.isArchived) return '';
+  if (!row || row.isArchived) return null;
   const workDir = clean(route.runtimeProvider === 'yeaft' ? row.workDir : row.work_dir, 4096);
+  if (!workDir) return '';
   return workbenchWorkspaceGeneration(workbenchRouteKey(route), workDir);
 }
 

package/local-runtime/version.json

@@ -1 +1 @@
-{"version":"1.0.494"}
+{"version":"1.0.495"}

package/package.json

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