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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: jarvis-ai-assistant
-Version: 0.1.18
+Version: 0.1.22
 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.18 → jarvis_ai_assistant-0.1.22}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "jarvis-ai-assistant"
-version = "0.1.18"
+version = "0.1.22"
 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.18 → jarvis_ai_assistant-0.1.22}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="jarvis-ai-assistant",
-    version="0.1.18",
+    version="0.1.22",
     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.18 → jarvis_ai_assistant-0.1.22}/src/jarvis/__init__.py

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

{jarvis_ai_assistant-0.1.18 → jarvis_ai_assistant-0.1.22}/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:
@@ -103,55 +104,32 @@ class Agent:
                 tools_prompt += f"  描述: {tool['description']}\n"
                 tools_prompt += f"  参数: {tool['parameters']}\n"
 
-            self.prompt = f"""你是 {self.name}，一个严格遵循 ReAct 框架进行逐步推理和行动的 AI 助手。
+            self.prompt = f"""你是 {self.name}，一个严格遵循 ReAct 框架的 AI 助手。
 
 {tools_prompt}
 
-关键规则：
-‼️ 禁止创建虚假对话
-‼️ 禁止假设用户回应
-‼️ 禁止在没有实际用户输入时继续
-‼️ 只回应用户实际说的内容
-‼️ 每个动作后停止并等待
+核心能力：
+1. 使用现有工具完成任务
+2. 通过 generate_tool 创建新工具扩展功能
+3. 遵循 ReAct (思考-行动-观察) 框架
 
-ReAct 框架：
+工作流程：
 1. 思考
-   - 分析当前情况
-   - 考虑可用工具
-   - 规划下一步行动
-   - 仅基于事实
-   - 不做用户回应的假设
-   - 不想象对话内容
-
-2. 行动（可选）
-   - 不是每次回应都需要调用工具
-   - 如果需要更多信息，直接询问用户
-   - 使用工具时：
-     - 只使用下面列出的工具
-     - 每次只执行一个工具
-     - 工具由用户手动执行
-     - 必须使用有效合法的YAML格式：
-     <START_TOOL_CALL>
-     name: tool_name
-     arguments:
-         param1: value1       # 所有参数必须正确缩进
-         param2: |           # 使用YAML块样式表示多行字符串
-             line1
-             line2
-     <END_TOOL_CALL>
-
+   - 分析需求和可用工具
+   - 评估是否需要新工具
+   - 规划解决方案
+
+2. 行动 (如果需要)
+   - 使用现有工具
+   - 创建新工具
+   - 询问更多信息
+   
 3. 观察
-   - 等待工具执行结果或用户回应
-   - 工具执行结果由用户提供
-   - 不要假设或想象回应
-   - 不要创建虚假对话
-   - 停止并等待实际输入
-
-回应格式：
-思考：我分析当前情况[具体情况]... 基于[事实]，我需要[目标]...
+   - 等待执行结果
+   - 分析反馈
+   - 规划下一步
 
-[如果需要使用工具：]
-行动：我将使用[工具]来[具体目的]...
+工具使用格式：
 <START_TOOL_CALL>
 name: tool_name
 arguments:
@@ -161,37 +139,28 @@ arguments:
         value
 <END_TOOL_CALL>
 
-[如果需要更多信息：]
-我需要了解更多关于[具体细节]的信息。请提供[需要的信息]。
+创建新工具示例：
+<START_TOOL_CALL>
+name: generate_tool
+arguments:
+    tool_name: custom_tool_name
+    class_name: CustomToolName
+    description: 详细的工具描述
+    parameters:
+        type: object
+        properties:
+            param1: 
+                type: string
+                description: 参数1的描述
+        required: [param1]
+<END_TOOL_CALL>
 
 严格规则：
