jarvis-ai-assistant 0.1.177__py3-none-any.whl → 0.1.179__py3-none-any.whl

jarvis/__init__.py

@@ -1,4 +1,4 @@
 # -*- coding: utf-8 -*-
 """Jarvis AI Assistant"""
 
-__version__ = "0.1.177"
+__version__ = "0.1.179"

jarvis/jarvis_agent/__init__.py

@@ -1,26 +1,33 @@
 # -*- coding: utf-8 -*-
+# 标准库导入
 import datetime
 import platform
-from typing import Any, Callable, List, Optional, Tuple, Union, Protocol
+from typing import Any, Callable, List, Optional, Protocol, Tuple, Union
 
-from yaspin import yaspin # type: ignore
+# 第三方库导入
+from yaspin import yaspin  # type: ignore
 
-from jarvis.jarvis_agent.output_handler import OutputHandler
+# 本地库导入
+# jarvis_agent 相关
+# jarvis_platform 相关
 from jarvis.jarvis_platform.base import BasePlatform
 from jarvis.jarvis_platform.registry import PlatformRegistry
-from jarvis.jarvis_utils.output import PrettyOutput, OutputType
+# jarvis_utils 相关
+from jarvis.jarvis_utils.config import (get_max_token_count,
+                                        get_max_tool_call_count,
+                                        is_auto_complete,
+                                        is_execute_tool_confirm,
+                                        is_use_analysis, is_use_methodology)
 from jarvis.jarvis_utils.embedding import get_context_token_count
-from jarvis.jarvis_utils.config import get_max_tool_call_count, is_auto_complete, is_execute_tool_confirm
-from jarvis.jarvis_utils.methodology import load_methodology, upload_methodology
-from jarvis.jarvis_utils.globals import make_agent_name, set_agent, delete_agent
+from jarvis.jarvis_utils.globals import (delete_agent, make_agent_name,
+                                         set_agent)
 from jarvis.jarvis_utils.input import get_multiline_input
-from jarvis.jarvis_utils.config import get_max_token_count
+from jarvis.jarvis_utils.methodology import (load_methodology,
+                                             upload_methodology)
+from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 from jarvis.jarvis_utils.tag import ct, ot
 from jarvis.jarvis_utils.utils import user_confirm
 
-from jarvis.jarvis_platform.registry import PlatformRegistry
-
-
 origin_agent_system_prompt = f"""
 <role>
 # 🤖 角色
@@ -105,14 +112,20 @@ origin_agent_system_prompt = f"""
 
 
 class OutputHandlerProtocol(Protocol):
-    def name(self) -> str: ...
-    def can_handle(self, response: str) -> bool: ...
-    def prompt(self) -> str: ...
-    def handle(self, response: str, agent: Any) -> Tuple[bool, Any]: ...
+    def name(self) -> str:
+        ...
 
+    def can_handle(self, response: str) -> bool:
+        ...
 
-class Agent:
+    def prompt(self) -> str:
+        ...
 
+    def handle(self, response: str, agent: Any) -> Tuple[bool, Any]:
+        ...
+
+
+class Agent:
     def set_summary_prompt(self, summary_prompt: str):
         """设置任务完成时的总结提示模板。
 
@@ -129,28 +142,31 @@ class Agent:
         2. 重置对话长度计数器
         3. 清空当前提示
         """
-        self.model.reset() # type: ignore
+        self.model.reset()  # type: ignore
         self.conversation_length = 0
         self.prompt = ""
 
     def __del__(self):
         delete_agent(self.name)
 
