jarvis-ai-assistant 0.1.192__py3-none-any.whl → 0.1.194__py3-none-any.whl

jarvis/jarvis_utils/output.py

@@ -42,6 +42,7 @@ class OutputType(Enum):
         USER: 用户输入
         TOOL: 工具调用
     """
+
     SYSTEM = "SYSTEM"
     CODE = "CODE"
     RESULT = "RESULT"
@@ -54,6 +55,8 @@ class OutputType(Enum):
     DEBUG = "DEBUG"
     USER = "USER"
     TOOL = "TOOL"
+
+
 class PrettyOutput:
     """
     使用rich库格式化和显示富文本输出的类。
@@ -64,6 +67,7 @@ class PrettyOutput:
     - 结构化内容的面板显示
     - 渐进显示的流式输出
     """
+
     # 不同输出类型的图标
     _ICONS = {
         OutputType.SYSTEM: "🤖",
@@ -81,39 +85,40 @@ class PrettyOutput:
     }
     # 语法高亮的语言映射
     _lang_map = {
-        'Python': 'python',
-        'JavaScript': 'javascript',
-        'TypeScript': 'typescript',
-        'Java': 'java',
-        'C++': 'cpp',
-        'C#': 'csharp',
-        'Ruby': 'ruby',
-        'PHP': 'php',
-        'Go': 'go',
-        'Rust': 'rust',
-        'Bash': 'bash',
-        'HTML': 'html',
-        'CSS': 'css',
-        'SQL': 'sql',
-        'R': 'r',
-        'Kotlin': 'kotlin',
-        'Swift': 'swift',
-        'Scala': 'scala',
-        'Perl': 'perl',
-        'Lua': 'lua',
-        'YAML': 'yaml',
-        'JSON': 'json',
-        'XML': 'xml',
-        'Markdown': 'markdown',
-        'Text': 'text',
-        'Shell': 'bash',
-        'Dockerfile': 'dockerfile',
-        'Makefile': 'makefile',
-        'INI': 'ini',
-        'TOML': 'toml',
+        "Python": "python",
+        "JavaScript": "javascript",
+        "TypeScript": "typescript",
+        "Java": "java",
+        "C++": "cpp",
+        "C#": "csharp",
+        "Ruby": "ruby",
+        "PHP": "php",
+        "Go": "go",
+        "Rust": "rust",
+        "Bash": "bash",
+        "HTML": "html",
+        "CSS": "css",
+        "SQL": "sql",
+        "R": "r",
+        "Kotlin": "kotlin",
+        "Swift": "swift",
+        "Scala": "scala",
+        "Perl": "perl",
+        "Lua": "lua",
+        "YAML": "yaml",
+        "JSON": "json",
+        "XML": "xml",
+        "Markdown": "markdown",
+        "Text": "text",
+        "Shell": "bash",
+        "Dockerfile": "dockerfile",
+        "Makefile": "makefile",
+        "INI": "ini",
+        "TOML": "toml",
     }
+
     @staticmethod
-    def _detect_language(text: str, default_lang: str = 'markdown') -> str:
+    def _detect_language(text: str, default_lang: str = "markdown") -> str:
         """
         检测给定文本的编程语言。
 
@@ -130,6 +135,7 @@ class PrettyOutput:
             return PrettyOutput._lang_map.get(detected_lang, default_lang)
         except (ClassNotFound, Exception):
             return default_lang
+
     @staticmethod
     def _format(output_type: OutputType, timestamp: bool = True) -> str:
         """
@@ -145,13 +151,20 @@ class PrettyOutput:
         icon = PrettyOutput._ICONS.get(output_type, "")
         formatted = f"{icon}  "
         if timestamp:
-            formatted+=f"[{datetime.now().strftime('%H:%M:%S')}][{output_type.value}]"
+            formatted += f"[{datetime.now().strftime('%H:%M:%S')}][{output_type.value}]"
         agent_info = get_agent_list()
         if agent_info:
-            formatted+=f"[{agent_info}]"
+            formatted += f"[{agent_info}]"
         return formatted
+
     @staticmethod
-    def print(text: str, output_type: OutputType, timestamp: bool = True, lang: Optional[str] = None, traceback: bool = False):
+    def print(
+        text: str,
+        output_type: OutputType,
+        timestamp: bool = True,
+        lang: Optional[str] = None,
+        traceback: bool = False,
+    ):
         """
         使用样式和语法高亮打印格式化输出。
 
