jarvis-ai-assistant 0.1.164__tar.gz → 0.1.166__tar.gz

{jarvis_ai_assistant-0.1.164/src/jarvis_ai_assistant.egg-info → jarvis_ai_assistant-0.1.166}/PKG-INFO

@@ -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` | - | 使用方法论功能 |
 
 ---
 

{jarvis_ai_assistant-0.1.164 → jarvis_ai_assistant-0.1.166}/README.md

@@ -36,11 +36,19 @@
 
 
 ## 🚀 快速开始 <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
 ```
 
 ### 最小化配置
@@ -79,55 +87,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` | - | 使用方法论功能 |
 
 ---
 

{jarvis_ai_assistant-0.1.164 → jarvis_ai_assistant-0.1.166}/pyproject.toml

@@ -8,7 +8,7 @@ default = true
 
 [project]
 name = "jarvis-ai-assistant"
-version = "0.1.164"
+version = "0.1.166"
 description = "Jarvis: An AI assistant that uses tools to interact with the system"
 readme = "README.md"
 authors = [{ name = "skyfire", email = "skyfireitdiy@hotmail.com" }]
@@ -41,6 +41,7 @@ dependencies = [
     "python-Levenshtein==0.25.1",
     "sseclient==0.0.27",
     "pillow==10.2.0",
+    "openai==1.78.1",
 ]
 requires-python = ">=3.8"
 

{jarvis_ai_assistant-0.1.164 → jarvis_ai_assistant-0.1.166}/setup.py

@@ -1,8 +1,9 @@
+# -*- coding: utf-8 -*-
 from setuptools import setup, find_packages
 
 setup(
     name="jarvis-ai-assistant",
-    version="0.1.164",
+    version="0.1.166",
     author="skyfire",
     author_email="skyfireitdiy@hotmail.com",
     description="An AI assistant that uses various tools to interact with the system",
@@ -12,6 +13,9 @@ setup(
     packages=find_packages(where="src"),
     package_dir={"": "src"},
     include_package_data=True,
+    package_data={
+        "jarvis": ["jarvis_data/huggingface.tar.gz"],
+    },
     install_requires=[
         "requests==2.32.3",
         "colorama==0.4.6",
@@ -28,6 +32,7 @@ setup(
         "python-Levenshtein==0.25.1",
         "sseclient==0.0.27",
         "pillow==10.2.0",
+        "openai==1.78.1",
     ],
     extras_require={
         "dev": ["pytest", "black", "isort", "mypy", "build", "twine"]

jarvis_ai_assistant-0.1.166/src/jarvis/__init__.py

@@ -0,0 +1,4 @@
+# -*- coding: utf-8 -*-
+"""Jarvis AI Assistant"""
+
+__version__ = "0.1.166"

{jarvis_ai_assistant-0.1.164 → jarvis_ai_assistant-0.1.166}/src/jarvis/jarvis_agent/__init__.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 import datetime
 import platform
 from typing import Any, Callable, List, Optional, Tuple, Union
@@ -309,6 +310,17 @@ class Agent:
         """
         self.after_tool_call_cb = cb
 
+    def get_tool_registry(self) -> Optional[ToolRegistry]:
+        """获取工具注册器。
+
+        返回:
+            ToolRegistry: 工具注册器实例
+        """
+        for handler in self.output_handler:
+            if isinstance(handler, ToolRegistry):
+                return handler
+        return None
+
     def make_default_addon_prompt(self, need_complete: bool) -> str:
         """生成附加提示。
 
@@ -657,7 +669,7 @@ arguments:
                 msg = user_input
                 for handler in self.input_handler:
                     msg, _ = handler(msg, self)
-                self.prompt = f"{user_input}\n\n以下是历史类似问题的执行经验，可参考：\n{load_methodology(msg)}"
+                self.prompt = f"{user_input}\n\n以下是历史类似问题的执行经验，可参考：\n{load_methodology(msg, self.get_tool_registry())}"
                 self.first = False
 
             while True:

{jarvis_ai_assistant-0.1.164 → jarvis_ai_assistant-0.1.166}/src/jarvis/jarvis_agent/builtin_input_handler.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 import re
 from typing import Any, Tuple
 from jarvis.jarvis_utils.config import get_replace_map

{jarvis_ai_assistant-0.1.164 → jarvis_ai_assistant-0.1.166}/src/jarvis/jarvis_agent/file_input_handler.py

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

{jarvis_ai_assistant-0.1.164 → jarvis_ai_assistant-0.1.166}/src/jarvis/jarvis_agent/jarvis.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 import argparse
 import os
 import sys

{jarvis_ai_assistant-0.1.164 → jarvis_ai_assistant-0.1.166}/src/jarvis/jarvis_agent/main.py

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

{jarvis_ai_assistant-0.1.164 → jarvis_ai_assistant-0.1.166}/src/jarvis/jarvis_agent/output_handler.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 from abc import ABC, abstractmethod
 from typing import Any, Tuple
 

{jarvis_ai_assistant-0.1.164 → jarvis_ai_assistant-0.1.166}/src/jarvis/jarvis_agent/shell_input_handler.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 from typing import Any, Tuple
 
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput

{jarvis_ai_assistant-0.1.164 → jarvis_ai_assistant-0.1.166}/src/jarvis/jarvis_code_agent/code_agent.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 """Jarvis代码代理模块。
 
 该模块提供CodeAgent类，用于处理代码修改任务。
@@ -86,32 +87,10 @@ class CodeAgent:
 - 代码阅读：优先使用read_code工具
 - 仅在命令行工具不足时使用专用工具
 
-## 代码编辑规范
-### 代码修改工具选择
-- 对于所有代码修改任务，优先使用edit_file工具，对于特别简单的修改，可以通过execute_script工具使用shell命令修改
-- edit_file通过搜索和替换实现精确的代码编辑，是首选的代码修改方式
-- 每个搜索文本在文件中必须唯一匹配，确保修改的准确性
-- 为新文件创建时，使用空字符串作为搜索文本，替换文本作为完整内容
-- 保持与原代码完全一致的格式和缩进风格
-
-### 文件重写工具使用
-- rewrite_file工具用于完全重写文件内容或创建新文件
-- 当需要替换整个文件内容时，优先使用rewrite_file工具
-- 创建新文件时，使用rewrite_file工具提供完整的文件内容
-- 保持与原代码相同的格式、缩进和编码风格
-- 确保提供格式良好的完整文件内容
-
-### 最小补丁原则
-- 对于任何代码修改，只提供需要修改的代码部分，不提供完整文件内容
-- 严格保持原始代码的缩进、空行和格式风格
-- 提供最小必要的上下文，确保修改位置精确
-- 每个修改必须包含清晰的修改理由
-
-### 格式兼容要求
-- 保持与原代码相同的缩进方式（空格或制表符）
-- 保持原代码的空行数量和位置
-- 不改变原代码的换行风格
-- 对新文件可提供完整内容，对现有文件只提供差异部分
+## 文件编辑工具使用规范
+- 对于部分文件内容修改，使用edit_file工具
+- 对于需要重写整个文件内容，使用rewrite_file工具
+- 对于简单的修改，可以使用execute_script工具执行shell命令完成
 </code_engineer_guide>
 """
         # Dynamically add ask_codebase based on task complexity if really needed

{jarvis_ai_assistant-0.1.164 → jarvis_ai_assistant-0.1.166}/src/jarvis/jarvis_code_analysis/checklists/__init__.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 """
 Language-specific checklists for code review.
 """ 

{jarvis_ai_assistant-0.1.164 → jarvis_ai_assistant-0.1.166}/src/jarvis/jarvis_code_analysis/checklists/c_cpp.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 """
 C/C++ language-specific code review checklist.
 """

{jarvis_ai_assistant-0.1.164 → jarvis_ai_assistant-0.1.166}/src/jarvis/jarvis_code_analysis/checklists/csharp.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 """
 C# language-specific code review checklist.
 """

{jarvis_ai_assistant-0.1.164 → jarvis_ai_assistant-0.1.166}/src/jarvis/jarvis_code_analysis/checklists/data_format.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 """
 Data format (XML, JSON, YAML) specific code review checklist.
 """

{jarvis_ai_assistant-0.1.164 → jarvis_ai_assistant-0.1.166}/src/jarvis/jarvis_code_analysis/checklists/devops.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 """
 DevOps (Makefile, CI/CD) specific code review checklist.
 """

{jarvis_ai_assistant-0.1.164 → jarvis_ai_assistant-0.1.166}/src/jarvis/jarvis_code_analysis/checklists/docs.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 """
 Documentation (Markdown, reStructuredText) specific code review checklist.
 """

{jarvis_ai_assistant-0.1.164 → jarvis_ai_assistant-0.1.166}/src/jarvis/jarvis_code_analysis/checklists/go.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 """
 Go language-specific code review checklist.
 """

{jarvis_ai_assistant-0.1.164 → jarvis_ai_assistant-0.1.166}/src/jarvis/jarvis_code_analysis/checklists/infrastructure.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 """
 Infrastructure as Code (Docker, Terraform) specific code review checklist.
 """

{jarvis_ai_assistant-0.1.164 → jarvis_ai_assistant-0.1.166}/src/jarvis/jarvis_code_analysis/checklists/java.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 """
 Java language-specific code review checklist.
 """

{jarvis_ai_assistant-0.1.164 → jarvis_ai_assistant-0.1.166}/src/jarvis/jarvis_code_analysis/checklists/javascript.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 """
 JavaScript/TypeScript language-specific code review checklist.
 """

{jarvis_ai_assistant-0.1.164 → jarvis_ai_assistant-0.1.166}/src/jarvis/jarvis_code_analysis/checklists/kotlin.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 """
 Kotlin-specific code review checklist.
 """

{jarvis_ai_assistant-0.1.164 → jarvis_ai_assistant-0.1.166}/src/jarvis/jarvis_code_analysis/checklists/loader.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 """
 Utility module for loading language-specific code review checklists.
 """

{jarvis_ai_assistant-0.1.164 → jarvis_ai_assistant-0.1.166}/src/jarvis/jarvis_code_analysis/checklists/php.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 """
 PHP language-specific code review checklist.
 """

{jarvis_ai_assistant-0.1.164 → jarvis_ai_assistant-0.1.166}/src/jarvis/jarvis_code_analysis/checklists/python.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 """
 Python language-specific code review checklist.
 """

{jarvis_ai_assistant-0.1.164 → jarvis_ai_assistant-0.1.166}/src/jarvis/jarvis_code_analysis/checklists/ruby.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 """
 Ruby-specific code review checklist.
 """

{jarvis_ai_assistant-0.1.164 → jarvis_ai_assistant-0.1.166}/src/jarvis/jarvis_code_analysis/checklists/rust.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 """
 Rust language-specific code review checklist.
 """

{jarvis_ai_assistant-0.1.164 → jarvis_ai_assistant-0.1.166}/src/jarvis/jarvis_code_analysis/checklists/shell.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 """
 Shell/Bash language-specific code review checklist.
 """

{jarvis_ai_assistant-0.1.164 → jarvis_ai_assistant-0.1.166}/src/jarvis/jarvis_code_analysis/checklists/sql.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 """
 SQL language-specific code review checklist.
 """

{jarvis_ai_assistant-0.1.164 → jarvis_ai_assistant-0.1.166}/src/jarvis/jarvis_code_analysis/checklists/swift.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 """
 Swift language-specific code review checklist.
 """

{jarvis_ai_assistant-0.1.164 → jarvis_ai_assistant-0.1.166}/src/jarvis/jarvis_code_analysis/checklists/web.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 """
 HTML/CSS specific code review checklist.
 """

{jarvis_ai_assistant-0.1.164 → jarvis_ai_assistant-0.1.166}/src/jarvis/jarvis_code_analysis/code_review.py

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

{jarvis_ai_assistant-0.1.164 → jarvis_ai_assistant-0.1.166}/src/jarvis/jarvis_dev/main.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 from jarvis.jarvis_platform.registry import PlatformRegistry
 from jarvis.jarvis_multi_agent import MultiAgent
 from jarvis.jarvis_tools.registry import ToolRegistry

{jarvis_ai_assistant-0.1.164 → jarvis_ai_assistant-0.1.166}/src/jarvis/jarvis_git_details/main.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 """Git Commit分析工具模块
 
 该模块提供了一个GitCommitAnalyzer类，用于获取和分析指定Git commit的详细信息，

{jarvis_ai_assistant-0.1.164 → jarvis_ai_assistant-0.1.166}/src/jarvis/jarvis_git_squash/main.py

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