entari-plugin-hyw 4.0.0rc5__py3-none-any.whl → 4.0.0rc7__py3-none-any.whl

entari_plugin_hyw/stage_instruct.py

@@ -8,7 +8,7 @@ Analyze user query and execute initial searches.
 import json
 import time
 import asyncio
-from typing import Any, Dict, List, Optional, Tuple
+from typing import Any, Dict, List, Optional, Tuple, Callable, Awaitable
 from loguru import logger
 from openai import AsyncOpenAI
 
@@ -17,6 +17,7 @@ from .definitions import (
     get_refuse_answer_tool,
     get_web_search_tool,
     get_crawl_page_tool,
+    get_set_mode_tool,
     INSTRUCT_SP
 )
 
@@ -25,13 +26,15 @@ class InstructStage(BaseStage):
     def name(self) -> str:
         return "Instruct"
     
-    def __init__(self, config: Any, search_service: Any, client: AsyncOpenAI):
+    def __init__(self, config: Any, search_service: Any, client: AsyncOpenAI, send_func: Optional[Callable[[str], Awaitable[None]]] = None):
         super().__init__(config, search_service, client)
+        self.send_func = send_func
         
         self.refuse_answer_tool = get_refuse_answer_tool()
         self.web_search_tool = get_web_search_tool()
         self.crawl_page_tool = get_crawl_page_tool()
-    
+        self.set_mode_tool = get_set_mode_tool()
+
     async def execute(self, context: StageContext) -> StageResult:
         start_time = time.time()
         
@@ -48,7 +51,7 @@ class InstructStage(BaseStage):
         # Execute Round 1 LLM
         r1_response, r1_usage, r1_tool_calls, r1_content = await self._call_llm(
             messages=r1_messages,
-            tools=[self.refuse_answer_tool, self.web_search_tool, self.crawl_page_tool],
+            tools=[self.refuse_answer_tool, self.web_search_tool, self.crawl_page_tool, self.set_mode_tool],
             tool_choice="auto"
         )
         
@@ -111,6 +114,7 @@ class InstructStage(BaseStage):
         model = model_cfg.get("model_name") or self.config.model_name
         
         try:
+            logger.info(f"Instruct: Sending LLM request to {model}...")
             response = await client.chat.completions.create(
                 model=model,
                 messages=messages,
@@ -167,7 +171,7 @@ class InstructStage(BaseStage):
                      "id": tc_id, "name": name, "content": f"Refused: {reason}"
                 })
             
-            elif name == "internal_web_search":
+            elif name == "web_search":
                 query = args.get("query")
                 if query: 
                     logger.info(f"Instruct: Planned search query -> '{query}'")
@@ -178,6 +182,26 @@ class InstructStage(BaseStage):
                 if url: 
                     logger.info(f"Instruct: Planned page crawl -> {url}")
                     pending_crawls.append((url, tc_id))
+            
+            elif name == "set_mode":
+                mode = args.get("mode", "fast")
+                if mode in ("fast", "deepsearch"):
+                    context.selected_mode = mode
+                    logger.info(f"Instruct: Mode set to '{mode}'")
+                    
+                    # Notify immediately if deepsearch
+                    if mode == "deepsearch" and self.send_func:
+                        try:
+                            await self.send_func("🔍 正在进行深度研究，可能需要一些时间，请耐心等待...")
+                        except Exception as e:
+                            logger.warning(f"Instruct: Failed to send notification: {e}")
+
+                    results_for_context.append({
+                        "id": tc_id, "name": name, "content": f"Mode set to: {mode}"
+                    })
+                else:
+                    logger.warning(f"Instruct: Invalid mode '{mode}', defaulting to 'fast'")
+                    context.selected_mode = "fast"
         
         # Execute Batches
         
@@ -189,7 +213,8 @@ class InstructStage(BaseStage):
             # Start fetch
             fetch_task = asyncio.create_task(self.search_service.fetch_pages_batch(urls))
             
