jarvis-ai-assistant 0.1.207__py3-none-any.whl → 0.1.209__py3-none-any.whl

jarvis/jarvis_tools/chdir.py

@@ -1,72 +0,0 @@
-# -*- coding: utf-8 -*-
-import os
-from typing import Any, Dict
-
-
-class ChdirTool:
-    name = "chdir"
-    description = "更改当前工作目录"
-    parameters = {
-        "type": "object",
-        "properties": {
-            "path": {
-                "type": "string",
-                "description": "要切换到的目录路径，支持相对路径和绝对路径",
-            }
-        },
-        "required": ["path"],
-    }
-
-    def execute(self, args: Dict[str, Any]) -> Dict[str, Any]:
-        """执行目录切换操作，并提供全面的错误处理。
-
-        参数:
-            args: 包含 'path' 键的字典，目标目录路径
-
-        返回:
-            字典，包含以下内容：
-                - success: 布尔值，表示操作状态
-                - stdout: 成功消息或空字符串
-                - stderr: 错误消息或空字符串
-
-        异常处理:
-            处理并返回适当的错误消息：
-                - 不存在的路径
-                - 非目录路径
-                - 权限错误
-                - 其他通用异常
-        """
-        # 主执行块，包含全面的错误处理
-        try:
-            # 规范化并展开输入路径（处理 ~ 和相对路径）
-            path = os.path.expanduser(args["path"].strip())
-            path = os.path.abspath(path)
-
-            # 验证目标路径是否存在
-            if not os.path.exists(path):
-                return {"success": False, "stdout": "", "stderr": f"目录不存在: {path}"}
-
-            # 确保路径指向的是目录，而不是文件
-            if not os.path.isdir(path):
-                return {
-                    "success": False,
-                    "stdout": "",
-                    "stderr": f"路径不是目录: {path}",
-                }
-
-            # 获取当前目录并尝试切换到新路径
-            old_path = os.getcwd()
-            os.chdir(path)
-
-            return {
-                "success": True,
-                "stdout": f"成功切换工作目录:\n原目录: {old_path}\n新目录: {path}",
-                "stderr": "",
-            }
-
-        # 处理用户没有目录访问权限的情况
-        except PermissionError:
-            return {"success": False, "stdout": "", "stderr": f"无权限访问目录: {path}"}
-        # 捕获在目录切换过程中可能出现的其他意外错误
-        except Exception as e:
-            return {"success": False, "stdout": "", "stderr": f"切换目录失败: {str(e)}"}

jarvis/jarvis_tools/code_plan.py