@@ -163,38 +176,101 @@ class PrettyOutput:
             traceback: 是否显示错误的回溯信息
         """
         styles = {
-            OutputType.SYSTEM: dict( bgcolor="#1e2b3c"),
-            OutputType.CODE: dict( bgcolor="#1c2b1c"),
-            OutputType.RESULT: dict( bgcolor="#1c1c2b"),
-            OutputType.ERROR: dict( bgcolor="#2b1c1c"),
-            OutputType.INFO: dict( bgcolor="#2b2b1c", meta={"icon": "ℹ️"}),
-            OutputType.PLANNING: dict(  bgcolor="#2b1c2b"),
-            OutputType.PROGRESS: dict(  bgcolor="#1c1c1c"),
-            OutputType.SUCCESS: dict( bgcolor="#1c2b1c"),
-            OutputType.WARNING: dict( bgcolor="#2b2b1c"),
-            OutputType.DEBUG: dict(  bgcolor="#1c1c1c"),
-            OutputType.USER: dict( bgcolor="#1c2b2b"),
-            OutputType.TOOL: dict( bgcolor="#1c2b2b"),
+            OutputType.SYSTEM: dict(bgcolor="#1e2b3c"),
+            OutputType.CODE: dict(bgcolor="#1c2b1c"),
+            OutputType.RESULT: dict(bgcolor="#1c1c2b"),
+            OutputType.ERROR: dict(bgcolor="#2b1c1c"),
+            OutputType.INFO: dict(bgcolor="#2b2b1c", meta={"icon": "ℹ️"}),
+            OutputType.PLANNING: dict(bgcolor="#2b1c2b"),
+            OutputType.PROGRESS: dict(bgcolor="#1c1c1c"),
+            OutputType.SUCCESS: dict(bgcolor="#1c2b1c"),
+            OutputType.WARNING: dict(bgcolor="#2b2b1c"),
+            OutputType.DEBUG: dict(bgcolor="#1c1c1c"),
+            OutputType.USER: dict(bgcolor="#1c2b2b"),
+            OutputType.TOOL: dict(bgcolor="#1c2b2b"),
         }
 
         header_styles = {
-            OutputType.SYSTEM: RichStyle(color="bright_cyan", bgcolor="#1e2b3c", frame=True, meta={"icon": "🤖"}),
-            OutputType.CODE: RichStyle(color="green", bgcolor="#1c2b1c", frame=True, meta={"icon": "📝"}),
-            OutputType.RESULT: RichStyle(color="bright_blue", bgcolor="#1c1c2b", frame=True, meta={"icon": "✨"}),
-            OutputType.ERROR: RichStyle(color="red", frame=True, bgcolor="#2b1c1c", meta={"icon": "❌"}),
-            OutputType.INFO: RichStyle(color="gold1", frame=True, bgcolor="#2b2b1c", meta={"icon": "ℹ️"}),
-            OutputType.PLANNING: RichStyle(color="purple", bold=True, frame=True, bgcolor="#2b1c2b", meta={"icon": "📋"}),
-            OutputType.PROGRESS: RichStyle(color="white", encircle=True, frame=True, bgcolor="#1c1c1c", meta={"icon": "⏳"}),
-            OutputType.SUCCESS: RichStyle(color="bright_green", bold=True, strike=False, bgcolor="#1c2b1c", meta={"icon": "✅"}),
-            OutputType.WARNING: RichStyle(color="yellow", bold=True, blink2=True, bgcolor="#2b2b1c", meta={"icon": "⚠️"}),
-            OutputType.DEBUG: RichStyle(color="grey58", dim=True, conceal=True, bgcolor="#1c1c1c", meta={"icon": "🔍"}),
-            OutputType.USER: RichStyle(color="spring_green2", frame=True, bgcolor="#1c2b2b", meta={"icon": "👤"}),
-            OutputType.TOOL: RichStyle(color="dark_sea_green4", bgcolor="#1c2b2b", frame=True, meta={"icon": "🔧"}),
+            OutputType.SYSTEM: RichStyle(
+                color="bright_cyan", bgcolor="#1e2b3c", frame=True, meta={"icon": "🤖"}
+            ),
+            OutputType.CODE: RichStyle(
+                color="green", bgcolor="#1c2b1c", frame=True, meta={"icon": "📝"}
+            ),
+            OutputType.RESULT: RichStyle(
+                color="bright_blue", bgcolor="#1c1c2b", frame=True, meta={"icon": "✨"}
+            ),
+            OutputType.ERROR: RichStyle(
+                color="red", frame=True, bgcolor="#2b1c1c", meta={"icon": "❌"}
+            ),
+            OutputType.INFO: RichStyle(
+                color="gold1", frame=True, bgcolor="#2b2b1c", meta={"icon": "ℹ️"}
+            ),
+            OutputType.PLANNING: RichStyle(
+                color="purple",
+                bold=True,
+                frame=True,
+                bgcolor="#2b1c2b",
+                meta={"icon": "📋"},
+            ),
+            OutputType.PROGRESS: RichStyle(
+                color="white",
+                encircle=True,
+                frame=True,
+                bgcolor="#1c1c1c",
+                meta={"icon": "⏳"},
+            ),
+            OutputType.SUCCESS: RichStyle(
+                color="bright_green",
+                bold=True,
+                strike=False,
+                bgcolor="#1c2b1c",
+                meta={"icon": "✅"},
+            ),
+            OutputType.WARNING: RichStyle(
+                color="yellow",
+                bold=True,
+                blink2=True,
+                bgcolor="#2b2b1c",
+                meta={"icon": "⚠️"},
+            ),
+            OutputType.DEBUG: RichStyle(
+                color="grey58",
+                dim=True,
+                conceal=True,
+                bgcolor="#1c1c1c",
+                meta={"icon": "🔍"},
+            ),
+            OutputType.USER: RichStyle(
+                color="spring_green2",
+                frame=True,
+                bgcolor="#1c2b2b",
+                meta={"icon": "👤"},
+            ),
+            OutputType.TOOL: RichStyle(
+                color="dark_sea_green4",
+                bgcolor="#1c2b2b",
+                frame=True,
+                meta={"icon": "🔧"},
+            ),
         }
 
-        lang = lang if lang is not None else PrettyOutput._detect_language(text, default_lang='markdown')
-        header = Text(PrettyOutput._format(output_type, timestamp), style=header_styles[output_type])
-        content = Syntax(text, lang, theme="monokai", word_wrap=True, background_color=styles[output_type]["bgcolor"])
+        lang = (
+            lang
+            if lang is not None
+            else PrettyOutput._detect_language(text, default_lang="markdown")
+        )
+        header = Text(
+            PrettyOutput._format(output_type, timestamp),
+            style=header_styles[output_type],
+        )
+        content = Syntax(
+            text,
+            lang,
+            theme="monokai",
+            word_wrap=True,
+            background_color=styles[output_type]["bgcolor"],
+        )
         panel = Panel(
             content,
             border_style=header_styles[output_type],
@@ -213,6 +289,7 @@ class PrettyOutput:
                 console.print(header, content)
         if traceback:
             console.print_exception()
+
     @staticmethod
     def section(title: str, output_type: OutputType = OutputType.INFO):
         """
