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

jarvis/jarvis_tools/find_symbol.py

@@ -11,7 +11,7 @@ class SymbolTool:
     符号查找工具
     使用agent查找代码库中的符号引用、定义和声明位置
     """
-    
+
     name = "find_symbol"
     description = "查找代码符号的引用、定义和声明位置"
     parameters = {
@@ -50,20 +50,20 @@ class SymbolTool:
         },
         "required": ["symbol"]
     }
-    
+
     def execute(self, args: Dict[str, Any]) -> Dict[str, Any]:
         """
         执行符号查找工具
-        
+
         Args:
             args: 包含参数的字典
-            
+
         Returns:
             包含执行结果的字典
         """
         # 存储原始目录
         original_dir = os.getcwd()
-        
+
         try:
             # 解析参数
             symbol = args.get("symbol", "")
@@ -71,7 +71,7 @@ class SymbolTool:
             file_extensions = args.get("file_extensions", [])
             exclude_dirs = args.get("exclude_dirs", [])
             objective = args.get("objective", "")
-            
+
             # 验证参数
             if not symbol:
                 return {
@@ -79,23 +79,23 @@ class SymbolTool:
                     "stdout": "",
                     "stderr": "必须提供符号名称"
                 }
-            
+
             # 创建agent的system prompt
             system_prompt = self._create_system_prompt(
                 symbol, root_dir, file_extensions, exclude_dirs, objective
             )
-            
+
             # 创建agent的summary prompt
             summary_prompt = self._create_summary_prompt(symbol)
-            
+
             # 切换到根目录
             os.chdir(root_dir)
-            
+
             # 构建使用的工具
             from jarvis.jarvis_tools.registry import ToolRegistry
             tool_registry = ToolRegistry()
             tool_registry.use_tools(["execute_shell", "read_code"])
-            
+
             # 创建并运行agent
             symbol_agent = Agent(
                 system_prompt=system_prompt,
@@ -107,17 +107,17 @@ class SymbolTool:
                 execute_tool_confirm=False,
                 auto_complete=True
             )
-            
+
             # 运行agent并获取结果
             task_input = f"查找符号 '{symbol}' 在代码库中的引用、定义和声明位置"
             result = symbol_agent.run(task_input)
-            
+
             return {
                 "success": True,
                 "stdout": result,
                 "stderr": ""
             }
-                
+
         except Exception as e:
             PrettyOutput.print(str(e), OutputType.ERROR)
             return {
@@ -128,39 +128,39 @@ class SymbolTool:
         finally:
             # 恢复原始目录
             os.chdir(original_dir)
-    
-    def _create_system_prompt(self, symbol: str, root_dir: str, 
+
+    def _create_system_prompt(self, symbol: str, root_dir: str,
                              file_extensions: List[str], exclude_dirs: List[str],
                              objective: str) -> str:
         """
         创建Agent的system prompt
-        
+
         Args:
             symbol: 符号名称
             root_dir: 代码库根目录
             file_extensions: 文件扩展名列表
             exclude_dirs: 排除目录列表
             objective: 分析目标
-            
+
         Returns:
             系统提示文本
         """
         file_ext_str = " ".join([f"*{ext}" for ext in file_extensions]) if file_extensions else ""
         exclude_str = " ".join([f"--glob '!{excl}'" for excl in exclude_dirs]) if exclude_dirs else ""
         objective_text = f"\n\n## 分析目标\n{objective}" if objective else ""
-        
+
         return f"""# 代码符号分析专家
 
 ## 任务描述
 查找符号 `{symbol}` 在代码库中的定义、声明和引用位置，专注于分析目标所需的信息，生成有针对性的符号分析报告。{objective_text}
 
 ## 工具使用优先级
-1. **优先使用 execute_shell 执行 rg 命令**: 
+1. **优先使用 execute_shell 执行 rg 命令**:
    - `rg -w "{symbol}" --type py` 查找Python文件中的符号
    - `rg "class\\s+{symbol}|def\\s+{symbol}" --type py` 查找定义
    - `rg -w "{symbol}" -A 2 -B 2` 查看符号上下文
 
-2. **辅以 read_code**: 
+2. **辅以 read_code**:
    - 找到符号位置后使用read_code阅读上下文
    - 读取符号定义和关键引用的完整实现
 
@@ -183,7 +183,7 @@ class SymbolTool:
 ## 符号查找工具指南
 
 ### execute_shell 搜索命令