-
-    def __init__(self,
-                 system_prompt: str,
-                 name: str = "Jarvis",
-                 description: str = "",
-                 platform: Union[Optional[BasePlatform], Optional[str]] = None,
-                 model_name: Optional[str] = None,
-                 summary_prompt: Optional[str] = None,
-                 auto_complete: Optional[bool] = None,
-                 output_handler: List[OutputHandlerProtocol] = [],
-                 use_tools: List[str] = [],
-                 input_handler: Optional[List[Callable[[str, Any], Tuple[str, bool]]]] = None,
-                 execute_tool_confirm: Optional[bool] = None,
-                 need_summary: bool = True,
-                 multiline_inputer: Optional[Callable[[str], str]] = None):
+    def __init__(
+        self,
+        system_prompt: str,
+        name: str = "Jarvis",
+        description: str = "",
+        platform: Union[Optional[BasePlatform], Optional[str]] = None,
+        model_name: Optional[str] = None,
+        summary_prompt: Optional[str] = None,
+        auto_complete: Optional[bool] = None,
+        output_handler: List[OutputHandlerProtocol] = [],
+        use_tools: List[str] = [],
+        input_handler: Optional[List[Callable[[str, Any], Tuple[str, bool]]]] = None,
+        execute_tool_confirm: Optional[bool] = None,
+        need_summary: bool = True,
+        multiline_inputer: Optional[Callable[[str], str]] = None,
+        use_methodology: Optional[bool] = None,
+        use_analysis: Optional[bool] = None,
+    ):
         """初始化Jarvis Agent实例
 
         参数:
@@ -167,6 +183,8 @@ class Agent:
             execute_tool_confirm: 执行工具前是否需要确认
             need_summary: 是否需要生成总结
             multiline_inputer: 多行输入处理器
+            use_methodology: 是否使用方法论
+            use_analysis: 是否使用任务分析
         """
         self.name = make_agent_name(name)
         self.description = description
@@ -180,7 +198,9 @@ class Agent:
             else:
                 self.model = platform
         else:
-            self.model = PlatformRegistry.get_global_platform_registry().get_normal_platform()
+            self.model = (
+                PlatformRegistry.get_global_platform_registry().get_normal_platform()
+            )
 
         if model_name is not None:
             self.model.set_model_name(model_name)
@@ -188,11 +208,20 @@ class Agent:
         self.model.set_suppress_output(False)
 
         from jarvis.jarvis_tools.registry import ToolRegistry
+
         self.output_handler = output_handler if output_handler else [ToolRegistry()]
         self.set_use_tools(use_tools)
 
-        self.multiline_inputer = multiline_inputer if multiline_inputer else get_multiline_input
+        self.multiline_inputer = (
+            multiline_inputer if multiline_inputer else get_multiline_input
+        )
 
+        self.use_methodology = (
+            use_methodology if use_methodology is not None else is_use_methodology()
+        )
+        self.use_analysis = (
+            use_analysis if use_analysis is not None else is_use_analysis()
+        )
         self.prompt = ""
         self.conversation_length = 0  # Use length counter instead
         self.system_prompt = system_prompt
@@ -205,10 +234,16 @@ class Agent:
         self.max_tool_call_count = get_max_tool_call_count()
         self.after_tool_call_cb: Optional[Callable[[Agent], None]] = None
 
+        self.execute_tool_confirm = (
+            execute_tool_confirm
+            if execute_tool_confirm is not None
+            else is_execute_tool_confirm()
+        )
 
-        self.execute_tool_confirm = execute_tool_confirm if execute_tool_confirm is not None else is_execute_tool_confirm()
-
-        self.summary_prompt = summary_prompt if summary_prompt else f"""<report>
+        self.summary_prompt = (
+            summary_prompt
+            if summary_prompt
+            else f"""<report>
 请生成任务执行的简明总结报告，包括：
 
 <content>
@@ -224,9 +259,12 @@ class Agent:
 </format>
 </report>
 """
+        )
 
-        self.max_token_count =  get_max_token_count()
-        self.auto_complete = auto_complete if auto_complete is not None else is_auto_complete()
+        self.max_token_count = get_max_token_count()
+        self.auto_complete = (
+            auto_complete if auto_complete is not None else is_auto_complete()
+        )
         welcome_message = f"{name} 初始化完成 - 使用 {self.model.name()} 模型"
 
         PrettyOutput.print(welcome_message, OutputType.SYSTEM)
@@ -239,7 +277,9 @@ class Agent:
 
         # 添加工具列表概览
         action_prompt += "\n<overview>\n## Action List\n"
-        action_prompt += ", ".join([handler.name() for handler in self.output_handler])
+        action_prompt += (
+            "[" + ", ".join([handler.name() for handler in self.output_handler]) + "]"
+        )
         action_prompt += "\n</overview>"
 
         # 添加每个工具的详细说明
@@ -249,8 +289,10 @@ class Agent:
             # 获取工具的提示词并确保格式正确
             handler_prompt = handler.prompt().strip()
             # 调整缩进以保持层级结构
-            handler_prompt = "\n".join("   " + line if line.strip() else line
-                                      for line in handler_prompt.split("\n"))
+            handler_prompt = "\n".join(
+                "   " + line if line.strip() else line
+                for line in handler_prompt.split("\n")
+            )
             action_prompt += handler_prompt + "\n</tool>\n"
 
         # 添加工具使用总结
@@ -268,40 +310,25 @@ class Agent:
 </actions>
 """
 
