jarvis-ai-assistant 0.1.180__tar.gz → 0.1.181__tar.gz

{jarvis_ai_assistant-0.1.180/src/jarvis_ai_assistant.egg-info → jarvis_ai_assistant-0.1.181}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: jarvis-ai-assistant
-Version: 0.1.180
+Version: 0.1.181
 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: python-Levenshtein==0.25.1
 Requires-Dist: pillow==10.2.0
 Requires-Dist: openai==1.78.1
 Requires-Dist: tabulate==0.9.0
+Requires-Dist: pyte==0.8.2
 Provides-Extra: dev
 Requires-Dist: pytest; extra == "dev"
 Requires-Dist: black; extra == "dev"
@@ -212,8 +213,8 @@ OPENAI_API_BASE: https://api.openai.com/v1  # 可选，默认为官方API地址
 | `JARVIS_PRETTY_OUTPUT` | false | 是否启用PrettyOutput |
 | `JARVIS_GIT_COMMIT_PROMPT` | "" | 自定义git提交信息生成提示模板 |
 | `JARVIS_PRINT_PROMPT` | false | 是否打印提示 |
-| `JARVIS_USE_METHODOLOGY` | false | 是否启用方法论功能 |
-| `JARVIS_USE_ANALYSIS` | false | 是否启用任务分析功能 |
+| `JARVIS_USE_METHODOLOGY` | true | 是否启用方法论功能 |
+| `JARVIS_USE_ANALYSIS` | true | 是否启用任务分析功能 |
 | `JARVIS_DATA_PATH` | ~/.jarvis | Jarvis数据存储目录路径 |
 
 所有配置编写到`~/.jarvis/config.yaml`文件中即可生效。
@@ -232,7 +233,6 @@ ENV:
 ### 内置工具
 | 工具名称 | 描述 |
 |----------|------|
-| ask_codebase | 智能代码库查询和分析，用于定位功能所在文件和理解单点实现，适合查询特定功能位置和实现原理 |
 | ask_user | 交互式用户输入收集 |
 | chdir | 更改当前工作目录 |
 | rewrite_file | 文件重写工具，用于完全重写或创建文件，提供完整的文件内容替换 |
@@ -243,7 +243,6 @@ ENV:
 | execute_script | 执行脚本并返回结果，支持任意解释器。 |
 | file_analyzer | 分析文件内容并提取关键信息。支持的文件：文本文件、word文档、pdf文件、图片 |
 | file_operation | 文件批量操作工具，可批量读写多个文件，支持文本文件，适用于需要同时处理多个文件的场景（读取配置文件、保存生成内容等） |
-| find_methodology | 方法论查找工具，用于在执行过程中查看历史方法论辅助决策 |
 | methodology | 方法论管理工具，支持添加、更新和删除操作 |
 | read_code | 代码阅读与分析工具，用于读取源代码文件并添加行号，针对代码文件优化，提供更好的格式化输出和行号显示，适用于代码分析、审查和理解代码实现的场景 |
 | read_webpage | 读取网页内容并分析 |

{jarvis_ai_assistant-0.1.180 → jarvis_ai_assistant-0.1.181}/README.md

@@ -143,8 +143,8 @@ OPENAI_API_BASE: https://api.openai.com/v1  # 可选，默认为官方API地址
 | `JARVIS_PRETTY_OUTPUT` | false | 是否启用PrettyOutput |
 | `JARVIS_GIT_COMMIT_PROMPT` | "" | 自定义git提交信息生成提示模板 |
 | `JARVIS_PRINT_PROMPT` | false | 是否打印提示 |
-| `JARVIS_USE_METHODOLOGY` | false | 是否启用方法论功能 |
-| `JARVIS_USE_ANALYSIS` | false | 是否启用任务分析功能 |
+| `JARVIS_USE_METHODOLOGY` | true | 是否启用方法论功能 |
+| `JARVIS_USE_ANALYSIS` | true | 是否启用任务分析功能 |
 | `JARVIS_DATA_PATH` | ~/.jarvis | Jarvis数据存储目录路径 |
 
 所有配置编写到`~/.jarvis/config.yaml`文件中即可生效。
@@ -163,7 +163,6 @@ ENV:
 ### 内置工具
 | 工具名称 | 描述 |
 |----------|------|
