jarvis-ai-assistant 0.1.138__py3-none-any.whl → 0.1.141__py3-none-any.whl

jarvis/jarvis_dev/main.py

@@ -55,8 +55,7 @@ PM_PROMPT = f"""
 - **ask_user**：获取用户需求和反馈，澄清不明确的需求点
 - **file_operation**：创建和管理项目文档，跟踪项目状态
 - **search_web**：研究相关领域知识，寻找最佳实践
-- **rag**：访问项目知识库，参考历史经验
-- **execute_shell**：监控项目状态，执行自动化任务
+- **execute_script**：监控项目状态，执行自动化任务
 - **read_webpage**：收集行业信息和最新技术动态
 - **project_analyzer**：分析项目结构和架构，了解整体情况
 - **methodology**：采用适当的项目方法论和最佳实践
@@ -88,13 +87,13 @@ content: |
 
 ## 文档管理规范
 ### 项目文档结构
-- `/requirements/`：存放需求相关文档
+- `requirements/`：存放需求相关文档
   - `project_requirements_v<version>.md`：项目需求文档
   - `change_log.md`：需求变更记录
-- `/status_reports/`：存放状态报告
+- `status_reports/`：存放状态报告
   - `weekly_status_report.md`：周报
   - `risk_register.md`：风险登记册
-- `/communication/`：存放沟通记录
+- `communication/`：存放沟通记录
   - `team_communication_log.md`：团队沟通日志
   - `decision_log.md`：决策记录
 
@@ -384,8 +383,7 @@ def create_dev_team() -> MultiAgent:
         "ask_user",
         "file_operation",
         "search_web",
-        "rag",
-        "execute_shell",
+        "execute_script",
         "read_webpage",
         "project_analyzer",
         "methodology",
@@ -397,8 +395,7 @@ def create_dev_team() -> MultiAgent:
         "ask_user",
         "file_operation",
         "search_web",
-        "rag",
-        "execute_shell",
+        "execute_script",
         "read_webpage",
         "methodology",
         "ask_codebase"
@@ -408,12 +405,8 @@ def create_dev_team() -> MultiAgent:
     SA_output_handler.use_tools([
         "file_operation",
         "search_web",
-        "rag",
         "ask_codebase",
-        "execute_shell",
-        "project_analyzer",
-        "file_analyzer",
-        "function_analyzer",
+        "execute_script",
         "read_code",
         "methodology"
     ])
@@ -423,12 +416,9 @@ def create_dev_team() -> MultiAgent:
         "file_operation",
         "ask_codebase",
         "lsp_get_diagnostics",
-        "execute_shell",
-        "code_review",
-        "find_symbol",
-        "find_caller",
-        "function_analyzer",
-        "project_analyzer"
+        "execute_script",
+        "project_analyzer",
+        "methodology",
     ])
 
     DEV_output_handler = ToolRegistry()
@@ -436,12 +426,10 @@ def create_dev_team() -> MultiAgent:
         "create_code_agent",
         "file_operation",
         "ask_codebase",
-        "execute_shell",
-        "find_symbol",
-        "function_analyzer",
-        "file_analyzer",
+        "execute_script",
         "read_code",
-        "create_sub_agent"
+        "create_sub_agent",
+        "methodology",
     ])
 
     QA_output_handler = ToolRegistry()
@@ -449,11 +437,11 @@ def create_dev_team() -> MultiAgent:
         "create_code_agent",
         "file_operation",
         "ask_codebase",
-        "execute_shell",
+        "execute_script",
         "lsp_get_diagnostics",
-        "code_review",
-        "execute_shell_script",
+        "execute_script",
         "read_code",
+        "methodology",
     ])
 
     # Update PM prompt with tool usage guidance
@@ -462,8 +450,7 @@ def create_dev_team() -> MultiAgent:
 - **ask_user**：获取用户需求和反馈，澄清不明确的需求点
 - **file_operation**：创建和管理项目文档，跟踪项目状态
 - **search_web**：研究相关领域知识，寻找最佳实践
-- **rag**：访问项目知识库，参考历史经验
-- **execute_shell**：监控项目状态，执行自动化任务
+- **execute_script**：监控项目状态，执行自动化任务
 - **read_webpage**：收集行业信息和最新技术动态
 - **project_analyzer**：分析项目结构和架构，了解整体情况
 - **methodology**：采用适当的项目方法论和最佳实践
@@ -471,11 +458,11 @@ def create_dev_team() -> MultiAgent:
 
 ## 文档管理规范
 每一步工作后，必须使用file_operation工具将结论性输出记录到项目文档中：
-1. 需求确认后，创建`/requirements/project_requirements_v<version>.md`记录需求文档
-2. 任务分配后，创建`/status_reports/task_assignments.md`记录任务分配情况
-3. 项目阶段完成后，创建`/status_reports/project_status_report.md`记录项目进度
-4. 遇到的风险和问题，记录到`/status_reports/risk_register.md`
-5. 重要决策和变更，记录到`/communication/decision_log.md`
+1. 需求确认后，创建`requirements/project_requirements_v<version>.md`记录需求文档
+2. 任务分配后，创建`status_reports/task_assignments.md`记录任务分配情况
+3. 项目阶段完成后，创建`status_reports/project_status_report.md`记录项目进度
+4. 遇到的风险和问题，记录到`status_reports/risk_register.md`
+5. 重要决策和变更，记录到`communication/decision_log.md`
 
 文档命名需规范，内容需要结构化，使用Markdown格式，便于团队成员理解和跟进。
 """