@@ -223,38 +300,42 @@ class PrettyOutput:
             output_type: 输出类型（影响样式）
         """
         text = Text(title, style=output_type.value, justify="center")
-        panel = Panel(
-            text,
-            border_style=output_type.value
-        )
+        panel = Panel(text, border_style=output_type.value)
         if get_pretty_output():
             console.print(panel)
         else:
             console.print(text)
-    
+
     @staticmethod
-    def print_gradient_text(text: str, start_color: Tuple[int, int, int], end_color: Tuple[int, int, int]) -> None:
+    def print_gradient_text(
+        text: str, start_color: Tuple[int, int, int], end_color: Tuple[int, int, int]
+    ) -> None:
         """打印带有渐变色彩的文本。
-        
+
         Args:
             text: 要打印的文本
             start_color: 起始RGB颜色元组 (r, g, b)
             end_color: 结束RGB颜色元组 (r, g, b)
         """
-        lines = text.strip('\n').split('\n')
+        lines = text.strip("\n").split("\n")
         total_lines = len(lines)
         colored_lines = []
         for i, line in enumerate(lines):
             # 计算当前行的渐变颜色
-            r = int(start_color[0] + (end_color[0] - start_color[0]) * i / (total_lines - 1))
-            g = int(start_color[1] + (end_color[1] - start_color[1]) * i / (total_lines - 1))
-            b = int(start_color[2] + (end_color[2] - start_color[2]) * i / (total_lines - 1))
-            
+            r = int(
+                start_color[0] + (end_color[0] - start_color[0]) * i / (total_lines - 1)
+            )
+            g = int(
+                start_color[1] + (end_color[1] - start_color[1]) * i / (total_lines - 1)
+            )
+            b = int(
+                start_color[2] + (end_color[2] - start_color[2]) * i / (total_lines - 1)
+            )
+
             # 使用ANSI转义序列设置颜色
             colored_lines.append(f"\033[38;2;{r};{g};{b}m{line}\033[0m")
-        colored_text = Text('\n'.join(colored_lines), style=OutputType.TOOL.value, justify="center")
-        panel = Panel(
-            colored_text,
-            box=SIMPLE
+        colored_text = Text(
+            "\n".join(colored_lines), style=OutputType.TOOL.value, justify="center"
         )
-        console.print(panel)
+        panel = Panel(colored_text, box=SIMPLE)
+        console.print(panel)

jarvis/jarvis_utils/tag.py

@@ -10,6 +10,7 @@ def ot(tag_name: str) -> str:
     """
     return f"<{tag_name}>"
 
+
 def ct(tag_name: str) -> str:
     """生成HTML标签结束标记
 
@@ -19,4 +20,4 @@ def ct(tag_name: str) -> str:
     返回：
     格式化的结束标签字符串
     """
-    return f"</{tag_name}>"
+    return f"</{tag_name}>"

jarvis/jarvis_utils/utils.py

@@ -11,12 +11,17 @@ from typing import Any, Callable, Dict, Optional
 import yaml
 
 from jarvis import __version__
-from jarvis.jarvis_utils.config import get_data_dir, get_max_big_content_size, set_global_env_data
+from jarvis.jarvis_utils.config import (
+    get_data_dir,
+    get_max_big_content_size,
+    set_global_env_data,
+)
 from jarvis.jarvis_utils.embedding import get_context_token_count
 from jarvis.jarvis_utils.input import get_single_line_input
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 from jarvis.jarvis_utils.globals import get_in_chat, get_interrupt, set_interrupt
 
+g_config_file = None
 
 
 def init_env(welcome_str: str, config_file: Optional[str] = None) -> None:
@@ -35,7 +40,7 @@ def init_env(welcome_str: str, config_file: Optional[str] = None) -> None:
     """
     # 保存原始信号处理函数
     original_sigint = signal.getsignal(signal.SIGINT)
-    
+
     def sigint_handler(signum, frame):
         if get_in_chat():
             set_interrupt(True)
@@ -44,7 +49,7 @@ def init_env(welcome_str: str, config_file: Optional[str] = None) -> None:
         else:
             if original_sigint and callable(original_sigint):
                 original_sigint(signum, frame)
-    
+
     signal.signal(signal.SIGINT, sigint_handler)
     count_cmd_usage()
 
@@ -63,22 +68,16 @@ def init_env(welcome_str: str, config_file: Optional[str] = None) -> None:
     if welcome_str:
         PrettyOutput.print_gradient_text(jarvis_ascii_art, (0, 120, 255), (0, 255, 200))
 
-    config_file_path = Path(config_file) if config_file is not None else Path(os.path.expanduser("~/.jarvis/config.yaml"))
+    global g_config_file
+    g_config_file = config_file
 
-    # 加载配置文件
-    if not config_file_path.exists():
-        old_config_file = config_file_path.parent / "env"
-        if old_config_file.exists():# 旧的配置文件存在
-            _read_old_config_file(old_config_file)
-    else:
-        _read_config_file(config_file_path.parent, config_file_path)
+    load_config()
 
     # 现在获取最终的数据目录(可能被配置文件修改)
     data_dir = Path(get_data_dir())
     script_dir = Path(os.path.dirname(os.path.dirname(__file__)))
     hf_archive = script_dir / "jarvis_data" / "huggingface.tar.gz"
 
-
     # 检查并解压huggingface模型
     hf_dir = data_dir / "huggingface" / "hub"
     if not hf_dir.exists() and hf_archive.exists():
@@ -88,22 +87,40 @@ def init_env(welcome_str: str, config_file: Optional[str] = None) -> None:
                 tar.extractall(path=data_dir)
             PrettyOutput.print("HuggingFace模型解压完成", OutputType.SUCCESS)
         except Exception as e:
-            PrettyOutput.print(f"解压HuggingFace模型失败: {e}", OutputType.ERROR)   
+            PrettyOutput.print(f"解压HuggingFace模型失败: {e}", OutputType.ERROR)
 
         # 检查是否是git仓库并更新
     from jarvis.jarvis_utils.git_utils import check_and_update_git_repo
 
     check_and_update_git_repo(str(script_dir))
 
+
+def load_config():
+    config_file = g_config_file
+    config_file_path = (
+        Path(config_file)
+        if config_file is not None
+        else Path(os.path.expanduser("~/.jarvis/config.yaml"))
+    )
+
+    # 加载配置文件
+    if not config_file_path.exists():
+        old_config_file = config_file_path.parent / "env"
+        if old_config_file.exists():  # 旧的配置文件存在
+            _read_old_config_file(old_config_file)
+    else:
+        _read_config_file(config_file_path.parent, config_file_path)
+
+
 def _read_config_file(jarvis_dir, config_file):
     """读取并解析YAML格式的配置文件
-    
+
     功能：
     1. 读取配置文件内容
     2. 检查并添加schema声明(如果缺失)
     3. 将配置数据保存到全局变量
     4. 设置环境变量(如果配置中有ENV字段)
-    
+
     参数:
         jarvis_dir: Jarvis数据目录路径
         config_file: 配置文件路径
@@ -112,31 +129,38 @@ def _read_config_file(jarvis_dir, config_file):
         content = f.read()
         config_data = yaml.safe_load(content) or {}
         if isinstance(config_data, dict):
-                # 检查是否已有schema声明，没有则添加
+            # 检查是否已有schema声明，没有则添加
             if "# yaml-language-server: $schema=" not in content:
-                schema_path = Path(os.path.relpath(
-                        Path(__file__).parent.parent / "jarvis_data" / "config_schema.json",
-                        start=jarvis_dir
-                    ))
+                schema_path = Path(
+                    os.path.relpath(
+                        Path(__file__).parent.parent
+                        / "jarvis_data"
+                        / "config_schema.json",
+                        start=jarvis_dir,
+                    )
+                )
                 with open(config_file, "w", encoding="utf-8") as f:
                     f.write(f"# yaml-language-server: $schema={schema_path}\n")
                     f.write(content)
             # 保存到全局变量
         set_global_env_data(config_data)
-            # 如果配置中有ENV键值对，则设置环境变量
+        # 如果配置中有ENV键值对，则设置环境变量
         if "ENV" in config_data and isinstance(config_data["ENV"], dict):
-            os.environ.update({str(k): str(v) for k, v in config_data["ENV"].items() if v is not None})
+            os.environ.update(
+                {str(k): str(v) for k, v in config_data["ENV"].items() if v is not None}
+            )
+
 
 def _read_old_config_file(config_file):
     """读取并解析旧格式的env配置文件
-    
+
     功能：
     1. 解析键值对格式的旧配置文件
     2. 支持多行值的处理
     3. 自动去除值的引号和空格
     4. 将配置数据保存到全局变量
     5. 设置环境变量并显示迁移警告
-    
+
     参数:
         config_file: 旧格式配置文件路径
     """
@@ -149,7 +173,7 @@ def _read_old_config_file(config_file):
             if not line or line.startswith(("#", ";")):
                 continue
             if "=" in line and not line.startswith((" ", "\t")):
-                        # 处理之前收集的多行值
+                # 处理之前收集的多行值
                 if current_key is not None:
                     value = "\n".join(current_value).strip().strip("'").strip('"')
                     # 将字符串"true"/"false"转换为bool类型
@@ -159,12 +183,12 @@ def _read_old_config_file(config_file):
                         value = False
                     config_data[current_key] = value
                     current_value = []
-                        # 解析新的键值对
+                    # 解析新的键值对
                 key, value = line.split("=", 1)
                 current_key = key.strip()
                 current_value.append(value.strip())
             elif current_key is not None:
-                        # 多行值的后续行
+                # 多行值的后续行
                 current_value.append(line.strip())
                 # 处理最后一个键值对
         if current_key is not None:
@@ -175,11 +199,16 @@ def _read_old_config_file(config_file):
             elif value.lower() == "false":
                 value = False
             config_data[current_key] = value
-        os.environ.update({str(k): str(v) for k, v in config_data.items() if v is not None})
+        os.environ.update(
+            {str(k): str(v) for k, v in config_data.items() if v is not None}
+        )
         set_global_env_data(config_data)
-    PrettyOutput.print(f"检测到旧格式配置文件，旧格式以后将不再支持，请尽快迁移到新格式", OutputType.WARNING)
+    PrettyOutput.print(
+        f"检测到旧格式配置文件，旧格式以后将不再支持，请尽快迁移到新格式",
+        OutputType.WARNING,
+    )
+
 
-    
 def while_success(func: Callable[[], Any], sleep_time: float = 0.1) -> Any:
     """循环执行函数直到成功
 
@@ -194,19 +223,23 @@ def while_success(func: Callable[[], Any], sleep_time: float = 0.1) -> Any:
         try:
             return func()
         except Exception as e:
-            PrettyOutput.print(f"执行失败: {str(e)}, 等待 {sleep_time}s...", OutputType.WARNING)
+            PrettyOutput.print(
+                f"执行失败: {str(e)}, 等待 {sleep_time}s...", OutputType.WARNING
+            )
             time.sleep(sleep_time)
             continue
+
+
 def while_true(func: Callable[[], bool], sleep_time: float = 0.1) -> Any:
     """循环执行函数直到返回True
-    
+
     参数:
         func: 要执行的函数，必须返回布尔值
         sleep_time: 每次失败后的等待时间(秒)
-        
+
     返回:
         函数最终返回的True值
-        
+
     注意:
         与while_success不同，此函数只检查返回是否为True，
         不捕获异常，异常会直接抛出
@@ -218,7 +251,9 @@ def while_true(func: Callable[[], bool], sleep_time: float = 0.1) -> Any:
         PrettyOutput.print(f"执行失败, 等待 {sleep_time}s...", OutputType.WARNING)
         time.sleep(sleep_time)
     return ret
-def get_file_md5(filepath: str)->str:
+
+
+def get_file_md5(filepath: str) -> str:
     """计算文件内容的MD5哈希值
 
     参数:
@@ -227,7 +262,9 @@ def get_file_md5(filepath: str)->str:
     返回:
         str: 文件内容的MD5哈希值
     """
-    return hashlib.md5(open(filepath, "rb").read(100*1024*1024)).hexdigest()
+    return hashlib.md5(open(filepath, "rb").read(100 * 1024 * 1024)).hexdigest()
+
+
 def user_confirm(tip: str, default: bool = True) -> bool:
     """提示用户确认是/否问题
 
@@ -242,6 +279,7 @@ def user_confirm(tip: str, default: bool = True) -> bool:
     ret = get_single_line_input(f"{tip} {suffix}: ")
     return default if ret == "" else ret.lower() == "y"
 
+
 def get_file_line_count(filename: str) -> int:
     """计算文件中的行数
 
@@ -257,7 +295,6 @@ def get_file_line_count(filename: str) -> int:
         return 0
 
 
-
 def _get_cmd_stats() -> Dict[str, int]:
     """从数据目录获取命令调用统计"""
     stats_file = Path(get_data_dir()) / "cmd_stat.yaml"
@@ -266,11 +303,10 @@ def _get_cmd_stats() -> Dict[str, int]:
             with open(stats_file, "r", encoding="utf-8") as f:
                 return yaml.safe_load(f) or {}
         except Exception as e:
-            PrettyOutput.print(
-                f"加载命令调用统计失败: {str(e)}", OutputType.WARNING
-            )
+            PrettyOutput.print(f"加载命令调用统计失败: {str(e)}", OutputType.WARNING)
     return {}
 
+
 def _update_cmd_stats(cmd_name: str) -> None:
     """更新命令调用统计"""
     stats = _get_cmd_stats()
@@ -280,31 +316,29 @@ def _update_cmd_stats(cmd_name: str) -> None:
         with open(stats_file, "w", encoding="utf-8") as f:
             yaml.safe_dump(stats, f, allow_unicode=True)
     except Exception as e:
-        PrettyOutput.print(
-            f"保存命令调用统计失败: {str(e)}", OutputType.WARNING
-        )
+        PrettyOutput.print(f"保存命令调用统计失败: {str(e)}", OutputType.WARNING)
+
 
 def count_cmd_usage() -> None:
     """统计当前命令的使用次数"""
     import sys
+
     _update_cmd_stats(sys.argv[0])
 
+
 def is_context_overflow(content: str) -> bool:
     """判断文件内容是否超出上下文限制"""
-    return get_context_token_count(content) > get_max_big_content_size() 
+    return get_context_token_count(content) > get_max_big_content_size()
+
 
 def get_loc_stats() -> str:
     """使用loc命令获取当前目录的代码统计信息
-    
+
     返回:
         str: loc命令输出的原始字符串，失败时返回空字符串
     """
     try:
-        result = subprocess.run(
-            ['loc'],
-            capture_output=True,
-            text=True
-        )
+        result = subprocess.run(["loc"], capture_output=True, text=True)
         return result.stdout if result.returncode == 0 else ""
     except FileNotFoundError:
-        return ""
+        return ""