@saber2pr/ai-agent 0.0.50 → 0.0.52

package/lib/core/agent-graph.d.ts

@@ -6,9 +6,6 @@ interface TokenUsage {
 }
 declare const AgentState: import("@langchain/langgraph").AnnotationRoot<{
     messages: import("@langchain/langgraph").BinaryOperatorAggregate<BaseMessage[], BaseMessage[]>;
-    auditedFiles: import("@langchain/langgraph").BinaryOperatorAggregate<string[], string[]>;
-    targetCount: import("@langchain/langgraph").BinaryOperatorAggregate<number, number>;
-    mode: import("@langchain/langgraph").BinaryOperatorAggregate<"chat" | "auto", "chat" | "auto">;
     tokenUsage: import("@langchain/langgraph").BinaryOperatorAggregate<TokenUsage, TokenUsage>;
     totalDuration: import("@langchain/langgraph").BinaryOperatorAggregate<number, number>;
 }>;
@@ -24,7 +21,6 @@ export default class McpGraphAgent<T extends AgentGraphModel = any> {
     private alwaysSystem;
     private recursionLimit;
     private apiConfig;
-    private maxTargetCount;
     private maxTokens;
     private mcpClients;
     private streamEnabled;
@@ -66,36 +62,21 @@ export default class McpGraphAgent<T extends AgentGraphModel = any> {
         totalDuration: number;
     }>;
     private getRecentToolCalls;
-    trackProgress(state: typeof AgentState.State): Promise<{
-        auditedFiles: string[];
-    }>;
     private printFinalSummary;
     createGraph(): Promise<import("@langchain/langgraph").CompiledStateGraph<import("@langchain/langgraph").StateType<{
         messages: import("@langchain/langgraph").BinaryOperatorAggregate<BaseMessage[], BaseMessage[]>;
-        auditedFiles: import("@langchain/langgraph").BinaryOperatorAggregate<string[], string[]>;
-        targetCount: import("@langchain/langgraph").BinaryOperatorAggregate<number, number>;
-        mode: import("@langchain/langgraph").BinaryOperatorAggregate<"chat" | "auto", "chat" | "auto">;
         tokenUsage: import("@langchain/langgraph").BinaryOperatorAggregate<TokenUsage, TokenUsage>;
         totalDuration: import("@langchain/langgraph").BinaryOperatorAggregate<number, number>;
     }>, import("@langchain/langgraph").UpdateType<{
         messages: import("@langchain/langgraph").BinaryOperatorAggregate<BaseMessage[], BaseMessage[]>;
-        auditedFiles: import("@langchain/langgraph").BinaryOperatorAggregate<string[], string[]>;
-        targetCount: import("@langchain/langgraph").BinaryOperatorAggregate<number, number>;
-        mode: import("@langchain/langgraph").BinaryOperatorAggregate<"chat" | "auto", "chat" | "auto">;
         tokenUsage: import("@langchain/langgraph").BinaryOperatorAggregate<TokenUsage, TokenUsage>;
         totalDuration: import("@langchain/langgraph").BinaryOperatorAggregate<number, number>;
-    }>, "tools" | "__start__" | "agent" | "progress", {
+    }>, "tools" | "__start__" | "agent", {
         messages: import("@langchain/langgraph").BinaryOperatorAggregate<BaseMessage[], BaseMessage[]>;
-        auditedFiles: import("@langchain/langgraph").BinaryOperatorAggregate<string[], string[]>;
-        targetCount: import("@langchain/langgraph").BinaryOperatorAggregate<number, number>;
-        mode: import("@langchain/langgraph").BinaryOperatorAggregate<"chat" | "auto", "chat" | "auto">;
         tokenUsage: import("@langchain/langgraph").BinaryOperatorAggregate<TokenUsage, TokenUsage>;
         totalDuration: import("@langchain/langgraph").BinaryOperatorAggregate<number, number>;
     }, {
         messages: import("@langchain/langgraph").BinaryOperatorAggregate<BaseMessage[], BaseMessage[]>;
-        auditedFiles: import("@langchain/langgraph").BinaryOperatorAggregate<string[], string[]>;
-        targetCount: import("@langchain/langgraph").BinaryOperatorAggregate<number, number>;
-        mode: import("@langchain/langgraph").BinaryOperatorAggregate<"chat" | "auto", "chat" | "auto">;
         tokenUsage: import("@langchain/langgraph").BinaryOperatorAggregate<TokenUsage, TokenUsage>;
         totalDuration: import("@langchain/langgraph").BinaryOperatorAggregate<number, number>;
     }, import("@langchain/langgraph").StateDefinition>>;

package/lib/core/agent-graph.js

@@ -29,18 +29,6 @@ const AgentState = langgraph_1.Annotation.Root({
         reducer: (x, y) => x.concat(y),
         default: () => [],
     }),
-    auditedFiles: (0, langgraph_1.Annotation)({
-        reducer: (x, y) => Array.from(new Set([...x, ...y])),
-        default: () => [],
-    }),
-    targetCount: (0, langgraph_1.Annotation)({
-        reducer: (x, y) => y !== null && y !== void 0 ? y : x,
-        default: () => 4,
-    }),
-    mode: (0, langgraph_1.Annotation)({
-        reducer: (x, y) => y !== null && y !== void 0 ? y : x,
-        default: () => 'chat',
-    }),
     // ✅ Token 累加器
     tokenUsage: (0, langgraph_1.Annotation)({
         reducer: (x, y) => ({
@@ -67,8 +55,7 @@ class McpGraphAgent {
         this.targetDir = options.targetDir || process.cwd();
         this.recursionLimit = options.recursionLimit || 80;
         this.apiConfig = options.apiConfig;
-        this.maxTargetCount = options.maxTargetCount || 4;
-        this.maxTokens = options.maxTokens || 8000;
+        this.maxTokens = options.maxTokens || 100000;
         this.streamEnabled = options.stream || false;
         process.setMaxListeners(100);
         // ✅ 修复 AbortSignal 监听器数量警告
@@ -283,8 +270,6 @@ class McpGraphAgent {
             const app = await this.createGraph();
             const graphStream = await app.stream({
                 messages: [new messages_1.HumanMessage(query)],
-                mode: 'auto',
-                targetCount: this.maxTargetCount,
             }, {
                 configurable: { thread_id: 'auto_worker' },
                 recursionLimit: this.recursionLimit,
@@ -312,8 +297,6 @@ class McpGraphAgent {
             const app = await this.createGraph();
             const graphStream = await app.stream({
                 messages: [new messages_1.HumanMessage(query)],
-                mode: 'auto',
-                targetCount: this.maxTargetCount,
             }, {
                 configurable: { thread_id: 'stream_worker' },
                 recursionLimit: this.recursionLimit,
@@ -346,7 +329,7 @@ class McpGraphAgent {
                     rl.close();
                     return;
                 }
-                const graphStream = await app.stream({ messages: [new messages_1.HumanMessage(input)], mode: 'chat' }, {
+                const graphStream = await app.stream({ messages: [new messages_1.HumanMessage(input)] }, {
                     configurable: { thread_id: 'session' },
                     recursionLimit: this.recursionLimit,
                     debug: this.verbose,
@@ -416,7 +399,6 @@ class McpGraphAgent {
         }
     }
     async callModel(state) {
-        const auditedListStr = state.auditedFiles.length > 0 ? state.auditedFiles.map(f => `\n  - ${f}`).join('') : '暂无';
         const recentToolCalls = this.getRecentToolCalls(state.messages);
         const recentToolCallsStr = recentToolCalls.length > 0
             ? `\n\n⚠️ 最近调用的工具（避免重复调用相同工具和参数）：\n${recentToolCalls
@@ -425,9 +407,14 @@ class McpGraphAgent {
             : '';
         // 1. 构建当前的系统提示词模板
         const systemPromptTemplate = `你是一个代码专家。工作目录：${this.targetDir}。
-当前模式：{mode}
-进度：{doneCount}/{targetCount}
-已审计文件：{auditedList}
+
+# 🧠 思考要求 (Mandatory Thinking Process)
+在进行任何输出或调用工具之前，你**必须**先进行深度的逻辑思考。请将你的思考过程包裹在 <think> 标签内。
+你的思考应包含：
+1. **当前状态分析**：目前任务进展到哪一步了？
+2. **逻辑推导**：基于已知信息，下一步应该做什么？为什么？
+3. **工具选择建议**：如果需要调用工具，预期通过该工具获得什么结果？
+4. **风险/边界评估**：有哪些潜在的代码缺陷或执行风险需要注意？
 
 # 🛠️ 工具调用规范
 1. Arguments 必须是纯粹的 JSON 对象，严禁将其作为字符串放入引号中。
@@ -458,10 +445,6 @@ class McpGraphAgent {
         try {
             const promptParams = {
                 messages: inputMessages,
-                mode: state.mode,
-                targetCount: state.targetCount,
-                doneCount: state.auditedFiles.length,
-                auditedList: auditedListStr,
                 recentToolCalls: recentToolCallsStr,
                 extraPrompt: this.options.extraSystemPrompt || '',
             };
@@ -611,21 +594,6 @@ class McpGraphAgent {
         }
         return toolCalls;
     }
-    async trackProgress(state) {
-        var _a;
-        const lastAiMsg = state.messages[state.messages.length - 1];
-        const currentAudited = new Set(state.auditedFiles);
-        if ((_a = lastAiMsg === null || lastAiMsg === void 0 ? void 0 : lastAiMsg.tool_calls) === null || _a === void 0 ? void 0 : _a.length) {
-            for (const tc of lastAiMsg.tool_calls) {
-                // 兼容所有可能的路径参数字段
-                const file = tc.args.path || tc.args.filePath || tc.args.file || tc.args.file_path;
-                if (file && typeof file === 'string') {
-                    currentAudited.add(file);
-                }
-            }
-        }
-        return { auditedFiles: Array.from(currentAudited) };
-    }
     printFinalSummary(state) {
         var _a;
         const totalTokens = ((_a = state.tokenUsage) === null || _a === void 0 ? void 0 : _a.total) || 0;
@@ -635,7 +603,6 @@ class McpGraphAgent {
             console.log(`🏁 \x1b[32;1m[审计任务全量结算]\x1b[0m`);
             console.log(`   - 累计消耗总额: \x1b[33m${totalTokens}\x1b[0m Tokens`);
             console.log(`   - 累计执行耗时: \x1b[36m${(totalMs / 1000).toFixed(2)}\x1b[0m s`);
-            console.log(`   - 审计文件总数: ${state.auditedFiles.length} 个`);
             console.log('═'.repeat(50) + '\n');
         }
     }
@@ -643,7 +610,6 @@ class McpGraphAgent {
         const workflow = new langgraph_1.StateGraph(AgentState)
             .addNode('agent', state => this.callModel(state))
             .addNode('tools', this.toolNode)
-            .addNode('progress', state => this.trackProgress(state))
             .addEdge(langgraph_1.START, 'agent')
             .addConditionalEdges('agent', state => {
             const { messages } = state;
@@ -653,6 +619,9 @@ class McpGraphAgent {
             // 如果已消耗 Token 超过了 options 中设置的 maxTokens (假设是总限额)
             if (this.options.maxTokens && state.tokenUsage.total >= this.options.maxTokens) {
                 console.warn('⚠️ [警告] 已达到最大 Token 限制，强制结束任务。');
+                state.messages.push(new messages_1.AIMessage({
+                    content: `⚠️ 抱歉，当前任务消耗的 Token 已达上限 ${this.options.maxTokens}。为了防止无限循环，我已停止执行。你可以尝试分步骤指令，继续发送指令，直到任务完成。`
+                }));
                 this.printFinalSummary(state);
                 return langgraph_1.END;
             }
@@ -660,22 +629,23 @@ class McpGraphAgent {
             if (lastMsg.tool_calls && lastMsg.tool_calls.length > 0) {
                 return 'tools';
             }
-            // 2. 判定结束的条件：
-            // - 模式是 auto 且审计完成
-            // - 或者 AI 明确输出了结束语
-            // - 或者 AI 输出了普通内容且没有工具调用（针对问答模式）
-            const isAutoFinished = state.mode === 'auto' && state.auditedFiles.length > state.targetCount;
             const isFinalAnswer = content.includes('Final Answer');
-            // ✅ 修复核心：如果 AI 只是在聊天（没有工具调用），直接结束，不要跳回 agent
-            if (isAutoFinished || isFinalAnswer || state.mode === 'chat') {
+            if (isFinalAnswer) {
                 this.printFinalSummary(state);
                 return langgraph_1.END;
             }
-            // 兜底：如果是在 auto 模式且还没干完活，才跳回 agent（通常不会走到这里）
+            if (messages.length > 50) {
+                state.messages.push(new messages_1.AIMessage({
+                    content: `💡 **嘿，我们已经聊了很多了！**
+
+为了保证服务质量，我通常会在 50 步操作后停下来和您确认一下。这样可以避免我产生“幻觉”或者做一些无用功。
+
+**任务还没完成吗？** 没关系，只要您回复“继续”，我马上就回来接着干！或者您有什么新的想法要告诉我？`
+                }));
+                return langgraph_1.END;
+            }
             return langgraph_1.END;
-        })
-            .addEdge('tools', 'progress')
-            .addEdge('progress', 'agent');
+        }).addEdge('tools', 'agent');
         return workflow.compile({ checkpointer: this.checkpointer });
     }
 }

package/lib/types/type.d.ts

@@ -7,7 +7,7 @@ export interface ApiConfig {
     model: string;
 }
 export interface ToolInfo {
-    type: "function";
+    type: 'function';
     function: {
         name: string;
         description?: string;
@@ -38,7 +38,6 @@ export interface GraphAgentOptions<T extends AgentGraphModel = any> extends Agen
     apiModel?: T;
     alwaysSystem?: boolean;
     recursionLimit?: number;
-    maxTargetCount?: number;
     /** 是否启用流式输出，默认 false */
     stream?: boolean;
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@saber2pr/ai-agent",
-  "version": "0.0.50",
+  "version": "0.0.52",
   "description": "AI Assistant CLI.",
   "author": "saber2pr",
   "license": "ISC",