-        complete_prompt = ""
-        if self.auto_complete:
-            complete_prompt = f"""
-<completion>
-<instruction>
-## 任务完成
-当任务完成时，你应该打印以下信息：
-</instruction>
-
-<marker>
-{ot("!!!COMPLETE!!!")}
-</marker>
-</completion>
-"""
-
-        self.model.set_system_message(f"""
+        self.model.set_system_message(
+            f"""
 {self.system_prompt}
 
 {action_prompt}
-
-{complete_prompt}
-""")
+"""
+        )
         self.first = True
 
     def set_use_tools(self, use_tools):
         """设置要使用的工具列表"""
         from jarvis.jarvis_tools.registry import ToolRegistry
+
         for handler in self.output_handler:
             if isinstance(handler, ToolRegistry):
                 if use_tools:
                     handler.use_tools(use_tools)
                 break
 
-
     def set_addon_prompt(self, addon_prompt: str):
         """设置附加提示。
 
@@ -310,7 +337,7 @@ class Agent:
         """
         self.addon_prompt = addon_prompt
 
-    def set_after_tool_call_cb(self, cb: Callable[[Any], None]): # type: ignore
+    def set_after_tool_call_cb(self, cb: Callable[[Any], None]):  # type: ignore
         """设置工具调用后回调函数。
 
         参数:
@@ -321,6 +348,7 @@ class Agent:
     def get_tool_registry(self) -> Optional[Any]:
         """获取工具注册表实例"""
         from jarvis.jarvis_tools.registry import ToolRegistry
+
         for handler in self.output_handler:
             if isinstance(handler, ToolRegistry):
                 return handler
@@ -334,22 +362,25 @@ class Agent:
 
         """
         # 结构化系统指令
-        action_handlers = '\n'.join([f'- {handler.name()}' for handler in self.output_handler])
+        action_handlers = ", ".join([handler.name() for handler in self.output_handler])
 
         # 任务完成提示
-        complete_prompt = f"- 输出{ot('!!!COMPLETE!!!')}" if need_complete and self.auto_complete else ""
+        complete_prompt = (
+            f"- 输出{ot('!!!COMPLETE!!!')}"
+            if need_complete and self.auto_complete
+            else ""
+        )
 
         addon_prompt = f"""
 [系统提示开始]
     请判断是否已经完成任务，如果已经完成：
-    - 说明完成原因，不需要再有新的操作
+    - 直接输出完成原因，不需要再有新的操作，不要输出{ot("TOOL_CALL")}标签
     {complete_prompt}
     如果没有完成，请进行下一步操作：
     - 仅包含一个操作
     - 如果信息不明确，请请求用户补充
     - 如果执行过程中连续失败5次，请使用ask_user询问用户操作
-    - 操作列表：
-    {action_handlers}
+    - 操作列表：{action_handlers}
 [系统提示结束]
 
 请继续。
@@ -390,8 +421,7 @@ class Agent:
         if self.conversation_length > self.max_token_count:
             message = self._summarize_and_clear_history() + "\n\n" + message
             self.conversation_length += get_context_token_count(message)
-        return self.model.chat_until_success(message)   # type: ignore
-
+        return self.model.chat_until_success(message)  # type: ignore
 
     def _summarize_and_clear_history(self) -> str:
         """总结当前对话并清理历史记录
@@ -412,7 +442,6 @@ class Agent:
         # Create a new model instance to summarize, avoid affecting the main conversation
 
         with yaspin(text="正在总结对话历史...", color="cyan") as spinner:
-
             summary_prompt = """
 <summary_request>
 <objective>
@@ -438,9 +467,9 @@ class Agent:
 
             try:
                 with spinner.hidden():
-                    summary = self.model.chat_until_success(self.prompt + "\n" + summary_prompt) # type: ignore
+                    summary = self.model.chat_until_success(self.prompt + "\n" + summary_prompt)  # type: ignore
 
-                self.model.reset() # type: ignore
+                self.model.reset()  # type: ignore
 
                 # 清空当前对话历史，但保留系统消息
                 self.conversation_length = 0  # Reset conversation length
@@ -448,7 +477,7 @@ class Agent:
                 # 添加总结作为新的上下文
                 spinner.text = "总结对话历史完成"
                 spinner.ok("✅")
-                return  f"""<summary>
+                return f"""<summary>
 <header>
 以下是之前对话的关键信息总结：
 </header>
@@ -488,18 +517,29 @@ class Agent:
             if handler.can_handle(response):
                 tool_list.append(handler)
         if len(tool_list) > 1:
-            PrettyOutput.print(f"操作失败：检测到多个操作。一次只能执行一个操作。尝试执行的操作：{', '.join([handler.name() for handler in tool_list])}", OutputType.WARNING)
-            return False, f"操作失败：检测到多个操作。一次只能执行一个操作。尝试执行的操作：{', '.join([handler.name() for handler in tool_list])}"
+            PrettyOutput.print(
+                f"操作失败：检测到多个操作。一次只能执行一个操作。尝试执行的操作：{', '.join([handler.name() for handler in tool_list])}",
+                OutputType.WARNING,
+            )
+            return (
+                False,
+                f"操作失败：检测到多个操作。一次只能执行一个操作。尝试执行的操作：{', '.join([handler.name() for handler in tool_list])}",
+            )
         if len(tool_list) == 0:
             return False, ""
-        if self.max_tool_call_count > 0 and self.tool_call_count >= self.max_tool_call_count:
+        if (
+            self.max_tool_call_count > 0
+            and self.tool_call_count >= self.max_tool_call_count
+        ):
             if user_confirm(f"工具调用次数超过限制，是否继续执行？", True):
                 self.reset_tool_call_count()
             else:
                 return False, ""
         if self.execute_tool_confirm:
             self.reset_tool_call_count()
-        if not self.execute_tool_confirm or user_confirm(f"需要执行{tool_list[0].name()}确认执行？", True):
+        if not self.execute_tool_confirm or user_confirm(
+            f"需要执行{tool_list[0].name()}确认执行？", True
+        ):
             with yaspin(text=f"正在执行{tool_list[0].name()}...", color="cyan") as spinner:
                 with spinner.hidden():
                     result = tool_list[0].handle(response, self)
@@ -508,11 +548,10 @@ class Agent:
                 self.tool_call_count += 1
                 return result
         return False, ""
-    
+
     def reset_tool_call_count(self):
         self.tool_call_count = 0
 
-
     def _complete_task(self) -> str:
         """完成任务并生成总结(如果需要)
 
@@ -524,34 +563,43 @@ class Agent:
             2. 对于子Agent: 可能会生成总结(如果启用)
             3. 使用spinner显示生成状态
         """
+        if self.use_analysis:
+            self._analysis_task()
+        if self.need_summary:
+            with yaspin(text="正在生成总结...", color="cyan") as spinner:
+                self.prompt = self.summary_prompt
+                with spinner.hidden():
+                    ret = self.model.chat_until_success(self.prompt)  # type: ignore
+                    spinner.text = "总结生成完成"
+                    spinner.ok("✅")
+                    return ret
+
+        return "任务完成"
+
+    def _analysis_task(self):
         with yaspin(text="正在分析任务...", color="cyan") as spinner:
             try:
                 # 让模型判断是否需要生成方法论
                 analysis_prompt = f"""<task_analysis>
 <request>
 当前任务已结束，请分析该任务的解决方案：
-
 1. 首先检查现有工具或方法论是否已经可以完成该任务，如果可以，直接说明即可，无需生成新内容
 2. 如果现有工具/方法论不足，评估当前任务是否可以通过编写新工具来自动化解决
 3. 如果可以通过工具解决，请设计并提供工具代码
 4. 如果无法通过编写通用工具完成，评估当前的执行流程是否可以总结为通用方法论
 5. 如果以上都不可行，给出详细理由
-
 请根据分析结果采取相应行动：说明现有工具/方法论、创建新工具、生成新方法论或说明原因。
 </request>
-
 <evaluation_criteria>
 现有资源评估:
 1. 现有工具 - 检查系统中是否已有可以完成该任务的工具
 2. 现有方法论 - 检查是否已有适用于该任务的方法论
 3. 组合使用 - 评估现有工具和方法论组合使用是否可以解决问题
-
 工具评估标准:
 1. 通用性 - 该工具是否可以解决一类问题，而不仅仅是当前特定问题
 2. 自动化 - 该工具是否可以减少人工干预，提高效率
 3. 可靠性 - 该工具是否可以在不同场景下稳定工作
 4. 简单性 - 该工具是否易于使用，参数设计是否合理
-
 方法论评估标准:
 1. 方法论应聚焦于通用且可重复的解决方案流程
 2. 方法论应该具备足够的通用性，可应用于同类问题
