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

jarvis/jarvis_tools/git_commiter.py

@@ -5,20 +5,23 @@ from typing import Dict, Any
 import tempfile
 import yaml
 from jarvis.jarvis_platform.registry import PlatformRegistry
-from jarvis.jarvis_utils import OutputType, PrettyOutput, has_uncommitted_changes, init_env
 import sys
 import argparse
 
+from jarvis.jarvis_utils.git_utils import has_uncommitted_changes
+from jarvis.jarvis_utils.output import OutputType, PrettyOutput
+from jarvis.jarvis_utils.utils import init_env
+
 
 class GitCommitTool:
     name = "git_commit_agent"
-    description = "Automatically generate and execute git commits based on code changes"
+    description = "根据代码变更自动生成并执行Git提交"
     parameters = {
         "type": "object",
         "properties": {
             "lang": {
                 "type": "string",
-                "description": "Language for commit message",
+                "description": "提交信息的语言",
                 "default": "Chinese"
             }
         },
@@ -67,21 +70,21 @@ class GitCommitTool:
             diff = process.communicate()[0].decode()
             PrettyOutput.print(diff, OutputType.CODE, lang="diff")
             
-            prompt = f'''Generate commit message with the paranoia of someone who's lost production data:
-            You should write commit message in {args.get('lang', 'Chinese')}
-# Required Structure
-YOU MUST USE EXACTLY THIS FORMAT:
+            prompt = f'''根据以下规则生成提交信息：
+            提交信息应使用{args.get('lang', '中文')}书写
+# 必需结构
+必须使用以下格式：
 <COMMIT_MESSAGE>
-<type>(<scope>): <subject>
-Body description in imperative mood
+<类型>(<范围>): <主题>
+使用祈使语气描述变更内容
 </COMMIT_MESSAGE>
-# Format Rules
-1. Types: fix, feat, docs, style, refactor, test, chore
-2. Scope indicates module (e.g. auth, database)
-3. Subject line <= 72 chars, no period
-4. Body explains WHAT and WHY for every change, using present tense
-5. Do not omit any changes
-# Analysis Material
+# 格式规则
+1. 类型: fix, feat, docs, style, refactor, test, chore
+2. 范围表示模块 (例如: auth, database)
+3. 主题行 <= 72个字符，不以句号结尾
+4. 正文使用现在时态解释每个变更的内容和原因
+5. 不要遗漏任何变更
+# 分析材料
 {diff}
 '''
             
@@ -131,4 +134,4 @@ def main():
     tool.execute({"lang": args.lang if hasattr(args, 'lang') else 'Chinese'})
 
 if __name__ == "__main__":
-    sys.exit(main())
+    sys.exit(main())

jarvis/jarvis_tools/lsp_find_definition.py

@@ -3,15 +3,15 @@ from typing import Dict, Any
 from jarvis.jarvis_lsp.registry import LSPRegistry
 
 class LSPFindDefinitionTool:
-    """Tool for finding symbol definitions in code using LSP."""
+    """使用LSP在代码中查找符号定义的工具"""
     
     name = "lsp_find_definition"
-    description = "Find the definition of a symbol in code"
+    description = "在代码中查找符号的定义"
     parameters = {
-        "file_path": "Path to the file containing the symbol",
-        "line": "Line number (0-based) of the symbol",
-        "character": "Character position in the line",
-        "language": f"Programming language of the file ({', '.join(LSPRegistry.get_global_lsp_registry().get_supported_languages())})"
+        "file_path": "包含符号的文件路径",
+        "line": "符号所在的行号（从0开始）",
+        "character": "符号在行中的字符位置",
+        "language": f"文件的编程语言（{', '.join(LSPRegistry.get_global_lsp_registry().get_supported_languages())}）"
     }
     
     @staticmethod
@@ -65,7 +65,7 @@ class LSPFindDefinitionTool:
             
         try:
             # Initialize LSP
-            if not lsp.initialize(os.path.dirname(os.path.abspath(file_path))):
+            if not lsp.initialize(os.path.abspath(os.getcwd())):
                 return {
                     "success": False,
                     "stderr": "LSP initialization failed",
@@ -131,4 +131,4 @@ class LSPFindDefinitionTool:
             }
         finally:
             if lsp:
-                lsp.shutdown()
+                lsp.shutdown()

jarvis/jarvis_tools/lsp_find_references.py

@@ -65,7 +65,7 @@ class LSPFindReferencesTool:
             
         try:
             # Initialize LSP
-            if not lsp.initialize(os.path.dirname(os.path.abspath(file_path))):
+            if not lsp.initialize(os.path.abspath(os.getcwd())):
                 return {
                     "success": False,
                     "stderr": "LSP initialization failed",

jarvis/jarvis_tools/lsp_get_diagnostics.py

@@ -1,12 +1,14 @@
 import os
 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"
+    # 工具描述
     description = "Get diagnostic information (errors, warnings) from code files"
+    # 工具参数定义
     parameters = {
         "file_path": "Path to the file to analyze",
         "language": f"Programming language of the file ({', '.join(LSPRegistry.get_global_lsp_registry().get_supported_languages())})"
@@ -14,16 +16,16 @@ class LSPGetDiagnosticsTool:
     
     @staticmethod
     def check() -> bool:
-        """Check if any LSP server is available."""
+        """检查是否有可用的LSP服务器"""
         registry = LSPRegistry.get_global_lsp_registry()
         return len(registry.get_supported_languages()) > 0
     
     def execute(self, args: Dict[str, Any]) -> Dict[str, Any]:
-        """Execute the tool."""
+        """执行工具的主要逻辑"""
         file_path = args.get("file_path", "")
         language = args.get("language", "")
         
-        # Validate inputs
+        # 验证输入参数
         if not all([file_path, language]):
             return {
                 "success": False,
@@ -31,6 +33,7 @@ class LSPGetDiagnosticsTool:
                 "stdout": ""
             }
             
+        # 检查文件是否存在
         if not os.path.exists(file_path):
             return {
                 "success": False,
@@ -38,10 +41,11 @@ class LSPGetDiagnosticsTool:
                 "stdout": ""
             }
             
-        # Get LSP instance
+        # 获取LSP实例
         registry = LSPRegistry.get_global_lsp_registry()
         lsp = registry.create_lsp(language)
         
+        # 检查语言是否支持
         if not lsp:
             return {
                 "success": False,
@@ -50,17 +54,18 @@ class LSPGetDiagnosticsTool:
             }
             
         try:
-            # Initialize LSP
-            if not lsp.initialize(os.path.dirname(os.path.abspath(file_path))):
+            # 初始化LSP
+            if not lsp.initialize(os.path.abspath(os.getcwd())):
                 return {
                     "success": False,
                     "stderr": "LSP initialization failed",
                     "stdout": ""
                 }
                 
-            # Get diagnostics
+            # 获取诊断信息
             diagnostics = lsp.get_diagnostics(file_path)
             
+            # 如果没有诊断信息
             if not diagnostics:
                 return {
                     "success": True,
@@ -68,16 +73,18 @@ class LSPGetDiagnosticsTool:
                     "stderr": ""
                 }
                 
-            # Format output
+            # 格式化输出
             output = ["Diagnostics:"]
+            # 严重程度映射
             severity_map = {1: "Error", 2: "Warning", 3: "Info", 4: "Hint"}
             
-            # Sort diagnostics by severity and line number
+            # 按严重程度和行号排序诊断信息
             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"]
@@ -90,7 +97,7 @@ class LSPGetDiagnosticsTool:
                     "-" * 60
                 ])
                 
-                # Add related information if available
+                # 处理相关附加信息
                 if diag.get("relatedInformation"):
                     output.append("Related information:")
                     for info in diag["relatedInformation"]:
@@ -117,5 +124,6 @@ class LSPGetDiagnosticsTool:
                 "stdout": ""
             }
         finally:
+            # 确保关闭LSP连接
             if lsp:
                 lsp.shutdown()

jarvis/jarvis_tools/lsp_get_document_symbols.py

@@ -50,7 +50,7 @@ class LSPGetDocumentSymbolsTool:
             
         try:
             # Initialize LSP
-            if not lsp.initialize(os.path.dirname(os.path.abspath(file_path))):
+            if not lsp.initialize(os.path.abspath(os.getcwd())):
                 return {
                     "success": False,
                     "stderr": "LSP initialization failed",

jarvis/jarvis_tools/lsp_prepare_rename.py

@@ -3,15 +3,15 @@ from typing import Dict, Any
 from jarvis.jarvis_lsp.registry import LSPRegistry
 
 class LSPPrepareRenameTool:
-    """Tool for checking if a symbol can be renamed using LSP."""
+    """使用LSP检查符号是否可以安全重命名并显示所有受影响位置的工具"""
     
     name = "lsp_prepare_rename"
-    description = "Check if a symbol can be safely renamed and show all locations that would be affected"
+    description = "检查符号是否可以安全重命名，并显示所有受影响的位置"
     parameters = {
-        "file_path": "Path to the file containing the symbol",
-        "line": "Line number (0-based) of the symbol",
-        "character": "Character position in the line",
-        "language": f"Programming language of the file ({', '.join(LSPRegistry.get_global_lsp_registry().get_supported_languages())})"
+        "file_path": "包含符号的文件路径",
+        "line": "符号所在的行号（从0开始）",
+        "character": "符号在行中的字符位置",
+        "language": f"文件的编程语言（{', '.join(LSPRegistry.get_global_lsp_registry().get_supported_languages())}）"
     }
     
     @staticmethod
@@ -65,7 +65,7 @@ class LSPPrepareRenameTool:
             
         try:
             # Initialize LSP
-            if not lsp.initialize(os.path.dirname(os.path.abspath(file_path))):
+            if not lsp.initialize(os.path.abspath(os.getcwd())):
                 return {
                     "success": False,
                     "stderr": "LSP initialization failed",
@@ -127,4 +127,4 @@ class LSPPrepareRenameTool:
             }
         finally:
             if lsp:
-                lsp.shutdown()
+                lsp.shutdown()

jarvis/jarvis_tools/methodology.py

@@ -1,29 +1,32 @@
 import os
 import yaml
 from typing import Dict, Optional, Any
-from jarvis.jarvis_utils import OutputType, PrettyOutput, is_use_methodology
+
+from jarvis.jarvis_utils.config import is_use_methodology
+from jarvis.jarvis_utils.output import OutputType, PrettyOutput
+
 
 
 class MethodologyTool:
-    """Experience management tool"""
+    """经验管理工具"""
     
     name = "methodology"
-    description = "Manage problem-solving methodologies, supporting add, update, and delete operations"
+    description = "管理问题解决方法论，支持添加、更新和删除操作"
     parameters = {
         "type": "object",
         "properties": {
             "operation": {
                 "type": "string",
-                "description": "Operation type (delete/update/add)",
+                "description": "操作类型（delete/update/add）",
                 "enum": ["delete", "update", "add"]
             },
             "problem_type": {
                 "type": "string",
-                "description": "Problem type, e.g., code_review, bug_fix, etc."
+                "description": "问题类型，例如：code_review, bug_fix 等"
             },
             "content": {
                 "type": "string",
-                "description": "Methodology content (required for update/add)",
+                "description": "方法论内容（更新和添加时必填）",
                 "optional": True
             }
         },
@@ -57,7 +60,7 @@ class MethodologyTool:
         except Exception as e:
             PrettyOutput.print(f"加载方法论失败: {str(e)}", OutputType.ERROR)
             return {}
-            
+                
     def _save_methodologies(self, methodologies: Dict):
         """Save all methodologies"""
         try:

jarvis/jarvis_tools/rag.py

@@ -1,25 +1,26 @@
 from typing import Dict, Any
 import os
-from jarvis.jarvis_utils import OutputType, PrettyOutput, dont_use_local_model
 from jarvis.jarvis_rag.main import RAGTool as RAGCore
+from jarvis.jarvis_utils.config import dont_use_local_model
+from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 
 class RAGTool:
     name = "rag"
-    description = "Ask questions based on a document directory, supporting multiple document formats (txt, pdf, docx, etc.)"
+    description = "基于文档目录进行问答，支持多种文档格式（txt、pdf、docx等）"
     parameters = {
         "type": "object",
         "properties": {
             "dir": {
                 "type": "string",
-                "description": "Document directory path, supports both relative and absolute paths"
+                "description": "文档目录路径，支持相对路径和绝对路径"
             },
             "question": {
                 "type": "string",
-                "description": "The question to ask"
+                "description": "要询问的问题"
             },
             "rebuild_index": {
                 "type": "boolean",
-                "description": "Whether to rebuild the index",
+                "description": "是否重建索引",
                 "default": False
             }
         },
@@ -48,25 +49,28 @@ class RAGTool:
         return self.rag_instances[dir_path]
 
     def execute(self, args: Dict[str, Any]) -> Dict[str, Any]:
-        """Execute document question and answer
+        """执行文档问答
         
         Args:
-            args: A dictionary containing parameters
-                - dir: The document directory path
-                - question: The question to ask
-                - rebuild_index: Whether to rebuild the index
+            args: 包含参数的字典
+                - dir: 文档目录路径
+                - question: 要询问的问题
+                - rebuild_index: 是否重建索引
                 
         Returns:
-            Dict[str, Any]: The execution result
+            Dict[str, Any]: 执行结果，包含以下字段：
+                - success: 布尔值，表示操作是否成功
+                - stdout: 如果成功，包含问题的答案
+                - stderr: 如果失败，包含错误信息
         """
         try:
-            # Get parameters
-            dir_path = os.path.expanduser(args["dir"])  # Expand ~ paths
-            dir_path = os.path.abspath(dir_path)  # Convert to absolute path
+            # 获取参数
+            dir_path = os.path.expanduser(args["dir"])
+            dir_path = os.path.abspath(dir_path)
             question = args["question"]
             rebuild_index = args.get("rebuild_index", False)
             
-            # Check if the directory exists
+            # 检查目录是否存在
             if not os.path.exists(dir_path):
                 return {
                     "success": False,
@@ -74,7 +78,7 @@ class RAGTool:
                     "stderr": f"Directory does not exist: {dir_path}"
                 }
                 
-            # Check if it is a directory
+            # 检查路径是否为目录
             if not os.path.isdir(dir_path):
                 return {
                     "success": False,
@@ -82,18 +86,19 @@ class RAGTool:
                     "stderr": f"The path is not a directory: {dir_path}"
                 }
                 
-            # Get RAG instance
+            # 获取RAG实例
             rag = self._get_rag_instance(dir_path)
             
-            # If you need to rebuild the index or the index does not exist
+            # 如果需要重建索引或索引不存在
             if rebuild_index or not rag.is_index_built():
                 PrettyOutput.print("正在构建文档索引...", OutputType.INFO)
                 rag.build_index(dir_path)
             
-            # Execute question and answer
+            # 执行问答
             PrettyOutput.print(f"问题: {question}", OutputType.INFO)
             response = rag.ask(question)
             
+            # 处理未找到相关文档的情况
             if response is None:
                 return {
                     "success": False,
@@ -101,6 +106,7 @@ class RAGTool:
                     "stderr": "Failed to get answer, possibly no relevant documents found"
                 }
                 
+            # 返回成功响应
             return {
                 "success": True,
                 "stdout": response,
@@ -108,6 +114,7 @@ class RAGTool:
             }
             
         except Exception as e:
+            # 处理任何意外错误
             PrettyOutput.print(f"文档问答失败：{str(e)}", OutputType.ERROR)
             return {
                 "success": False,
@@ -138,4 +145,4 @@ def main():
         PrettyOutput.print(result["stderr"], OutputType.WARNING)
 
 if __name__ == "__main__":
-    main() 
+    main()

jarvis/jarvis_tools/read_webpage.py

@@ -1,17 +1,18 @@
 from typing import Dict, Any
 import requests
 from bs4 import BeautifulSoup
-from jarvis.jarvis_utils import PrettyOutput, OutputType
+
+from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 
 class WebpageTool:
     name = "read_webpage"
-    description = "Read webpage content, extract title, text and hyperlinks"
+    description = "读取网页内容，提取标题、文本和超链接"
     parameters = {
         "type": "object",
         "properties": {
             "url": {
                 "type": "string",
-                "description": "The URL of the webpage to read"
+                "description": "要读取的网页URL"
             }
         },
         "required": ["url"]