jarvis-ai-assistant 0.1.120__py3-none-any.whl → 0.1.122__py3-none-any.whl

jarvis/jarvis_platform/ai8.py

@@ -60,12 +60,12 @@ class AI8Model(BasePlatform):
             )
             
             if response.status_code != 200:
-                PrettyOutput.print(f"创建会话失败: {response.status_code}", OutputType.ERROR)
+                PrettyOutput.print(f"创建会话失败: {response.status_code}", OutputType.WARNING)
                 return False
             
             data = response.json()
             if data['code'] != 0:
-                PrettyOutput.print(f"创建会话失败: {data.get('msg', '未知错误')}", OutputType.ERROR)
+                PrettyOutput.print(f"创建会话失败: {data.get('msg', '未知错误')}", OutputType.WARNING)
                 return False
             
             self.conversation = data['data']
@@ -93,10 +93,10 @@ class AI8Model(BasePlatform):
                     self.conversation = data['data']
                     return True
                 else:
-                    PrettyOutput.print(f"更新会话设置失败: {data.get('msg', '未知错误')}", OutputType.ERROR)
+                    PrettyOutput.print(f"更新会话设置失败: {data.get('msg', '未知错误')}", OutputType.WARNING)
                     return False
             else:
-                PrettyOutput.print(f"更新会话设置失败: {response.status_code}", OutputType.ERROR)
+                PrettyOutput.print(f"更新会话设置失败: {response.status_code}", OutputType.WARNING)
                 return False
             
         except Exception as e:
@@ -162,7 +162,7 @@ class AI8Model(BasePlatform):
             
             if response.status_code != 200:
                 error_msg = f"Failed to chat: {response.status_code} {response.text}"
-                PrettyOutput.print(error_msg, OutputType.ERROR)
+                PrettyOutput.print(error_msg, OutputType.WARNING)
                 raise Exception(error_msg)
             
             # 处理流式响应
@@ -229,11 +229,11 @@ class AI8Model(BasePlatform):
                     return True
                 else:
                     error_msg = f"删除会话失败: {data.get('msg', '未知错误')}"
-                    PrettyOutput.print(error_msg, OutputType.ERROR)
+                    PrettyOutput.print(error_msg, OutputType.WARNING)
                     return False
             else:
                 error_msg = f"删除会话请求失败: {response.status_code}"
-                PrettyOutput.print(error_msg, OutputType.ERROR)
+                PrettyOutput.print(error_msg, OutputType.WARNING)
                 return False
             
         except Exception as e:
@@ -265,12 +265,12 @@ class AI8Model(BasePlatform):
             )
             
             if response.status_code != 200:
-                PrettyOutput.print(f"获取模型列表失败: {response.status_code}", OutputType.ERROR)
+                PrettyOutput.print(f"获取模型列表失败: {response.status_code}", OutputType.WARNING)
                 return []
             
             data = response.json()
             if data['code'] != 0:
-                PrettyOutput.print(f"获取模型列表失败: {data.get('msg', '未知错误')}", OutputType.ERROR)
+                PrettyOutput.print(f"获取模型列表失败: {data.get('msg', '未知错误')}", OutputType.WARNING)
                 return []
             
             # 保存模型信息
@@ -309,6 +309,6 @@ class AI8Model(BasePlatform):
             return list(self.models.keys())
             
         except Exception as e:
-            PrettyOutput.print(f"获取模型列表失败: {str(e)}", OutputType.WARNING)
+            PrettyOutput.print(f"获取模型列表失败: {str(e)}", OutputType.ERROR)
             return []
         

jarvis/jarvis_platform/kimi.py

@@ -198,11 +198,11 @@ class KimiModel(BasePlatform):
                         if self._wait_for_parse(file_info["id"]):
                             uploaded_files.append(file_info)
                         else:
-                            PrettyOutput.print(f"✗ 文件解析失败: {file_path}", OutputType.ERROR)
+                            PrettyOutput.print(f"✗ 文件解析失败: {file_path}", OutputType.WARNING)
                     else:
                         uploaded_files.append(file_info)
                 else:
-                    PrettyOutput.print(f"错误：文件上传失败: {file_path}", OutputType.ERROR)
+                    PrettyOutput.print(f"错误：文件上传失败: {file_path}", OutputType.WARNING)
                     
             except Exception as e:
                 PrettyOutput.print(f"✗ 处理文件出错 {file_path}: {str(e)}", OutputType.ERROR)
