jarvis-ai-assistant 0.3.30__py3-none-any.whl → 0.7.6__py3-none-any.whl

jarvis/jarvis_tools/clear_memory.py

@@ -4,7 +4,6 @@ from pathlib import Path
 from typing import Any, Dict, List, Optional
 
 from jarvis.jarvis_utils.config import get_data_dir
-from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 from jarvis.jarvis_utils.globals import (
     clear_short_term_memories,
     short_term_memories,
@@ -15,21 +14,7 @@ class ClearMemoryTool:
     """清除记忆工具，用于批量清除指定的记忆"""
 
     name = "clear_memory"
-    description = """批量清除指定的记忆。
-    
-    支持的清除方式：
-    1. 按记忆类型清除所有记忆
-    2. 按标签清除特定记忆
-    3. 按记忆ID清除单个记忆
-    
-    支持的记忆类型：
-    - project_long_term: 项目长期记忆
-    - global_long_term: 全局长期记忆
-    - short_term: 短期记忆
-    - all: 所有类型的记忆
-    
-    注意：清除操作不可恢复，请谨慎使用
-    """
+    description = "批量清除指定的记忆。支持按类型、标签或ID清除。记忆类型：project_long_term/global_long_term/short_term/all。注意：清除操作不可恢复。"
 
     parameters = {
         "type": "object",
@@ -160,9 +145,7 @@ class ClearMemoryTool:
                     removed_count += 1
 
             except Exception as e:
-                PrettyOutput.print(
-                    f"处理记忆文件 {memory_file} 时出错: {str(e)}", OutputType.WARNING
-                )
+                print(f"⚠️ 处理记忆文件 {memory_file} 时出错: {str(e)}")
 
         # 如果目录为空，可以删除目录
         if not any(memory_dir.iterdir()) and memory_dir != self.project_memory_dir:
@@ -235,5 +218,5 @@ class ClearMemoryTool:
 
         except Exception as e:
             error_msg = f"清除记忆失败: {str(e)}"
-            PrettyOutput.print(error_msg, OutputType.ERROR)
+            print(f"❌ {error_msg}")
             return {"success": False, "stdout": "", "stderr": error_msg}

jarvis/jarvis_tools/cli/main.py

@@ -11,6 +11,7 @@ from jarvis.jarvis_utils.utils import init_env
 app = typer.Typer(help="Jarvis 工具系统命令行界面")
 
 
+
 @app.command("list")
 def list_tools(
     as_json: bool = typer.Option(False, "--json", help="以JSON格式输出"),
@@ -37,7 +38,7 @@ def list_tools(
                 lang="json",
             )
     else:
-        PrettyOutput.section("可用工具列表", OutputType.SYSTEM)
+        print("📋 可用工具列表")
         # 为避免 PrettyOutput 对每行加框造成信息稀疏，先拼接字符串再统一打印
         lines = []
         import json as _json  # local import to ensure available
@@ -53,7 +54,7 @@ def list_tools(
                 except Exception:
                     lines.append(str(tool.get("parameters")))
                 lines.append("```")
-        PrettyOutput.print("\n".join(lines), OutputType.INFO, lang="markdown")
+        PrettyOutput.print("\n".join(lines), OutputType.CODE, lang="markdown")
 
 
 @app.command("stat")
@@ -89,7 +90,7 @@ def stat_tools(
             )
         else:
             time_desc = f"最近{last_days}天" if last_days else "所有历史"
-            PrettyOutput.section(f"工具调用统计 ({time_desc})", OutputType.SYSTEM)
+            print(f"📊 工具调用统计 ({time_desc})")
             if table_data:
                 PrettyOutput.print(
                     tabulate(
@@ -98,15 +99,12 @@ def stat_tools(
                     OutputType.CODE,
                     lang="text",
                 )
-                PrettyOutput.print(
-                    f"\n总计: {len(table_data)} 个工具被使用，共 {sum(x[1] for x in table_data)} 次调用",
-                    OutputType.INFO,
-                )
+                print(f"ℹ️ 总计: {len(table_data)} 个工具被使用，共 {sum(x[1] for x in table_data)} 次调用")
             else:
-                PrettyOutput.print("暂无工具调用记录", OutputType.INFO)
+                print("ℹ️ 暂无工具调用记录")
     else:
         # 使用 stats 系统的高级功能
-        PrettyOutput.section("工具组统计", OutputType.SYSTEM)
+        print("📊 工具组统计")
         # 显示所有标记为 tool 组的指标
         metrics = StatsManager.list_metrics()
         tool_metrics = []
@@ -165,7 +163,7 @@ def stat_tools(
                             tags={"group": "tool"},
                         )
         else:
-            PrettyOutput.print("暂无工具调用记录", OutputType.INFO)
+            print("ℹ️ 暂无工具调用记录")
 
 
 @app.command("call")
@@ -181,24 +179,24 @@ def call_tool(
     tool_obj = registry.get_tool(tool_name)
 
     if not tool_obj:
-        PrettyOutput.print(f"错误: 工具 '{tool_name}' 不存在", OutputType.ERROR)
+        print(f"❌ 错误: 工具 '{tool_name}' 不存在")
         available_tools = ", ".join([t["name"] for t in registry.get_all_tools()])
-        PrettyOutput.print(f"可用工具: {available_tools}", OutputType.INFO)
+        print(f"ℹ️ 可用工具: {available_tools}")
         raise typer.Exit(code=1)
 
     tool_args = {}
     if args:
         try:
             tool_args = json.loads(args)
-        except json.JSONDecodeError:
-            PrettyOutput.print("错误: 参数必须是有效的JSON格式", OutputType.ERROR)
+        except Exception:
+            print("❌ 错误: 参数必须是有效的JSON格式")
             raise typer.Exit(code=1)
     elif args_file:
         try:
             with open(args_file, "r", encoding="utf-8") as f:
                 tool_args = json.load(f)
-        except (json.JSONDecodeError, FileNotFoundError) as e:
-            PrettyOutput.print(f"错误: 无法从文件加载参数: {str(e)}", OutputType.ERROR)
+        except (Exception, FileNotFoundError) as e:
+            print(f"❌ 错误: 无法从文件加载参数: {str(e)}")
             raise typer.Exit(code=1)
 
     required_params = tool_obj.parameters.get("required", [])
@@ -216,23 +214,23 @@ def call_tool(
             param_info = params.get(param_name, {})
             desc = param_info.get("description", "无描述")
             lines.append(f"  - {param_name}: {desc}")
-        PrettyOutput.print("\n".join(lines), OutputType.ERROR)
+        print("❌ " + "\n❌ ".join(lines))
         raise typer.Exit(code=1)
 
     result = registry.execute_tool(tool_name, tool_args)
 
     if result["success"]:
-        PrettyOutput.section(f"工具 {tool_name} 执行成功", OutputType.SUCCESS)
+        print(f"✅ 工具 {tool_name} 执行成功")
     else:
-        PrettyOutput.section(f"工具 {tool_name} 执行失败", OutputType.ERROR)
+        print(f"❌ 工具 {tool_name} 执行失败")
 
     if result.get("stdout"):
-        PrettyOutput.print("\n输出:", OutputType.INFO)
+        print("\n📤 输出:")
         PrettyOutput.print(result["stdout"], OutputType.CODE, lang="text")
 
     if result.get("stderr"):
-        PrettyOutput.print("\n错误:", OutputType.ERROR)
-        PrettyOutput.print(result["stderr"], OutputType.ERROR, lang="text")
+        print("\n❌ 错误:")
+        PrettyOutput.print(result["stderr"], OutputType.CODE, lang="text")
 
     if not result["success"]:
         raise typer.Exit(code=1)
@@ -240,7 +238,7 @@ def call_tool(
 
 def cli():
     """Typer application entry point"""
-    init_env("欢迎使用 Jarvis-Tools，您的工具系统已准备就绪！")
+    init_env()
     app()