@yeaft/webchat-agent 1.0.294 → 1.0.296

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@yeaft/webchat-agent",
-  "version": "1.0.294",
+  "version": "1.0.296",
   "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/engine.js

@@ -1838,7 +1838,7 @@ export class Engine {
    *   string-prompt shape (no regression for existing callers).
    * @yields {EngineEvent}
    */
-  async *query({ prompt, promptParts = null, messages = [], signal, userEffort = null, scenario = 'chat', vpPersona, router, senderVpId, inboundEnvelope, taskId, taskMembers, sessionId, sessionMembers, sessionTopics = null, vpPlan, sessionAnnouncement, workCenterInstructions, workDir, userAlreadyPersisted = false, getCurrentTodos = null, setCurrentTodos = null, askUser = null, threadId = MAIN_THREAD_ID, vpTurnId = null, drainPendingUserMessages = null, closePendingUserInput = null, collabToolPolicy = null } = {}) {
+  async *query({ prompt, promptParts = null, messages = [], signal, userEffort = null, scenario = 'chat', vpPersona, router, senderVpId, inboundEnvelope, taskId, taskMembers, sessionId, sessionMembers, sessionTopics = null, vpPlan, sessionAnnouncement, workCenterInstructions, workDir, userAlreadyPersisted = false, getCurrentTodos = null, setCurrentTodos = null, askUser = null, threadId = MAIN_THREAD_ID, vpTurnId = null, drainPendingUserMessages = null, prepareProviderRequest = null, startProviderRequest = null, finishProviderRequest = null, failProviderRequest = null, closePendingUserInput = null, collabToolPolicy = null } = {}) {
     if (!prompt || typeof prompt !== 'string' || !prompt.trim()) {
       yield {
         type: 'error',
@@ -1915,7 +1915,7 @@ export class Engine {
     };
     try {
       this.#currentThreadId = threadId || MAIN_THREAD_ID;
-      yield* this.#runQuery({ prompt: effectivePrompt, promptParts: effectivePromptParts, messages, signal: runSignal, userEffort: effectiveUserEffort, scenario, vpPersona, router, senderVpId, inboundEnvelope, taskId, taskMembers, sessionId, sessionMembers, sessionTopics, vpPlan, sessionAnnouncement, workCenterInstructions, workDir, userAlreadyPersisted, getCurrentTodos, setCurrentTodos, askUser, threadId: this.#currentThreadId, vpTurnId, drainPendingUserMessages, closePendingUserInput, collabToolPolicy: effectiveCollabToolPolicy, explicitSkillName: parsedSkill.skillName, retryLifecycle });
+      yield* this.#runQuery({ prompt: effectivePrompt, promptParts: effectivePromptParts, messages, signal: runSignal, userEffort: effectiveUserEffort, scenario, vpPersona, router, senderVpId, inboundEnvelope, taskId, taskMembers, sessionId, sessionMembers, sessionTopics, vpPlan, sessionAnnouncement, workCenterInstructions, workDir, userAlreadyPersisted, getCurrentTodos, setCurrentTodos, askUser, threadId: this.#currentThreadId, vpTurnId, drainPendingUserMessages, prepareProviderRequest, startProviderRequest, finishProviderRequest, failProviderRequest, closePendingUserInput, collabToolPolicy: effectiveCollabToolPolicy, explicitSkillName: parsedSkill.skillName, retryLifecycle });
     } finally {
       // Closing the async generator at a visible retry boundary means the
       // continuation never reached a provider. Keep it out of history and
@@ -1963,7 +1963,7 @@ export class Engine {
    * in a try/finally without indenting the whole loop.
    * @private
    */
-  async *#runQuery({ prompt, promptParts = null, messages, signal, userEffort = null, scenario = 'chat', vpPersona, router, senderVpId, inboundEnvelope, taskId, taskMembers, sessionId, sessionMembers, sessionTopics = null, vpPlan, sessionAnnouncement, workCenterInstructions, workDir, userAlreadyPersisted = false, getCurrentTodos = null, setCurrentTodos = null, askUser = null, threadId = MAIN_THREAD_ID, vpTurnId = null, drainPendingUserMessages = null, closePendingUserInput = null, collabToolPolicy = null, explicitSkillName = null, retryLifecycle }) {
+  async *#runQuery({ prompt, promptParts = null, messages, signal, userEffort = null, scenario = 'chat', vpPersona, router, senderVpId, inboundEnvelope, taskId, taskMembers, sessionId, sessionMembers, sessionTopics = null, vpPlan, sessionAnnouncement, workCenterInstructions, workDir, userAlreadyPersisted = false, getCurrentTodos = null, setCurrentTodos = null, askUser = null, threadId = MAIN_THREAD_ID, vpTurnId = null, drainPendingUserMessages = null, prepareProviderRequest = null, startProviderRequest = null, finishProviderRequest = null, failProviderRequest = null, closePendingUserInput = null, collabToolPolicy = null, explicitSkillName = null, retryLifecycle }) {
 
     const effectiveCollabToolPolicy = collabToolPolicy === COLLAB_TOOL_POLICY.SINGLE_VP || collabToolPolicy === COLLAB_TOOL_POLICY.MULTI_VP
       ? collabToolPolicy
@@ -2367,6 +2367,7 @@ export class Engine {
     let currentModel = this.#config.model;
     let cumulativeInputTokens = 0;
     let cumulativeOutputTokens = 0;
+    let activeProviderRequest = null;
     // task-707: tool-callable end-turn signal. Tools (currently only
     // `route_forward`) can set this via toolCtx.requestEndTurn(reason)
     // to break out of the tool-loop after the current batch finishes
@@ -2645,6 +2646,16 @@ export class Engine {
           retryLifecycle.pendingContinuation = null;
         }
 
+        activeProviderRequest = typeof prepareProviderRequest === 'function'
+          ? prepareProviderRequest({
+              turnNumber,
+              entries: appendedBeforeStream,
+              system: systemPrompt,
+              messages: wireMessages.map(mapDebugMessage),
+              model: currentModel,
+            }) || null
+          : null;
+
         // Snapshot task results carried by this exact request. Request start
         // is not delivery: fetch may remain pending and then be aborted before
         // the provider processes anything. Ack only after a normal stream end
@@ -2661,6 +2672,7 @@ export class Engine {
           effortSource: userEffort ? 'user' : 'auto',
           signal,
           onRawExchange: captureRawExchange,
+          onRequestStart: () => startProviderRequest?.(activeProviderRequest),
         })) {
           // task-325a (abort-stop fix): per-event abort short-circuit.
           // The adapter is expected to throw AbortError when fetch's
@@ -2768,6 +2780,14 @@ export class Engine {
         // escrow so retry or final rescue can deliver the payload.
         if (sawProviderStop) {
           this.#confirmAsyncTaskResults(requestAsyncTaskIds);
+          const completedProviderRequest = activeProviderRequest;
+          activeProviderRequest = null;
+          finishProviderRequest?.(completedProviderRequest, {
+            responseText,
+            stopReason,
+            toolCalls: toolCalls.map(tc => ({ id: tc.id, name: tc.name, input: tc.input })),
+            thinkingBlocks,
+          });
         }
         traceRequest('llm.request_complete', {
           durationMs: perfNowMs() - requestPerfStart,
@@ -2785,6 +2805,10 @@ export class Engine {
         consecutiveRetryableErrors = 0;
       } catch (err) {
         const latencyMs = Date.now() - startTime;
+        if (activeProviderRequest) {
+          failProviderRequest?.(activeProviderRequest, err);
+          activeProviderRequest = null;
+        }
 
         const endAttemptTrace = (attemptStopReason) => {
           this.#trace.endTurn(turnId, {

package/yeaft/work-center/bridge.js

@@ -20,7 +20,7 @@ let shutdownPromise = null;
 let serviceFactory = null;
 
 const BROWSER_DETAIL_OPS = new Set([
-  'get', 'create', 'update', 'start', 'cancel', 'resume', 'action_input', 'retry_action', 'guide', 'retry',
+  'get', 'create', 'update', 'start', 'cancel', 'resume', 'post_work_item_message', 'action_input', 'retry_action', 'guide', 'retry',
 ]);
 const BROWSER_ACTION_DEBUG_OPS = new Set(['get_action_messages', 'get_action_requests', 'get_action_request']);
 // `files` is an internal server-to-Agent field. The browser relay rejects any
@@ -29,6 +29,10 @@ const BROWSER_FILE_FIELDS = Object.freeze({
   create: [
     'title', 'goal', 'acceptanceCriteria', 'workItemType', 'workDir', 'reuseMemory', 'files', 'start',
   ],
+  post_work_item_message: [
+    'id', 'clientMessageId', 'text', 'target', 'revision', 'planRevision', 'ledgerRevision',
+    'coordinatorRevision', 'files',
+  ],
   work_item_message: [
     'id', 'text', 'revision', 'planRevision', 'ledgerRevision', 'coordinatorRevision', 'files',
   ],

package/yeaft/work-center/controller.js

@@ -227,6 +227,8 @@ export class WorkflowController {
   }
 
   input(id, input = {}) {
+    const existingClientMessage = this.store.hasActionInputClientMessage(id, input.actionId, input.clientMessageId);
+    if (existingClientMessage) return this.store.getWorkItemDetail(id);
     const text = typeof input.text === 'string' ? input.text.trim().slice(0, 8_000) : '';
     const addedAttachmentCount = Math.max(0, Number(input.addedAttachmentCount) || 0);
     if (!text && addedAttachmentCount === 0) throw new Error('Action input or attachments are required');
@@ -253,7 +255,7 @@ export class WorkflowController {
         actionId: input.actionId,
         generation: expectedGeneration,
         revision: input.revision,
-      }, input.attachments, input.addedAttachments);
+      }, input.attachments, input.addedAttachments, input.clientMessageId);
     }
     if (!['waiting', 'failed'].includes(targetAction.status)) {
       throw new Error(`Action in ${targetAction.status} cannot accept input`);
@@ -264,7 +266,9 @@ export class WorkflowController {
       expected: { actionId: input.actionId, generation: input.generation, revision: input.revision },
       attachments: input.attachments,
       inputEvent: {
-        inputId: randomUUID(),
+        inputId: input.clientMessageId || randomUUID(),
+        clientMessageId: input.clientMessageId || null,
+        targetActionId: input.actionId,
         text: text || `The user added ${addedAttachmentCount} attachment(s) as additional context for this Action.`,
         attachments: input.addedAttachments,
       },

package/yeaft/work-center/coordinator.js

@@ -558,8 +558,12 @@ export class WorkItemCoordinator {
     }, {
       attachments: input.attachments,
       addedAttachments,
+      clientMessageId: input.clientMessageId,
     });
     if (!started) throw new Error(`WorkItem not found: ${id}`);
+    if (started.duplicate) {
+      return { detail: started.detail, task: Promise.resolve(started.detail), duplicate: true };
+    }
     options.onUpdate?.('coordinator.turn_started', started.detail);
     return this.#scheduleTurn(started, {
       text: promptText,
@@ -569,6 +573,18 @@ export class WorkItemCoordinator {
     });
   }
 
+  resume(started, options = {}) {
+    if (!started?.turnId || !started?.detail || !started?.fence) {
+      throw new Error('Coordinator provider recovery target is invalid');
+    }
+    return this.#scheduleTurn(started, {
+      text: typeof options.text === 'string' ? options.text : '',
+      recovery: options.recovery === true,
+      addedAttachments: Array.isArray(options.addedAttachments) ? options.addedAttachments : [],
+      options,
+    });
+  }
+
   recover(id, options = {}) {
     if (this.shuttingDown) throw new Error('Work Center Coordinator is shutting down');
     const detail = this.store.getWorkItemDetail(id);
@@ -689,19 +705,42 @@ export class WorkItemCoordinator {
               const content = attachmentContext.promptParts.length > 0
                 ? [{ type: 'text', text: latestMessage }, ...attachmentContext.promptParts]
                 : latestMessage;
-              result = await Promise.race([
+              const requestBody = {
+                model: resolved.model,
+                system: coordinatorSystemPrompt(language),
+                messages: [{ role: 'user', content }],
+                maxTokens: Math.min(
+                  resolveMaxOutputTokens(resolved.model, runtime.config),
+                  COORDINATOR_MAX_OUTPUT_TOKENS,
+                ),
+                effort: resolved.effort,
+                effortSource: resolved.source,
+                effortContext: { scenario: 'work-center-coordinator' },
+              };
+              const providerTurn = this.store.prepareCoordinatorProviderTurn(
+                started.detail.id, started.turnId, attemptCount, requestBody,
+              );
+              if (providerTurn.status === 'unknown') {
+                throw new Error('Coordinator provider dispatch outcome is unknown and requires review');
+              }
+              if (providerTurn.status === 'responded') {
+                result = providerTurn.response;
+              } else {
+                result = await Promise.race([
                 runtime.adapter.call({
-                  model: resolved.model,
-                  system: coordinatorSystemPrompt(language),
-                  messages: [{ role: 'user', content }],
-                  maxTokens: Math.min(
-                    resolveMaxOutputTokens(resolved.model, runtime.config),
-                    COORDINATOR_MAX_OUTPUT_TOKENS,
-                  ),
-                  effort: resolved.effort,
-                  effortSource: resolved.source,
-                  effortContext: { scenario: 'work-center-coordinator' },
+                  ...requestBody,
                   signal: abortController.signal,
+                  onRequestStart: () => {
+                    if (!this.store.dispatchCoordinatorProviderTurn(providerTurn.id)) {
+                      throw new Error('Coordinator provider turn lost its dispatch fence');
+                    }
+                  },
+                }).then(response => {
+                  const persisted = this.store.respondCoordinatorProviderTurn(
+                    providerTurn.id, providerTurn.requestHash, response,
+                  );
+                  if (!persisted) throw new Error('Coordinator provider response lost its CAS fence');
+                  return response;
                 }),
                 new Promise((_, reject) => {
                   abortController.signal.addEventListener('abort', () => {
@@ -709,6 +748,7 @@ export class WorkItemCoordinator {
                   }, { once: true });
                 }),
               ]);
+              }
             } catch (error) {
               if (abortController.signal.aborted || this.shuttingDown) throw error;
               throw coordinatorExecutionError(error, 'provider', language);