@@ -486,20 +473,19 @@ def create_dev_team() -> MultiAgent:
 - **ask_user**：深入了解用户需求，进行需求挖掘和验证
 - **file_operation**：创建和管理需求文档与分析资料
 - **search_web**：研究行业标准和最佳实践
-- **rag**：访问项目知识库，参考相似需求历史
-- **execute_shell**：查询系统环境和配置信息
+- **execute_script**：查询系统环境和配置信息
 - **read_webpage**：收集用户体验和行业趋势信息
 - **methodology**：应用需求分析和用户故事映射方法论
 - **ask_codebase**：分析代码库中的功能实现，了解现有系统能力和限制，分析业务逻辑
 
 ## 文档管理规范
 每一步分析后，必须使用file_operation工具将结论性输出记录到分析文档中：
-1. 需求分析完成后，创建`/analysis/requirements_analysis_v<version>.md`记录分析结果
-2. 用户故事编写后，创建`/analysis/user_stories_v<version>.md`记录用户故事
-3. 功能规格确定后，创建`/specs/functional_specs.md`记录功能规格
-4. 数据需求确定后，创建`/specs/data_dictionary.md`记录数据字典
-5. 业务流程分析后，创建`/models/process_flows.md`记录业务流程
-6. 数据模型设计后，创建`/models/data_models.md`记录数据模型
+1. 需求分析完成后，创建`analysis/requirements_analysis_v<version>.md`记录分析结果
+2. 用户故事编写后，创建`analysis/user_stories_v<version>.md`记录用户故事
+3. 功能规格确定后，创建`specs/functional_specs.md`记录功能规格
+4. 数据需求确定后，创建`specs/data_dictionary.md`记录数据字典
+5. 业务流程分析后，创建`models/process_flows.md`记录业务流程
+6. 数据模型设计后，创建`models/data_models.md`记录数据模型
 
 文档需要结构化，使用Markdown格式，包含清晰的需求描述、优先级、验收标准和依赖关系。
 """
@@ -509,23 +495,19 @@ def create_dev_team() -> MultiAgent:
 ## 工具使用指南
 - **file_operation**：创建和管理架构文档和技术规格
 - **search_web**：研究架构模式和技术趋势
-- **rag**：访问架构知识库，参考历史设计决策
 - **ask_codebase**：分析代码库，理解系统实现
-- **execute_shell**：检查系统环境和依赖关系
-- **project_analyzer**：分析项目结构，理解模块划分
-- **file_analyzer**：深入分析关键文件的结构和功能
-- **function_analyzer**：分析核心函数的实现和设计
+- **execute_script**：检查系统环境和依赖关系
 - **read_code**：阅读和理解关键代码段
 - **methodology**：应用架构设计方法论和模式
 
 ## 文档管理规范
 每一步架构设计后，必须使用file_operation工具将结论性输出记录到架构文档中：
-1. 系统架构设计后，创建`/architecture/system_architecture_v<version>.md`记录架构文档
-2. 架构图创建后，保存到`/architecture/architecture_diagrams/`目录
-3. 组件规格定义后，创建`/technical_specs/component_specs/<component_name>.md`
-4. API规格设计后，创建`/technical_specs/api_specs/<api_name>.md`
-5. 架构决策后，创建`/decisions/adr_<number>_<decision_name>.md`记录架构决策
-6. 技术选型评估后，创建`/architecture/technology_evaluation.md`记录评估结果
+1. 系统架构设计后，创建`architecture/system_architecture_v<version>.md`记录架构文档
+2. 架构图创建后，保存到`architecture/architecture_diagrams/`目录
+3. 组件规格定义后，创建`technical_specs/component_specs/<component_name>.md`
+4. API规格设计后，创建`technical_specs/api_specs/<api_name>.md`
+5. 架构决策后，创建`decisions/adr_<number>_<decision_name>.md`记录架构决策
+6. 技术选型评估后，创建`architecture/technology_evaluation.md`记录评估结果
 
 文档需使用图表、表格等方式清晰展示架构设计，包含各组件职责、接口、性能考量及安全措施。
 """
