@yeaft/webchat-agent 1.0.345 → 1.0.347

package/yeaft/work-center/runner.js

@@ -12,6 +12,7 @@ import { runPreflow } from '../memory/preflow.js';
 import { formatPickedForInjection } from '../sessions/pre-flow.js';
 import { existsSync, lstatSync, realpathSync } from 'node:fs';
 import path from 'node:path';
+import { sessionMessageQuotePrompt } from '../session-message-quote.js';
 import { buildWorkItemAttachmentContext } from './attachments.js';
 import { withUsageAccounting } from '../llm/usage-accounting.js';
 import {
@@ -62,6 +63,21 @@ const WORK_ITEM_TOOL_NAMES = Object.freeze([
 ]);
 const WORK_ITEM_TOOL_ALLOWLIST = new Set(WORK_ITEM_TOOL_NAMES);
 const DEFAULT_PROGRESS_INTERVAL_MS = 200;
+const ACTION_INPUT_QUOTE_MAX_BYTES = 8 * 1024;
+
+export function renderPendingActionInput(item, attachmentFileById = new Map()) {
+  const attachments = Array.isArray(item?.attachments) ? item.attachments : [];
+  const attachmentLines = attachments.map(attachment => {
+    const file = attachmentFileById.get(attachment.id);
+    return file ? `- ${attachment.name}: ${file.ref}` : `- ${attachment.name}`;
+  });
+  const quoteContext = sessionMessageQuotePrompt(item?.quote, {
+    maxBytes: ACTION_INPUT_QUOTE_MAX_BYTES,
+  });
+  return [item?.text || '', quoteContext, attachmentLines.length > 0
+    ? `Additional WorkItem attachments:\n${attachmentLines.join('\n')}` : '']
+    .filter(Boolean).join('\n\n');
+}
 
 function structuredOutcome(value) {
   return value && typeof value === 'object'
@@ -1197,13 +1213,7 @@ export class WorkItemRunner {
       ) || [];
       const accepted = [];
       for (const item of pending) {
-        const attachmentLines = item.attachments.map(attachment => {
-          const file = attachmentFileById.get(attachment.id);
-          return file ? `- ${attachment.name}: ${file.ref}` : `- ${attachment.name}`;
-        });
-        const content = [item.text, attachmentLines.length > 0
-          ? `Additional WorkItem attachments:\n${attachmentLines.join('\n')}` : '']
-          .filter(Boolean).join('\n\n');
+        const content = renderPendingActionInput(item, attachmentFileById);
         if (!content) continue;
         pendingEntriesById.set(String(item.id), item);
         accepted.push({

package/yeaft/work-center/service.js

@@ -388,6 +388,7 @@ export class WorkCenterService {
             ledgerRevision: payload.ledgerRevision,
             coordinatorRevision: payload.coordinatorRevision,
             clientMessageId: typeof payload.clientMessageId === 'string' ? payload.clientMessageId : null,
+            quote: payload.quote,
             addedAttachments,
             attachments: [...(workItem.attachments || []), ...addedAttachments],
           }, {
@@ -450,6 +451,7 @@ export class WorkCenterService {
             revision: payload.revision,
             generation,
             clientMessageId: typeof payload.clientMessageId === 'string' ? payload.clientMessageId : null,
+            quote: payload.quote,
             addedAttachmentCount: addedAttachments.length,
             addedAttachments,
             attachments: [...(workItem.attachments || []), ...addedAttachments],
@@ -610,6 +612,7 @@ export class WorkCenterService {
         recovery: Boolean(recoverable.recovery),
         addedAttachments: Array.isArray(recoverable.addedAttachments)
           ? recoverable.addedAttachments : [],
+        quote: recoverable.quote || null,
         onUpdate: (type, detail) => this.#emit({ type, workItem: detail }),
       });
       turn?.task?.catch?.(() => {});

package/yeaft/work-center/store.js

@@ -385,6 +385,8 @@ function carryCurrentActionInputContext(db, action, extraInputs = [], explicitOn
     return [{
       ...entry,
       inputId,
+      summary: event?.data?.text || entry.summary || '',
+      quote: event?.data?.quote || entry.quote || null,
       attachments: Array.isArray(entry.attachments) && entry.attachments.length > 0
         ? entry.attachments
         : inputAttachments(event),
@@ -400,6 +402,7 @@ function carryCurrentActionInputContext(db, action, extraInputs = [], explicitOn
       role: 'user',
       inputId,
       summary: event.data?.text || '',
+      quote: event.data?.quote || null,
       attachments: inputAttachments(event),
       evidence: [],
     });
@@ -1755,7 +1758,7 @@ export class WorkItemStore {
     return true;
   }
 
-  addActionInput(id, input, expected, attachments = null, addedAttachments = [], clientMessageId = null) {
+  addActionInput(id, input, expected, attachments = null, addedAttachments = [], clientMessageId = null, quote = null) {
     return withTransaction(this.db, () => {
       const sourceKey = typeof clientMessageId === 'string' && clientMessageId
         ? `client:message:${id}:${clientMessageId}` : null;
@@ -1818,6 +1821,7 @@ export class WorkItemStore {
           role: 'user',
           inputId,
           summary: input,
+          quote,
           attachments: projectedAttachments,
           evidence: [],
         });
@@ -1871,6 +1875,7 @@ export class WorkItemStore {
         inputId,
         clientMessageId,
         text: input,
+        quote,
         attachments: projectedAttachments,
       }, { actionId: action.id, runId: eventRunId, actionGeneration: eventGeneration });
       if (action.status === 'running') {
@@ -1896,7 +1901,7 @@ export class WorkItemStore {
           status: 'pending',
           text: input,
           attachments: projectedAttachments,
-          payload: { eventId, inputId, actionGeneration: eventGeneration, actionSpecHash: eventSpecHash },
+          payload: { eventId, inputId, quote, actionGeneration: eventGeneration, actionSpecHash: eventSpecHash },
           createdAt: now,
         }, sourceKey || `pending_action_inputs:event:${eventId}`);
       } else if (sourceKey) {
@@ -1909,7 +1914,7 @@ export class WorkItemStore {
           status: 'consumed',
           text: input,
           attachments: projectedAttachments,
-          payload: { eventId, inputId, actionGeneration: eventGeneration, actionSpecHash: eventSpecHash },
+          payload: { eventId, inputId, quote, actionGeneration: eventGeneration, actionSpecHash: eventSpecHash },
           createdAt: now,
         }, sourceKey);
       }
@@ -1920,8 +1925,8 @@ export class WorkItemStore {
   listPendingActionInputs(actionId, runId, ownerBootId, leaseEpoch) {
     const active = this.#activeRunRow(runId, ownerBootId, leaseEpoch, true);
     if (!active || active.action_id !== actionId) return [];
-    return this.db.prepare(`SELECT p.*, ae.id AS action_entry_id, ae.sequence AS action_entry_sequence
-      FROM pending_action_inputs p
+    return this.db.prepare(`SELECT p.*, ae.id AS action_entry_id, ae.sequence AS action_entry_sequence,
+      ae.payload AS action_entry_payload FROM pending_action_inputs p
       LEFT JOIN action_entries ae ON CAST(json_extract(ae.payload, '$.eventId') AS INTEGER) = p.event_id
         AND ae.action_id = p.action_id
       WHERE p.action_id = ? AND p.run_id = ? AND p.action_generation = ? AND p.action_spec_hash = ?
@@ -1932,6 +1937,7 @@ export class WorkItemStore {
       actionEntryId: row.action_entry_id || null,
       sequence: row.action_entry_sequence == null ? null : Number(row.action_entry_sequence),
       text: row.text || '',
+      quote: parseJson(row.action_entry_payload, null)?.quote || null,
       attachments: parseJson(row.attachments, []),
     }));
   }
@@ -3155,9 +3161,11 @@ export class WorkItemStore {
         throw new Error('WorkItem Coordinator message or attachments are required');
       }
       const turnId = randomUUID();
+      const quote = options.quote && typeof options.quote === 'object' ? options.quote : null;
       const userMessage = automaticRecovery ? null : {
         id: randomUUID(), turnId, role: 'user', text, attachments: projectedAttachments,
         status: 'completed', createdAt: now,
+        ...(quote ? { quote } : {}),
       };
       const assistantMessage = {
         id: randomUUID(), turnId, role: 'assistant', text: '', status: 'thinking',
@@ -3177,6 +3185,7 @@ export class WorkItemStore {
         text,
         recovery,
         clientMessageId,
+        ...(quote ? { quote } : {}),
         addedAttachments: projectedAttachments,
       }, clientMessageId ? `client:message:${id}:${clientMessageId}` : `coordinator:turn:${turnId}`);
       const messages = [...(workItem.messages || []), ...(userMessage ? [userMessage] : []), assistantMessage]

package/yeaft/work-center/workflow.js

@@ -1,3 +1,4 @@
+import { sessionMessageQuotePrompt } from '../session-message-quote.js';
 import { renderSessionContextSnapshot } from './session-context.js';
 
 export const BUILT_IN_ACTION_TYPES = Object.freeze([
@@ -22,6 +23,7 @@ const MODEL_MODES = new Set(['inherit', 'primary', 'fast', 'specific']);
 const MODEL_EFFORTS = new Set(['minimal', 'low', 'medium', 'high', 'xhigh', 'max']);
 const WORKSPACE_MODES = new Set(['shared', 'read', 'isolated-write', 'integrate']);
 const HIGH_EFFORT_ACTION_TYPES = new Set(['triage', 'research', 'design', 'diagnose', 'review']);
+const ACTION_CONTEXT_QUOTE_MAX_BYTES = 8 * 1024;
 
 const DEFAULT_STAGE_INSTRUCTIONS = Object.freeze({
   triage: 'Turn the request into an executable contract. Inspect relevant repository facts before deciding the flow. Classify the WorkItem, identify constraints, risks, dependencies, and missing acceptance criteria, then plan only the Actions needed for this task. Do not implement. If the goal or acceptance criteria must change, submit a contractPatch and explain why.',
@@ -716,8 +718,11 @@ function renderContext(context = []) {
     const decision = entry.reviewDecision ? `\nReview decision: ${entry.reviewDecision}` : '';
     const waitingReason = entry.waitingReason ? `\nWaiting reason: ${entry.waitingReason}` : '';
     const answer = entry.answer ? `\nUser answer: ${entry.answer}` : '';
+    const quote = entry.quote
+      ? sessionMessageQuotePrompt(entry.quote, { maxBytes: ACTION_CONTEXT_QUOTE_MAX_BYTES })
+      : '';
     const source = entry.sourceTitle ? ` from ${entry.sourceTitle}` : '';
-    return `### ${entry.type}${source} (${entry.vpId || entry.role || 'unknown VP'})\n${entry.summary || '(no summary)'}${decision}${waitingReason}${answer}${evidence}`;
+    return `### ${entry.type}${source} (${entry.vpId || entry.role || 'unknown VP'})\n${entry.summary || '(no summary)'}${decision}${waitingReason}${answer}${quote}${evidence}`;
   });
   return `\n\nReusable Work Center context and prior Action results:\n${blocks.join('\n\n')}`;
 }