fr-cli 2.1.0__py3-none-any.whl

fr_cli/core/core.py

@@ -0,0 +1,111 @@
+"""
+全局状态管理容器 (AppState)
+统一管理配置、子系统实例、运行时状态，实现依赖注入。
+"""
+from zhipuai import ZhipuAI
+from fr_cli.weapon.fs import VFS
+from fr_cli.weapon.mail import MailClient
+from fr_cli.weapon.web import WebRaider
+from fr_cli.weapon.disk import CloudDisk
+from fr_cli.addon.plugin import init_plugins
+from fr_cli.command.security import SecurityManager
+from fr_cli.command.executor import CommandExecutor
+from fr_cli.weapon.loader import load_weapon_md
+from fr_cli.weapon.mcp import MCPManager
+
+
+class AppState:
+    """应用程序运行时状态容器 —— 本命元神"""
+
+    def __init__(self, cfg):
+        self.cfg = cfg
+        self.lang = cfg.get("lang", "zh")
+        self.model_name = cfg.get("model", "glm-4-flash")
+        self.limit = cfg.get("limit", 4096)
+        self.api_key = cfg.get("key", "")
+        self.sn = cfg.get("session_name", "")
+        self.aliases = cfg.get("aliases", {})
+        self.thinking_mode = cfg.get("thinking_mode", "direct")
+
+        # 核心子系统实例化
+        self.client = ZhipuAI(api_key=self.api_key)
+        self.vfs = VFS(cfg.get("allowed_dirs", []))
+        self.plugins = init_plugins()
+        self.mail_c = MailClient(cfg.get("mail", {}))
+        self.web_c = WebRaider()
+        self.disk_c = CloudDisk(cfg.get("disk", {}))
+        self.security = SecurityManager(self.lang, cfg)
+
+        # MCP 法宝管理器
+        self.mcp = MCPManager(cfg)
+
+        # 运行时消息与上下文
+        self.messages = []
+        self.context_summary = ""
+        self.weapon_tools, self.weapon_triggers = load_weapon_md()
+        self.mcp_tools = []  # 延迟加载，避免启动阻塞
+
+        # 自动会话存档路径（按日期编号）
+        self.auto_session_path = None
+
+        # 命令执行引擎
+        self.executor = CommandExecutor(self)
+
+        # 主控 Agent（自我进化型）
+        from fr_cli.agent.master import MasterAgent
+        self.master_agent = MasterAgent(self)
+
+        # Agent HTTP 服务守护
+        self.agent_server = None
+
+        # Gatekeeper 守护进程管理器
+        from fr_cli.gatekeeper.manager import GatekeeperManager
+        self.gatekeeper = GatekeeperManager()
+
+    def reinit_client(self):
+        """API Key 或模型变更后重铸客户端"""
+        self.api_key = self.cfg.get("key", "")
+        self.client = ZhipuAI(api_key=self.api_key)
+
+    def save_cfg(self):
+        """持久化当前配置"""
+        from fr_cli.conf.config import save_config
+        save_config(self.cfg)
+
+    def update_model(self, name):
+        """切换法器模型"""
+        self.cfg["model"] = name
+        self.model_name = name
+        self.save_cfg()
+        self.reinit_client()
+
+    def update_key(self, key):
+        """重铸 API 密钥"""
+        self.cfg["key"] = key
+        self.save_cfg()
+        self.reinit_client()
+
+    def update_limit(self, limit):
+        """设置 Token 上限"""
+        self.cfg["limit"] = limit
+        self.limit = limit
+        self.save_cfg()
+
+    def update_lang(self, lang):
+        """切换界面语言"""
+        self.cfg["lang"] = lang
+        self.lang = lang
+        self.save_cfg()
+        self.security = SecurityManager(self.lang, self.cfg)
+
+    def update_session_name(self, name):
+        """更新轮回名"""
+        self.sn = name
+        self.cfg["session_name"] = name
+        self.save_cfg()
+
+    def update_thinking_mode(self, mode):
+        """切换思维模式"""
+        self.cfg["thinking_mode"] = mode
+        self.thinking_mode = mode
+        self.save_cfg()

fr_cli/core/intent.py

