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

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: 用户的输入，如果取消则返回空字符串
     """