@code4bug/jarvis-agent 1.0.2

package/dist/config/constants.d.ts

@@ -0,0 +1,25 @@
+/** 应用全局常量 */
+export declare const APP_VERSION: string;
+export declare const PROJECT_NAME: string;
+/** Agentic Loop 最大迭代次数 */
+export declare const MAX_ITERATIONS = 50;
+export declare const SESSIONS_DIR: string;
+/** 输入后是否隐藏 WelcomeHeader，默认 false（不隐藏） */
+export declare const HIDE_WELCOME_AFTER_INPUT = false;
+export declare const MODEL_NAME: string;
+/** 是否支持思考/非思考模式切换，默认 false（隐藏该功能） */
+export declare const ENABLE_THINKING_MODE_TOGGLE: boolean;
+/** 上下文 token 上限 */
+export declare const CONTEXT_TOKEN_LIMIT: number;
+/** 默认智能体名称（硬编码兜底值） */
+export declare const DEFAULT_AGENT_FALLBACK = "Jarvis";
+/** 智能体定义文件目录（相对项目根目录） */
+export declare const AGENTS_DIR = "src/agents";
+/** 智能体默认配色 */
+export declare const DEFAULT_AGENT_COLOR = "green";
+/** 智能体默认标识符 */
+export declare const DEFAULT_AGENT_EMOJI = ">";
+/** 当前激活的智能体名称 — 启动时从 ~/.jarvis/agent.json 读取，运行时可切换 */
+export declare const DEFAULT_AGENT: string;
+/** 应用名称 — 取自当前激活智能体的 name */
+export declare const APP_NAME: string;

package/dist/config/constants.js

@@ -0,0 +1,67 @@
+import fs from 'fs';
+import path from 'path';
+import { fileURLToPath } from 'url';
+/** 从 package.json 动态读取版本号 */
+function resolveAppVersion() {
+    try {
+        const __dirname = path.dirname(fileURLToPath(import.meta.url));
+        const pkgPath = path.resolve(__dirname, '../../package.json');
+        const pkg = JSON.parse(fs.readFileSync(pkgPath, 'utf-8'));
+        return `v${pkg.version}`;
+    }
+    catch {
+        return 'v0.0.0';
+    }
+}
+/** 应用全局常量 */
+export const APP_VERSION = resolveAppVersion();
+export const PROJECT_NAME = path.basename(process.cwd());
+/** Agentic Loop 最大迭代次数 */
+export const MAX_ITERATIONS = 50;
+/** 会话存储目录（~/.jarvis/sessions/） */
+import os from 'os';
+export const SESSIONS_DIR = path.join(os.homedir(), '.jarvis', 'sessions');
+/** 输入后是否隐藏 WelcomeHeader，默认 false（不隐藏） */
+export const HIDE_WELCOME_AFTER_INPUT = false;
+/** 从配置文件获取当前模型名称 */
+import { loadConfig, getActiveModel } from './loader.js';
+const _cfg = loadConfig();
+function resolveModelName() {
+    try {
+        const active = getActiveModel(_cfg);
+        return active?.model ?? _cfg.system.model ?? 'unknown';
+    }
+    catch {
+        return 'unknown';
+    }
+}
+export const MODEL_NAME = resolveModelName();
+/** 是否支持思考/非思考模式切换，默认 false（隐藏该功能） */
+export const ENABLE_THINKING_MODE_TOGGLE = _cfg.system.enable_thinking_mode_toggle ?? false;
+/** 上下文 token 上限 */
+export const CONTEXT_TOKEN_LIMIT = _cfg.system.context_token_limit ?? 18000;
+// ===== 智能体默认配置 =====
+/** 默认智能体名称（硬编码兜底值） */
+export const DEFAULT_AGENT_FALLBACK = 'Jarvis';
+/** 智能体定义文件目录（相对项目根目录） */
+export const AGENTS_DIR = 'src/agents';
+/** 智能体默认配色 */
+export const DEFAULT_AGENT_COLOR = 'green';
+/** 智能体默认标识符 */
+export const DEFAULT_AGENT_EMOJI = '>';
+// ===== 动态应用名称（跟随激活智能体） =====
+import { getAgent } from '../agents/index.js';
+import { getActiveAgent } from './agentState.js';
+/** 当前激活的智能体名称 — 启动时从 ~/.jarvis/agent.json 读取，运行时可切换 */
+export const DEFAULT_AGENT = getActiveAgent(DEFAULT_AGENT_FALLBACK);
+function resolveAppName() {
+    try {
+        const agent = getAgent(DEFAULT_AGENT);
+        return agent?.meta.name ?? 'Jarvis';
+    }
+    catch {
+        return 'Jarvis';
+    }
+}
+/** 应用名称 — 取自当前激活智能体的 name */
+export const APP_NAME = resolveAppName();

