jarvis-ai-assistant 0.1.149__tar.gz → 0.1.151__tar.gz

{jarvis_ai_assistant-0.1.149 → jarvis_ai_assistant-0.1.151}/MANIFEST.in

@@ -2,4 +2,5 @@ include README.md
 include LICENSE
 include pyproject.toml
 recursive-include src/jarvis * 
-exclude __pycache__
+exclude __pycache__
+exclude .venv

{jarvis_ai_assistant-0.1.149/src/jarvis_ai_assistant.egg-info → jarvis_ai_assistant-0.1.151}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: jarvis-ai-assistant
-Version: 0.1.149
+Version: 0.1.151
 Summary: Jarvis: An AI assistant that uses tools to interact with the system
 Home-page: https://github.com/skyfireitdiy/Jarvis
 Author: skyfire
@@ -43,25 +43,27 @@ License-File: LICENSE
 Requires-Dist: requests==2.32.3
 Requires-Dist: colorama==0.4.6
 Requires-Dist: prompt_toolkit==3.0.50
-Requires-Dist: PyMuPDF==1.24.11
 Requires-Dist: yaspin==2.4.0
 Requires-Dist: pygments==2.19.1
 Requires-Dist: fuzzywuzzy==0.18.0
 Requires-Dist: jedi==0.19.2
 Requires-Dist: fastapi==0.115.12
 Requires-Dist: uvicorn==0.33.0
-Requires-Dist: pandas==2.0.3
 Requires-Dist: rich==14.0.0
 Requires-Dist: transformers==4.46.3
-Requires-Dist: python-docx==1.1.2
-Requires-Dist: python-pptx==1.0.2
 Requires-Dist: torch==2.4.1
 Requires-Dist: python-Levenshtein==0.25.1
+Requires-Dist: sseclient==0.0.27
+Requires-Dist: pillow==10.2.1
 Provides-Extra: dev
 Requires-Dist: pytest; extra == "dev"
 Requires-Dist: black; extra == "dev"
 Requires-Dist: isort; extra == "dev"
 Requires-Dist: mypy; extra == "dev"
+Dynamic: author
+Dynamic: home-page
+Dynamic: license-file
+Dynamic: requires-python
 
 # 🤖 Jarvis AI 助手
 <p align="center">
@@ -191,13 +193,12 @@ jarvis-methodology --help
 |------|----------|--------|------|
 | 核心配置 | `JARVIS_MAX_TOKEN_COUNT` | 102400000 | 上下文窗口的最大token数量 |
 | 核心配置 | `JARVIS_MAX_INPUT_TOKEN_COUNT` | 32000 | 输入的最大token数量 |
-| 核心配置 | `JARVIS_THREAD_COUNT` | 1 | 并行处理的线程数量 |
-| 核心配置 | `JARVIS_AUTO_COMPLETE` | false | 是否启用自动补全功能 
+| 核心配置 | `JARVIS_AUTO_COMPLETE` | false | 是否启用自动完成功能（任务判定完成的时候会自动终止） |
 | 核心配置 | `JARVIS_SHELL_NAME` | bash | 系统shell名称 |
 | 核心配置 | `JARVIS_PLATFORM` | yuanbao | 默认AI平台 |
 | 核心配置 | `JARVIS_MODEL` | deep_seek_v3 | 默认模型 |
-| 核心配置 | `JARVIS_THINKING_PLATFORM` | JARVIS_PLATFORM | 思考任务使用的平台 |
-| 核心配置 | `JARVIS_THINKING_MODEL` | JARVIS_MODEL | 思考任务使用的模型 |
+| 核心配置 | `JARVIS_THINKING_PLATFORM` | JARVIS_PLATFORM | 推理任务使用的平台 |
+| 核心配置 | `JARVIS_THINKING_MODEL` | JARVIS_MODEL | 推理任务使用的模型 |
 | 核心配置 | `JARVIS_EXECUTE_TOOL_CONFIRM` | false | 执行工具前是否需要确认 |
 | 核心配置 | `JARVIS_CONFIRM_BEFORE_APPLY_PATCH` | true | 应用补丁前是否需要确认 |
 | 核心配置 | `JARVIS_MAX_TOOL_CALL_COUNT` | 20 | 最大连续工具调用次数 |