@@ -0,0 +1,129 @@
+"""
+意图判定引擎
+
+封装关键词预检与 LLM 意图分类逻辑。
+负责判定用户输入是直接问答还是需要调用工具。
+"""
+
+from fr_cli.core.stream import stream_cnt
+
+
+# 明确的工具操作关键词（兜底规则，避免大模型漏判）
+# 同时包含中英文，覆盖用户在任何语言界面下输入任意语言的场景
+_FORCE_TOOL_KEYWORDS = [
+    # 中文关键词
+    "保存", "保存到", "保存文件", "写入", "写到", "写入文件", "写到文件",
+    "创建文件", "生成文件", "输出到文件", "导出到", "导出文件",
+    "搜索", "查找", "查一下", "搜一下",
+    "发送邮件", "发邮件", "发信", "发邮件给", "发信给",
+    "查看目录", "列出文件", "查看文件", "打开文件",
+    "画图", "生成图片", "画一张", "画个", "生成图像",
+    "运行代码", "执行代码", "执行脚本", "运行脚本",
+    "定时任务", "定时执行", "循环任务",
+    "上传", "上传到", "下载", "下载到",
+    "保存会话", "导出会话", "切换模型", "设置密钥",
+    # 英文关键词
+    "save", "save to", "save file", "write", "write to", "write file",
+    "create file", "generate file", "output to file", "export", "export to",
+    "search", "look up", "look for", "find", "google", "bing",
+    "send email", "send mail", "send an email", "email to", "mail to",
+    "list files", "list directory", "show files", "show directory", "open file",
+    "draw", "generate image", "create image", "paint", "image of",
+    "run code", "execute code", "run script", "execute script",
+    "schedule", "scheduled task", "cron job", "timer",
+    "upload", "upload to", "download", "download to",
+    "save session", "export session", "switch model", "set key", "set api key",
+    # MCP 外部神通关键词
+    "mcp", "外部工具", "外部神通", "调用工具", "use tool", "invoke tool",
+]
+
+# 保存意图关键词（用于第二轮强制保存检测）
+_SAVE_KEYWORDS = [
+    "保存", "保存到", "保存文件", "写入", "写到", "写入文件", "写到文件",
+    "存储", "存到", "存为", "导出到", "导出文件",
+    "save", "save to", "save file", "save as", "write", "write to", "write file",
+    "store", "store to", "export to", "export file",
+]
+
+# 信息获取关键词（触发"先回答再调用工具"双源模式）
+_INFO_FETCH_KEYWORDS = [
+    # 搜索/查询
+    "搜索", "查询", "查一下", "搜一下", "什么是", "是什么", "介绍", "了解一下",
+    "最新", "新闻", "资讯", "百科", "定义", "概念", "解释",
+    "search", "look up", "what is", "what are", "introduction to", "latest",
+    "news", "wikipedia", "who is", "how to", "define", "explain",
+    # 远程/RAG/Agent
+    "远程", "rag", "知识库", "agent", "mcp", "外部工具", "外部神通",
+    "remote", "knowledge base", "external tool",
+    # 文件/数据读取
+    "读取", "查看内容", "分析文件", "总结", "提取",
+    "read", "analyze file", "summarize", "extract",
+]
+
+
+def should_force_tool(user_input):
+    """快速关键词预检：如果包含明确的工具操作关键词，直接判定为需要工具。
+    同时检测中英文关键词，不依赖当前界面语言。"""
+    u = user_input.lower()
+    for kw in _FORCE_TOOL_KEYWORDS:
+        if kw.lower() in u:
+            return True
+    return False
+
+
+def classify_intent(state, user_input, tools, lang):
+    """
+    意图判定：让大模型判定用户提问是直接查询还是需要调用工具。
+    将用户提问内容和 fr-cli 功能列表发给大模型，由大模型做判定。
+    返回 "DIRECT"（直接回答）或 "TOOL"（需要调用工具）。
+    根据 lang 自动切换中英文 prompt。
+    """
+    tools_desc = "\n".join([
+        f"- {t['name']}: {t['description']} (commands: {', '.join(t['commands'])}"
+        for t in tools
+    ])
+
+    if lang == "en":
+        classify_prompt = f"""You are an intent classifier. Based on the user's question, determine whether they need a direct answer or need to use the following tools to complete their task.
+
+Available tools:
+{tools_desc}
+
+Rules (strict):
+- DIRECT: The user is only asking for information, concepts, advice, or chatting. No action is required.
+- TOOL: The user requests any specific action, including but not limited to saving files, searching the web, sending emails, listing directories, running code, generating images, etc. If the user mentions any action word like "save", "write", "search", "send", "look up", "list", etc., even if the first half is a question, it MUST be classified as TOOL.
+
+User question: {user_input}
+
+Output only one word: DIRECT or TOOL. No explanation."""
+    else:
+        classify_prompt = f"""你是一个意图分类器。请根据用户的提问，判定用户是需要直接获得回答，还是需要调用以下工具来完成任务。
+
+可用工具列表：
+{tools_desc}
+
+判定规则（请严格遵守）：
+- DIRECT：用户只是单纯询问信息、概念、建议、闲聊，没有任何操作要求。
+- TOOL：用户要求执行任何具体操作，包括但不限于保存文件、搜索网页、发送邮件、查看目录、运行代码、画图等。只要用户提到了"保存"、"写入"、"搜索"、"发送"、"查看"等操作性词汇，即使前半句是询问信息，也必须判定为 TOOL。
+
+用户提问：{user_input}
+
+请只输出一个单词：DIRECT 或 TOOL。不要输出任何解释。"""
+
+    messages = [{"role": "user", "content": classify_prompt}]
+    txt, _, _ = stream_cnt(
+        state.client, state.model_name, messages, lang,
+        custom_prefix="", max_tokens=10, silent=True
+    )
+
+    return "TOOL" if "TOOL" in txt.upper() else "DIRECT"
+
+
+def has_info_fetch_intent(user_input):
+    """检测用户输入是否包含信息获取关键词（触发双源回答模式）。"""
+    return any(kw in user_input.lower() for kw in _INFO_FETCH_KEYWORDS)
+
+
+def has_save_intent(user_input):
+    """检测用户输入是否包含保存意图关键词。"""
+    return any(kw in user_input.lower() for kw in _SAVE_KEYWORDS)

