infoquest support image-search (#1255)

backend/packages/harness/deerflow/community/infoquest/tools.py

@@ -13,6 +13,7 @@ def _get_infoquest_client() -> InfoQuestClient:
     search_time_range = -1
     if search_config is not None and "search_time_range" in search_config.model_extra:
         search_time_range = search_config.model_extra.get("search_time_range")
+        
     fetch_config = get_app_config().get_tool_config("web_fetch")
     fetch_time = -1
     if fetch_config is not None and "fetch_time" in fetch_config.model_extra:
@@ -23,12 +24,24 @@ def _get_infoquest_client() -> InfoQuestClient:
     navigation_timeout = -1
     if fetch_config is not None and "navigation_timeout" in fetch_config.model_extra:
         navigation_timeout = fetch_config.model_extra.get("navigation_timeout")
+        
+    image_search_config = get_app_config().get_tool_config("image_search")
+    image_search_time_range = -1
+    if image_search_config is not None and "image_search_time_range" in image_search_config.model_extra:
+        image_search_time_range = image_search_config.model_extra.get("image_search_time_range")
+    image_size = "i"
+    if image_search_config is not None and "image_size" in image_search_config.model_extra:
+        image_size = image_search_config.model_extra.get("image_size")
+    
+    
 
     return InfoQuestClient(
         search_time_range=search_time_range,
         fetch_timeout=fetch_timeout,
         fetch_navigation_timeout=navigation_timeout,
         fetch_time=fetch_time,
+        image_search_time_range=image_search_time_range,
+        image_size=image_size,
     )
 
 
@@ -61,3 +74,22 @@ def web_fetch_tool(url: str) -> str:
         return result
     article = readability_extractor.extract_article(result)
     return article.to_markdown()[:4096]
+
+
+@tool("image_search", parse_docstring=True)
+def image_search_tool(query: str) -> str:
+    """Search for images online. Use this tool BEFORE image generation to find reference images for characters, portraits, objects, scenes, or any content requiring visual accuracy.
+
+    **When to use:**
+    - Before generating character/portrait images: search for similar poses, expressions, styles
+    - Before generating specific objects/products: search for accurate visual references
+    - Before generating scenes/locations: search for architectural or environmental references
+    - Before generating fashion/clothing: search for style and detail references
+
+    The returned image URLs can be used as reference images in image generation to significantly improve quality.
+
+    Args:
+        query: The query to search for images.
+    """
+    client = _get_infoquest_client()
+    return client.image_search(query)