zhitalk 0.0.11 → 0.1.0

package/dist/agent/agent.js

@@ -6,7 +6,8 @@ import { DB_PATH } from './db.js';
 import { tools, maybePersistedOutput, initTools } from './tools.js';
 import { compressMessages } from './context.js';
 import { systemPrompt } from './prompt.js';
-import { formatToolLog } from './colors.js';
+import { formatToolLog, formatTodoList } from './colors.js';
+import { processTodoCalls, formatTodoListForPrompt } from './todo.js';
 import { checkReadPermission } from './permission/read.js';
 import { checkWritePermission } from './permission/write.js';
 import { checkExecPermission } from './permission/exec.js';
@@ -33,6 +34,10 @@ const StateAnnotation = Annotation.Root({
         reducer: (_prev, next) => next,
         default: () => 0,
     }),
+    todoList: Annotation({
+        reducer: (_prev, next) => next,
+        default: () => null,
+    }),
 });
 // ── Graph Nodes ───────────────────────────────────────────
 function shouldContinue(state) {
@@ -121,7 +126,11 @@ function createAgentGraph(toolList) {
             }
             return true;
         });
-        const messages = [new SystemMessage(systemPrompt), ...modelMessages];
+        const messages = [new SystemMessage(systemPrompt)];
+        if (state.todoList && state.todoList.length > 0) {
+            messages.push(new SystemMessage(`当前任务进度：\n${formatTodoListForPrompt(state.todoList)}`));
+        }
+        messages.push(...modelMessages);
         const response = await modelWithTheseTools.invoke(messages, config);
         return { messages: [response] };
     }
@@ -240,9 +249,25 @@ function createAgentGraph(toolList) {
                 }
             }));
         }
-        const allowOutputs = await executeCalls(allowCalls);
+        // 分离 todo 工具调用
+        const regularAllowCalls = [];
+        const todoAllowCalls = [];
+        for (const call of allowCalls) {
+            if (call.name === 'create_todo_list' || call.name === 'update_todo_status') {
+                todoAllowCalls.push(call);
+            }
+            else {
+                regularAllowCalls.push(call);
+            }
+        }
+        const regularOutputs = await executeCalls(regularAllowCalls);
+        const todoResult = processTodoCalls(todoAllowCalls, state.todoList);
+        if (todoResult.todoList && todoResult.todoList.length > 0) {
+            console.log(formatTodoList(todoResult.todoList));
+        }
+        const allowOutputs = [...regularOutputs, ...todoResult.messages];
         if (confirmCalls.length === 0) {
-            return { messages: [...allowOutputs, ...blockMessages] };
+            return { messages: [...allowOutputs, ...blockMessages], todoList: todoResult.todoList };
         }
         const result = interrupt({ toolCalls: confirmCalls });
         if (result !== 'approved') {
@@ -253,10 +278,11 @@ function createAgentGraph(toolList) {
             }));
             return {
                 messages: [...allowOutputs, ...blockMessages, ...deniedMessages],
+                todoList: todoResult.todoList,
             };
         }
         const confirmOutputs = await executeCalls(confirmCalls);
-        return { messages: [...allowOutputs, ...blockMessages, ...confirmOutputs] };
+        return { messages: [...allowOutputs, ...blockMessages, ...confirmOutputs], todoList: todoResult.todoList };
     }
     const workflow = new StateGraph(StateAnnotation)
         .addNode('model_request', modelRequest)

package/dist/agent/colors.d.ts

@@ -15,5 +15,11 @@ export declare const color: {
     warn: (text: string) => string;
     info: (text: string) => string;
     good: (text: string) => string;
+    todoTag: () => string;
 };
+export interface TodoDisplayItem {
+    content: string;
+    status: 'pending' | 'in_progress' | 'completed' | 'failed';
+}
+export declare function formatTodoList(items: TodoDisplayItem[]): string;
 export declare function formatToolLog(name: string, detail?: string): string;

package/dist/agent/colors.js

@@ -33,7 +33,25 @@ export const color = {
     warn: (text) => ck().yellow(text),
     info: (text) => ck().blueBright(text),
     good: (text) => ck().greenBright(text),
+    todoTag: () => ck().magentaBright('[Todo]'),
 };