-            is_image_mode = getattr(self.config, "page_content_mode", "text") == "image"
+            # Use image capability from context to determine content mode
+            is_image_mode = getattr(context, "image_input_supported", True)
             tab_ids = []
             if is_image_mode:
                 from .render_vue import get_content_renderer
@@ -285,6 +310,8 @@ class InstructStage(BaseStage):
                 visible_results = [r for r in web_results if not r.get("_hidden")]
                 
                 # Update global context
+                total_images = sum(len(item.get("images", []) or []) for item in web_results)
+                logger.debug(f"Instruct: Search '{query}' returned {len(web_results)} items with {total_images} images total")
                 for item in web_results:
                     item["_id"] = context.next_id()
                     if "type" in item:
@@ -301,7 +328,7 @@ class InstructStage(BaseStage):
                 
                 results_for_context.append({
                      "id": tc_id, 
-                     "name": "internal_web_search", 
+                     "name": "web_search", 
                      "content": summary
                 })
                 

entari_plugin_hyw/stage_instruct_deepsearch.py

@@ -0,0 +1,104 @@
+"""
+Instruct Deepsearch Stage
+
+Handles the deepsearch loop: Supplement information until sufficient or max iterations reached.
+Inherits from InstructStage to reuse tool execution logic.
+"""
+
+import time
+from typing import Any, List
+from loguru import logger
+from openai import AsyncOpenAI
+
+from .stage_base import StageContext, StageResult
+from .stage_instruct import InstructStage
+from .definitions import INSTRUCT_DEEPSEARCH_SP
+
+class InstructDeepsearchStage(InstructStage):
+    @property
+    def name(self) -> str:
+        return "Instruct Deepsearch"
+    
+    def __init__(self, config: Any, search_service: Any, client: AsyncOpenAI):
+        super().__init__(config, search_service, client)
+        # Inherits tools from InstructStage (web_search, crawl_page)
+    
+    async def execute(self, context: StageContext) -> StageResult:
+        start_time = time.time()
+        logger.info("Instruct Deepsearch: Starting supplementary research")
+        
+        # Check if we have context to review
+        if not context.review_context:
+            logger.warning("Instruct Deepsearch: No context found. Skipping.")
+            return StageResult(
+                success=True, 
+                data={"reasoning": "Skipped due to missing context.", "should_stop": True}
+            )
+            
+        # Build System Prompt (Clean)
+        system_prompt = INSTRUCT_DEEPSEARCH_SP
+        
+        # Build Messages
+        # Inject context as a separate user message explaining the background
+        context_message = f"## 已收集的信息\n\n```context\n{context.review_context}\n```"
+        
+        messages = [
+            {"role": "system", "content": system_prompt},
+            {"role": "user", "content": context_message},
+            {"role": "user", "content": self._build_user_message(context)}
+        ]
+        
+        # Call LLM
+        # We use only web_search and crawl_page tools (no set_mode, no refuse_answer in this stage)
+        tools = [self.web_search_tool, self.crawl_page_tool]
+        
+        response, usage, tool_calls, content = await self._call_llm(
+            messages=messages,
+            tools=tools,
+            tool_choice="auto"
+        )
+        
+        # Check for empty response = signal to stop
+        should_stop = False
+        if not tool_calls or len(tool_calls) == 0:
+            logger.info("Instruct Deepsearch: No tool calls, signaling to stop loop.")
+            should_stop = True
+        else:
+            # Execute Tools
+            tool_outputs = await self._process_tool_calls(context, tool_calls)
+            
+            # Update context for next iteration
+            iteration_summary = f"\n## Deepsearch Iteration\n"
+            if content:
+                iteration_summary += f"Thought: {content}\n"
+            for output in tool_outputs:
+                iteration_summary += f"- {output['name']}: {output['content'][:200]}...\n"
+            context.review_context += iteration_summary
+            
+            # Update history
+            context.instruct_history.append({
+                "role": "assistant",
+                "content": f"[Deepsearch]: {content}\n[Actions]: {len(tool_outputs)} tools"
+            })
+        
+        return self._build_result(start_time, usage, content, len(tool_calls or []), should_stop)
+
+    def _build_result(self, start_time, usage, content, tool_calls_count, should_stop=False):
+        model_cfg = self.config.get_model_config("instruct")
+        model = model_cfg.get("model_name") or self.config.model_name
+
+        trace = {
+            "stage": "Instruct Deepsearch",
+            "model": model, 
+            "usage": usage,
+            "output": content,
+            "tool_calls": tool_calls_count,
+            "time": time.time() - start_time,
+        }
+        
+        return StageResult(
+            success=True,
+            data={"reasoning": content, "should_stop": should_stop},
+            usage=usage,
+            trace=trace
+        )