-- **基本搜索**: 
+- **基本搜索**:
   - `rg -w "{symbol}" --type=文件类型`
   - 示例: `rg -w "{symbol}" --type py` 搜索Python文件中的符号
 
@@ -272,10 +272,10 @@ class SymbolTool:
     def _create_summary_prompt(self, symbol: str) -> str:
         """
         创建Agent的summary prompt
-        
+
         Args:
             symbol: 符号名称
-            
+
         Returns:
             总结提示文本
         """

jarvis/jarvis_tools/function_analyzer.py

@@ -11,7 +11,7 @@ class FunctionAnalyzerTool:
     函数分析工具
     使用agent深入分析函数内部实现，包括子函数调用、全局变量使用等
     """
-    
+
     name = "function_analyzer"
     description = "深入分析函数内部实现，查找子函数调用、全局变量使用等详细信息"
     parameters = {
@@ -60,20 +60,20 @@ class FunctionAnalyzerTool:
         },
         "required": ["function_name"]
     }
-    
+
     def execute(self, args: Dict[str, Any]) -> Dict[str, Any]:
         """
         执行函数分析工具
-        
+
         Args:
             args: 包含参数的字典
-            
+
         Returns:
             包含执行结果的字典
         """
         # 存储原始目录
         original_dir = os.getcwd()
-        
+
         try:
             # 解析参数
             function_name = args.get("function_name", "")
@@ -83,7 +83,7 @@ class FunctionAnalyzerTool:
             exclude_dirs = args.get("exclude_dirs", [])
             analysis_depth = args.get("analysis_depth", 1)
             objective = args.get("objective", "")
-            
+
             # 验证参数
             if not function_name:
                 return {
@@ -91,24 +91,24 @@ class FunctionAnalyzerTool:
                     "stdout": "",
                     "stderr": "必须提供函数名称"
                 }
-            
+
             # 创建agent的system prompt
             system_prompt = self._create_system_prompt(
-                function_name, file_path, root_dir, 
+                function_name, file_path, root_dir,
                 file_extensions, exclude_dirs, analysis_depth, objective
             )
-            
+
             # 创建agent的summary prompt
             summary_prompt = self._create_summary_prompt(function_name, analysis_depth)
-            
+
             # 切换到根目录
             os.chdir(root_dir)
-            
+
             # 构建使用的工具
             from jarvis.jarvis_tools.registry import ToolRegistry
             tool_registry = ToolRegistry()
             tool_registry.use_tools(["execute_shell", "read_code"])
-            
+
             # 创建并运行agent
             analyzer_agent = Agent(
                 system_prompt=system_prompt,
@@ -120,17 +120,17 @@ class FunctionAnalyzerTool:
                 execute_tool_confirm=False,
                 auto_complete=True
             )
-            
+
             # 运行agent并获取结果
             task_input = f"深入分析 '{function_name}' 函数的内部实现，包括子函数调用、全局变量使用等详细信息"
             result = analyzer_agent.run(task_input)
-            
+
             return {
                 "success": True,
                 "stdout": result,
                 "stderr": ""
             }
-                
+
         except Exception as e:
             PrettyOutput.print(str(e), OutputType.ERROR)
             return {
@@ -141,13 +141,13 @@ class FunctionAnalyzerTool:
         finally:
             # 恢复原始目录
             os.chdir(original_dir)
-    
-    def _create_system_prompt(self, function_name: str, file_path: str, root_dir: str, 
+
+    def _create_system_prompt(self, function_name: str, file_path: str, root_dir: str,
                              file_extensions: List[str], exclude_dirs: List[str],
                              analysis_depth: int, objective: str) -> str:
         """
         创建Agent的system prompt
-        
+
         Args:
             function_name: 函数名称
             file_path: 函数所在文件路径
@@ -156,29 +156,29 @@ class FunctionAnalyzerTool:
             exclude_dirs: 排除目录列表
             analysis_depth: 子函数分析深度
             objective: 分析目标
-            
+
         Returns:
             系统提示文本
         """
         file_ext_str = " ".join([f"*{ext}" for ext in file_extensions]) if file_extensions else ""
         exclude_str = " ".join([f"--glob '!{excl}'" for excl in exclude_dirs]) if exclude_dirs else ""
-        
+
         depth_description = "不分析子函数" if analysis_depth == 0 else f"分析 {analysis_depth} 层子函数"
         file_info = f"已知文件路径: {file_path}" if file_path else "需要首先查找函数定义位置"
         objective_text = f"\n\n## 分析目标\n{objective}" if objective else ""
-        
+
         return f"""# 函数实现分析专家
 
 ## 任务描述
 分析函数 `{function_name}` 的实现，专注于分析目标所需的信息，生成有针对性的函数分析报告。{objective_text}
 
 ## 工具使用优先级