@@ -559,7 +607,6 @@ class Agent:
 4. 如果用户明确指出了某个解决步骤的优化方向，这应该被纳入方法论
 5. 方法论要严格按照实际的执行流程来总结，不要遗漏或增加任何步骤
 </evaluation_criteria>
-
 <tool_requirements>
 工具代码要求:
 1. 工具类名应与工具名称保持一致
@@ -569,13 +616,11 @@ class Agent:
 5. 工具描述应详细说明用途、适用场景和使用示例
 6. 参数定义应遵循JSON Schema格式
 7. 不要包含特定任务的细节，保持通用性
-
 工具设计关键点:
 1. **使用PrettyOutput打印执行过程**：强烈建议在工具中使用PrettyOutput显示执行过程，
    这样用户可以了解工具在做什么，提升用户体验。示例：
    ```python
    from jarvis.jarvis_utils.output import PrettyOutput, OutputType
-   
    # 执行中打印信息
    PrettyOutput.print("正在处理数据...", OutputType.INFO)
    # 成功信息
@@ -585,12 +630,10 @@ class Agent:
    # 错误信息
    PrettyOutput.print("操作失败", OutputType.ERROR)
    ```
-   
 2. **结构化返回结果**：工具应该始终返回结构化的结果字典，包含以下字段：
    - success: 布尔值，表示操作是否成功
    - stdout: 字符串，包含工具的主要输出内容
    - stderr: 字符串，包含错误信息（如果有）
-
 3. **异常处理**：工具应该妥善处理可能发生的异常，并在失败时清理已创建的资源
    ```python
    try:
@@ -610,7 +653,6 @@ class Agent:
        }}
    ```
 </tool_requirements>
-
 <methodology_requirements>
 方法论格式要求:
 1. 问题重述: 简明扼要的问题归纳，不含特定细节
@@ -618,15 +660,12 @@ class Agent:
 3. 注意事项: 执行中可能遇到的常见问题和注意点，尤其是用户指出的问题
 4. 可选步骤: 对于有多种解决路径的问题，标注出可选步骤和适用场景
 </methodology_requirements>
-
 <output_requirements>
 根据分析结果，输出以下三种情况之一：
-
 1. 如果现有工具/方法论可以解决，直接输出说明：
 已有工具/方法论可以解决该问题，无需创建新内容。
 可用的工具/方法论：[列出工具名称或方法论名称]
 使用方法：[简要说明如何使用]
-
 2. 工具创建（如果需要创建新工具）:
 {ot("TOOL_CALL")}
 want: 创建新工具来解决XXX问题
@@ -637,14 +676,12 @@ arguments:
     # -*- coding: utf-8 -*-
     from typing import Dict, Any
     from jarvis.jarvis_utils.output import PrettyOutput, OutputType
-    
     class 工具名称:
         name = "工具名称"
         description = "Tool for text transformation"
                 Tool description
         适用场景：1. 格式化文本; 2. 处理标题; 3. 标准化输出
         \"\"\"
-        
         parameters = {{
             "type": "object",
             "properties": {{
@@ -652,19 +689,15 @@ arguments:
             }},
             "required": []
         }}
-        
         @staticmethod
         def check() -> bool:
             return True
-        
         def execute(self, args: Dict[str, Any]) -> Dict[str, Any]:
             try:
                 # 使用PrettyOutput显示执行过程
                 PrettyOutput.print("开始执行操作...", OutputType.INFO)
-                
                 # 实现逻辑
                 # ...
-                
         PrettyOutput.print("操作已完成", OutputType.SUCCESS)
         return {{
             "success": True,
@@ -679,7 +712,6 @@ arguments:
             "stderr": f"操作失败: {{str(e)}}"
         }}
 {ct("TOOL_CALL")}
-
 3. 方法论创建（如果需要创建新方法论）:
 {ot("TOOL_CALL")}
 want: 添加/更新xxxx的方法论
@@ -690,14 +722,13 @@ arguments:
   content: |2
     方法论内容
 {ct("TOOL_CALL")}
