zhitalk 0.0.0 → 0.0.3

package/dist/agent/tools.js

@@ -0,0 +1,193 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.tools = exports.nativeTools = void 0;
+exports.maybePersistedOutput = maybePersistedOutput;
+exports.initTools = initTools;
+const tools_1 = require("@langchain/core/tools");
+const zod_1 = require("zod");
+const promises_1 = require("fs/promises");
+const path_1 = require("path");
+const config_1 = require("./config");
+const read_file_tool_1 = require("./tools/read_file_tool");
+const write_file_tool_1 = require("./tools/write_file_tool");
+const exec_tool_1 = require("./tools/exec_tool");
+const run_js_tool_1 = require("./tools/run_js_tool");
+const run_py_tool_1 = require("./tools/run_py_tool");
+const web_search_tool_1 = require("./tools/web_search_tool");
+const web_fetch_tool_1 = require("./tools/web_fetch_tool");
+const load_skill_tool_1 = require("./tools/load_skill_tool");
+const memory_create_tool_1 = require("./tools/memory_create_tool");
+const memory_retrieve_tool_1 = require("./tools/memory_retrieve_tool");
+const memory_delete_tool_1 = require("./tools/memory_delete_tool");
+const profile_update_tool_1 = require("./tools/profile_update_tool");
+const agent_tool_1 = require("./tools/agent_tool");
+const mcp_1 = require("./mcp");
+function createZhitalkTool(impl, options, permission_level) {
+    const t = (0, tools_1.tool)(impl, options);
+    t.permission_level = permission_level;
+    return t;
+}
+const readFileTool = createZhitalkTool(read_file_tool_1.readFileTool, {
+    name: 'read_file',
+    description: 'Read the contents of a file.',
+    schema: zod_1.z.object({
+        filepath: zod_1.z.string().describe('The path of the file to read. Can be relative, absolute, or start with ~.'),
+    }),
+}, 'read');
+const writeFileTool = createZhitalkTool(write_file_tool_1.writeFileTool, {
+    name: 'write_file',
+    description: 'Create or overwrite a file. Will create parent directories if needed.',
+    schema: zod_1.z.object({
+        filepath: zod_1.z.string().describe('The path of the file to write. Can be relative, absolute, or start with ~.'),
+        content: zod_1.z.string().describe('The content to write to the file.'),
+    }),
+}, 'write');
+const execTool = createZhitalkTool(exec_tool_1.execTool, {
+    name: 'exec',
+    description: 'Execute a safe shell command in the current directory. Dangerous commands (rm, rmdir, etc.), absolute paths, and parent directory references are blocked.',
+    schema: zod_1.z.object({
+        command: zod_1.z.string().describe('The shell command to execute.'),
+    }),
+}, 'exec');
+const runJsTool = createZhitalkTool(run_js_tool_1.runJsTool, {
+    name: 'run_js',
+    description: 'Execute JavaScript code using Node.js in the current directory. Returns stdout/stderr or error messages.',
+    schema: zod_1.z.object({
+        code: zod_1.z.string().describe('The JavaScript code to execute.'),
+    }),
+}, 'exec');
+const runPyTool = createZhitalkTool(run_py_tool_1.runPyTool, {
+    name: 'run_py',
+    description: 'Execute Python code using Python3 in the current directory. Returns stdout/stderr or error messages.',
+    schema: zod_1.z.object({
+        code: zod_1.z.string().describe('The Python code to execute.'),
+    }),
+}, 'exec');
+const webSearchTool = createZhitalkTool(web_search_tool_1.webSearchTool, {
+    name: 'web_search',
+    description: 'Search the web using Tavily. Useful for finding current information, news, and facts.',
+    schema: zod_1.z.object({
+        query: zod_1.z.string().describe('The search query.'),
+    }),
+}, 'network');
+const webFetchTool = createZhitalkTool(web_fetch_tool_1.webFetchTool, {
+    name: 'web_fetch',
+    description: 'Fetch the content of a web page by URL. Returns the raw HTML/text content. Useful when you need to read a specific page.',
+    schema: zod_1.z.object({
+        url: zod_1.z.string().describe('The full URL of the web page to fetch.'),
+    }),
+}, 'network');
+const loadSkillTool = createZhitalkTool(load_skill_tool_1.loadSkillTool, {
+    name: 'load_skill',
+    description: 'Load the full content of a skill by its name. Call this when you need to use a specific skill to handle the user request. You can only load one skill at a time.',
+    schema: zod_1.z.object({
+        name: zod_1.z.string().describe('The name of the skill to load.'),
+    }),
+}, 'read');
+const memoryCreateTool = createZhitalkTool(memory_create_tool_1.memoryCreateTool, {
+    name: 'memory_create',
+    description: 'Save a piece of memory to the database. Use this when the user shares something worth remembering, such as a personal fact, event, preference, or skill.',
+    schema: zod_1.z.object({
+        type: zod_1.z
+            .enum(['fact', 'event', 'preference', 'skill'])
+            .describe('The type of memory to save.'),
+        content: zod_1.z
+            .string()
+            .describe('The natural language description of the memory.'),
+        keywords: zod_1.z
+            .array(zod_1.z.string())
+            .optional()
+            .describe('Optional keywords for retrieval, as an array of strings.'),
+        importance: zod_1.z
+            .number()
+            .min(1)
+            .max(5)
+            .optional()
+            .describe('Importance level from 1 to 5. Default is 3.'),
+    }),
+}, 'db');
+const memoryRetrieveTool = createZhitalkTool(memory_retrieve_tool_1.memoryRetrieveTool, {
+    name: 'memory_retrieve',
+    description: 'Retrieve relevant memories from the database using full-text search. Use this when the user asks about something that may have been remembered before but is not in the current conversation context. Extract a few key keywords from the question and pass them as the query.',
+    schema: zod_1.z.object({
+        query: zod_1.z
+            .array(zod_1.z.string())
+            .describe('Keywords to search for in memories.'),
+        limit: zod_1.z
+            .number()
+            .min(1)
+            .max(50)
+            .optional()
+            .describe('Maximum number of memories to return. Default is 10.'),
+    }),
+}, 'db');
+const memoryDeleteTool = createZhitalkTool(memory_delete_tool_1.memoryDeleteTool, {
+    name: 'memory_delete',
+    description: 'Delete a memory from the database by its id. Use this when the user wants to forget or remove a specific memory.',
+    schema: zod_1.z.object({
+        id: zod_1.z
+            .number()
+            .int()
+            .positive()
+            .describe('The id of the memory to delete.'),
+    }),
+}, 'db');
+const profileUpdateTool = createZhitalkTool(profile_update_tool_1.profileUpdateTool, {
+    name: 'profile_update',
+    description: "Update the user's profile information. When providing new profile info, also include all other fields mentioned in <profile_info> — all profile information must be updated together.",
+    schema: zod_1.z.object({
+        profile_info: zod_1.z
+            .string()
+            .describe('The complete profile information in markdown format, including all existing and updated fields.'),
+    }),
+}, 'write');
+const agentTool = createZhitalkTool(agent_tool_1.agentTool, {
+    name: 'agent_tool',
+    description: 'Launch a sub-agent to execute an independent task. The sub-agent has access to all tools (except this one), skills, memory, and hooks. It will run to completion and return its final result. Use this for tasks that can be delegated, such as research, code generation, or data processing.',
+    schema: zod_1.z.object({
+        prompt: zod_1.z
+            .string()
+            .describe('A clear, self-contained task prompt for the sub-agent. It should include all necessary context because the sub-agent cannot see the current conversation history.'),
+    }),
+}, 'exec');
+async function maybePersistedOutput(content, toolCallId) {
+    const MAX_LENGTH = 50000;
+    if (content.length <= MAX_LENGTH) {
+        return content;
+    }
+    const id = toolCallId || Math.random().toString(36).slice(2, 9);
+    const dir = (0, path_1.resolve)(config_1.WORKSPACE_DIR, '.tool_output');
+    const filePath = (0, path_1.resolve)(dir, `tool_output_${id}.txt`);
+    await (0, promises_1.mkdir)(dir, { recursive: true });
+    await (0, promises_1.writeFile)(filePath, content, 'utf-8');
+    return `<persisted-output>
+Output too large (${(content.length / 1024).toFixed(1)}KB).
+Full output saved to: ${filePath}
+
+If you need the complete content, it is recommended to read it in segments.
+
+Preview (first 2KB):
+${content.slice(0, 2000)}
+...
+</persisted-output>`;
+}
+exports.nativeTools = [
+    readFileTool,
+    writeFileTool,
+    execTool,
+    runJsTool,
+    runPyTool,
+    webSearchTool,
+    webFetchTool,
+    loadSkillTool,
+    memoryCreateTool,
+    memoryRetrieveTool,
+    memoryDeleteTool,
+    profileUpdateTool,
+    agentTool,
+];
+exports.tools = [...exports.nativeTools];
+async function initTools() {
+    const mcpTools = await (0, mcp_1.initMcpTools)();
+    exports.tools = [...exports.nativeTools, ...mcpTools];
+}

package/dist/agent/utils.d.ts

@@ -0,0 +1,2 @@
+export declare function formatRelativeTime(ts: string): string;
+export declare function truncate(str: string | null, maxLen: number): string;

package/dist/agent/utils.js

@@ -0,0 +1,30 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.formatRelativeTime = formatRelativeTime;
+exports.truncate = truncate;
+function formatRelativeTime(ts) {
+    const diffMs = Date.now() - new Date(ts).getTime();
+    const diffMin = Math.floor(diffMs / 60000);
+    const diffHour = Math.floor(diffMin / 60);
+    const diffDay = Math.floor(diffHour / 24);
+    if (diffMin < 1)
+        return '刚刚';
+    if (diffMin < 5)
+        return '5分钟内';
+    if (diffMin < 10)
+        return '10分钟内';
+    if (diffMin < 30)
+        return '30分钟内';
+    if (diffMin < 60)
+        return '1小时内';
+    if (diffHour < 24)
+        return `${diffHour}小时内`;
+    if (diffDay < 2)
+        return '1天前';
+    return `${diffDay}天前`;
+}
+function truncate(str, maxLen) {
+    if (!str)
+        return '-';
+    return str.length > maxLen ? str.slice(0, maxLen) + '...' : str;
+}

package/dist/index.d.ts

