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

jarvis/__init__.py

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

jarvis/jarvis_agent/__init__.py

@@ -396,7 +396,7 @@ class Agent:
         """
         return execute_tool_call(response, self)
 
-    def _complete_task(self) -> str:
+    def _complete_task(self, auto_completed: bool = False) -> str:
         """完成任务并生成总结(如果需要)
 
         返回:
@@ -407,8 +407,26 @@ class Agent:
             2. 对于子Agent: 可能会生成总结(如果启用)
             3. 使用spinner显示生成状态
         """
+        satisfaction_feedback = ""
+
+        if not auto_completed and self.use_analysis:
+            if user_confirm("您对本次任务的完成是否满意？", True):
+                satisfaction_feedback = "\n\n用户对本次任务的完成表示满意。"
+            else:
+                feedback = self.multiline_inputer(
+                    "请提供您的反馈意见（可留空直接回车）:"
+                )
+                if feedback:
+                    satisfaction_feedback = (
+                        f"\n\n用户对本次任务的完成不满意，反馈意见如下：\n{feedback}"
+                    )
+                else:
+                    satisfaction_feedback = (
+                        "\n\n用户对本次任务的完成不满意，未提供具体反馈意见。"
+                    )
+
         if self.use_analysis:
-            self._analysis_task()
+            self._analysis_task(satisfaction_feedback)
         if self.need_summary:
             print("📄 正在生成总结...")
             self.session.prompt = self.summary_prompt
@@ -420,11 +438,13 @@ class Agent:
 
         return "任务完成"
 
-    def _analysis_task(self):
+    def _analysis_task(self, satisfaction_feedback: str = ""):
         print("🔍 正在分析任务...")
         try:
             # 让模型判断是否需要生成方法论
             analysis_prompt = TASK_ANALYSIS_PROMPT
+            if satisfaction_feedback:
+                analysis_prompt += satisfaction_feedback
 
             self.session.prompt = analysis_prompt
             if not self.model:
@@ -452,6 +472,23 @@ class Agent:
             else ""
         )
 
+        # 检查工具列表并添加记忆工具相关提示
+        memory_prompts = ""
+        tool_registry = self.get_tool_registry()
+        if tool_registry:
+            tool_names = [tool.name for tool in tool_registry.tools.values()]
+            
+            # 如果有save_memory工具，添加相关提示
+            if "save_memory" in tool_names:
+                memory_prompts += "\n    - 如果有关键信息需要记忆，请调用save_memory工具进行记忆："
+                memory_prompts += "\n      * project_long_term: 保存与当前项目相关的长期信息"
+                memory_prompts += "\n      * global_long_term: 保存通用的信息、用户喜好、知识、方法等"
+                memory_prompts += "\n      * short_term: 保存当前任务相关的临时信息"
+            
+            # 如果有retrieve_memory工具，添加相关提示
+            if "retrieve_memory" in tool_names:
+                memory_prompts += "\n    - 如果需要检索相关记忆信息，请调用retrieve_memory工具"
+
         addon_prompt = f"""
 <system_prompt>
     请判断是否已经完成任务，如果已经完成：
@@ -461,7 +498,7 @@ class Agent:
     - 仅包含一个操作
     - 如果信息不明确，请请求用户补充
     - 如果执行过程中连续失败5次，请使用ask_user询问用户操作
-    - 操作列表：{action_handlers}
+    - 操作列表：{action_handlers}{memory_prompts}
 </system_prompt>
 
 请继续。
@@ -511,7 +548,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 +556,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
@@ -541,7 +584,7 @@ class Agent:
                         continue
 
                     if self.auto_complete and ot("!!!COMPLETE!!!") in current_response:
-                        return self._complete_task()
+                        return self._complete_task(auto_completed=True)
 
                     # 获取用户输入
                     user_input = self.multiline_inputer(
@@ -554,7 +597,7 @@ class Agent:
                         continue
 
                     if not user_input:
-                        return self._complete_task()
+                        return self._complete_task(auto_completed=False)
 
                 except Exception as e:
                     PrettyOutput.print(f"任务失败: {str(e)}", OutputType.ERROR)
@@ -565,6 +608,23 @@ class Agent:
             return f"Task failed: {str(e)}"
 
     def _first_run(self):
+        # 获取所有记忆标签并添加到提示中
+        from jarvis.jarvis_utils.globals import get_all_memory_tags
+        
+        memory_tags = get_all_memory_tags()
+        memory_tags_prompt = ""
+        
+        if any(tags for tags in memory_tags.values()):
+            memory_tags_prompt = "\n\n系统中存在以下记忆标签，你可以使用 retrieve_memory 工具检索相关记忆："
+            for memory_type, tags in memory_tags.items():
+                if tags:
+                    type_name = {
+                        "short_term": "短期记忆",
+                        "project_long_term": "项目长期记忆", 
+                        "global_long_term": "全局长期记忆"
+                    }.get(memory_type, memory_type)
+                    memory_tags_prompt += f"\n- {type_name}: {', '.join(tags)}"
+        
         # 如果有上传文件，先上传文件
         if self.model and self.model.support_upload_files():
             if self.use_methodology:
@@ -577,12 +637,12 @@ class Agent:
                     msg = self.session.prompt
                     for handler in self.input_handler:
                         msg, _ = handler(msg, self)
-                    self.session.prompt = f"{self.session.prompt}\n\n以下是历史类似问题的执行经验，可参考：\n{load_methodology(msg, self.get_tool_registry())}"
+                    self.session.prompt = f"{self.session.prompt}\n\n以下是历史类似问题的执行经验，可参考：\n{load_methodology(msg, self.get_tool_registry())}{memory_tags_prompt}"
                 else:
                     if self.files:
-                        self.session.prompt = f"{self.session.prompt}\n\n上传的文件包含历史对话信息和方法论文件，可以从中获取一些经验信息。"
+                        self.session.prompt = f"{self.session.prompt}\n\n上传的文件包含历史对话信息和方法论文件，可以从中获取一些经验信息。{memory_tags_prompt}"
                     else:
-                        self.session.prompt = f"{self.session.prompt}\n\n上传的文件包含历史对话信息，可以从中获取一些经验信息。"
+                        self.session.prompt = f"{self.session.prompt}\n\n上传的文件包含历史对话信息，可以从中获取一些经验信息。{memory_tags_prompt}"
             elif self.files:
                 if not self.model.upload_files(self.files):
                     PrettyOutput.print(
@@ -598,6 +658,10 @@ class Agent:
                 for handler in self.input_handler:
                     msg, _ = handler(msg, self)
                 self.session.prompt = f"{self.session.prompt}\n\n以下是历史类似问题的执行经验，可参考：\n{load_methodology(msg, self.get_tool_registry())}"
+        
+        # 添加记忆标签提示
+        if memory_tags_prompt:
+            self.session.prompt = f"{self.session.prompt}{memory_tags_prompt}"
 
         self.first = False
 

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 = []
 
@@ -205,9 +204,9 @@ class EditFileHandler(OutputHandler):
                 found = False
 
                 if exact_search in modified_content:
-                    # 直接执行替换（保留所有原始格式）
+                    # 直接执行替换（保留所有原始格式），只替换第一个匹配
                     modified_content = modified_content.replace(
-                        exact_search, replace_text
+                        exact_search, replace_text, 1
                     )
                     print(f"✅ 补丁 #{patch_count} 应用成功")
                     found = True
@@ -223,7 +222,7 @@ class EditFileHandler(OutputHandler):
                         stripped_replace = replace_text[1:-1]
                         if stripped_search in modified_content:
                             modified_content = modified_content.replace(
-                                stripped_search, stripped_replace
+                                stripped_search, stripped_replace, 1
                             )
                             print(f"✅ 补丁 #{patch_count} 应用成功 (自动去除首尾换行)")
                             found = True
@@ -252,7 +251,7 @@ class EditFileHandler(OutputHandler):
                             )
                             if indented_search in modified_content:
                                 modified_content = modified_content.replace(
-                                    indented_search, indented_replace
+                                    indented_search, indented_replace, 1
                                 )
                                 print(
                                     f"✅ 补丁 #{patch_count} 应用成功 (自动增加 {space_count} 个空格缩进)"

jarvis/jarvis_code_agent/code_agent.py

@@ -65,6 +65,8 @@ class CodeAgent:
                 "ask_user",
                 "read_code",
                 "rewrite_file",
+                "save_memory",
+                "retrieve_memory",
             ]
         )
         code_system_prompt = """
@@ -269,7 +271,7 @@ class CodeAgent:
             return
 
         PrettyOutput.print(
-            "⚠️ 即将修改git换行符敏感设置，这会影响所有文件的换行符处理方式",
+            "⚠️ 正在修改git换行符敏感设置，这会影响所有文件的换行符处理方式",
             OutputType.WARNING,
         )
         print("将进行以下设置：")
@@ -277,18 +279,15 @@ class CodeAgent:
             current = current_settings.get(key, "未设置")
             print(f"  {key}: {current} -> {value}")
 
-        if user_confirm("是否继续修改git换行符敏感设置？", True):
-            for key, value in target_settings.items():
-                subprocess.run(["git", "config", key, value], check=True)
+        # 直接执行设置，不需要用户确认
+        for key, value in target_settings.items():
+            subprocess.run(["git", "config", key, value], check=True)
 
-            # 对于Windows系统，提示用户可以创建.gitattributes文件
-            if sys.platform.startswith("win"):
-                self._handle_windows_line_endings()
+        # 对于Windows系统，提示用户可以创建.gitattributes文件
+        if sys.platform.startswith("win"):
+            self._handle_windows_line_endings()
 
-            print("✅ git换行符敏感设置已更新")
-        else:
-            print("❌ 用户取消修改git换行符敏感设置")
-            sys.exit(0)
+        print("✅ git换行符敏感设置已更新")
 
     def _handle_windows_line_endings(self) -> None:
         """在Windows系统上处理换行符问题，提供建议而非强制修改"""
@@ -348,13 +347,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 +359,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 +394,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 +426,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 +451,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 +467,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 +575,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_multi_agent/main.py

@@ -7,6 +7,7 @@ import yaml
 from jarvis.jarvis_multi_agent import MultiAgent
 from jarvis.jarvis_utils.input import get_multiline_input
 from jarvis.jarvis_utils.utils import init_env
+from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 
 app = typer.Typer(help="多智能体系统启动器")
 

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():
     """运行演示，展示统计模块的功能"""