jarvis-ai-assistant 0.1.134__py3-none-any.whl → 0.1.138__py3-none-any.whl

jarvis/jarvis_utils/git_utils.py

@@ -16,10 +16,10 @@ from jarvis.jarvis_utils.output import PrettyOutput, OutputType
 def find_git_root(start_dir="."):
     """
     切换到给定路径的Git根目录，如果不是Git仓库则初始化。
-    
+
     参数:
         start_dir (str): 起始查找目录，默认为当前目录。
-    
+
     返回:
         str: Git仓库根目录路径。如果目录不是Git仓库，则会初始化一个新的Git仓库。
     """
@@ -39,28 +39,28 @@ def has_uncommitted_changes():
     """检查Git仓库中是否有未提交的更改"""
     # 静默添加所有更改
     subprocess.run(["git", "add", "."], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
-    
+
     # 检查工作目录更改
-    working_changes = subprocess.run(["git", "diff", "--exit-code"], 
-                                    stdout=subprocess.DEVNULL, 
+    working_changes = subprocess.run(["git", "diff", "--exit-code"],
+                                    stdout=subprocess.DEVNULL,
                                     stderr=subprocess.DEVNULL).returncode != 0
-    
+
     # 检查暂存区更改
-    staged_changes = subprocess.run(["git", "diff", "--cached", "--exit-code"], 
-                                   stdout=subprocess.DEVNULL, 
+    staged_changes = subprocess.run(["git", "diff", "--cached", "--exit-code"],
+                                   stdout=subprocess.DEVNULL,
                                    stderr=subprocess.DEVNULL).returncode != 0
-    
+
     # 静默重置更改
     subprocess.run(["git", "reset"], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
-    
+
     return working_changes or staged_changes
 def get_commits_between(start_hash: str, end_hash: str) -> List[Tuple[str, str]]:
     """获取两个提交哈希值之间的提交列表
-    
+
     参数：
         start_hash: 起始提交哈希值（不包含）
         end_hash: 结束提交哈希值（包含）
-        
+
     返回：
         List[Tuple[str, str]]: (提交哈希值, 提交信息) 元组列表
     """
@@ -75,20 +75,20 @@ def get_commits_between(start_hash: str, end_hash: str) -> List[Tuple[str, str]]
         if result.returncode != 0:
             PrettyOutput.print(f"获取commit历史失败: {result.stderr}", OutputType.ERROR)
             return []
-            
+
         commits = []
         for line in result.stdout.splitlines():
             if '|' in line:
                 commit_hash, message = line.split('|', 1)
                 commits.append((commit_hash, message))
         return commits
-        
+
     except Exception as e:
         PrettyOutput.print(f"获取commit历史异常: {str(e)}", OutputType.ERROR)
         return []
 def get_latest_commit_hash() -> str:
     """获取当前Git仓库的最新提交哈希值
-    
+
     返回：
         str: 提交哈希值，如果不在Git仓库或发生错误则返回空字符串
     """
@@ -106,25 +106,25 @@ def get_latest_commit_hash() -> str:
         return ""
 def get_modified_line_ranges() -> Dict[str, Tuple[int, int]]:
     """从Git差异中获取所有更改文件的修改行范围
-    
+
     返回：
         字典，将文件路径映射到包含修改部分的（起始行, 结束行）范围元组。
         行号从1开始。
     """
     # 获取所有文件的Git差异
     diff_output = os.popen("git show").read()
-    
+
     # 解析差异以获取修改的文件及其行范围
     result = {}
     current_file = None
-    
+
     for line in diff_output.splitlines():
         # 匹配类似"+++ b/path/to/file"的行
         file_match = re.match(r"^\+\+\+ b/(.*)", line)
         if file_match:
             current_file = file_match.group(1)
             continue
-            
+
         # 匹配类似"@@ -100,5 +100,7 @@"的行，其中+部分显示新行
         range_match = re.match(r"^@@ -\d+(?:,\d+)? \+(\d+)(?:,(\d+))? @@", line)
         if range_match and current_file:
@@ -132,5 +132,5 @@ def get_modified_line_ranges() -> Dict[str, Tuple[int, int]]:
             line_count = int(range_match.group(2)) if range_match.group(2) else 1
             end_line = start_line + line_count - 1
             result[current_file] = (start_line, end_line)
-    
+
     return result

jarvis/jarvis_utils/globals.py

@@ -6,7 +6,7 @@
 - 带有自定义主题的控制台配置
 - 环境初始化
 """
-from typing import Any, Set
+from typing import Any, Set, Dict
 import colorama
 import os
 from rich.console import Console
@@ -17,6 +17,7 @@ colorama.init()
 os.environ["TOKENIZERS_PARALLELISM"] = "false"
 # 全局代理管理
 global_agents: Set[str] = set()
+global_read_files: Dict[str, Set[str]] = {}
 current_agent_name: str = ""
 # 使用自定义主题配置rich控制台
 custom_theme = Theme({
@@ -37,10 +38,10 @@ console = Console(theme=custom_theme)
 def make_agent_name(agent_name: str) -> str:
     """
     通过附加后缀生成唯一的代理名称（如果必要）。
-    
+
     参数：
         agent_name: 基础代理名称
-        
+
     返回：
         str: 唯一的代理名称
     """
@@ -53,7 +54,7 @@ def make_agent_name(agent_name: str) -> str:
 def set_agent(agent_name: str, agent: Any) -> None:
     """
     设置当前代理并将其添加到全局代理集合中。
-    
+
     参数：
         agent_name: 代理名称
         agent: 代理对象
@@ -64,7 +65,7 @@ def set_agent(agent_name: str, agent: Any) -> None:
 def get_agent_list() -> str:
     """
     获取表示当前代理状态的格式化字符串。
-    
+
     返回：
         str: 包含代理数量和当前代理名称的格式化字符串
     """
@@ -72,7 +73,7 @@ def get_agent_list() -> str:
 def delete_agent(agent_name: str) -> None:
     """
     从全局代理集合中删除一个代理。
-    
+
     参数：
         agent_name: 要删除的代理名称
     """
@@ -80,3 +81,14 @@ def delete_agent(agent_name: str) -> None:
         global_agents.remove(agent_name)
         global current_agent_name
         current_agent_name = ""
+
+def add_read_file_record(file_path: str):
+    if current_agent_name not in global_read_files:
+        global_read_files[current_agent_name] = set()
+    global_read_files[current_agent_name].add(file_path)
+
+def has_read_file(file_path: str) -> bool:
+    return not os.path.exists(file_path) or (current_agent_name in global_read_files and file_path in global_read_files[current_agent_name])
+
+def clear_read_file_record():
+    global_read_files.pop(current_agent_name, None)

jarvis/jarvis_utils/input.py

@@ -19,10 +19,10 @@ from jarvis.jarvis_utils.output import PrettyOutput, OutputType
 def get_single_line_input(tip: str) -> str:
     """
     获取支持历史记录的单行输入。
-    
+
     参数：
         tip: 要显示的提示信息
-        
+
     返回：
         str: 用户的输入
     """
@@ -34,7 +34,7 @@ def get_single_line_input(tip: str) -> str:
 class FileCompleter(Completer):
     """
     带有模糊匹配的文件路径自定义补全器。
-    
+
     属性：
         path_completer: 基础路径补全器
         max_suggestions: 显示的最大建议数量
@@ -48,11 +48,11 @@ class FileCompleter(Completer):
     def get_completions(self, document: Document, complete_event) -> Completion: # type: ignore
         """
         生成带有模糊匹配的文件路径补全建议。
-        
+
         参数：
             document: 当前正在编辑的文档
             complete_event: 补全事件
-            
+
         生成：
             Completion: 建议的补全项
         """
@@ -81,6 +81,7 @@ class FileCompleter(Completer):
                 (ot("RAG"), '知识库检索'),
                 (ot("Summary"), '总结'),
                 (ot("Clear"), '清除历史'),
+                (ot("Methodology"), '查找相关方法论'),
             ]
             for name, desc in default_suggestions:
                 yield Completion(
@@ -98,8 +99,8 @@ class FileCompleter(Completer):
         all_files = []
         try:
             import subprocess
-            result = subprocess.run(['git', 'ls-files'], 
-                                 stdout=subprocess.PIPE, 
+            result = subprocess.run(['git', 'ls-files'],
+                                 stdout=subprocess.PIPE,
                                  stderr=subprocess.PIPE,
                                  text=True)
             if result.returncode == 0:
@@ -129,10 +130,10 @@ class FileCompleter(Completer):
 def get_multiline_input(tip: str) -> str:
     """
     获取带有增强补全和确认功能的多行输入。
-    
+
     参数：
         tip: 要显示的提示信息
-        
+
     返回：
         str: 用户的输入，如果取消则返回空字符串
     """