shennian 0.2.88 → 0.2.90

package/dist/src/session/handlers/agent-config.js

@@ -1,150 +1 @@
-// @arch docs/features/agent-provider-config.md
-// @test src/__tests__/agent-config-status.test.ts
-// @test src/__tests__/agent-config-handler.test.ts
-import { mkdir } from 'node:fs/promises';
-import { createAgent } from '../../agents/adapter.js';
-import { buildManagedAgentEnv, deleteManagedAgentProviderConfig, getAgentConfigSummary, upsertManagedAgentProviderConfig, } from '../../agents/config-status.js';
-import { resolveShennianPath } from '../../config/index.js';
-import { handleAgentsRefresh } from './agents.js';
-const AGENT_TEST_TIMEOUT_MS = 30_000;
-const AGENT_TEST_PROMPT = '这是一次神念连接测试。请只回复 OK，不要解释。';
-const AGENT_TEST_WORK_DIR = resolveShennianPath('tmp', 'agent-config-test');
-function normalizeAgent(value) {
-    if (value === 'codex' || value === 'claude' || value === 'pi')
-        return value;
-    throw new Error('Only Codex, Claude Code, and Nian provider config are supported');
-}
-export async function handleAgentConfigGet(runtime, req) {
-    const agent = normalizeAgent(req.params.agent);
-    runtime.client.sendRes({
-        type: 'res',
-        id: req.id,
-        ok: true,
-        payload: { agent, config: getAgentConfigSummary(agent) },
-    });
-}
-export async function handleAgentConfigUpsert(runtime, req) {
-    const agent = normalizeAgent(req.params.agent);
-    const baseUrl = typeof req.params.baseUrl === 'string' ? req.params.baseUrl.trim() : undefined;
-    const token = typeof req.params.token === 'string' ? req.params.token.trim() : undefined;
-    if (!baseUrl && !token)
-        throw new Error('Base URL or token is required');
-    upsertManagedAgentProviderConfig({ agent, baseUrl, token });
-    runtime.client.sendRes({
-        type: 'res',
-        id: req.id,
-        ok: true,
-        payload: { agent, config: getAgentConfigSummary(agent) },
-    });
-    await broadcastAgents(runtime);
-}
-export async function handleAgentConfigClear(runtime, req) {
-    const agent = normalizeAgent(req.params.agent);
-    deleteManagedAgentProviderConfig(agent);
-    runtime.client.sendRes({
-        type: 'res',
-        id: req.id,
-        ok: true,
-        payload: { agent, config: getAgentConfigSummary(agent) },
-    });
-    await broadcastAgents(runtime);
-}
-export async function handleAgentConfigTest(runtime, req) {
-    const agent = normalizeAgent(req.params.agent);
-    const summary = getAgentConfigSummary(agent);
-    try {
-        const result = await runAgentConfigTest(agent);
-        runtime.client.sendRes({
-            type: 'res',
-            id: req.id,
-            ok: true,
-            payload: { agent, config: getAgentConfigSummary(agent), test: result },
-        });
-    }
-    catch (err) {
-        runtime.client.sendRes({
-            type: 'res',
-            id: req.id,
-            ok: false,
-            payload: { agent, config: summary },
-            error: err instanceof Error ? err.message : String(err),
-        });
-    }
-}
-async function runAgentConfigTest(agent) {
-    const adapter = createAgent(agent);
-    if (!adapter)
-        throw new Error(`Unsupported agent: ${agent}`);
-    const sessionId = `agent-test-${agent}-${Date.now()}`;
-    await mkdir(AGENT_TEST_WORK_DIR, { recursive: true });
-    let settled = false;
-    let failWait = () => { };
-    const waitForReply = new Promise((resolve, reject) => {
-        const timeout = setTimeout(() => {
-            if (settled)
-                return;
-            settled = true;
-            reject(new Error('Agent test timed out after 30 seconds'));
-        }, AGENT_TEST_TIMEOUT_MS);
-        function finish(result) {
-            if (settled)
-                return;
-            settled = true;
-            clearTimeout(timeout);
-            resolve(result);
-        }
-        function fail(error) {
-            if (settled)
-                return;
-            settled = true;
-            clearTimeout(timeout);
-            reject(error);
-        }
-        failWait = fail;
-        adapter.on('agentEvent', (event) => {
-            if (event.state === 'delta' && event.text && !event.thinking) {
-                finish({ mode: 'agent-run', reply: event.text.slice(0, 120) });
-                return;
-            }
-            if (event.state === 'final') {
-                finish({ mode: 'agent-run' });
-                return;
-            }
-            if (event.state === 'error') {
-                fail(new Error(event.message || 'Agent test failed'));
-            }
-        });
-        adapter.on('error', fail);
-    });
-    try {
-        adapter.configure?.({
-            sessionId,
-            env: getManagedEnvForAgent(agent),
-        });
-        await adapter.start(sessionId, AGENT_TEST_WORK_DIR, null);
-        const sendPromise = adapter.send(AGENT_TEST_PROMPT).catch((err) => {
-            failWait(err instanceof Error ? err : new Error(String(err)));
-        });
-        return await Promise.race([waitForReply, sendPromise.then(() => waitForReply)]);
-    }
-    finally {
-        adapter.removeAllListeners();
-        await adapter.stop().catch(() => { });
-    }
-}
-async function broadcastAgents(runtime) {
-    const req = {
-        type: 'req',
-        id: `agent-config-refresh-${Date.now()}`,
-        method: 'agents.refresh',
-        params: {},
-    };
-    await handleAgentsRefresh(runtime, req);
-}
-export function getManagedEnvForAgent(agentType) {
-    // Re-export point kept here for session handlers to avoid reaching into storage internals.
-    // The implementation lives in agents/config-status.ts.
-    return agentType === 'codex' || agentType === 'claude' || agentType === 'pi'
-        ? buildManagedAgentEnv(agentType)
-        : {};
-}
+import{mkdir as y}from"node:fs/promises";import{createAgent as w}from"../../agents/adapter.js";import{buildManagedAgentEnv as h,deleteManagedAgentProviderConfig as A,getAgentConfigSummary as s,upsertManagedAgentProviderConfig as E}from"../../agents/config-status.js";import{resolveShennianPath as T}from"../../config/index.js";import{handleAgentsRefresh as C}from"./agents.js";const k=3e4,R="\u8FD9\u662F\u4E00\u6B21\u795E\u5FF5\u8FDE\u63A5\u6D4B\u8BD5\u3002\u8BF7\u53EA\u56DE\u590D OK\uFF0C\u4E0D\u8981\u89E3\u91CA\u3002",p=T("tmp","agent-config-test");function c(e){if(e==="codex"||e==="claude"||e==="pi")return e;throw new Error("Only Codex, Claude Code, and Nian provider config are supported")}async function O(e,t){const n=c(t.params.agent);e.client.sendRes({type:"res",id:t.id,ok:!0,payload:{agent:n,config:s(n)}})}async function G(e,t){const n=c(t.params.agent),r=typeof t.params.baseUrl=="string"?t.params.baseUrl.trim():void 0,o=typeof t.params.token=="string"?t.params.token.trim():void 0;if(!r&&!o)throw new Error("Base URL or token is required");E({agent:n,baseUrl:r,token:o}),e.client.sendRes({type:"res",id:t.id,ok:!0,payload:{agent:n,config:s(n)}}),await m(e)}async function N(e,t){const n=c(t.params.agent);A(n),e.client.sendRes({type:"res",id:t.id,ok:!0,payload:{agent:n,config:s(n)}}),await m(e)}async function $(e,t){const n=c(t.params.agent),r=s(n);try{const o=await x(n);e.client.sendRes({type:"res",id:t.id,ok:!0,payload:{agent:n,config:s(n),test:o}})}catch(o){e.client.sendRes({type:"res",id:t.id,ok:!1,payload:{agent:n,config:r},error:o instanceof Error?o.message:String(o)})}}async function x(e){const t=w(e);if(!t)throw new Error(`Unsupported agent: ${e}`);const n=`agent-test-${e}-${Date.now()}`;await y(p,{recursive:!0});let r=!1,o=()=>{};const g=new Promise((d,i)=>{const l=setTimeout(()=>{r||(r=!0,i(new Error("Agent test timed out after 30 seconds")))},k);function u(a){r||(r=!0,clearTimeout(l),d(a))}function f(a){r||(r=!0,clearTimeout(l),i(a))}o=f,t.on("agentEvent",a=>{if(a.state==="delta"&&a.text&&!a.thinking){u({mode:"agent-run",reply:a.text.slice(0,120)});return}if(a.state==="final"){u({mode:"agent-run"});return}a.state==="error"&&f(new Error(a.message||"Agent test failed"))}),t.on("error",f)});try{t.configure?.({sessionId:n,env:_(e)}),await t.start(n,p,null);const d=t.send(R).catch(i=>{o(i instanceof Error?i:new Error(String(i)))});return await Promise.race([g,d.then(()=>g)])}finally{t.removeAllListeners(),await t.stop().catch(()=>{})}}async function m(e){const t={type:"req",id:`agent-config-refresh-${Date.now()}`,method:"agents.refresh",params:{}};await C(e,t)}function _(e){return e==="codex"||e==="claude"||e==="pi"?h(e):{}}export{_ as getManagedEnvForAgent,N as handleAgentConfigClear,O as handleAgentConfigGet,$ as handleAgentConfigTest,G as handleAgentConfigUpsert};

