jarvis-ai-assistant 0.1.163__py3-none-any.whl → 0.1.165__py3-none-any.whl

jarvis/__init__.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 """Jarvis AI Assistant"""
 
-__version__ = "0.1.163"
+__version__ = "0.1.165"

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

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/jarvis_agent/file_input_handler.py

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

jarvis/jarvis_agent/jarvis.py

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

jarvis/jarvis_agent/main.py

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

jarvis/jarvis_agent/output_handler.py

@@ -1,23 +1,48 @@
-
-
+# -*- coding: utf-8 -*-
 from abc import ABC, abstractmethod
 from typing import Any, Tuple
 
 
-
 class OutputHandler(ABC):
     @abstractmethod
     def handle(self, response: str, agent: Any) -> Tuple[bool, Any]:
+        """处理响应数据
+        
+        Args:
+            response: 需要处理的响应字符串
+            agent: 执行处理的agent实例
+        
+        Returns:
+            Tuple[bool, Any]: 返回处理结果元组，第一个元素表示是否处理成功，第二个元素为处理后的数据
+        """
         pass
 
     @abstractmethod
     def can_handle(self, response: str) -> bool:
+        """判断是否能处理给定的响应
+        
+        Args:
+            response: 需要判断的响应字符串
+        
+        Returns:
+            bool: 返回是否能处理该响应
+        """
         pass
 
     @abstractmethod
     def prompt(self) -> str:
+        """获取处理器的提示信息
+        
+        Returns:
+            str: 返回处理器的提示字符串
+        """
         pass
 
     @abstractmethod
     def name(self) -> str:
+        """获取处理器的名称
+        
+        Returns:
+            str: 返回处理器的名称字符串
+        """
         pass

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/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/jarvis_code_analysis/checklists/__init__.py

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

jarvis/jarvis_code_analysis/checklists/c_cpp.py

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

jarvis/jarvis_code_analysis/checklists/csharp.py

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

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/jarvis_code_analysis/checklists/devops.py

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

jarvis/jarvis_code_analysis/checklists/docs.py

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

jarvis/jarvis_code_analysis/checklists/go.py

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

jarvis/jarvis_code_analysis/checklists/infrastructure.py

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

jarvis/jarvis_code_analysis/checklists/java.py

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

jarvis/jarvis_code_analysis/checklists/javascript.py

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

jarvis/jarvis_code_analysis/checklists/kotlin.py

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

jarvis/jarvis_code_analysis/checklists/loader.py

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

jarvis/jarvis_code_analysis/checklists/php.py

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

jarvis/jarvis_code_analysis/checklists/python.py

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

jarvis/jarvis_code_analysis/checklists/ruby.py

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

jarvis/jarvis_code_analysis/checklists/rust.py

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

jarvis/jarvis_code_analysis/checklists/shell.py

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

jarvis/jarvis_code_analysis/checklists/sql.py

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

jarvis/jarvis_code_analysis/checklists/swift.py

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

jarvis/jarvis_code_analysis/checklists/web.py

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

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/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/jarvis_git_details/main.py

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

jarvis/jarvis_git_squash/main.py

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

jarvis/jarvis_git_utils/git_commiter.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 import re
 import shlex
 import subprocess
@@ -32,6 +33,16 @@ class GitCommitTool:
                 "type": "string",
                 "description": "Git仓库的根目录路径（可选）",
                 "default": "."
+            },
+            "prefix": {
+                "type": "string",
+                "description": "提交信息前缀（可选）",
+                "default": ""
+            },
+            "suffix": {
+                "type": "string",
+                "description": "提交信息后缀（可选）",
+                "default": ""
             }
         },
         "required": []
@@ -60,6 +71,8 @@ class GitCommitTool:
         """Execute automatic commit process with support for multi-line messages and special characters"""
         try:
             root_dir = args.get("root_dir", ".")
+            prefix = args.get("prefix", "")
+            suffix = args.get("suffix", "")
 
             # Store current directory
             original_dir = os.getcwd()
