jarvis-ai-assistant 0.1.164__py3-none-any.whl → 0.1.166__py3-none-any.whl

jarvis/jarvis_tools/generate_new_tool.py

@@ -0,0 +1,288 @@
+# -*- coding: utf-8 -*-
+import re
+from pathlib import Path
+from typing import Dict, Any, Tuple
+
+from jarvis.jarvis_utils.config import get_data_dir
+from jarvis.jarvis_utils.output import OutputType, PrettyOutput
+
+class generate_new_tool:
+    name = "generate_new_tool"
+    description = """
+    生成并注册新的Jarvis工具。该工具会在用户数据目录下创建新的工具文件，
+    并自动注册到当前的工具注册表中。适用场景：1. 需要创建新的自定义工具；
+    2. 扩展Jarvis功能；3. 自动化重复性操作；4. 封装特定领域的功能。
+    
+    使用示例：
+    
+    ```
+    # 创建一个将文本转换为大写/小写的工具
+    name: generate_new_tool
+    arguments:
+        tool_name: text_transformer
+        tool_code: |
+            # -*- coding: utf-8 -*-
+            from typing import Dict, Any
+            
+            class text_transformer:
+                name = "text_transformer"
+                description = \"\"\"
+                文本转换工具，可以将输入的文本转换为大写、小写或首字母大写格式。
+                适用场景：1. 格式化文本; 2. 处理标题; 3. 标准化输出
+                \"\"\"
+                
+                parameters = {
+                    "type": "object",
+                    "properties": {
+                        "text": {
+                            "type": "string",
+                            "description": "需要转换格式的文本"
+                        },
+                        "transform_type": {
+                            "type": "string",
+                            "description": "转换类型，可选值为 upper（大写）、lower（小写）或 title（首字母大写）",
+                            "enum": ["upper", "lower", "title"]
+                        }
+                    },
+                    "required": ["text", "transform_type"]
+                }
+                
+                @staticmethod
+                def check() -> bool:
+                    return True
+                
+                def execute(self, args: Dict[str, Any]) -> Dict[str, Any]:
+                    try:
+                        text = args["text"]
+                        transform_type = args["transform_type"]
+                        
+                        if transform_type == "upper":
+                            result = text.upper()
+                        elif transform_type == "lower":
+                            result = text.lower()
+                        elif transform_type == "title":
+                            result = text.title()
+                        else:
+                            return {
+                                "success": False,
+                                "stdout": "",
+                                "stderr": f"不支持的转换类型: {transform_type}"
+                            }
+                        
+                        return {
+                            "success": True,
+                            "stdout": result,
+                            "stderr": ""
+                        }
+                        
+                    except Exception as e:
+                        return {
+                            "success": False,
+                            "stdout": "",
+                            "stderr": f"转换失败: {str(e)}"
+                        }
+    ```
+    
+    创建完成后可以立即使用：
+    
+    ```
+    name: text_transformer
+    arguments:
+        text: hello world
+        transform_type: upper
+    ```
+    """
+    
+    parameters = {
+        "type": "object",
+        "properties": {
+            "tool_name": {
+                "type": "string",
+                "description": "新工具的名称，将用作文件名和工具类名"
+            },
+            "tool_code": {
+                "type": "string",
+                "description": "工具的完整Python代码，包含类定义、名称、描述、参数和execute方法"
+            }
+        },
+        "required": ["tool_name", "tool_code"]
+    }
+
+    @staticmethod
+    def check() -> bool:
+        """检查工具是否可用"""
+        # 检查数据目录是否存在
+        data_dir = get_data_dir()
+        tools_dir = Path(data_dir) / "tools"
+        
+        # 如果tools目录不存在，尝试创建
+        if not tools_dir.exists():
+            try:
+                tools_dir.mkdir(parents=True, exist_ok=True)
+                return True
+            except Exception as e:
+                PrettyOutput.print(f"无法创建工具目录 {tools_dir}: {e}", OutputType.ERROR)
+                return False
+        
+        return True
+
+    def execute(self, args: Dict[str, Any]) -> Dict[str, Any]:
+        """
+        生成新工具并注册到当前的工具注册表中
+
+        参数:
+            args: 包含工具名称和工具代码的字典
+
+        返回:
+            Dict: 包含生成结果的字典
+        """
+        try:
+            # 从参数中获取工具信息
+            tool_name = args["tool_name"]
+            tool_code = args["tool_code"]
+            agent = args.get("agent", None)
+            
+            # 验证工具名称
+            if not tool_name.isidentifier():
+                return {
+                    "success": False,
+                    "stdout": "",
+                    "stderr": f"工具名称 '{tool_name}' 不是有效的Python标识符"
+                }
+            
+            # 准备工具目录
+            tools_dir = Path(get_data_dir()) / "tools"
+            tools_dir.mkdir(parents=True, exist_ok=True)
+            
+            # 生成工具文件路径
+            tool_file_path = tools_dir / f"{tool_name}.py"
+            
+            # 检查是否已存在同名工具
+            if tool_file_path.exists():
+                return {
+                    "success": False,
+                    "stdout": "",
+                    "stderr": f"工具 '{tool_name}' 已经存在于 {tool_file_path}"
+                }
+            
+            # 验证并处理工具代码
+            processed_code, error_msg = self._validate_and_process_code(tool_name, tool_code)
+            if error_msg:
+                return {
+                    "success": False,
+                    "stdout": "",
+                    "stderr": error_msg
+                }
+            
+            # 写入工具文件
+            with open(tool_file_path, "w", encoding="utf-8") as f:
+                f.write(processed_code)
+            
+            # 注册新工具到当前的工具注册表
+            success_message = f"工具 '{tool_name}' 已成功生成在 {tool_file_path}"
+            
+            if agent:
+                tool_registry = agent.get_tool_registry()
+                if tool_registry:
+                    # 尝试加载并注册新工具
+                    if tool_registry.register_tool_by_file(str(tool_file_path)):
+                        success_message += f"\n已成功注册到当前会话的工具注册表中"
+                else:
+                    success_message += f"\n注册到当前会话失败，可能需要重新启动Jarvis"
+            
+            return {
+                "success": True,
+                "stdout": success_message,
+                "stderr": ""
+            }
+            
+        except Exception as e:
+            # 如果发生异常，返回失败响应，包含错误信息
+            return {
+                "success": False,
+                "stdout": "",
+                "stderr": f"生成工具失败: {str(e)}"
+            }
+    
+    def _validate_and_process_code(self, tool_name: str, tool_code: str) -> Tuple[str, str]:
+        """
+        验证并处理工具代码
+        
+        参数:
+            tool_name: 工具名称
+            tool_code: 工具代码
+            
+        返回:
+            Tuple[str, str]: (处理后的代码, 错误信息)
+        """
+        # 检查工具代码中是否包含类定义
+        if f"class {tool_name}" not in tool_code:
+            # 尝试找到任何类定义
+            class_match = re.search(r"class\s+(\w+)", tool_code)
+            if class_match:
+                old_class_name = class_match.group(1)
+                # 替换类名为工具名
+                tool_code = tool_code.replace(f"class {old_class_name}", f"class {tool_name}")
+                tool_code = tool_code.replace(f'name = "{old_class_name}"', f'name = "{tool_name}"')
+            else:
+                # 没有找到类定义，返回错误
+                return "", f"工具代码中缺少类定义 'class {tool_name}'"
+        
+        # 检查工具代码中是否包含必要的属性和方法
+        missing_components = []
+        
+        if f'name = "{tool_name}"' not in tool_code and f"name = '{tool_name}'" not in tool_code:
+            # 尝试查找任何name属性并修复
+            name_match = re.search(r'name\s*=\s*["\'](\w+)["\']', tool_code)
+            if name_match:
+                old_name = name_match.group(1)
+                tool_code = re.sub(r'name\s*=\s*["\'](\w+)["\']', f'name = "{tool_name}"', tool_code)
+            else:
+                missing_components.append(f"name = \"{tool_name}\"")
+        
+        if "description = " not in tool_code:
+            missing_components.append("description 属性")
+        
+        if "parameters = " not in tool_code:
+            missing_components.append("parameters 属性")
+        
+        if "def execute(self, args:" not in tool_code:
+            missing_components.append("execute 方法")
+        
+        if "def check(" not in tool_code:
+            # 添加默认的check方法
+            class_match = re.search(r"class\s+(\w+).*?:", tool_code, re.DOTALL)
+            if class_match:
+                indent = "    "  # 默认缩进
+                # 找到类定义后的第一个属性
+                first_attr_match = re.search(r"class\s+(\w+).*?:(.*?)(\w+\s*=)", tool_code, re.DOTALL)
+                if first_attr_match:
+                    # 获取属性前的缩进
+                    attr_indent = re.search(r"\n([ \t]*)\w+\s*=", first_attr_match.group(2))
+                    if attr_indent:
+                        indent = attr_indent.group(1)
+                
+                check_method = f"\n{indent}@staticmethod\n{indent}def check() -> bool:\n{indent}    \"\"\"检查工具是否可用\"\"\"\n{indent}    return True\n"
+                
+                # 在类定义后插入check方法
+                pattern = r"(class\s+(\w+).*?:.*?)(\n\s*\w+\s*=|\n\s*@|\n\s*def)"
+                replacement = r"\1" + check_method + r"\3"
+                tool_code = re.sub(pattern, replacement, tool_code, 1, re.DOTALL)
+        
+        # 如果缺少必要组件，返回错误信息
+        if missing_components:
+            return "", f"工具代码中缺少以下必要组件: {', '.join(missing_components)}"
+        
+        # 确保代码有正确的Python文件头部
+        if not tool_code.startswith("# -*- coding:") and not tool_code.startswith("# coding="):
+            tool_code = "# -*- coding: utf-8 -*-\n" + tool_code
+        
+        # 确保导入了必要的模块
+        if "from typing import Dict, Any" not in tool_code:
+            imports_pos = tool_code.find("\n\n")
+            if imports_pos > 0:
+                tool_code = tool_code[:imports_pos] + "\nfrom typing import Dict, Any" + tool_code[imports_pos:]
+            else:
+                tool_code = "from typing import Dict, Any\n\n" + tool_code
+        
+        return tool_code, ""