@@ -536,22 +518,17 @@ def create_dev_team() -> MultiAgent:
 - **file_operation**：管理技术文档和指导文件
 - **ask_codebase**：分析代码库，理解实现细节
 - **lsp_get_diagnostics**：检查代码问题和警告
-- **execute_shell**：执行开发工具和命令
-- **code_review**：进行代码审查，确保代码质量
-- **find_symbol**：查找关键符号在代码中的使用
-- **find_caller**：分析函数调用关系和依赖
-- **function_analyzer**：深入分析函数实现和优化空间
-- **project_analyzer**：分析项目结构和技术架构
+- **execute_script**：执行开发工具和命令
 
 ## 文档管理规范
 每一步技术指导或审查后，必须使用file_operation工具将结论性输出记录到技术文档中：
-1. 实施计划制定后，创建`/technical/implementation_plan_v<version>.md`记录实施计划
-2. 任务分解后，创建`/technical/task_breakdown.md`记录任务分解详情
-3. 编码标准制定后，创建`/guidelines/coding_standards.md`记录编码标准
-4. 审查指南制定后，创建`/guidelines/review_guidelines.md`记录审查指南
-5. 代码审查后，创建`/quality/code_review_<date>.md`记录审查结果
-6. 技术债务识别后，更新`/quality/technical_debt.md`记录技术债务
-7. 性能优化后，创建`/quality/performance_metrics.md`记录性能指标
+1. 实施计划制定后，创建`technical/implementation_plan_v<version>.md`记录实施计划
+2. 任务分解后，创建`technical/task_breakdown.md`记录任务分解详情
+3. 编码标准制定后，创建`guidelines/coding_standards.md`记录编码标准
+4. 审查指南制定后，创建`guidelines/review_guidelines.md`记录审查指南
+5. 代码审查后，创建`quality/code_review_<date>.md`记录审查结果
+6. 技术债务识别后，更新`quality/technical_debt.md`记录技术债务
+7. 性能优化后，创建`quality/performance_metrics.md`记录性能指标
 
 文档需包含清晰的技术指导、代码质量标准、任务分解和时间估计，便于开发团队执行。
 """
@@ -562,22 +539,19 @@ def create_dev_team() -> MultiAgent:
 - **create_code_agent**：创建专业代码开发代理
 - **file_operation**：管理源代码和配置文件
 - **ask_codebase**：了解代码库实现细节
-- **execute_shell**：执行开发命令和测试脚本
-- **find_symbol**：查找关键符号在代码中的使用
-- **function_analyzer**：分析函数实现和优化空间
-- **file_analyzer**：分析文件结构和功能
+- **execute_script**：执行开发命令和测试脚本
 - **read_code**：阅读和理解关键代码段
 - **create_sub_agent**：创建专门的子代理处理特定任务
 
 ## 文档管理规范
 每一步代码实现或优化后，必须使用file_operation工具将结论性输出记录到开发文档中：
-1. 功能实现完成后，创建`/src/README.md`或更新模块说明文档，记录实现细节
-2. API实现后，更新`/docs/api/<module_name>.md`记录API使用说明
-3. 复杂算法实现后，创建`/docs/algorithms/<algorithm_name>.md`解释算法原理
-4. 配置变更后，更新`/docs/configuration.md`记录配置项变更
-5. 依赖更新后，更新`/docs/dependencies.md`记录依赖关系
-6. 开发过程中遇到的问题和解决方案，记录到`/docs/troubleshooting.md`
-7. 单元测试完成后，创建`/tests/README.md`记录测试覆盖情况
+1. 功能实现完成后，创建`src/README.md`或更新模块说明文档，记录实现细节
+2. API实现后，更新`docs/api/<module_name>.md`记录API使用说明
+3. 复杂算法实现后，创建`docs/algorithms/<algorithm_name>.md`解释算法原理
+4. 配置变更后，更新`docs/configuration.md`记录配置项变更
+5. 依赖更新后，更新`docs/dependencies.md`记录依赖关系
+6. 开发过程中遇到的问题和解决方案，记录到`docs/troubleshooting.md`
+7. 单元测试完成后，创建`tests/README.md`记录测试覆盖情况
 
 文档需要包含功能描述、使用示例、参数说明和注意事项，便于其他开发者理解和使用。
 """