entari_plugin_hyw/stage_summary.py

@@ -47,6 +47,11 @@ class SummaryStage(BaseStage):
         # Build Context Message
         context_message = f"## Web Search & Page Content\n\n```context\n{full_context}\n```"
         
+        # Add vision description if present (from VisionStage)
+        if context.vision_description:
+            vision_context = f"## 用户图片描述\n\n{context.vision_description}"
+            context_message = f"{vision_context}\n\n{context_message}"
+        
         # Build user content
         user_text = context.user_input or "..."
         if images:
@@ -104,6 +109,7 @@ class SummaryStage(BaseStage):
                 "provider": model_cfg.get("model_provider") or "Unknown",
                 "usage": usage,
                 "system_prompt": system_prompt,
+                "context_message": context_message,  # Includes vision description + search results
                 "output": content,
                 "time": time.time() - start_time,
                 "images_count": len(images) if images else 0,

entari_plugin_hyw/stage_vision.py

@@ -0,0 +1,113 @@
+"""
+Vision Stage
+
+Generates image description using a vision-capable model.
+The description is then passed as context to subsequent stages.
+"""
+
+import time
+from typing import Any, Dict, List, Optional
+
+from loguru import logger
+from openai import AsyncOpenAI
+
+from .stage_base import BaseStage, StageContext, StageResult
+from .definitions import VISION_DESCRIPTION_SP
+
+
+class VisionStage(BaseStage):
+    """
+    Vision Stage: Generate image description.
+    
+    Takes user images and text, calls a vision model to produce
+    a detailed description of the image content.
+    """
+    
+    @property
+    def name(self) -> str:
+        return "Vision"
+    
+    async def execute(
+        self, 
+        context: StageContext, 
+        images: List[str] = None
+    ) -> StageResult:
+        """Generate image description."""
+        start_time = time.time()
+        
+        if not images:
+            return StageResult(
+                success=True,
+                data={"description": ""},
+                trace={"skipped": True, "reason": "No images provided"}
+            )
+        
+        # Get model config for vision stage
+        model_cfg = self.config.get_model_config("vision")
+        model = model_cfg.get("model_name")
+        
+        if not model:
+            logger.warning("VisionStage: No vision model configured, skipping")
+            return StageResult(
+                success=True,
+                data={"description": ""},
+                trace={"skipped": True, "reason": "No vision model configured"}
+            )
+        
+        client = self._client_for(
+            api_key=model_cfg.get("api_key"),
+            base_url=model_cfg.get("base_url")
+        )
+        
+        # Build user content with images
+        user_text = context.user_input or "请描述这张图片"
+        user_content: List[Dict[str, Any]] = [{"type": "text", "text": user_text}]
+        
+        for img_b64 in images:
+            url = f"data:image/jpeg;base64,{img_b64}" if not img_b64.startswith("data:") else img_b64
+            user_content.append({"type": "image_url", "image_url": {"url": url}})
+        
+        messages = [
+            {"role": "system", "content": VISION_DESCRIPTION_SP},
+            {"role": "user", "content": user_content}
+        ]
+        
+        try:
+            logger.info(f"VisionStage: Calling model '{model}' with {len(images)} image(s)")
+            response = await client.chat.completions.create(
+                model=model,
+                messages=messages,
+                temperature=0.3,  # Lower temperature for factual description
+                extra_body=model_cfg.get("extra_body"),
+            )
+        except Exception as e:
+            logger.error(f"VisionStage LLM error: {e}")
+            return StageResult(
+                success=False,
+                error=str(e),
+                data={"description": ""},
+                trace={"error": str(e)}
+            )
+        
+        usage = {"input_tokens": 0, "output_tokens": 0}
+        if hasattr(response, "usage") and response.usage:
+            usage["input_tokens"] = getattr(response.usage, "prompt_tokens", 0) or 0
+            usage["output_tokens"] = getattr(response.usage, "completion_tokens", 0) or 0
+        
+        description = (response.choices[0].message.content or "").strip()
+        
+        logger.info(f"VisionStage: Generated description ({len(description)} chars)")
+        
+        return StageResult(
+            success=True,
+            data={"description": description},
+            usage=usage,
+            trace={
+                "model": model,
+                "provider": model_cfg.get("model_provider") or "Unknown",
+                "usage": usage,
+                "output": description,
+                "time": time.time() - start_time,
+                "images_count": len(images),
+            }
+        )