jarvis/jarvis_tools/lsp_get_diagnostics.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 import os
 from typing import Dict, Any
 from jarvis.jarvis_lsp.registry import LSPRegistry

jarvis/jarvis_tools/methodology.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 import os
 import json
 import hashlib

jarvis/jarvis_tools/read_code.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 from typing import Dict, Any
 import os
 

jarvis/jarvis_tools/read_webpage.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 from typing import Dict, Any
 import os
 from jarvis.jarvis_platform.registry import PlatformRegistry

jarvis/jarvis_tools/registry.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 import json
 from pathlib import Path
 import re

jarvis/jarvis_tools/rewrite_file.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 """
 文件重写工具类
 

jarvis/jarvis_tools/search_web.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 import os
 from typing import Any, Dict
 from jarvis.jarvis_platform.registry import PlatformRegistry

jarvis/jarvis_tools/virtual_tty.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 from typing import Dict, Any
 import os
 import time

jarvis/jarvis_utils/__init__.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 """
 Jarvis工具模块
 该模块提供了Jarvis系统中使用的各种实用函数和类。

jarvis/jarvis_utils/builtin_replace_map.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 """内置替换映射表模块。
 
 该模块定义了Jarvis系统内置的默认替换映射表。

jarvis/jarvis_utils/config.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 import os
 from functools import lru_cache
 import yaml

jarvis/jarvis_utils/embedding.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 import os
 from transformers import AutoTokenizer
 from typing import List

jarvis/jarvis_utils/file_processors.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 import unicodedata
 
 class FileProcessor:

jarvis/jarvis_utils/git_utils.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 """
 Git工具模块
 该模块提供了与Git仓库交互的工具。

