@bolloon/bolloon-agent 0.5.0 → 0.5.2

package/dist/llm/config-store.js

@@ -6,6 +6,7 @@ import * as fs from 'fs/promises';
 import * as path from 'path';
 import * as os from 'os';
 import * as fsSync from 'fs';
+import { absorbLegacyProviderEntries, ensureProviderRegistryMetadataSource, getProviderRegistryEntry, normalizeCustomProviders, setCustomProviderSnapshot } from './provider-registry.js';
 /** 2026-09-16 (M1): 不再在模块加载时固定 HOME —— 独立宿主/测试注入/长驻进程都走同一个解析 */
 function configDir() {
     const env = process.env.BOLLOON_HOME?.trim();
@@ -256,6 +257,11 @@ class LLMConfigStore {
         return next;
     }
     async initialize() {
+        // 2026-09-26 (P3): 任何一次配置读都顺手把注册表的元数据填充点接上。
+        //   为什么不放模块体: 本模块与注册表之间是循环 import, 模块体的执行时机取决于"谁是入口",
+        //   实测会撞上对方模块 `metadataSources` 的 TDZ (整个进程起不来)。配置路径是所有读模型
+        //   元数据的前置步骤, 放这里既能自动接上, 又不依赖 import 顺序 (详见 provider-registry)。
+        ensureProviderRegistryMetadataSource();
         // 目录变了 (换 HOME / 独立宿主 / 测试注入): 内存缓存作废, 重新读盘
         const dir = configDir();
         const sig = this.fileSignature();
@@ -290,19 +296,46 @@ class LLMConfigStore {
                     loadedConfig.providers[provider] = { ...DEFAULT_PROVIDER_CONFIGS[provider] };
                 }
             }
+            // 2026-09-26 (P3): 自定义供应商在**内存里**规范化 (数组形 → map / 别名 / 从 providers 吸收),
+            //   **不写盘** —— "读一下就把用户文件改了"是另一种意外。写回只发生在显式 add/update/remove。
+            const normCustom = normalizeCustomProviders(loadedConfig.customProviders);
+            const absorbed = absorbLegacyProviderEntries(loadedConfig.providers, normCustom.providers);
+            loadedConfig.customProviders = absorbed.providers;
+            if (normCustom.rejected.length) {
+                console.warn(`[config-store] customProviders 有 ${normCustom.rejected.length} 条被拒: `
+                    + normCustom.rejected.map((r) => `${r.key}(${r.reason})`).join('; '));
+            }
+            // 注册表的同步读口 (元数据填充点/长期任务门) 靠这份快照 —— 配置一读完就刷新, 不留陈旧结论
+            setCustomProviderSnapshot(absorbed.providers);
             // 确保有 activeProvider
             const activeProvider = loadedConfig.activeProvider;
-            if (!activeProvider || !DEFAULT_PROVIDER_CONFIGS[activeProvider]) {
+            if (!activeProvider || !this.isSelectableProvider(activeProvider, loadedConfig)) {
                 loadedConfig.activeProvider = 'ollama';
             }
             this.config = loadedConfig;
         }
         catch {
             this.config = getDefaultConfig();
+            setCustomProviderSnapshot({});
             await this.save();
         }
         this.initialized = true;
     }
