jarvis-ai-assistant 0.1.63__py3-none-any.whl → 0.1.64__py3-none-any.whl

jarvis/__init__.py

@@ -1,3 +1,3 @@
 """Jarvis AI Assistant"""
 
-__version__ = "0.1.63" 
+__version__ = "0.1.64" 

jarvis/agent.py

@@ -142,20 +142,6 @@ class Agent:
         except Exception as e:
             PrettyOutput.print(f"总结对话历史失败: {str(e)}", OutputType.ERROR)
 
-    
-    def _choose_methodology(self, methodology: Dict[str, str], task: str) -> str:
-        """选择方法论"""
-        prompt = f"""请根据任务内容选择合适的方法论，并返回方法论内容，格式如下：
-任务内容:
-{task}
-
-方法论:
-"""
-        for k, v in methodology.items():
-            prompt += f"问题类型：{k}\n"
-            prompt += f"方法论：{v}\n"
-        return self._call_model(prompt)
-
     def _complete_task(self) -> str:
         """完成任务并生成总结
         
@@ -164,6 +150,34 @@ class Agent:
         """
         PrettyOutput.section("任务完成", OutputType.SUCCESS)
         
+        # 询问是否生成方法论，带输入验证
+        while True:
+            user_input = input("是否要为此任务生成方法论？(y/n): ").strip().lower()
+            if user_input in ['y', 'n', '']:
+                break
+            PrettyOutput.print("无效输入，请输入 y 或 n", OutputType.WARNING)
+        
+        if user_input == 'y':
+            try:
+                # 让模型判断是否需要生成方法论
+                analysis_prompt = """本次任务已结束，请分析是否需要生成方法论。
+如果认为需要生成方法论，请先判断是创建新的方法论还是更新已有方法论。如果是更新已有方法论，使用update，否则使用add。
+如果认为不需要生成方法论，请说明原因。
+仅输出方法论工具的调用指令，或者是不需要生成方法论的说明，除此之外不要输出任何内容。
+"""
+                self.prompt = analysis_prompt
+                response = self._call_model(self.prompt)
+                
+                # 检查是否包含工具调用
+                try:
+                    result = Agent.extract_tool_calls(response)
+                    PrettyOutput.print(result, OutputType.RESULT)
+                except Exception as e:
+                    PrettyOutput.print(f"处理方法论生成失败: {str(e)}", OutputType.ERROR)
+                
+            except Exception as e:
+                PrettyOutput.print(f"生成方法论时发生错误: {str(e)}", OutputType.ERROR)
+        
         if not self.is_sub_agent:
             return "Task completed"
         
@@ -198,9 +212,6 @@ class Agent:
 
             # 加载方法论
             methodology = self._load_methodology()
-
-            methodology =self._choose_methodology(methodology, user_input)
-
             methodology_prompt = ""
             if methodology:
                 methodology_prompt = f"""这是以往处理问题的标准方法论，如果当前任务与此类似，可参考：
@@ -271,7 +282,6 @@ arguments:
 
 特殊指令：
 1. !<<SUMMARIZE>>! - 当你发现对话历史过长可能导致token超限时，可以使用此指令总结当前对话要点并清空历史。使用方法：直接回复"!<<SUMMARIZE>>!"即可。
-2. !<<FINISHED>>! - 当你确认任务已经完成时，使用此指令结束任务。使用方法：在回复中包含"!<<FINISHED>>!"即可。
 
 -------------------------------------------------------------
 
@@ -316,10 +326,6 @@ arguments:
                         self.prompt = tool_result
                         continue
                     
-                    # 检查是否完成任务
-                    if "!<<FINISHED>>!" in current_response:
-                        return self._complete_task()
-                    
                     # 获取用户输入
                     user_input = get_multiline_input(f"{self.name}: 您可以继续输入，或输入空行结束当前任务")
                     if user_input == "__interrupt__":

{jarvis_ai_assistant-0.1.63.dist-info → jarvis_ai_assistant-0.1.64.dist-info}/METADATA

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

@@ -1,5 +1,5 @@
-jarvis/__init__.py,sha256=sZ5DQXpur0P2tY_PCFzbnYK8EQLpCoUcnhPRL-xFwRg,50
-jarvis/agent.py,sha256=aIth7geQ2ZIugBYjW_Rx3sr-I6NuZtH1INEj4hOBVD4,14242
+jarvis/__init__.py,sha256=wMtxQxTakpumAd-K51g16_sMIW8so_8Pv15IEYnhfwI,50
+jarvis/agent.py,sha256=kl6pwNrluzb-9eZKgwmsk5Jh4CpWi4F8B3RvEQNvc5U,14921
 jarvis/main.py,sha256=gXXtnrkkvGwEswJL6qiYjVrg3bpzye-GJeAe0Nf2B9o,6509
 jarvis/utils.py,sha256=JlkuC9RtspXH2VWDmj9nR0vnb8ie1gIsKc4vC7WRco8,7321
 jarvis/jarvis_coder/main.py,sha256=TosDDiaYSjDpzKPNKcygxZ3XXWbcnvBmIIMn3UMBc60,35102
@@ -21,9 +21,9 @@ jarvis/tools/search.py,sha256=1EqOVvLhg2Csh-i03-XeCrusbyfmH69FZ8khwZt8Tow,6131
 jarvis/tools/shell.py,sha256=UPKshPyOaUwTngresUw-ot1jHjQIb4wCY5nkJqa38lU,2520
 jarvis/tools/sub_agent.py,sha256=rEtAmSVY2ZjFOZEKr5m5wpACOQIiM9Zr_3dT92FhXYU,2621
 jarvis/tools/webpage.py,sha256=d3w3Jcjcu1ESciezTkz3n3Zf-rp_l91PrVoDEZnckOo,2391
-jarvis_ai_assistant-0.1.63.dist-info/LICENSE,sha256=AGgVgQmTqFvaztRtCAXsAMryUymB18gZif7_l2e1XOg,1063
-jarvis_ai_assistant-0.1.63.dist-info/METADATA,sha256=8Kq0xkFkfbqX9sB6kR3h9yKJhtb2T5hgeVBy_1Onyx4,11213
-jarvis_ai_assistant-0.1.63.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-jarvis_ai_assistant-0.1.63.dist-info/entry_points.txt,sha256=ieRI4ilnGNx1R6LlzT2P510mJ27lhLesVZToezDjSd8,89
-jarvis_ai_assistant-0.1.63.dist-info/top_level.txt,sha256=1BOxyWfzOP_ZXj8rVTDnNCJ92bBGB0rwq8N1PCpoMIs,7
-jarvis_ai_assistant-0.1.63.dist-info/RECORD,,
+jarvis_ai_assistant-0.1.64.dist-info/LICENSE,sha256=AGgVgQmTqFvaztRtCAXsAMryUymB18gZif7_l2e1XOg,1063
+jarvis_ai_assistant-0.1.64.dist-info/METADATA,sha256=akHo9a0miOBYWvtITaAuo1G3PZxiE2yPSZxuh0QE1C4,11213
+jarvis_ai_assistant-0.1.64.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+jarvis_ai_assistant-0.1.64.dist-info/entry_points.txt,sha256=ieRI4ilnGNx1R6LlzT2P510mJ27lhLesVZToezDjSd8,89
+jarvis_ai_assistant-0.1.64.dist-info/top_level.txt,sha256=1BOxyWfzOP_ZXj8rVTDnNCJ92bBGB0rwq8N1PCpoMIs,7
+jarvis_ai_assistant-0.1.64.dist-info/RECORD,,