jarvis-ai-assistant 0.1.22__tar.gz → 0.1.23__tar.gz

{jarvis_ai_assistant-0.1.22/src/jarvis_ai_assistant.egg-info → jarvis_ai_assistant-0.1.23}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: jarvis-ai-assistant
-Version: 0.1.22
+Version: 0.1.23
 Summary: Jarvis: An AI assistant that uses tools to interact with the system
 Home-page: https://github.com/skyfireitdiy/Jarvis
 Author: skyfire

{jarvis_ai_assistant-0.1.22 → jarvis_ai_assistant-0.1.23}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "jarvis-ai-assistant"
-version = "0.1.22"
+version = "0.1.23"
 description = "Jarvis: An AI assistant that uses tools to interact with the system"
 readme = "README.md"
 authors = [{ name = "Your Name", email = "your.email@example.com" }]

{jarvis_ai_assistant-0.1.22 → jarvis_ai_assistant-0.1.23}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="jarvis-ai-assistant",
-    version="0.1.22",
+    version="0.1.23",
     author="skyfire",
     author_email="skyfireitdiy@hotmail.com",
     description="An AI assistant that uses various tools to interact with the system",

{jarvis_ai_assistant-0.1.22 → jarvis_ai_assistant-0.1.23}/src/jarvis/__init__.py

@@ -1,3 +1,3 @@
 """Jarvis AI Assistant"""
 
-__version__ = "0.1.22" 
+__version__ = "0.1.23" 

{jarvis_ai_assistant-0.1.22 → jarvis_ai_assistant-0.1.23}/src/jarvis/agent.py

@@ -111,17 +111,20 @@ class Agent:
 核心能力：
 1. 使用现有工具完成任务
 2. 通过 generate_tool 创建新工具扩展功能
-3. 遵循 ReAct (思考-行动-观察) 框架
+3. 通过 create_sub_agent 创建子代理处理独立任务
+4. 遵循 ReAct (思考-行动-观察) 框架
 
 工作流程：
 1. 思考
    - 分析需求和可用工具
    - 评估是否需要新工具
+   - 考虑是否需要拆分子任务
    - 规划解决方案
 
 2. 行动 (如果需要)
    - 使用现有工具
    - 创建新工具
+   - 创建子代理
    - 询问更多信息
    
 3. 观察
@@ -129,14 +132,25 @@ class Agent:
    - 分析反馈
    - 规划下一步
 
-工具使用格式：
+任务拆分建议：
+- 当任务包含多个独立步骤时
+- 当子任务需要独立的上下文时
+- 当子任务有明确的完成目标时
+- 当需要并行处理多个任务时
+
+创建子代理时，必须提供尽可能多的上下文信息，以确保其正确工作。
+
+创建子代理示例：
 <START_TOOL_CALL>
-name: tool_name
+name: create_sub_agent
 arguments:
-    param1: value1
-    param2: |
-        multiline
-        value
+    agent_name: CodeAnalyzer
+    task: 分析项目代码质量
+    context: |
+        这是一个Python项目
+        使用了Flask框架
+        需要关注性能和安全性
+    goal: 生成代码质量分析报告
 <END_TOOL_CALL>
 
 创建新工具示例：
@@ -155,6 +169,16 @@ arguments:
         required: [param1]
 <END_TOOL_CALL>
 
+工具使用格式：
+<START_TOOL_CALL>
+name: tool_name
+arguments:
+    param1: value1
+    param2: |
+        multiline
+        value
+<END_TOOL_CALL>
+
 严格规则：
 1. 每次只能执行一个工具
 2. 等待用户提供执行结果

jarvis_ai_assistant-0.1.23/src/jarvis/tools/sub_agent.py

@@ -0,0 +1,108 @@
+from typing import Dict, Any, Protocol, Optional
+from enum import Enum
+import os
+import sys
+from pathlib import Path
+
+# 添加项目根目录到 Python 路径
+project_root = Path(__file__).parent.parent.parent
+if str(project_root) not in sys.path:
+    sys.path.append(str(project_root))
+
+from jarvis.agent import Agent
+from jarvis.utils import OutputType
+
+class OutputHandler(Protocol):
+    def print(self, text: str, output_type: OutputType) -> None: ...
+
+class ModelHandler(Protocol):
+    def chat(self, message: str) -> str: ...
+
+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": ""
+            }
+        },
+        "required": ["agent_name", "task", "context", "goal"]
+    }
+
+    def __init__(self, **kwargs):
+        """初始化子代理工具
+        
+        Args:
+            model: 模型处理器
+            output_handler: 输出处理器
+            register: 工具注册器
+        """
+        self.model = kwargs.get('model')
+        if not self.model:
+            raise Exception("Model is required for SubAgentTool")
+        self.output = kwargs.get('output_handler')
+        self.register = kwargs.get('register')
+
+    def _print(self, text: str, output_type: OutputType = OutputType.INFO):
+        """输出信息"""
+        if self.output:
+            self.output.print(text, output_type)
+
+    def execute(self, args: Dict) -> Dict[str, Any]:
+        """创建并运行子代理"""
+        try:
+            agent_name = args["agent_name"]
+            task = args["task"]
+            context = args.get("context", "")
+            goal = args.get("goal", "")
+
+            self._print(f"创建子代理: {agent_name}")
+
+            # 构建任务描述
+            task_description = task
+            if context:
+                task_description = f"上下文信息:\n{context}\n\n任务:\n{task}"
+            if goal:
+                task_description += f"\n\n完成目标:\n{goal}"
+
+            # 创建子代理
+            sub_agent = Agent(
+                name=agent_name,
+                model=self.model,
+                tool_registry=self.register,
+                is_sub_agent=True
+            )
+
+            # 运行子代理
+            self._print(f"子代理开始执行任务...")
+            result = sub_agent.run(task_description)
+
+            return {
+                "success": True,
+                "stdout": f"子代理任务完成\n\n{result}",
+                "stderr": ""
+            }
+
+        except Exception as e:
+            self._print(str(e), OutputType.ERROR)
+            return {
+                "success": False,
+                "error": f"子代理执行失败: {str(e)}"
+            } 

{jarvis_ai_assistant-0.1.22 → jarvis_ai_assistant-0.1.23/src/jarvis_ai_assistant.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: jarvis-ai-assistant
-Version: 0.1.22
+Version: 0.1.23
 Summary: Jarvis: An AI assistant that uses tools to interact with the system
 Home-page: https://github.com/skyfireitdiy/Jarvis
 Author: skyfire

{jarvis_ai_assistant-0.1.22 → jarvis_ai_assistant-0.1.23}/src/jarvis_ai_assistant.egg-info/SOURCES.txt

@@ -25,6 +25,7 @@ src/jarvis/tools/base.py
 src/jarvis/tools/file_ops.py
 src/jarvis/tools/generator.py
 src/jarvis/tools/shell.py
+src/jarvis/tools/sub_agent.py
 src/jarvis/tools/__pycache__/__init__.cpython-313.pyc
 src/jarvis/tools/__pycache__/base.cpython-313.pyc
 src/jarvis/tools/__pycache__/bing_search.cpython-313.pyc