+    /**
+     * 这个 provider id 现在**能当 activeProvider** 吗。
+     *
+     * 2026-09-26 (P3) 之前: 只认内置表 → 用户把自定义端点当默认模型时会被**静默改成 ollama**
+     * (旧配置直接失效)。现在: 内置, 或已注册的自定义供应商, 或配置里**真有这一格且写了地址/模型**
+     * (旧的"手写 provider 条目"写法) 都算数; 三者都不是才退到 ollama。
+     */
+    isSelectableProvider(id, cfg) {
+        if (DEFAULT_PROVIDER_CONFIGS[id])
+            return true;
+        if (cfg.customProviders && cfg.customProviders[id])
+            return true;
+        const own = cfg.providers[id];
+        return !!own && (String(own.baseUrl || '').trim().length > 0 || String(own.model || '').trim().length > 0);
+    }
     async save() {
         if (!this.config)
             return;
@@ -341,31 +374,87 @@ class LLMConfigStore {
     }
     async setActiveProvider(provider) {
         await this.initialize();
-        if (!this.config?.providers[provider]) {
+        const hit = this.providerRowFor(String(provider));
+        if (!hit) {
             throw new Error(`Unknown provider: ${provider}`);
         }
-        const providerConfig = this.config.providers[provider];
+        const providerConfig = hit.row;
         if (providerConfig.requiresApiKey && !providerConfig.apiKey) {
             throw new Error(`${provider} requires an API key but none is configured`);
         }
         await this.withWriteLock(async () => {
+            if (!hit.existing)
+                this.config.providers[String(provider)] = hit.row;
             this.config.activeProvider = provider;
             await this.save();
         });
     }
     async updateProvider(provider, updates) {
         await this.initialize();
-        if (!this.config?.providers[provider]) {
+        const hit = this.providerRowFor(String(provider));
+        if (!hit) {
             throw new Error(`Unknown provider: ${provider}`);
         }
         await this.withWriteLock(async () => {
-            this.config.providers[provider] = {
-                ...this.config.providers[provider],
+            this.config.providers[String(provider)] = {
+                ...hit.row,
                 ...updates
             };
             await this.save();
         });
     }
+    /**
+     * 取（必要时**算出**）一个 provider 的配置行。`existing=false` 表示这一行还不存在, 由调用方
+     * 在自己的写锁里落进 map（本方法**不改任何状态** —— 校验失败时就该什么都不发生）。
+     *
+     * 2026-09-26 (P6): 统一入口现在**接受自定义供应商 id** (从注册表校验, 不再只看内置表),
+     * 于是切换时会把 `model/baseUrl/apiKey` 写进 `providers.<自定义 id>` 这一格 —— 而这一格此前
+     * 根本不存在, 旧实现直接 `Unknown provider` 抛错。
+     *
+     * 规矩 (不编默认值):
+     *   · 内置供应商 / 配置里本来就有这一格 → 原样返回 (老行为一字不变);
+     *   · 已注册的**自定义供应商** → 用它的**声明**起手 (地址/模型), 凭据那一格留空等写入 ——
+     *     不拿内置默认去顶;
+     *   · 其余 (既不是内置, 也不在 `customProviders` 里) → `null` (调用方照旧抛 Unknown provider)。
+     */
+    providerRowFor(id) {
+        const rows = this.config?.providers;
+        if (!rows)
+            return null;
+        if (rows[id])
+            return { row: rows[id], existing: true };
+        const spec = this.config?.customProviders?.[id];
+        if (!spec) {
+            // 2026-09-27 (目录驱动): **目录里的家**也走这条 —— 新家只要在目录里就能切进来 (不改代码)。
+            //   起手行只放目录**声明过的**东西: 基址 = 目录里的 api (目录没给就是空串, 让校验层如实报
+            //   "需自定义 baseUrl"); 模型留空 (公开目录没有"默认模型"这回事, 不许抓一个凑数);
+            //   凭据那一格留空, 等写入或等环境变量 (目录只给环境变量**名**)。
+            const catEntry = getProviderRegistryEntry(id);
+            if (!catEntry || catEntry.kind !== 'catalog')
+                return null;
+            return {
+                existing: false,
+                row: {
+                    enabled: true,
+                    apiKey: '',
+                    baseUrl: catEntry.defaultBaseUrl,
+                    model: '',
+                    requiresApiKey: catEntry.requiresApiKey !== false,
+                },
+            };
+        }
+        const entry = getProviderRegistryEntry(id);
+        return {
+            existing: false,
+            row: {
+                enabled: true,
+                apiKey: '',
+                baseUrl: spec.baseUrl,
+                model: String(spec.model || spec.models?.[0] || ''),
+                requiresApiKey: entry ? entry.requiresApiKey : true,
+            },
+        };
+    }
     async testProvider(provider) {
         await this.initialize();
         const config = this.config?.providers[provider];
@@ -494,6 +583,57 @@ class LLMConfigStore {
     getAllProviderInfo() {
         return PROVIDER_INFO;
     }
+    /**
+     * 2026-09-26: 丢弃内存缓存, 下一次 initialize() 强制重新读盘。
+     * 用途: 统一入口在做"读出→改写→写回"时, 必须在拿到跨进程锁之后重读,
+     * 否则两个进程各自拿旧内存值写回, 后写的会把先写的那份改动覆盖掉。
+     * 回滚路径同样需要它 (把文件还原后, 内存里的那一份也得跟着回来)。
+     */
+    invalidate() {
+        this.initialized = false;
+        this.config = null;
+        this.loadedDir = null;
+        this.loadedSig = null;
+    }
+    /** 配置文件绝对路径 (统一入口/锁/迁移都从这里取, 不再各自拼) */
+    configFilePath() {
+        return CONFIG_PATH;
+    }
+    /** 旧文件名 (迁移输入) */
+    legacyConfigFilePath() {
+        return LEGACY_CONFIG_PATH;
+    }
+    /**
+     * 现在在册的自定义供应商 (规范化后的内存视图; 读盘由 `initialize()` 保证已经发生过)。
+     * 凭据原样在里面 —— 调用方展示前必须走脱敏 (`custom-provider-store.redactCustomProviders`)。
+     */
+    async getCustomProviders() {
+        await this.initialize();
+        return { ...(this.config?.customProviders || {}) };
+    }
+    /**
+     * 覆盖 `customProviders` 一格 (唯一写口)。
+     *
+     * 只动这一格 —— 内置 `providers` / `activeProvider` 一个字节都不碰; 于是"加个自定义供应商"
+     * 不会顺手把用户在别处刚改的模型选择覆盖掉。写前调用方负责 `invalidate()` + `initialize()`
+     * (拿到跨进程锁之后重读), 否则两个进程各自拿旧内存值写回。
+     */
+    async setCustomProviders(map) {
+        await this.initialize();
+        const norm = normalizeCustomProviders(map && typeof map === 'object' ? map : {});
+        if (norm.rejected.length) {
+            throw new Error(`customProviders 规范化失败: ${norm.rejected.map((r) => `${r.key}(${r.reason})`).join('; ')}`);
+        }
+        await this.withWriteLock(async () => {
+            this.config.customProviders = norm.providers;
+            setCustomProviderSnapshot(norm.providers);
+            await this.save();
+        });
+    }
+    /** 配置目录 (锁文件、session 绑定都落在这里) */
+    configDirPath() {
+        return configDir();
+    }
 }
 export const llmConfigStore = new LLMConfigStore();
 export function getPiSDKConfig() {