@@ -212,17 +213,17 @@ jarvis-methodology --help
 | ask_user | 交互式用户输入收集 |
 | chdir | 更改当前工作目录 |
 | code_plan | 理解需求并制定详细的代码修改计划，在修改前获取用户确认 |
-| create_code_agent | 代码开发工具，当需要修改代码时使用，如果只是简单文件修改，使用文件操作或者脚本即可 |
+| create_code_agent | 代码开发工具，当需要修改代码时使用 |
 | create_sub_agent | 创建子代理以处理特定任务，子代理将生成任务总结报告 |
 | execute_script | 执行脚本并返回结果，支持任意解释器。 |
 | file_analyzer | 分析文件内容并提取关键信息。支持的文件：文本文件、word文档、pdf文件、图片 |
-| file_operation | 文件批量操作工具，可批量读写多个文件，支持文本、PDF、Word、Excel、PPT等格式，适用于需要同时处理多个文件的场景（读取配置文件、保存生成内容等） |
+| file_operation | 文件批量操作工具，可批量读写多个文件，支持文本文件，适用于需要同时处理多个文件的场景（读取配置文件、保存生成内容等） |
 | find_methodology | 方法论查找工具，用于在执行过程中查看历史方法论辅助决策 |
 | lsp_get_diagnostics | 获取代码诊断信息（错误、警告） |
-| methodology | 经验管理工具，支持添加、更新和删除操作 |
+| methodology | 方法论管理工具，支持添加、更新和删除操作 |
 | read_code | 代码阅读与分析工具，用于读取源代码文件并添加行号，针对代码文件优化，提供更好的格式化输出和行号显示，适用于代码分析、审查和理解代码实现的场景 |
-| read_webpage | 读取网页内容，提取标题、文本和超链接 |
-| search_web | 使用bing进行网络搜索 |
+| read_webpage | 读取网页内容并分析 |
+| search_web | 使用互联网搜索 |
 | virtual_tty | 控制虚拟终端执行各种操作，如启动终端、输入命令、获取输出等。 |
 
 
@@ -232,7 +233,7 @@ jarvis-methodology --help
 ---
 ## 🛠️ 扩展开发 <a id="extensions"></a>
 ### 添加新工具
-在 `~/.jarvis/tools/》 中创建新的 Python 文件：
+在 `~/.jarvis/tools/` 中创建新的 Python 文件：
 ```python
 from typing import Dict, Any
 from jarvis.utils import OutputType, PrettyOutput