-| ask_codebase | 智能代码库查询和分析，用于定位功能所在文件和理解单点实现，适合查询特定功能位置和实现原理 |
 | ask_user | 交互式用户输入收集 |
 | chdir | 更改当前工作目录 |
 | rewrite_file | 文件重写工具，用于完全重写或创建文件，提供完整的文件内容替换 |
@@ -174,7 +173,6 @@ ENV:
 | execute_script | 执行脚本并返回结果，支持任意解释器。 |
 | file_analyzer | 分析文件内容并提取关键信息。支持的文件：文本文件、word文档、pdf文件、图片 |
 | file_operation | 文件批量操作工具，可批量读写多个文件，支持文本文件，适用于需要同时处理多个文件的场景（读取配置文件、保存生成内容等） |
-| find_methodology | 方法论查找工具，用于在执行过程中查看历史方法论辅助决策 |
 | methodology | 方法论管理工具，支持添加、更新和删除操作 |
 | read_code | 代码阅读与分析工具，用于读取源代码文件并添加行号，针对代码文件优化，提供更好的格式化输出和行号显示，适用于代码分析、审查和理解代码实现的场景 |
 | read_webpage | 读取网页内容并分析 |

{jarvis_ai_assistant-0.1.180 → jarvis_ai_assistant-0.1.181}/pyproject.toml

@@ -8,7 +8,7 @@ default = true
 
 [project]
 name = "jarvis-ai-assistant"
-version = "0.1.180"
+version = "0.1.181"
 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 = [
     "pillow==10.2.0",
     "openai==1.78.1",
     "tabulate==0.9.0",
+    "pyte==0.8.2",
 ]
 requires-python = ">=3.8"
 
@@ -65,7 +66,5 @@ jarvis-git-squash = "jarvis.jarvis_git_squash.main:main"
 jarvis-multi-agent = "jarvis.jarvis_multi_agent.main:main"
 jarvis-agent = "jarvis.jarvis_agent.main:main"
 jarvis-tool = "jarvis.jarvis_tools.cli.main:main"
-jarvis-ask-codebase = "jarvis.jarvis_tools.ask_codebase:main"
-jac = "jarvis.jarvis_tools.ask_codebase:main"
 jarvis-git-details = "jarvis.jarvis_git_details.main:main"
 jarvis-methodology = "jarvis.jarvis_methodology.main:main"

{jarvis_ai_assistant-0.1.180 → jarvis_ai_assistant-0.1.181}/setup.py

@@ -3,7 +3,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="jarvis-ai-assistant",
-    version="0.1.180",
+    version="0.1.181",
     author="skyfire",
     author_email="skyfireitdiy@hotmail.com",
     description="An AI assistant that uses various tools to interact with the system",
@@ -35,6 +35,7 @@ setup(
         "pillow==10.2.0",
         "openai==1.78.1",
         "tabulate==0.9.0",
+        "pyte==0.8.2",
     ],
     extras_require={
         "dev": ["pytest", "black", "isort", "mypy", "build", "twine"]
@@ -55,8 +56,6 @@ setup(
             "jarvis-multi-agent=jarvis.jarvis_multi_agent.main:main",
             "jarvis-agent=jarvis.jarvis_agent.main:main",
             "jarvis-tool=jarvis.jarvis_tools.cli.main:main",
-            "jarvis-ask-codebase=jarvis.jarvis_tools.ask_codebase:main",
-            "jac=jarvis.jarvis_tools.ask_codebase:main",
             "jarvis-git-details=jarvis.jarvis_git_details.main:main",
             "jarvis-methodology=jarvis.jarvis_methodology.main:main",
         ],

{jarvis_ai_assistant-0.1.180 → jarvis_ai_assistant-0.1.181}/src/jarvis/__init__.py

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

{jarvis_ai_assistant-0.1.180 → jarvis_ai_assistant-0.1.181}/src/jarvis/jarvis_code_agent/code_agent.py

@@ -8,7 +8,7 @@ import argparse
 import os
 import subprocess
 import sys
-from typing import Any, Dict, List, Optional, Tuple
+from typing import List, Optional, Tuple
 
 from yaspin import yaspin  # type: ignore
 
@@ -24,12 +24,12 @@ from jarvis.jarvis_tools.registry import ToolRegistry
 from jarvis.jarvis_utils.config import is_confirm_before_apply_patch
 from jarvis.jarvis_utils.git_utils import (find_git_root, get_commits_between,
                                            get_diff, get_diff_file_list,
-                                           get_latest_commit_hash,
+                                           get_latest_commit_hash, get_recent_commits_with_files,
                                            handle_commit_workflow,
                                            has_uncommitted_changes)
 from jarvis.jarvis_utils.input import get_multiline_input
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput
-from jarvis.jarvis_utils.utils import init_env, user_confirm
+from jarvis.jarvis_utils.utils import get_loc_stats, init_env, user_confirm
 
 
 class CodeAgent:
@@ -47,11 +47,9 @@ class CodeAgent:
             "execute_script",
             "search_web",
             "ask_user",
-            "ask_codebase",
             "read_code",
             "methodology",
             "chdir",
-            "find_methodology",
             "edit_file",
             "rewrite_file"
         ])