jarvis/jarvis_utils/globals.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 """
 全局变量和配置模块
 该模块管理Jarvis系统的全局状态和配置。

jarvis/jarvis_utils/input.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 """
 输入处理模块
 该模块提供了处理Jarvis系统中用户输入的实用工具。

jarvis/jarvis_utils/methodology.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 """
 方法论管理模块
 该模块提供了加载和搜索方法论的实用工具。
@@ -9,7 +10,7 @@
 import os
 import json
 import tempfile
-from typing import Dict, Optional
+from typing import Any, Dict, Optional
 
 from jarvis.jarvis_utils.config import get_data_dir
 from jarvis.jarvis_utils.output import PrettyOutput, OutputType
@@ -91,7 +92,7 @@ def _create_methodology_temp_file(methodologies: Dict[str, str]) -> Optional[str
         PrettyOutput.print(f"创建方法论临时文件失败: {str(e)}", OutputType.ERROR)
         return None
 
-def load_methodology(user_input: str) -> str:
+def load_methodology(user_input: str, tool_registery: Optional[Any] = None) -> str:
     """
     加载方法论并上传到大模型。
 
@@ -102,6 +103,9 @@ def load_methodology(user_input: str) -> str:
         str: 相关的方法论提示，如果未找到方法论则返回空字符串
     """
     from yaspin import yaspin # type: ignore
+    from jarvis.jarvis_tools.registry import ToolRegistry
+
+    prompt = tool_registery.prompt() if tool_registery else ""
 
     # 获取方法论目录
     methodology_dir = _get_methodology_directory()
@@ -132,10 +136,13 @@ def load_methodology(user_input: str) -> str:
         full_content = base_prompt
         for problem_type, content in methodologies.items():
             full_content += f"## {problem_type}\n\n{content}\n\n---\n\n"
+
+        full_content += f"以下是所有可用的工具内容：\n\n"
+        full_content += prompt
         
         # 添加用户输入和输出要求
         full_content += f"""