package/dist/config/loader.d.ts

@@ -0,0 +1,30 @@
+/**
+ * 配置加载器
+ *
+ * 优先级（后者覆盖前者）：
+ *   1. ~/.jarvis/config.json   （全局配置）
+ *   2. ./.jarvis/config.json   （项目配置）
+ */
+export interface ModelConfig {
+    api_url: string;
+    api_key: string;
+    model: string;
+    temperature?: number;
+    max_tokens?: number;
+}
+export interface SystemConfig {
+    context_token_limit?: number;
+    context_compress_threshold?: number;
+    /** 当前使用的模型名称，对应 models 中的 key */
+    model?: string;
+    /** 是否支持思考/非思考模式切换，默认 false */
+    enable_thinking_mode_toggle?: boolean;
+}
+export interface JarvisConfig {
+    system: SystemConfig;
+    models: Record<string, ModelConfig>;
+}
+/** 加载合并后的最终配置（结果会被缓存，多次调用只解析一次） */
+export declare function loadConfig(): JarvisConfig;
+/** 根据当前配置获取活跃模型配置 */
+export declare function getActiveModel(config: JarvisConfig): ModelConfig | null;

package/dist/config/loader.js

@@ -0,0 +1,64 @@
+/**
+ * 配置加载器
+ *
+ * 优先级（后者覆盖前者）：
+ *   1. ~/.jarvis/config.json   （全局配置）
+ *   2. ./.jarvis/config.json   （项目配置）
+ */
+import fs from 'fs';
+import path from 'path';
+import os from 'os';
+const CONFIG_FILENAME = 'config.json';
+const JARVIS_DIR = '.jarvis';
+/** 读取并解析单个配置文件，失败返回 null */
+function loadJsonFile(filePath) {
+    try {
+        if (!fs.existsSync(filePath))
+            return null;
+        const raw = fs.readFileSync(filePath, 'utf-8');
+        return JSON.parse(raw);
+    }
+    catch (err) {
+        console.error(`[config] Failed to parse ${filePath}:`, err instanceof Error ? err.message : err);
+        return null;
+    }
+}
+/** 深度合并两个配置，local 覆盖 global */
+function mergeConfigs(global, local) {
+    const base = {
+        system: {},
+        models: {},
+    };
+    if (global) {
+        Object.assign(base.system, global.system);
+        Object.assign(base.models, global.models);
+    }
+    if (local) {
+        Object.assign(base.system, local.system);
+        // 项目级 models 逐个覆盖，而非整体替换
+        for (const [key, val] of Object.entries(local.models ?? {})) {
+            base.models[key] = { ...base.models[key], ...val };
+        }
+    }
+    return base;
+}
+/** 缓存已加载的配置 */
+let _cachedConfig = null;
+/** 加载合并后的最终配置（结果会被缓存，多次调用只解析一次） */
+export function loadConfig() {
+    if (_cachedConfig)
+        return _cachedConfig;
+    const globalPath = path.join(os.homedir(), JARVIS_DIR, CONFIG_FILENAME);
+    const localPath = path.join(process.cwd(), JARVIS_DIR, CONFIG_FILENAME);
+    const globalCfg = loadJsonFile(globalPath);
+    const localCfg = loadJsonFile(localPath);
+    _cachedConfig = mergeConfigs(globalCfg, localCfg);
+    return _cachedConfig;
+}
+/** 根据当前配置获取活跃模型配置 */
+export function getActiveModel(config) {
+    const modelName = config.system.model;
+    if (!modelName)
+        return null;
+    return config.models[modelName] ?? null;
+}