@@ -75,6 +73,12 @@ class CodeAgent:
 1. **项目分析**：分析项目结构，确定需修改的文件
 2. **需求分析**：理解需求意图，选择影响最小的实现方案
 3. **代码分析**：详细分析目标文件，禁止虚构现有代码
+   - 结构分析：优先使用 fd 命令或 find 工具快速定位文件和目录结构
+   - 内容搜索：优先使用 rg（ripgrep）进行函数、类、变量等内容的全文搜索，避免遗漏
+   - 依赖关系：如需分析依赖、调用关系，可结合 grep、ctags、pyan3 等工具辅助
+   - 代码阅读：使用 read_code 工具获取目标文件的完整内容或指定范围内容，禁止凭空假设代码
+   - 变更影响：如需分析变更影响范围，可结合 git diff、git log 等命令辅助判断
+   - 工具优先级：优先使用自动化工具，减少人工推断，确保分析结果准确
 4. **方案设计**：确定最小变更方案，保持代码结构
 5. **实施修改**：遵循"先读后写"原则，保持代码风格一致性
 
@@ -90,7 +94,6 @@ class CodeAgent:
 - 对于简单的修改，可以使用execute_script工具执行shell命令完成
 </code_engineer_guide>
 """
-        # Dynamically add ask_codebase based on task complexity if really needed
         # 处理platform参数
         platform_instance = (PlatformRegistry().create_platform(platform)  # type: ignore
             if platform
@@ -110,14 +113,6 @@ class CodeAgent:
             ],
             need_summary=need_summary
         )
-        self.agent.set_addon_prompt(
-            "请使用工具充分理解用户需求，然后根据需求一步步执行代码修改/开发，"
-            "如果不清楚要修改那些文件，可以使用ask_codebase工具，"
-            "以：xxxx功能在哪个文件中实现？类似句式提问。"
-            "所有代码修改任务都应优先使用edit_file工具，而非edit_file工具。"
-            "edit_file工具通过精确的搜索和替换实现代码编辑，"
-            "搜索文本需在目标文件中有且仅有一次精确匹配，确保修改的准确性。"
-        )
 
         self.agent.set_after_tool_call_cb(self.after_tool_call_cb)
 
@@ -129,82 +124,8 @@ class CodeAgent:
         """
         return self.root_dir
 