@@ -280,8 +281,32 @@ class CustomTool:
 ```
 
 
+### 添加MCP
+MCP(模型上下文协议)。在`~/.jarvis/tools/mcp/`中创建YAML配置文件：
+
+#### 本地MCP配置（`stdio`模式）
+```yaml
+type: local
+name: MCP名称
+command: 可执行命令
+args: [参数列表]  # 可选
+env:  # 可选环境变量
+  KEY: VALUE
+```
+
+#### 远程MCP配置（`sse`模式）
+```yaml
+type: remote
+name: MCP名称
+base_url: http://example.com/api
+auth_token: 认证令牌  # 可选
+headers:  # 可选HTTP头
+  X-Custom-Header: value
+```
+
+
 ### 添加新大模型平台
-在 `~/.jarvis/platforms/》 中创建新的 Python 文件：
+在 `~/.jarvis/platforms/` 中创建新的 Python 文件：
 ```python
 from jarvis.jarvis_platform.base import BasePlatform
 class CustomPlatform(BasePlatform):

{jarvis_ai_assistant-0.1.149 → jarvis_ai_assistant-0.1.151}/README.md

@@ -126,13 +126,12 @@ jarvis-methodology --help
 |------|----------|--------|------|
 | 核心配置 | `JARVIS_MAX_TOKEN_COUNT` | 102400000 | 上下文窗口的最大token数量 |
 | 核心配置 | `JARVIS_MAX_INPUT_TOKEN_COUNT` | 32000 | 输入的最大token数量 |
-| 核心配置 | `JARVIS_THREAD_COUNT` | 1 | 并行处理的线程数量 |
-| 核心配置 | `JARVIS_AUTO_COMPLETE` | false | 是否启用自动补全功能 
+| 核心配置 | `JARVIS_AUTO_COMPLETE` | false | 是否启用自动完成功能（任务判定完成的时候会自动终止） |
 | 核心配置 | `JARVIS_SHELL_NAME` | bash | 系统shell名称 |
 | 核心配置 | `JARVIS_PLATFORM` | yuanbao | 默认AI平台 |
 | 核心配置 | `JARVIS_MODEL` | deep_seek_v3 | 默认模型 |
-| 核心配置 | `JARVIS_THINKING_PLATFORM` | JARVIS_PLATFORM | 思考任务使用的平台 |
-| 核心配置 | `JARVIS_THINKING_MODEL` | JARVIS_MODEL | 思考任务使用的模型 |
+| 核心配置 | `JARVIS_THINKING_PLATFORM` | JARVIS_PLATFORM | 推理任务使用的平台 |
+| 核心配置 | `JARVIS_THINKING_MODEL` | JARVIS_MODEL | 推理任务使用的模型 |
 | 核心配置 | `JARVIS_EXECUTE_TOOL_CONFIRM` | false | 执行工具前是否需要确认 |
 | 核心配置 | `JARVIS_CONFIRM_BEFORE_APPLY_PATCH` | true | 应用补丁前是否需要确认 |
 | 核心配置 | `JARVIS_MAX_TOOL_CALL_COUNT` | 20 | 最大连续工具调用次数 |
@@ -147,17 +146,17 @@ jarvis-methodology --help
 | ask_user | 交互式用户输入收集 |
 | chdir | 更改当前工作目录 |
 | code_plan | 理解需求并制定详细的代码修改计划，在修改前获取用户确认 |
-| create_code_agent | 代码开发工具，当需要修改代码时使用，如果只是简单文件修改，使用文件操作或者脚本即可 |
+| create_code_agent | 代码开发工具，当需要修改代码时使用 |
 | create_sub_agent | 创建子代理以处理特定任务，子代理将生成任务总结报告 |
 | execute_script | 执行脚本并返回结果，支持任意解释器。 |
 | file_analyzer | 分析文件内容并提取关键信息。支持的文件：文本文件、word文档、pdf文件、图片 |
-| file_operation | 文件批量操作工具，可批量读写多个文件，支持文本、PDF、Word、Excel、PPT等格式，适用于需要同时处理多个文件的场景（读取配置文件、保存生成内容等） |
+| file_operation | 文件批量操作工具，可批量读写多个文件，支持文本文件，适用于需要同时处理多个文件的场景（读取配置文件、保存生成内容等） |
 | find_methodology | 方法论查找工具，用于在执行过程中查看历史方法论辅助决策 |
 | lsp_get_diagnostics | 获取代码诊断信息（错误、警告） |
-| methodology | 经验管理工具，支持添加、更新和删除操作 |
+| methodology | 方法论管理工具，支持添加、更新和删除操作 |
 | read_code | 代码阅读与分析工具，用于读取源代码文件并添加行号，针对代码文件优化，提供更好的格式化输出和行号显示，适用于代码分析、审查和理解代码实现的场景 |
-| read_webpage | 读取网页内容，提取标题、文本和超链接 |
-| search_web | 使用bing进行网络搜索 |
+| read_webpage | 读取网页内容并分析 |
+| search_web | 使用互联网搜索 |
 | virtual_tty | 控制虚拟终端执行各种操作，如启动终端、输入命令、获取输出等。 |
 
 
@@ -167,7 +166,7 @@ jarvis-methodology --help
 ---
 ## 🛠️ 扩展开发 <a id="extensions"></a>
 ### 添加新工具
-在 `~/.jarvis/tools/》 中创建新的 Python 文件：
+在 `~/.jarvis/tools/` 中创建新的 Python 文件：
 ```python
 from typing import Dict, Any
 from jarvis.utils import OutputType, PrettyOutput
@@ -215,8 +214,32 @@ class CustomTool:
 ```
 
 
+### 添加MCP
+MCP(模型上下文协议)。在`~/.jarvis/tools/mcp/`中创建YAML配置文件：
+
+#### 本地MCP配置（`stdio`模式）
+```yaml
+type: local
+name: MCP名称
+command: 可执行命令
+args: [参数列表]  # 可选
+env:  # 可选环境变量
+  KEY: VALUE
+```
+
+#### 远程MCP配置（`sse`模式）
+```yaml
+type: remote
+name: MCP名称
+base_url: http://example.com/api
+auth_token: 认证令牌  # 可选
+headers:  # 可选HTTP头
+  X-Custom-Header: value
+```
+
+
 ### 添加新大模型平台
-在 `~/.jarvis/platforms/》 中创建新的 Python 文件：
+在 `~/.jarvis/platforms/` 中创建新的 Python 文件：
 ```python
 from jarvis.jarvis_platform.base import BasePlatform
 class CustomPlatform(BasePlatform):

{jarvis_ai_assistant-0.1.149 → jarvis_ai_assistant-0.1.151}/pyproject.toml

@@ -2,9 +2,13 @@
 requires = ["setuptools>=45", "wheel"]
 build-backend = "setuptools.build_meta"
 
+[[tool.uv.index]]
+url = "https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple"
+default = true
+
 [project]
 name = "jarvis-ai-assistant"
-version = "0.1.149"
+version = "0.1.151"
 description = "Jarvis: An AI assistant that uses tools to interact with the system"
 readme = "README.md"
 authors = [{ name = "skyfire", email = "skyfireitdiy@hotmail.com" }]
@@ -25,20 +29,18 @@ dependencies = [
     "requests==2.32.3",
     "colorama==0.4.6",
     "prompt_toolkit==3.0.50",
-    "PyMuPDF==1.24.11",
     "yaspin==2.4.0",
     "pygments==2.19.1",
     "fuzzywuzzy==0.18.0",
     "jedi==0.19.2",
     "fastapi==0.115.12",
     "uvicorn==0.33.0",
-    "pandas==2.0.3",
     "rich==14.0.0",
     "transformers==4.46.3",
-    "python-docx==1.1.2",
-    "python-pptx==1.0.2",
     "torch==2.4.1",
     "python-Levenshtein==0.25.1",
+    "sseclient==0.0.27",
+    "pillow==10.2.1",
 ]
 requires-python = ">=3.8"
 

{jarvis_ai_assistant-0.1.149 → jarvis_ai_assistant-0.1.151}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="jarvis-ai-assistant",
-    version="0.1.149",
+    version="0.1.151",
     author="skyfire",
     author_email="skyfireitdiy@hotmail.com",
     description="An AI assistant that uses various tools to interact with the system",
@@ -16,20 +16,18 @@ setup(
         "requests==2.32.3",
         "colorama==0.4.6",
         "prompt_toolkit==3.0.50",
-        "PyMuPDF==1.24.11",
         "yaspin==2.4.0",
         "pygments==2.19.1",
         "fuzzywuzzy==0.18.0",
         "jedi==0.19.2",
         "fastapi==0.115.12",
         "uvicorn==0.33.0",
-        "pandas==2.0.3",
         "rich==14.0.0",
         "transformers==4.46.3",
-        "python-docx==1.1.2",
-        "python-pptx==1.0.2",
         "torch==2.4.1",
         "python-Levenshtein==0.25.1",
+        "sseclient==0.0.27",
+        "pillow==10.2.1",
     ],
     entry_points={
         "console_scripts": [

{jarvis_ai_assistant-0.1.149 → jarvis_ai_assistant-0.1.151}/src/jarvis/__init__.py

@@ -1,3 +1,3 @@
 """Jarvis AI Assistant"""
 
-__version__ = "0.1.149"
+__version__ = "0.1.151"

{jarvis_ai_assistant-0.1.149 → jarvis_ai_assistant-0.1.151}/src/jarvis/jarvis_agent/jarvis.py

@@ -4,6 +4,7 @@ import sys
 
 from typing import Dict  # 仅保留实际使用的类型导入
 
+from jarvis.jarvis_utils.config import get_data_dir
 from prompt_toolkit import prompt
 import yaml
 from yaspin import yaspin
@@ -25,13 +26,14 @@ def _load_tasks() -> Dict[str, str]:
     """Load tasks from .jarvis files in user home and current directory."""
     tasks: Dict[str, str] = {}
 
-    # Check .jarvis/pre-command in user directory
-    user_jarvis = os.path.expanduser("~/.jarvis/pre-command")
-    if os.path.exists(user_jarvis):
-        spinner_text = f"从{user_jarvis}加载预定义任务..."
+    # Check pre-command in data directory
+    data_dir = get_data_dir()
+    pre_command_path = os.path.join(data_dir, "pre-command")
+    if os.path.exists(pre_command_path):
+        spinner_text = f"从{pre_command_path}加载预定义任务..."
         with yaspin(text=spinner_text, color="cyan") as spinner:
             try:
-                with open(user_jarvis, "r", encoding="utf-8", errors="ignore") as f:
+                with open(pre_command_path, "r", encoding="utf-8", errors="ignore") as f:
                     user_tasks = yaml.safe_load(f)
                 if isinstance(user_tasks, dict):
                     for name, desc in user_tasks.items():

{jarvis_ai_assistant-0.1.149 → jarvis_ai_assistant-0.1.151}/src/jarvis/jarvis_git_utils/git_commiter.py

@@ -177,6 +177,13 @@ class GitCommitTool:
                             # 如果成功提取，就跳出循环
                             if commit_message:
                                 break
+                            prompt = f"""格式错误，请按照以下格式重新生成提交信息：
+                            {ot("COMMIT_MESSAGE")}
+        <类型>(<范围>): <主题>
+        
+        [可选] 详细描述变更内容和原因
+        {ct("COMMIT_MESSAGE")}
+                            """
                         spinner.write("✅ 生成提交消息")
 
                         # 执行提交

{jarvis_ai_assistant-0.1.149 → jarvis_ai_assistant-0.1.151}/src/jarvis/jarvis_lsp/registry.py

@@ -6,6 +6,7 @@ import sys
 from typing import Dict, Type, Optional, List
 from jarvis.jarvis_lsp.base import BaseLSP
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput
+from jarvis.jarvis_utils.config import get_data_dir
 
 REQUIRED_METHODS = [
     ('initialize', ['workspace_path']),
@@ -21,7 +22,7 @@ class LSPRegistry:
     @staticmethod
     def get_lsp_dir() -> str:
         """Get LSP implementation directory."""
-        user_lsp_dir = os.path.expanduser("~/.jarvis/lsp")
+        user_lsp_dir = os.path.join(get_data_dir(), "lsp")
         if not os.path.exists(user_lsp_dir):
             try:
                 os.makedirs(user_lsp_dir)

jarvis_ai_assistant-0.1.151/src/jarvis/jarvis_mcp/__init__.py

@@ -0,0 +1,65 @@
+from abc import ABC, abstractmethod
+from typing import Any, Dict, List
+
+
+class McpClient(ABC):
+    """MCP客户端抽象基类"""
+    
+    @abstractmethod
+    def get_tool_list(self) -> List[Dict[str, Any]]:
+        """获取工具列表
+        
+        返回:
+            List[Dict[str, Any]]: 工具列表，每个工具包含以下字段：
+                - name: str - 工具名称
+                - description: str - 工具描述
+                - parameters: Dict - 工具参数
+        """
+        pass
+    
+    @abstractmethod
+    def execute(self, tool_name: str, arguments: Dict[str, Any]) -> Dict[str, Any]:
+        """执行工具
+        
+        参数:
+            tool_name: 工具名称
+            arguments: 参数字典，包含工具执行所需的参数
+            
+        返回:
+            Dict[str, Any]: 执行结果，包含以下字段：
+                - success: bool - 是否执行成功
+                - stdout: str - 标准输出
+                - stderr: str - 标准错误
+        """
+        pass
+
+    @abstractmethod
+    def get_resource_list(self) -> List[Dict[str, Any]]:
+        """获取资源列表
+        
+        返回:
+            List[Dict[str, Any]]: 资源列表，每个资源包含以下字段：
+                - uri: str - 资源的唯一标识符
+                - name: str - 资源的名称
+                - description: str - 资源的描述（可选）
+                - mimeType: str - 资源的MIME类型（可选）
+        """
+        pass
+
+    @abstractmethod
+    def get_resource(self, uri: str) -> Dict[str, Any]:
+        """获取指定资源的内容
+        
+        参数:
+            uri: str - 资源的URI标识符
+            
+        返回:
+            Dict[str, Any]: 资源内容，包含以下字段：
+                - uri: str - 资源的URI
+                - mimeType: str - 资源的MIME类型（可选）
+                - text: str - 文本内容（如果是文本资源）
+                - blob: str - 二进制内容（如果是二进制资源，base64编码）
+        """
+        pass
+
+