-
 如果以上三种情况都不适用，则直接输出原因分析，不要使用工具调用格式。
 </output_requirements>
 </task_analysis>"""
 
                 self.prompt = analysis_prompt
                 with spinner.hidden():
-                    response = self.model.chat_until_success(self.prompt) # type: ignore
+                    response = self.model.chat_until_success(self.prompt)  # type: ignore
 
                 with spinner.hidden():
                     self._call_tools(response)
@@ -706,17 +737,6 @@ arguments:
             except Exception as e:
                 spinner.text = "分析失败"
                 spinner.fail("❌")
-        if self.need_summary:
-            with yaspin(text="正在生成总结...", color="cyan") as spinner:
-                self.prompt = self.summary_prompt
-                with spinner.hidden():
-                    ret = self.model.chat_until_success(self.prompt) # type: ignore
-                    spinner.text = "总结生成完成"
-                    spinner.ok("✅")
-                    return ret
-
-        return "任务完成"
-
 
     def run(self, user_input: str) -> Any:
         """处理用户输入并执行任务
@@ -738,10 +758,9 @@ arguments:
 
             self.prompt = f"{user_input}"
 
-            if self.first:
-                
+            if self.first and self.use_methodology:
                 # 先尝试上传方法轮
-                platform = self.model if hasattr(self.model, 'upload_files') else None
+                platform = self.model if hasattr(self.model, "upload_files") else None
                 if platform and upload_methodology(platform):
                     self.prompt = f"{user_input}\n\n方法论已上传到平台，请参考平台上的方法论内容"
                 else:
@@ -750,7 +769,7 @@ arguments:
                         msg, _ = handler(msg, self)
                     # 上传失败则回退到本地加载
                     self.prompt = f"{user_input}\n\n以下是历史类似问题的执行经验，可参考：\n{load_methodology(msg, self.get_tool_registry())}"
-                
+
                 self.first = False
 
             self.conversation_length = get_context_token_count(self.prompt)
@@ -760,7 +779,9 @@ arguments:
 
                     current_response = self._call_model(self.prompt, True)
                     self.prompt = ""
-                    self.conversation_length += get_context_token_count(current_response)
+                    self.conversation_length += get_context_token_count(
+                        current_response
+                    )
 
                     need_return, self.prompt = self._call_tools(current_response)
 
@@ -775,11 +796,13 @@ arguments:
 
                     if self.auto_complete and ot("!!!COMPLETE!!!") in current_response:
                         return self._complete_task()
-                    
+
                     self.reset_tool_call_count()
 
                     # 获取用户输入
-                    user_input = self.multiline_inputer(f"{self.name}: 请输入，或输入空行来结束当前任务：")
+                    user_input = self.multiline_inputer(
+                        f"{self.name}: 请输入，或输入空行来结束当前任务："
+                    )
 
                     if user_input:
                         self.prompt = user_input
@@ -808,8 +831,5 @@ arguments:
             用于重置Agent状态而不影响系统消息
         """
         self.prompt = ""
-        self.model.reset() # type: ignore
+        self.model.reset()  # type: ignore
         self.conversation_length = 0  # 重置对话长度
-
-
-

jarvis/jarvis_agent/builtin_input_handler.py

@@ -1,8 +1,8 @@
 # -*- coding: utf-8 -*-
 import re
 from typing import Any, Tuple
-from jarvis.jarvis_utils.config import get_replace_map
 
+from jarvis.jarvis_utils.config import get_replace_map
 
 
 def builtin_input_handler(user_input: str, agent: Any) -> Tuple[str, bool]:

jarvis/jarvis_agent/jarvis.py

@@ -2,24 +2,20 @@
 import argparse
 import os
 import sys
+from typing import Dict
 
-from typing import Dict, List, Tuple  # 增加必要的类型导入
-
-from jarvis.jarvis_utils.config import get_data_dir
-from prompt_toolkit import prompt
 import yaml
+from prompt_toolkit import prompt
 from yaspin import yaspin
-from jarvis.jarvis_agent import (
-     PrettyOutput, OutputType,
-     get_multiline_input,
-     user_confirm,
-     Agent,
-     origin_agent_system_prompt
-)
+
+from jarvis.jarvis_agent import (Agent, OutputType, PrettyOutput,
+                                 get_multiline_input,
+                                 origin_agent_system_prompt, user_confirm)
+from jarvis.jarvis_agent.builtin_input_handler import builtin_input_handler
+from jarvis.jarvis_agent.shell_input_handler import shell_input_handler
 from jarvis.jarvis_tools.registry import ToolRegistry
+from jarvis.jarvis_utils.config import get_data_dir
 from jarvis.jarvis_utils.utils import init_env
-from jarvis.jarvis_agent.shell_input_handler import shell_input_handler
-from jarvis.jarvis_agent.builtin_input_handler import builtin_input_handler
 
 
 def _load_tasks() -> Dict[str, str]: