@yeaft/webchat-agent 1.0.173 → 1.0.175

package/yeaft/work-center/service.js

@@ -12,7 +12,13 @@ import {
   removeWorkItemAttachments,
 } from './attachments.js';
 import { normalizeSessionContextSnapshot } from './session-context.js';
-import { projectWorkItemDetail, projectWorkItemSummary } from './projection.js';
+import {
+  projectActionMessagePage,
+  projectActionRequestDetail,
+  projectActionRequestIndex,
+  projectWorkItemDetail,
+  projectWorkItemSummary,
+} from './projection.js';
 import { readWorkCenterSettings, writeWorkCenterSettings } from './settings.js';
 import {
   defaultWorkCenterStageInstructions,
@@ -81,7 +87,38 @@ export class WorkCenterService {
           watcher: this.watcher.status(),
         };
       case 'get':
-        return projectWorkItemDetail(this.#requiredItem(payload.id));
+        return this.#requiredItem(payload.id);
+      case 'get_action_messages': {
+        const detail = this.#requiredItem(payload.id);
+        const action = this.#requiredAction(detail, payload.actionId);
+        return projectActionMessagePage(action, detail.runs, detail.events, {
+          cursor: payload.cursor,
+          limit: payload.limit,
+        });
+      }
+      case 'get_action_requests': {
+        const detail = this.#requiredItem(payload.id);
+        const action = this.#requiredAction(detail, payload.actionId);
+        const entries = [];
+        for (const run of detail.runs.filter(item => item.actionId === action.id)) {
+          const history = await this.#debugHistory(run, { indexOnly: true });
+          for (const turn of Array.isArray(history?.turns) ? history.turns : []) {
+            entries.push({ run, turn });
+          }
+        }
+        return projectActionRequestIndex(action, entries);
+      }
+      case 'get_action_request': {
+        const detail = this.#requiredItem(payload.id);
+        const action = this.#requiredAction(detail, payload.actionId);
+        const requestId = requiredString(payload.requestId, 'requestId');
+        const run = detail.runs.find(item => item.actionId === action.id && item.id === payload.runId);
+        if (!run) throw new Error('Action request not found');
+        const history = await this.#debugHistory(run, { detailTurnId: requestId });
+        const projected = projectActionRequestDetail(action, run, history);
+        if (!projected) throw new Error('Action request detail is no longer available');
+        return projected;
+      }
       case 'get_settings': {
         const settings = this.settingsReader(this.yeaftDir);
         return { settings, runtime: await this.runtimeInfo() };
@@ -168,6 +205,38 @@ export class WorkCenterService {
         this.#emit({ type: 'work_item.cancelled', workItem: detail });
         return detail;
       }
+      case 'action_input': {
+        const id = requiredString(payload.id, 'id');
+        const workItem = this.#requiredItem(id);
+        let addedAttachments = [];
+        let detail;
+        try {
+          addedAttachments = appendWorkItemAttachments(workItem.attachments, payload.files, {
+            root: this.attachmentRoot,
+            workItemId: id,
+          });
+          detail = this.controller.input(id, {
+            text: typeof payload.text === 'string' ? payload.text : '',
+            actionId: typeof payload.actionId === 'string' ? payload.actionId : '',
+            revision: payload.revision,
+            addedAttachmentCount: addedAttachments.length,
+            addedAttachments,
+            attachments: [...(workItem.attachments || []), ...addedAttachments],
+          });
+        } catch (error) {
+          try {
+            if ((workItem.attachments || []).length === 0 && addedAttachments.length > 0) {
+              removeWorkItemAttachments(this.attachmentRoot, id);
+            } else {
+              removeWorkItemAttachmentFiles(this.attachmentRoot, id, addedAttachments);
+            }
+          } catch {}
+          throw error;
+        }
+        this.watcher.abortInvalidWorkItemRuns(id);
+        this.#emit({ type: 'action.input_added', workItem: detail });
+        return detail;
+      }
       case 'guide': {
         const id = requiredString(payload.id, 'id');
         const workItem = this.#requiredItem(id);
@@ -183,6 +252,7 @@ export class WorkCenterService {
             actionId: typeof payload.actionId === 'string' ? payload.actionId : '',
             revision: payload.revision,
             addedAttachmentCount: addedAttachments.length,
+            addedAttachments,
             attachments: [...(workItem.attachments || []), ...addedAttachments],
           });
         } catch (error) {
@@ -221,13 +291,6 @@ export class WorkCenterService {
           },
         };
       }
-      case 'retry': {
-        const detail = this.controller.retry(requiredString(payload.id, 'id'), {
-          answer: typeof payload.answer === 'string' ? payload.answer : '',
-        });
-        this.#emit({ type: 'work_item.retried', workItem: detail });
-        return detail;
-      }
       case 'set_watcher':
         if (payload.enabled === false) await this.watcher.stop();
         else this.watcher.start();
@@ -243,6 +306,28 @@ export class WorkCenterService {
     return item;
   }
 
+  #requiredAction(detail, actionId) {
+    const id = requiredString(actionId, 'actionId');
+    const action = detail.actions.find(item => item.id === id);
+    if (!action) throw new Error(`Action not found: ${id}`);
+    return action;
+  }
+
+  async #debugHistory(run, options = {}) {
+    const trace = this.watcher.runner?.trace;
+    if (!trace || typeof trace.fetchRecentDebugHistory !== 'function') {
+      return { turns: [], loops: [] };
+    }
+    return trace.fetchRecentDebugHistory({
+      limit: 10,
+      dreamLimit: 0,
+      sessionId: `work-item-${run.workItemId}`,
+      threadId: run.id,
+      indexOnly: options.indexOnly === true,
+      detailTurnId: options.detailTurnId || null,
+    });
+  }
+
   #emit(event) {
     try { this.onEvent(event); } catch {}
   }
