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

{jarvis_ai_assistant-0.1.19/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.19
+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.19 → 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.19"
+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.19 → 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.19",
+    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.19 → jarvis_ai_assistant-0.1.23}/src/jarvis/__init__.py

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

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

@@ -11,7 +11,7 @@ import os
 from datetime import datetime
 
 class Agent:
-    def __init__(self, model: BaseModel, tool_registry: ToolRegistry, name: str = "Jarvis"):
+    def __init__(self, model: BaseModel, tool_registry: ToolRegistry, name: str = "Jarvis", is_sub_agent: bool = False):
         """Initialize Agent with a model, optional tool registry and name
         
         Args:
@@ -23,6 +23,7 @@ class Agent:
         self.model = model
         self.tool_registry = tool_registry or ToolRegistry(model)
         self.name = name
+        self.is_sub_agent = is_sub_agent
         self.prompt = ""
 
 
@@ -77,7 +78,7 @@ class Agent:
         except Exception as e:
             raise Exception(f"{self.name}: 模型调用失败: {str(e)}")
 
-    def run(self, user_input: str, file_list: Optional[List[str]] = None, keep_history: bool = False):
+    def run(self, user_input: str, file_list: Optional[List[str]] = None, keep_history: bool = False) -> str:
         """处理用户输入并返回响应，返回任务总结报告
         
         Args:
@@ -110,17 +111,20 @@ class Agent:
 核心能力：
 1. 使用现有工具完成任务
 2. 通过 generate_tool 创建新工具扩展功能
-3. 遵循 ReAct (思考-行动-观察) 框架
+3. 通过 create_sub_agent 创建子代理处理独立任务
+4. 遵循 ReAct (思考-行动-观察) 框架
 
 工作流程：
 1. 思考
    - 分析需求和可用工具
    - 评估是否需要新工具
+   - 考虑是否需要拆分子任务
    - 规划解决方案
 
 2. 行动 (如果需要)
    - 使用现有工具
    - 创建新工具
+   - 创建子代理
    - 询问更多信息
    
 3. 观察
@@ -128,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>
 
 创建新工具示例：
@@ -154,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. 等待用户提供执行结果
@@ -203,7 +228,23 @@ arguments:
                     
                     if not user_input:
                         PrettyOutput.section("任务完成", OutputType.SUCCESS)
-                        return 
+                        if self.is_sub_agent:
+                            # 生成任务总结
+                            summary_prompt = f"""请对以上任务执行情况生成一个简洁的总结报告，包括：
+
+1. 任务目标: xxxx
+2. 执行结果: 成功/失败
+3. 关键信息: 提取执行过程中的重要信息
+4. 重要发现: 任何值得注意的发现
+5. 后续建议: 如果有的话
+
+请用简洁的要点形式描述，突出重要信息。
+"""
+                            self.prompt = summary_prompt
+                            summary = self.model.chat(self.prompt)
+                            return summary
+                        else:
+                            return "Task completed"
 
                 except Exception as e:
                     PrettyOutput.print(str(e), OutputType.ERROR)

{jarvis_ai_assistant-0.1.19 → jarvis_ai_assistant-0.1.23}/src/jarvis/tools/base.py

@@ -115,7 +115,7 @@ class ToolRegistry:
                         parameters=tool_instance.parameters,
                         func=tool_instance.execute
                     )
-                    self.output_handler.print(f"已加载工具: {tool_instance.name}", OutputType.INFO)
+                    self.output_handler.print(f"已加载工具: {tool_instance.name}: {tool_instance.description}", OutputType.INFO)
                     tool_found = True
                     
             if not tool_found:

{jarvis_ai_assistant-0.1.19 → jarvis_ai_assistant-0.1.23}/src/jarvis/tools/generator.py

@@ -15,7 +15,7 @@ class ModelHandler(Protocol):
 
 class ToolGeneratorTool:
     name = "generate_tool"
-    description = "生成新的工具代码并自动注册到ToolRegistry"
+    description = "生成新的工具代码并自动注册到Jarvis，自动扩充Jarvis的能力"
     parameters = {
         "type": "object",
         "properties": {

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.19 → 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.19
+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.19 → 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