@@ -588,21 +562,20 @@ def create_dev_team() -> MultiAgent:
 - **create_code_agent**：创建测试代码开发代理
 - **file_operation**：管理测试文档和测试脚本
 - **ask_codebase**：了解代码库实现以设计测试
-- **execute_shell**：执行测试命令和测试套件
+- **execute_script**：执行测试命令和测试套件
 - **lsp_get_diagnostics**：检查代码问题和警告
-- **code_review**：从质量保证角度审查代码
-- **execute_shell_script**：执行自动化测试脚本
+- **execute_script**：执行各类脚本（Shell命令、Shell脚本、Python脚本）
 - **read_code**：阅读和理解代码以设计测试用例
 
 ## 文档管理规范
 每一步测试或质量评估后，必须使用file_operation工具将结论性输出记录到测试文档中：
-1. 测试计划制定后，创建`/testing/test_plan.md`记录测试计划
-2. 测试用例设计后，创建`/testing/test_cases/<feature_name>_test_cases.md`记录测试用例
-3. 测试执行后，创建`/testing/test_reports/test_report_<date>.md`记录测试报告
-4. 发现缺陷后，更新`/defects/defect_log.md`记录缺陷详情
-5. 自动化测试脚本开发后，在`/automation/README.md`中记录脚本使用说明
-6. 性能测试结果，记录到`/testing/performance_test_results.md`
-7. 缺陷统计和趋势分析，记录到`/defects/defect_metrics.md`
+1. 测试计划制定后，创建`testing/test_plan.md`记录测试计划
+2. 测试用例设计后，创建`testing/test_cases/<feature_name>_test_cases.md`记录测试用例
+3. 测试执行后，创建`testing/test_reports/test_report_<date>.md`记录测试报告
+4. 发现缺陷后，更新`defects/defect_log.md`记录缺陷详情
+5. 自动化测试脚本开发后，在`automation/README.md`中记录脚本使用说明
+6. 性能测试结果，记录到`testing/performance_test_results.md`
+7. 缺陷统计和趋势分析，记录到`defects/defect_metrics.md`
 
 测试文档需包含测试范围、测试环境、测试用例、预期结果、实际结果和缺陷级别，便于跟踪和修复。
 """

jarvis/jarvis_git_squash/main.py

@@ -1,8 +1,7 @@
-import re
 import sys
 import argparse
-from typing import Dict, Any
-from jarvis.jarvis_tools.git_commiter import GitCommitTool
+from typing import Dict
+from jarvis.jarvis_git_utils.git_commiter import GitCommitTool
 import subprocess
 
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput
@@ -10,6 +9,7 @@ from jarvis.jarvis_utils.utils import init_env, user_confirm
 class GitSquashTool:
     name = "git_squash_agent"
     description = "Squash commits interactively using a base commit hash"
+    labels = ['git', 'version_control', 'squash']
 
     def _confirm_squash(self) -> bool:
         """Prompt user for confirmation to squash commits"""

jarvis/jarvis_git_utils/git_commiter.py

@@ -0,0 +1,242 @@
+import re
+import shlex
+import subprocess
+from typing import Dict, Any, Optional
+import tempfile
+from click import Option
+import yaml
+from yaspin import yaspin
+from jarvis.jarvis_platform.registry import PlatformRegistry
+import sys
+import argparse
+import os
+
+from jarvis.jarvis_utils.config import get_max_input_token_count
+from jarvis.jarvis_utils.embedding import get_context_token_count
+from jarvis.jarvis_utils.git_utils import find_git_root, has_uncommitted_changes
+from jarvis.jarvis_utils.output import OutputType, PrettyOutput
+from jarvis.jarvis_utils.utils import ct, ot, init_env
+
+
+class GitCommitTool:
+    name = "git_commit_agent"
+    description = "根据代码变更自动生成并执行Git提交"
+    labels = ['git', 'version_control']
+    parameters = {
+        "type": "object",
+        "properties": {
+            "lang": {
+                "type": "string",
+                "description": "提交信息的语言",
+                "default": "Chinese"
+            },
+            "root_dir": {
+                "type": "string",
+                "description": "Git仓库的根目录路径（可选）",
+                "default": "."
+            }
+        },
+        "required": []
+    }
+    def _extract_commit_message(self, message)->Optional[str]:
+        """Raw extraction preserving all characters"""
+        r = re.search(
+            r"(?i)" + ot("COMMIT_MESSAGE") + r"\s*([\s\S]*?)\s*" + ct("COMMIT_MESSAGE"),
+            message
+        )
+        if r:
+            # 直接返回原始内容，仅去除外围空白
+            return shlex.quote(r.group(1).strip())
+        return None
+
+    def _get_last_commit_hash(self):
+        process = subprocess.Popen(
+            ["git", "log", "-1", "--pretty=%H"],
+            stdout=subprocess.PIPE,
+            stderr=subprocess.PIPE
+        )
+        stdout, _ = process.communicate()
+        return stdout.decode().strip()
+
+    def execute(self, args: Dict) -> Dict[str, Any]:
+        """Execute automatic commit process with support for multi-line messages and special characters"""
+        try:
+            root_dir = args.get("root_dir", ".")
+
+            # Store current directory
+            original_dir = os.getcwd()
+
+            try:
+                # Change to root_dir
+                os.chdir(root_dir)
+
+                find_git_root()
+                if not has_uncommitted_changes():
+                    PrettyOutput.print("没有未提交的更改", OutputType.SUCCESS)
+                    return {"success": True, "stdout": "No changes to commit", "stderr": ""}
+
+                with yaspin(text="正在初始化提交流程...", color="cyan") as spinner:
+                    # 添加文件
+                    spinner.text = "正在添加文件到提交..."
+                    subprocess.Popen(
+                        ["git", "add", "."],
+                        stdout=subprocess.DEVNULL,
+                        stderr=subprocess.DEVNULL
+                    ).wait()
+                    spinner.write("✅ 添加文件到提交")
+
+                    # 获取差异
+                    spinner.text = "正在获取代码差异..."
+                    # 获取文件列表
+                    files_cmd = ["git", "diff", "--cached", "--name-only"]
+                    process = subprocess.Popen(
+                        files_cmd,
+                        stdout=subprocess.PIPE,
+                        stderr=subprocess.PIPE
+                    )
+                    files_output = process.communicate()[0].decode()
+                    files = [f.strip() for f in files_output.split("\n") if f.strip()]
+                    file_count = len(files)
+                    
+                    # 获取完整差异
+                    process = subprocess.Popen(
+                        ["git", "diff", "--cached", "--exit-code"],
+                        stdout=subprocess.PIPE,
+                        stderr=subprocess.PIPE
+                    )
+                    diff = process.communicate()[0].decode()
+                    spinner.write(f"✅ 获取差异 ({file_count} 个文件)")
+
+                    # 写入差异文件
+                    temp_diff_file = None
+                    try:
+                        # 创建临时文件并写入差异内容
+                        temp_diff_file = tempfile.NamedTemporaryFile(mode='w', suffix='.diff', delete=False)
+                        temp_diff_file_path = temp_diff_file.name
+                        temp_diff_file.write(diff)
+                        temp_diff_file.flush()
+                        temp_diff_file.close()
+                        spinner.write(f"✅ 差异内容已写入临时文件")
+                        # 生成提交信息
+                        spinner.text = "正在生成提交消息..."
+                        
+                        # 准备提示信息
+                        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. 仅输出提交信息，不要输出其他内容
+        '''
+
+                        # 获取模型并尝试上传文件
+                        platform = PlatformRegistry().get_normal_platform()
+                        upload_success = False
+                        
+                        if hasattr(platform, 'upload_files') and os.path.exists(temp_diff_file_path):
+                            spinner.text = "正在上传代码差异文件..."
+                            try:
+                                with spinner.hidden():
+                                    upload_success = platform.upload_files([temp_diff_file_path])
+                                if upload_success:
+                                    spinner.write("✅ 成功上传代码差异文件")
+                                else:
+                                    spinner.write("⚠️ 上传代码差异文件失败，将回退到直接提供差异内容")
+                            except Exception as e:
+                                spinner.write(f"⚠️ 上传文件时出错: {str(e)}")
+                                upload_success = False
+                        
+                        # 根据上传状态准备完整的提示
+                        if upload_success:
+                            # 使用上传的文件
+                            prompt = base_prompt + f'''
+        # 变更概述
+        - 变更文件数量: {file_count} 个文件
+        - 已上传包含完整代码差异的文件
+        
+        请详细分析已上传的代码差异文件，生成符合上述格式的提交信息。
+        '''
+                        else:
+                            # 直接在提示中包含差异内容
+                            prompt = base_prompt + f'''
+        # 分析材料
+        {diff}
+        '''
+                        
+                        # 尝试生成提交信息
+                        spinner.text = "正在生成提交消息..."
+                        while True:
+                            commit_message = platform.chat_until_success(prompt)
+                            commit_message = self._extract_commit_message(commit_message)
+                            # 如果成功提取，就跳出循环
+                            if commit_message:
+                                break
+                        spinner.write("✅ 生成提交消息")
+
+                        # 执行提交
+                        spinner.text = "正在准备提交..."
+                        with tempfile.NamedTemporaryFile(mode='w', delete=True) as tmp_file:
+                            tmp_file.write(commit_message)
+                            tmp_file.flush()
+                            spinner.text = "正在执行提交..."
+                            commit_cmd = ["git", "commit", "-F", tmp_file.name]
+                            subprocess.Popen(
+                                commit_cmd,
+                                stdout=subprocess.DEVNULL,
+                                stderr=subprocess.DEVNULL
+                            ).wait()
+                            spinner.write("✅ 提交")
+
+                        commit_hash = self._get_last_commit_hash()
+                        spinner.text = "完成提交"
+                        spinner.ok("✅")
+                    finally:
+                        # 清理临时差异文件
+                        if temp_diff_file is not None and os.path.exists(temp_diff_file_path):
+                            try:
+                                os.unlink(temp_diff_file_path)
+                            except Exception as e:
+                                spinner.write(f"⚠️ 无法删除临时文件: {str(e)}")
+
+                PrettyOutput.print(f"提交哈希: {commit_hash}\n提交消息: {commit_message}", OutputType.SUCCESS)
+
+                return {
+                    "success": True,
+                    "stdout": yaml.safe_dump({
+                        "commit_hash": commit_hash,
+                        "commit_message": commit_message
+                    }),
+                    "stderr": ""
+                    }
+            finally:
+                # Always restore original directory
+                os.chdir(original_dir)
+
+        except Exception as e:
+            return {
+                "success": False,
+                "stdout": "",
+                "stderr": f"Commit failed: {str(e)}"
+            }
+
+def main():
+    init_env()
+    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')
+    args = parser.parse_args()
+    tool = GitCommitTool()
+    tool.execute({"lang": args.lang if hasattr(args, 'lang') else 'Chinese', "root_dir": args.root_dir})
+
+if __name__ == "__main__":
+    sys.exit(main())

jarvis/jarvis_init/main.py

@@ -0,0 +1,62 @@
+from typing import Any, Tuple
+import subprocess
+from jarvis.jarvis_utils.embedding import (
+    get_context_token_count,
+    load_tokenizer
+)
+from jarvis.jarvis_utils.output import PrettyOutput, OutputType
+
+def init_models() -> Tuple[Any, Any, Any, Any]:
+    """
+    初始化所需的模型和分词器。
+    
+    返回：
+        Tuple[Any, Any, Any, Any]: (embedding_model, rerank_model, rerank_tokenizer, tokenizer)
+    """
+    try:
+        
+        # 加载GPT2分词器
+        PrettyOutput.print("正在加载GPT2分词器...", OutputType.INFO)
+        tokenizer = load_tokenizer()
+        
+        # 测试token计算
+        test_text = "这是一个测试文本，用于验证token计算功能。"
+        token_count = get_context_token_count(test_text)
+        PrettyOutput.print(f"Token计算测试成功，测试文本token数: {token_count}", OutputType.SUCCESS)
+        
+        return embedding_model, rerank_model, rerank_tokenizer, tokenizer
+        
+    except Exception as e:
+        PrettyOutput.print(f"模型加载失败: {str(e)}", OutputType.ERROR)
+        raise
+
+def install_playwright():
+    """安装Playwright及其依赖。"""
+    try:
+        PrettyOutput.print("正在安装Playwright...", OutputType.INFO)
+        subprocess.run(["playwright", "install", "chromium"], check=True)
+        PrettyOutput.print("Playwright安装成功", OutputType.SUCCESS)
+    except subprocess.CalledProcessError as e:
+        PrettyOutput.print(f"Playwright安装失败: {str(e)}", OutputType.ERROR)
+        raise
+    except Exception as e:
+        PrettyOutput.print(f"安装过程中发生错误: {str(e)}", OutputType.ERROR)
+        raise
+
+def main():
+    """主函数，初始化所有必要的组件。"""
+    try:
+        # 初始化模型
+        init_models()
+        
+        # 安装Playwright
+        install_playwright()
+        
+        PrettyOutput.print("所有组件初始化完成", OutputType.SUCCESS)
+        
+    except Exception as e:
+        PrettyOutput.print(f"初始化失败: {str(e)}", OutputType.ERROR)
+        raise
+
+if __name__ == "__main__":
+    main()

jarvis/jarvis_platform/base.py

@@ -31,6 +31,10 @@ class BasePlatform(ABC):
     def chat(self, message: str) -> str:
         """Execute conversation"""
         raise NotImplementedError("chat is not implemented")
+    
+    @abstractmethod
+    def upload_files(self, file_list: List[str]) -> bool:
+        raise NotImplementedError("upload_files is not implemented")
 
     def chat_until_success(self, message: str) -> str:
         def _chat():