package/dist/config/systemInfo.d.ts

@@ -0,0 +1,12 @@
+/**
+ * 系统信息收集器
+ *
+ * 启动时执行一次，收集当前操作系统和硬件信息，
+ * 格式化后追加到智能体上下文中，帮助 LLM 更好地理解用户环境。
+ */
+/**
+ * 获取格式化的系统信息文本，用于追加到 system prompt
+ *
+ * 返回 markdown 格式的系统环境描述
+ */
+export declare function getSystemInfoPrompt(): string;

package/dist/config/systemInfo.js

@@ -0,0 +1,95 @@
+/**
+ * 系统信息收集器
+ *
+ * 启动时执行一次，收集当前操作系统和硬件信息，
+ * 格式化后追加到智能体上下文中，帮助 LLM 更好地理解用户环境。
+ */
+import os from 'os';
+import { execSync } from 'child_process';
+/** 安全执行命令，失败返回空字符串 */
+function safeExec(cmd) {
+    try {
+        return execSync(cmd, { encoding: 'utf-8', timeout: 5000 }).trim();
+    }
+    catch {
+        return '';
+    }
+}
+/** 收集系统信息 */
+function collectSystemInfo() {
+    const cpus = os.cpus();
+    const platform = os.platform();
+    const extras = {};
+    // 平台特定信息
+    if (platform === 'darwin') {
+        const macModel = safeExec('sysctl -n hw.model');
+        if (macModel)
+            extras['Mac 型号'] = macModel;
+        const macOSName = safeExec('sw_vers -productName');
+        const macOSVer = safeExec('sw_vers -productVersion');
+        if (macOSName && macOSVer)
+            extras['macOS'] = `${macOSName} ${macOSVer}`;
+    }
+    else if (platform === 'linux') {
+        const distro = safeExec('cat /etc/os-release 2>/dev/null | grep PRETTY_NAME | cut -d= -f2 | tr -d \'"\'');
+        if (distro)
+            extras['发行版'] = distro;
+        const kernel = safeExec('uname -r');
+        if (kernel)
+            extras['内核版本'] = kernel;
+    }
+    // 通用工具版本检测
+    const gitVersion = safeExec('git --version');
+    if (gitVersion)
+        extras['Git'] = gitVersion.replace('git version ', '');
+    const pythonVersion = safeExec('python3 --version 2>/dev/null || python --version 2>/dev/null');
+    if (pythonVersion)
+        extras['Python'] = pythonVersion.replace('Python ', '');
+    return {
+        osType: `${platform} (${os.type()})`,
+        osVersion: os.release(),
+        arch: os.arch(),
+        hostname: os.hostname(),
+        cpuModel: cpus[0]?.model ?? 'unknown',
+        cpuCores: cpus.length,
+        totalMemoryGB: (os.totalmem() / 1024 / 1024 / 1024).toFixed(1),
+        freeMemoryGB: (os.freemem() / 1024 / 1024 / 1024).toFixed(1),
+        username: os.userInfo().username,
+        shell: process.env.SHELL || process.env.COMSPEC || 'unknown',
+        cwd: process.cwd(),
+        nodeVersion: process.version,
+        extras,
+    };
+}
+/** 缓存，只收集一次 */
+let _cached = null;
+/**
+ * 获取格式化的系统信息文本，用于追加到 system prompt
+ *
+ * 返回 markdown 格式的系统环境描述
+ */
+export function getSystemInfoPrompt() {
+    if (_cached)
+        return _cached;
+    const info = collectSystemInfo();
+    const lines = [
+        '\n\n---',
+        '[系统环境] 以下是用户当前的操作系统和硬件信息，可用于辅助判断和生成适配当前环境的命令或代码：',
+        '',
+        `- 操作系统: ${info.osType}`,
+        `- 系统版本: ${info.osVersion}`,
+        `- 架构: ${info.arch}`,
+        `- 主机名: ${info.hostname}`,
+        `- CPU: ${info.cpuModel} (${info.cpuCores} 核)`,
+        `- 内存: ${info.totalMemoryGB} GB (可用 ${info.freeMemoryGB} GB)`,
+        `- 用户: ${info.username}`,
+        `- Shell: ${info.shell}`,
+        `- 工作目录: ${info.cwd}`,
+        `- Node.js: ${info.nodeVersion}`,
+    ];
+    for (const [key, val] of Object.entries(info.extras)) {
+        lines.push(`- ${key}: ${val}`);
+    }
+    _cached = lines.join('\n');
+    return _cached;
+}

