@yeaft/webchat-agent 0.1.443 → 0.1.445

package/connection/message-router.js

@@ -25,7 +25,7 @@ import { sendToServer, flushMessageBuffer } from './buffer.js';
 import { handleRestartAgent, handleUpgradeAgent } from './upgrade.js';
 import { loadMcpServers, updateMcpConfig } from '../mcp.js';
 import { getLlmConfig, updateLlmConfig } from '../unify/config-api.js';
-import { handleUnifyChat, handleUnifyModeSwitch, resetUnifySession } from '../unify/web-bridge.js';
+import { handleUnifyChat, handleUnifyModeSwitch, handleUnifyModelSwitch, resetUnifySession } from '../unify/web-bridge.js';
 
 export async function handleMessage(msg) {
   switch (msg.type) {
@@ -329,6 +329,10 @@ export async function handleMessage(msg) {
       handleUnifyModeSwitch(msg);
       break;
 
+    case 'unify_model_switch':
+      handleUnifyModelSwitch(msg);
+      break;
+
     case 'unify_reset':
       await resetUnifySession();
       break;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@yeaft/webchat-agent",
-  "version": "0.1.443",
+  "version": "0.1.445",
   "description": "Remote agent for Yeaft WebChat — connects worker machines to the central server",
   "main": "index.js",
   "type": "module",

package/unify/config.js

@@ -260,6 +260,24 @@ export function loadConfig(overrides = {}) {
     adapter: null,
   };
 
+  // Aggregate all available models from providers
+  config.availableModels = [];
+  if (providers) {
+    for (const p of providers) {
+      if (!Array.isArray(p.models)) continue;
+      for (const m of p.models) {
+        // Avoid duplicates (first provider wins)
+        if (!config.availableModels.some(am => am.id === m)) {
+          config.availableModels.push({
+            id: m,
+            provider: p.name,
+            label: m,
+          });
+        }
+      }
+    }
+  }
+
   return config;
 }
 

package/unify/web-bridge.js

@@ -32,6 +32,11 @@ let unifyConversationId = null;
 /** Current query mode: 'chat' or 'work' */
 let currentMode = 'chat';
 
+/** Accumulated conversation messages for context continuity across queries.
+ *  Each entry is { role: 'user'|'assistant', content: string|Array }.
+ *  Cleared on session reset or consolidation. */
+let conversationMessages = [];
+
 /** Whether we've already sent a permission warning to the UI */
 let _permissionDiagnosticSent = false;
 
@@ -103,6 +108,7 @@ export async function handleUnifyChat(msg) {
         type: 'session_ready',
         conversationId: unifyConversationId,
         model: session.config.model,
+        availableModels: session.config.availableModels || [],
         skills: session.status.skills,
         mcpServers: session.status.mcpServers,
         tools: session.status.tools,
@@ -132,10 +138,14 @@ export async function handleUnifyChat(msg) {
     resetQueryTimer();
 
     try {
+    // ─── Collect assistant response for conversation history ──
+    let assistantTextParts = [];
+
     // ─── Stream Engine events → claude_output format ──
     for await (const event of session.engine.query({
       prompt,
       mode: currentMode,
+      messages: conversationMessages,
       signal: currentAbort.signal,
     })) {
       // Reset timeout on every event — activity means the query is alive
@@ -143,6 +153,7 @@ export async function handleUnifyChat(msg) {
       switch (event.type) {
         // ── Text streaming ──
         case 'text_delta':
+          assistantTextParts.push(event.text);
           sendUnifyOutput({
             type: 'assistant',
             message: {
@@ -231,6 +242,9 @@ export async function handleUnifyChat(msg) {
 
         // ── Context consolidation ──
         case 'consolidate':
+          // Engine has compressed the context — clear our accumulated history.
+          // The engine's compactSummary will provide context on next query.
+          conversationMessages = [];
           sendUnifyEvent({
             type: 'consolidate',
             archivedCount: event.archivedCount,
@@ -302,7 +316,15 @@ export async function handleUnifyChat(msg) {
       }
     }
 
-    // ─── Query complete — signal turn end ──
+    // ─── Query complete — accumulate messages for context continuity ──
+    conversationMessages.push({ role: 'user', content: prompt });
+
+    const fullText = assistantTextParts.join('');
+    if (fullText) {
+      conversationMessages.push({ role: 'assistant', content: fullText });
+    }
+
+    // ─── Signal turn end to UI ──
     // Finish any streaming text
     sendUnifyOutput({
       type: 'assistant',
@@ -385,6 +407,32 @@ export function handleUnifyModeSwitch(msg) {
   }
 }
 
+/**
+ * Handle model switch from the web UI.
+ * Updates Engine's config so the next query uses the new model.
+ * @param {{ model: string }} msg
+ */
+export function handleUnifyModelSwitch(msg) {
+  if (!session || !msg.model) return;
+
+  // Validate: model must be in availableModels list
+  const available = session.config.availableModels || [];
+  const found = available.some(m => m.id === msg.model);
+  if (!found) {
+    console.warn(`[Unify] model switch rejected — "${msg.model}" not in availableModels`);
+    return;
+  }
+
+  // Update Engine's model for subsequent queries
+  session.config.model = msg.model;
+
+  // Confirm switch to frontend
+  sendUnifyEvent({
+    type: 'model_switched',
+    model: msg.model,
+  });
+}
+
 /**
  * Reset Unify session (for clear messages).
  */
@@ -399,4 +447,5 @@ export async function resetUnifySession() {
   }
   unifyConversationId = null;
   currentMode = 'chat';
+  conversationMessages = [];
 }