@@ -1,218 +0,0 @@
-# -*- coding: utf-8 -*-
-"""代码修改规划工具模块
-
-该模块提供CodePlanTool类，用于分析代码修改需求并制定详细的修改计划。
-包含以下主要功能：
-1. 需求理解与分析
-2. 代码修改计划制定
-3. 用户确认流程
-4. 修改计划输出
-"""
-
-import os
-from typing import Any, Dict
-
-from jarvis.jarvis_agent import Agent
-from jarvis.jarvis_platform.registry import PlatformRegistry
-from jarvis.jarvis_tools.registry import ToolRegistry
-from jarvis.jarvis_utils.git_utils import find_git_root
-from jarvis.jarvis_utils.output import OutputType, PrettyOutput
-
-
-class CodePlanTool:
-    """用于代码修改规划和需求分析的工具
-
-    适用场景：
-    - 理解复杂代码修改需求
-    - 制定详细的代码修改步骤
-    - 协调多文件修改计划
-    - 在修改前获取用户确认
-
-    工作流程：
-    1. 理解需求：通过多种方式(代码查询、用户交互、网络搜索)完全理解需求
-    2. 制定计划：分析代码后制定详细的修改步骤
-    3. 用户确认：将修改计划呈现给用户确认
-    4. 输出计划：按照标准格式输出最终修改计划
-    """
-
-    name = "code_plan"
-    description = "理解需求并制定详细的代码修改计划，在修改前获取用户确认"
-    parameters = {
-        "type": "object",
-        "properties": {
-            "requirement": {"type": "string", "description": "代码修改需求描述"},
-            "root_dir": {
-                "type": "string",
-                "description": "代码库根目录路径（可选）",
-                "default": ".",
-            },
-        },
-        "required": ["requirement"],
-    }
-
-    def execute(self, args: Dict[str, Any]) -> Dict[str, Any]:
-        """执行代码规划流程
-
-        Args:
-            args: 包含:
-                - requirement: 代码修改需求描述
-                - root_dir: 代码库根目录路径(可选)
-
-        Returns:
-            包含执行结果的字典:
-                - success: 是否成功
-                - stdout: 执行结果
-                - stderr: 错误信息
-        """
-        try:
-            requirement = args["requirement"]
-            root_dir = args.get("root_dir", ".")
-
-            # 存储当前目录
-            original_dir = os.getcwd()
-
-            try:
-                # 切换到root_dir
-                os.chdir(root_dir)
-
-                # 获取git根目录
-                git_root: str = find_git_root() or os.getcwd()
-
-                # 创建系统提示
-                system_prompt = self._create_system_prompt(requirement, git_root)
-
-                # 创建总结提示
-                summary_prompt = self._create_summary_prompt(requirement)
-
-                # 创建工具注册表
-                tool_registry: ToolRegistry = ToolRegistry()
-                tool_registry.use_tools(
-                    ["execute_script", "read_code", "search_web", "ask_user"]
-                )
-
-                # 创建并运行Agent
-                platform_registry: PlatformRegistry = PlatformRegistry()
-                planner_agent = Agent(
-                    system_prompt=system_prompt,
-                    name="CodePlanner",
-                    description="分析代码修改需求并制定详细计划",
-                    summary_prompt=summary_prompt,
-                    platform=platform_registry.get_normal_platform(),
-                    output_handler=[tool_registry],
-                    execute_tool_confirm=False,
-                    auto_complete=False,
-                )
-
-                # 运行agent并获取结果
-                task_input = f"分析并规划代码修改: {requirement}"
-                result = planner_agent.run(task_input)
-
-                return {"success": True, "stdout": result, "stderr": ""}
-            except (OSError, RuntimeError) as e:
-                error_msg = f"代码规划失败: {str(e)}"
-                PrettyOutput.print(error_msg, OutputType.WARNING)
-                return {"success": False, "stdout": "", "stderr": error_msg}
-            finally:
-                # 恢复原始目录
-                os.chdir(original_dir)
-        except (KeyError, ValueError) as e:
-            error_msg = f"代码规划失败: {str(e)}"
-            PrettyOutput.print(error_msg, OutputType.WARNING)
-            return {"success": False, "stdout": "", "stderr": error_msg}
-
-    def _create_system_prompt(self, requirement: str, git_root: str) -> str:
-        """创建Agent的system prompt"""
-        return f"""# 代码修改规划专家
-
-## 任务描述
-分析代码修改需求，理解需求后制定详细的代码修改计划，并在执行前获取用户确认。
-
-## 重要原则
-- **禁止直接修改代码**：仅提供修改计划，不执行实际代码修改
-- **只读分析**：所有代码分析必须基于现有代码，不得假设或虚构代码
-- **配置变更优先**：优先考虑通过配置而非代码修改实现需求
-
-## 需求信息
-- 需求: {requirement}
-- 代码库根目录: {git_root}
-
-## 工作流程
-1. **需求理解阶段**:
-   - 使用execute_script工具和read_code工具理解代码
-   - 必要时使用search_web搜索补充信息
-   - 使用ask_user工具向用户确认模糊点
-
-2. **代码分析阶段**:
-   - 使用fd/rg查找相关文件
-   - 使用read_code分析关键文件
-   - 确定需要修改的文件和范围
-   - **严格禁止修改代码**：仅分析不修改
-
-3. **计划制定阶段**:
-   - 制定最小变更方案
-   - 按功能模块分组修改
-   - 预估修改范围和影响
-   - **仅输出计划**：不执行实际修改
-
-4. **用户确认阶段**:
-   - 将完整修改计划呈现给用户
-   - 获取用户明确确认
-   - 根据反馈调整计划
-
-5. **计划输出阶段**:
-   - 按照summary_prompt格式输出最终计划
-   - 确保计划清晰、可执行
-   - **明确标注**：所有修改需用户手动执行
-
-## 工具使用优先级
-1. **代码查询工具**:
-   - fd/rg: 查找文件和代码模式
-   - read_code: 读取文件内容
-
-2. **信息补充工具**:
-   - search_web: 搜索技术实现方案
-   - ask_user: 确认需求细节
-
-## 计划制定原则
-- **只读原则**：所有分析必须基于现有代码，不得修改
-- 最小变更原则: 保持现有代码结构
-- 模块化修改: 按功能分组修改
-- 影响分析: 评估修改的影响范围
-- 风格一致: 保持代码风格统一"""
-
-    def _create_summary_prompt(self, requirement: str) -> str:
-        """创建Agent的summary prompt"""
-        return f"""# 代码修改计划报告
-
-## 报告要求
-生成关于以下需求的清晰、可执行的代码修改计划:
-
-**需求**: {requirement}
-
-报告应包含:
-
-1. **需求分析**:
-   - 需求的完整理解
-   - 关键功能点和边界条件
-   - 任何假设和前提条件
-
-2. **代码现状**:
-   - 相关文件和模块
-   - 现有实现分析
-   - 需要修改的部分
-
-3. **修改计划**:
-   - 需要修改的文件列表
-   - 每个文件的修改内容概述
-   - 修改步骤和顺序
-   - 预估影响范围
-
-4. **验证方案**:
-   - 如何验证修改的正确性
-   - 需要添加的测试用例
-
-5. **用户确认**:
-   - 明确标记已获得用户确认
-   - 记录用户的任何特殊要求
-
-使用清晰的Markdown格式，重点突出修改计划和验证方案。"""