{entari_plugin_hyw-4.0.0rc5.dist-info → entari_plugin_hyw-4.0.0rc7.dist-info}/METADATA

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

{entari_plugin_hyw-4.0.0rc5.dist-info → entari_plugin_hyw-4.0.0rc7.dist-info}/RECORD

@@ -1,16 +1,17 @@
-entari_plugin_hyw/__init__.py,sha256=502FHKMiXsq7RoOfQs1RiY6B9-1JMnT1caDGCy3mxDU,20073
-entari_plugin_hyw/definitions.py,sha256=QKvhk8q6Wa22QlF72HsoikbdNxfveoZJR71zpkxTiLQ,5912
-entari_plugin_hyw/history.py,sha256=6d1e8DPfv4XQyM9hMjrx08_WKrxqVSJEvSIGRX4JOB0,11085
+entari_plugin_hyw/__init__.py,sha256=mh0aArXgQxFMNa3XUN2bfYv2IORn1CODNRRwKKz8oi0,22950
+entari_plugin_hyw/definitions.py,sha256=WOTweqy-qZm6Y3jJblETGqYGVDWJs5ztIuDepOuCdMM,7288
+entari_plugin_hyw/history.py,sha256=d3lsQevhtq0PbPR_6n53q0-kqmPwPGvQp8yo3dmyAbw,11343
 entari_plugin_hyw/image_cache.py,sha256=t8pr1kgH2ngK9IhrBAhzUqhBWERNztUywMzgCFZEtQk,9899
-entari_plugin_hyw/misc.py,sha256=b0KPCUrf1GuCY0Mx29Sn5sREg1fxR1OyYEl4O4Omsys,4553
-entari_plugin_hyw/modular_pipeline.py,sha256=QjuvmM6d1fYudyzK3a7tlZS24zpUlFRwo21Skn_gVW0,14798
+entari_plugin_hyw/misc.py,sha256=ZGDXeXbSugG4tRrBfUDVd94i2SCaPOmaGtVmPz8mqtY,5413
+entari_plugin_hyw/modular_pipeline.py,sha256=M8wmpfDuGUpPEf-Fvp9YcefzEaK5MiT8czXacs9QJEg,21747
 entari_plugin_hyw/render_vue.py,sha256=4i5xTZCb9amPgSWo6f7Ev279ZOk-D8Kfmxe2HdcA_vI,14737
