shennian 0.2.89 → 0.2.90

package/dist/src/agents/pi.js

@@ -1,723 +1,14 @@
-// @arch docs/architecture/cli/agent-adapters.md#pi-agent-上下文管理
-// @test src/__tests__/pi-context.test.ts
-import { randomUUID } from 'node:crypto';
-import fs from 'node:fs';
-import path from 'node:path';
-import { execFile } from 'node:child_process';
-import { promisify } from 'node:util';
-import { Agent, streamProxy } from '@mariozechner/pi-agent-core';
-import { streamOpenAICompletions } from '@mariozechner/pi-ai/openai-completions';
-import { Type } from '@sinclair/typebox';
-import { AgentAdapter, registerAgent } from './adapter.js';
-import { buildExternalChannelInstructions } from './external-channel-instructions.js';
-import { loadConfig } from '../config/index.js';
-import { getManagedAgentProviderConfig } from './config-status.js';
-import { SERVERS } from '../region.js';
-import { buildRollingSummary, buildShellCommandSpec, cloneMessages, CONTEXT_TOKEN_THRESHOLD, createPiModel, estimateTokens, getSessionDir, KEEP_RECENT_MESSAGES, LEGACY_SUMMARY_FILENAME, loadAgentsMdInstructions, longestCommonPrefixLength, MESSAGES_FILENAME, messagesToText, PI_DEFAULT_MODEL_ID, requestProxySummary, SNAPSHOT_FILENAME, SYSTEM_PROMPT, SUMMARY_FILENAME, } from './pi-context.js';
-export { buildShellCommandSpec } from './pi-context.js';
-const execFileAsync = promisify(execFile);
-const DASHSCOPE_COMPATIBLE_BASE_URL = 'https://dashscope.aliyuncs.com/compatible-mode/v1';
-// ── Local tools ───────────────────────────────────────────────────────────────
-function resolvePiToolPath(workDir, filePath) {
-    return path.resolve(workDir, filePath);
-}
-async function executePiShellCommand(workDir, command, extraEnv, signal) {
-    const spec = buildShellCommandSpec(command);
-    const { stdout, stderr } = await execFileAsync(spec.file, spec.args, {
-        cwd: workDir,
-        env: { ...process.env, ...extraEnv },
-        timeout: 30_000,
-        signal,
-        maxBuffer: 1024 * 1024,
-        windowsHide: true,
-    });
-    return { stdout, stderr, shell: spec.shell };
-}
-// eslint-disable-next-line @typescript-eslint/no-explicit-any
-function makeTools(workDir, extraEnv = {}) {
-    return [
-        {
-            name: 'read_file',
-            label: '读取文件',
-            description: 'Read the full content of a file at the given path.',
-            parameters: Type.Object({
-                path: Type.String({ description: 'Absolute or relative file path' }),
-            }),
-            async execute(_id, { path: filePath }) {
-                let resolved = '';
-                try {
-                    resolved = resolvePiToolPath(workDir, filePath);
-                    const content = fs.readFileSync(resolved, 'utf-8');
-                    return {
-                        content: [{ type: 'text', text: content }],
-                        details: { path: resolved },
-                    };
-                }
-                catch (err) {
-                    return {
-                        content: [
-                            {
-                                type: 'text',
-                                text: `Error: ${err instanceof Error ? err.message : String(err)}`,
-                            },
-                        ],
-                        details: { path: resolved },
-                    };
-                }
-            },
-        },
-        {
-            name: 'write_file',
-            label: '写入文件',
-            description: 'Write content to a file, creating it or overwriting if it exists.',
-            parameters: Type.Object({
-                path: Type.String({ description: 'Absolute or relative file path' }),
-                content: Type.String({ description: 'File content to write' }),
-            }),
-            async execute(_id, { path: filePath, content }) {
-                let resolved = '';
-                try {
-                    resolved = resolvePiToolPath(workDir, filePath);
-                    fs.mkdirSync(path.dirname(resolved), { recursive: true });
-                    fs.writeFileSync(resolved, content, 'utf-8');
-                    return {
-                        content: [
-                            { type: 'text', text: `Written ${content.length} bytes to ${resolved}` },
-                        ],
-                        details: { path: resolved },
-                    };
-                }
-                catch (err) {
-                    return {
-                        content: [
-                            {
-                                type: 'text',
-                                text: `Error: ${err instanceof Error ? err.message : String(err)}`,
-                            },
-                        ],
-                        details: { path: resolved },
-                    };
-                }
-            },
-        },
-        {
-            name: 'list_directory',
-            label: '列出目录',
-            description: 'List files and directories at the given path.',
-            parameters: Type.Object({
-                path: Type.String({ description: 'Directory path to list', default: '.' }),
-            }),
-            async execute(_id, { path: dirPath }) {
-                let resolved = '';
-                try {
-                    resolved = resolvePiToolPath(workDir, dirPath);
-                    const entries = fs.readdirSync(resolved, { withFileTypes: true });
-                    const lines = entries.map((e) => `${e.isDirectory() ? 'd' : 'f'} ${e.name}`).join('\n');
-                    return {
-                        content: [{ type: 'text', text: lines || '(empty)' }],
-                        details: { path: resolved },
-                    };
-                }
-                catch (err) {
-                    return {
-                        content: [
-                            {
-                                type: 'text',
-                                text: `Error: ${err instanceof Error ? err.message : String(err)}`,
-                            },
-                        ],
-                        details: { path: resolved },
-                    };
-                }
-            },
-        },
-        {
-            name: 'bash',
-            label: '执行命令',
-            description: 'Execute a shell command in the working directory. Uses PowerShell on Windows and bash on macOS/Linux. Timeout: 30s.',
-            parameters: Type.Object({
-                command: Type.String({ description: 'Shell command to execute' }),
-            }),
-            async execute(_id, { command }, signal) {
-                try {
-                    const { stdout, stderr, shell } = await executePiShellCommand(workDir, command, extraEnv, signal);
-                    const output = [stdout, stderr].filter(Boolean).join('\n---stderr---\n');
-                    return {
-                        content: [{ type: 'text', text: output || '(no output)' }],
-                        details: { command, shell },
-                    };
-                }
-                catch (err) {
-                    const msg = err instanceof Error ? err.message : String(err);
-                    return {
-                        content: [{ type: 'text', text: `Error: ${msg}` }],
-                        details: { command, shell: buildShellCommandSpec(command).shell },
-                    };
-                }
-            },
-        },
-        {
-            name: 'read_image',
-            label: '读取图片',
-            description: 'Read an image file and return its content for visual analysis. Supports png, jpg, jpeg, gif, webp. Max 10 MB.',
-            parameters: Type.Object({
-                path: Type.String({ description: 'Absolute or relative path to the image file' }),
-            }),
-            async execute(_id, { path: filePath }) {
-                let resolved;
-                try {
-                    resolved = resolvePiToolPath(workDir, filePath);
-                }
-                catch (err) {
-                    return {
-                        content: [
-                            {
-                                type: 'text',
-                                text: `Error: ${err instanceof Error ? err.message : String(err)}`,
-                            },
-                        ],
-                        details: { path: filePath },
-                    };
-                }
-                const ext = path.extname(resolved).toLowerCase().slice(1);
-                const mimeMap = {
-                    png: 'image/png',
-                    jpg: 'image/jpeg',
-                    jpeg: 'image/jpeg',
-                    gif: 'image/gif',
-                    webp: 'image/webp',
-                };
-                const mimeType = mimeMap[ext];
-                if (!mimeType) {
-                    return {
-                        content: [
-                            {
-                                type: 'text',
-                                text: `Error: unsupported image type ".${ext}". Supported: png, jpg, jpeg, gif, webp`,
-                            },
-                        ],
-                        details: { path: resolved },
-                    };
-                }
-                try {
-                    const stat = fs.statSync(resolved);
-                    if (stat.size > 10 * 1024 * 1024) {
-                        return {
-                            content: [
-                                {
-                                    type: 'text',
-                                    text: `Error: image too large (${(stat.size / 1024 / 1024).toFixed(1)} MB, max 10 MB)`,
-                                },
-                            ],
-                            details: { path: resolved },
-                        };
-                    }
-                    const data = fs.readFileSync(resolved).toString('base64');
-                    const imageContent = { type: 'image', data, mimeType };
-                    return { content: [imageContent], details: { path: resolved, mimeType, size: stat.size } };
-                }
-                catch (err) {
-                    return {
-                        content: [
-                            {
-                                type: 'text',
-                                text: `Error: ${err instanceof Error ? err.message : String(err)}`,
-                            },
-                        ],
-                        details: { path: resolved },
-                    };
-                }
-            },
-        },
-    ];
-}
-// ── PiAdapter ─────────────────────────────────────────────────────────────────
-export class PiAdapter extends AgentAdapter {
-    type = 'pi';
-    agent = null;
-    sessionId = null;
-    workDir = null;
-    seq = 0;
-    runId = '';
-    emittedLengths = new Map();
-    terminalState = 'open';
-    authToken = null;
-    proxyUrl = null;
-    providerConfig;
-    sessionDir = null;
-    messagesPath = null;
-    snapshotPath = null;
-    summaryPath = null;
-    cachedSummary = null;
-    restoredMessages = [];
-    compressing = false;
-    lastSummaryMsgCount = 0;
-    pendingBaseMessages = [];
-    finalizePromise = Promise.resolve();
-    sendGeneration = 0;
-    externalChannel = null;
-    shennianSessionId = null;
-    extraEnv = {};
-    pendingSendStart = null;
-    configure(options) {
-        this.shennianSessionId = options.sessionId ?? null;
-        this.externalChannel = options.externalChannel ?? null;
-        this.extraEnv = options.env ?? {};
-    }
-    async start(sessionId, workDir, _agentSessionId) {
-        this.sessionId = sessionId;
-        this.workDir = workDir;
-        this.seq = 0;
-        this.sessionDir = getSessionDir(sessionId);
-        this.messagesPath = path.join(this.sessionDir, MESSAGES_FILENAME);
-        this.snapshotPath = path.join(this.sessionDir, SNAPSHOT_FILENAME);
-        this.summaryPath = path.join(this.sessionDir, SUMMARY_FILENAME);
-        this.loadSnapshot();
-        this.loadSummary();
-    }
-    async send(text, modelId) {
-        const generation = ++this.sendGeneration;
-        const interruptedMessages = this.agent && this.terminalState === 'open'
-            ? cloneMessages(this.pendingBaseMessages)
-            : null;
-        if (interruptedMessages) {
-            this.agent?.abort();
-            this.agent = null;
-            this.restoredMessages = interruptedMessages;
-            this.finalizePromise = Promise.resolve();
-        }
-        const config = loadConfig();
-        const authToken = config.machineToken ?? config.accessToken;
-        if (!authToken) {
-            this.emit('agentEvent', {
-                state: 'error',
-                runId: this.runId,
-                seq: ++this.seq,
-                message: '未配对，请先运行 shennian 完成机器配对',
-            });
-            return;
-        }
-        this.authToken = authToken;
-        this.proxyUrl = (config.serverUrl ?? SERVERS.cn.url).replace(/\/$/, '');
-        const managedProviderConfig = getManagedAgentProviderConfig('pi');
-        const dashscopeKey = config.apiKeys?.dashscope?.trim();
-        this.providerConfig = managedProviderConfig ?? (dashscopeKey
-            ? { agent: 'pi', token: dashscopeKey, updatedAt: '' }
-            : undefined);
-        this.runId = randomUUID();
-        this.seq = 0;
-        this.emittedLengths.clear();
-        this.terminalState = 'open';
-        if (!this.agent) {
-            this.initAgent();
-        }
-        await this.finalizePromise.catch(() => { });
-        this.agent?.setModel(createPiModel(modelId ?? PI_DEFAULT_MODEL_ID));
-        this.pendingBaseMessages = cloneMessages(this.agent?.state.messages ?? []);
-        const runId = this.runId;
-        const sendAccepted = new Promise((resolve, reject) => {
-            this.pendingSendStart = { runId, resolve, reject };
-        });
-        this.emit('agentEvent', { state: 'init', runId, seq: ++this.seq });
-        void this.agent.prompt(text)
-            .then(async () => {
-            if (generation !== this.sendGeneration)
-                return;
-            this.resolvePendingSendStart(runId);
-            await this.finalizePromise.catch(() => { });
-        })
-            .catch((err) => {
-            if (generation !== this.sendGeneration)
-                return;
-            this.rejectPendingSendStart(runId, err);
-            if (this.terminalState !== 'open')
-                return;
-            this.terminalState = 'error';
-            const message = err instanceof Error ? err.message : String(err);
-            if (message.includes('429') || message.includes('daily_quota_exceeded') || message.includes('nian_quota_exceeded')) {
-                this.emit('agentEvent', {
-                    state: 'error',
-                    runId,
-                    seq: ++this.seq,
-                    message: 'Nian 今日额度已用完，次日自动恢复。',
-                });
-            }
-            else {
-                this.emit('agentEvent', {
-                    state: 'error',
-                    runId,
-                    seq: ++this.seq,
-                    message,
-                });
-            }
-        });
-        await sendAccepted;
-    }
-    // ── Agent lifecycle ──────────────────────────────────────────────────────────
-    initAgent() {
-        const workDir = this.workDir ?? process.cwd();
-        const tools = makeTools(workDir, this.extraEnv);
-        const agentsMdInstructions = loadAgentsMdInstructions(workDir);
-        const agent = new Agent({
-            initialState: {
-                systemPrompt: [
-                    SYSTEM_PROMPT,
-                    agentsMdInstructions,
-                    `当前工作目录：${workDir}`,
-                    buildExternalChannelInstructions(this.externalChannel, workDir, this.shennianSessionId ?? undefined),
-                ].filter(Boolean).join('\n\n'),
-                model: createPiModel(),
-                tools,
-            },
-            streamFn: (model, context, options) => {
-                const providerConfig = this.providerConfig;
-                if (providerConfig?.token) {
-                    return streamOpenAICompletions(createPiModel(model.id, {
-                        provider: 'dashscope',
-                        baseUrl: providerConfig.baseUrl || DASHSCOPE_COMPATIBLE_BASE_URL,
-                        compat: {
-                            supportsDeveloperRole: false,
-                            supportsStore: false,
-                            supportsReasoningEffort: false,
-                            maxTokensField: 'max_tokens',
-                            thinkingFormat: 'qwen',
-                        },
-                    }), context, {
-                        ...options,
-                        apiKey: providerConfig.token,
-                    });
-                }
-                return streamProxy(model, context, {
-                    ...options,
-                    authToken: this.authToken,
-                    proxyUrl: this.proxyUrl,
-                });
-            },
-            transformContext: (messages) => this.compressContext(messages),
-        });
-        this.agent = agent;
-        if (this.restoredMessages.length > 0) {
-            agent.replaceMessages(cloneMessages(this.restoredMessages));
-        }
-        else if (this.cachedSummary) {
-            agent.appendMessage({
-                role: 'user',
-                content: [{ type: 'text', text: `[之前的对话摘要]\n${this.cachedSummary}` }],
-                timestamp: Date.now(),
-            });
-            agent.appendMessage({
-                role: 'assistant',
-                content: [{ type: 'text', text: '好的，我已了解之前的对话上下文，请继续。' }],
-                timestamp: Date.now(),
-            });
-        }
-        agent.subscribe((evt) => {
-            if (this.agent !== agent)
-                return;
-            if (this.terminalState !== 'open')
-                return;
-            const seq = ++this.seq;
-            const runId = this.runId;
-            switch (evt.type) {
-                case 'agent_start':
-                    this.resolvePendingSendStart(runId);
-                    this.emit('agentEvent', { state: 'start', runId, seq });
-                    break;
-                case 'message_update': {
-                    const msg = evt.message;
-                    if (msg.role !== 'assistant')
-                        break;
-                    const content = msg.content ?? [];
-                    for (let i = 0; i < content.length; i++) {
-                        const c = content[i];
-                        if (c.type === 'text' && c.text) {
-                            const key = `text:${i}`;
-                            const prev = this.emittedLengths.get(key) ?? 0;
-                            if (c.text.length > prev) {
-                                const delta = c.text.slice(prev);
-                                this.emittedLengths.set(key, c.text.length);
-                                this.emit('agentEvent', { state: 'delta', runId, seq, text: delta });
-                            }
-                        }
-                        else if (c.type === 'thinking' && c.thinking) {
-                            const key = `thinking:${i}`;
-                            const prev = this.emittedLengths.get(key) ?? 0;
-                            if (c.thinking.length > prev) {
-                                const delta = c.thinking.slice(prev);
-                                this.emittedLengths.set(key, c.thinking.length);
-                                this.emit('agentEvent', { state: 'delta', runId, seq, text: delta, thinking: true });
-                            }
-                        }
-                    }
-                    break;
-                }
-                case 'tool_execution_start':
-                    this.emit('agentEvent', {
-                        state: 'tool-call',
-                        runId,
-                        seq,
-                        name: evt.toolName,
-                        args: evt.args,
-                    });
-                    break;
-                case 'tool_execution_end':
-                    this.emit('agentEvent', {
-                        state: 'tool-result',
-                        runId,
-                        seq,
-                        name: evt.toolName,
-                        result: typeof evt.result === 'string' ? evt.result : JSON.stringify(evt.result),
-                    });
-                    break;
-                case 'agent_end': {
-                    this.terminalState = 'final';
-                    const msgs = evt.messages;
-                    const lastAssistant = [...msgs].reverse().find((m) => m.role === 'assistant');
-                    const usage = lastAssistant?.usage;
-                    this.emit('agentEvent', {
-                        state: 'final',
-                        runId,
-                        seq,
-                        usage: usage
-                            ? {
-                                inputTokens: usage.input ?? 0,
-                                outputTokens: usage.output ?? 0,
-                            }
-                            : undefined,
-                    });
-                    this.finalizePromise = this.finalizeTurn(msgs);
-                    break;
-                }
-            }
-        });
-    }
-    // ── Context compression ────────────────────────────────────────────────────
-    /**
-     * After each turn, compress old messages and actually replace them in the
-     * Agent's internal state. This is the primary compression mechanism.
-     */
-    async compressAndReplace() {
-        if (!this.agent || this.compressing)
-            return;
-        const messages = this.agent.state.messages;
-        const tokens = estimateTokens(messages);
-        if (tokens < CONTEXT_TOKEN_THRESHOLD)
-            return;
-        const split = this.splitMessages(messages);
-        if (!split)
-            return;
-        const summary = await this.generateSummary(split.toCompress);
-        if (!summary)
-            return;
-        this.cachedSummary = summary;
-        this.lastSummaryMsgCount += split.toCompress.length;
-        const replacement = [
-            {
-                role: 'user',
-                content: [{ type: 'text', text: `[历史对话摘要]\n${summary}` }],
-                timestamp: Date.now(),
-            },
-            {
-                role: 'assistant',
-                content: [{ type: 'text', text: '好的，我已了解上下文。' }],
-                timestamp: Date.now(),
-            },
-            ...split.toKeep,
-        ];
-        this.agent.replaceMessages(replacement);
-    }
-    /**
-     * Safety fallback: if messages somehow exceed the threshold at LLM call time
-     * (e.g. compressAndReplace hasn't run yet), return a truncated view using
-     * the cached summary. Does NOT call LLM — avoids nested async delays.
-     */
-    async compressContext(messages) {
-        const tokens = estimateTokens(messages);
-        if (tokens < CONTEXT_TOKEN_THRESHOLD)
-            return messages;
-        if (!this.cachedSummary)
-            return messages;
-        const split = this.splitMessages(messages);
-        if (!split)
-            return messages;
-        return [
-            {
-                role: 'user',
-                content: [{ type: 'text', text: `[历史对话摘要]\n${this.cachedSummary}` }],
-                timestamp: Date.now(),
-            },
-            ...split.toKeep,
-        ];
-    }
-    splitMessages(messages) {
-        const nonSystem = messages.filter((m) => m.role !== 'system');
-        if (nonSystem.length <= KEEP_RECENT_MESSAGES)
-            return null;
-        let splitIdx = nonSystem.length - KEEP_RECENT_MESSAGES;
-        while (splitIdx > 0 &&
-            nonSystem[splitIdx].role === 'toolResult') {
-            splitIdx--;
-        }
-        if (splitIdx <= 0)
-            return null;
-        return { toCompress: nonSystem.slice(0, splitIdx), toKeep: nonSystem.slice(splitIdx) };
-    }
-    async generateSummary(messages) {
-        if (!this.authToken || !this.proxyUrl || this.compressing)
-            return this.cachedSummary;
-        this.compressing = true;
-        try {
-            const text = messagesToText(messages);
-            if (text.length < 100)
-                return null;
-            const truncated = text.length > 10000 ? text.slice(0, 10000) + '\n...(已截断)' : text;
-            return await requestProxySummary(this.proxyUrl, this.authToken, `请将以下对话历史压缩为简洁摘要，保留关键信息：讨论的文件和目录、做出的技术决定、当前任务进度、未完成的工作。摘要控制在 300 字以内。\n\n${truncated}`);
-        }
-        catch {
-            return this.cachedSummary;
-        }
-        finally {
-            this.compressing = false;
-        }
-    }
-    // ── Summary persistence ────────────────────────────────────────────────────
-    loadSnapshot() {
-        if (!this.snapshotPath)
-            return;
-        try {
-            const raw = fs.readFileSync(this.snapshotPath, 'utf-8');
-            const data = JSON.parse(raw);
-            this.restoredMessages = Array.isArray(data.messages) ? data.messages : [];
-            this.cachedSummary = data.summary ?? this.cachedSummary;
-            this.lastSummaryMsgCount = data.summarizedCount ?? this.lastSummaryMsgCount;
-        }
-        catch {
-            this.restoredMessages = [];
-        }
-    }
-    loadSummary() {
-        if (!this.summaryPath)
-            return;
-        try {
-            const raw = fs.readFileSync(this.summaryPath, 'utf-8');
-            const data = JSON.parse(raw);
-            this.cachedSummary = data.summary ?? null;
-            this.lastSummaryMsgCount = data.summarizedCount ?? 0;
-        }
-        catch {
-            if (!this.sessionDir)
-                return;
-            try {
-                const legacyPath = path.join(this.sessionDir, LEGACY_SUMMARY_FILENAME);
-                const raw = fs.readFileSync(legacyPath, 'utf-8');
-                const data = JSON.parse(raw);
-                this.cachedSummary = data.summary ?? null;
-                this.lastSummaryMsgCount = data.summarizedCount ?? 0;
-            }
-            catch {
-                // File doesn't exist or is corrupt — start fresh
-            }
-        }
-    }
-    saveSummary(messages) {
-        if (!this.summaryPath)
-            return;
-        try {
-            fs.mkdirSync(path.dirname(this.summaryPath), { recursive: true });
-            const summary = buildRollingSummary(this.cachedSummary, messages);
-            this.cachedSummary = summary;
-            fs.writeFileSync(this.summaryPath, JSON.stringify({
-                version: 1,
-                summary,
-                summarizedCount: this.lastSummaryMsgCount,
-                updatedAt: Date.now(),
-            }, null, 2));
-        }
-        catch {
-            // Best-effort persistence
-        }
-    }
-    saveSnapshot(messages) {
-        if (!this.snapshotPath || !this.sessionId || !this.workDir)
-            return;
-        try {
-            fs.mkdirSync(path.dirname(this.snapshotPath), { recursive: true });
-            fs.writeFileSync(this.snapshotPath, JSON.stringify({
-                version: 1,
-                sessionId: this.sessionId,
-                workDir: this.workDir,
-                summary: this.cachedSummary,
-                summarizedCount: this.lastSummaryMsgCount,
-                messages,
-                updatedAt: Date.now(),
-            }, null, 2));
-            this.restoredMessages = cloneMessages(messages);
-        }
-        catch {
-            // Best-effort persistence
-        }
-    }
-    appendMessagesLog(messages) {
-        if (!this.messagesPath || messages.length === 0)
-            return;
-        try {
-            fs.mkdirSync(path.dirname(this.messagesPath), { recursive: true });
-            const lines = messages.map((message) => JSON.stringify(message)).join('\n') + '\n';
-            fs.appendFileSync(this.messagesPath, lines, 'utf-8');
-        }
-        catch {
-            // Best-effort persistence
-        }
-    }
-    getTurnMessages(finalMessages) {
-        const prefixLength = longestCommonPrefixLength(this.pendingBaseMessages, finalMessages);
-        return cloneMessages(finalMessages.slice(prefixLength));
-    }
-    async finalizeTurn(finalMessages) {
-        const clonedFinalMessages = cloneMessages(finalMessages);
-        const newMessages = this.getTurnMessages(clonedFinalMessages);
-        this.appendMessagesLog(newMessages);
-        await this.compressAndReplace();
-        const currentMessages = this.agent
-            ? cloneMessages(this.agent.state.messages)
-            : clonedFinalMessages;
-        if (!this.cachedSummary) {
-            this.cachedSummary = buildRollingSummary(null, clonedFinalMessages);
-        }
-        this.saveSummary(currentMessages);
-        this.saveSnapshot(currentMessages);
-    }
-    async resume(_agentSessionId) {
-        this.loadSnapshot();
-        this.loadSummary();
-        if (this.agent && this.restoredMessages.length > 0) {
-            this.agent.replaceMessages(cloneMessages(this.restoredMessages));
-        }
-    }
-    async stop() {
-        this.rejectPendingSendStart(this.runId, new Error('Pi run stopped before dispatch completed'));
-        await this.finalizePromise.catch(() => { });
-        if (this.agent) {
-            const currentMessages = cloneMessages(this.agent.state.messages ?? []);
-            this.saveSummary(currentMessages);
-            this.saveSnapshot(currentMessages);
-        }
-        this.agent?.abort();
-        this.agent = null;
-    }
-    resolvePendingSendStart(runId) {
-        if (!this.pendingSendStart || this.pendingSendStart.runId !== runId)
-            return;
-        const { resolve } = this.pendingSendStart;
-        this.pendingSendStart = null;
-        resolve();
-    }
-    rejectPendingSendStart(runId, error) {
-        if (!this.pendingSendStart || this.pendingSendStart.runId !== runId)
-            return;
-        const { reject } = this.pendingSendStart;
-        this.pendingSendStart = null;
-        reject(error);
-    }
-}
-registerAgent('pi', () => new PiAdapter());
+import{randomUUID as C}from"node:crypto";import c from"node:fs";import g from"node:path";import{execFile as _}from"node:child_process";import{promisify as b}from"node:util";import{Agent as A,streamProxy as T}from"@mariozechner/pi-agent-core";import{streamOpenAICompletions as I}from"@mariozechner/pi-ai/openai-completions";import{Type as d}from"@sinclair/typebox";import{AgentAdapter as D,registerAgent as j}from"./adapter.js";import{buildExternalChannelInstructions as F}from"./external-channel-instructions.js";import{loadConfig as L}from"../config/index.js";import{getManagedAgentProviderConfig as O}from"./config-status.js";import{SERVERS as $}from"../region.js";import{buildRollingSummary as E,buildShellCommandSpec as P,cloneMessages as y,CONTEXT_TOKEN_THRESHOLD as v,createPiModel as M,estimateTokens as w,getSessionDir as N,KEEP_RECENT_MESSAGES as k,LEGACY_SUMMARY_FILENAME as q,loadAgentsMdInstructions as z,longestCommonPrefixLength as R,MESSAGES_FILENAME as U,messagesToText as B,PI_DEFAULT_MODEL_ID as K,requestProxySummary as G,SNAPSHOT_FILENAME as J,SYSTEM_PROMPT as H,SUMMARY_FILENAME as Y}from"./pi-context.js";import{buildShellCommandSpec as yt}from"./pi-context.js";const W=b(_),V="https://dashscope.aliyuncs.com/compatible-mode/v1";function f(p,t){return g.resolve(p,t)}async function X(p,t,e,n){const i=P(t),{stdout:s,stderr:a}=await W(i.file,i.args,{cwd:p,env:{...process.env,...e},timeout:3e4,signal:n,maxBuffer:1024*1024,windowsHide:!0});return{stdout:s,stderr:a,shell:i.shell}}function Q(p,t={}){return[{name:"read_file",label:"\u8BFB\u53D6\u6587\u4EF6",description:"Read the full content of a file at the given path.",parameters:d.Object({path:d.String({description:"Absolute or relative file path"})}),async execute(e,{path:n}){let i="";try{return i=f(p,n),{content:[{type:"text",text:c.readFileSync(i,"utf-8")}],details:{path:i}}}catch(s){return{content:[{type:"text",text:`Error: ${s instanceof Error?s.message:String(s)}`}],details:{path:i}}}}},{name:"write_file",label:"\u5199\u5165\u6587\u4EF6",description:"Write content to a file, creating it or overwriting if it exists.",parameters:d.Object({path:d.String({description:"Absolute or relative file path"}),content:d.String({description:"File content to write"})}),async execute(e,{path:n,content:i}){let s="";try{return s=f(p,n),c.mkdirSync(g.dirname(s),{recursive:!0}),c.writeFileSync(s,i,"utf-8"),{content:[{type:"text",text:`Written ${i.length} bytes to ${s}`}],details:{path:s}}}catch(a){return{content:[{type:"text",text:`Error: ${a instanceof Error?a.message:String(a)}`}],details:{path:s}}}}},{name:"list_directory",label:"\u5217\u51FA\u76EE\u5F55",description:"List files and directories at the given path.",parameters:d.Object({path:d.String({description:"Directory path to list",default:"."})}),async execute(e,{path:n}){let i="";try{return i=f(p,n),{content:[{type:"text",text:c.readdirSync(i,{withFileTypes:!0}).map(o=>`${o.isDirectory()?"d":"f"} ${o.name}`).join(`
+`)||"(empty)"}],details:{path:i}}}catch(s){return{content:[{type:"text",text:`Error: ${s instanceof Error?s.message:String(s)}`}],details:{path:i}}}}},{name:"bash",label:"\u6267\u884C\u547D\u4EE4",description:"Execute a shell command in the working directory. Uses PowerShell on Windows and bash on macOS/Linux. Timeout: 30s.",parameters:d.Object({command:d.String({description:"Shell command to execute"})}),async execute(e,{command:n},i){try{const{stdout:s,stderr:a,shell:o}=await X(p,n,t,i);return{content:[{type:"text",text:[s,a].filter(Boolean).join(`
+---stderr---
+`)||"(no output)"}],details:{command:n,shell:o}}}catch(s){return{content:[{type:"text",text:`Error: ${s instanceof Error?s.message:String(s)}`}],details:{command:n,shell:P(n).shell}}}}},{name:"read_image",label:"\u8BFB\u53D6\u56FE\u7247",description:"Read an image file and return its content for visual analysis. Supports png, jpg, jpeg, gif, webp. Max 10 MB.",parameters:d.Object({path:d.String({description:"Absolute or relative path to the image file"})}),async execute(e,{path:n}){let i;try{i=f(p,n)}catch(r){return{content:[{type:"text",text:`Error: ${r instanceof Error?r.message:String(r)}`}],details:{path:n}}}const s=g.extname(i).toLowerCase().slice(1),o={png:"image/png",jpg:"image/jpeg",jpeg:"image/jpeg",gif:"image/gif",webp:"image/webp"}[s];if(!o)return{content:[{type:"text",text:`Error: unsupported image type ".${s}". Supported: png, jpg, jpeg, gif, webp`}],details:{path:i}};try{const r=c.statSync(i);return r.size>10*1024*1024?{content:[{type:"text",text:`Error: image too large (${(r.size/1024/1024).toFixed(1)} MB, max 10 MB)`}],details:{path:i}}:{content:[{type:"image",data:c.readFileSync(i).toString("base64"),mimeType:o}],details:{path:i,mimeType:o,size:r.size}}}catch(r){return{content:[{type:"text",text:`Error: ${r instanceof Error?r.message:String(r)}`}],details:{path:i}}}}}]}class Z extends D{type="pi";agent=null;sessionId=null;workDir=null;seq=0;runId="";emittedLengths=new Map;terminalState="open";authToken=null;proxyUrl=null;providerConfig;sessionDir=null;messagesPath=null;snapshotPath=null;summaryPath=null;cachedSummary=null;restoredMessages=[];compressing=!1;lastSummaryMsgCount=0;pendingBaseMessages=[];finalizePromise=Promise.resolve();sendGeneration=0;externalChannel=null;shennianSessionId=null;extraEnv={};pendingSendStart=null;configure(t){this.shennianSessionId=t.sessionId??null,this.externalChannel=t.externalChannel??null,this.extraEnv=t.env??{}}async start(t,e,n){this.sessionId=t,this.workDir=e,this.seq=0,this.sessionDir=N(t),this.messagesPath=g.join(this.sessionDir,U),this.snapshotPath=g.join(this.sessionDir,J),this.summaryPath=g.join(this.sessionDir,Y),this.loadSnapshot(),this.loadSummary()}async send(t,e){const n=++this.sendGeneration,i=this.agent&&this.terminalState==="open"?y(this.pendingBaseMessages):null;i&&(this.agent?.abort(),this.agent=null,this.restoredMessages=i,this.finalizePromise=Promise.resolve());const s=L(),a=s.machineToken??s.accessToken;if(!a){this.emit("agentEvent",{state:"error",runId:this.runId,seq:++this.seq,message:"\u672A\u914D\u5BF9\uFF0C\u8BF7\u5148\u8FD0\u884C shennian \u5B8C\u6210\u673A\u5668\u914D\u5BF9"});return}this.authToken=a,this.proxyUrl=(s.serverUrl??$.cn.url).replace(/\/$/,"");const o=O("pi"),r=s.apiKeys?.dashscope?.trim();this.providerConfig=o??(r?{agent:"pi",token:r,updatedAt:""}:void 0),this.runId=C(),this.seq=0,this.emittedLengths.clear(),this.terminalState="open",this.agent||this.initAgent(),await this.finalizePromise.catch(()=>{}),this.agent?.setModel(M(e??K)),this.pendingBaseMessages=y(this.agent?.state.messages??[]);const u=this.runId,l=new Promise((h,m)=>{this.pendingSendStart={runId:u,resolve:h,reject:m}});this.emit("agentEvent",{state:"init",runId:u,seq:++this.seq}),this.agent.prompt(t).then(async()=>{n===this.sendGeneration&&(this.resolvePendingSendStart(u),await this.finalizePromise.catch(()=>{}))}).catch(h=>{if(n!==this.sendGeneration||(this.rejectPendingSendStart(u,h),this.terminalState!=="open"))return;this.terminalState="error";const m=h instanceof Error?h.message:String(h);m.includes("429")||m.includes("daily_quota_exceeded")||m.includes("nian_quota_exceeded")?this.emit("agentEvent",{state:"error",runId:u,seq:++this.seq,message:"Nian \u4ECA\u65E5\u989D\u5EA6\u5DF2\u7528\u5B8C\uFF0C\u6B21\u65E5\u81EA\u52A8\u6062\u590D\u3002"}):this.emit("agentEvent",{state:"error",runId:u,seq:++this.seq,message:m})}),await l}initAgent(){const t=this.workDir??process.cwd(),e=Q(t,this.extraEnv),n=z(t),i=new A({initialState:{systemPrompt:[H,n,`\u5F53\u524D\u5DE5\u4F5C\u76EE\u5F55\uFF1A${t}`,F(this.externalChannel,t,this.shennianSessionId??void 0)].filter(Boolean).join(`
+
+`),model:M(),tools:e},streamFn:(s,a,o)=>{const r=this.providerConfig;return r?.token?I(M(s.id,{provider:"dashscope",baseUrl:r.baseUrl||V,compat:{supportsDeveloperRole:!1,supportsStore:!1,supportsReasoningEffort:!1,maxTokensField:"max_tokens",thinkingFormat:"qwen"}}),a,{...o,apiKey:r.token}):T(s,a,{...o,authToken:this.authToken,proxyUrl:this.proxyUrl})},transformContext:s=>this.compressContext(s)});this.agent=i,this.restoredMessages.length>0?i.replaceMessages(y(this.restoredMessages)):this.cachedSummary&&(i.appendMessage({role:"user",content:[{type:"text",text:`[\u4E4B\u524D\u7684\u5BF9\u8BDD\u6458\u8981]
+${this.cachedSummary}`}],timestamp:Date.now()}),i.appendMessage({role:"assistant",content:[{type:"text",text:"\u597D\u7684\uFF0C\u6211\u5DF2\u4E86\u89E3\u4E4B\u524D\u7684\u5BF9\u8BDD\u4E0A\u4E0B\u6587\uFF0C\u8BF7\u7EE7\u7EED\u3002"}],timestamp:Date.now()})),i.subscribe(s=>{if(this.agent!==i||this.terminalState!=="open")return;const a=++this.seq,o=this.runId;switch(s.type){case"agent_start":this.resolvePendingSendStart(o),this.emit("agentEvent",{state:"start",runId:o,seq:a});break;case"message_update":{const r=s.message;if(r.role!=="assistant")break;const u=r.content??[];for(let l=0;l<u.length;l++){const h=u[l];if(h.type==="text"&&h.text){const m=`text:${l}`,S=this.emittedLengths.get(m)??0;if(h.text.length>S){const x=h.text.slice(S);this.emittedLengths.set(m,h.text.length),this.emit("agentEvent",{state:"delta",runId:o,seq:a,text:x})}}else if(h.type==="thinking"&&h.thinking){const m=`thinking:${l}`,S=this.emittedLengths.get(m)??0;if(h.thinking.length>S){const x=h.thinking.slice(S);this.emittedLengths.set(m,h.thinking.length),this.emit("agentEvent",{state:"delta",runId:o,seq:a,text:x,thinking:!0})}}}break}case"tool_execution_start":this.emit("agentEvent",{state:"tool-call",runId:o,seq:a,name:s.toolName,args:s.args});break;case"tool_execution_end":this.emit("agentEvent",{state:"tool-result",runId:o,seq:a,name:s.toolName,result:typeof s.result=="string"?s.result:JSON.stringify(s.result)});break;case"agent_end":{this.terminalState="final";const r=s.messages,l=[...r].reverse().find(h=>h.role==="assistant")?.usage;this.emit("agentEvent",{state:"final",runId:o,seq:a,usage:l?{inputTokens:l.input??0,outputTokens:l.output??0}:void 0}),this.finalizePromise=this.finalizeTurn(r);break}}})}async compressAndReplace(){if(!this.agent||this.compressing)return;const t=this.agent.state.messages;if(w(t)<v)return;const n=this.splitMessages(t);if(!n)return;const i=await this.generateSummary(n.toCompress);if(!i)return;this.cachedSummary=i,this.lastSummaryMsgCount+=n.toCompress.length;const s=[{role:"user",content:[{type:"text",text:`[\u5386\u53F2\u5BF9\u8BDD\u6458\u8981]
+${i}`}],timestamp:Date.now()},{role:"assistant",content:[{type:"text",text:"\u597D\u7684\uFF0C\u6211\u5DF2\u4E86\u89E3\u4E0A\u4E0B\u6587\u3002"}],timestamp:Date.now()},...n.toKeep];this.agent.replaceMessages(s)}async compressContext(t){if(w(t)<v||!this.cachedSummary)return t;const n=this.splitMessages(t);return n?[{role:"user",content:[{type:"text",text:`[\u5386\u53F2\u5BF9\u8BDD\u6458\u8981]
+${this.cachedSummary}`}],timestamp:Date.now()},...n.toKeep]:t}splitMessages(t){const e=t.filter(i=>i.role!=="system");if(e.length<=k)return null;let n=e.length-k;for(;n>0&&e[n].role==="toolResult";)n--;return n<=0?null:{toCompress:e.slice(0,n),toKeep:e.slice(n)}}async generateSummary(t){if(!this.authToken||!this.proxyUrl||this.compressing)return this.cachedSummary;this.compressing=!0;try{const e=B(t);if(e.length<100)return null;const n=e.length>1e4?e.slice(0,1e4)+`
+...(\u5DF2\u622A\u65AD)`:e;return await G(this.proxyUrl,this.authToken,`\u8BF7\u5C06\u4EE5\u4E0B\u5BF9\u8BDD\u5386\u53F2\u538B\u7F29\u4E3A\u7B80\u6D01\u6458\u8981\uFF0C\u4FDD\u7559\u5173\u952E\u4FE1\u606F\uFF1A\u8BA8\u8BBA\u7684\u6587\u4EF6\u548C\u76EE\u5F55\u3001\u505A\u51FA\u7684\u6280\u672F\u51B3\u5B9A\u3001\u5F53\u524D\u4EFB\u52A1\u8FDB\u5EA6\u3001\u672A\u5B8C\u6210\u7684\u5DE5\u4F5C\u3002\u6458\u8981\u63A7\u5236\u5728 300 \u5B57\u4EE5\u5185\u3002
+
+${n}`)}catch{return this.cachedSummary}finally{this.compressing=!1}}loadSnapshot(){if(this.snapshotPath)try{const t=c.readFileSync(this.snapshotPath,"utf-8"),e=JSON.parse(t);this.restoredMessages=Array.isArray(e.messages)?e.messages:[],this.cachedSummary=e.summary??this.cachedSummary,this.lastSummaryMsgCount=e.summarizedCount??this.lastSummaryMsgCount}catch{this.restoredMessages=[]}}loadSummary(){if(this.summaryPath)try{const t=c.readFileSync(this.summaryPath,"utf-8"),e=JSON.parse(t);this.cachedSummary=e.summary??null,this.lastSummaryMsgCount=e.summarizedCount??0}catch{if(!this.sessionDir)return;try{const t=g.join(this.sessionDir,q),e=c.readFileSync(t,"utf-8"),n=JSON.parse(e);this.cachedSummary=n.summary??null,this.lastSummaryMsgCount=n.summarizedCount??0}catch{}}}saveSummary(t){if(this.summaryPath)try{c.mkdirSync(g.dirname(this.summaryPath),{recursive:!0});const e=E(this.cachedSummary,t);this.cachedSummary=e,c.writeFileSync(this.summaryPath,JSON.stringify({version:1,summary:e,summarizedCount:this.lastSummaryMsgCount,updatedAt:Date.now()},null,2))}catch{}}saveSnapshot(t){if(!(!this.snapshotPath||!this.sessionId||!this.workDir))try{c.mkdirSync(g.dirname(this.snapshotPath),{recursive:!0}),c.writeFileSync(this.snapshotPath,JSON.stringify({version:1,sessionId:this.sessionId,workDir:this.workDir,summary:this.cachedSummary,summarizedCount:this.lastSummaryMsgCount,messages:t,updatedAt:Date.now()},null,2)),this.restoredMessages=y(t)}catch{}}appendMessagesLog(t){if(!(!this.messagesPath||t.length===0))try{c.mkdirSync(g.dirname(this.messagesPath),{recursive:!0});const e=t.map(n=>JSON.stringify(n)).join(`
+`)+`
+`;c.appendFileSync(this.messagesPath,e,"utf-8")}catch{}}getTurnMessages(t){const e=R(this.pendingBaseMessages,t);return y(t.slice(e))}async finalizeTurn(t){const e=y(t),n=this.getTurnMessages(e);this.appendMessagesLog(n),await this.compressAndReplace();const i=this.agent?y(this.agent.state.messages):e;this.cachedSummary||(this.cachedSummary=E(null,e)),this.saveSummary(i),this.saveSnapshot(i)}async resume(t){this.loadSnapshot(),this.loadSummary(),this.agent&&this.restoredMessages.length>0&&this.agent.replaceMessages(y(this.restoredMessages))}async stop(){if(this.rejectPendingSendStart(this.runId,new Error("Pi run stopped before dispatch completed")),await this.finalizePromise.catch(()=>{}),this.agent){const t=y(this.agent.state.messages??[]);this.saveSummary(t),this.saveSnapshot(t)}this.agent?.abort(),this.agent=null}resolvePendingSendStart(t){if(!this.pendingSendStart||this.pendingSendStart.runId!==t)return;const{resolve:e}=this.pendingSendStart;this.pendingSendStart=null,e()}rejectPendingSendStart(t,e){if(!this.pendingSendStart||this.pendingSendStart.runId!==t)return;const{reject:n}=this.pendingSendStart;this.pendingSendStart=null,n(e)}}j("pi",()=>new Z);export{Z as PiAdapter,yt as buildShellCommandSpec};