+export function formatTodoList(items) {
+    const lines = [color.todoTag()];
+    for (const item of items) {
+        const symbol = item.status === 'completed'
+            ? color.good('✓')
+            : item.status === 'in_progress'
+                ? color.info('→')
+                : item.status === 'failed'
+                    ? color.error('✗')
+                    : color.gray('○');
+        const desc = item.status === 'completed' || item.status === 'failed'
+            ? color.gray(item.content)
+            : item.content;
+        lines.push(`  ${symbol} ${desc}`);
+    }
+    return '\n' + lines.join('\n') + '\n';
+}
 export function formatToolLog(name, detail) {
     const parts = [color.toolTag(), color.toolName(name), color.toolAction()];
     if (detail) {

package/dist/agent/prompt.js

@@ -47,12 +47,22 @@ ${skillsText}`
 const skillPrompt = `## Skills
 
 ${skillList}\n\nNote: If you want to add a new skill, it must be placed in the ${path.join(WORKSPACE_DIR, 'skills')} directory.`;
+const todoPrompt = `## Task Planning
+
+When a user's request is a complex, multi-step task (such as data analysis, writing a document, planning a task, or any task clearly requiring more than 3 steps), you MUST:
+1. First call \`create_todo_list\` to create a structured plan with clear, actionable steps
+2. Then execute each step one by one
+3. After completing each step, call \`update_todo_status\` to mark it as completed (or failed if it didn't work)
+4. Only provide the final answer after all steps are done
+
+For simple, single-step tasks, do NOT use the todo list tools.`;
 export const systemPrompt = [
     basePrompt,
     profilePrompt,
     memoryPrompt,
     skillPrompt,
     dateTimePrompt,
+    todoPrompt,
 ]
     .filter(Boolean)
     .join('\n\n');

package/dist/agent/todo.d.ts

@@ -0,0 +1,17 @@
+import { ToolMessage } from '@langchain/core/messages';
+export interface TodoItem {
+    id: number;
+    content: string;
+    status: 'pending' | 'in_progress' | 'completed' | 'failed';
+}
+interface TodoCall {
+    id?: string;
+    name: string;
+    args: Record<string, any>;
+}
+export declare function processTodoCalls(calls: TodoCall[], currentTodoList: TodoItem[] | null): {
+    messages: ToolMessage[];
+    todoList: TodoItem[] | null;
+};
+export declare function formatTodoListForPrompt(items: TodoItem[]): string;
+export {};

package/dist/agent/todo.js

@@ -0,0 +1,86 @@
+import { ToolMessage } from '@langchain/core/messages';
+export function processTodoCalls(calls, currentTodoList) {
+    const messages = [];
+    let todoList = currentTodoList;
+    for (const call of calls) {
+        if (call.name === 'create_todo_list') {
+            const items = call.args.items;
+            if (!Array.isArray(items) || items.length === 0) {
+                messages.push(new ToolMessage({
+                    content: 'Error: items must be a non-empty array.',
+                    tool_call_id: call.id ?? '',
+                    name: call.name,
+                }));
+                continue;
+            }
+            todoList = items.map((content, i) => ({
+                id: i,
+                content: typeof content === 'string' ? content.trim() : String(content),
+                status: 'pending',
+            }));
+            messages.push(new ToolMessage({
+                content: 'Todo list created successfully.',
+                tool_call_id: call.id ?? '',
+                name: call.name,
+            }));
+        }
+        else if (call.name === 'update_todo_status') {
+            const index = call.args.index;
+            const status = call.args.status;
+            if (!todoList) {
+                messages.push(new ToolMessage({
+                    content: 'Error: no todo list exists.',
+                    tool_call_id: call.id ?? '',
+                    name: call.name,
+                }));
+                continue;
+            }
+            if (typeof index !== 'number' ||
+                !Number.isInteger(index) ||
+                index < 0 ||
+                index >= todoList.length) {
+                messages.push(new ToolMessage({
+                    content: `Error: index ${index} is out of range.`,
+                    tool_call_id: call.id ?? '',
+                    name: call.name,
+                }));
+                continue;
+            }
+            const validStatuses = [
+                'pending',
+                'in_progress',
+                'completed',
+                'failed',
+            ];
+            if (!validStatuses.includes(status)) {
+                messages.push(new ToolMessage({
+                    content: `Error: invalid status "${status}".`,
+                    tool_call_id: call.id ?? '',
+                    name: call.name,
+                }));
+                continue;
+            }
+            todoList[index].status = status;
+            messages.push(new ToolMessage({
+                content: `Todo item ${index} updated to ${status}.`,
+                tool_call_id: call.id ?? '',
+                name: call.name,
+            }));
+        }
+    }
+    return { messages, todoList };
+}
+export function formatTodoListForPrompt(items) {
+    return items
+        .map((item) => {
+        const statusMark = item.status === 'completed'
+            ? '[x]'
+            : item.status === 'in_progress'
+                ? '[~]'
+                : item.status === 'failed'
+                    ? '[!]'
+                    : '[ ]';
+        return `${statusMark} ${item.content}`;
+    })
+        .join('\n');
+}

package/dist/agent/tools.js

@@ -145,6 +145,35 @@ const agentTool = createZhitalkTool(agentToolImpl, {
             .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 createTodoListDummy() {
+    return 'Error: create_todo_list should be handled internally by the agent.';
+}
+async function updateTodoStatusDummy() {
+    return 'Error: update_todo_status should be handled internally by the agent.';
+}
+const createTodoListTool = createZhitalkTool(createTodoListDummy, {
+    name: 'create_todo_list',
+    description: 'Create a structured todo list for a complex, multi-step task. Must be called BEFORE starting execution. Each item should be a clear, actionable step.',
+    schema: z.object({
+        items: z
+            .array(z.string().describe('A single todo item description.'))
+            .describe('The list of todo items to create. Each item should be a clear, actionable step.'),
+    }),
+}, 'read');
+const updateTodoStatusTool = createZhitalkTool(updateTodoStatusDummy, {
+    name: 'update_todo_status',
+    description: 'Update the status of a todo item in the current todo list. Call this after completing EACH step.',
+    schema: z.object({
+        index: z
+            .number()
+            .int()
+            .min(0)
+            .describe('The zero-based index of the todo item to update.'),
+        status: z
+            .enum(['in_progress', 'completed', 'failed'])
+            .describe('The new status of the todo item.'),
+    }),
+}, 'read');
 export async function maybePersistedOutput(content, toolCallId) {
     const MAX_LENGTH = 50000;
     if (content.length <= MAX_LENGTH) {
@@ -180,6 +209,8 @@ export const nativeTools = [
     memoryDeleteTool,
     profileUpdateTool,
     agentTool,
+    createTodoListTool,
+    updateTodoStatusTool,
 ];
 export let tools = [...nativeTools];
 export async function initTools() {

package/dist/install.js

@@ -5,34 +5,112 @@ import { execSync } from 'child_process';
 import { WORKSPACE_DIR, CONFIG_PATH } from './agent/config.js';
 import { initDb } from './agent/db.js';
 const SKILLS_DIR = path.join(WORKSPACE_DIR, 'skills');
+// 核心 skills
+const CORE_SKILLS = [
+    {
+        name: 'find-skills',
+        repo: 'vercel-labs/skills',
+        path: 'skills/find-skills',
+    },
+    {
+        name: 'skill-creator',
+        repo: 'anthropics/skills',
+        path: 'skills/skill-creator',
+    },
+];
+// 可选的常用 skills，默认一起安装
+const OPTIONAL_SKILLS = [
+    {
+        name: 'canvas-design',
+        repo: 'anthropics/skills',
+        path: 'skills/canvas-design',
+    },
+    { name: 'docx', repo: 'anthropics/skills', path: 'skills/docx' },
+    { name: 'pdf', repo: 'anthropics/skills', path: 'skills/pdf' },
+    { name: 'pptx', repo: 'anthropics/skills', path: 'skills/pptx' },
+    { name: 'xlsx', repo: 'anthropics/skills', path: 'skills/xlsx' },
+    {
+        name: 'frontend-design',
+        repo: 'anthropics/skills',
+        path: 'skills/frontend-design',
+    },
+    {
+        name: 'webapp-testing',
+        repo: 'anthropics/skills',
+        path: 'skills/webapp-testing',
+    },
+];
+const ALL_SKILLS = [...CORE_SKILLS, ...OPTIONAL_SKILLS];
 function run(command, options) {
     execSync(command, { stdio: 'inherit', ...options });
 }
-function installSkill(name, repo, skillPath, targetDir, tempDir) {
-    if (fs.existsSync(targetDir)) {
-        console.log(`  ✅ ${name} 已存在，跳过`);
-        return;
+function installSkills(skills, targetBaseDir) {
+    const installed = [];
+    const skipped = [];
+    const failed = [];
+    // 按 repo 分组，避免重复下载
+    const byRepo = new Map();
+    for (const skill of skills) {
+        const list = byRepo.get(skill.repo) ?? [];
+        list.push(skill);
+        byRepo.set(skill.repo, list);
     }
-    console.log(`  📥 正在下载 ${name}...`);
-    const tarPath = path.join(tempDir, `${name}.tar.gz`);
-    const extractDir = path.join(tempDir, `${name}-extract`);
-    fs.mkdirSync(extractDir, { recursive: true });
-    try {
-        run(`curl -fsSL -o "${tarPath}" "https://github.com/${repo}/archive/refs/heads/main.tar.gz"`, { timeout: 30000 });
-        run(`tar -xzf "${tarPath}" -C "${extractDir}"`, { timeout: 30000 });
-        const subDir = fs.readdirSync(extractDir)[0];
-        fs.cpSync(path.join(extractDir, subDir, skillPath), targetDir, {
-            recursive: true,
-        });
-        console.log(`  ✅ ${name} 安装完成`);
-    }
-    catch (error) {
-        if (error.killed || error.code === 'ETIMEDOUT') {
-            console.log(`  ⏱️  ${name} 安装超时（网络较慢），已跳过，继续下一步`);
-            return;
+    for (const [repo, repoSkills] of byRepo) {
+        const tempDir = fs.mkdtempSync(path.join(os.tmpdir(), 'zhitalk-'));
+        const markFailed = () => {
+            const repoFailed = repoSkills.filter((s) => !fs.existsSync(path.join(targetBaseDir, s.name)));
+            if (repoFailed.length === 0)
+                return;
+            for (const skill of repoFailed) {
+                console.log(`  ❌ ${skill.name} 安装失败，已跳过`);
+                failed.push(skill.name);
+            }
+            console.log(`  ⚠️  ${repo} 下载失败，你可以手动安装：`);
+            console.log(`     1. 下载 https://github.com/${repo}/archive/refs/heads/main.tar.gz`);
+            console.log(`     2. 解压后找到以下目录并拷贝：`);
+            for (const skill of repoFailed) {
+                console.log(`        ${skill.path} → ${path.join(targetBaseDir, skill.name)}`);
+            }
+        };
+        console.log(`  📥 正在下载 ${repo}...`);
+        try {
+            const tarPath = path.join(tempDir, 'repo.tar.gz');
+            run(`curl --connect-timeout 10 --max-time 30 -fsSL -o "${tarPath}" "https://github.com/${repo}/archive/refs/heads/main.tar.gz"`, { timeout: 45000 });
+            run(`tar -xzf "${tarPath}" -C "${tempDir}"`, { timeout: 30000 });
+            const subDir = fs.readdirSync(tempDir).find((d) => d !== 'repo.tar.gz');
+            if (!subDir) {
+                markFailed();
+                continue;
+            }
+            const extractDir = path.join(tempDir, subDir);
+            for (const skill of repoSkills) {
+                const targetDir = path.join(targetBaseDir, skill.name);
+                if (fs.existsSync(targetDir)) {
+                    console.log(`  ✅ ${skill.name} 已存在，跳过`);
+                    skipped.push(skill.name);
+                    continue;
+                }
+                console.log(`  📥 正在安装 ${skill.name}...`);
+                fs.cpSync(path.join(extractDir, skill.path), targetDir, {
+                    recursive: true,
+                });
+                console.log(`  ✅ ${skill.name} 安装完成`);
+                installed.push(skill.name);
+            }
+        }
+        catch {
+            markFailed();
+        }
+        finally {
+            try {
+                fs.rmSync(tempDir, { recursive: true, force: true });
+            }
+            catch {
+                // 忽略清理失败
+            }
         }
-        throw error;
     }
+    return { installed, skipped, failed };
 }
 export async function runInstall() {
     console.log('🚀 欢迎使用 Zhitalk！首次使用需要进行初始化配置。\n');
@@ -41,17 +119,13 @@ export async function runInstall() {
     initDb();
     console.log('✅ 数据库初始化完成\n');
     // 2. 安装 skills
-    console.log('🔧 正在安装默认 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 });
+    const { installed, skipped, failed } = installSkills(ALL_SKILLS, SKILLS_DIR);
+    if (failed.length > 0) {
+        console.log(`\n⚠️  以下 ${failed.length} 个 skill 安装失败：${failed.join(', ')}`);
     }
-    console.log('✅ Skills 安装完成\n');
+    console.log(`\n✅ Skills 安装完成（${installed.length} 个新安装，${skipped.length} 个已存在跳过）\n`);
     // 3. 创建配置文件
     console.log('⚙️  正在创建配置文件...');
     const configTemplate = {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "zhitalk",
-  "version": "0.0.11",
+  "version": "0.1.0",
   "description": "A personal AI Agent like OpenClaw, including tools, skills, memory, hook, sub-agent, MCP server, etc. Run in the terminal.",
   "type": "module",
   "main": "dist/index.js",