-‼️ 只使用下面列出的工具
-‼️ 工具调用是可选的 - 需要时询问用户
-‼️ 每次只能调用一个工具
-‼️ 工具调用必须是有效的YAML格式
-‼️ 参数必须正确缩进
-‼️ 使用YAML块样式(|)表示多行值
-‼️ 工具由用户手动执行
-‼️ 等待用户提供工具执行结果
-‼️ 不要假设或想象用户回应
-‼️ 没有用户输入时不要继续对话
-‼️ 不要创建虚假对话
-‼️ 每个动作后停止
-‼️ 不要假设结果
-‼️ 不要假设行动
-
-注意事项：
-- 先思考再行动
-- 需要时询问用户
-- 只使用列出的工具
-- 一次一个工具
-- 严格遵循YAML格式
-- 等待用户回应
-- 工具结果来自用户
-- 不要假设回应
-- 不要虚构对话
-- 每个动作后停止
-- 只在有实际用户输入时继续
+1. 每次只能执行一个工具
+2. 等待用户提供执行结果
+3. 不要假设或想象结果
+4. 不要创建虚假对话
+5. 每个动作后停止等待
 
 任务:
 {user_input}
@@ -235,7 +204,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.18 → jarvis_ai_assistant-0.1.22}/src/jarvis/main.py

@@ -6,6 +6,9 @@ import yaml
 import os
 import sys
 from pathlib import Path
+from prompt_toolkit import prompt
+from prompt_toolkit.completion import WordCompleter
+from prompt_toolkit.validation import Validator, ValidationError
 
 # 添加父目录到Python路径以支持导入
 sys.path.insert(0, str(Path(__file__).parent.parent))
@@ -55,6 +58,19 @@ def load_tasks() -> dict:
     
     return tasks
 
+class NumberValidator(Validator):
+    def validate(self, document):
+        text = document.text.strip()
+        if not text:  # Allow empty input
+            return
+            
+        try:
+            number = int(text)
+            if number < 0:
+                raise ValidationError(message='Please enter a non-negative number')
+        except ValueError:
+            raise ValidationError(message='Please enter a valid number')
+
 def select_task(tasks: dict) -> str:
     """Let user select a task from the list or skip. Returns task description if selected."""
     if not tasks:
@@ -68,9 +84,20 @@ def select_task(tasks: dict) -> str:
         PrettyOutput.print(f"[{i}] {name}", OutputType.INFO)
     PrettyOutput.print("[0] Skip predefined tasks", OutputType.INFO)
     
+    # Create completer with valid numbers
+    valid_numbers = [str(i) for i in range(len(task_names) + 1)]
+    number_completer = WordCompleter(valid_numbers)
+    
     while True:
         try:
-            choice = input("\nSelect a task number (0 to skip): ").strip()
+            choice = prompt(
+                "\nSelect a task number (0 to skip): ",
+                completer=number_completer,
+                validator=NumberValidator(),
+                validate_while_typing=False,
+                enable_history_search=True,
+            ).strip()
+            
             if not choice:
                 return ""
             
@@ -81,9 +108,12 @@ def select_task(tasks: dict) -> str:
                 selected_name = task_names[choice - 1]
                 return tasks[selected_name]  # Return the task description
             else:
-                PrettyOutput.print("Invalid choice. Please try again.", OutputType.ERROR)
-        except ValueError:
-            PrettyOutput.print("Please enter a valid number.", OutputType.ERROR)
+                PrettyOutput.print("Invalid choice. Please select a number from the list.", OutputType.ERROR)
+                
+        except KeyboardInterrupt:
+            return ""  # Return empty on Ctrl+C
+        except EOFError:
+            return ""  # Return empty on Ctrl+D
 
 def main():
     """Main entry point for Jarvis."""

{jarvis_ai_assistant-0.1.18 → jarvis_ai_assistant-0.1.22}/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.18 → jarvis_ai_assistant-0.1.22}/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": {
@@ -113,9 +113,13 @@ class ExampleTool:
     def __init__(self, **kwargs):
         '''初始化工具
         Args:
-            model: 模型处理器 
+            model: 模型处理器  
+                与大模型交互: model.chat(prompt) 
+                使用完成后删除会话: model.delete_chat()
             output_handler: 输出处理器 
+                输出信息: output_handler.print(text, output_type)
             register: 注册器
+                注册工具: register.register_tool_by_file(tool_file)
         '''
         self.model = kwargs.get('model')
         self.output = kwargs.get('output_handler')

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

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