entari-plugin-hyw 4.0.0rc12__py3-none-any.whl → 4.0.0rc13__py3-none-any.whl

{entari_plugin_hyw-4.0.0rc12.dist-info → entari_plugin_hyw-4.0.0rc13.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: entari_plugin_hyw
-Version: 4.0.0rc12
+Version: 4.0.0rc13
 Summary: Use large language models to interpret chat messages
 Author-email: kumoSleeping <zjr2992@outlook.com>
 License: MIT

{entari_plugin_hyw-4.0.0rc12.dist-info → entari_plugin_hyw-4.0.0rc13.dist-info}/RECORD

@@ -5,10 +5,10 @@ entari_plugin_hyw/history.py,sha256=0XJwbfvXH5T1EPt4G1J5wWMJsKi0FfmajY5cvw8CQWE,
 entari_plugin_hyw/misc.py,sha256=5IqF5Z2C_6Ufy5TI89uX5hX5fVYcXOTZIQUIu_tvf54,6855
 entari_plugin_hyw/search_cache.py,sha256=7MIhTm5_YnZjc0aBaX7AE4AJp0VT8eU6ObR6mTkoerc,4285
 hyw_core/__init__.py,sha256=Jlr9Ic-BLOPTnff6OctUCdjDMdK4nssTF_vHie4QKTo,1958
-hyw_core/agent.py,sha256=lztXefMYjDedeTlTulEXq-5S5_D0BcQp3YVgFnNOX4E,26893
+hyw_core/agent.py,sha256=xKvO9CIo0MX7mBQ6DuLDaDLvREyRIZMNXKk4tpOMi1U,29496
 hyw_core/config.py,sha256=DHxwToUVLm1nT88gG05e3hVzSLxXMk9BjgjAnhGCADk,4918
 hyw_core/core.py,sha256=_jN4831OeHQ_aM7sIlzcwYb5_Lp82kp2XmqpJD_tsLA,16097
-hyw_core/definitions.py,sha256=SrIYngRELNSsWlM4wDKdlAmy_kPwGX2onlkYF9Ti0aM,4137
+hyw_core/definitions.py,sha256=pH46L-N25pSuPaIiN7l7yfoD6oHK6BLHigE0eYLFmJQ,4270
 hyw_core/image_cache.py,sha256=t8pr1kgH2ngK9IhrBAhzUqhBWERNztUywMzgCFZEtQk,9899
 hyw_core/pipeline.py,sha256=ZWwF0DHa29-65lUMU1_Fem3xQmxl7X_vgeni0ErOb8Q,22826
 hyw_core/search.py,sha256=VvfNSb9Hf7ZQWlNtnZfYe2eO9qPjYtwJxVlud6OdeCQ,7787
@@ -66,7 +66,7 @@ hyw_core/crawling/models.py,sha256=pCKe0k9xT3taSAlTlh0PazcLV0xYsm8p3XIkLHGf-LM,2
 hyw_core/stages/__init__.py,sha256=W89cWpq-HBLi2FprtJQSjQNLzpbhM8ZCkqPG61D_imE,521
 hyw_core/stages/base.py,sha256=EfnTkISXbBNxjARykqIhmMrVqw2tqZl7ozJbJEbRnhI,2806
 hyw_core/stages/summary.py,sha256=sgHCm_Leq_pkJ4YcgQuf8croiOP1oKz171TnzJwRwVs,7080
-entari_plugin_hyw-4.0.0rc12.dist-info/METADATA,sha256=R0rURsgj-bZ9i3MQrbP7txMhegpEthRncRZIOz4j6Nc,3845
-entari_plugin_hyw-4.0.0rc12.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
-entari_plugin_hyw-4.0.0rc12.dist-info/top_level.txt,sha256=ah76OrufRX0okOl4Fv8MO6PXiT0IaZ1oG0eDrdAPoNo,27
-entari_plugin_hyw-4.0.0rc12.dist-info/RECORD,,
+entari_plugin_hyw-4.0.0rc13.dist-info/METADATA,sha256=NhbBeH0svhxJ6MvHl6vKF-BCzg-jRczAhlg_3fXD_lM,3845
+entari_plugin_hyw-4.0.0rc13.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
+entari_plugin_hyw-4.0.0rc13.dist-info/top_level.txt,sha256=ah76OrufRX0okOl4Fv8MO6PXiT0IaZ1oG0eDrdAPoNo,27
+entari_plugin_hyw-4.0.0rc13.dist-info/RECORD,,

hyw_core/agent.py

@@ -2,7 +2,7 @@
 Agent Pipeline
 
 Tool-calling agent that can autonomously use web_tool to search/screenshot.
-Maximum 2 tool calls, then forced summary.
+Maximum 2 rounds of tool calls, up to 3 parallel calls per round.
 """
 
 import asyncio
@@ -31,6 +31,9 @@ class AgentSession:
     messages: List[Dict] = field(default_factory=list)  # LLM conversation
     created_at: float = field(default_factory=time.time)
     
+    # Round tracking (each round can have up to 3 parallel tool calls)
+    round_count: int = 0
+    
     # Image tracking
     user_image_count: int = 0  # Number of images from user input
     total_image_count: int = 0  # Total images including web screenshots
@@ -45,12 +48,13 @@ class AgentSession:
     
     @property
     def call_count(self) -> int:
+        """Total number of individual tool calls."""
         return len(self.tool_calls)
     
     @property
     def should_force_summary(self) -> bool:
-        """第3次调用时强制总结"""
-        return self.call_count >= 2
+        """Force summary after 2 rounds of tool calls."""
+        return self.round_count >= 2
 
 
 def parse_filter_syntax(query: str, max_count: int = 3):
@@ -130,12 +134,13 @@ class AgentPipeline:
     
     Flow:
     1. 用户输入 → LLM (with tools)
-    2. If tool_call: execute tool → notify user → loop
-    3. If call_count >= 2: force summary on next call
+    2. If tool_call: execute all tools in parallel → notify user with batched message → loop
+    3. If call_count >= 2 rounds: force summary on next call
     4. Return final content
     """
     
-    MAX_TOOL_CALLS = 2
+    MAX_TOOL_ROUNDS = 2  # Maximum rounds of tool calls
+    MAX_PARALLEL_TOOLS = 3  # Maximum parallel tool calls per round
     
     def __init__(
         self, 
@@ -183,7 +188,9 @@ class AgentPipeline:
         
         # Build initial messages
         language = getattr(self.config, "language", "Simplified Chinese")
-        system_prompt = AGENT_SYSTEM_PROMPT + f"\n\n用户要求的语言: {language}"
+        from datetime import datetime
+        current_time = datetime.now().strftime("%Y-%m-%d %H:%M")
+        system_prompt = AGENT_SYSTEM_PROMPT + f"\n\n用户要求的语言: {language}\n当前时间: {current_time}"
         
         # Build user content with images if provided
         user_image_count = len(images) if images else 0
@@ -230,7 +237,7 @@ class AgentPipeline:
         while True:
             # Check if we need to force summary (no tools)
             if session.should_force_summary:
-                logger.info(f"AgentPipeline: Max tool calls ({self.MAX_TOOL_CALLS}) reached, forcing summary")
+                logger.info(f"AgentPipeline: Max tool rounds ({self.MAX_TOOL_ROUNDS}) reached, forcing summary")
                 # Add context message about collected info
                 if context.web_results:
                     context_msg = self._format_web_context(context)
@@ -307,7 +314,12 @@ class AgentPipeline:
                 ]
             })
             
-            # Execute tool calls
+            # Execute all tool calls in parallel
+            tool_tasks = []
+            tool_call_ids = []
+            tool_call_names = []
+            tool_call_args_list = []
+            
             for tool_call in message.tool_calls:
                 tc_id = tool_call.id
                 func_name = tool_call.function.name
@@ -317,17 +329,22 @@ class AgentPipeline:
                 except json.JSONDecodeError:
                     args = {}
                 
-                logger.info(f"AgentPipeline: Executing tool '{func_name}' with args: {args}")
-                
+                tool_call_ids.append(tc_id)
+                tool_call_names.append(func_name)
+                tool_call_args_list.append(args)
+                logger.info(f"AgentPipeline: Queueing tool '{func_name}' with args: {args}")
+            
+            # Check for refuse_answer first (handle immediately)
+            for idx, func_name in enumerate(tool_call_names):
                 if func_name == "refuse_answer":
-                    # Handle refusal
+                    args = tool_call_args_list[idx]
                     reason = args.get("reason", "Refused")
                     context.should_refuse = True
                     context.refuse_reason = reason
                     
                     session.messages.append({
                         "role": "tool",
-                        "tool_call_id": tc_id,
+                        "tool_call_id": tool_call_ids[idx],
                         "content": f"已拒绝回答: {reason}"
                     })
                     
@@ -339,23 +356,50 @@ class AgentPipeline:
                         "stats": {"total_time": time.time() - start_time},
                         "usage": usage_totals,
                     }
