@yeaft/webchat-agent 1.0.456 → 1.0.458

package/connection/message-router.js

@@ -28,9 +28,10 @@ import { handleRestartAgent, handleUpgradeAgent } from './upgrade.js';
 import { loadMcpServers, updateMcpConfig } from '../mcp.js';
 import { getLlmConfig, updateLlmConfig, getYeaftSettings, updateYeaftSettings, getPluginConfig, updatePluginConfig, getTelemetrySettings, updateTelemetrySettings, getSearchSettings, updateSearchSettings, fetchTavilyUsage } from '../yeaft/config-api.js';
 import { loadConfig } from '../yeaft/config.js';
+import { mutateAgentConfig } from '../yeaft/config-store.js';
 import { discoverLlmModels } from '../llm-model-discovery.js';
 import { fetchModelsDev } from '../yeaft/llm/models-dev.js';
-import { handleYeaftSessionSend, handleYeaftAskUserAnswer, handleYeaftSubAgentPrompt, handleYeaftTaskCancel, handleYeaftModeSwitch, handleYeaftModelSwitch, resetYeaftSession, refreshLiveSessionConfig, handleYeaftLoadHistory, handleYeaftLoadHistoryOutline, handleYeaftSearchHistory, handleYeaftLoadHistoryWindow, handleYeaftLoadMoreHistory, handleYeaftAbortThread, handleYeaftAbortAll, handleYeaftAbortTurn, handleYeaftVpSubscribe, handleYeaftVpCreate, handleYeaftVpUpdate, handleYeaftVpDelete, handleYeaftVpRead, handleYeaftListSessions, handleYeaftProjectContextSync, handleYeaftProjectMutation, handleYeaftCreateSession, handleYeaftRenameSession, handleYeaftUpdateSession, handleYeaftUpdateSessionConfig, handleYeaftArchiveSession, handleYeaftDeleteSession, handleYeaftSessionAddMember, handleYeaftSessionRemoveMember, handleYeaftSessionSetDefaultVp, handleYeaftScanWorkdirSessions, handleYeaftRestoreSession, handleYeaftDreamTrigger, handleYeaftFetchToolStats, handleYeaftFetchDebugHistory, handleYeaftMcpList, handleYeaftMcpAdd, handleYeaftMcpRemove, handleYeaftMcpReload, handleYeaftPluginCatalog, handleYeaftManagedSkill, broadcastLanguageChange, broadcastYeaftSessionSnapshotEager, broadcastYeaftVpSnapshotEager, preloadYeaftSkillSlashCommands } from '../yeaft/web-bridge.js';
+import { handleYeaftSessionSend, handleYeaftAskUserAnswer, handleYeaftSubAgentPrompt, handleYeaftTaskCancel, handleYeaftModeSwitch, handleYeaftModelSwitch, resetYeaftSession, refreshLiveSessionConfig, handleYeaftLoadHistory, handleYeaftLoadHistoryOutline, handleYeaftSearchHistory, handleYeaftLoadHistoryWindow, handleYeaftLoadMoreHistory, handleYeaftAbortThread, handleYeaftAbortAll, handleYeaftAbortTurn, handleYeaftVpSubscribe, handleYeaftVpCreate, handleYeaftVpUpdate, handleYeaftVpDelete, handleYeaftVpRead, handleYeaftListSessions, handleYeaftProjectContextSync, handleYeaftProjectMutation, handleYeaftCreateSession, handleYeaftRenameSession, handleYeaftUpdateSession, handleYeaftUpdateSessionConfig, handleYeaftArchiveSession, handleYeaftDeleteSession, handleYeaftSessionAddMember, handleYeaftSessionRemoveMember, handleYeaftSessionSetDefaultVp, handleYeaftScanWorkdirSessions, handleYeaftRestoreSession, handleYeaftDreamTrigger, handleYeaftFetchToolStats, handleYeaftFetchDebugHistory, handleYeaftMcpList, handleYeaftMcpAdd, handleYeaftMcpRemove, handleYeaftMcpReload, handleYeaftPluginCatalog, handleYeaftManagedSkill, ensureSessionLoaded, broadcastLanguageChange, broadcastYeaftSessionSnapshotEager, broadcastYeaftVpSnapshotEager, preloadYeaftSkillSlashCommands } from '../yeaft/web-bridge.js';
 import { startYeaftStatusRefresh, forceRefreshYeaftStatus } from '../yeaft/status-cache.js';
 import { handleWorkCenterRequest } from '../yeaft/work-center/bridge.js';
 import { handleBrowserRuntimeMessage } from '../browser-runtime/messages.js';