@@ -0,0 +1,2 @@
+#!/usr/bin/env node
+export {};

package/dist/index.js

@@ -0,0 +1,72 @@
+#!/usr/bin/env node
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+const commander_1 = require("commander");
+const fs_1 = require("fs");
+const path_1 = require("path");
+const config_1 = require("./agent/config");
+const pkg = JSON.parse((0, fs_1.readFileSync)((0, path_1.join)(__dirname, '../package.json'), 'utf-8'));
+async function main() {
+    if (!(0, fs_1.existsSync)(config_1.CONFIG_PATH)) {
+        const { runInstall } = await Promise.resolve().then(() => __importStar(require('./install')));
+        await runInstall();
+        return;
+    }
+    const { initColors } = await Promise.resolve().then(() => __importStar(require('./agent/colors')));
+    const { initDb } = await Promise.resolve().then(() => __importStar(require('./agent/db')));
+    const { initAgent } = await Promise.resolve().then(() => __importStar(require('./agent/agent')));
+    const { interactiveChat } = await Promise.resolve().then(() => __importStar(require('./agent/cli')));
+    const { shutdownMcp } = await Promise.resolve().then(() => __importStar(require('./agent/mcp')));
+    const { checkModel } = await Promise.resolve().then(() => __importStar(require('./agent/model')));
+    await initColors();
+    await checkModel();
+    initDb();
+    await initAgent();
+    process.once('SIGINT', async () => {
+        await shutdownMcp();
+        process.exit(0);
+    });
+    const program = new commander_1.Command();
+    program.name(pkg.name).description(pkg.description).version(pkg.version);
+    if (process.argv.length > 2) {
+        program.parse();
+    }
+    else {
+        await interactiveChat();
+    }
+    await shutdownMcp();
+}
+main();

package/dist/install.d.ts

@@ -0,0 +1 @@
+export declare function runInstall(): Promise<void>;

package/dist/install.js

