jarvis-ai-assistant 0.1.125__py3-none-any.whl → 0.1.126__py3-none-any.whl

jarvis/jarvis_lsp/base.py

@@ -125,18 +125,6 @@ class BaseLSP(ABC):
         """
         return None
     
-    @abstractmethod
-    def validate_edit(self, file_path: str, edit: Dict[str, Any]) -> bool:
-        """Validate if proposed edit is syntactically correct.
-        
-        Args:
-            file_path: Path to the file
-            edit: Edit operation in LSP format
-            
-        Returns:
-            bool: True if edit is valid
-        """
-        return False
     
     def shutdown(self):
         """Shutdown LSP server cleanly."""

jarvis/jarvis_lsp/cpp.py

@@ -114,15 +114,6 @@ class CPPLSP(BaseLSP):
         })
         return result
     
-    def validate_edit(self, file_path: str, edit: Dict[str, Any]) -> bool:
-        # Send workspace/willRenameFiles request to check validity
-        result = self._send_request("workspace/willRenameFiles", {
-            "files": [{
-                "oldUri": f"file://{file_path}",
-                "newUri": f"file://{file_path}.tmp"
-            }]
-        })
-        return bool(result)
     
     def shutdown(self):
         if self.clangd_process:

jarvis/jarvis_lsp/go.py

@@ -120,15 +120,6 @@ class GoLSP(BaseLSP):
         })
         return result
     
-    def validate_edit(self, file_path: str, edit: Dict[str, Any]) -> bool:
-        # Send workspace/willRenameFiles request to check validity
-        result = self._send_request("workspace/willRenameFiles", {
-            "files": [{
-                "oldUri": f"file://{file_path}",
-                "newUri": f"file://{file_path}.tmp"
-            }]
-        })
-        return bool(result)
     
     def shutdown(self):
         if self.gopls_process:

jarvis/jarvis_lsp/python.py

@@ -100,34 +100,6 @@ class PythonLSP(BaseLSP):
             return None
         return None
     
-    def validate_edit(self, file_path: str, edit: Dict[str, Any]) -> bool:
-        try:
-            # Simple syntax check of the edited content
-            content = ""
-            with open(file_path, 'r') as f:
-                content = f.read()
-            
-            # Apply edit
-            start = edit["range"]["start"]
-            end = edit["range"]["end"]
-            new_text = edit["newText"]
-            
-            lines = content.splitlines(True)
-            before = "".join(lines[:start["line"]])
-            after = "".join(lines[end["line"] + 1:])
-            current_line = lines[start["line"]]
-            
-            edited_line = (current_line[:start["character"]] + 
-                         new_text + 
-                         current_line[end["character"]:])
-            
-            new_content = before + edited_line + after
-            
-            # Check if new content is valid Python
-            jedi.Script(code=new_content)
-            return True
-        except Exception:
-            return False
     
     def shutdown(self):
         self.script_cache.clear()

jarvis/jarvis_lsp/registry.py

@@ -14,7 +14,6 @@ REQUIRED_METHODS = [
     ('get_document_symbols', ['file_path']),
     ('get_diagnostics', ['file_path']),
     ('prepare_rename', ['file_path', 'position']),
-    ('validate_edit', ['file_path', 'edit']),
     ('shutdown', [])
 ]
 

jarvis/jarvis_lsp/rust.py

@@ -122,15 +122,6 @@ class RustLSP(BaseLSP):
         })
         return result
     
-    def validate_edit(self, file_path: str, edit: Dict[str, Any]) -> bool:
-        # Send workspace/willRenameFiles request to check validity
-        result = self._send_request("workspace/willRenameFiles", {
-            "files": [{
-                "oldUri": f"file://{file_path}",
-                "newUri": f"file://{file_path}.tmp"
-            }]
-        })
-        return bool(result)
     
     def shutdown(self):
         if self.analyzer_process:

jarvis/jarvis_multi_agent/__init__.py

@@ -32,65 +32,65 @@ class MultiAgent(OutputHandler):
 
     def prompt(self) -> str:
         return f"""