@@ -112,21 +125,21 @@ class GitCommitTool:
                         # 准备提示信息
                         base_prompt = f'''根据代码差异生成提交信息：
                         提交信息应使用{args.get('lang', '中文')}书写
-        # 必需结构
-        必须使用以下格式：
-        {ot("COMMIT_MESSAGE")}
-        <类型>(<范围>): <主题>
-        
-        [可选] 详细描述变更内容和原因
-        {ct("COMMIT_MESSAGE")}
-        # 格式规则
-        1. 类型: fix(修复bug), feat(新功能), docs(文档), style(格式), refactor(重构), test(测试), chore(其他)
-        2. 范围表示变更的模块或组件 (例如: auth, database, ui)
-        3. 主题行不超过72个字符，不以句号结尾，使用祈使语气
-        4. 如有详细描述，使用空行分隔主题和详细描述
-        5. 详细描述部分应解释"是什么"和"为什么"，而非"如何"
-        6. 仅输出提交信息，不要输出其他内容
-        '''
+    # 必需结构
+    必须使用以下格式：
+    {ot("COMMIT_MESSAGE")}
+    <类型>(<范围>): <主题>
+    
+    [可选] 详细描述变更内容和原因
+    {ct("COMMIT_MESSAGE")}
+    # 格式规则
+    1. 类型: fix(修复bug), feat(新功能), docs(文档), style(格式), refactor(重构), test(测试), chore(其他)
+    2. 范围表示变更的模块或组件 (例如: auth, database, ui)
+    3. 主题行不超过72个字符，不以句号结尾，使用祈使语气
+    4. 如有详细描述，使用空行分隔主题和详细描述
+    5. 详细描述部分应解释"是什么"和"为什么"，而非"如何"
+    6. 仅输出提交信息，不要输出其他内容
+    '''
 
                         # 获取模型并尝试上传文件
                         platform = PlatformRegistry().get_normal_platform()
@@ -159,12 +172,12 @@ class GitCommitTool:
                             spinner.text = "正在生成提交消息..."
                             # 使用上传的文件
                             prompt = base_prompt + f'''
-        # 变更概述
-        - 变更文件数量: {file_count} 个文件
-        - 已上传包含完整代码差异的文件
-        
-        请详细分析已上传的代码差异文件，生成符合上述格式的提交信息。
-        '''
+    # 变更概述
+    - 变更文件数量: {file_count} 个文件
+    - 已上传包含完整代码差异的文件
+    
+    请详细分析已上传的代码差异文件，生成符合上述格式的提交信息。
+    '''
                             commit_message = platform.chat_until_success(prompt)
                         else:
                             # 如果上传失败但内容较大，使用chat_big_content