@@ -350,8 +350,6 @@ class KimiModel(BasePlatform):
                     output.append("")
 
                 PrettyOutput.print("\n".join(output), OutputType.PROGRESS)
-
-            PrettyOutput.print(full_response, OutputType.RESULT)
             
             return full_response
 
@@ -375,7 +373,7 @@ class KimiModel(BasePlatform):
                 self.reset()
                 return True
             else:
-                PrettyOutput.print(f"删除会话失败: HTTP {response.status_code}", OutputType.ERROR)
+                PrettyOutput.print(f"删除会话失败: HTTP {response.status_code}", OutputType.WARNING)
                 return False
         except Exception as e:
             PrettyOutput.print(f"删除会话时发生错误: {str(e)}", OutputType.ERROR)

jarvis/jarvis_platform/oyi.py

@@ -83,10 +83,10 @@ class OyiModel(BasePlatform):
                     self.conversation = data
                     return True
                 else:
-                    PrettyOutput.print(f"创建会话失败: {data['message']}", OutputType.ERROR)
+                    PrettyOutput.print(f"创建会话失败: {data['message']}", OutputType.WARNING)
                     return False
             else:
-                PrettyOutput.print(f"创建会话失败: {response.status_code}", OutputType.ERROR)
+                PrettyOutput.print(f"创建会话失败: {response.status_code}", OutputType.WARNING)
                 return False
                 
         except Exception as e:
@@ -157,13 +157,13 @@ class OyiModel(BasePlatform):
             
             if response.status_code != 200:
                 error_msg = f"聊天请求失败: {response.status_code}"
-                PrettyOutput.print(error_msg, OutputType.ERROR)
+                PrettyOutput.print(error_msg, OutputType.WARNING)
                 raise Exception(error_msg)
             
             data = response.json()
             if data['code'] != 200 or data['type'] != 'success':
                 error_msg = f"聊天失败: {data.get('message', '未知错误')}"
-                PrettyOutput.print(error_msg, OutputType.ERROR)
+                PrettyOutput.print(error_msg, OutputType.WARNING)
                 raise Exception(error_msg)
             
             message_id = data['result'][-1]
@@ -196,7 +196,7 @@ class OyiModel(BasePlatform):
                 return full_response
             else:
                 error_msg = f"获取响应失败: {response.status_code}"
-                PrettyOutput.print(error_msg, OutputType.ERROR)
+                PrettyOutput.print(error_msg, OutputType.WARNING)
                 raise Exception(error_msg)
         except Exception as e:
             PrettyOutput.print(f"聊天失败: {str(e)}", OutputType.ERROR)
@@ -241,11 +241,11 @@ class OyiModel(BasePlatform):
                     return True
                 else:
                     error_msg = f"删除会话失败: {data.get('message', '未知错误')}"
-                    PrettyOutput.print(error_msg, OutputType.ERROR)
+                    PrettyOutput.print(error_msg, OutputType.WARNING)
                     return False
             else:
                 error_msg = f"删除会话请求失败: {response.status_code}"
-                PrettyOutput.print(error_msg, OutputType.ERROR)
+                PrettyOutput.print(error_msg, OutputType.WARNING)
                 return False
             
         except Exception as e:
@@ -281,7 +281,7 @@ class OyiModel(BasePlatform):
                 # 检查文件类型
                 file_type = mimetypes.guess_type(file_path)[0]
                 if not file_type or not file_type.startswith(('image/', 'text/', 'application/')):
-                    PrettyOutput.print(f"文件类型 {file_type} 不支持", OutputType.ERROR)
+                    PrettyOutput.print(f"文件类型 {file_type} 不支持", OutputType.WARNING)
                     continue
                 
                 with open(file_path, 'rb') as f:
@@ -300,10 +300,10 @@ class OyiModel(BasePlatform):
                         if data.get('code') == 200:
                             self.files.append(data)
                         else:
-                            PrettyOutput.print(f"文件上传失败: {data.get('message')}", OutputType.ERROR)
+                            PrettyOutput.print(f"文件上传失败: {data.get('message')}", OutputType.WARNING)
                             return []
                     else:
-                        PrettyOutput.print(f"文件上传失败: {response.status_code}", OutputType.ERROR)
+                        PrettyOutput.print(f"文件上传失败: {response.status_code}", OutputType.WARNING)
                         return []
                 
             return self.files
@@ -335,7 +335,7 @@ class OyiModel(BasePlatform):
             )
             
             if response.status_code != 200:
-                PrettyOutput.print(f"获取模型列表失败: {response.status_code}", OutputType.ERROR)
+                PrettyOutput.print(f"获取模型列表失败: {response.status_code}", OutputType.WARNING)
                 return []
             
             data = response.json()

jarvis/jarvis_platform/registry.py

@@ -72,7 +72,7 @@ class PlatformRegistry:
         if missing_methods:
             PrettyOutput.print(
                 f"平台 {platform_class.__name__} 缺少必要的方法: {', '.join(missing_methods)}", 
-                OutputType.ERROR
+                OutputType.WARNING
             )
             return False
             
@@ -92,7 +92,7 @@ class PlatformRegistry:
         
         # 确保目录存在
         if not os.path.exists(directory):
-            PrettyOutput.print(f"平台目录不存在: {directory}", OutputType.ERROR)
+            PrettyOutput.print(f"平台目录不存在: {directory}", OutputType.WARNING)
             return platforms
             
         # 获取目录的包名
@@ -201,7 +201,7 @@ class PlatformRegistry:
             BasePlatform: Platform instance
         """
         if name not in self.platforms:
-            PrettyOutput.print(f"未找到平台: {name}", OutputType.ERROR)
+            PrettyOutput.print(f"未找到平台: {name}", OutputType.WARNING)
             return None
             
         try:

jarvis/jarvis_platform_manager/main.py

@@ -55,7 +55,7 @@ def chat_with_model(platform_name: str, model_name: str):
     # Create platform instance
     platform = registry.create_platform(platform_name)
     if not platform:
-        PrettyOutput.print(f"创建平台 {platform_name} 失败", OutputType.ERROR)
+        PrettyOutput.print(f"创建平台 {platform_name} 失败", OutputType.WARNING)
         return
     
     try:
@@ -108,7 +108,7 @@ def chat_with_model(platform_name: str, model_name: str):
 # Helper function for platform and model validation
 def validate_platform_model(args):
     if not args.platform or not args.model:
-        PrettyOutput.print("请指定平台和模型。使用 'jarvis info' 查看可用平台和模型。", OutputType.ERROR)
+        PrettyOutput.print("请指定平台和模型。使用 'jarvis info' 查看可用平台和模型。", OutputType.WARNING)
         return False
     return True
 

jarvis/jarvis_tools/ask_codebase.py

@@ -89,7 +89,7 @@ def main():
     if result["success"]:
         PrettyOutput.print(result["stdout"], OutputType.INFO, lang="markdown")
     else:
-        PrettyOutput.print(result["stderr"], OutputType.ERROR)
+        PrettyOutput.print(result["stderr"], OutputType.WARNING)
         
 
 if __name__ == "__main__":

jarvis/jarvis_tools/code_review.py

@@ -241,7 +241,7 @@ def main():
         PrettyOutput.print(report, OutputType.SUCCESS, lang="yaml")
         
     else:
-        PrettyOutput.print(result["stderr"], OutputType.ERROR)
+        PrettyOutput.print(result["stderr"], OutputType.WARNING)
 
 if __name__ == "__main__":
     main()

jarvis/jarvis_tools/create_code_agent.py

@@ -109,7 +109,7 @@ def main():
     if result["success"]:
         PrettyOutput.print(result["stdout"], OutputType.SUCCESS)
     else:
-        PrettyOutput.print(result["stderr"], OutputType.ERROR)
+        PrettyOutput.print(result["stderr"], OutputType.WARNING)
 
 if __name__ == "__main__":
     main()

jarvis/jarvis_tools/file_operation.py

@@ -57,9 +57,6 @@ class FileOperationTool:
                 content = open(abs_path, 'r', encoding='utf-8').read()
                 output = f"File: {abs_path}\n{content}"
                 
-                # Print file content
-                PrettyOutput.print(f"读取文件: {abs_path}\n{content}", OutputType.INFO)
-                
                 return {
                     "success": True,
                     "stdout": output,

jarvis/jarvis_tools/git_commiter.py

@@ -1,9 +1,9 @@
 import os
 import re
 import shlex
+import subprocess
 from typing import Dict, Any
 import tempfile
-
 import yaml
 from jarvis.jarvis_platform.registry import PlatformRegistry
 from jarvis.jarvis_utils import OutputType, PrettyOutput, has_uncommitted_changes, init_env
@@ -27,7 +27,13 @@ class GitCommitTool:
         return "<<FORMAT VIOLATION>> Invalid commit message structure"
     
     def _get_last_commit_hash(self):
-        return os.popen("git log -1 --pretty=%H").read().strip()
+        process = subprocess.Popen(
+            ["git", "log", "-1", "--pretty=%H"],
+            stdout=subprocess.PIPE,
+            stderr=subprocess.PIPE
+        )
+        stdout, _ = process.communicate()
+        return stdout.decode().strip()
 
     def execute(self, args: Dict) -> Dict[str, Any]:
         """Execute automatic commit process with support for multi-line messages and special characters"""
@@ -37,10 +43,19 @@ class GitCommitTool:
                 return {"success": True, "stdout": "No changes to commit", "stderr": ""}
             
             PrettyOutput.print("准备添加文件到提交...", OutputType.SYSTEM)
-            os.system("git add .")
+            subprocess.Popen(
+                ["git", "add", "."],
+                stdout=subprocess.DEVNULL,
+                stderr=subprocess.DEVNULL
+            ).wait()
             
             PrettyOutput.print("获取差异...", OutputType.SYSTEM)
-            diff = os.popen("git diff --cached --exit-code").read()
+            process = subprocess.Popen(
+                ["git", "diff", "--cached", "--exit-code"],
+                stdout=subprocess.PIPE,
+                stderr=subprocess.PIPE
+            )
+            diff = process.communicate()[0].decode()
             PrettyOutput.print(diff, OutputType.CODE, lang="diff")
             
             prompt = f'''Generate commit message with the paranoia of someone who's lost production data:
@@ -69,7 +84,7 @@ YOU MUST USE EXACTLY THIS FORMAT:
             
             PrettyOutput.print("生成提交消息...", OutputType.SYSTEM)
             platform = PlatformRegistry().get_codegen_platform()
-            # platform.set_suppress_output(True)
+            platform.set_suppress_output(True)
             commit_message = platform.chat_until_success(prompt)
             commit_message = self._extract_commit_message(commit_message)
             
@@ -77,9 +92,13 @@ YOU MUST USE EXACTLY THIS FORMAT:
             with tempfile.NamedTemporaryFile(mode='w', delete=True) as tmp_file:
                 tmp_file.write(commit_message)
                 tmp_file.flush()  # 确保内容写入文件
-                commit_cmd = f"git commit -F {tmp_file.name}"
+                commit_cmd = ["git", "commit", "-F", tmp_file.name]
                 PrettyOutput.print("提交...", OutputType.INFO)
-                os.system(commit_cmd)
+                subprocess.Popen(
+                    commit_cmd,
+                    stdout=subprocess.DEVNULL,
+                    stderr=subprocess.DEVNULL
+                ).wait()
 
             commit_hash = self._get_last_commit_hash()
             PrettyOutput.print(f"提交哈希: {commit_hash}\n提交消息: {commit_message}", OutputType.SUCCESS)

jarvis/jarvis_tools/rag.py

@@ -135,7 +135,7 @@ def main():
     if result["success"]:
         PrettyOutput.print(f"{result['stdout']}", OutputType.INFO, lang="markdown")
     else:
-        PrettyOutput.print(result["stderr"], OutputType.ERROR)
+        PrettyOutput.print(result["stderr"], OutputType.WARNING)
 
 if __name__ == "__main__":
     main() 

jarvis/jarvis_tools/read_code.py

@@ -52,7 +52,7 @@ class ReadCodeTool:
         """
         try:
             abs_path = os.path.abspath(filepath.strip())
