jarvis-ai-assistant 0.2.4__py3-none-any.whl → 0.2.5__py3-none-any.whl

jarvis/__init__.py

@@ -1,4 +1,4 @@
 # -*- coding: utf-8 -*-
 """Jarvis AI Assistant"""
 
-__version__ = "0.2.4"
+__version__ = "0.2.5"

jarvis/jarvis_agent/__init__.py

@@ -511,7 +511,7 @@ class Agent:
                         )
                         if user_input:
                             run_input_handlers = True
-                            # 如果有工具调用且用户确认继续，则将干预信息和工具执行结果拼接为prompt
+                            # 如果有工具调用且用户确认继续，则继续执行工具调用
                             if any(
                                 handler.can_handle(current_response)
                                 for handler in self.output_handler
@@ -519,13 +519,19 @@ class Agent:
                                 if user_confirm(
                                     "检测到有工具调用，是否继续处理工具调用？", True
                                 ):
-                                    self.session.prompt = (
-                                        f"{user_input}\n\n{current_response}"
-                                    )
-                                    run_input_handlers = False
+                                    # 先添加用户干预信息到session
+                                    self.session.prompt = f"被用户中断，用户补充信息为：{user_input}\n\n用户同意继续工具调用。"
+                                    # 继续执行下面的工具调用逻辑，不要continue跳过
+                                else:
+                                    # 用户选择不继续处理工具调用
+                                    self.session.prompt = f"被用户中断，用户补充信息为：{user_input}\n\n检测到有工具调用，但被用户拒绝执行。请根据用户的补充信息重新考虑下一步操作。"
                                     continue
-                            self.session.prompt += f"{user_input}"
-                            continue
+                            else:
+                                # 没有检测到工具调用
+                                self.session.prompt = (
+                                    f"被用户中断，用户补充信息为：{user_input}"
+                                )
+                                continue
 
                     need_return, self.session.prompt = self._call_tools(
                         current_response

jarvis/jarvis_agent/edit_file_handler.py

@@ -58,8 +58,7 @@ class EditFileHandler(OutputHandler):
         
         # 记录 edit_file 工具调用统计
         from jarvis.jarvis_stats.stats import StatsManager
-        stats_manager = StatsManager()
-        stats_manager.increment("edit_file", group="tool")
+        StatsManager.increment("edit_file", group="tool")
 
         results = []
 

jarvis/jarvis_code_agent/code_agent.py

@@ -348,13 +348,11 @@ class CodeAgent:
         from jarvis.jarvis_stats.stats import StatsManager
         import re
 
-        stats_manager = StatsManager()
-
         # 匹配插入行数
         insertions_match = re.search(r"(\d+)\s+insertions?\(\+\)", diff_text)
         if insertions_match:
             insertions = int(insertions_match.group(1))
-            stats_manager.increment(
+            StatsManager.increment(
                 "code_lines_inserted", amount=insertions, group="code_agent"
             )
 
@@ -362,7 +360,7 @@ class CodeAgent:
         deletions_match = re.search(r"(\d+)\s+deletions?\(\-\)", diff_text)
         if deletions_match:
             deletions = int(deletions_match.group(1))
-            stats_manager.increment(
+            StatsManager.increment(
                 "code_lines_deleted", amount=deletions, group="code_agent"
             )
 
@@ -397,8 +395,7 @@ class CodeAgent:
             # 用户确认修改，统计修改次数
             from jarvis.jarvis_stats.stats import StatsManager
 
-            stats_manager = StatsManager()
-            stats_manager.increment("code_modification_confirmed", group="code_agent")
+            StatsManager.increment("code_modification_confirmed", group="code_agent")
 
             try:
                 confirm_add_new_files()
@@ -430,7 +427,7 @@ class CodeAgent:
                 )
 
                 # 统计提交次数
-                stats_manager.increment("code_commits_accepted", group="code_agent")
+                StatsManager.increment("code_commits_accepted", group="code_agent")
             except subprocess.CalledProcessError as e:
                 PrettyOutput.print(f"提交失败: {str(e)}", OutputType.ERROR)
 
@@ -455,8 +452,7 @@ class CodeAgent:
             # 统计生成的commit数量
             from jarvis.jarvis_stats.stats import StatsManager
 
-            stats_manager = StatsManager()
-            stats_manager.increment("commits_generated", group="code_agent")
+            StatsManager.increment("commits_generated", group="code_agent")
 
             commit_messages = "检测到以下提交记录:\n" + "\n".join(
                 f"- {commit_hash[:7]}: {message}" for commit_hash, message in commits
@@ -472,8 +468,7 @@ class CodeAgent:
             # 统计接受的commit数量
             from jarvis.jarvis_stats.stats import StatsManager
 
-            stats_manager = StatsManager()
-            stats_manager.increment("commits_accepted", group="code_agent")
+            StatsManager.increment("commits_accepted", group="code_agent")
 
             subprocess.run(
                 ["git", "reset", "--mixed", str(start_commit)],
@@ -581,8 +576,7 @@ class CodeAgent:
                 # 统计修改次数
                 from jarvis.jarvis_stats.stats import StatsManager
 
-                stats_manager = StatsManager()
-                stats_manager.increment("code_modifications", group="code_agent")
+                StatsManager.increment("code_modifications", group="code_agent")
 
                 # 获取提交信息
                 end_hash = get_latest_commit_hash()

jarvis/jarvis_data/config_schema.json

@@ -111,14 +111,9 @@
       "description": "Git提交信息生成提示模板",
       "default": ""
     },
-    "JARVIS_MAX_TOKEN_COUNT": {
-      "type": "number",
-      "description": "模型能处理的最大token数量",
-      "default": 960000
-    },
     "JARVIS_MAX_INPUT_TOKEN_COUNT": {
       "type": "number",
-      "description": "模型能处理的最大输入token数量",
+      "description": "模型能处理的最大输入token数量。其他token限制基于此值计算：最大token数量=此值×100，最大大内容尺寸=此值×5",
       "default": 32000
     },
     "JARVIS_PLATFORM": {
@@ -171,17 +166,9 @@
               "type": "string",
               "default": "deep_seek_v3"
             },
-            "JARVIS_MAX_TOKEN_COUNT": {
-              "type": "number",
-              "default": 960000
-            },
             "JARVIS_MAX_INPUT_TOKEN_COUNT": {
               "type": "number",
               "default": 32000
-            },
-            "JARVIS_MAX_BIG_CONTENT_SIZE": {
-              "type": "number",
-              "default": 160000
             }
           },
           "required": [
@@ -206,11 +193,7 @@
       "description": "Jarvis数据存储目录路径",
       "default": "~/.jarvis"
     },
-    "JARVIS_MAX_BIG_CONTENT_SIZE": {
-      "type": "number",
-      "description": "最大大内容尺寸",
-      "default": 160000
-    },
+
     "JARVIS_PRETTY_OUTPUT": {
       "type": "boolean",
       "description": "是否启用美化输出",

jarvis/jarvis_stats/cli.py

@@ -4,6 +4,7 @@
 使用 typer 提供友好的命令行交互
 """
 
+import builtins
 from datetime import datetime, timedelta
 from typing import Optional, List
 import typer
@@ -183,13 +184,14 @@ def list():
     table.add_column("单位", style="green")
     table.add_column("最后更新", style="yellow")
     table.add_column("7天数据点", style="magenta")
+    table.add_column("标签", style="blue")
 
     # 获取每个指标的信息
     end_time = datetime.now()
     start_time = end_time - timedelta(days=7)
 
     for metric in metrics:
-        info = stats.storage.get_metric_info(metric)
+        info = stats._get_storage().get_metric_info(metric)
         if info:
             unit = info.get("unit", "-")
             last_updated = info.get("last_updated", "-")
@@ -202,11 +204,36 @@ def list():
                 except:
                     pass
 
-            # 获取数据点数
-            records = stats.storage.get_metrics(metric, start_time, end_time)
+            # 获取数据点数和标签
+            records = stats._get_storage().get_metrics(metric, start_time, end_time)
             count = len(records)
-
-            table.add_row(metric, unit, last_updated, str(count))
+            
+            # 收集所有唯一的标签
+            all_tags = {}
+            for record in records:
+                tags = record.get("tags", {})
+                for k, v in tags.items():
+                    if k not in all_tags:
+                        all_tags[k] = set()
+                    all_tags[k].add(v)
+            
+            # 格式化标签显示
+            tag_str = ""
+            if all_tags:
+                tag_parts = []
+                for k, values in sorted(all_tags.items()):
+                    # 使用内置的list函数
+                    values_list = sorted(builtins.list(values))
+                    if len(values_list) == 1:
+                        tag_parts.append(f"{k}={values_list[0]}")
+                    else:
+                        # 转义方括号以避免Rich markup错误
+                        tag_parts.append(f"{k}=\\[{', '.join(values_list)}\\]")
+                tag_str = ", ".join(tag_parts)
+            else:
+                tag_str = "-"
+
+            table.add_row(metric, unit, last_updated, str(count), tag_str)
 
     console.print(table)
     rprint(f"\n[green]总计: {len(metrics)} 个指标[/green]")
@@ -278,6 +305,46 @@ def export(
             rprint("[yellow]没有找到数据[/yellow]", file=sys.stderr)
 
 
+@app.command()
+def remove(
+    metric: str = typer.Argument(..., help="要删除的指标名称"),
+    yes: bool = typer.Option(False, "--yes", "-y", help="跳过确认"),
+):
+    """删除指定的指标及其所有数据"""
+    if not yes:
+        # 显示指标信息供用户确认
+        stats = StatsManager(_get_stats_dir())
+        metrics = stats.list_metrics()
+        
+        if metric not in metrics:
+            rprint(f"[red]错误：指标 '{metric}' 不存在[/red]")
+            return
+            
+        # 获取指标的基本信息
+        info = stats._get_storage().get_metric_info(metric)
+        if info:
+            unit = info.get("unit", "-")
+            last_updated = info.get("last_updated", "-")
+            
+            rprint(f"\n[yellow]准备删除指标:[/yellow]")
+            rprint(f"  名称: {metric}")
+            rprint(f"  单位: {unit}")
+            rprint(f"  最后更新: {last_updated}")
+            
+        confirm = typer.confirm(f"\n确定要删除指标 '{metric}' 及其所有数据吗？")
+        if not confirm:
+            rprint("[yellow]已取消操作[/yellow]")
+            return
+    
+    stats = StatsManager(_get_stats_dir())
+    success = stats.remove_metric(metric)
+    
+    if success:
+        rprint(f"[green]✓[/green] 已成功删除指标: {metric}")
+    else:
+        rprint(f"[red]✗[/red] 删除失败：指标 '{metric}' 不存在")
+
+
 @app.command()
 def demo():
     """运行演示，展示统计模块的功能"""