jarvis/jarvis_tools/create_code_agent.py

@@ -1,95 +0,0 @@
-# -*- coding: utf-8 -*-
-import os
-from typing import Any, Dict
-
-from jarvis.jarvis_code_agent.code_agent import CodeAgent
-from jarvis.jarvis_git_utils.git_commiter import GitCommitTool
-from jarvis.jarvis_utils.git_utils import (get_latest_commit_hash,
-                                           has_uncommitted_changes)
-from jarvis.jarvis_utils.output import OutputType, PrettyOutput
-
-
-class CreateCodeAgentTool:
-    """用于管理代码开发工作流的工具"""
-
-    name = "create_code_agent"
-    description = "代码开发工具，当需要修改代码时使用，如果只是简单文件修改，使用文件操作或者脚本即可"
-    parameters = {
-        "requirement": """代码实现的技术规范，必须包含以下完整信息：
-1. 项目代码目录 - 项目根目录的绝对路径
-2. 项目功能 - 项目的主要功能和目标
-3. 本次要实现的feature - 本次开发要完成的具体功能需求
-4. 涉及的文件 - 需要修改或新增的文件列表
-5. 额外需要注意的信息 - 开发时需要额外注意的信息
-""",
-        "root_dir": {
-            "type": "string",
-            "description": "代码库根目录路径（可选）",
-            "default": ".",
-        },
-    }
-
-    def execute(self, args: Dict[str, Any]) -> Dict[str, Any]:
-        try:
-            requirement = args.get("requirement", "")
-            root_dir = args.get("root_dir", ".")
-
-            # Store current directory
-            original_dir = os.getcwd()
-
-            try:
-                # Change to root_dir
-                os.chdir(root_dir)
-                if not requirement:
-                    return {
-                        "success": False,
-                        "stderr": "Requirement must be provided",
-                        "stdout": "",
-                    }
-
-                # Step 1: Handle uncommitted changes
-                start_commit = None
-                if has_uncommitted_changes():
-                    PrettyOutput.print("发现未提交的更改，正在提交...", OutputType.INFO)
-                    git_commiter = GitCommitTool()
-                    result = git_commiter.execute({})
-                    if not result["success"]:
-                        return {
-                            "success": False,
-                            "stderr": "Failed to commit changes: " + result["stderr"],
-                            "stdout": "",
-                        }
-
-                # Get current commit hash
-                start_commit = get_latest_commit_hash()
-
-                # Step 2: Development
-                PrettyOutput.print("开始开发...", OutputType.INFO)
-                agent = CodeAgent()
-                agent.run(requirement)
-
-                # Get new commit hash after development
-                end_commit = get_latest_commit_hash()
-
-                # Step 4: Generate Summary
-                summary = f"""开发总结:
-
-开始提交: {start_commit}
-结束提交: {end_commit}
-
-需求:
-{requirement}
-
-"""
-
-                return {"success": True, "stdout": summary, "stderr": ""}
-            finally:
-                # Always restore original directory
-                os.chdir(original_dir)
-
-        except Exception as e:
-            return {
-                "success": False,
-                "stderr": f"Development workflow failed: {str(e)}",
-                "stdout": "",
-            }