-请根据以上方法论内容，规划/总结出以下用户需求的执行步骤: {user_input}
+请根据以上方法论和可调用的工具内容，规划/总结出以下用户需求的执行步骤: {user_input}
 
 请按以下格式回复：
 ### 与该任务/需求相关的方法论
@@ -174,7 +181,7 @@ def load_methodology(user_input: str) -> str:
                     if upload_success:
                         # 使用上传的文件生成摘要
                         return platform.chat_until_success(base_prompt + f"""
-请根据已上传的方法论文件内容，规划/总结出以下用户需求的执行步骤: {user_input}
+请根据已上传的方法论和可调用的工具文件内容，规划/总结出以下用户需求的执行步骤: {user_input}
 
 请按以下格式回复：
 ### 与该任务/需求相关的方法论
@@ -191,7 +198,7 @@ def load_methodology(user_input: str) -> str:
                     elif hasattr(platform, 'chat_big_content'):
                         # 如果上传失败但支持大内容处理，使用chat_big_content
                         return platform.chat_big_content(full_content, base_prompt + f"""
-请根据以上方法论内容，规划/总结出以下用户需求的执行步骤: {user_input}
+请根据以上方法论和可调用的工具文件内容，规划/总结出以下用户需求的执行步骤: {user_input}
 
 请按以下格式回复：
 ### 与该任务/需求相关的方法论

jarvis/jarvis_utils/output.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 """
 输出格式化模块
 该模块为Jarvis系统提供了丰富的文本格式化和显示工具。

jarvis/jarvis_utils/tag.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 def ot(tag_name: str) -> str:
     """生成HTML标签开始标记
 

jarvis/jarvis_utils/utils.py

@@ -1,6 +1,9 @@
+# -*- coding: utf-8 -*-
+from doctest import script_from_examples
 import os
 import time
 import hashlib
+import tarfile
 from pathlib import Path
 from typing import List, Any, Callable
 from jarvis.jarvis_utils.config import INPUT_WINDOW_REVERSE_SIZE, get_max_input_token_count, get_data_dir
@@ -18,9 +21,24 @@ def init_env() -> None:
     jarvis_dir = Path(get_data_dir())
     env_file = jarvis_dir / "env"
 
+    script_dir = Path(os.path.dirname(os.path.dirname(__file__)))
+    hf_archive = script_dir / "jarvis_data" / "huggingface.tar.gz"
+
     # 检查jarvis_data目录是否存在
     if not jarvis_dir.exists():
         jarvis_dir.mkdir(parents=True)
+
+    # 检查并解压huggingface模型
+    hf_dir = jarvis_dir / "huggingface" / "hub"
+    if not hf_dir.exists() and hf_archive.exists():
+        try:
+            PrettyOutput.print("正在解压HuggingFace模型...", OutputType.INFO)
+            with tarfile.open(hf_archive, "r:gz") as tar:
+                tar.extractall(path=jarvis_dir)
+            PrettyOutput.print("HuggingFace模型解压完成", OutputType.SUCCESS)
+        except Exception as e:
+            PrettyOutput.print(f"解压HuggingFace模型失败: {e}", OutputType.ERROR)
+
     if env_file.exists():
         try:
             with open(env_file, "r", encoding="utf-8", errors="ignore") as f:
@@ -48,7 +66,7 @@ def while_success(func: Callable[[], Any], sleep_time: float = 0.1) -> Any:
         try:
             return func()
         except Exception as e:
-            PrettyOutput.print(f"执行失败: {str(e)}, 等待 {sleep_time}s...", OutputType.ERROR)
+            PrettyOutput.print(f"执行失败: {str(e)}, 等待 {sleep_time}s...", OutputType.WARNING)
             time.sleep(sleep_time)
             continue
 def while_true(func: Callable[[], bool], sleep_time: float = 0.1) -> Any:

{jarvis_ai_assistant-0.1.164.dist-info → jarvis_ai_assistant-0.1.166.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: jarvis-ai-assistant
-Version: 0.1.164
+Version: 0.1.166
 Summary: Jarvis: An AI assistant that uses tools to interact with the system
 Home-page: https://github.com/skyfireitdiy/Jarvis
 Author: skyfire
@@ -55,6 +55,7 @@ Requires-Dist: torch==2.4.1
 Requires-Dist: python-Levenshtein==0.25.1
 Requires-Dist: sseclient==0.0.27
 Requires-Dist: pillow==10.2.0
+Requires-Dist: openai==1.78.1
 Provides-Extra: dev
 Requires-Dist: pytest; extra == "dev"
 Requires-Dist: black; extra == "dev"
@@ -105,11 +106,19 @@ Dynamic: requires-python
 
 
 ## 🚀 快速开始 <a id="quick-start"></a>
+### 系统要求
+- 目前只能在Linux系统下使用（很多工具依赖Linux系统）
+- Windows没有测试过，但Windows 10以上的用户可以在WSL上使用此工具
+
 ### 安装
 ```bash
+# 从源码安装（推荐）
 git clone https://github.com/skyfireitdiy/Jarvis
 cd Jarvis
 pip3 install -e .
+
+# 或者从PyPI安装（可能更新不及时）
+pip3 install jarvis-ai-assistant
 ```
 
 ### 最小化配置
@@ -148,55 +157,44 @@ Kimi API Key获取方式：
 
 删除Bearer前缀，剩下的内容就是Kimi API Key。
 
-以上配置编写到`~/.jarvis/env`文件中。
 
-### 基本使用
+#### OpenAI
 ```bash
-# 使用通用代理
-jarvis
-
-# 使用代码代理
-jarvis-code-agent
-# 或者 jca
-
-# 使用智能shell的功能
-jarvis-smart-shell --help
-# 或者 jss
-
-# 使用平台管理的功能
-jarvis-platform-manager --help
-
-# 使用代码审查的功能
-jarvis-code-review --help
-
-# 使用自动化git commit的功能
-jarvis-git-commit --help
-# 或者 jgc
-
-# 使用dev功能（开发中）
-jarvis-dev --help
-
-# 使用git squash的功能
-jarvis-git-squash --help
-
-# 使用多代理的功能
-jarvis-multi-agent --help
-
-# 使用agent的功能
-jarvis-agent --help
+JARVIS_PLATFORM=openai
+JARVIS_MODEL=gpt-4o  # 默认模型，可选gpt-4-turbo, gpt-3.5-turbo等
+JARVIS_THINKING_PLATFORM=openai
+JARVIS_THINKING_MODEL=gpt-4o
+
+OPENAI_API_KEY=<OpenAI API Key>
+OPENAI_API_BASE=https://api.openai.com/v1  # 可选，默认为官方API地址
+OPENAI_MODEL_NAME=gpt-4o  # 可选，覆盖JARVIS_MODEL设置
+```
 
-# 使用工具的功能
-jarvis-tool --help
+配置说明：
+1. `OPENAI_API_KEY`: 必填。
+2. `OPENAI_API_BASE`: 可选，用于自定义API端点
 
-# 使用代码库查询功能
-jarvis-ask-codebase --help
+以上配置编写到`~/.jarvis/env`文件中。
 
-# 使用git details的功能
-jarvis-git-details --help
+支持的模型可通过`jarvis-platform-manager --list-models`查看完整列表。
 
-# 使用方法论的功能
-jarvis-methodology --help
-```
+### 基本使用
+| 命令 | 快捷方式 | 功能描述 |
+|------|----------|----------|
+| `jarvis` | - | 使用通用代理 |
+| `jarvis-code-agent` | `jca` | 使用代码代理 |
+| `jarvis-smart-shell` | `jss` | 使用智能shell功能 |
+| `jarvis-platform-manager` | - | 使用平台管理功能 |
+| `jarvis-code-review` | - | 使用代码审查功能 |
+| `jarvis-git-commit` | `jgc` | 使用自动化git commit功能 |
+| `jarvis-dev` | - | 使用dev功能（开发中） |
+| `jarvis-git-squash` | - | 使用git squash功能 |
+| `jarvis-multi-agent` | - | 使用多代理功能 |
+| `jarvis-agent` | - | 使用agent功能 |
+| `jarvis-tool` | - | 使用工具功能 |
+| `jarvis-ask-codebase` | `jac` | 使用代码库查询功能 |
+| `jarvis-git-details` | - | 使用git details功能 |
+| `jarvis-methodology` | - | 使用方法论功能 |
 
 ---