@yeaft/webchat-agent 0.1.413 → 0.1.414

package/connection/message-router.js

@@ -24,6 +24,7 @@ import {
 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';
 
 export async function handleMessage(msg) {
   switch (msg.type) {
@@ -304,5 +305,18 @@ export async function handleMessage(msg) {
       sendToServer({ type: 'mcp_config_updated', servers: updated });
       break;
     }
+
+    // LLM configuration (read/write ~/.yeaft/config.json)
+    case 'get_llm_config': {
+      const config = getLlmConfig(ctx.CONFIG?.yeaftDir);
+      sendToServer({ type: 'llm_config', ...config });
+      break;
+    }
+
+    case 'update_llm_config': {
+      const result = updateLlmConfig(msg.config || {}, ctx.CONFIG?.yeaftDir);
+      sendToServer({ type: 'llm_config_updated', ...result });
+      break;
+    }
   }
 }

package/package.json

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

package/unify/config-api.js

@@ -0,0 +1,117 @@
+/**
+ * config-api.js — Read/write Yeaft config.json for remote management
+ *
+ * Provides functions to get and update the LLM-related portion of
+ * ~/.yeaft/config.json via WebSocket messages from the web UI.
+ *
+ * Only exposes provider/model configuration — not internal fields
+ * like maxContinueTurns or debug that don't belong in the UI.
+ */
+
+import { existsSync, readFileSync, writeFileSync } from 'fs';
+import { join } from 'path';
+import { DEFAULT_YEAFT_DIR } from './init.js';
+
+/**
+ * Read the LLM-relevant portion of config.json.
+ *
+ * @param {string} [dir] — Yeaft data directory
+ * @returns {{ providers, primaryModel, fastModel, language } | { error: string }}
+ */
+export function getLlmConfig(dir) {
+  const root = dir || process.env.YEAFT_DIR || DEFAULT_YEAFT_DIR;
+  const configPath = join(root, 'config.json');
+
+  if (!existsSync(configPath)) {
+    return { providers: [], primaryModel: null, fastModel: null, language: 'en', needsSetup: true };
+  }
+
+  try {
+    const raw = readFileSync(configPath, 'utf8');
+    const json = JSON.parse(raw);
+    const providers = Array.isArray(json.providers) ? json.providers : [];
+
+    // Detect if config still has default/placeholder values (first-time setup needed)
+    const needsSetup = providers.length === 0 || providers.every(p =>
+      p.apiKey === 'proxy' || p.apiKey === '' || !p.apiKey
+    );
+
+    return {
+      providers,
+      primaryModel: json.primaryModel || null,
+      fastModel: json.fastModel || null,
+      language: json.language || 'en',
+      needsSetup,
+    };
+  } catch (e) {
+    return { error: `Failed to read config.json: ${e.message}` };
+  }
+}
+
+/**
+ * Update the LLM-relevant portion of config.json.
+ * Merges into existing config — preserves fields like debug, maxContextTokens, etc.
+ *
+ * @param {object} update — { providers?, primaryModel?, fastModel?, language? }
+ * @param {string} [dir] — Yeaft data directory
+ * @returns {{ providers, primaryModel, fastModel, language } | { error: string }}
+ */
+export function updateLlmConfig(update, dir) {
+  const root = dir || process.env.YEAFT_DIR || DEFAULT_YEAFT_DIR;
+  const configPath = join(root, 'config.json');
+
+  // Read existing config (preserve non-LLM fields)
+  let existing = {};
+  if (existsSync(configPath)) {
+    try {
+      existing = JSON.parse(readFileSync(configPath, 'utf8'));
+    } catch {
+      // Start fresh if corrupt
+      existing = {};
+    }
+  }
+
+  // Validate providers structure
+  if (update.providers !== undefined) {
+    if (!Array.isArray(update.providers)) {
+      return { error: 'providers must be an array' };
+    }
+    for (const p of update.providers) {
+      if (!p.name || typeof p.name !== 'string') {
+        return { error: 'Each provider must have a name' };
+      }
+      if (!p.baseUrl || typeof p.baseUrl !== 'string') {
+        return { error: `Provider "${p.name}" must have a baseUrl` };
+      }
+      if (!Array.isArray(p.models) || p.models.length === 0) {
+        return { error: `Provider "${p.name}" must have at least one model` };
+      }
+    }
+    existing.providers = update.providers;
+  }
+
+  // Update model selections
+  if (update.primaryModel !== undefined) {
+    existing.primaryModel = update.primaryModel || null;
+  }
+  if (update.fastModel !== undefined) {
+    existing.fastModel = update.fastModel || null;
+  }
+  if (update.language !== undefined) {
+    existing.language = update.language;
+  }
+
+  // Write back
+  try {
+    writeFileSync(configPath, JSON.stringify(existing, null, 2) + '\n', 'utf8');
+  } catch (e) {
+    return { error: `Failed to write config.json: ${e.message}` };
+  }
+
+  return {
+    providers: Array.isArray(existing.providers) ? existing.providers : [],
+    primaryModel: existing.primaryModel || null,
+    fastModel: existing.fastModel || null,
+    language: existing.language || 'en',
+  };
+}