jarvis/jarvis_tools/create_sub_agent.py

@@ -1,82 +0,0 @@
-# -*- coding: utf-8 -*-
-import os
-from typing import Any, Dict
-
-from jarvis.jarvis_agent import Agent, origin_agent_system_prompt
-from jarvis.jarvis_utils.output import OutputType, PrettyOutput
-
-
-class SubAgentTool:
-    name = "create_sub_agent"
-    description = "创建子代理以处理特定任务，子代理将生成任务总结报告"
-    parameters = {
-        "type": "object",
-        "properties": {
-            "agent_name": {"type": "string", "description": "子代理名称"},
-            "task": {"type": "string", "description": "要完成的特定任务"},
-            "context": {
-                "type": "string",
-                "description": "与任务相关的上下文信息",
-                "default": "",
-            },
-            "goal": {"type": "string", "description": "任务的完成目标", "default": ""},
-            "root_dir": {
-                "type": "string",
-                "description": "任务执行的根目录路径（可选）",
-                "default": ".",
-            },
-        },
-        "required": ["agent_name", "task"],
-    }
-
-    def execute(self, args: Dict) -> Dict[str, Any]:
-        """Create and run sub-agent"""
-        try:
-            agent_name = args["agent_name"]
-            task = args["task"]
-            context = args.get("context", "")
-            goal = args.get("goal", "")
-            root_dir = args.get("root_dir", ".")
-
-            PrettyOutput.print(f"创建子代理: {agent_name}", OutputType.INFO)
-
-            # Build task description
-            task_description = task
-            if context:
-                task_description = f"Context information:\n{context}\n\nTask:\n{task}"
-            if goal:
-                task_description += f"\n\nCompletion goal:\n{goal}"
-
-            # Store current directory
-            original_dir = os.getcwd()
-
-            try:
-                # Change to root_dir
-                os.chdir(root_dir)
-
-                # Create sub-agent
-                sub_agent = Agent(
-                    system_prompt=origin_agent_system_prompt,
-                    name=f"Agent({agent_name})",
-                )
-
-                # Run sub-agent, pass file list
-                PrettyOutput.print("子代理开始执行任务...", OutputType.INFO)
-                result = sub_agent.run(task_description)
-
-                return {
-                    "success": True,
-                    "stdout": f"Sub-agent task completed\n\n{result}",
-                    "stderr": "",
-                }
-            finally:
-                # Always restore original directory
-                os.chdir(original_dir)
-
-        except Exception as e:
-            PrettyOutput.print(str(e), OutputType.ERROR)
-            return {
-                "success": False,
-                "stdout": "",
-                "stderr": f"Sub-agent execution failed: {str(e)}",
-            }