@@ -174,9 +187,9 @@ class GitCommitTool:
                             else:
                                 # 直接在提示中包含差异内容
                                 prompt = base_prompt + f'''
-        # 分析材料
-        {diff}
-        '''
+    # 分析材料
+    {diff}
+    '''
                                 commit_message = platform.chat_until_success(prompt)
                         
                         while True:
@@ -187,13 +200,18 @@ class GitCommitTool:
                             # 如果成功提取，就跳出循环
                             if extracted_message:
                                 commit_message = extracted_message
+                                # 应用prefix和suffix
+                                if prefix:
+                                    commit_message = f"{prefix} {commit_message}"
+                                if suffix:
+                                    commit_message = f"{commit_message}\n{suffix}"
                                 break
                             prompt = f"""格式错误，请按照以下格式重新生成提交信息：
                             {ot("COMMIT_MESSAGE")}
-        <类型>(<范围>): <主题>
-        
-        [可选] 详细描述变更内容和原因
-        {ct("COMMIT_MESSAGE")}
+    <类型>(<范围>): <主题>
+    
+    [可选] 详细描述变更内容和原因
+    {ct("COMMIT_MESSAGE")}
                             """
                             commit_message = platform.chat_until_success(prompt)
                         spinner.write("✅ 生成提交消息")
@@ -250,9 +268,16 @@ def main():
     parser = argparse.ArgumentParser(description='Git commit tool')
     parser.add_argument('--lang', type=str, default='Chinese', help='Language for commit messages')
     parser.add_argument('--root-dir', type=str, default='.', help='Root directory of the Git repository')
+    parser.add_argument('--prefix', type=str, default='', help='Prefix to prepend to commit message (separated by space)')
+    parser.add_argument('--suffix', type=str, default='', help='Suffix to append to commit message (separated by newline)')
     args = parser.parse_args()
     tool = GitCommitTool()
-    tool.execute({"lang": args.lang if hasattr(args, 'lang') else 'Chinese', "root_dir": args.root_dir})
+    tool.execute({
+        "lang": args.lang if hasattr(args, 'lang') else 'Chinese',
+        "root_dir": args.root_dir,
+        "prefix": args.prefix if hasattr(args, 'prefix') else '',
+        "suffix": args.suffix if hasattr(args, 'suffix') else ''
+    })
 
 if __name__ == "__main__":
-    sys.exit(main())
+    sys.exit(main())

jarvis/jarvis_lsp/base.py

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

jarvis/jarvis_lsp/cpp.py

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

jarvis/jarvis_lsp/go.py

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

jarvis/jarvis_lsp/python.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 from typing import List, Dict, Any
 import jedi
 from jarvis.jarvis_lsp.base import BaseLSP

jarvis/jarvis_lsp/registry.py

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

jarvis/jarvis_lsp/rust.py

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

jarvis/jarvis_mcp/__init__.py

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

jarvis/jarvis_mcp/sse_mcp_client.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 from typing import Any, Dict, List, Callable
 import requests
 import json

jarvis/jarvis_mcp/stdio_mcp_client.py

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

jarvis/jarvis_methodology/main.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 """
 方法论导入导出命令行工具
 

jarvis/jarvis_multi_agent/__init__.py

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

jarvis/jarvis_multi_agent/main.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 import yaml
 from jarvis.jarvis_multi_agent import MultiAgent
 from jarvis.jarvis_utils.utils import init_env

jarvis/jarvis_platform/__init__.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 from jarvis.jarvis_platform.base import BasePlatform
 
 __all__ = ['BasePlatform']

jarvis/jarvis_platform/base.py

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

jarvis/jarvis_platform/human.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 from typing import Dict, List, Tuple
 import random
 import string

jarvis/jarvis_platform/kimi.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 from typing import Dict, List, Tuple
 import requests # type: ignore
 import json

jarvis/jarvis_platform/openai.py

@@ -0,0 +1,117 @@
+# -*- coding: utf-8 -*-
+from typing import Dict, List, Tuple
+import os
+from openai import OpenAI
+from jarvis.jarvis_platform.base import BasePlatform
+from jarvis.jarvis_utils.output import OutputType, PrettyOutput
+
+class OpenAIModel(BasePlatform):
+    platform_name = "openai"
+
+    def __init__(self):
+        """
+        Initialize OpenAI model
+        """
+        super().__init__()
+        self.system_message = ""
+        self.api_key = os.getenv("OPENAI_API_KEY")
+        if not self.api_key:
+            message = (
+                "需要设置以下环境变量才能使用 OpenAI 模型:\n"
+                "  • OPENAI_API_KEY: API 密钥\n"
+                "  • OPENAI_API_BASE: (可选) API 基础地址, 默认使用 https://api.openai.com/v1\n"
+                "您可以通过以下方式设置它们:\n"
+                "1. 创建或编辑 ~/.jarvis/env 文件:\n"
+                "   OPENAI_API_KEY=your_api_key\n"
+                "   OPENAI_API_BASE=your_api_base\n"
+                "   OPENAI_MODEL_NAME=your_model_name\n"
+                "2. 直接设置环境变量:\n"
+                "   export OPENAI_API_KEY=your_api_key\n"
+                "   export OPENAI_API_BASE=your_api_base\n"
+                "   export OPENAI_MODEL_NAME=your_model_name"
+            )
+            PrettyOutput.print(message, OutputType.INFO)
+            PrettyOutput.print("OPENAI_API_KEY 未设置", OutputType.WARNING)
+
+        self.base_url = os.getenv("OPENAI_API_BASE", "https://api.openai.com/v1")
+        self.model_name =  os.getenv("JARVIS_MODEL") or "gpt-4o"
+
+
+        self.client = OpenAI(
+            api_key=self.api_key,
+            base_url=self.base_url
+        )
+        self.messages: List[Dict[str, str]] = []
+        self.system_message = ""
+
+    def upload_files(self, file_list: List[str]) -> bool:
+        return False
+
+    def get_model_list(self) -> List[Tuple[str, str]]:
+        """Get model list"""
+        try:
+            models = self.client.models.list()
+            model_list = []
+            for model in models:
+                model_list.append((model.id, model.id))
+            return model_list
+        except Exception as e:
+            PrettyOutput.print(f"获取模型列表失败：{str(e)}", OutputType.ERROR)
+            return []
+
+    def set_model_name(self, model_name: str):
+        """Set model name"""
+
+        self.model_name = model_name
+
+    def set_system_message(self, message: str):
+        """Set system message"""
+        self.system_message = message
+        self.messages.append({"role": "system", "content": self.system_message})
+
+    def chat(self, message: str) -> str:
+        """Execute conversation"""
+        try:
+
+            # Add user message to history
+            self.messages.append({"role": "user", "content": message})
+
+            response = self.client.chat.completions.create(
+                model=self.model_name,  # Use the configured model name
+                messages=self.messages, # type: ignore
+                stream=True
+            ) # type: ignore
+
+            full_response = ""
+
+            for chunk in response:
+                if chunk.choices and chunk.choices[0].delta.content:
+                    text = chunk.choices[0].delta.content
+                    if not self.suppress_output:
+                        PrettyOutput.print_stream(text)
+                    full_response += text
+
+            if not self.suppress_output:
+                PrettyOutput.print_stream_end()
+
+            # Add assistant reply to history
+            self.messages.append({"role": "assistant", "content": full_response})
+
+            return full_response
+
+        except Exception as e:
+            PrettyOutput.print(f"对话失败：{str(e)}", OutputType.ERROR)
+            raise Exception(f"Chat failed: {str(e)}")
+
+    def name(self) -> str:
+        """Return model name"""
+        return self.model_name
+
+
+    def delete_chat(self)->bool:
+        """Delete conversation"""
+        if self.system_message:
+            self.messages = [{"role": "system", "content": self.system_message}]
+        else:
+            self.messages = []
+        return True

jarvis/jarvis_platform/registry.py

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