shuiyu67andWorkBuddy
9dd418ec89
- 新增 lsp_client.py：LspClient（subprocess stdio，Content-Length 分帧 JSON-RPC， reader 线程，initialize/initialized/didOpen/didChange/hover/definition/references/ completion + publishDiagnostics 通知捕获）+ LspManager（按扩展名→语言→server 路由、 connect_all 读 lspServers、get_context 返回已分析文件诊断、cleanup 幂等） - unified_tool_manager.py：initialize 末尾加载 LSP server，注册 5 个 lsp__* 工具 （diagnostics/hover/definition/references/completion，带真实 inputSchema、protocol='lsp'、 instance 存 manager 供 shutdown 清理） - cli_core.py：_build_system_prompt 末尾注入 LSP 诊断上下文（模型持续看到真实诊断） - config.json：顶层加 lspServers 段（python/rust 示例，含 _说明；该文件被 gitignore 不入库） - 新增 tests/test_lsp_client.py（9 项：fake server 全链路+路由+桥接+fc_tools 保真+缺失/失败优雅降级） - 修 LSP 线缆坑：didOpen/didChange 是通知（无响应），初版误当请求等响应导致超时 Co-Authored-By: WorkBuddy <workbuddy@tencent.com>
1610 lines
87 KiB
Python
1610 lines
87 KiB
Python
import os
|
||
import sys
|
||
import json
|
||
import re
|
||
import time
|
||
import shutil
|
||
import threading
|
||
import random
|
||
from typing import Optional
|
||
from colorama import Fore, Style
|
||
|
||
from .constants import TEXTUAL_AVAILABLE, LOVE_FILE_PATH, DEFAULT_MAX_HISTORY, VERSION
|
||
from .config import get_system_config, set_system_config, logger
|
||
from .cli_base import BaseAICLI
|
||
from .safety import get_safety, MODE_UNRESTRICTED, MODE_NORMAL, MODE_MANUAL
|
||
from .cli_clawli import ClawliMixin
|
||
from .cli_tools import ToolMixin
|
||
from .cli_code_exec import CodeExecMixin
|
||
from .cli_display import DisplayMixin
|
||
from .cli_history import HistoryMixin
|
||
from .plugin_market import PluginMarketMixin
|
||
from .notification import notify_task_complete, get_notification_manager
|
||
from .process_protection import get_protection_status
|
||
|
||
# ── Plan 模式 system prompt 追加段（只读规划约束）──
|
||
PLAN_MODE_SYSTEM_APPEND = """
|
||
|
||
【PLAN 模式 — 只读规划】
|
||
你当前处于 PLAN 模式。此模式下你只能进行调查、阅读、分析，不得执行任何修改。
|
||
- 你可以使用只读工具调研代码库：code_search（搜索/结构/符号/统计）、file_manager 的 list/read/info/search、code_editor 的 read_range/find/diff、git_tools 的 status/log/diff/show/blame、tool_search（查询工具用法）、ai_search（联网搜索）。
|
||
- 你**不能**执行写操作（编辑/创建/删除文件、运行 shell 命令、git commit/push/checkout、委派会修改的子代理等）——这些会被系统拦截并返回提示。
|
||
- 调研完成后，输出一份清晰的、带编号步骤的「实施计划」，让用户审批。计划应包含：每步要做什么、涉及哪些文件/函数、预期结果、潜在风险。
|
||
- 计划以如下标记开头：## 实施计划
|
||
- 完成计划后停止，等待用户输入 /build 批准执行，或继续追问以完善计划。不要自行开始修改。"""
|
||
|
||
|
||
class AICLI(BaseAICLI, ClawliMixin, ToolMixin, CodeExecMixin, DisplayMixin, HistoryMixin, PluginMarketMixin):
|
||
"""AI CLI主程序 - 组合所有 Mixin"""
|
||
|
||
# ── 深度思考处理 ──
|
||
|
||
def process_thinking_response(self, response):
|
||
"""处理深度思考AI返回的内容, 分离思考和回复内容"""
|
||
try:
|
||
response_data = json.loads(response)
|
||
if isinstance(response_data, dict) and response_data.get('action') == 'use_tool':
|
||
return response
|
||
if isinstance(response_data, dict):
|
||
if 'answer' in response_data:
|
||
extracted_content = response_data['answer']
|
||
elif 'response' in response_data:
|
||
extracted_content = response_data['response']
|
||
elif 'content' in response_data:
|
||
extracted_content = response_data['content']
|
||
elif 'text' in response_data:
|
||
extracted_content = response_data['text']
|
||
else:
|
||
return json.dumps(response_data, ensure_ascii=False, indent=2)
|
||
return self._process_content_with_thinking(extracted_content)
|
||
except json.JSONDecodeError:
|
||
return self._process_content_with_thinking(response)
|
||
|
||
def _process_content_with_thinking(self, content):
|
||
"""处理包含思考标记的内容"""
|
||
thinking_start_marker = getattr(self.current_engine, 'thinking_start_marker', None) if self.current_engine else None
|
||
thinking_end_marker = getattr(self.current_engine, 'thinking_end_marker', None) if self.current_engine else None
|
||
|
||
if '<think/>' in content:
|
||
parts = content.split('<think/>')
|
||
if len(parts) >= 2:
|
||
thinking_content = parts[0].strip()
|
||
reply_content = parts[1].strip()
|
||
if thinking_content:
|
||
print(f"{Fore.LIGHTBLACK_EX}{Style.DIM}[思考: {thinking_content}]{Style.RESET_ALL}")
|
||
return reply_content if reply_content else ""
|
||
|
||
if thinking_start_marker and thinking_end_marker and thinking_start_marker in content and thinking_end_marker in content:
|
||
parts = content.split(thinking_start_marker)
|
||
if len(parts) > 1:
|
||
remaining = parts[1]
|
||
if thinking_end_marker in remaining:
|
||
thinking_part, reply_part = remaining.split(thinking_end_marker, 1)
|
||
thinking_content = thinking_part.strip()
|
||
reply_content = reply_part.strip()
|
||
if thinking_content:
|
||
print(f"{Fore.LIGHTBLACK_EX}{Style.DIM}[思考: {thinking_content}]{Style.RESET_ALL}")
|
||
return reply_content if reply_content else ""
|
||
|
||
return content
|
||
|
||
# ── 系统提示构建 ──
|
||
|
||
def _build_system_prompt(self, liugin_prompts=None):
|
||
"""构建系统提示词 - 增强版，支持编程任务"""
|
||
base_prompt = """你叫小狸，是一个强大的智能编程助手。你具备以下核心能力：
|
||
|
||
1. **精准代码编辑** - 使用 code_editor 工具进行搜索替换、多文件批量编辑（修改后自动检查语法）
|
||
2. **代码搜索与理解** - 使用 code_search 工具跨文件搜索、提取符号、分析依赖
|
||
3. **Git 版本控制** - 使用 git_tools 工具管理代码版本
|
||
4. **Shell 命令执行** - 使用 cmd_executor 执行系统命令
|
||
5. **文件管理** - 使用 file_manager 管理文件和目录
|
||
6. **浏览器自动化** - 使用 browser_auto 工具操控浏览器
|
||
7. **子 Agent 协作** - 使用 sub_agent 创建子 Agent 并行处理任务
|
||
8. **工程化自动化** - 使用 auto_engineer 进行 lint/test/build
|
||
9. **网络工具** - 使用 network_tools 进行 HTTP 请求和网络诊断
|
||
10. **AI 联网搜索** - 使用 ai_search 搜索实时信息
|
||
|
||
你在说话时可以适当加上'nyan'来表现可爱性格，但不要过度使用。"""
|
||
|
||
has_tool_search = any(t.get('name') == 'tool_search' for t in self.liugin_manager.tools)
|
||
|
||
tool_search_rule = ""
|
||
if has_tool_search:
|
||
tool_search_rule = """【⚠️ 强制工具查询规则 — 必须遵守】
|
||
在调用任何工具之前，你必须先使用 tool_search 查询该工具的详细用法、参数格式和示例。
|
||
这是硬性要求，没有例外。即使你认为自己知道工具的用法，也必须先查询确认。
|
||
|
||
【⚠️ 工具调用格式 — 必须使用 JSON】
|
||
所有工具都必须通过 JSON 指令调用，格式如下：
|
||
{{"action": "use_tool", "tool": "工具名", "args": "操作 参数"}}
|
||
|
||
禁止直接写 "工具名 操作 参数"，必须包在 JSON 里。
|
||
例如：
|
||
✅ 正确: {{"action": "use_tool", "tool": "code_editor", "args": "find . keyword *.py"}}
|
||
❌ 错误: code_editor find . keyword *.py
|
||
❌ 错误: code_editor("find . keyword *.py")
|
||
|
||
执行流程：
|
||
第一步：用 tool_search 查询工具用法
|
||
{{"action": "use_tool", "tool": "tool_search", "args": "你要用的工具名"}}
|
||
第二步：阅读 tool_search 返回的详细说明（包括参数格式、操作列表、示例）
|
||
第三步：根据返回的说明，用 JSON 格式正确调用工具
|
||
|
||
为什么要这样做：
|
||
- 工具的参数格式可能已更新，你的记忆可能过时
|
||
- 先查询可以避免参数格式错误导致的调用失败
|
||
- 确保你使用的是最优的操作和参数组合
|
||
- 减少因格式不对而反复重试的浪费
|
||
|
||
示例流程（修改代码）：
|
||
用户: "帮我修复 src/app.py 第 10 行的 bug"
|
||
你的思考过程:
|
||
1. 需要先查看代码 → 先查 code_editor 怎么读取文件
|
||
2. {{"action": "use_tool", "tool": "tool_search", "args": "code_editor"}} → 发现 read_range 操作
|
||
3. {{"action": "use_tool", "tool": "code_editor", "args": "read_range src/app.py 5 15"}} → 拿到代码
|
||
4. 需要修改代码 → 已经在第一步查过 code_editor，知道 edit 格式
|
||
5. {{"action": "use_tool", "tool": "code_editor", "args": "edit src/app.py
|
||
旧代码
|
||
<<<>>>
|
||
新代码"}}
|
||
|
||
示例流程（Git 操作）：
|
||
用户: "帮我提交代码"
|
||
你的思考过程:
|
||
1. 需要 git 操作 → 先查 git_tools 怎么用
|
||
2. {{"action": "use_tool", "tool": "tool_search", "args": "git_tools"}} → 发现 smart-commit 操作
|
||
3. {{"action": "use_tool", "tool": "git_tools", "args": "smart-commit"}} → 自动提交"""
|
||
|
||
if liugin_prompts:
|
||
prompt = f"""{base_prompt}
|
||
|
||
{liugin_prompts}
|
||
|
||
{tool_search_rule}
|
||
|
||
【工具调用格式】
|
||
在回复中包含 JSON 指令来执行操作：
|
||
|
||
1. 纯 JSON: {{"action": "use_tool", "tool": "code_editor", "args": "edit file.py old <<<>>> new"}}
|
||
2. 文本+JSON: 好的，我来修改。{{"action": "use_tool", "tool": "code_editor", "args": "edit file.py old <<<>>> new"}}
|
||
|
||
【编程工作流 — 每一步都必须先查询工具】
|
||
1. 理解需求 → 先 tool_search "code_editor"，再用 structure 查看项目结构
|
||
2. 定位代码 → 已查询过，直接用 find 或 regex 搜索
|
||
3. 查看上下文 → 用 read_range 查看相关代码
|
||
4. 精准修改 → 用 edit 替换代码（old <<<>>> new 分隔）
|
||
5. 语法自动检查 → 编辑操作后自动执行，通过显示 ✓，失败显示详细错误
|
||
6. 验证变更 → 先 tool_search "git_tools"，再用 diff 查看变更
|
||
7. 提交代码 → 已查询过，用 commit 提交
|
||
|
||
【语法检查 — 自动 + 主动】
|
||
- 编辑操作(edit/multi/insert/delete_lines/create/write/append)完成后自动检查语法
|
||
- 通过: 显示 " 语法检查通过 ✓"
|
||
- 失败: 显示具体错误（行号、列号、错误描述）
|
||
- 主动检查: syntax_check <文件路径> 或 check <文件路径>
|
||
- 支持语言: Python, JavaScript, TypeScript, JSON, YAML, TOML, HTML, XML, CSS, Shell, SQL
|
||
- 如果语法检查失败，根据错误信息修复代码后再次编辑
|
||
|
||
【Diff 弹窗 — 自动弹出修改对比】
|
||
- 编辑操作完成后自动显示修改前后的完整内容和行号对比
|
||
- 新增行用绿色标记，删除行用红色标记
|
||
- 支持两种模式: 弹窗模式（新终端窗口）/ 主终端内显示（SSH 兼容）
|
||
- 开关: diff_popup on / diff_popup off / diff_popup popup / diff_popup inline
|
||
- 默认开启，无需手动调用
|
||
|
||
【code_editor 编辑格式】
|
||
edit 操作使用 <<<>>> 分隔旧代码和新代码：
|
||
{{"action": "use_tool", "tool": "code_editor", "args": "edit path/to/file.py
|
||
def old_function():
|
||
pass
|
||
<<<>>>
|
||
def new_function():
|
||
return True"}}
|
||
|
||
【批量编辑】
|
||
multi 操作支持一次修改多处：
|
||
{{"action": "use_tool", "tool": "code_editor", "args": "multi path/to/file.py [{{\\"old\\": \\"old1\\", \\"new\\": \\"new1\\"}}, {{\\"old\\": \\"old2\\", \\"new\\": \\"new2\\"}}]"}}
|
||
|
||
【搜索代码】
|
||
- 关键词搜索: {{"action": "use_tool", "tool": "code_search", "args": "find . keyword *.py"}}
|
||
- 正则搜索: {{"action": "use_tool", "tool": "code_search", "args": "regex . def\\\\s+\\\\w+ *.py"}}
|
||
- 查看结构: {{"action": "use_tool", "tool": "code_search", "args": "structure ."}}
|
||
- 查找调用: {{"action": "use_tool", "tool": "code_search", "args": "callers . function_name"}}
|
||
|
||
【Git 操作】
|
||
- 查看状态: {{"action": "use_tool", "tool": "git_tools", "args": "status"}}
|
||
- 查看变更: {{"action": "use_tool", "tool": "git_tools", "args": "diff"}}
|
||
- 提交代码: {{"action": "use_tool", "tool": "git_tools", "args": "commit 描述信息"}}
|
||
- 查看历史: {{"action": "use_tool", "tool": "git_tools", "args": "log --oneline 10"}}
|
||
|
||
【重要提醒】
|
||
- 调用任何不熟悉的工具前，必须先 tool_search 查询
|
||
- code_editor.edit 的 old_text 必须与文件中的内容完全匹配（包括缩进）
|
||
- 如果 edit 失败，先用 read_range 查看实际内容再重试
|
||
- args 必须是字符串
|
||
- 可以一次返回多个 JSON 指令
|
||
- 使用 continue 补充回答: {{"action": "continue", "content": "补充内容"}}"""
|
||
else:
|
||
prompt = f"""{base_prompt}
|
||
|
||
{tool_search_rule}
|
||
|
||
【工具调用格式】
|
||
{{"action": "use_tool", "tool": "工具名", "args": "参数"}}
|
||
|
||
【⚠️ 核心规则：调用任何工具前必须先查询】
|
||
1. 先调用 tool_search 查询工具用法
|
||
2. 阅读返回的参数说明和示例
|
||
3. 再根据说明正确调用工具
|
||
|
||
【核心工具】
|
||
- tool_search: 查询任何工具的用法（必须先调用这个）
|
||
- code_editor: edit/search/diff/insert/create/write/read_range
|
||
- code_search: find/regex/symbols/imports/structure/stats
|
||
- git_tools: status/diff/log/add/commit/branch
|
||
- cmd_executor: 执行 shell 命令
|
||
- file_manager: 文件管理
|
||
|
||
【code_editor.edit 格式】
|
||
{{"action": "use_tool", "tool": "code_editor", "args": "edit 文件路径
|
||
旧代码内容
|
||
<<<>>>
|
||
新代码内容"}}
|
||
|
||
【continue 格式】
|
||
{{"action": "continue", "content": "要补充的内容"}}"""
|
||
|
||
# ── 注入持久化记忆上下文 ──
|
||
if hasattr(self, 'memory_manager') and self.memory_manager:
|
||
try:
|
||
memory_ctx = self.memory_manager.build_memory_context()
|
||
if memory_ctx:
|
||
prompt += memory_ctx
|
||
except Exception:
|
||
pass
|
||
|
||
# ── Plan 模式：追加只读规划约束 ──
|
||
if getattr(self, 'plan_mode', False):
|
||
prompt += PLAN_MODE_SYSTEM_APPEND
|
||
|
||
# ── LSP 诊断自动注入：让模型持续看到真实语言诊断 ──
|
||
liugin_manager = getattr(self, 'liugin_manager', None)
|
||
if liugin_manager and hasattr(liugin_manager, 'get_lsp_context'):
|
||
try:
|
||
lsp_ctx = liugin_manager.get_lsp_context()
|
||
if lsp_ctx:
|
||
prompt += "\n\n# 当前代码诊断（来自 LSP 语言服务器）\n" + lsp_ctx
|
||
except Exception:
|
||
pass
|
||
|
||
return prompt
|
||
|
||
# ── JSON 解析 ──
|
||
|
||
def _clean_json_response(self, response):
|
||
"""清理AI响应中的Markdown代码块标记"""
|
||
if response.startswith("```json"):
|
||
response = response[7:]
|
||
elif response.startswith("```"):
|
||
response = response[3:]
|
||
if response.endswith("```"):
|
||
response = response[:-3]
|
||
return response.strip()
|
||
|
||
def get_liugin_usage_prompts(self):
|
||
"""获取可用工具提示词"""
|
||
if not self.liugin_manager.tools:
|
||
return "当前没有可用的工具."
|
||
|
||
prompts = []
|
||
prompts.append("可用的工具列表：\n")
|
||
|
||
plugins = []
|
||
skills = []
|
||
for tool in self.liugin_manager.tools:
|
||
protocol = tool.get('protocol', 'liugin')
|
||
if protocol == 'skill':
|
||
skills.append(tool)
|
||
else:
|
||
plugins.append(tool)
|
||
|
||
if plugins:
|
||
prompts.append("## 内置工具（使用 tool_search 查询详细用法）：")
|
||
for tool in plugins:
|
||
tool_name = tool.get('name', '')
|
||
tool_desc = tool.get('description', '')
|
||
prompts.append(f"- {tool_name}: {tool_desc}")
|
||
|
||
if skills:
|
||
prompts.append("\n## 扩展技能（已加载详细指令）：")
|
||
for tool in skills:
|
||
tool_name = tool.get('name', '')
|
||
tool_desc = tool.get('description', '')
|
||
prompts.append(f"\n### {tool_name}")
|
||
prompts.append(f"描述: {tool_desc}")
|
||
instance = tool.get('instance')
|
||
if instance:
|
||
instructions = getattr(instance, 'instructions', '')
|
||
if instructions:
|
||
if len(instructions) > 2000:
|
||
instructions = instructions[:2000] + "\n...(内容已截断)"
|
||
prompts.append(f"\n{instructions}")
|
||
|
||
if self.is_clawli_mode:
|
||
prompts.append("\n## Clawli 远程模式专属工具：")
|
||
prompts.append("""
|
||
### send_image - 发送图片到手机端
|
||
当用户请求发送图片到手机时，使用此工具：
|
||
|
||
```json
|
||
{
|
||
"action": "use_tool",
|
||
"tool": "send_image",
|
||
"args": {
|
||
"image_path": "图片文件的绝对路径",
|
||
"caption": "图片说明文字（可选）"
|
||
}
|
||
}
|
||
```
|
||
|
||
支持的图片格式：JPG、PNG、GIF、BMP、WEBP
|
||
""")
|
||
|
||
return "\n".join(prompts)
|
||
|
||
def _parse_mixed_response(self, response):
|
||
"""
|
||
解析混合响应，处理同时包含文本和JSON指令的内容
|
||
返回格式: (text_content, json_data)
|
||
"""
|
||
response = response.strip()
|
||
|
||
# 方法0: 处理 <tool_call>... 格式
|
||
if response.startswith("<tool_call>"):
|
||
json_str = response[11:]
|
||
json_str = "{" + json_str
|
||
try:
|
||
parsed = json.loads(json_str)
|
||
if isinstance(parsed, dict):
|
||
return "", parsed
|
||
except json.JSONDecodeError:
|
||
pass
|
||
|
||
# 方法1: 直接解析为纯JSON
|
||
try:
|
||
parsed = json.loads(response)
|
||
if isinstance(parsed, (list, dict)):
|
||
return "", parsed
|
||
except json.JSONDecodeError:
|
||
pass
|
||
|
||
# 方法2: 修复JSON中的控制字符问题
|
||
try:
|
||
def fix_json_string(match):
|
||
content = match.group(1)
|
||
content = content.replace('\\', '\\\\')
|
||
content = content.replace('\n', '\\n')
|
||
content = content.replace('\r', '\\r')
|
||
content = content.replace('\t', '\\t')
|
||
content = content.replace('"', '\\"')
|
||
return f'"{content}"'
|
||
fixed_response = re.sub(r'"([^"]*(?:\n[^"]*)*)"', fix_json_string, response, flags=re.DOTALL)
|
||
parsed = json.loads(fixed_response)
|
||
if isinstance(parsed, (list, dict)):
|
||
return "", parsed
|
||
except:
|
||
pass
|
||
|
||
# 方法3: 查找代码块中的JSON
|
||
json_pattern = r"```(?:json)?\s*({.*?})\s*```"
|
||
matches = re.findall(json_pattern, response, re.DOTALL)
|
||
if matches:
|
||
try:
|
||
json_data = json.loads(matches[-1])
|
||
if isinstance(json_data, (dict, list)):
|
||
text_content = re.sub(json_pattern, "", response, flags=re.DOTALL).strip()
|
||
text_content = re.sub(r"\n\s*\n", "\n\n", text_content).strip()
|
||
return text_content, json_data
|
||
except json.JSONDecodeError:
|
||
pass
|
||
|
||
# 方法4: 查找每行独立的JSON对象
|
||
lines_resp = response.split("\n")
|
||
json_objects = []
|
||
text_lines = []
|
||
for line in lines_resp:
|
||
line = line.strip()
|
||
if line.startswith("{") and line.endswith("}"):
|
||
try:
|
||
parsed = json.loads(line)
|
||
if isinstance(parsed, dict):
|
||
json_objects.append(parsed)
|
||
continue
|
||
except json.JSONDecodeError:
|
||
pass
|
||
text_lines.append(line)
|
||
|
||
if json_objects:
|
||
text_content = "\n".join(text_lines).strip()
|
||
if len(json_objects) == 1:
|
||
return text_content, json_objects[0]
|
||
else:
|
||
return text_content, json_objects
|
||
|
||
# 方法5: 查找内联的JSON对象
|
||
json_start_markers = [
|
||
'{"action": "use_tool"', '{"action": "continue"',
|
||
'{"action":"use_tool"', '{"action":"continue"',
|
||
'{ "action": "use_tool"', '{ "action": "continue"',
|
||
'{ "action":"use_tool"', '{ "action":"continue"'
|
||
]
|
||
for marker in json_start_markers:
|
||
json_start = response.find(marker)
|
||
if json_start != -1:
|
||
brace_count = 0
|
||
in_json = False
|
||
for i in range(json_start, len(response)):
|
||
char = response[i]
|
||
if char == '{':
|
||
if not in_json:
|
||
in_json = True
|
||
brace_count += 1
|
||
elif char == '}':
|
||
brace_count -= 1
|
||
if in_json and brace_count == 0:
|
||
json_str = response[json_start:i+1]
|
||
try:
|
||
json_data = json.loads(json_str)
|
||
if isinstance(json_data, dict) and json_data.get('action') in ['use_tool', 'continue']:
|
||
text_content = response[:json_start].strip()
|
||
return text_content, json_data
|
||
except json.JSONDecodeError:
|
||
pass
|
||
break
|
||
|
||
# 方法6: 通用JSON对象查找
|
||
potential_jsons = re.findall(r'\{[^{}]*(?:\{[^{}]*\}[^{}]*)*\}', response, re.DOTALL)
|
||
for json_str in potential_jsons:
|
||
try:
|
||
json_data = json.loads(json_str)
|
||
if isinstance(json_data, dict) and json_data.get('action') in ['use_tool', 'continue']:
|
||
json_start = response.find(json_str)
|
||
text_content = response[:json_start].strip()
|
||
return text_content, json_data
|
||
except json.JSONDecodeError:
|
||
continue
|
||
|
||
# 方法7: 批量JSON
|
||
try:
|
||
cleaned = response.strip()
|
||
for action in ['use_tool', 'continue']:
|
||
last_brace_pos = cleaned.rfind(f'{{"action": "{action}"')
|
||
if last_brace_pos == -1:
|
||
last_brace_pos = cleaned.rfind(f'{{"action":"{action}"')
|
||
if last_brace_pos != -1:
|
||
json_part = cleaned[last_brace_pos:]
|
||
if json_part.startswith('['):
|
||
parsed = json.loads(json_part)
|
||
if isinstance(parsed, list):
|
||
text_content = cleaned[:last_brace_pos].strip()
|
||
return text_content, parsed
|
||
else:
|
||
parsed = json.loads(json_part)
|
||
if isinstance(parsed, dict):
|
||
text_content = cleaned[:last_brace_pos].strip()
|
||
return text_content, parsed
|
||
except:
|
||
pass
|
||
|
||
return response, None
|
||
|
||
# ── 对话处理 ──
|
||
|
||
def process_conversation(self, user_input):
|
||
"""处理完整的对话循环, 支持工具调用循环和AI继续操作循环"""
|
||
self.shared_conversation_history.append({
|
||
"role": "user",
|
||
"content": user_input
|
||
})
|
||
liugin_prompts = self.get_liugin_usage_prompts()
|
||
loop_count = 0
|
||
current_input = user_input
|
||
max_loops = 20
|
||
|
||
while loop_count < max_loops:
|
||
response = self._generate_response_with_animation(current_input, liugin_prompts=liugin_prompts)
|
||
processed_response = self.process_thinking_response(response)
|
||
processed_response = self._process_code_blocks(processed_response)
|
||
|
||
# ── Function Calling 响应检测 ──
|
||
fc_handled = self._handle_fc_response(processed_response, user_input)
|
||
if fc_handled:
|
||
# FC 工具已执行，用结果继续循环
|
||
current_input = fc_handled
|
||
loop_count += 1
|
||
continue
|
||
|
||
try:
|
||
text_content, json_data = self._parse_mixed_response(processed_response)
|
||
|
||
if json_data:
|
||
if text_content:
|
||
text_content = self._process_code_blocks(text_content)
|
||
self.shared_conversation_history.append({
|
||
"role": "assistant",
|
||
"content": text_content
|
||
})
|
||
self._display_response(text_content)
|
||
|
||
if isinstance(json_data, dict):
|
||
response_data = json_data
|
||
|
||
if response_data.get('action') == 'use_tool':
|
||
canceled = self._execute_single_tool(response_data)
|
||
if canceled:
|
||
break
|
||
break
|
||
|
||
if (response_data.get('continue') is True or
|
||
response_data.get('need_continue') is True or
|
||
response_data.get('think_more') is True):
|
||
continue_message = response_data.get('message', '')
|
||
if continue_message:
|
||
processed_response = continue_message
|
||
self.shared_conversation_history.append({
|
||
"role": "assistant",
|
||
"content": processed_response
|
||
})
|
||
self._display_response(processed_response, is_continue=True)
|
||
loop_count += 1
|
||
elif response_data.get('message'):
|
||
message = response_data.get('message')
|
||
if text_content:
|
||
text_content = self._process_code_blocks(text_content)
|
||
self.shared_conversation_history.append({
|
||
"role": "assistant",
|
||
"content": text_content
|
||
})
|
||
self._display_response(text_content)
|
||
self.shared_conversation_history.append({
|
||
"role": "assistant",
|
||
"content": message
|
||
})
|
||
self._display_response(message)
|
||
break
|
||
else:
|
||
if text_content:
|
||
text_content = self._process_code_blocks(text_content)
|
||
self.shared_conversation_history.append({
|
||
"role": "assistant",
|
||
"content": text_content
|
||
})
|
||
self._display_response(text_content)
|
||
else:
|
||
self.shared_conversation_history.append({
|
||
"role": "assistant",
|
||
"content": processed_response
|
||
})
|
||
self._display_response(processed_response)
|
||
break
|
||
|
||
elif isinstance(json_data, list):
|
||
tool_calls = [item for item in json_data if isinstance(item, dict) and item.get('action') == 'use_tool']
|
||
if tool_calls:
|
||
self._execute_concurrent_tools(tool_calls)
|
||
break
|
||
continue_items = [item for item in json_data if isinstance(item, dict) and
|
||
(item.get('continue') is True or
|
||
item.get('need_continue') is True or
|
||
item.get('think_more') is True)]
|
||
if continue_items:
|
||
continue_message = continue_items[0].get('message', '')
|
||
if continue_message:
|
||
processed_response = continue_message
|
||
self.shared_conversation_history.append({
|
||
"role": "assistant",
|
||
"content": processed_response
|
||
})
|
||
self._display_response(processed_response, is_continue=True)
|
||
loop_count += 1
|
||
else:
|
||
if text_content:
|
||
self.shared_conversation_history.append({
|
||
"role": "assistant",
|
||
"content": text_content
|
||
})
|
||
self._display_response(text_content or processed_response)
|
||
break
|
||
else:
|
||
display_content = text_content if text_content else processed_response
|
||
self.shared_conversation_history.append({
|
||
"role": "assistant",
|
||
"content": display_content
|
||
})
|
||
self._display_response(display_content)
|
||
break
|
||
except json.JSONDecodeError:
|
||
text_content, json_data = self._parse_mixed_response(processed_response)
|
||
if json_data:
|
||
if text_content:
|
||
text_content = self._process_code_blocks(text_content)
|
||
self.shared_conversation_history.append({
|
||
"role": "assistant",
|
||
"content": text_content
|
||
})
|
||
self._display_response(text_content)
|
||
|
||
if isinstance(json_data, dict):
|
||
response_data = json_data
|
||
if response_data.get('action') == 'continue':
|
||
continue_content = response_data.get('content', '')
|
||
if continue_content:
|
||
self.shared_conversation_history.append({
|
||
"role": "assistant",
|
||
"content": continue_content
|
||
})
|
||
self._display_response(continue_content)
|
||
continue
|
||
|
||
if response_data.get('action') == 'use_tool':
|
||
canceled = self._execute_single_tool(response_data)
|
||
if canceled:
|
||
break
|
||
break
|
||
|
||
if (response_data.get('continue') is True or
|
||
response_data.get('need_continue') is True or
|
||
response_data.get('think_more') is True):
|
||
continue_message = response_data.get('message', '')
|
||
if continue_message:
|
||
processed_response = continue_message
|
||
self.shared_conversation_history.append({
|
||
"role": "assistant",
|
||
"content": processed_response
|
||
})
|
||
self._display_response(processed_response, is_continue=True)
|
||
loop_count += 1
|
||
else:
|
||
message = response_data.get('message', '')
|
||
if message:
|
||
self.shared_conversation_history.append({
|
||
"role": "assistant",
|
||
"content": message
|
||
})
|
||
self._display_response(message)
|
||
loop_count += 1
|
||
else:
|
||
if text_content:
|
||
self.shared_conversation_history.append({
|
||
"role": "assistant",
|
||
"content": text_content
|
||
})
|
||
self._display_response(text_content or processed_response)
|
||
break
|
||
|
||
elif isinstance(json_data, list):
|
||
tool_calls = [item for item in json_data if isinstance(item, dict) and item.get('action') == 'use_tool']
|
||
if tool_calls:
|
||
self._execute_concurrent_tools(tool_calls)
|
||
break
|
||
continue_items = [item for item in json_data if isinstance(item, dict) and
|
||
(item.get('continue') is True or
|
||
item.get('need_continue') is True or
|
||
item.get('think_more') is True)]
|
||
if continue_items:
|
||
continue_message = continue_items[0].get('message', '')
|
||
if continue_message:
|
||
processed_response = continue_message
|
||
self.shared_conversation_history.append({
|
||
"role": "assistant",
|
||
"content": processed_response
|
||
})
|
||
self._display_response(processed_response, is_continue=True)
|
||
loop_count += 1
|
||
else:
|
||
if text_content:
|
||
self.shared_conversation_history.append({
|
||
"role": "assistant",
|
||
"content": text_content
|
||
})
|
||
self._display_response(text_content or processed_response)
|
||
break
|
||
else:
|
||
display_content = text_content if text_content else processed_response
|
||
self.shared_conversation_history.append({
|
||
"role": "assistant",
|
||
"content": display_content
|
||
})
|
||
self._display_response(display_content)
|
||
break
|
||
else:
|
||
self._output(f"AI 已达到最大循环次数 ({max_loops})，已停止自动处理")
|
||
self.shared_conversation_history.append({
|
||
"role": "assistant",
|
||
"content": f"已达到最大处理次数 ({max_loops})，已自动停止。如需继续处理，请重新输入。"
|
||
})
|
||
|
||
# ── 任务完成通知 ──
|
||
task_summary = self._extract_task_summary(user_input)
|
||
notify_task_complete(task_summary)
|
||
|
||
# ── 自动保存聊天记录 ──
|
||
if hasattr(self, 'memory_manager') and self.memory_manager:
|
||
try:
|
||
self.memory_manager.auto_save_chat(
|
||
self.shared_conversation_history,
|
||
label=user_input[:30]
|
||
)
|
||
except Exception:
|
||
pass
|
||
|
||
# ── 自动压缩上下文（token 感知优先，消息数兜底）──
|
||
try:
|
||
if hasattr(self, 'memory_manager') and self.memory_manager:
|
||
engine = self.current_engine
|
||
prompt_tokens = getattr(engine, 'last_prompt_tokens', None) if engine else None
|
||
ctx = None
|
||
if engine and hasattr(engine, 'context_window'):
|
||
try:
|
||
ctx = engine.context_window()
|
||
except Exception:
|
||
ctx = None
|
||
ratio = get_system_config('compress_ratio', 0.7)
|
||
self.memory_manager.set_compress_ratio(ratio)
|
||
if self.memory_manager.should_compress(
|
||
self.shared_conversation_history, prompt_tokens, ctx, ratio
|
||
):
|
||
before = len(self.shared_conversation_history)
|
||
self.shared_conversation_history = self.memory_manager.compress_context(
|
||
self.shared_conversation_history,
|
||
engine=engine
|
||
)
|
||
token_info = f"，token {prompt_tokens}/{ctx}" if (prompt_tokens and ctx) else ""
|
||
self._output(
|
||
f"{Fore.CYAN}🗜 上下文已自动压缩 (原 {before} 条 → {len(self.shared_conversation_history)} 条{token_info}，阈值 {ratio}){Style.RESET_ALL}"
|
||
)
|
||
except Exception as e:
|
||
logger.warning(f"自动压缩上下文失败: {e}")
|
||
|
||
def _extract_task_summary(self, user_input: str) -> str:
|
||
"""提取任务摘要用于通知显示"""
|
||
# 取用户输入的前 80 字符作为摘要
|
||
summary = user_input.strip().replace('\n', ' ')
|
||
if len(summary) > 80:
|
||
summary = summary[:77] + "..."
|
||
return summary
|
||
|
||
def _handle_fc_response(self, response: str, original_input: str) -> Optional[str]:
|
||
"""
|
||
处理 Function Calling 响应
|
||
如果响应是 FC 格式，执行工具并返回结果（供下一轮循环使用）
|
||
如果不是 FC 响应，返回 None
|
||
"""
|
||
if not response or '_fc' not in response:
|
||
return None
|
||
|
||
try:
|
||
data = json.loads(response)
|
||
if not data.get("_fc"):
|
||
return None
|
||
|
||
tool_calls = data.get("tool_calls", [])
|
||
if not tool_calls:
|
||
return None
|
||
|
||
# 记录 assistant 的 FC 消息到历史
|
||
fc_history_entry = {
|
||
"role": "assistant",
|
||
"content": None,
|
||
"tool_calls": [{
|
||
"id": tc.get("id", ""),
|
||
"type": "function",
|
||
"function": {
|
||
"name": tc.get("name", ""),
|
||
"arguments": json.dumps(tc.get("arguments", {}), ensure_ascii=False)
|
||
}
|
||
} for tc in tool_calls]
|
||
}
|
||
self.shared_conversation_history.append(fc_history_entry)
|
||
|
||
# 执行每个工具调用
|
||
results = []
|
||
for tc in tool_calls:
|
||
tool_name = tc.get("tool", tc.get("name", ""))
|
||
tool_args = tc.get("args", "")
|
||
call_id = tc.get("id", "")
|
||
|
||
self._output(f"[green] FC 调用: {tool_name}({tool_args[:60]})[/]")
|
||
|
||
# 通过插件管理器执行
|
||
result = self._execute_tool_by_name(tool_name, tool_args)
|
||
|
||
# 记录工具结果到历史（官方 OpenAI tool 消息不含 name 字段）
|
||
tool_result_msg = {
|
||
"role": "tool",
|
||
"tool_call_id": call_id,
|
||
"content": result[:5000] if result else "执行完成"
|
||
}
|
||
self.shared_conversation_history.append(tool_result_msg)
|
||
|
||
results.append(f"[{tool_name}]: {result[:200] if result else '完成'}")
|
||
|
||
# 返回工具结果，供下一轮 AI 处理
|
||
return "工具执行结果:\n" + "\n".join(results)
|
||
|
||
except (ValueError, KeyError, TypeError):
|
||
return None
|
||
|
||
def _execute_tool_by_name(self, tool_name: str, tool_args: str) -> str:
|
||
"""根据名称执行工具（FC 调用路径）
|
||
|
||
注意：插件 handler 可能返回 ToolResult / dict / str，
|
||
这里统一归一成 str，保证调用方能安全做切片与拼接。
|
||
"""
|
||
from .tool_result import normalize_tool_text
|
||
|
||
# 安全检查
|
||
safety = get_safety()
|
||
allowed, msg = safety.check(tool_name, tool_args)
|
||
if not allowed:
|
||
return f"操作被拒绝: {msg}"
|
||
|
||
# 查找工具
|
||
tool = self.liugin_manager.get_tool_by_name(tool_name)
|
||
if tool and 'handler' in tool:
|
||
try:
|
||
return normalize_tool_text(tool['handler'](tool_args), default="执行完成")
|
||
except Exception as e:
|
||
return f"工具执行失败: {e}"
|
||
|
||
# memory 内置工具
|
||
if tool_name == 'memory' and hasattr(self, 'memory_manager'):
|
||
return normalize_tool_text(self.memory_manager.handle_tool(tool_args), default="执行完成")
|
||
|
||
# scheduler 内置工具
|
||
if tool_name == 'scheduler':
|
||
for t in self.liugin_manager.tools:
|
||
if t.get('name') == 'scheduler' and 'handler' in t:
|
||
try:
|
||
return normalize_tool_text(t['handler'](tool_args), default="执行完成")
|
||
except Exception as e:
|
||
return f"工具执行失败: {e}"
|
||
|
||
return f"未找到工具: {tool_name}"
|
||
|
||
def _generate_response_with_animation(self, current_input, liugin_prompts=None):
|
||
"""生成AI响应并显示等待动画 — 支持 ESC 跨平台取消"""
|
||
# TUI 模式：不做动画
|
||
if self.tui_output_callback:
|
||
system_prompt = self._build_system_prompt(liugin_prompts)
|
||
if not self.current_engine:
|
||
return "错误: 当前没有可用的AI引擎，请检查ai_engines目录中的引擎插件"
|
||
# TUI 模式也支持 FC
|
||
fc_tools = None
|
||
try:
|
||
from .fc_tools import mcp_to_openai_tools
|
||
fc_tools = mcp_to_openai_tools(self.liugin_manager)
|
||
except Exception:
|
||
pass
|
||
response = self.current_engine.generate_response(current_input, system_prompt=system_prompt, tools=fc_tools)
|
||
return response
|
||
|
||
# CLI 模式：带动画和跨平台 ESC 取消
|
||
animation_running = threading.Event()
|
||
animation_running.set()
|
||
esc_pressed = threading.Event()
|
||
response_ready = threading.Event()
|
||
response_result = [None] # 用列表存储，方便线程内修改
|
||
|
||
love_sentences = []
|
||
love_file_path = os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), LOVE_FILE_PATH)
|
||
if os.path.exists(love_file_path):
|
||
try:
|
||
with open(love_file_path, 'r', encoding='utf-8') as f:
|
||
love_sentences = [line.strip() for line in f.readlines() if line.strip()]
|
||
except Exception:
|
||
pass
|
||
if not love_sentences:
|
||
love_sentences = ["AI正在思考中...", "请稍等片刻...", "正在处理您的请求..."]
|
||
|
||
# ── 跨平台 ESC 检测 ──
|
||
def check_for_esc():
|
||
"""跨平台 ESC 键检测"""
|
||
import platform
|
||
if platform.system() == 'Windows':
|
||
try:
|
||
import msvcrt
|
||
while animation_running.is_set():
|
||
if msvcrt.kbhit():
|
||
key = msvcrt.getch()
|
||
if ord(key) == 27: # ESC
|
||
esc_pressed.set()
|
||
break
|
||
time.sleep(0.05)
|
||
except ImportError:
|
||
pass
|
||
else:
|
||
# Linux / macOS — 用 select 做非阻塞读取
|
||
import select
|
||
import sys
|
||
try:
|
||
# 保存原始终端设置
|
||
import tty, termios
|
||
fd = sys.stdin.fileno()
|
||
old_settings = termios.tcgetattr(fd)
|
||
try:
|
||
tty.setcbreak(fd) # cbreak 模式：无需回车即可读取
|
||
while animation_running.is_set():
|
||
if select.select([sys.stdin], [], [], 0.05)[0]:
|
||
ch = sys.stdin.read(1)
|
||
if ord(ch) == 27: # ESC
|
||
esc_pressed.set()
|
||
break
|
||
finally:
|
||
termios.tcsetattr(fd, termios.TCSADRAIN, old_settings)
|
||
except Exception:
|
||
# 降级：无法检测 ESC，只能等 AI 完成
|
||
pass
|
||
|
||
# ── 动画线程 ──
|
||
def show_animation():
|
||
last_change_time = time.time()
|
||
current_sentence = random.choice(love_sentences)
|
||
while animation_running.is_set():
|
||
current_time = time.time()
|
||
if current_time - last_change_time >= 5:
|
||
current_sentence = random.choice(love_sentences)
|
||
last_change_time = current_time
|
||
animation_chars = "|/-\\"
|
||
char_idx = int((current_time * 10) % len(animation_chars))
|
||
display_sentence = current_sentence[:47] + "..." if len(current_sentence) > 50 else current_sentence
|
||
print(f"\r{Fore.MAGENTA}{display_sentence} {animation_chars[char_idx]}{Style.RESET_ALL}", end="", flush=True)
|
||
time.sleep(0.1)
|
||
|
||
# ── AI 响应线程 ──
|
||
def generate_in_thread():
|
||
try:
|
||
system_prompt = self._build_system_prompt(liugin_prompts)
|
||
if not self.current_engine:
|
||
response_result[0] = "错误: 当前没有可用的AI引擎，请检查ai_engines目录中的引擎插件"
|
||
else:
|
||
# ── 收集 FC 工具定义 ──
|
||
fc_tools = None
|
||
try:
|
||
from .fc_tools import mcp_to_openai_tools
|
||
fc_tools = mcp_to_openai_tools(self.liugin_manager)
|
||
if fc_tools:
|
||
fc_tools = fc_tools # 已经是列表
|
||
except Exception:
|
||
fc_tools = None
|
||
|
||
response_result[0] = self.current_engine.generate_response(
|
||
current_input, system_prompt=system_prompt, tools=fc_tools
|
||
)
|
||
except Exception as e:
|
||
response_result[0] = f"AI 响应生成失败: {e}"
|
||
finally:
|
||
response_ready.set()
|
||
|
||
# 启动三个线程
|
||
threads = []
|
||
for target in (show_animation, check_for_esc, generate_in_thread):
|
||
t = threading.Thread(target=target, daemon=True)
|
||
t.start()
|
||
threads.append(t)
|
||
|
||
try:
|
||
# 主线程等待：ESC 或 响应完成
|
||
while not response_ready.is_set() and not esc_pressed.is_set():
|
||
time.sleep(0.1)
|
||
|
||
# ESC 被按下 → 取消
|
||
if esc_pressed.is_set():
|
||
# 清除动画行
|
||
print("\r" + " " * 80 + "\r", end="", flush=True)
|
||
print(f"\n{Fore.YELLOW}⚡ AI 请求已取消 (ESC){Style.RESET_ALL}")
|
||
return "[AI请求已取消]"
|
||
|
||
# 响应正常返回
|
||
response = response_result[0]
|
||
if response:
|
||
# 清除动画行后再输出
|
||
print("\r" + " " * 80 + "\r", end="", flush=True)
|
||
self._typeprint(response, Fore.CYAN)
|
||
|
||
if response and self.shared_conversation_history is not None:
|
||
self.shared_conversation_history.append({
|
||
"role": "assistant",
|
||
"content": response
|
||
})
|
||
|
||
return response
|
||
|
||
finally:
|
||
animation_running.clear()
|
||
for t in threads:
|
||
t.join(timeout=1)
|
||
|
||
# ── Plan 模式（只读规划 → 审批 → 执行）──
|
||
|
||
def _sync_current_plan(self):
|
||
"""Plan 模式下，把对话历史中最后一条 assistant 内容记为当前计划。"""
|
||
if not getattr(self, 'plan_mode', False):
|
||
return
|
||
for msg in reversed(self.shared_conversation_history):
|
||
if isinstance(msg, dict) and msg.get("role") == "assistant" and msg.get("content"):
|
||
self.current_plan = msg["content"]
|
||
break
|
||
|
||
def handle_plan_command(self, args):
|
||
"""处理 /plan 命令。进入 PLAN 模式（只读规划）。
|
||
|
||
/plan 进入模式，等待描述任务
|
||
/plan <任务> 进入模式并立即只读调研、生成实施计划
|
||
/plan off|exit 退出模式（计划不执行）
|
||
"""
|
||
args = (args or "").strip()
|
||
if args in ('off', 'exit', '退出', 'cancel'):
|
||
self.plan_mode = False
|
||
print(f"{Fore.YELLOW}已退出 PLAN 模式（计划未执行）{Style.RESET_ALL}")
|
||
return
|
||
|
||
self.plan_mode = True
|
||
if args:
|
||
print(f"{Fore.CYAN}已进入 PLAN 模式，正在只读调研并生成实施计划...{Style.RESET_ALL}")
|
||
self.process_conversation(args)
|
||
self._autosave_session()
|
||
self._sync_current_plan()
|
||
else:
|
||
print(f"{Fore.CYAN}已进入 PLAN 模式。{Style.RESET_ALL}")
|
||
print(f" 描述你的任务，AI 将只做只读调研并给出实施计划；"
|
||
f"完成后用 {Fore.WHITE}/build{Style.RESET_ALL} 批准执行，或 "
|
||
f"{Fore.WHITE}/plan off{Style.RESET_ALL} 取消。{Style.RESET_ALL}")
|
||
|
||
def handle_build_command(self):
|
||
"""处理 /build 命令。从 PLAN 模式进入执行（用已批准计划驱动）。"""
|
||
plan = getattr(self, 'current_plan', '') or ""
|
||
if not getattr(self, 'plan_mode', False) and not plan.strip():
|
||
print(f"{Fore.YELLOW}当前不在 PLAN 模式，且无已生成的计划{Style.RESET_ALL}")
|
||
return
|
||
|
||
self.plan_mode = False
|
||
if not plan.strip():
|
||
print(f"{Fore.YELLOW}尚未生成实施计划，先 /plan 描述任务让 AI 调研{Style.RESET_ALL}")
|
||
return
|
||
|
||
print(f"{Fore.GREEN}已批准计划，开始执行...{Style.RESET_ALL}")
|
||
instruction = ("【已批准的实施计划，请现在严格按照以下步骤执行，利用可用工具完成每一步；"
|
||
"遇到与计划不符的情况先说明再继续】\n\n" + plan)
|
||
self.process_conversation(instruction)
|
||
self._autosave_session()
|
||
|
||
# ── CLI 主循环 ──
|
||
|
||
def run_tui(self):
|
||
"""运行 TUI 模式（v8.0 默认真·全屏 textual TUI；textual 不可用时回退 rich 面板）"""
|
||
if TEXTUAL_AVAILABLE:
|
||
from .tui import XiaoliTUI
|
||
app = XiaoliTUI(self)
|
||
app.run()
|
||
return
|
||
# 回退 rich 面板 TUI（textual 不可用时）
|
||
try:
|
||
from .rich_tui import RichTUI
|
||
from xcli_core.constants import VERSION as _V
|
||
tui = RichTUI(self)
|
||
tui.version = _V
|
||
tui.set_status(
|
||
engine=getattr(self.current_engine, 'name', '') if self.current_engine else '',
|
||
model=getattr(self.current_engine, 'model', '') if self.current_engine else '',
|
||
mode='TUI',
|
||
)
|
||
self.tui_output_callback = tui.push
|
||
tui.run()
|
||
self.tui_output_callback = None
|
||
except ImportError:
|
||
print(f"{Fore.RED}TUI 不可用: 请安装 textual (pip install textual){Style.RESET_ALL}")
|
||
|
||
def run(self):
|
||
"""运行 CLI"""
|
||
import uuid
|
||
self.user_id = str(uuid.getnode())
|
||
current_engine_name = getattr(self.current_engine, 'name', '未设置') if self.current_engine else '未设置'
|
||
model_name = getattr(self.current_engine, 'model', '') if self.current_engine else ''
|
||
engine_desc = f"{current_engine_name} ({model_name})" if model_name else current_engine_name
|
||
|
||
print(f"{Fore.GREEN}小狸 Pro-CLI v{VERSION} 已启动!{Style.RESET_ALL} "
|
||
f"{Fore.WHITE}引擎: {engine_desc}{Style.RESET_ALL}")
|
||
|
||
# 一行速览，完整命令清单收进 /help
|
||
print(f"{Fore.CYAN}/help 全部命令 · /quit 退出 · /tui 图形界面 · "
|
||
f"/model 换模型 · /resume 恢复会话 · /plan 规划模式 · @文件路径 读文件给 AI{Style.RESET_ALL}")
|
||
|
||
# 历史会话提示（opencode 式 /resume）
|
||
try:
|
||
sessions = self.session_manager.list_sessions()
|
||
if sessions:
|
||
latest = sessions[0]
|
||
print(f"{Fore.YELLOW}有 {len(sessions)} 个历史会话，/resume 可恢复"
|
||
f"（最近: {latest.id} · {latest.title}）{Style.RESET_ALL}")
|
||
except Exception:
|
||
pass
|
||
|
||
extras = []
|
||
loaded_engines = list(self.engines.keys())
|
||
if len(loaded_engines) > 1:
|
||
extras.append(f"可切换引擎: {', '.join(loaded_engines)}")
|
||
# 进程保护：只在有未生效项时才提示，全绿就闭嘴
|
||
if getattr(self, 'protection_results', None):
|
||
r = self.protection_results
|
||
failed = [n for k, n in (('single_instance', '单实例'), ('priority', '优先级'),
|
||
('ppl_protection', 'PPL保护'), ('watchdog', '看门狗'))
|
||
if not r.get(k)]
|
||
if failed:
|
||
extras.append(f"进程保护未生效: {', '.join(failed)}")
|
||
if extras:
|
||
print(f"{Fore.BLACK}{Style.BRIGHT}{' · '.join(extras)}{Style.RESET_ALL}")
|
||
|
||
print(f"{Fore.BLACK}{Style.BRIGHT}用户ID: {self.user_id}{Style.RESET_ALL}")
|
||
|
||
# 恢复 diff 显示模式（首次运行才询问）
|
||
self._ask_diff_popup_mode()
|
||
|
||
self._run_cli_loop()
|
||
|
||
def _get_code_editor_plugin(self):
|
||
"""拿到 code_editor 插件实例（找不到返回 None）"""
|
||
for tool in self.liugin_manager.tools:
|
||
if tool.get('name') == 'code_editor':
|
||
return getattr(tool.get('handler'), '__self__', None)
|
||
return None
|
||
|
||
def _ask_diff_popup_mode(self):
|
||
"""设置 diff 显示模式：已保存过就静默沿用，只有首次运行才交互询问"""
|
||
plugin_instance = self._get_code_editor_plugin()
|
||
if plugin_instance is None:
|
||
return
|
||
|
||
saved = get_system_config('diff_popup_mode', None)
|
||
if saved is not None:
|
||
plugin_instance.diff_popup_mode = bool(saved)
|
||
return
|
||
|
||
print(f"\n{Fore.CYAN} Diff 显示模式（AI 修改文件后的对比方式）:{Style.RESET_ALL}")
|
||
print(f" {Fore.WHITE}1{Style.RESET_ALL} - 弹窗模式（新终端窗口显示）")
|
||
print(f" {Fore.WHITE}2{Style.RESET_ALL} - 主终端内显示（SSH 兼容，推荐）")
|
||
|
||
try:
|
||
choice = input(f"\n{Fore.WHITE} 请选择 [1/2]（默认 2，之后可用 /diff 改）: {Style.RESET_ALL}").strip()
|
||
except (EOFError, KeyboardInterrupt):
|
||
choice = '2'
|
||
|
||
self._set_diff_popup_mode(choice == '1', plugin_instance)
|
||
print()
|
||
|
||
def _set_diff_popup_mode(self, popup, plugin_instance=None):
|
||
"""写入并持久化 diff 显示模式"""
|
||
if plugin_instance is None:
|
||
plugin_instance = self._get_code_editor_plugin()
|
||
if plugin_instance is not None:
|
||
plugin_instance.diff_popup_mode = bool(popup)
|
||
set_system_config('diff_popup_mode', bool(popup))
|
||
label = '弹窗模式' if popup else '主终端内显示'
|
||
print(f"{Fore.GREEN} ✓ Diff 显示模式: {label}{Style.RESET_ALL}")
|
||
|
||
def _run_cli_loop(self):
|
||
"""运行 CLI 主循环"""
|
||
while True:
|
||
try:
|
||
user_input = input(f"\n{Fore.WHITE}> {Style.RESET_ALL}").strip()
|
||
|
||
if user_input == '/quit':
|
||
print(f"{Fore.GREEN}再见!{Style.RESET_ALL}")
|
||
break
|
||
|
||
if user_input == '/help':
|
||
self.show_help()
|
||
continue
|
||
|
||
if user_input.startswith('/help '):
|
||
liugin_name = user_input[6:].strip()
|
||
self.show_liugin_help(liugin_name)
|
||
continue
|
||
|
||
if user_input.startswith('/engine '):
|
||
engine_command = user_input[8:].strip()
|
||
parts = engine_command.split(' ', 1)
|
||
cmd = parts[0]
|
||
args = parts[1] if len(parts) > 1 else ''
|
||
if cmd in self.engine_commands:
|
||
self.engine_commands[cmd](args)
|
||
else:
|
||
available_commands = ', '.join(self.engine_commands.keys())
|
||
print(f"{Fore.RED}未知的引擎命令.可用命令: {available_commands}{Style.RESET_ALL}")
|
||
continue
|
||
|
||
if user_input.startswith('/engine.'):
|
||
engine_command = user_input[8:].strip()
|
||
if not self.handle_engine_command(engine_command):
|
||
print(f"{Fore.RED}当前引擎不支持该命令或命令执行失败{Style.RESET_ALL}")
|
||
continue
|
||
|
||
# /model 命令族 — OpenAI 引擎多模型在线增删切换
|
||
if user_input == '/model' or user_input.startswith('/model '):
|
||
self.handle_model_command(user_input[7:].strip())
|
||
continue
|
||
|
||
# /plan 进入 PLAN 模式（只读规划 → 审批 → 执行）
|
||
if user_input == '/plan' or user_input.startswith('/plan '):
|
||
sub = user_input[6:].strip() if user_input.startswith('/plan ') else ''
|
||
self.handle_plan_command(sub)
|
||
continue
|
||
if user_input == '/build':
|
||
self.handle_build_command()
|
||
continue
|
||
|
||
# /resume 会话恢复（opencode 式自动持久化 + 一键恢复）
|
||
if user_input in ('/resume', '/sessions') or \
|
||
user_input.startswith('/resume ') or user_input.startswith('/sessions ') or \
|
||
user_input.startswith('/session '):
|
||
if user_input.startswith('/resume'):
|
||
self.handle_resume_command(user_input[7:].strip())
|
||
elif user_input.startswith('/sessions'):
|
||
self.handle_resume_command(user_input[9:].strip())
|
||
else: # /session <子命令>
|
||
sub = user_input[8:].strip()
|
||
self.handle_resume_command(sub if sub != 'new' else 'new')
|
||
continue
|
||
|
||
# /manual 快捷命令 — 切换到 manual 引擎或执行 manual 子命令
|
||
if user_input == '/manual' or user_input.startswith('/manual '):
|
||
args = user_input[8:].strip() if user_input.startswith('/manual ') else ''
|
||
if 'manual' not in self.engines:
|
||
print(f"{Fore.RED}manual 引擎未加载{Style.RESET_ALL}")
|
||
elif not args:
|
||
# 无参数：切换到 manual 引擎
|
||
self.switch_engine('manual')
|
||
print(f"{Fore.GREEN}已切换到 manual 引擎{Style.RESET_ALL}")
|
||
else:
|
||
# 有参数：执行 manual 引擎命令（如 status, restart）
|
||
result = self.engines['manual'].handle_command(args)
|
||
if result:
|
||
print(f"{Fore.GREEN}{result}{Style.RESET_ALL}")
|
||
continue
|
||
|
||
if user_input.startswith('/file.read '):
|
||
args = user_input[11:].strip()
|
||
parts = args.split(' ', 2)
|
||
if len(parts) < 1:
|
||
print(f"{Fore.RED}请提供文件名.用法: /file.read <文件名> [行数]{Style.RESET_ALL}")
|
||
continue
|
||
filename = parts[0]
|
||
max_lines = None
|
||
if len(parts) >= 2:
|
||
try:
|
||
max_lines = int(parts[1])
|
||
if max_lines <= 0:
|
||
print(f"{Fore.RED}行数必须是正整数{Style.RESET_ALL}")
|
||
continue
|
||
except ValueError:
|
||
print(f"{Fore.RED}行数必须是正整数{Style.RESET_ALL}")
|
||
continue
|
||
result = self._read_file_direct(filename, max_lines)
|
||
print(f"{Fore.GREEN}{result}{Style.RESET_ALL}")
|
||
continue
|
||
|
||
if user_input.startswith('/chat '):
|
||
chat_command = user_input[6:].strip()
|
||
if chat_command.startswith('save '):
|
||
name = chat_command[5:].strip()
|
||
if name:
|
||
self.save_chat_history(name)
|
||
else:
|
||
print(f"{Fore.RED}请提供聊天记录名称.用法: /chat save <名称>{Style.RESET_ALL}")
|
||
continue
|
||
elif chat_command == 'list':
|
||
self.list_chat_history()
|
||
continue
|
||
elif chat_command.startswith('open '):
|
||
name = chat_command[5:].strip()
|
||
if name:
|
||
self.load_chat_history(name)
|
||
else:
|
||
print(f"{Fore.RED}请提供聊天记录名称.用法: /chat open <名称>{Style.RESET_ALL}")
|
||
continue
|
||
else:
|
||
print(f"{Fore.RED}未知的聊天命令.可用命令: save, list, open{Style.RESET_ALL}")
|
||
continue
|
||
|
||
if user_input.startswith('/plugin'):
|
||
plugin_args = user_input[7:].strip()
|
||
result = self.handle_plugin_command(plugin_args)
|
||
if result:
|
||
print(result)
|
||
continue
|
||
|
||
if user_input.startswith('/protect'):
|
||
parts = user_input.split()
|
||
if len(parts) > 1 and parts[1] == 'test':
|
||
# 重新测试所有保护
|
||
from .process_protection import enable_process_protection
|
||
results = enable_process_protection()
|
||
print(f"\n{Fore.CYAN} 进程保护测试结果:{Style.RESET_ALL}")
|
||
for key, val in results.items():
|
||
icon = '' if val else ''
|
||
print(f" {icon} {key}: {val}")
|
||
else:
|
||
status = get_protection_status()
|
||
print(f"\n{Fore.CYAN} 进程保护状态:{Style.RESET_ALL}")
|
||
print(f" 平台: {status.get('platform', '未知')}")
|
||
icon = '' if status.get('is_protected') else ''
|
||
print(f" {icon} 保护已启用: {status.get('is_protected', False)}")
|
||
icon = '' if status.get('single_instance_held') else ''
|
||
print(f" {icon} 单实例锁: {status.get('single_instance_held', False)}")
|
||
icon = '' if status.get('watchdog_running') else ''
|
||
print(f" {icon} 看门狗: {status.get('watchdog_running', False)}")
|
||
print(f" 关闭钩子: {status.get('shutdown_hooks_count', 0)} 个")
|
||
continue
|
||
|
||
if user_input.startswith('/remote'):
|
||
remote_command = user_input[7:].strip()
|
||
self._handle_remote_command(remote_command)
|
||
continue
|
||
|
||
if user_input == '/about':
|
||
self._read_about_file()
|
||
continue
|
||
|
||
if user_input == '/tui':
|
||
print(f"{Fore.CYAN}正在切换到 TUI 模式...{Style.RESET_ALL}")
|
||
self.run_tui()
|
||
print(f"{Fore.CYAN}已从 TUI 模式返回 CLI 模式{Style.RESET_ALL}")
|
||
continue
|
||
|
||
if user_input == '/compress':
|
||
if hasattr(self, 'memory_manager') and self.memory_manager:
|
||
engine = self.current_engine
|
||
before = len(self.shared_conversation_history)
|
||
self.shared_conversation_history = self.memory_manager.compress_context(
|
||
self.shared_conversation_history, engine=engine)
|
||
self._set_shared_conversation_history()
|
||
print(f"{Fore.CYAN}已手动压缩上下文: {before} → {len(self.shared_conversation_history)} 条{Style.RESET_ALL}")
|
||
else:
|
||
print(f"{Fore.YELLOW}记忆系统未加载，无法压缩{Style.RESET_ALL}")
|
||
continue
|
||
|
||
if user_input.startswith('/memory mode '):
|
||
mode = user_input[len('/memory mode '):].strip().lower()
|
||
if mode in ('companion', 'work'):
|
||
try:
|
||
set_system_config('memory_mode', mode)
|
||
except Exception:
|
||
pass
|
||
print(f"{Fore.GREEN}记忆模式已设为: {mode}{Style.RESET_ALL}")
|
||
else:
|
||
print(f"{Fore.RED}用法: /memory mode <companion|work>{Style.RESET_ALL}")
|
||
continue
|
||
|
||
if user_input.startswith('/safe'):
|
||
parts = user_input.split()
|
||
safety = get_safety()
|
||
if len(parts) > 1:
|
||
mode_arg = parts[1].lower()
|
||
if mode_arg in ('off', 'unrestricted', '无限制'):
|
||
safety.set_mode(MODE_UNRESTRICTED)
|
||
elif mode_arg in ('on', 'normal', '普通'):
|
||
safety.set_mode(MODE_NORMAL)
|
||
elif mode_arg in ('manual', '人工', 'all'):
|
||
safety.set_mode(MODE_MANUAL)
|
||
else:
|
||
print(f"{Fore.RED}用法: /safe [off|on|manual]{Style.RESET_ALL}")
|
||
continue
|
||
else:
|
||
# 无参数：循环切换
|
||
safety.cycle_mode()
|
||
mode_name = safety.get_mode_name()
|
||
icons = {"无限制": "", "普通": "", "人工确认": ""}
|
||
icon = icons.get(mode_name, "")
|
||
print(f"{icon} 安全模式: {mode_name}")
|
||
continue
|
||
|
||
if user_input.startswith('/diff'):
|
||
parts = user_input.split()
|
||
if self._get_code_editor_plugin() is None:
|
||
print(f"{Fore.YELLOW}code_editor 插件未加载，无法设置 diff 模式{Style.RESET_ALL}")
|
||
continue
|
||
if len(parts) > 1:
|
||
arg = parts[1].lower()
|
||
if arg in ('popup', '弹窗', '1'):
|
||
self._set_diff_popup_mode(True)
|
||
elif arg in ('inline', '内联', '主终端', '2'):
|
||
self._set_diff_popup_mode(False)
|
||
else:
|
||
print(f"{Fore.RED}用法: /diff [popup|inline]{Style.RESET_ALL}")
|
||
else:
|
||
cur = bool(get_system_config('diff_popup_mode', False))
|
||
self._set_diff_popup_mode(not cur)
|
||
continue
|
||
|
||
if user_input.startswith('/fc'):
|
||
from .fc_tools import get_fc_cache, reset_fc_cache
|
||
parts = user_input.split()
|
||
if len(parts) > 1 and parts[1].lower() in ('reset', 'clear', '清除'):
|
||
n = reset_fc_cache()
|
||
print(f"{Fore.GREEN}已清除 {n} 条 FC 探测记录，下次请求会重新试探{Style.RESET_ALL}")
|
||
else:
|
||
cache = get_fc_cache()
|
||
if not cache:
|
||
print(f"{Fore.GREEN}FC 探测缓存为空：所有模型都按支持 function calling 处理{Style.RESET_ALL}")
|
||
else:
|
||
print(f"{Fore.CYAN}已探测到不支持 function calling 的模型:{Style.RESET_ALL}")
|
||
for key, ok in cache.items():
|
||
if ok is False:
|
||
print(f" {Fore.YELLOW}✗{Style.RESET_ALL} {key}")
|
||
print(f"{Fore.WHITE}这些请求会自动跳过 tools 字段；/fc reset 可清除重测{Style.RESET_ALL}")
|
||
continue
|
||
|
||
if user_input.startswith('/notify'):
|
||
parts = user_input.split()
|
||
nm = get_notification_manager()
|
||
if len(parts) > 1:
|
||
arg = parts[1].lower()
|
||
if arg in ('on', '开启', '启用'):
|
||
nm.set_enabled(True)
|
||
print(f"{Fore.GREEN}🔔 系统通知已开启{Style.RESET_ALL}")
|
||
elif arg in ('off', '关闭', '禁用'):
|
||
nm.set_enabled(False)
|
||
print(f"{Fore.YELLOW}🔕 系统通知已关闭{Style.RESET_ALL}")
|
||
else:
|
||
print(f"{Fore.RED}用法: /notify [on|off]{Style.RESET_ALL}")
|
||
else:
|
||
# 无参数：切换状态
|
||
nm.set_enabled(not nm.enabled)
|
||
if nm.enabled:
|
||
print(f"{Fore.GREEN}🔔 系统通知已开启{Style.RESET_ALL}")
|
||
else:
|
||
print(f"{Fore.YELLOW}🔕 系统通知已关闭{Style.RESET_ALL}")
|
||
continue
|
||
|
||
if user_input.startswith('/'):
|
||
command_parts = user_input[1:].split(' ', 1)
|
||
command = command_parts[0]
|
||
args = command_parts[1] if len(command_parts) > 1 else ""
|
||
if command in self.liugin_commands:
|
||
try:
|
||
result = self.liugin_commands[command](args)
|
||
if result:
|
||
result = self._limit_output_lines(result, max_lines=5)
|
||
print(f"{Fore.GREEN}{result}{Style.RESET_ALL}")
|
||
except Exception as e:
|
||
print(f"{Fore.RED}插件命令执行失败: {e}{Style.RESET_ALL}")
|
||
continue
|
||
|
||
if user_input.startswith('/thinking '):
|
||
thinking_input = user_input[10:].strip()
|
||
processed_response = self.process_thinking_response(thinking_input)
|
||
processed_response = self._process_code_blocks(processed_response)
|
||
user_id_display = f"[用户ID: {self.user_id}]"
|
||
response_lines = processed_response.split('\n')
|
||
for i, line in enumerate(response_lines):
|
||
if i == 0:
|
||
line = " " + line
|
||
if i == len(response_lines) - 1:
|
||
print(f"{line} {user_id_display}")
|
||
else:
|
||
print(f"{line}")
|
||
continue
|
||
|
||
if '@' in user_input:
|
||
processed_input = self._process_file_paths(user_input)
|
||
if processed_input != user_input:
|
||
print(f"{Fore.CYAN}{'='*50}{Style.RESET_ALL}")
|
||
user_input = processed_input
|
||
print(f"{Fore.CYAN}{'='*50}{Style.RESET_ALL}")
|
||
|
||
self.process_conversation(user_input)
|
||
# 自动持久化当前会话（opencode 式 /resume 落盘）
|
||
self._autosave_session()
|
||
# Plan 模式下：把本轮最后一条 assistant 内容记为当前计划
|
||
self._sync_current_plan()
|
||
except KeyboardInterrupt:
|
||
user_id_display = f"[用户ID: {self.user_id}]"
|
||
print(f"\n{Fore.GREEN}再见! {user_id_display}{Style.RESET_ALL}")
|
||
break
|
||
except Exception as e:
|
||
user_id_display = f"[用户ID: {self.user_id}]"
|
||
print(f"{Fore.RED}发生错误: {e} {user_id_display}{Style.RESET_ALL}")
|
||
|
||
|
||
# ============================================================================
|
||
# Main Entry Point
|
||
# ============================================================================
|
||
def main():
|
||
"""主入口函数"""
|
||
import time
|
||
import sys
|
||
|
||
art_lines = [
|
||
"█████████████████████████████████████████████████████████████████████████████████████████████████████",
|
||
"█████████████████████████████████████████████████████████████████████████████████████████████████████",
|
||
"█████████████████████████████████████████████████████████████████████████████████████████████████████",
|
||
"█████████████████████████████████████████████████████████████████████████████████████████████████████",
|
||
"█████████████████████████████████████████████████████████████████████████████████████████████████████",
|
||
"███████████▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓▓██████████████████",
|
||
"██████████ ▓████████████████",
|
||
"██████████ ▒███████████████",
|
||
"██████████ ░██████████████",
|
||
"██████████ ░█████████████",
|
||
"██████████ ████████████",
|
||
"██████████ ███████████",
|
||
"██████████ ▒████████████████████████████████████████████████████",
|
||
"██████████ ░██████████████████████████████████████████████████████",
|
||
"██████████ ████████████████████████████████████████████████████████",
|
||
"██████████ ▓█████████████████████████████████████████████████████████",
|
||
"██████████ ▒███████████████████████████████████████████████████████████",
|
||
"██████████ ███████████▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▓██████████",
|
||
"██████████ ███████████▓▒▓▒▓▒▒▓▒▒▓▒▒▓▒▒▓▒▒▓▒▒▓▒▒▓▒▒▓▒▒▓▒▒▓▒▒▓▒▒▓▒▒▓▒██████████",
|
||
"██████████ ▓██████████▓▒▒▓▒▒▒▓▒▒▓▒▒▓▒▒▓▒▒▓▒▒▓▒▒▓▒▒▓▒▒▓▒▒▓▒▒▓▒▒▒▒▒▓██████████",
|
||
"██████████ ░███████████▒▒▓▒▒▒▓▒▒▓▒▒▓▒▒▓▒▒▓▒▒▓▒▒▓▒▒▓▒▒▓▒▒▓▒▒▓▒▒▓▒▓▒▒▓▒▒▒██████████",
|
||
"██████████ ███████████▒▒▓▒▒▓▒▓▒▒▓▒▒▓▒▒▓▒▒▒▒▓▒▒▒▓▒▒▒▒▓▒▒▒▓▒▒▒▒▓▓▒██████████",
|
||
"██████████ ███████████▓▒▓▒▒▓▒▒▒▒▓▒▒▓▒▒▒▓▒▓▒▒▓▒▒▒▓▒▓▒▒▓▒▒▒▓▒▓▒▒▓▒▒▒▓▒▓▒▓██████████",
|
||
"██████████ ▓██████████▓▒▒▒▒▓▒▒▓▒▓▒▒▓▒▒▓▒▒▓▒▒▓▒▒▓▒▒▓▒▒▓▒▒▓▒▒▓▒▒▒▒▒▓▒▒▒▓██████████",
|
||
"██████████ ░██████████▓▒▒▓▒▓▒▒▓▒▒▒▒▓▒▒▓▒▓▒▓▒▒▒▓▒▒▓▒▓▒▒▒▓▒▒▓▒▓▒▒▒▓▒▒▓▒▓▒▒▒▓▓██████████",
|
||
"██████████ ███████████▒▓▒▓▒▒▒▓▒▒▓▒▓▒▒▓▒▒▒▒▒▒▓▒▓▒▒▓▒▒▒▓▒▓▒▒▓▒▒▒▓▒▓▒▒▒▒▒▒▓▒▓▒▓██████████",
|
||
"██████████ ███████████▒▒▒▒▒▒▓▒▓▒▒▓▒▒▒▓▒▒▓▒▓▒▓▒▒▒▒▓▒▒▒▓▒▒▒▒▒▓▒▒▓▒▒▒▒▒▓▒▓▒▓▒▒▒▓▓██████████",
|
||
"██████████ ░██████████▒▓▒▓▒▓▒▓▒▒▒▓▒▒▒▓▒▒▓▒▒▒▓▒▒▓▒▓▒▒▓▒▒▓▒▓▒▓▒▒▒▒▓▒▓▒▓▒▒▓▒▒▒▓▒▒▒██████████",
|
||
"██████████ █████████▓▒▒▒▓▒▒▓▒▒▒▓▒▒▓▒▓▒▒▓▒▒▓▒▓▒▒▓▒▒▓▒▒▓▒▒▓▒▒▓▒▒▓▒▒▓▒▒▓▒▒▓▒▓▒▒██████████",
|
||
"██████████ █████████▒▒▓▒▒▓▒▒▓▒▓▒▒▓▒▒▒▓▒▒▓▒▒▓▒▒▒▒▒▒▓▒▓▒▒▓▒▒▒▓▒▒▓▒▒▓▒▒▓▒▒▒▒▒▓██████████",
|
||
"██████████ ▓████████▒▓▒▓▒▒▓▒▒▒▒▓▒▒▓▒▒▓▒▒▓▒▒▓▒▒▒▒▒▒▓▒▒▓▒▒▓▒▓▒▒▒▓▒▒▓▒▒▓▓▒▒▒██████████",
|
||
"██████████ █████████▒▒▒▒▓▒▒▒▓▒▓▒▒▓▒▒▓▒▒▓▒▒▓▒▒▒▒▒▒▓▒▒▒▓▒▒▒▓▒▒▓▒▒▒▒▓▒▓▒▒▓▒▒▒▒▒▓▒██████████",
|
||
"██████████ █████████▒▓▒▓▒▒▓▒▒▓▒▒▒▒▓▒▒▒▓▒▒▓▒▒▓▒▒▓▒▒▒▒▓▒▒▓▒▒▓▒▓▒▒▒▓▒▓▒▒▓▒▒▒▒▒▓▒▒▒██████████",
|
||
"██████████ ▓████████▒▒▓▒▒▒▓▒▒▓▒▒▒▒▓▒▒▓▒▒▓▒▓▒▒▒▒▒▒▒▓▒▒▓▒▒▓▒▓▒▒▒▒▓▒▓▒▒▓▒▒▒▓▒▒██████████",
|
||
"██████████ █████████▒▓▒▒▒▒▓▒▒▒▒▓▒▓▒▒▓▒▒▒▓▒▒▒▒▒▒▒▒▓▒▒▒▓▒▒▒▓▒▒▓▒▒▒▒▓▒▓▒▒▒▒▒▒▒██████████",
|
||
"██████████ █████████▒▒▓▒▓▒▒▓▒▓▒▒▒▒▓▒▒▒▓▒▒▓▒▒▓▒▒▓▒▒▒▒▓▒▒▓▒▒▓▒▒▒▒▓▒▒▒▒▓▒▒▒▒▒▓▒██████████",
|
||
"██████████ ▓████████▒▓▒▒▒▓▒▒▒▒▒▒▓▒▓▒▒▓▒▒▒▓▒▒▒▒▒▒▒▓▒▒▒▓▒▒▒▓▒▒▓▒▒▒▒▒▓▒▓▒▒▒▒▒▓██████████",
|
||
"██████████ █████████▒▒▓▒▓▒▒▓▒▓▒▒▒▒▓▒▒▒▓▒▒▓▒▒▒▒▒▒▒▒▓▒▒▒▓▒▒▒▓▒▒▓▒▒▒▒▒▓▒▒▒▒▒▒██████████",
|
||
"██████████ █████████▒▓▒▒▒▓▒▒▒▒▒▒▓▒▓▒▒▓▒▒▒▓▒▒▒▒▒▒▒▓▒▒▒▓▒▒▒▓▒▒▓▒▒▒▒▒▓▒▒▒▒▒▒██████████",
|
||
"██████████ ▓████████▒▒▓▒▒▒▓▒▒▓▒▒▒▒▓▒▒▓▒▒▓▒▓▒▒▒▒▒▒▒▓▒▒▓▒▒▓▒▓▒▒▒▒▓▒▓▒▒▓▒▒▒▓▒▒▒██████████",
|
||
"██████████ █████████▒▓▒▒▒▒▓▒▒▒▒▒▒▓▒▓▒▒▓▒▒▒▓▒▒▒▒▒▒▒▓▒▒▒▓▒▒▒▓▒▒▓▒▒▒▒▒▓▒▒▒▒▒▒██████████",
|
||
"██████████ █████████▒▒▓▒▓▒▒▓▒▓▒▒▒▒▓▒▒▒▓▒▒▓▒▒▒▒▒▒▒▒▓▒▒▒▓▒▒▒▓▒▒▓▒▒▒▒▒▓▒▒▒▒▒▒██████████",
|
||
"██████████ ▓████████▒▓▒▒▒▓▒▒▒▒▒▒▓▒▓▒▒▓▒▒▒▓▒▒▒▒▒▒▒▓▒▒▒▓▒▒▒▓▒▒▓▒▒▒▒▒▓▒▒▒▒▒▒██████████",
|
||
"██████████ █████████▒▒▓▒▒▒▓▒▒▓▒▒▒▒▓▒▒▒▓▒▒▒▓▒▓▒▒▒▒▒▓▒▒▒▓▒▒▒▓▒▒▓▒▒▒▒▒▓▒▒▒▒▒▒██████████",
|
||
"██████████ █████████▒▓▒▒▒▒▒▒▓▒▒▒▓▒▒▓▒▒▒▒▒▒▒▒▓▒▒▓▒▒▒▒▓▒▒▒▓▒▒▒▓▒▒▓▒▒▒▒▒▓▒▒▒▒▒▒██████████",
|
||
"██████████ ▓████████▒▒▓▒▒▒▒▒▓▒▒▓▒▒▒▒▓▒▒▒▒▒▒▓▒▒▒▒▓▒▒▒▓▒▒▒▓▒▒▒▓▒▒▓▒▒▒▒▒▓▒▒▒▒▒▒██████████",
|
||
"███████████ █████████▒▓▒▒▒▒▒▒▓▒▒▒▒▓▒▒▓▒▒▒▒▒▒▒▒▓▒▒▓▒▒▒▒▓▒▒▒▓▒▒▒▓▒▒▓▒▒▒▒▒▓▒▒▒▒▒▒██████████",
|
||
"█████████████▓ █████████▒▒▓▒▒▓▒▒▓▒▒▒▒▓▒▒▒▓▒▒▓▒▒▒▒▓▒▒▓▒▒▒▒▓▒▒▒▓▒▒▒▓▒▒▓▒▒▒▒▒▓▒▒▒▒▒▒██████████",
|
||
"█████████████████ ▓████████▒▓▒▒▒▒▒▒▓▒▒▒▓▒▒▓▒▒▒▒▒▒▒▒▒▓▒▒▓▒▒▒▒▓▒▒▒▓▒▒▒▓▒▒▓▒▒▒▒▒▓▒▒▒▒▒▒██████████",
|
||
"███████████████████▒ █████████▒▒▓▒▒▒▓▒▒▓▒▒▒▒▓▒▒▒▓▒▒▒▒▒▒▒▒▓▒▒▓▒▒▒▓▒▒▒▓▒▒▒▓▒▒▓▒▒▒▒▒▓▒▒▒▒▒▒██████████",
|
||
"█████████████████████████████████████████████████████████████████████████████████████████████████████",
|
||
"█████████████████████████████████████████████████████████████████████████████████████████████████████",
|
||
"█████████████████████████████████████████████████████████████████████████████████████████████████████",
|
||
"█████████████████████████████████████████████████████████████████████████████████████████████████████",
|
||
"█████████████████████████████████████████████████████████████████████████████████████████████████████",
|
||
"█████████████████████████████████████████████████████████████████████████████████████████████████████",
|
||
]
|
||
|
||
if sys.platform == 'win32':
|
||
import codecs
|
||
sys.stdout = codecs.getwriter('utf-8')(sys.stdout.buffer, 'strict')
|
||
|
||
def _safe_print(line):
|
||
try:
|
||
print(line, flush=True)
|
||
except UnicodeEncodeError:
|
||
print(line.encode('gbk', errors='ignore').decode('gbk'), flush=True)
|
||
|
||
# 完整开机动画（57 行 + 逐行 0.05s）默认不放，用 --logo 显式召唤
|
||
if '--logo' in sys.argv or '--banner' in sys.argv:
|
||
for line in art_lines:
|
||
_safe_print(line)
|
||
time.sleep(0.05)
|
||
else:
|
||
_safe_print(f"\n{Fore.CYAN} ▄▀▄ 小狸 Pro-CLI v{VERSION}{Style.RESET_ALL}")
|
||
_safe_print(f"{Fore.CYAN} ▀▄▀ {Style.RESET_ALL}"
|
||
f"{Fore.WHITE}AI 智能编程助手{Style.RESET_ALL}"
|
||
f" {Fore.BLACK}{Style.BRIGHT}(--logo 看完整开机动画){Style.RESET_ALL}")
|
||
|
||
cli = AICLI()
|
||
cli.run()
|