shennian 0.2.89 → 0.2.90

package/dist/src/agents/pi-context.js

@@ -1,217 +1,12 @@
-// @arch docs/architecture/cli/agent-adapters.md#pi-agent-上下文管理
-// @test src/__tests__/pi-context.test.ts
-import fs from 'node:fs';
-import path from 'node:path';
-import { resolveShennianPath } from '../config/index.js';
-export const PI_DEFAULT_MODEL_ID = 'qwen3.6-plus';
-export function buildShellCommandSpec(command, platform = process.platform) {
-    if (platform === 'win32') {
-        return {
-            file: 'powershell.exe',
-            args: ['-NoLogo', '-NoProfile', '-NonInteractive', '-ExecutionPolicy', 'Bypass', '-Command', command],
-            shell: 'powershell',
-        };
-    }
-    return {
-        file: 'bash',
-        args: ['-c', command],
-        shell: 'bash',
-    };
-}
-export function createPiModel(modelId = PI_DEFAULT_MODEL_ID, overrides = {}) {
-    return {
-        id: modelId,
-        name: modelId,
-        api: 'openai-completions',
-        provider: overrides.provider ?? 'shennian-proxy',
-        baseUrl: overrides.baseUrl ?? '',
-        reasoning: false,
-        input: ['text', 'image'],
-        cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
-        contextWindow: 128000,
-        maxTokens: 8192,
-        ...(overrides.compat ? { compat: overrides.compat } : {}),
-    };
-}
-export const SYSTEM_PROMPT = `你是神念内置编程助手，运行在用户本地机器上。
-你可以读写文件、执行 shell 命令、帮助用户完成编程和系统管理任务。
-工作目录已设置，操作文件时使用相对路径或绝对路径均可。
-当前 shell 会随操作系统选择：Windows 使用 PowerShell，macOS/Linux 使用 bash。Windows 下需要真实 curl 时使用 curl.exe，避免 PowerShell 的 curl 别名。
-保持回复简洁、准确，中文回复。`;
-const AGENTS_MD_MAX_CHARS = 24_000;
-export function findAgentsMd(startDir) {
-    let current = path.resolve(startDir || process.cwd());
-    try {
-        const stat = fs.existsSync(current) ? fs.statSync(current) : null;
-        if (stat && !stat.isDirectory())
-            current = path.dirname(current);
-    }
-    catch {
-        // Fall back to the resolved path.
-    }
-    while (true) {
-        const candidate = path.join(current, 'AGENTS.md');
-        if (fs.existsSync(candidate))
-            return candidate;
-        const parent = path.dirname(current);
-        if (parent === current)
-            return null;
-        current = parent;
-    }
-}
-export function loadAgentsMdInstructions(workDir) {
-    const filePath = findAgentsMd(workDir);
-    if (!filePath)
-        return null;
-    try {
-        const content = fs.readFileSync(filePath, 'utf-8').trim();
-        if (!content)
-            return null;
-        const truncated = content.length > AGENTS_MD_MAX_CHARS
-            ? `${content.slice(0, AGENTS_MD_MAX_CHARS)}\n\n...（AGENTS.md 已截断）`
-            : content;
-        return `# AGENTS.md instructions for ${path.dirname(filePath)}\n\n${truncated}`;
-    }
-    catch {
-        return null;
-    }
-}
-// ── Context compression ──────────────────────────────────────────────────────
-export const CONTEXT_TOKEN_THRESHOLD = 90_000;
-export const KEEP_RECENT_MESSAGES = 6;
-const SESSIONS_DIR = resolveShennianPath('sessions');
-export const SUMMARY_FILENAME = 'summary.json';
-export const SNAPSHOT_FILENAME = 'snapshot.json';
-export const MESSAGES_FILENAME = 'messages.jsonl';
-export const LEGACY_SUMMARY_FILENAME = 'pi-context.json';
-export function estimateTokens(messages) {
-    let chars = 0;
-    for (const msg of messages)
-        chars += JSON.stringify(msg).length;
-    return Math.ceil(chars / 3);
-}
-export function messagesToText(messages) {
-    const lines = [];
-    for (const msg of messages) {
-        const m = msg;
-        const role = String(m.role ?? 'unknown');
-        if (!Array.isArray(m.content))
-            continue;
-        for (const c of m.content) {
-            if (c.type === 'text' && typeof c.text === 'string') {
-                lines.push(`[${role}] ${c.text.slice(0, 500)}`);
-            }
-            else if (c.type === 'toolCall') {
-                const name = typeof c.name === 'string' ? c.name : typeof c.toolName === 'string' ? c.toolName : 'tool';
-                const args = c.arguments ?? c.args ?? {};
-                lines.push(`[${role}] tool:${name}(${JSON.stringify(args).slice(0, 200)})`);
-            }
-        }
-    }
-    return lines.join('\n');
-}
-export function cloneMessages(messages) {
-    return JSON.parse(JSON.stringify(messages));
-}
-export function getSessionDir(sessionId) {
-    return path.join(SESSIONS_DIR, sessionId);
-}
-function getMessageContentText(content) {
-    if (!content || typeof content !== 'object')
-        return null;
-    const chunk = content;
-    if (chunk.type === 'text' && typeof chunk.text === 'string')
-        return chunk.text;
-    if (chunk.type === 'thinking' && typeof chunk.thinking === 'string')
-        return `[thinking] ${chunk.thinking}`;
-    if (chunk.type === 'toolCall') {
-        const name = typeof chunk.name === 'string'
-            ? chunk.name
-            : typeof chunk.toolName === 'string'
-                ? chunk.toolName
-                : 'tool';
-        const args = chunk.arguments ?? chunk.args ?? {};
-        return `[toolCall] ${name}(${JSON.stringify(args).slice(0, 300)})`;
-    }
-    if (chunk.type === 'image' && typeof chunk.mimeType === 'string') {
-        return `[image] ${chunk.mimeType}`;
-    }
-    return null;
-}
-export function buildRollingSummary(cachedSummary, messages) {
-    const recentLines = [];
-    for (const msg of messages) {
-        const role = String(msg.role ?? 'unknown');
-        const content = Array.isArray(msg.content)
-            ? msg.content
-            : [];
-        for (const chunk of content) {
-            const text = getMessageContentText(chunk);
-            if (text)
-                recentLines.push(`[${role}] ${text}`);
-        }
-    }
-    const recent = recentLines.slice(-8).join('\n').trim();
-    const summary = recent || cachedSummary?.trim() || null;
-    if (!summary)
-        return null;
-    return summary.length > 2000 ? summary.slice(summary.length - 2000) : summary;
-}
-export function longestCommonPrefixLength(left, right) {
-    const max = Math.min(left.length, right.length);
-    for (let i = 0; i < max; i++) {
-        if (JSON.stringify(left[i]) !== JSON.stringify(right[i]))
-            return i;
-    }
-    return max;
-}
-export async function requestProxySummary(proxyUrl, authToken, prompt) {
-    const response = await fetch(`${proxyUrl}/api/stream`, {
-        method: 'POST',
-        headers: {
-            'Content-Type': 'application/json',
-            Authorization: `Bearer ${authToken}`,
-        },
-        body: JSON.stringify({
-            model: createPiModel('qwen-plus'),
-            context: {
-                messages: [
-                    {
-                        role: 'user',
-                        content: [{ type: 'text', text: prompt }],
-                    },
-                ],
-            },
-            options: {
-                temperature: 0.3,
-                maxTokens: 500,
-            },
-        }),
-    });
-    if (!response.ok || !response.body)
-        return null;
-    const reader = response.body.getReader();
-    const decoder = new TextDecoder();
-    const parts = [];
-    let buffer = '';
-    while (true) {
-        const { done, value } = await reader.read();
-        if (done)
-            break;
-        buffer += decoder.decode(value, { stream: true });
-        const lines = buffer.split('\n');
-        buffer = lines.pop() ?? '';
-        for (const line of lines) {
-            if (!line.startsWith('data: '))
-                continue;
-            const raw = line.slice(6).trim();
-            if (!raw)
-                continue;
-            const event = JSON.parse(raw);
-            if (event.type === 'text_delta' && typeof event.delta === 'string') {
-                parts.push(event.delta);
-            }
-        }
-    }
-    return parts.join('').trim() || null;
-}
+import u from"node:fs";import l from"node:path";import{resolveShennianPath as x}from"../config/index.js";const S="qwen3.6-plus";function w(e,t=process.platform){return t==="win32"?{file:"powershell.exe",args:["-NoLogo","-NoProfile","-NonInteractive","-ExecutionPolicy","Bypass","-Command",e],shell:"powershell"}:{file:"bash",args:["-c",e],shell:"bash"}}function d(e=S,t={}){return{id:e,name:e,api:"openai-completions",provider:t.provider??"shennian-proxy",baseUrl:t.baseUrl??"",reasoning:!1,input:["text","image"],cost:{input:0,output:0,cacheRead:0,cacheWrite:0},contextWindow:128e3,maxTokens:8192,...t.compat?{compat:t.compat}:{}}}const O=`\u4F60\u662F\u795E\u5FF5\u5185\u7F6E\u7F16\u7A0B\u52A9\u624B\uFF0C\u8FD0\u884C\u5728\u7528\u6237\u672C\u5730\u673A\u5668\u4E0A\u3002
+\u4F60\u53EF\u4EE5\u8BFB\u5199\u6587\u4EF6\u3001\u6267\u884C shell \u547D\u4EE4\u3001\u5E2E\u52A9\u7528\u6237\u5B8C\u6210\u7F16\u7A0B\u548C\u7CFB\u7EDF\u7BA1\u7406\u4EFB\u52A1\u3002
+\u5DE5\u4F5C\u76EE\u5F55\u5DF2\u8BBE\u7F6E\uFF0C\u64CD\u4F5C\u6587\u4EF6\u65F6\u4F7F\u7528\u76F8\u5BF9\u8DEF\u5F84\u6216\u7EDD\u5BF9\u8DEF\u5F84\u5747\u53EF\u3002
+\u5F53\u524D shell \u4F1A\u968F\u64CD\u4F5C\u7CFB\u7EDF\u9009\u62E9\uFF1AWindows \u4F7F\u7528 PowerShell\uFF0CmacOS/Linux \u4F7F\u7528 bash\u3002Windows \u4E0B\u9700\u8981\u771F\u5B9E curl \u65F6\u4F7F\u7528 curl.exe\uFF0C\u907F\u514D PowerShell \u7684 curl \u522B\u540D\u3002
+\u4FDD\u6301\u56DE\u590D\u7B80\u6D01\u3001\u51C6\u786E\uFF0C\u4E2D\u6587\u56DE\u590D\u3002`,h=24e3;function E(e){let t=l.resolve(e||process.cwd());try{const n=u.existsSync(t)?u.statSync(t):null;n&&!n.isDirectory()&&(t=l.dirname(t))}catch{}for(;;){const n=l.join(t,"AGENTS.md");if(u.existsSync(n))return n;const o=l.dirname(t);if(o===t)return null;t=o}}function $(e){const t=E(e);if(!t)return null;try{const n=u.readFileSync(t,"utf-8").trim();if(!n)return null;const o=n.length>h?`${n.slice(0,h)}
+
+...\uFF08AGENTS.md \u5DF2\u622A\u65AD\uFF09`:n;return`# AGENTS.md instructions for ${l.dirname(t)}
+
+${o}`}catch{return null}}const P=9e4,b=6,N=x("sessions"),k="summary.json",C="snapshot.json",L="messages.jsonl",j="pi-context.json";function R(e){let t=0;for(const n of e)t+=JSON.stringify(n).length;return Math.ceil(t/3)}function D(e){const t=[];for(const n of e){const o=n,s=String(o.role??"unknown");if(Array.isArray(o.content)){for(const r of o.content)if(r.type==="text"&&typeof r.text=="string")t.push(`[${s}] ${r.text.slice(0,500)}`);else if(r.type==="toolCall"){const c=typeof r.name=="string"?r.name:typeof r.toolName=="string"?r.toolName:"tool",i=r.arguments??r.args??{};t.push(`[${s}] tool:${c}(${JSON.stringify(i).slice(0,200)})`)}}}return t.join(`
+`)}function I(e){return JSON.parse(JSON.stringify(e))}function J(e){return l.join(N,e)}function A(e){if(!e||typeof e!="object")return null;const t=e;if(t.type==="text"&&typeof t.text=="string")return t.text;if(t.type==="thinking"&&typeof t.thinking=="string")return`[thinking] ${t.thinking}`;if(t.type==="toolCall"){const n=typeof t.name=="string"?t.name:typeof t.toolName=="string"?t.toolName:"tool",o=t.arguments??t.args??{};return`[toolCall] ${n}(${JSON.stringify(o).slice(0,300)})`}return t.type==="image"&&typeof t.mimeType=="string"?`[image] ${t.mimeType}`:null}function G(e,t){const n=[];for(const r of t){const c=String(r.role??"unknown"),i=Array.isArray(r.content)?r.content:[];for(const p of i){const a=A(p);a&&n.push(`[${c}] ${a}`)}}const s=n.slice(-8).join(`
+`).trim()||e?.trim()||null;return s?s.length>2e3?s.slice(s.length-2e3):s:null}function F(e,t){const n=Math.min(e.length,t.length);for(let o=0;o<n;o++)if(JSON.stringify(e[o])!==JSON.stringify(t[o]))return o;return n}async function U(e,t,n){const o=await fetch(`${e}/api/stream`,{method:"POST",headers:{"Content-Type":"application/json",Authorization:`Bearer ${t}`},body:JSON.stringify({model:d("qwen-plus"),context:{messages:[{role:"user",content:[{type:"text",text:n}]}]},options:{temperature:.3,maxTokens:500}})});if(!o.ok||!o.body)return null;const s=o.body.getReader(),r=new TextDecoder,c=[];let i="";for(;;){const{done:p,value:a}=await s.read();if(p)break;i+=r.decode(a,{stream:!0});const m=i.split(`
+`);i=m.pop()??"";for(const g of m){if(!g.startsWith("data: "))continue;const y=g.slice(6).trim();if(!y)continue;const f=JSON.parse(y);f.type==="text_delta"&&typeof f.delta=="string"&&c.push(f.delta)}}return c.join("").trim()||null}export{P as CONTEXT_TOKEN_THRESHOLD,b as KEEP_RECENT_MESSAGES,j as LEGACY_SUMMARY_FILENAME,L as MESSAGES_FILENAME,S as PI_DEFAULT_MODEL_ID,C as SNAPSHOT_FILENAME,k as SUMMARY_FILENAME,O as SYSTEM_PROMPT,G as buildRollingSummary,w as buildShellCommandSpec,I as cloneMessages,d as createPiModel,R as estimateTokens,E as findAgentsMd,J as getSessionDir,$ as loadAgentsMdInstructions,F as longestCommonPrefixLength,D as messagesToText,U as requestProxySummary};