+            
+            # Execute web_tool calls in parallel
+            search_start = time.time()
+            tasks_to_run = []
+            task_indices = []
+            
+            for idx, func_name in enumerate(tool_call_names):
+                if func_name == "web_tool":
+                    tasks_to_run.append(self._execute_web_tool(tool_call_args_list[idx], context))
+                    task_indices.append(idx)
+            
+            # Run all web_tool calls in parallel
+            if tasks_to_run:
+                results = await asyncio.gather(*tasks_to_run, return_exceptions=True)
+            else:
+                results = []
+            
+            session.search_time += time.time() - search_start
+            
+            # Process results and collect notifications
+            notifications = []
+            result_map = {}  # Map task index to result
+            
+            for i, result in enumerate(results):
+                task_idx = task_indices[i]
+                if isinstance(result, Exception):
+                    result_map[task_idx] = {"summary": f"执行失败: {result}", "results": []}
+                else:
+                    result_map[task_idx] = result
+            
+            # Add all tool results to messages and collect notifications
+            for idx, func_name in enumerate(tool_call_names):
+                tc_id = tool_call_ids[idx]
+                args = tool_call_args_list[idx]
                 
-                elif func_name == "web_tool":
-                    # Execute web tool with time tracking
-                    search_start = time.time()
-                    result = await self._execute_web_tool(args, context)
-                    session.search_time += time.time() - search_start
+                if func_name == "web_tool":
+                    result = result_map.get(idx, {"summary": "未执行", "results": []})
                     
                     # Track tool call
                     session.tool_calls.append({"name": func_name, "args": args})
                     session.tool_results.append(result)
                     
-                    # Send IM notification with search result (NOT "正在搜索...")
-                    if self.send_func:
-                        try:
-                            await self.send_func(f"🔍 {result['summary']}")
-                        except Exception as e:
-                            logger.warning(f"AgentPipeline: Failed to send notification: {e}")
+                    # Collect notification
+                    notifications.append(f"🔍 {result['summary']}")
                     
                     # Add tool result to messages
                     result_content = f"搜索完成: {result['summary']}\n\n找到 {len(result.get('results', []))} 个结果"
@@ -368,15 +412,15 @@ class AgentPipeline:
                     # Add image source hint for web screenshots
                     screenshot_count = result.get("screenshot_count", 0)
                     if screenshot_count > 0:
-                        start_idx = session.total_image_count + 1
-                        end_idx = session.total_image_count + screenshot_count
-                        session.total_image_count = end_idx
+                        start_idx_img = session.total_image_count + 1
+                        end_idx_img = session.total_image_count + screenshot_count
+                        session.total_image_count = end_idx_img
                         
                         source_desc = result.get("source_desc", "网页截图")
-                        if start_idx == end_idx:
-                            hint = f"第{start_idx}张图片来自{source_desc}，作为查询的参考资料"
+                        if start_idx_img == end_idx_img:
+                            hint = f"第{start_idx_img}张图片来自{source_desc}，作为查询的参考资料"
                         else:
-                            hint = f"第{start_idx}-{end_idx}张图片来自{source_desc}，作为查询的参考资料"
+                            hint = f"第{start_idx_img}-{end_idx_img}张图片来自{source_desc}，作为查询的参考资料"
                         session.messages.append({"role": "system", "content": hint})
                 else:
                     # Unknown tool
@@ -385,6 +429,19 @@ class AgentPipeline:
                         "tool_call_id": tc_id,
                         "content": f"Unknown tool: {func_name}"
                     })
+            
+            # Send batched notification (up to 3 lines)
+            if self.send_func and notifications:
+                try:
+                    # Join notifications with newlines, max 3 lines
+                    notification_msg = "\n".join(notifications[:3])
+                    await self.send_func(notification_msg)
+                except Exception as e:
+                    logger.warning(f"AgentPipeline: Failed to send notification: {e}")
+            
+            # Increment round count after processing all tool calls in this round
+            if tasks_to_run:
+                session.round_count += 1
         
         # Build final response
         total_time = time.time() - start_time

hyw_core/definitions.py

@@ -46,6 +46,7 @@ def get_web_tool() -> Dict[str, Any]:
         "function": {
             "name": "web_tool",
             "description": """搜索网页或截图指定URL。用于获取最新信息、查找资料。
+## 使用方式
 网页搜索(大部分问题优先使用此方法): 
 直接传入搜索词如 "python async" 会返回搜索结果列表
 
@@ -84,10 +85,12 @@ AGENT_SYSTEM_PROMPT = """# 你是一个智能助手 (Agent), 你的职责是使
 最小限度使用自身知识, 尽可能使用 web_tool 获取信息.
 
 ## 工具使用指南
+- 并行调用工具
+    - 网页搜索: 可以同时调用3次, 其中URL截图消耗较大, 最多同时调用1个
 - 积极使用 web_tool 获取信息
     - 搜索时, 关键词保证简单、指向准确、利于传统搜索引擎.
     - 获取页面截图时, 只使用官方性较强的 wiki、官方网站、资源站等等, 不使用第三方转载新闻网站.
-- 最多可调用2次工具, 之后必须给出最终回答
+- 最多可调用2轮工具, 之后必须给出最终回答
 - 适当时候调用 `refuse_answer`
 - 对于具体任务, 如果是转述、格式化、翻译等, 请直接给出最终回答, 不再调用工具