jarvis-ai-assistant 0.1.18__py3-none-any.whl → 0.1.19__py3-none-any.whl

jarvis/__init__.py

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

jarvis/agent.py

@@ -103,55 +103,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 +138,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}

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/tools/generator.py

@@ -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.dist-info → jarvis_ai_assistant-0.1.19.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: jarvis-ai-assistant
-Version: 0.1.18
+Version: 0.1.19
 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.dist-info → jarvis_ai_assistant-0.1.19.dist-info}/RECORD

@@ -1,10 +1,10 @@
-jarvis/__init__.py,sha256=_f28_I38xy1i_kwbYg6iIU59t_tF8apmhCpqrc4yWvQ,50
-jarvis/agent.py,sha256=f8RIOFIiwRAuql4O32P_nDXByzBKbrNM-ergf_F8FfU,9409
-jarvis/main.py,sha256=CwqsXKTxeZ3rWwdHfhMyzX1DFJC3NA7D-Sbla0YkdSE,7046
+jarvis/__init__.py,sha256=ZS0T4CRb_B0dOsiPYPRNrunjMPVvZzmD9IIaUB29ig8,50
+jarvis/agent.py,sha256=r00CssF8HzSj4r2GXR9KNJ3XJk46HIjw4BEfzx-fKMQ,8114
+jarvis/main.py,sha256=tptQp6rjhHOSsPphmrtoxf7l78btnP_hJmr3hVMO4zg,8103
 jarvis/utils.py,sha256=YipQpEuIRwFE3y3YrgGlSVLEPDrlbBNY1gRiOJix9DU,6602
-jarvis/__pycache__/__init__.cpython-313.pyc,sha256=PGPqHZavqZ2fSStY_yJGNQjMt49Af70vSbxInRVxFwM,209
-jarvis/__pycache__/agent.cpython-313.pyc,sha256=anFGVa4pfpsOCDbstPAPSsUmglCZUrRkuDN-jnvhSyI,12152
-jarvis/__pycache__/main.cpython-313.pyc,sha256=fvBChS8JvPs2jNR4cYyRJi10n7TVph1RAZtp-AkESuo,9669
+jarvis/__pycache__/__init__.cpython-313.pyc,sha256=TnbFSQwCFYqlSRaAgJgfajYnPQVIUrqM9Pyj4QRQjDQ,209
+jarvis/__pycache__/agent.cpython-313.pyc,sha256=ItAH6Mb964vpqvziIsw-62avHUVu54mNgTa02VWK3lM,10855
+jarvis/__pycache__/main.cpython-313.pyc,sha256=uQ1QHMHR6EukOb7z-YOz8CicXRPEfiHyS7KRq15xxVw,10943
 jarvis/__pycache__/models.cpython-313.pyc,sha256=uWuRIjGrY4YDB3dGW5PGDLWaS03et8g11O725TjY_eU,5960
 jarvis/__pycache__/tools.cpython-313.pyc,sha256=lAD4LrnnWzNZQmHXGfZ_2l7oskOpr2_2OC-gdFhxQY8,33933
 jarvis/__pycache__/utils.cpython-313.pyc,sha256=Jpkf2GOA1d_fgrRftqZch1yiO0mB75LWHzQk8IfhHRU,8977
@@ -18,7 +18,7 @@ jarvis/models/__pycache__/kimi.cpython-313.pyc,sha256=-hOwhZrfvqfd5K_c34Tm-uR3ga
 jarvis/tools/__init__.py,sha256=AjmFYLXt-_ulZXdYiXya2fD7QCJ-RaK91PEZCNj-UbA,262
 jarvis/tools/base.py,sha256=ZHmRczYks7CuaYHU9uB5QnsD40G_jeqE6ubuhlaSFR8,7318
 jarvis/tools/file_ops.py,sha256=KbQLVCCXw-MtJg-12iyMeVGu8BTtLq7Mk7fpVKau40U,4296
-jarvis/tools/generator.py,sha256=A_J7U9_Vjh37zotGkt5VDM8As1BfNawM-6vbclJ4PR4,7142
+jarvis/tools/generator.py,sha256=VJXfZg0Xt3OZr1LtX8m-RJh9807VxZRah5xV7UtX6is,7406
 jarvis/tools/shell.py,sha256=MWe9-BAGApbGJfR60XG4nElGYHNBnbdF9vsOQTnEZ4g,2989
 jarvis/tools/__pycache__/__init__.cpython-313.pyc,sha256=KC48FS57427navbkxZ0eDKLePMCH2BKc5vAIu0b8lFc,404
 jarvis/tools/__pycache__/base.cpython-313.pyc,sha256=r-jPPg5L_Rok2c_VsMEAXkhuiiUjHv6bLCwVjnONDRo,9861