-# 🤖 Message Handling System
-You are part of a multi-agent system that communicates through structured messages.
-
-# 🎯 Core Rules
-## Critical Action Rules
-- Execute ONLY ONE action per turn:
-  - Either use ONE tool (file_operation, ask_user, etc.)
-  - OR send ONE message to another agent
-  - NEVER combine both in same turn
-
-## Message Flow Control
-- Wait for response after sending message
-- Process response before next action
-- Never send multiple messages at once
-- Never combine messages with tool calls
-
-# 📝 Message Format
+# 🤖 多智能体消息处理系统
+您是多智能体系统的一部分，通过结构化消息进行通信。
+
+# 🎯 核心规则
+## 关键操作规则
+- 每轮只能执行一个操作：
+  - 要么使用一个工具（文件操作、询问用户等）
+  - 要么发送一条消息给其他智能体
+  - 切勿在同一轮中同时进行这两种操作
+
+## 消息流控制
+- 发送消息后等待响应
+- 处理响应后再进行下一步操作
+- 切勿同时发送多条消息
+- 切勿将消息与工具调用混合使用
+
+# 📝 消息格式
 ```
 <SEND_MESSAGE>
-to: agent_name    # Target agent name
+to: 智能体名称    # 目标智能体名称
 content: |
-    message_content    # Message content
-    use multiple lines    # If needed
-    with proper indentation
+    消息内容    # 消息内容
+    可使用多行    # 如果需要
+    保持正确的缩进
 </SEND_MESSAGE>
 ```
 
-# 🔄 Action Sequence
-1. Choose Most Important Action
-   - Evaluate priority
-   - Select ONE action
-   - Execute action
+# 🔄 操作顺序
+1. 选择最重要的操作
+   - 评估优先级
+   - 选择一个操作
+   - 执行该操作
 
-2. Wait for Response
-   - Process result/response
-   - Plan next action
-   - Wait for next turn
+2. 等待响应
+   - 处理结果/响应
+   - 计划下一步操作
+   - 等待下一轮
 
-3. Handle Responses
-   - Process incoming messages
-   - Reply to sender when needed
-   - Continue task based on response
+3. 处理响应
+   - 处理收到的消息
+   - 需要时回复发送者
+   - 根据响应继续任务
 
-# 👥 Available Agents
+# 👥 可用智能体
 {chr(10).join([f"- {c.name}: {c.description}" for c in self.agents_config])}
 
-# ❗ Important Rules
-1. ONE action per turn only
-2. Wait for responses
-3. Process before next action
-4. Reply to messages
-5. Forward task if needed
-
-# 💡 Tips
-- First action will be executed
-- Additional actions will be ignored
-- Always process responses first
-- Send message to continue task if needed
-- Handle and reply to received messages
+# ❗ 重要规则
+1. 每轮只能执行一个操作
+2. 等待响应
+3. 处理后再进行下一步
+4. 回复消息
+5. 需要时转发任务
+
+# 💡 提示
+- 第一个操作将被执行
+- 额外的操作将被忽略
+- 总是先处理响应
+- 需要时发送消息以继续任务
+- 处理并回复收到的消息
 """
 
     def can_handle(self, response: str) -> bool:
@@ -161,10 +161,10 @@ from: {last_agent}
 content: {msg['content']}
 """
                 if msg['to'] not in self.agents:
-                    PrettyOutput.print(f"没有找到{msg['to']}，重试...", OutputType.WARNING)
-                    msg = self.agents[last_agent].run(f"The agent {msg['to']} is not found, agent list: {self.agents.keys()}")
+                    PrettyOutput.print(f"未找到智能体 {msg['to']}，正在重试...", OutputType.WARNING)
+                    msg = self.agents[last_agent].run(f"未找到智能体 {msg['to']}，可用智能体列表: {self.agents.keys()}")
                     continue
-                PrettyOutput.print(f"{last_agent} 发送消息给 {msg['to']}...", OutputType.INFO)
+                PrettyOutput.print(f"{last_agent} 正在向 {msg['to']} 发送消息...", OutputType.INFO)
                 last_agent = self.agents[msg['to']].name
                 msg = self.agents[msg['to']].run(prompt)
-        return ""
+        return ""

jarvis/jarvis_tools/ask_codebase.py