-    def get_loc_stats(self) -> str:
-        """使用loc命令获取当前目录的代码统计信息
-        
-        返回:
-            str: loc命令输出的原始字符串，失败时返回空字符串
-        """
-        try:
-            result = subprocess.run(
-                ['loc'],
-                cwd=self.root_dir,
-                capture_output=True,
-                text=True
-            )
-            return result.stdout if result.returncode == 0 else ""
-        except FileNotFoundError:
-            return ""
-
-    def get_recent_commits_with_files(self) -> List[Dict[str, Any]]:
-        """获取最近5次提交的commit信息和文件清单
-        
-        返回:
-            List[Dict[str, Any]]: 包含commit信息和文件清单的字典列表，格式为:
-                [
-                    {
-                        'hash': 提交hash,
-                        'message': 提交信息,
-                        'author': 作者,
-                        'date': 提交日期,
-                        'files': [修改的文件列表] (最多20个文件)
-                    },
-                    ...
-                ]
-                失败时返回空列表
-        """
-        try:
-            # 获取最近5次提交的基本信息
-            result = subprocess.run(
-                ['git', 'log', '-5', '--pretty=format:%H%n%s%n%an%n%ad'],
-                cwd=self.root_dir,
-                capture_output=True,
-                text=True
-            )
-            if result.returncode != 0:
-                return []
-
-            # 解析提交信息
-            commits = []
-            lines = result.stdout.splitlines()
-            for i in range(0, len(lines), 4):
-                if i + 3 >= len(lines):
-                    break
-                commit = {
-                    'hash': lines[i],
-                    'message': lines[i+1],
-                    'author': lines[i+2],
-                    'date': lines[i+3],
-                    'files': []
-                }
-                commits.append(commit)
-
-            # 获取每个提交的文件修改清单
-            for commit in commits:
-                files_result = subprocess.run(
-                    ['git', 'show', '--name-only', '--pretty=format:', commit['hash']],
-                    cwd=self.root_dir,
-                    capture_output=True,
-                    text=True
-                )
-                if files_result.returncode == 0:
-                    files = list(set(filter(None, files_result.stdout.splitlines())))
-                    commit['files'] = files[:20]  # 限制最多20个文件
 
-            return commits
 
-        except subprocess.CalledProcessError:
-            return []
 
     def _init_env(self) -> None:
         """初始化环境，包括：
@@ -317,8 +238,8 @@ class CodeAgent:
             start_commit = get_latest_commit_hash()
 
             # 获取项目统计信息并附加到用户输入
-            loc_stats = self.get_loc_stats()
-            commits_info = self.get_recent_commits_with_files()
+            loc_stats = get_loc_stats()
+            commits_info = get_recent_commits_with_files()
             
             project_info = []
             if loc_stats:
@@ -331,8 +252,20 @@ class CodeAgent:
                     for i, commit in enumerate(commits_info)
                 )
                 project_info.append(f"最近提交:\n{commits_str}")
-            
-            enhanced_input = f"{user_input}\n\n项目概况:\n" + "\n\n".join(project_info) if project_info else user_input
+
+            first_tip = """请严格遵循以下规范进行代码修改任务：
+            1. 每次响应仅执行一步操作，先分析再修改，避免一步多改。
+            2. 充分利用工具理解用户需求和现有代码，禁止凭空假设。
+            3. 如果不清楚要修改的文件，必须先分析并找出需要修改的文件，明确目标后再进行编辑。
+            4. 代码编辑任务优先使用 edit_file 工具，确保搜索文本在目标文件中有且仅有一次精确匹配，保证修改的准确性和安全性。
+            5. 如需大范围重写，才可使用 rewrite_file 工具。
+            6. 如遇信息不明，优先调用工具补充分析，不要主观臆断。
+            """
+
+            if project_info:
+                enhanced_input = f"项目概况:\n" + "\n\n".join(project_info) + "\n\n" + first_tip + "\n\n任务描述：\n" + user_input
+            else:
+                enhanced_input = first_tip + "\n\n任务描述：\n" + user_input
 
             try:
                 self.agent.run(enhanced_input)
@@ -379,21 +312,16 @@ class CodeAgent:
                         for file in modified_files 
                         if get_lint_tools(file)
                     )
-                    addon_prompt = """
-1. 请对以下修改的文件进行静态扫描:
-""" + "\n".join(f"   - {file}" for file in modified_files) + (
-                        f"""
-2. 建议使用以下lint工具进行检查:
-{lint_tools_info}""" if lint_tools_info else ""
-                    ) + """
-3. 如果本次修改引入了警告和错误，请根据警告和错误信息修复代码
-4. 在引入的警告和错误都被修复的前提下，如果用户的需求未完成，请继续修改代码，如果已经完成，请终止，不要实现任何超出用户需求外的内容
-5. 如果有任何信息不明确，调用工具获取信息
-6. 每次响应必须且只能包含一个操作
+                    file_list = "\n".join(f"   - {file}" for file in modified_files)
+                    tool_info = f"建议使用以下lint工具进行检查:\n{lint_tools_info}" if lint_tools_info else ""
+                    if lint_tools_info:
+                        addon_prompt = f"""
+请对以下修改的文件进行静态扫描:
+{file_list}
+{tool_info}
+如果本次修改引入了警告和错误，请根据警告和错误信息修复代码
                     """