@@ -0,0 +1,108 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.runInstall = runInstall;
+const fs = __importStar(require("fs"));
+const path = __importStar(require("path"));
+const os = __importStar(require("os"));
+const child_process_1 = require("child_process");
+const config_1 = require("./agent/config");
+const db_1 = require("./agent/db");
+const SKILLS_DIR = path.join(config_1.WORKSPACE_DIR, '.agents', 'skills');
+function run(command) {
+    (0, child_process_1.execSync)(command, { stdio: 'inherit' });
+}
+function installSkill(name, repo, skillPath, targetDir, tempDir) {
+    if (fs.existsSync(targetDir)) {
+        console.log(`  ✅ ${name} 已存在，跳过`);
+        return;
+    }
+    console.log(`  📥 正在下载 ${name}...`);
+    const tarPath = path.join(tempDir, `${name}.tar.gz`);
+    const extractDir = path.join(tempDir, `${name}-extract`);
+    fs.mkdirSync(extractDir, { recursive: true });
+    run(`curl -fsSL -o "${tarPath}" "https://github.com/${repo}/archive/refs/heads/main.tar.gz"`);
+    run(`tar -xzf "${tarPath}" -C "${extractDir}"`);
+    const subDir = fs.readdirSync(extractDir)[0];
+    fs.cpSync(path.join(extractDir, subDir, skillPath), targetDir, {
+        recursive: true,
+    });
+    console.log(`  ✅ ${name} 安装完成`);
+}
+async function runInstall() {
+    console.log('🚀 欢迎使用 Zhitalk！首次使用需要进行初始化配置。\n');
+    // 1. 初始化数据库
+    console.log('📦 正在初始化数据库...');
+    (0, db_1.initDb)();
+    console.log('✅ 数据库初始化完成\n');
+    // 2. 安装 skills
+    console.log('🔧 正在安装默认 skills...');
+    fs.mkdirSync(SKILLS_DIR, { recursive: true });
+    const tempDir = fs.mkdtempSync(path.join(os.tmpdir(), 'zhitalk-'));
+    try {
+        installSkill('find-skills', 'vercel-labs/skills', 'skills/find-skills', path.join(SKILLS_DIR, 'find-skills'), tempDir);
+        installSkill('skill-creator', 'anthropics/skills', 'skills/skill-creator', path.join(SKILLS_DIR, 'skill-creator'), tempDir);
+    }
+    finally {
+        fs.rmSync(tempDir, { recursive: true, force: true });
+    }
+    console.log('✅ Skills 安装完成\n');
+    // 3. 创建配置文件
+    console.log('⚙️  正在创建配置文件...');
+    const configTemplate = {
+        model: {
+            model: 'kimi-k2.6',
+            apiKey: 'your-api-key',
+            baseURL: 'https://api.moonshot.cn/v1',
+        },
+        env: {
+            TAVILY_API_KEY: 'your-tavily-api-key',
+        },
+    };
+    fs.mkdirSync(path.dirname(config_1.CONFIG_PATH), { recursive: true });
+    fs.writeFileSync(config_1.CONFIG_PATH, JSON.stringify(configTemplate, null, 2), 'utf-8');
+    console.log(`✅ 配置文件已创建: ${config_1.CONFIG_PATH}\n`);
+    // 4. 提示用户
+    console.log('━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━');
+    console.log('🎉 初始化完成！');
+    console.log('');
+    console.log('⚠️  请编辑配置文件，填入你的 API 密钥：');
+    console.log(`   ${config_1.CONFIG_PATH}`);
+    console.log('');
+    console.log('📖 配置文档: https://zhitalk.chat/#config');
+    console.log('');
+    console.log('🔄 修改完成后，请重新运行 zhitalk');
+    console.log('━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━');
+}

package/package.json

@@ -1,13 +1,50 @@
 {
   "name": "zhitalk",
-  "version": "0.0.0",
-  "description": "",
-  "main": "index.js",
+  "version": "0.0.3",
+  "description": "A personal AI Agent like OpenClaw, including tools, skills, memory, hook, sub-agent, MCP server, etc. Run in the terminal.",
+  "type": "commonjs",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "bin": {
+    "zhitalk": "dist/index.js"
+  },
   "scripts": {
-    "test": "echo \"Error: no test specified\" && exit 1"
+    "fix:perms": "chmod +x dist/index.js",
+    "build": "pnpm clean && tsc && pnpm fix:perms",
+    "dev": "ts-node src/index.ts",
+    "start": "node dist/index.js",
+    "test": "jest",
+    "test:watch": "jest --watch",
+    "test:coverage": "jest --coverage",
+    "clean": "rm -rf dist"
   },
   "keywords": [],
-  "author": "",
+  "author": "双越",
   "license": "ISC",
-  "type": "commonjs"
+  "docs": "https://zhitalk.chat/",
+  "devDependencies": {
+    "@types/jest": "^29.5.12",
+    "@types/node": "^20.11.0",
+    "jest": "^29.7.0",
+    "ts-jest": "^29.1.2",
+    "ts-node": "^10.9.2",
+    "typescript": "^5.3.3"
+  },
+  "dependencies": {
+    "@langchain/core": "^1.1.46",
+    "@langchain/langgraph": "^1.3.0",
+    "@langchain/langgraph-checkpoint-sqlite": "^1.0.1",
+    "@langchain/openai": "^1.4.5",
+    "@langchain/tavily": "^1.2.0",
+    "@modelcontextprotocol/sdk": "^1.29.0",
+    "@types/better-sqlite3": "^7.6.13",
+    "better-sqlite3": "^12.10.0",
+    "boxen": "^8.0.1",
+    "chalk": "^5.6.2",
+    "cli-table3": "^0.6.5",
+    "commander": "^14.0.3",
+    "figlet": "^1.11.0",
+    "langchain": "^1.4.0",
+    "zod": "^4.4.3"
+  }
 }

package/index.js

@@ -1 +0,0 @@
-console.log('hello zhitalk')