-entari_plugin_hyw/search.py,sha256=oAC9vmpSBwvOHkf4mBSqgvN5W9USsJT5Znubn41llYk,4826
-entari_plugin_hyw/stage_base.py,sha256=-A8CArfnL7eDaLv-DCv-mjtSkMrdowlHq0w_gg8btQ4,2625
-entari_plugin_hyw/stage_instruct.py,sha256=7b_3vGB1KQ7Ek3l55_OmLgVgARXYDHcVqWYWVBPo4tg,13676
-entari_plugin_hyw/stage_instruct_review.py,sha256=e0yblVnEI-11dwZ-H5ydctx9O5PejYLNpkGWyHc1M1s,3392
-entari_plugin_hyw/stage_summary.py,sha256=1g1tL7alHbDdp6Mrq2HrXD8k65VMOL-zdyo3bWAS5RQ,5736
-entari_plugin_hyw/assets/card-dist/index.html,sha256=s_RU1rCFowRwFMTuIiUsyGp25WfsIyk-XNWyD9jYqVc,2204580
+entari_plugin_hyw/search.py,sha256=rcEUpE7AVUerbQfMQA2X00KTlf2NN_nz3kBLky2eP2k,7162
+entari_plugin_hyw/stage_base.py,sha256=YvHU2bYRRKjjaJXbKUyHCRKfeZbULqIUyCwfxxHuHUM,2834
+entari_plugin_hyw/stage_instruct.py,sha256=qHJpG6vnQQaNItX4ihGEQtEwKpobciBLjAzSFynngfE,15243
+entari_plugin_hyw/stage_instruct_deepsearch.py,sha256=_I_xZ-M1xovM4I7NL5eyc21Wg9xdFKnJDQVQActTjUI,3969
+entari_plugin_hyw/stage_summary.py,sha256=o5SjgWuWEG__oKLot5sDl6gB8E_6cdOiYQ6FYbp9ncs,6104
+entari_plugin_hyw/stage_vision.py,sha256=fbudjmzcS3lfOjRKqNwzz1AGDrxHJBiaWVXlQfy1_S4,3864
+entari_plugin_hyw/assets/card-dist/index.html,sha256=jiRhPVQQDv3dgEQ-pqPUm0R2wA6wlInI8pLJyTQN1wM,2205760
 entari_plugin_hyw/assets/card-dist/vite.svg,sha256=SnSK_UQ5GLsWWRyDTEAdrjPoeGGrXbrQgRw6O0qSFPs,1497
 entari_plugin_hyw/assets/card-dist/logos/anthropic.svg,sha256=ASsy1ypo3osNc3n-B0R81tk_dIFsVgg7qQORrd5T2kA,558
 entari_plugin_hyw/assets/card-dist/logos/cerebras.svg,sha256=bpmiiYTODwc06knTmPj3GQ7NNtosMog5lkggvB_Z-7M,44166
@@ -50,11 +51,13 @@ entari_plugin_hyw/assets/icon/xiaomi.png,sha256=WHxlDFGU5FCjb-ure3ngdGG18-efYZUU
 entari_plugin_hyw/assets/icon/zai.png,sha256=K-gnabdsjMLInppHA1Op7Nyt33iegrx1x-yNlvCZ0Tc,2351
 entari_plugin_hyw/browser/__init__.py,sha256=Cht-i5MowfAdmfW3kiY4sV7oXKDb-DmhZ-_eKwDl6r0,321
 entari_plugin_hyw/browser/landing.html,sha256=wgqldumdylz69T83pvOkrigT1Mdb9GY0_KU0ceLGwdY,4642
-entari_plugin_hyw/browser/manager.py,sha256=9VuTyXBr1QBcTV8Nj2vSrozstMmNxAytXESUQ6vxLrM,5234
-entari_plugin_hyw/browser/service.py,sha256=ekrHDEWesDZw2PLex9bGL6SMYxfElxEOAxIegRu_zEo,12013
+entari_plugin_hyw/browser/manager.py,sha256=aT1DQTDC5WMYUEt5678Xbb5XkFEFYwgFb3XrGDbB0Rc,5233
+entari_plugin_hyw/browser/service.py,sha256=cqvS-O7taTvOdvbnwI-6GlHmJpgfGglh6JqfQgYQvek,23040
 entari_plugin_hyw/browser/engines/base.py,sha256=q5y4SM1G6xS7-6TQ-nZz9iTWw3XonjJn01fWzoTxr6c,414
 entari_plugin_hyw/browser/engines/bing.py,sha256=rIWcvjzvm700xji_OBl6COUAtwXg87DcXQ97DlTzleA,3838