fr_cli/core/recommender.py

@@ -0,0 +1,71 @@
+"""
+智能功能推荐系统
+根据用户输入推荐相关功能
+"""
+
+def recommend_features(user_input):
+    """根据用户输入推荐相关功能"""
+    recommendations = []
+    input_lower = user_input.lower()
+
+    # 文件相关
+    if any(kw in input_lower for kw in ['文件', 'file', '目录', 'folder', '读取', 'read', '查看', 'view', 'ls', 'cat']):
+        recommendations.append({"cmd": "/ls", "desc": "列出当前目录文件"})
+        recommendations.append({"cmd": "/cat <file>", "desc": "查看文件内容"})
+        recommendations.append({"cmd": "/cd <dir>", "desc": "切换目录"})
+
+    # 文件写入相关
+    if any(kw in input_lower for kw in ['保存', '写入', 'write', '创建', 'create', '生成', 'generate']):
+        recommendations.append({"cmd": "/write <file> <content>", "desc": "写入文件内容"})
+        recommendations.append({"cmd": "/append <file> <content>", "desc": "追加文件内容"})
+
+    # 图片相关
+    if any(kw in input_lower for kw in ['图片', 'image', 'photo', '看图', 'see', '识别', 'recognize']):
+        recommendations.append({"cmd": "/see <image>", "desc": "查看并分析图片"})
+
+    # 邮件相关
+    if any(kw in input_lower for kw in ['邮件', 'mail', 'email', '发送', 'send', '收件', 'inbox']):
+        recommendations.append({"cmd": "/mail_inbox", "desc": "查看收件箱"})
+        recommendations.append({"cmd": "/mail_send <to> <subject>", "desc": "发送邮件"})
+
+    # 网络搜索相关
+    if any(kw in input_lower for kw in ['搜索', 'search', 'web', '网页', 'website', '查询', 'query']):
+        recommendations.append({"cmd": "/web <query>", "desc": "网络搜索"})
+        recommendations.append({"cmd": "/fetch <url>", "desc": "获取网页内容"})
+
+    # 定时任务相关
+    if any(kw in input_lower for kw in ['定时', 'schedule', 'cron', '任务', 'task', '周期', 'period']):
+        recommendations.append({"cmd": "/cron_add <seconds> <command>", "desc": "添加定时任务"})
+        recommendations.append({"cmd": "/cron_list", "desc": "列出定时任务"})
+
+    # 云盘相关
+    if any(kw in input_lower for kw in ['云盘', 'cloud', '上传', 'upload', '下载', 'download', 'disk']):
+        recommendations.append({"cmd": "/disk_ls", "desc": "列出云盘文件"})
+        recommendations.append({"cmd": "/disk_up <local> <remote>", "desc": "上传到云盘"})
+        recommendations.append({"cmd": "/disk_down <remote> <local>", "desc": "从云盘下载"})
+
+    # 会话相关
+    if any(kw in input_lower for kw in ['保存会话', 'save session', '加载会话', 'load session', '会话', 'session', '记录', 'record']):
+        recommendations.append({"cmd": "/save <name>", "desc": "保存当前会话"})
+        recommendations.append({"cmd": "/load", "desc": "加载历史会话"})
+
+    # 配置相关
+    if any(kw in input_lower for kw in ['模型', 'model', '密钥', 'key', '配置', 'config', '设置', 'setting']):
+        recommendations.append({"cmd": "/model <name>", "desc": "切换AI模型"})
+        recommendations.append({"cmd": "/key <key>", "desc": "设置API密钥"})
+        recommendations.append({"cmd": "/lang <zh/en>", "desc": "切换语言"})
+
+    # 插件相关
+    if any(kw in input_lower for kw in ['插件', 'plugin', '技能', 'skill', '工具', 'tool']):
+        recommendations.append({"cmd": "/skills", "desc": "查看可用插件"})
+
+    # 导出相关
+    if any(kw in input_lower for kw in ['导出', 'export', '文档', 'document']):
+        recommendations.append({"cmd": "/export", "desc": "导出会话为Markdown"})
+
+    # 命令执行相关
+    if any(kw in input_lower for kw in ['执行', 'exec', '命令', 'command', 'shell', 'bash', 'terminal']):
+        recommendations.append({"cmd": "!<command>", "desc": "执行系统命令"})
+        recommendations.append({"cmd": "!<cmd> | <prompt>", "desc": "命令输出管道到AI"})
+
+    return recommendations

fr_cli/core/stream.py

@@ -0,0 +1,83 @@
+"""
+流式输出与高亮状态机引擎
+"""
+import sys
+import time
+from fr_cli.ui.ui import RESET, DIM, CYAN, RED, CODE_BG, CODE_FG
+from fr_cli.lang.i18n import T
+
+def stream_cnt(client, model, messages, lang, custom_prefix=None, max_tokens=None, silent=False):
+    """
+    流式调用智谱API并实时打印，带有简易代码块高亮状态机
+    :param silent: 如果为True，则不输出到终端，仅返回文本
+    :return: tuple (完整回复文本 str, 使用情况 dict, 响应时间 float)
+    """
+    if not silent:
+        p = custom_prefix or f"{CYAN}{T('prompt_ai', lang)}{RESET} "
+        sys.stdout.write(p); sys.stdout.flush()
+    
+    start_time = time.time()
+    full_text = ""
+    in_code = False
+    usage = {}
+    
+    try:
+        # 验证API密钥
+        if not client.api_key or len(client.api_key) < 10:
+            print(f"{RED}[错误] API密钥未配置或格式不正确{RESET}")
+            return "[错误] 请先配置有效的API密钥", {}, 0.0
+        
+        response = client.chat.completions.create(
+            model=model,
+            messages=messages,
+            stream=True,
+            max_tokens=max_tokens if max_tokens else 4096
+        )
+        
+        for chunk in response:
+            if chunk.choices and chunk.choices[0].delta:
+                txt = chunk.choices[0].delta.content
+                if txt:
+                    full_text += txt
+                    # 简易状态机：检测 ``` 切换代码背景
+                    if "```" in txt:
+                        parts = txt.split("```")
+                        for i, part in enumerate(parts):
+                            if i > 0:  # 遇到了一个 ```
+                                in_code = not in_code
+                                if not silent:
+                                    if in_code:
+                                        sys.stdout.write(f"{CODE_BG}{CODE_FG}")
+                                    else:
+                                        sys.stdout.write(f"{RESET}")
+                            if not silent:
+                                sys.stdout.write(part)
+                            sys.stdout.flush()
+                    else:
+                        if not silent:
+                            if in_code:
+                                sys.stdout.write(f"{CODE_BG}{CODE_FG}{txt}{CODE_FG}")
+                            else:
+                                sys.stdout.write(txt)
+                        sys.stdout.flush()
+                    
+            if hasattr(chunk, 'usage') and chunk.usage:
+                usage = chunk.usage.model_dump()
+                
+    except Exception as e:
+        sys.stdout.write(f"\n{DIM}{str(e)[:50]}{RESET}")
+        sys.stdout.flush()
+        
+    if not silent:
+        sys.stdout.write(RESET)
+        sys.stdout.write("\n")
+        sys.stdout.flush()
+    
+    end_time = time.time()
+    response_time = end_time - start_time
+    
+    # 如果没有收到任何内容，返回提示信息
+    if not full_text:
+        return "[错误] 无法获取AI回复，请检查API密钥配置", usage, response_time
+    
+    return full_text, usage, response_time

fr_cli/core/sysmon.py

@@ -0,0 +1,117 @@
+"""
+系统状态监控器
+提供 CPU、内存、网络带宽的实时采样
+优先使用 psutil，无依赖时优雅降级
+"""
+import time
+
+# 全局采样状态（单例）
+_last_sample = {
+    "time": 0.0,
+    "bytes_sent": 0,
+    "bytes_recv": 0,
+}
+
+
+def _has_psutil():
+    try:
+        import psutil
+        return True
+    except ImportError:
+        return False
+
+
+def get_cpu_percent():
+    """获取 CPU 使用百分比，无 psutil 时返回 None"""
+    if not _has_psutil():
+        return None
+    import psutil
+    return psutil.cpu_percent(interval=0.1)
+
+
+def get_memory_info():
+    """
+    获取内存使用情况
+    :return: (used_mb, total_mb, percent) 或 None
+    """
+    if not _has_psutil():
+        return None
+    import psutil
+    mem = psutil.virtual_memory()
+    used_mb = mem.used / (1024 * 1024)
+    total_mb = mem.total / (1024 * 1024)
+    return used_mb, total_mb, mem.percent
+
+
+def get_network_speed():
+    """
+    获取网络带宽（MB/s）
+    :return: (upload_mb_s, download_mb_s) 或 None
+    """
+    if not _has_psutil():
+        return None
+    import psutil
+    global _last_sample
+
+    net = psutil.net_io_counters()
+    now = time.time()
+
+    # 首次采样，只记录不返回速率
+    if _last_sample["time"] == 0:
+        _last_sample = {
+            "time": now,
+            "bytes_sent": net.bytes_sent,
+            "bytes_recv": net.bytes_recv,
+        }
+        return 0.0, 0.0
+
+    elapsed = now - _last_sample["time"]
+    if elapsed <= 0:
+        return 0.0, 0.0
+
+    sent_diff = net.bytes_sent - _last_sample["bytes_sent"]
+    recv_diff = net.bytes_recv - _last_sample["bytes_recv"]
+
+    upload_mb_s = (sent_diff / (1024 * 1024)) / elapsed
+    download_mb_s = (recv_diff / (1024 * 1024)) / elapsed
+
+    _last_sample = {
+        "time": now,
+        "bytes_sent": net.bytes_sent,
+        "bytes_recv": net.bytes_recv,
+    }
+
+    return upload_mb_s, download_mb_s
+
+
+def get_sys_stats(lang="zh"):
+    """
+    获取完整的系统状态字符串（用于直接拼接显示）
+    :param lang: 语言代码 "zh" 或 "en"
+    :return: 形如 "CPU: 12% | 内存: 4.2/16.0GB(26%) | 网络: ↑0.5 ↓1.2 MB/s" 的字符串
+             或空字符串（无 psutil 时）
+    """
+    if not _has_psutil():
+        return ""
+
+    cpu = get_cpu_percent()
+    mem = get_memory_info()
+    net = get_network_speed()
+
+    L_CPU = "CPU" if lang == "en" else "CPU"
+    L_MEM = "Mem" if lang == "en" else "内存"
+    L_NET = "Net" if lang == "en" else "网络"
+
+    parts = []
+    if cpu is not None:
+        parts.append(f"{L_CPU}:{cpu:.0f}%")
+
+    if mem is not None:
+        used_gb = mem[0] / 1024
+        total_gb = mem[1] / 1024
+        parts.append(f"{L_MEM}:{used_gb:.1f}/{total_gb:.1f}GB({mem[2]:.0f}%)")
+
+    if net is not None:
+        parts.append(f"{L_NET}:↑{net[0]:.1f}↓{net[1]:.1f}MB/s")
+
+    return " | ".join(parts) if parts else ""