-            PrettyOutput.print(f"正在读取代码文件：{abs_path}", OutputType.INFO)
+            PrettyOutput.print(f"正在读取代码文件：{abs_path} [范围: [{start_line},{end_line}]]", OutputType.INFO)
             
             if not os.path.exists(abs_path):
                 PrettyOutput.print(f"文件不存在: {abs_path}", OutputType.WARNING)
@@ -120,7 +120,6 @@ class ReadCodeTool:
                 
             content = "".join(formatted_lines)
             output = f"File: {filepath}\nLines: [{start_line}, {end_line}]\n{content}"
-            PrettyOutput.print(output, OutputType.CODE)
             return {
                 "success": True,
                 "stdout": output,

jarvis/jarvis_utils/__init__.py

@@ -23,9 +23,11 @@ import psutil
 from rich.console import Console
 from rich.theme import Theme
 from rich.panel import Panel
+from rich.box import HEAVY
 from rich.text import Text
 from rich.traceback import install as install_rich_traceback
 from rich.syntax import Syntax
+from rich.style import Style as RichStyle
 
 from prompt_toolkit.completion import Completer, Completion, PathCompleter
 from prompt_toolkit.document import Document
@@ -45,18 +47,18 @@ install_rich_traceback()
 
 # Create console with custom theme
 custom_theme = Theme({
-    "info": "yellow",
-    "warning": "yellow",
-    "error": "red",
-    "success": "green",
-    "system": "cyan",
-    "code": "green",
-    "result": "blue",
-    "planning": "magenta",
-    "progress": "white",
-    "debug": "blue",
-    "user": "green",
-    "tool": "yellow",
+    "INFO": "yellow",
+    "WARNING": "yellow",
+    "ERROR": "red",
+    "SUCCESS": "green",
+    "SYSTEM": "cyan",
+    "CODE": "green",
+    "RESULT": "blue",
+    "PLANNING": "magenta",
+    "PROGRESS": "white",
+    "DEBUG": "blue",
+    "USER": "green",
+    "TOOL": "yellow",
 })
 
 console = Console(theme=custom_theme)
@@ -76,7 +78,7 @@ def set_agent(agent_name: str, agent: Any):
     current_agent_name = agent_name
 
 def get_agent_list():
-    return "[" + str(len(global_agents)) + "]" + current_agent_name if global_agents else "No Agent"
+    return "[" + str(len(global_agents)) + "]" + current_agent_name if global_agents else ""
 
 def delete_agent(agent_name: str):
     if agent_name in global_agents:
@@ -85,18 +87,18 @@ def delete_agent(agent_name: str):
         current_agent_name = ""
 
 class OutputType(Enum):
-    SYSTEM = "system"      # AI assistant message
-    CODE = "code"         # Code related
-    RESULT = "result"     # Tool execution result
-    ERROR = "error"       # Error information
-    INFO = "info"         # System prompt
-    PLANNING = "planning" # Task planning
-    PROGRESS = "progress" # Execution progress
-    SUCCESS = "success"   # Success information
-    WARNING = "warning"   # Warning information
-    DEBUG = "debug"       # Debug information
-    USER = "user"         # User input
-    TOOL = "tool"         # Tool call
+    SYSTEM = "SYSTEM"      # AI assistant message
+    CODE = "CODE"         # Code related
+    RESULT = "RESULT"     # Tool execution result
+    ERROR = "ERROR"       # Error information
+    INFO = "INFO"         # System prompt
+    PLANNING = "PLANNING" # Task planning
+    PROGRESS = "PROGRESS" # Execution progress
+    SUCCESS = "SUCCESS"   # Success information
+    WARNING = "WARNING"   # Warning information
+    DEBUG = "DEBUG"       # Debug information
+    USER = "USER"         # User input
+    TOOL = "TOOL"         # Tool call
 
 class PrettyOutput:
     """Pretty output using rich"""
@@ -164,18 +166,20 @@ class PrettyOutput:
             return default_lang
 
     @staticmethod
-    def _format(text: str, output_type: OutputType, timestamp: bool = True) -> Text:
+    def _format(output_type: OutputType, timestamp: bool = True) -> Text:
         """Format output text using rich Text"""
         # Create rich Text object
         formatted = Text()
         
         # Add timestamp and agent info
         if timestamp:
-            formatted.append(f"[{datetime.now().strftime('%H:%M:%S')}] ", style="white")
-        formatted.append(f"{get_agent_list()}", style="blue")
+            formatted.append(f"[{datetime.now().strftime('%H:%M:%S')}][{output_type.value}]", style=output_type.value)
+        agent_info = get_agent_list()
+        if agent_info:  # Only add brackets if there's agent info
+            formatted.append(f"[{agent_info}]", style="blue")
         # Add icon
         icon = PrettyOutput._ICONS.get(output_type, "")
-        formatted.append(f"{icon} ", style=output_type.value)
+        formatted.append(f" {icon} ", style=output_type.value)
         
         return formatted
 
@@ -190,27 +194,86 @@ class PrettyOutput:
             lang: Language for syntax highlighting
             traceback: Whether to show traceback for errors
         """
-        from rich.style import Style as RichStyle
+        
         
         # Define styles for different output types
         styles = {
-            OutputType.SYSTEM: RichStyle(color="cyan", bold=True),
-            OutputType.CODE: RichStyle(color="green"),
-            OutputType.RESULT: RichStyle(color="blue"),
-            OutputType.ERROR: RichStyle(color="red", bold=True),
-            OutputType.INFO: RichStyle(color="yellow"),
-            OutputType.PLANNING: RichStyle(color="magenta"),
-            OutputType.PROGRESS: RichStyle(color="white"),
-            OutputType.SUCCESS: RichStyle(color="green", bold=True),
-            OutputType.WARNING: RichStyle(color="yellow", bold=True),
-            OutputType.DEBUG: RichStyle(color="blue", dim=True),
-            OutputType.USER: RichStyle(color="green"),
-            OutputType.TOOL: RichStyle(color="yellow", italic=True)
+            OutputType.SYSTEM: RichStyle(
+                color="bright_cyan", 
+                italic=True,
+                bold=True, 
+            ),
+            OutputType.CODE: RichStyle(
+                color="green", 
+                italic=True,
+                bgcolor="#1a1a1a",
+                frame=True
+            ),
+            OutputType.RESULT: RichStyle(
+                color="bright_blue",
+                bold=True,
+                italic=True,
+                bgcolor="navy_blue"
+            ),
+            OutputType.ERROR: RichStyle(
+                color="red", 
+                bold=True,
+                italic=True,
+                blink=True,
+                bgcolor="dark_red",
+            ),
+            OutputType.INFO: RichStyle(
+                color="gold1",
+                dim=True,
+                bgcolor="grey11",
+                italic=True
+            ),
+            OutputType.PLANNING: RichStyle(
+                color="purple", 
+                italic=True,
+                bold=True,
+            ),
+            OutputType.PROGRESS: RichStyle(
+                color="white", 
+                encircle=True,
+                italic=True,
+            ),
+            OutputType.SUCCESS: RichStyle(
+                color="bright_green", 
+                bold=True,
+                strike=False,
+                meta={"icon": "✓"},
+                italic=True
+            ),
+            OutputType.WARNING: RichStyle(
+                color="yellow", 
+                bold=True,
+                blink2=True,
+                bgcolor="dark_orange",
+                italic=True
+            ),
+            OutputType.DEBUG: RichStyle(
+                color="grey58",
+                dim=True,
+                italic=True,
+                conceal=True
+            ),
+            OutputType.USER: RichStyle(
+                color="spring_green2",
+                reverse=True,
+                frame=True,
+                italic=True
+            ),
+            OutputType.TOOL: RichStyle(
+                color="dark_sea_green4",
+                italic=True,
+                bgcolor="grey19",
+            )
         }
         
         # Get formatted header
         lang = lang if lang is not None else PrettyOutput._detect_language(text, default_lang='markdown')
-        header = PrettyOutput._format("", output_type, timestamp)
+        header = PrettyOutput._format(output_type, timestamp)
         
         # Create syntax highlighted content
         content = Syntax(
@@ -218,7 +281,6 @@ class PrettyOutput:
             lang,
             theme="monokai",
             word_wrap=True,
-            background_color="default"
         )
         
         # Create panel with styling
@@ -229,7 +291,8 @@ class PrettyOutput:
             title=header,
             title_align="left",
             padding=(0, 0),
-            highlight=True
+            highlight=True,
+            box=HEAVY,
         )
         
         # Print panel
@@ -253,12 +316,22 @@ class PrettyOutput:
     @staticmethod
     def print_stream(text: str):
         """Print stream output without line break"""
-        console.print(text, style="system", end="")
+        # 使用进度类型样式
+        style = PrettyOutput._get_style(OutputType.SYSTEM)
+        console.print(text, style=style, end="")
 
     @staticmethod
     def print_stream_end():
         """End stream output with line break"""
-        console.print()
+        # 结束符样式
+        end_style = PrettyOutput._get_style(OutputType.SUCCESS)
+        console.print("\n", style=end_style)
+        console.file.flush()
+
+    @staticmethod
+    def _get_style(output_type: OutputType) -> RichStyle:
+        """Get pre-defined RichStyle for output type"""
+        return console.get_style(output_type.value)
 
 def get_single_line_input(tip: str) -> str:
     """Get single line input, support direction key, history function, etc."""

{jarvis_ai_assistant-0.1.120.dist-info → jarvis_ai_assistant-0.1.122.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: jarvis-ai-assistant
-Version: 0.1.120
+Version: 0.1.122
 Summary: Jarvis: An AI assistant that uses tools to interact with the system
 Home-page: https://github.com/skyfireitdiy/Jarvis
 Author: skyfire