-1. **首先使用 execute_shell 执行 rg 命令查找函数定义**: 
+1. **首先使用 execute_shell 执行 rg 命令查找函数定义**:
    - `rg "def\\s+{function_name}\\s*\\(" --type py` 查找Python函数定义
    - `rg "function\\s+{function_name}\\s*\\(" --type js` 查找JavaScript函数定义
    - `rg "func\\s+{function_name}\\s*\\(" --type go` 查找Go函数定义
 
-2. **优先使用 read_code 阅读函数实现**: 
+2. **优先使用 read_code 阅读函数实现**:
    - 找到函数位置后使用read_code阅读完整实现
    - 对于长函数可分段读取关键部分
 
@@ -305,16 +305,16 @@ class FunctionAnalyzerTool:
     def _create_summary_prompt(self, function_name: str, analysis_depth: int) -> str:
         """
         创建Agent的summary prompt
-        
+
         Args:
             function_name: 函数名称
             analysis_depth: 子函数分析深度
-            
+
         Returns:
             总结提示文本
         """
         depth_description = "不包含子函数分析" if analysis_depth == 0 else f"包含 {analysis_depth} 层子函数分析"
-        
+
         return f"""# 函数分析报告: `{function_name}`
 
 ## 报告要求
@@ -328,4 +328,4 @@ class FunctionAnalyzerTool:
 - 使用具体的代码片段支持你的观点
 - 以清晰的Markdown格式呈现，简洁明了
 
-在分析中保持灵活性，避免固定思维模式。你的任务不是提供全面的函数概览，而是直接解决分析目标中提出的具体问题。""" 
+在分析中保持灵活性，避免固定思维模式。你的任务不是提供全面的函数概览，而是直接解决分析目标中提出的具体问题。"""

jarvis/jarvis_tools/git_commiter.py

@@ -37,14 +37,14 @@ class GitCommitTool:
     def _extract_commit_message(self, message):
         """Raw extraction preserving all characters"""
         r = re.search(
-            r"(?i)" + ot("COMMIT_MESSAGE") + r"\s*([\s\S]*?)\s*" + ct("COMMIT_MESSAGE"), 
+            r"(?i)" + ot("COMMIT_MESSAGE") + r"\s*([\s\S]*?)\s*" + ct("COMMIT_MESSAGE"),
             message
         )
         if r:
             # 直接返回原始内容，仅去除外围空白
             return shlex.quote(r.group(1).strip())
         return "<<FORMAT VIOLATION>> Invalid commit message structure"
-    
+
     def _get_last_commit_hash(self):
         process = subprocess.Popen(
             ["git", "log", "-1", "--pretty=%H"],
@@ -58,19 +58,19 @@ class GitCommitTool:
         """Execute automatic commit process with support for multi-line messages and special characters"""
         try:
             root_dir = args.get("root_dir", ".")
-            
+
             # Store current directory
             original_dir = os.getcwd()
-            
+
             try:
                 # Change to root_dir
                 os.chdir(root_dir)
-                
+
                 find_git_root()
                 if not has_uncommitted_changes():
                     PrettyOutput.print("没有未提交的更改", OutputType.SUCCESS)
                     return {"success": True, "stdout": "No changes to commit", "stderr": ""}
-                
+
                 with yaspin(text="正在初始化提交流程...", color="cyan") as spinner:
                     # 添加文件
                     spinner.text = "正在添加文件到提交..."
@@ -80,7 +80,7 @@ class GitCommitTool:
                         stderr=subprocess.DEVNULL
                     ).wait()
                     spinner.write("✅ 添加文件到提交")
-                    
+
                     # 获取差异
                     spinner.text = "正在获取代码差异..."
                     process = subprocess.Popen(
@@ -90,7 +90,7 @@ class GitCommitTool:
                     )
                     diff = process.communicate()[0].decode()
                     spinner.write("✅ 获取差异")
-                    
+
                     # 生成提交信息
                     spinner.text = "正在生成提交消息..."
                     prompt = f'''根据以下规则生成提交信息：
@@ -114,7 +114,7 @@ class GitCommitTool:
                     commit_message = platform.chat_until_success(prompt)
                     commit_message = self._extract_commit_message(commit_message)
                     spinner.write("✅ 生成提交消息")
-                    
+
                     # 执行提交
                     spinner.text = "正在准备提交..."
                     with tempfile.NamedTemporaryFile(mode='w', delete=True) as tmp_file:

jarvis/jarvis_tools/lsp_get_diagnostics.py

@@ -3,7 +3,7 @@ from typing import Dict, Any
 from jarvis.jarvis_lsp.registry import LSPRegistry
 class LSPGetDiagnosticsTool:
     """Tool for getting diagnostics (errors, warnings) from code using LSP."""
-    
+
     # 工具名称
     name = "lsp_get_diagnostics"
     # 工具描述
@@ -18,19 +18,19 @@ class LSPGetDiagnosticsTool:
             "default": "."
         }
     }
-    
+
     @staticmethod
     def check() -> bool:
         """检查是否有可用的LSP服务器"""
         registry = LSPRegistry.get_global_lsp_registry()
         return len(registry.get_supported_languages()) > 0
-    
+
     def execute(self, args: Dict[str, Any]) -> Dict[str, Any]:
         """执行工具的主要逻辑"""
         file_path = args.get("file_path", "")
         language = args.get("language", "")
         root_dir = args.get("root_dir", ".")
-        
+
         # 验证输入参数
         if not all([file_path, language]):
             return {
@@ -38,7 +38,7 @@ class LSPGetDiagnosticsTool:
                 "stderr": "Both file_path and language must be provided",
                 "stdout": ""
             }
-            
+
         # 检查文件是否存在
         if not os.path.exists(file_path):
             return {
@@ -46,18 +46,18 @@ class LSPGetDiagnosticsTool:
                 "stderr": f"File not found: {file_path}",
                 "stdout": ""
             }
-            
+
         # 存储当前目录
         original_dir = os.getcwd()
-        
+
         try:
             # 切换到root_dir
             os.chdir(root_dir)
-            
+
             # 获取LSP实例
             registry = LSPRegistry.get_global_lsp_registry()
             lsp = registry.create_lsp(language)
-        
+
             # 检查语言是否支持
             if not lsp:
                 return {
@@ -65,7 +65,7 @@ class LSPGetDiagnosticsTool:
                     "stderr": f"No LSP support for language: {language}",
                     "stdout": ""
                 }
-                
+
             try:
                 # 初始化LSP
                 if not lsp.initialize(os.path.abspath(os.getcwd())):
@@ -74,10 +74,10 @@ class LSPGetDiagnosticsTool:
                         "stderr": "LSP initialization failed",
                         "stdout": ""
                     }
-                    
+
                 # 获取诊断信息
                 diagnostics = lsp.get_diagnostics(file_path)
-                
+
                 # 如果没有诊断信息
                 if not diagnostics:
                     return {
@@ -85,31 +85,31 @@ class LSPGetDiagnosticsTool:
                         "stdout": "No issues found in the file",
                         "stderr": ""
                     }
-                    
+
                 # 格式化输出
                 output = ["Diagnostics:"]
                 # 严重程度映射
                 severity_map = {1: "Error", 2: "Warning", 3: "Info", 4: "Hint"}
-                
+
                 # 按严重程度和行号排序诊断信息
                 sorted_diagnostics = sorted(
                     diagnostics,
                     key=lambda x: (x["severity"], x["range"]["start"]["line"])
                 )
-                
+
                 # 处理每个诊断信息
                 for diag in sorted_diagnostics:
                     severity = severity_map.get(diag["severity"], "Unknown")
                     start = diag["range"]["start"]
                     line = LSPRegistry.get_line_at_position(file_path, start["line"]).strip()
-                    
+
                     output.extend([
                         f"\n{severity} at line {start['line'] + 1}, column {start['character'] + 1}:",
                         f"Message: {diag['message']}",
                         f"Code: {line}",
                         "-" * 60
                     ])
-                    
+
                     # 处理相关附加信息
                     if diag.get("relatedInformation"):
                         output.append("Related information:")
@@ -123,13 +123,13 @@ class LSPGetDiagnosticsTool:
                                 f"    at {info['location']['uri']}:{info['location']['range']['start']['line'] + 1}",
                                 f"    {info_line}"
                             ])
-                
+
                 return {
                     "success": True,
                     "stdout": "\n".join(output),
                     "stderr": ""
                 }
-                
+
             except Exception as e:
                 return {
                     "success": False,