-entari_plugin_hyw/browser/engines/searxng.py,sha256=m58Vzj807-70mT4j63SjzGijS85-O06l-Mmd3M-sd7Y,5152
+entari_plugin_hyw/browser/engines/default.py,sha256=BlHCQI4-rN9cEzLLfqvRD4bvhyP2G2KUGlo92J4kFNw,6092
+entari_plugin_hyw/browser/engines/duckduckgo.py,sha256=jlCIXR1mpJ0LjSbuHMYNGMYjHiQ9Lw_x-A8242Zajgo,5154
+entari_plugin_hyw/browser/engines/google.py,sha256=vCtyOxr63F40hDMW70sS1CyoMsqc0HzyESYrK_qcZLg,6091
 entari_plugin_hyw/card-ui/.gitignore,sha256=_nGOe6uxTzy60tl_CIibnOUhXtP-DkOyuM-_s7m4ROg,253
 entari_plugin_hyw/card-ui/README.md,sha256=fN9IawCcxEcJ8LM-RfKiAH835fRqyY_iqrRsgSkxiSk,442
 entari_plugin_hyw/card-ui/index.html,sha256=Hd7vk8v8PtATbfiEWLYoKDpUT0dlyozW_K5gR_cObfo,328
@@ -83,7 +86,7 @@ entari_plugin_hyw/card-ui/public/logos/qwen.png,sha256=eqLbnIPbjh2_PsODU_mmqjeD8
 entari_plugin_hyw/card-ui/public/logos/xai.png,sha256=uSulvvDVqoA4RUOW0ZAkdvBVM2rpyGJRZIbn5dEFspw,362
 entari_plugin_hyw/card-ui/public/logos/xiaomi.png,sha256=WHxlDFGU5FCjb-ure3ngdGG18-efYZUUfqA3_lqCUN0,4084
 entari_plugin_hyw/card-ui/public/logos/zai.png,sha256=K-gnabdsjMLInppHA1Op7Nyt33iegrx1x-yNlvCZ0Tc,2351
-entari_plugin_hyw/card-ui/src/App.vue,sha256=ofRlf3hTSqDg1l8jgnFwzWg1dBzQVVg5MLOaMW01z68,28858
+entari_plugin_hyw/card-ui/src/App.vue,sha256=yuZtrI3PeY8F7WvsiF7PUFkWKTKa5wRCnyC70cuXgik,30693
 entari_plugin_hyw/card-ui/src/main.ts,sha256=rm653lPnK5fuTIj-iNLpgr8GAmayuCoKop7IWfo0IBk,111
 entari_plugin_hyw/card-ui/src/style.css,sha256=Qm0sv9em6k4VI_j5PI0_E_ng6u20eFpf2lN44H19zzc,671
 entari_plugin_hyw/card-ui/src/test_regex.js,sha256=cWmclm6LRKYfjeN1RT5HECdltmo1HvS2BwGCYY_4l14,3040
@@ -93,7 +96,7 @@ entari_plugin_hyw/card-ui/src/components/HelloWorld.vue,sha256=yvBIzJua9BfikUOR1
 entari_plugin_hyw/card-ui/src/components/MarkdownContent.vue,sha256=SV95Vuj99tQN2yrU9GqiyhiemWAW8omhYnS8AsH1YIU,13325
 entari_plugin_hyw/card-ui/src/components/SectionCard.vue,sha256=owcDNx2JYVmF2J5SYCroR2gvg_cPApQsNunjK1WJpVI,1433
 entari_plugin_hyw/card-ui/src/components/StageCard.vue,sha256=MgpOaBlPR--LJoRenN37i72BV8qVgzDdurpoKCvzKyk,11133