@@ -5,20 +5,20 @@ from jarvis.jarvis_utils.git_utils import find_git_root
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 
 class AskCodebaseTool:
-    """Tool for intelligent codebase querying and analysis using CodeBase"""
-    
+    """用于智能代码库查询和分析的工具"""
+
     name = "ask_codebase"
-    description = "Ask questions about the codebase and get detailed analysis"
+    description = "查询代码库问题并获取详细分析"
     parameters = {
         "type": "object",
         "properties": {
             "question": {
                 "type": "string",
-                "description": "Question about the codebase"
+                "description": "关于代码库的问题"
             },
             "top_k": {
                 "type": "integer",
-                "description": "Number of most relevant files to analyze (optional)",
+                "description": "要分析的最相关文件数量（可选）",
                 "default": 20
             }
         },
@@ -98,4 +98,4 @@ def main():
         
 
 if __name__ == "__main__":
-    main()
+    main()

jarvis/jarvis_tools/ask_user.py

@@ -5,13 +5,13 @@ from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 
 class AskUserTool:
     name="ask_user"
-    description="""Ask the user when information needed to complete the task is missing or when critical decision information is lacking. Users can input multiple lines of text, ending with an empty line. Use cases: 1. Need user to provide more information to complete the task; 2. Need user to make critical decisions; 3. Need user to confirm important operations; 4. Need user to provide additional information"""
+    description="""当完成任务所需的信息缺失或关键决策信息不足时，向用户提问。用户可以输入多行文本，以空行结束。使用场景：1. 需要用户提供更多信息以完成任务；2. 需要用户做出关键决策；3. 需要用户确认重要操作；4. 需要用户提供额外信息"""
     parameters={
         "type": "object",
         "properties": {
             "question": {
                 "type": "string",
-                "description": "The question to ask the user"
+                "description": "要向用户提出的问题"
             }
         },
         "required": ["question"]

jarvis/jarvis_tools/base.py

@@ -7,7 +7,7 @@ class Tool:
         """
         初始化工具对象
         
-        Args:
+        参数:
             name (str): 工具名称
             description (str): 工具描述
             parameters (Dict): 工具参数定义
@@ -28,10 +28,10 @@ class Tool:
         """
         执行工具函数
         
-        Args:
+        参数:
             arguments (Dict): 工具执行所需的参数
             
-        Returns:
+        返回:
             Dict[str, Any]: 工具执行结果
         """
-        return self.func(arguments)
+        return self.func(arguments)

jarvis/jarvis_tools/chdir.py

@@ -3,13 +3,13 @@ import os
 
 class ChdirTool:
     name = "chdir"
-    description = "Change current working directory"
+    description = "更改当前工作目录"
     parameters = {
         "type": "object",
         "properties": {
             "path": {
                 "type": "string",
-                "description": "Directory path to switch to, supports both relative and absolute paths"
+                "description": "要切换到的目录路径，支持相对路径和绝对路径"
             }
         },
         "required": ["path"]
@@ -45,7 +45,7 @@ class ChdirTool:
                 return {
                     "success": False,
                     "stdout": "",
-                    "stderr": f"Directory does not exist: {path}"
+                    "stderr": f"目录不存在: {path}"
                 }
                 
             # Ensure the path points to a directory, not a file
@@ -53,7 +53,7 @@ class ChdirTool:
                 return {
                     "success": False,
                     "stdout": "",
-                    "stderr": f"The path is not a directory: {path}"
+                    "stderr": f"路径不是目录: {path}"
                 }
                 
             # Capture current directory and attempt to change to new path
@@ -62,7 +62,7 @@ class ChdirTool:
             
             return {
                 "success": True,
-                "stdout": f"Changed working directory:\nFrom: {old_path}\nTo: {path}",
+                "stdout": f"成功切换工作目录:\n原目录: {old_path}\n新目录: {path}",
                 "stderr": ""
             }
             
@@ -71,12 +71,12 @@ class ChdirTool:
             return {
                 "success": False,
                 "stdout": "",
-                "stderr": f"No permission to access directory: {path}"
+                "stderr": f"无权限访问目录: {path}"
             }
         # Catch-all for any other unexpected errors during directory change
         except Exception as e:
             return {
                 "success": False,
                 "stdout": "",
-                "stderr": f"Failed to switch directory: {str(e)}"
-            }
+                "stderr": f"切换目录失败: {str(e)}"
+            }

jarvis/jarvis_tools/code_review.py

@@ -10,27 +10,27 @@ from jarvis.jarvis_utils.utils import init_env
 
 class CodeReviewTool:
     name = "code_review"
-    description = "Autonomous code review agent for code changes analysis"
+    description = "自动代码审查工具，用于分析代码变更"
     parameters = {
         "type": "object",
         "properties": {
             "review_type": {
                 "type": "string",
-                "description": "Type of review: 'commit' for specific commit, 'current' for current changes, 'range' for commit range",
+                "description": "审查类型：'commit' 审查特定提交，'current' 审查当前变更，'range' 审查提交范围",
                 "enum": ["commit", "current", "range"],
                 "default": "current"
             },
             "commit_sha": {
                 "type": "string",
-                "description": "Target commit SHA to analyze (required for review_type='commit')"
+                "description": "要分析的提交SHA（review_type='commit'时必填）"
             },
             "start_commit": {
                 "type": "string",
-                "description": "Start commit SHA (required for review_type='range')"
+                "description": "起始提交SHA（review_type='range'时必填）"
             },
             "end_commit": {
                 "type": "string",
-                "description": "End commit SHA (required for review_type='range')"
+                "description": "结束提交SHA（review_type='range'时必填）"
             }
         },
         "required": []
@@ -243,4 +243,4 @@ def main():
         PrettyOutput.print(result["stderr"], OutputType.WARNING)
 
 if __name__ == "__main__":
-    main()
+    main()

jarvis/jarvis_tools/create_code_agent.py

@@ -6,12 +6,12 @@ from jarvis.jarvis_utils.git_utils import get_latest_commit_hash, has_uncommitte
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 
 class CreateCodeAgentTool:
-    """Tool for managing the code development workflow."""
+    """用于管理代码开发工作流的工具"""
     
     name = "create_code_agent"
-    description = "Technical code implementation and development process management"
+    description = "技术代码实现和开发过程管理工具"
     parameters = {
-        "requirement": "Technical specifications for code implementation"
+        "requirement": "代码实现的技术规范"
     }
     
     
@@ -109,4 +109,4 @@ def main():
         PrettyOutput.print(result["stderr"], OutputType.WARNING)
 
 if __name__ == "__main__":
-    main()
+    main()

jarvis/jarvis_tools/create_sub_agent.py

@@ -8,32 +8,32 @@ from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 
 class SubAgentTool:
     name = "create_sub_agent"
-    description = "Create a sub-agent to handle specific tasks, the sub-agent will generate a task summary report"
+    description = "创建子代理以处理特定任务，子代理将生成任务总结报告"
     parameters = {
         "type": "object",
         "properties": {
             "agent_name": {
                 "type": "string",
-                "description": "Sub-agent name"
+                "description": "子代理名称"
             },
             "task": {
                 "type": "string",
-                "description": "Specific task to complete"
+                "description": "要完成的特定任务"
             },
             "context": {
                 "type": "string",
-                "description": "Context information related to the task",
+                "description": "与任务相关的上下文信息",
                 "default": ""
             },
             "goal": {
                 "type": "string",
-                "description": "Completion goal of the task",
+                "description": "任务的完成目标",
                 "default": ""
             },
             "files": {
                 "type": "array",
                 "items": {"type": "string"},
-                "description": "Related file path list, used for file question answering and processing",
+                "description": "相关文件路径列表，用于文件问答和处理",
                 "default": []
             }
         },
@@ -83,4 +83,4 @@ class SubAgentTool:
                 "success": False,
                 "stdout": "",
                 "stderr": f"Sub-agent execution failed: {str(e)}"
-            } 
+            }

jarvis/jarvis_tools/execute_shell.py

@@ -1,57 +1,89 @@
+# Shell command execution module
+# 
+# Provides functionality to execute shell commands safely with:
+# - Command escaping
+# - Output capturing
+# - Temporary file management
+# - Error handling
 from typing import Dict, Any
 import os
 import tempfile
 from pathlib import Path
-
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput
-
-
-
-
 class ShellTool:
+    """Shell command execution tool
+    
+    Attributes:
+        name: Tool identifier used in API
+        description: Tool description for API documentation
+        parameters: JSON schema for command parameters
+    """
     name = "execute_shell"
-    description = """Execute shell command and return result"""
-
+    description = "执行Shell命令并返回结果"
     parameters = {
         "type": "object",
         "properties": {
             "command": {
                 "type": "string",
-                "description": "Shell command to execute"
+                "description": "要执行的Shell命令"
             }
         },
         "required": ["command"]
     }
-
-
     def _escape_command(self, cmd: str) -> str:
-        """Escape special characters in command"""
+        """Escape special characters in command to prevent shell injection
+        
+        Args:
+            cmd: Raw command string
+            
+        Returns:
+            Escaped command string with single quotes properly handled
+        """
         return cmd.replace("'", "'\"'\"'")
-
     def execute(self, args: Dict) -> Dict[str, Any]:
-        """Execute shell command"""
+        """Execute shell command and capture output
+        
+        Steps:
+        1. Validate and clean input command
+        2. Create temporary file for output capture
+        3. Execute command with output redirection
+        4. Read and process output file
+        5. Clean up temporary resources
+        6. Return structured results
+        
+        Args:
+            args: Dictionary containing 'command' parameter
+            
+        Returns:
+            Dictionary with:
+            - success: Boolean indicating command execution status
+            - stdout: Command output
+            - stderr: Error message if execution failed
+        """
         try:
+            # Get and clean command input
             command = args["command"].strip()
             
-            # Generate temporary file name
+            # Generate temporary file name using process ID for uniqueness
             output_file = os.path.join(tempfile.gettempdir(), f"jarvis_shell_{os.getpid()}.log")
             
-            # Escape special characters in command
+            # Escape special characters in command to prevent injection
             escaped_command = self._escape_command(command)
             
-            # Modify command to use script
+            # Use script command to capture both stdout and stderr
             tee_command = f"script -q -c '{escaped_command}' {output_file}"
             
+            # Log command execution
             PrettyOutput.print(f"执行命令: {command}", OutputType.INFO)
             
-            # Execute command
+            # Execute command and capture return code
             return_code = os.system(tee_command)
             
-            # Read output file
+            # Read and process output file
             try:
                 with open(output_file, 'r', encoding='utf-8', errors='replace') as f:
                     output = f.read()
-                    # Remove header and footer added by script
+                    # Remove header and footer added by script command
                     if output:
                         lines = output.splitlines()
                         if len(lines) > 2:
@@ -62,6 +94,7 @@ class ShellTool:
                 # Clean up temporary file
                 Path(output_file).unlink(missing_ok=True)
             
+            # Return successful result
             return {
                 "success": True,
                 "stdout": output,
@@ -69,7 +102,7 @@ class ShellTool:
             }
                 
         except Exception as e:
-            # Ensure temporary file is cleaned up
+            # Ensure temporary file is cleaned up even if error occurs
             if 'output_file' in locals():
                 Path(output_file).unlink(missing_ok=True)
             PrettyOutput.print(str(e), OutputType.ERROR)
@@ -77,4 +110,4 @@ class ShellTool:
                 "success": False,
                 "stdout": "",
                 "stderr": str(e)
-            } 
+            }

jarvis/jarvis_tools/execute_shell_script.py

@@ -8,13 +8,13 @@ from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 
 class ShellScriptTool:
     name = "execute_shell_script"
-    description = """Execute shell script file and return result"""
+    description = "执行Shell脚本文件并返回结果"
     parameters = {
         "type": "object",
         "properties": {
             "script_content": {
                 "type": "string",
-                "description": "Content of the shell script to execute"
+                "description": "要执行的Shell脚本内容"
             }
         },
         "required": ["script_content"]
@@ -55,4 +55,4 @@ class ShellScriptTool:
                 "success": False,
                 "stdout": "",
                 "stderr": str(e)
-            }
+            }