package/dist/core/QueryEngine.d.ts

@@ -0,0 +1,52 @@
+import { Message, Session, LoopState } from '../types/index.js';
+import { DangerConfirmResult } from './query.js';
+export interface EngineCallbacks {
+    onMessage: (msg: Message) => void;
+    onUpdateMessage: (id: string, updates: Partial<Message>) => void;
+    onStreamText: (text: string) => void;
+    /** 清空流式文本（每轮迭代结束时调用） */
+    onClearStreamText?: () => void;
+    onLoopStateChange: (state: LoopState) => void;
+    onSessionUpdate: (session: Session) => void;
+    /** 危险命令交互式确认 */
+    onConfirmDangerousCommand?: (command: string, reason: string, ruleName: string) => Promise<DangerConfirmResult>;
+}
+export declare class QueryEngine {
+    private service;
+    private session;
+    private transcript;
+    private abortSignal;
+    constructor();
+    private createSession;
+    private ensureSessionDir;
+    /** 处理用户输入 */
+    handleQuery(userInput: string, callbacks: EngineCallbacks): Promise<void>;
+    /** 终止当前任务 */
+    abort(): void;
+    /** 重置会话 */
+    reset(): void;
+    /**
+     * 切换智能体：持久化选择 + 重建 LLM service + 重置会话
+     */
+    switchAgent(agentName: string): void;
+    /** 保存会话到文件 */
+    private saveSession;
+    /** 列出所有历史会话（按更新时间倒序），返回摘要信息 */
+    static listSessions(): {
+        id: string;
+        summary: string;
+        updatedAt: number;
+        totalTokens: number;
+    }[];
+    /** 恢复指定会话：加载历史消息和 transcript */
+    loadSession(sessionId: string): {
+        session: Session;
+        messages: Message[];
+    } | null;
+    getSession(): Session;
+    /**
+     * 清理非当前会话的所有历史会话文件
+     * @returns 删除的会话数量
+     */
+    clearOtherSessions(): number;
+}

package/dist/core/QueryEngine.js