-entari_plugin_hyw-4.0.0rc5.dist-info/METADATA,sha256=oe3M_DRBIEBJu8n90HXBsK7CuwahB8DanrFdx2MO7bA,3766
-entari_plugin_hyw-4.0.0rc5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-entari_plugin_hyw-4.0.0rc5.dist-info/top_level.txt,sha256=TIDsn6XPs6KA5e3ezsE65JoXsy03ejDdrB41I4SPjmo,18
-entari_plugin_hyw-4.0.0rc5.dist-info/RECORD,,
+entari_plugin_hyw-4.0.0rc7.dist-info/METADATA,sha256=4wYIdgmoSZRZ4VFfd14jVS5_3kHjvHik_4zPUnJPZrs,3766
+entari_plugin_hyw-4.0.0rc7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+entari_plugin_hyw-4.0.0rc7.dist-info/top_level.txt,sha256=TIDsn6XPs6KA5e3ezsE65JoXsy03ejDdrB41I4SPjmo,18
+entari_plugin_hyw-4.0.0rc7.dist-info/RECORD,,

entari_plugin_hyw/stage_instruct_review.py

@@ -1,92 +0,0 @@
-"""
-Instruct Review Stage
-
-Handles the second round of instruction: Review and Refine.
-Inherits from InstructStage to reuse tool execution logic.
-"""
-
-import time
-from typing import Any, List
-from loguru import logger
-from openai import AsyncOpenAI
-
-from .stage_base import StageContext, StageResult
-from .stage_instruct import InstructStage
-from .definitions import INSTRUCT_REVIEW_SP
-
-class InstructReviewStage(InstructStage):
-    @property
-    def name(self) -> str:
-        return "Instruct Review"
-    
-    def __init__(self, config: Any, search_service: Any, client: AsyncOpenAI):
-        super().__init__(config, search_service, client)
-        # Inherits tools from InstructStage
-    
-    async def execute(self, context: StageContext) -> StageResult:
-        start_time = time.time()
-        logger.info("Instruct Review: Starting Round 2 (Review & Refine)")
-        
-        # Check if we have context to review
-        if not context.review_context:
-            logger.warning("Instruct Review: No context found from Round 1. Skipping.")
-            return StageResult(success=True, data={"reasoning": "Skipped due to missing context."})
-            
-        # Build System Prompt (Clean)
-        system_prompt = INSTRUCT_REVIEW_SP
-        
-        # Build Messages
-        # Inject context as a separate user message explaining the background
-        context_message = f"## Previous Round Context\n\n```context\n{context.review_context}\n```"
-        
-        messages = [
-            {"role": "system", "content": system_prompt},
-            {"role": "user", "content": context_message},
-            {"role": "user", "content": self._build_user_message(context)}
-        ]
-        
-        # Call LLM
-        # We reuse _call_llm from parent
-        # We reuse tools from parent (refuse_answer might be redundant but harmless, or we can filter)
-        tools = [self.web_search_tool, self.crawl_page_tool] # Review prompt doesn't mention refuse_answer explicitly, but usually fine.
-        
-        response, usage, tool_calls, content = await self._call_llm(
-            messages=messages,
-            tools=tools,
-            tool_choice="auto"
-        )
-        
-        # Execute Tools
-        tool_outputs = []
-        if tool_calls:
-            tool_outputs = await self._process_tool_calls(context, tool_calls)
-            
-        # Update history logic? 
-        # The prompt says "上下文". It is "independent". 
-        # But for the record, we might want to log it.
-        context.instruct_history.append({
-             "role": "assistant",
-             "content": f"[Round 2 Review]: {content}\n[Round 2 Actions]: {len(tool_outputs)} tools"
-        })
-        
-        return self._build_result(start_time, usage, content, len(tool_calls or []))
-
-    def _build_result(self, start_time, usage, content, tool_calls_count):
-        model_cfg = self.config.get_model_config("instruct")
-        model = model_cfg.get("model_name") or self.config.model_name
-
-        trace = {
-            "stage": "Instruct Review",
-            "model": model, 
-            "usage": usage,
-            "output": content,
-            "tool_calls": tool_calls_count,
-            "time": time.time() - start_time,
-        }
-        
-        return StageResult(
-            success=True,
-            data={"reasoning": content},
-            usage=usage,
-            trace=trace
-        )