@@ -253,6 +338,7 @@ export class WorkCenterService {
 
   async shutdown() {
     await this.watcher.stop();
+    try { await this.watcher.runner?.trace?.close?.(); } catch {}
     this.store.close();
   }
 }

package/yeaft/work-center/store.js

@@ -755,7 +755,7 @@ export class WorkItemStore {
     });
   }
 
-  addActionGuidance(id, guidance, expected, makeAction, attachments = null) {
+  addActionGuidance(id, guidance, expected, makeAction, attachments = null, addedAttachments = []) {
     return withTransaction(this.db, () => {
       const workItem = this.getWorkItem(id);
       if (!workItem) return null;
@@ -788,7 +788,16 @@ export class WorkItemStore {
         now,
         id,
       );
-      this.appendEvent(id, 'action.guidance_added', { guidance }, { actionId: action.id });
+      this.appendEvent(id, 'action.guidance_added', {
+        guidance,
+        attachments: (Array.isArray(addedAttachments) ? addedAttachments : []).map(attachment => ({
+          id: attachment.id,
+          name: attachment.name,
+          mimeType: attachment.mimeType,
+          size: Math.max(0, Number(attachment.size) || 0),
+          isImage: attachment.isImage === true,
+        })),
+      }, { actionId: action.id });
       return this.getWorkItemDetail(id);
     });
   }
@@ -906,13 +915,18 @@ export class WorkItemStore {
     });
   }
 
-  retryWorkItemAtomic(id, makeAction) {
+  retryWorkItemAtomic(id, makeAction, options = {}) {
     return withTransaction(this.db, () => {
       const workItem = this.getWorkItem(id);
       if (!workItem) return null;
       if (!['waiting', 'needs_attention'].includes(workItem.status)) {
         throw new Error(`WorkItem in ${workItem.status} does not need retry`);
       }
+      if (options.expected) {
+        if (workItem.currentActionId !== options.expected.actionId || workItem.revision !== options.expected.revision) {
+          throw new Error('Action changed before input was applied; refresh and try again');
+        }
+      }
       const previous = workItem.currentActionId ? this.getAction(workItem.currentActionId) : null;
       const previousRun = previous
         ? mapRun(this.db.prepare(`SELECT * FROM runs WHERE work_item_id = ? AND action_id = ?
@@ -939,8 +953,20 @@ export class WorkItemStore {
         contractRevision: workItem.revision,
       }, this.#nextSequence(id), now);
       this.db.prepare(`UPDATE work_items SET status = 'ready', current_action_id = ?,
-        current_run_id = NULL, updated_at = ? WHERE id = ?`).run(action.id, now, id);
-      this.appendEvent(id, 'work_item.retried', {}, { actionId: action.id });
+        current_run_id = NULL, attachments = ?, updated_at = ? WHERE id = ?`).run(
+        action.id,
+        stringify(Array.isArray(options.attachments) ? options.attachments : workItem.attachments),
+        now,
+        id,
+      );
+      const inputEvent = options.inputEvent && typeof options.inputEvent === 'object'
+        ? options.inputEvent
+        : null;
+      if (inputEvent) {
+        this.appendEvent(id, 'action.input_added', inputEvent, { actionId: action.id });
+      } else {
+        this.appendEvent(id, 'work_item.retried', {}, { actionId: action.id });
+      }
       return this.getWorkItemDetail(id);
     });
   }