@@ -0,0 +1,246 @@
+import { v4 as uuid } from 'uuid';
+import fs from 'fs';
+import path from 'path';
+import { getAllTools } from '../tools/index.js';
+import { executeQuery } from './query.js';
+import { MockService } from '../services/api/mock.js';
+import { LLMServiceImpl } from '../services/api/llm.js';
+import { loadConfig, getActiveModel } from '../config/loader.js';
+import { SESSIONS_DIR } from '../config/constants.js';
+import { setActiveAgent } from '../config/agentState.js';
+import { clearAuthorizations } from './safeguard.js';
+export class QueryEngine {
+    service;
+    session;
+    transcript = [];
+    abortSignal = { aborted: false };
+    constructor() {
+        // 尝试从配置文件加载 LLM，失败则回退 MockService
+        const config = loadConfig();
+        const activeModel = getActiveModel(config);
+        if (activeModel) {
+            try {
+                this.service = new LLMServiceImpl();
+            }
+            catch {
+                this.service = new MockService();
+            }
+        }
+        else {
+            this.service = new MockService();
+        }
+        this.session = this.createSession();
+        this.ensureSessionDir();
+    }
+    createSession() {
+        return {
+            id: uuid(),
+            messages: [],
+            createdAt: Date.now(),
+            updatedAt: Date.now(),
+            totalTokens: 0,
+            totalCost: 0,
+        };
+    }
+    ensureSessionDir() {
+        if (!fs.existsSync(SESSIONS_DIR)) {
+            fs.mkdirSync(SESSIONS_DIR, { recursive: true });
+        }
+    }
+    /** 处理用户输入 */
+    async handleQuery(userInput, callbacks) {
+        this.abortSignal = { aborted: false };
+        const userMsg = {
+            id: uuid(),
+            type: 'user',
+            status: 'success',
+            content: userInput,
+            timestamp: Date.now(),
+        };
+        callbacks.onMessage(userMsg);
+        this.session.messages.push(userMsg);
+        const queryCallbacks = {
+            onMessage: (msg) => {
+                callbacks.onMessage(msg);
+                this.session.messages.push(msg);
+            },
+            onUpdateMessage: callbacks.onUpdateMessage,
+            onStreamText: (text) => {
+                // 每收到一个 chunk 视为一个 token，实时递增并通知 UI
+                this.session.totalTokens++;
+                callbacks.onStreamText(text);
+                callbacks.onSessionUpdate(this.session);
+            },
+            onClearStreamText: callbacks.onClearStreamText,
+            onLoopStateChange: callbacks.onLoopStateChange,
+            onConfirmDangerousCommand: callbacks.onConfirmDangerousCommand,
+        };
+        try {
+            this.transcript = await executeQuery(userInput, this.transcript, getAllTools(), this.service, queryCallbacks, this.abortSignal);
+        }
+        catch (err) {
+            const errMsg = {
+                id: uuid(),
+                type: 'error',
+                status: 'error',
+                content: `错误: ${err.message}`,
+                timestamp: Date.now(),
+            };
+            callbacks.onMessage(errMsg);
+        }
+        this.session.updatedAt = Date.now();
+        callbacks.onSessionUpdate(this.session);
+        this.saveSession();
+    }
+    /** 终止当前任务 */
+    abort() {
+        this.abortSignal.aborted = true;
+    }
+    /** 重置会话 */
+    reset() {
+        this.saveSession();
+        this.session = this.createSession();
+        this.transcript = [];
+        clearAuthorizations();
+    }
+    /**
+     * 切换智能体：持久化选择 + 重建 LLM service + 重置会话
+     */
+    switchAgent(agentName) {
+        setActiveAgent(agentName);
+        // 重建 LLM service 以加载新 agent 的 system prompt
+        const config = loadConfig();
+        const activeModel = getActiveModel(config);
+        if (activeModel) {
+            try {
+                this.service = new LLMServiceImpl();
+            }
+            catch {
+                this.service = new MockService();
+            }
+        }
+        else {
+            this.service = new MockService();
+        }
+        // 重置会话上下文
+        this.saveSession();
+        this.session = this.createSession();
+        this.transcript = [];
+    }
+    /** 保存会话到文件 */
+    saveSession() {
+        try {
+            // 自动提取摘要：取首条用户消息前 80 个字符
+            if (!this.session.summary) {
+                const firstUser = this.session.messages.find((m) => m.type === 'user');
+                if (firstUser) {
+                    this.session.summary = firstUser.content.slice(0, 80).replace(/\n/g, ' ');
+                }
+            }
+            const filePath = path.join(SESSIONS_DIR, `${this.session.id}.json`);
+            fs.writeFileSync(filePath, JSON.stringify(this.session, null, 2), 'utf-8');
+        }
+        catch { /* 静默失败 */ }
+    }
+    /** 列出所有历史会话（按更新时间倒序），返回摘要信息 */
+    static listSessions() {
+        try {
+            if (!fs.existsSync(SESSIONS_DIR))
+                return [];
+            const files = fs.readdirSync(SESSIONS_DIR).filter((f) => f.endsWith('.json'));
+            const sessions = [];
+            for (const file of files) {
+                try {
+                    const raw = fs.readFileSync(path.join(SESSIONS_DIR, file), 'utf-8');
+                    const s = JSON.parse(raw);
+                    sessions.push({
+                        id: s.id,
+                        summary: s.summary || '(无摘要)',
+                        updatedAt: s.updatedAt,
+                        totalTokens: s.totalTokens,
+                    });
+                }
+                catch { /* 跳过损坏文件 */ }
+            }
+            // 按更新时间倒序
+            sessions.sort((a, b) => b.updatedAt - a.updatedAt);
+            return sessions;
+        }
+        catch {
+            return [];
+        }
+    }
+    /** 恢复指定会话：加载历史消息和 transcript */
+    loadSession(sessionId) {
+        try {
+            const filePath = path.join(SESSIONS_DIR, `${sessionId}.json`);
+            if (!fs.existsSync(filePath))
+                return null;
+            const raw = fs.readFileSync(filePath, 'utf-8');
+            const loaded = JSON.parse(raw);
+            // 保存当前会话
+            this.saveSession();
+            // 恢复会话状态
+            this.session = loaded;
+            // 从历史消息重建 transcript
+            this.transcript = [];
+            for (const msg of loaded.messages) {
+                if (msg.type === 'user') {
+                    this.transcript.push({ role: 'user', content: msg.content });
+                }
+                else if (msg.type === 'reasoning' && msg.status === 'success' && msg.content) {
+                    // assistant 消息的 content 必须是 ContentBlock[]，与 query.ts 中的构建方式一致
+                    this.transcript.push({
+                        role: 'assistant',
+                        content: [{ type: 'text', text: msg.content }],
+                    });
+                }
+                else if (msg.type === 'tool_exec' && msg.toolName && msg.toolResult !== undefined) {
+                    // 工具调用：assistant tool_use + tool_result
+                    this.transcript.push({
+                        role: 'assistant',
+                        content: [{ type: 'tool_use', id: msg.id, name: msg.toolName, input: msg.toolArgs ?? {} }],
+                    });
+                    this.transcript.push({
+                        role: 'tool_result',
+                        content: msg.toolResult,
+                        toolUseId: msg.id,
+                    });
+                }
+            }
+            return { session: this.session, messages: loaded.messages };
+        }
+        catch {
+            return null;
+        }
+    }
+    getSession() {
+        return this.session;
+    }
+    /**
+     * 清理非当前会话的所有历史会话文件
+     * @returns 删除的会话数量
+     */
+    clearOtherSessions() {
+        try {
+            if (!fs.existsSync(SESSIONS_DIR))
+                return 0;
+            const files = fs.readdirSync(SESSIONS_DIR).filter((f) => f.endsWith('.json'));
+            const currentFile = `${this.session.id}.json`;
+            let count = 0;
+            for (const file of files) {
+                if (file === currentFile)
+                    continue;
+                try {
+                    fs.unlinkSync(path.join(SESSIONS_DIR, file));
+                    count++;
+                }
+                catch { /* 跳过删除失败的文件 */ }
+            }
+            return count;
+        }
+        catch {
+            return 0;
+        }
+    }
+}

package/dist/core/hint.d.ts

@@ -0,0 +1,14 @@
+/**
+ * 智能提示生成器
+ *
+ * 根据当前激活的智能体角色 + 项目上下文，调用 LLM 生成一条
+ * 符合角色设定且贴合当前项目的输入提示信息。
+ */
+/** 获取角色对应的静态兜底提示（同步，可用于初始 placeholder） */
+export declare function getFallbackHint(agentName?: string): string;
+/**
+ * 调用 LLM 生成一条符合当前角色 + 项目上下文的输入提示
+ *
+ * @returns 生成的提示文本，失败时返回静态兜底
+ */
+export declare function generateAgentHint(): Promise<string>;