@@ -26,7 +26,7 @@ jarvis/tools/__pycache__/bing_search.cpython-313.pyc,sha256=1G_wPbk5wcQYh7H0drLI
 jarvis/tools/__pycache__/calculator.cpython-313.pyc,sha256=C_qwTDGm6gc7QNxtPzPZXyStdKEintJVQIt5NMHQ8oY,4205
 jarvis/tools/__pycache__/calculator_tool.cpython-313.pyc,sha256=PI4LZNDTPdSe3ffWDRovLZ-r-vF8Kl-n6xdGdFWiBpY,4296
 jarvis/tools/__pycache__/file_ops.cpython-313.pyc,sha256=OBR2EDQ2Ana7CTtLOoM5AYJzyVwP683LS7QrOR8_I3g,5208
-jarvis/tools/__pycache__/generator.cpython-313.pyc,sha256=sOpj8PBTUwYSVrWhBr-ESK43lSP0HOkRUJdEpLD6PPY,8482
+jarvis/tools/__pycache__/generator.cpython-313.pyc,sha256=3XTMBphRw7gqRqJXPuQrA4_6FirinFZ3khd7LiCy7uU,8746
 jarvis/tools/__pycache__/python_script.cpython-313.pyc,sha256=8JpryqTovEiTvBlWAK1KjZmPvHUuPc9GT9rTXBEQoJc,6693
 jarvis/tools/__pycache__/rag.cpython-313.pyc,sha256=JH6-PSZRMKAvTZqCwlRXJGClxYXNMs-vetU0q7hBLz0,6064
 jarvis/tools/__pycache__/search.cpython-313.pyc,sha256=wLMIkFwT-h4NGHgssytT4xme7sGO6ZhEnex7kjcy0-k,5990
@@ -36,9 +36,9 @@ jarvis/tools/__pycache__/user_confirmation.cpython-313.pyc,sha256=wK3Ev10lHSUSRv
 jarvis/tools/__pycache__/user_input.cpython-313.pyc,sha256=JjTFOhObKsKF4Pn8KBRuKfV1_Ssj083fjU7Mfc_5z7c,2531
 jarvis/tools/__pycache__/user_interaction.cpython-313.pyc,sha256=RuVZ-pmiPBDywY3efgXSfohMAciC1avMGPmBK5qlnew,3305
 jarvis/tools/__pycache__/webpage.cpython-313.pyc,sha256=BjzSfnNzsKCrLETCcWjt32lNDLzwnjqcVGg4JfWd9OM,3008
-jarvis_ai_assistant-0.1.18.dist-info/LICENSE,sha256=AGgVgQmTqFvaztRtCAXsAMryUymB18gZif7_l2e1XOg,1063
-jarvis_ai_assistant-0.1.18.dist-info/METADATA,sha256=3n1MHwr3pmlVGHHwG-CX3tLdb-SI7lqLA7UJlFhM528,10193
-jarvis_ai_assistant-0.1.18.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-jarvis_ai_assistant-0.1.18.dist-info/entry_points.txt,sha256=iKu7OMfew9dtfGhW71gIMTg4wvafuPqKb4wyQOnMAGU,44
-jarvis_ai_assistant-0.1.18.dist-info/top_level.txt,sha256=1BOxyWfzOP_ZXj8rVTDnNCJ92bBGB0rwq8N1PCpoMIs,7
-jarvis_ai_assistant-0.1.18.dist-info/RECORD,,
+jarvis_ai_assistant-0.1.19.dist-info/LICENSE,sha256=AGgVgQmTqFvaztRtCAXsAMryUymB18gZif7_l2e1XOg,1063
+jarvis_ai_assistant-0.1.19.dist-info/METADATA,sha256=SPaJ0q7U024HA6W3sagtO6SlmDizohRfXhCBEuFMdrM,10193
+jarvis_ai_assistant-0.1.19.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+jarvis_ai_assistant-0.1.19.dist-info/entry_points.txt,sha256=iKu7OMfew9dtfGhW71gIMTg4wvafuPqKb4wyQOnMAGU,44
+jarvis_ai_assistant-0.1.19.dist-info/top_level.txt,sha256=1BOxyWfzOP_ZXj8rVTDnNCJ92bBGB0rwq8N1PCpoMIs,7
+jarvis_ai_assistant-0.1.19.dist-info/RECORD,,