package/dist/src/session/handlers/agents.js

@@ -1,55 +1 @@
-// @arch docs/architecture/cli/daemon.md#会话管理
-// @test src/__tests__/model-switching.test.ts
-import { SERVERS } from '../../region.js';
-import { loadConfig } from '../../config/index.js';
-import { detectAgents } from '../../agents/detect.js';
-import { refreshAgentInfos } from '../../agents/model-registry.js';
-import { resetCommandSpecCache } from '../../agents/command-spec.js';
-async function refreshMachineAgents(runtime, req) {
-    const config = loadConfig();
-    const serverUrl = config.serverUrl ?? SERVERS.cn.url;
-    const detectedAgents = detectAgents();
-    const agentList = detectedAgents.map((agent) => agent.type);
-    if (detectedAgents.length === 0) {
-        runtime.client.sendRes({
-            type: 'res',
-            id: req.id,
-            ok: true,
-            payload: { agentList: [], agents: [] },
-        });
-        return;
-    }
-    try {
-        const agents = await refreshAgentInfos(detectedAgents, {
-            serverUrl,
-            authToken: config.machineToken ?? config.accessToken,
-        });
-        runtime.client.sendRes({
-            type: 'res',
-            id: req.id,
-            ok: true,
-            payload: { agentList, agents },
-        });
-        runtime.client.sendEvent({
-            type: 'event',
-            event: 'machine.agents',
-            payload: { agentList, agents },
-        });
-    }
-    catch (err) {
-        runtime.client.sendRes({
-            type: 'res',
-            id: req.id,
-            ok: false,
-            error: err instanceof Error ? err.message : String(err),
-        });
-    }
-}
-export async function handleModelsRefresh(runtime, req) {
-    await refreshMachineAgents(runtime, req);
-}
-export async function handleAgentsRefresh(runtime, req) {
-    resetCommandSpecCache();
-    runtime.reloadCustomAgents();
-    await refreshMachineAgents(runtime, req);
-}
+import{SERVERS as i}from"../../region.js";import{loadConfig as d}from"../../config/index.js";import{detectAgents as f}from"../../agents/detect.js";import{refreshAgentInfos as l}from"../../agents/model-registry.js";import{resetCommandSpecCache as p}from"../../agents/command-spec.js";async function a(e,n){const s=d(),c=s.serverUrl??i.cn.url,o=f(),r=o.map(t=>t.type);if(o.length===0){e.client.sendRes({type:"res",id:n.id,ok:!0,payload:{agentList:[],agents:[]}});return}try{const t=await l(o,{serverUrl:c,authToken:s.machineToken??s.accessToken});e.client.sendRes({type:"res",id:n.id,ok:!0,payload:{agentList:r,agents:t}}),e.client.sendEvent({type:"event",event:"machine.agents",payload:{agentList:r,agents:t}})}catch(t){e.client.sendRes({type:"res",id:n.id,ok:!1,error:t instanceof Error?t.message:String(t)})}}async function k(e,n){await a(e,n)}async function A(e,n){p(),e.reloadCustomAgents(),await a(e,n)}export{A as handleAgentsRefresh,k as handleModelsRefresh};