-
-                    agent.set_addon_prompt(addon_prompt)
-
+                        agent.set_addon_prompt(addon_prompt)
                 else:
                     final_ret += "✅ 补丁已应用（没有新的提交）"
             else:

{jarvis_ai_assistant-0.1.180 → jarvis_ai_assistant-0.1.181}/src/jarvis/jarvis_data/config_schema.json

@@ -112,23 +112,19 @@
       "default": ""
     },
     "JARVIS_MAX_TOKEN_COUNT": {
-      "type": "string",
+      "type": "number",
       "description": "模型能处理的最大token数量",
-      "default": "960000"
+      "default": 960000
     },
     "JARVIS_MAX_INPUT_TOKEN_COUNT": {
-      "type": "string",
+      "type": "number",
       "description": "模型能处理的最大输入token数量",
-      "default": "32000"
+      "default": 32000
     },
     "JARVIS_AUTO_COMPLETE": {
-      "type": "string",
+      "type": "boolean",
       "description": "是否启用自动补全功能",
-      "default": "false",
-      "enum": [
-        "true",
-        "false"
-      ]
+      "default": false
     },
     "SHELL": {
       "type": "string",
@@ -156,27 +152,19 @@
       "default": "deep_seek"
     },
     "JARVIS_EXECUTE_TOOL_CONFIRM": {
-      "type": "string",
+      "type": "boolean",
       "description": "执行工具前是否需要确认",
-      "default": "false",
-      "enum": [
-        "true",
-        "false"
-      ]
+      "default": false
     },
     "JARVIS_CONFIRM_BEFORE_APPLY_PATCH": {
-      "type": "string",
+      "type": "boolean",
       "description": "应用补丁前是否需要确认",
-      "default": "true",
-      "enum": [
-        "true",
-        "false"
-      ]
+      "default": true
     },
     "JARVIS_MAX_TOOL_CALL_COUNT": {
-      "type": "string",
+      "type": "number",
       "description": "最大连续工具调用次数",
-      "default": "20"
+      "default": 20
     },
     "JARVIS_DATA_PATH": {
       "type": "string",
@@ -184,54 +172,34 @@
       "default": "~/.jarvis"
     },
     "JARVIS_AUTO_UPDATE": {
-      "type": "string",
+      "type": "boolean",
       "description": "是否自动更新git仓库",
-      "default": "true",
-      "enum": [
-        "true",
-        "false"
-      ]
+      "default": true
     },
     "JARVIS_MAX_BIG_CONTENT_SIZE": {
-      "type": "string",
+      "type": "number",
       "description": "最大大内容尺寸",
-      "default": "1024000"
+      "default": 1024000
     },
     "JARVIS_PRETTY_OUTPUT": {
-      "type": "string",
+      "type": "boolean",
       "description": "是否启用美化输出",
-      "default": "false",
-      "enum": [
-        "true",
-        "false"
-      ]
+      "default": false
     },
     "JARVIS_USE_METHODOLOGY": {
-      "type": "string",
+      "type": "boolean",
       "description": "是否启用方法论",
-      "default": "false",
-      "enum": [
-        "true",
-        "false"
-      ]
+      "default": true
     },
     "JARVIS_USE_ANALYSIS": {
-      "type": "string",
+      "type": "boolean",
       "description": "是否启用任务分析",
-      "default": "false",
-      "enum": [
-        "true",
-        "false"
-      ]
+      "default": true
     },
     "JARVIS_PRINT_PROMPT": {
-      "type": "string",
+      "type": "boolean",
       "description": "是否打印提示",
-      "default": "false",
-      "enum": [
-        "true",
-        "false"
-      ]
+      "default": false
     },
     "JARVIS_REPLACE_MAP": {
       "type": "object",
@@ -252,9 +220,11 @@
             "description": "Description of the replacement"
           }
         },
-        "required": ["template"]
+        "required": [
+          "template"
+        ]
       }
     }
   },
   "additionalProperties": false
-}
+}