@@ -143,7 +144,8 @@ export async function handleMessage(msg) {
       await ctx.assetOutbox?.drain();
 
       // ★ Phase 1: 通知 server 同步完成
-      sendToServer({ type: 'agent_sync_complete' });
+      const runtimeConfig = loadConfig({ dir: ctx.CONFIG?.yeaftDir });
+      sendToServer({ type: 'agent_sync_complete', dreamEnabled: runtimeConfig.dream?.enabled !== false });
 
       // ★ 发送 MCP servers 列表给 server（供前端 Settings > Tools tab 使用）
       if (ctx.mcpServers.length > 0) {
@@ -351,6 +353,23 @@ export async function handleMessage(msg) {
       await handleRestartAgent();
       break;
 
+    case 'set_dream_enabled': {
+      const enabled = msg.enabled !== false;
+      try {
+        mutateAgentConfig(ctx.CONFIG?.yeaftDir, (config) => {
+          if (!config.dream || typeof config.dream !== 'object' || Array.isArray(config.dream)) config.dream = {};
+          config.dream.enabled = enabled;
+        });
+        const runtime = await ensureSessionLoaded();
+        runtime.config.dream = { ...(runtime.config.dream || {}), enabled };
+        runtime.dreamScheduler?.setEnabled?.(enabled);
+        sendToServer({ type: 'dream_enabled_changed', enabled });
+      } catch (error) {
+        sendToServer({ type: 'dream_enabled_changed', enabled: !enabled, error: error?.message || String(error) });
+      }
+      break;
+    }
+
     case 'upgrade_agent':
       await handleUpgradeAgent();
       break;

package/local-runtime/server/handlers/agent-sync.js

@@ -65,6 +65,7 @@ export async function handleAgentSync(agentId, agent, msg) {
     // Phase 1: Agent 同步完成
     case 'agent_sync_complete': {
       agent.status = 'ready';
+      agent.dreamEnabled = msg.dreamEnabled !== false;
       if (agent._syncTimeout) {
         clearTimeout(agent._syncTimeout);
         delete agent._syncTimeout;
@@ -74,6 +75,24 @@ export async function handleAgentSync(agentId, agent, msg) {
       break;
     }
 
+    case 'dream_enabled_changed': {
+      if (!msg.error) {
+        agent.dreamEnabled = msg.enabled !== false;
+        await broadcastAgentList();
+      }
+      for (const [, client] of webClients) {
+        if (!client.authenticated) continue;
+        if (!CONFIG.skipAuth && agent.ownerId !== client.userId && !(!agent.ownerId && client.role === 'admin')) continue;
+        await sendToWebClient(client, {
+          type: 'dream_enabled_changed',
+          agentId,
+          enabled: msg.error ? agent.dreamEnabled !== false : msg.enabled !== false,
+          ...(msg.error ? { error: msg.error } : {}),
+        });
+      }
+      break;
+    }
+
     case 'agent_capabilities_updated': {
       const capabilities = Array.isArray(msg.capabilities)
         ? [...new Set(msg.capabilities.filter(value => (

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

@@ -56,6 +56,14 @@ export async function handleClientMisc(clientId, client, msg, checkAgentAccess)
       break;
     }
 
+    case 'set_dream_enabled': {
+      const agentId = msg.agentId;
+      if (!agentId) break;
+      if (!await checkAgentAccess(agentId)) break;
+      await forwardToAgent(agentId, { type: 'set_dream_enabled', enabled: msg.enabled !== false });
+      break;
+    }
+
     case 'upgrade_agent': {
       const upgradeAgentId = msg.agentId;
       if (!upgradeAgentId) break;

package/local-runtime/server/ws-utils.js

@@ -171,6 +171,7 @@ export async function broadcastAgentList() {
           capabilities: agent.capabilities || ['terminal', 'file_editor', 'background_tasks'],
           ...(agent.capabilityMetadataProvided === true ? { capabilityMetadataProvided: true } : {}),
           version: agent.version || null,
+          dreamEnabled: agent.dreamEnabled !== false,
           yeaftStatus: agent.yeaftStatus || null,
           proxyPorts: agent.proxyPorts || [],
           conversations: Array.from(agent.conversations.values()).filter(c =>

package/local-runtime/version.json

@@ -1 +1 @@
-{"version":"1.0.456"}
+{"version":"1.0.458"}