Rename _search_provider -> _search and _image_provider -> _image extensions

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

Author: Jeomon

operator_use/agent/service.py

@@ -111,8 +111,8 @@ def __init__(
         self.tool_register.set_extension("_process_store", self.process_store)
         self.tool_register.set_extension("_acp_registry", acp_registry or {})
         self.tool_register.set_extension("_llm", self.llm)
-        self.tool_register.set_extension("_image_provider", image)
-        self.tool_register.set_extension("_search_provider", search)
+        self.tool_register.set_extension("_image", image)
+        self.tool_register.set_extension("_search", search)
         self.tool_register.set_extension("_agent", self)
         self.tool_register.set_extension("_agent_id", self.agent_id)
 

operator_use/agent/tools/builtin/imagegen.py

@@ -81,7 +81,7 @@ async def imagegen(
     send_result: bool = True,
     **kwargs,
 ) -> ToolResult:
-    provider = kwargs.get("_image_provider")
+    provider = kwargs.get("_image")
     if provider is None:
         return ToolResult.error_result(
             "No image provider is configured. "

operator_use/agent/tools/builtin/web.py

@@ -35,7 +35,7 @@ class WebSearch(BaseModel):
 
 @Tool(name="web_search", description="Search the web and return titles, URLs, and snippets. Use for current events, documentation, package info, or error messages. Follow up with web_fetch to read the full content of a result.", model=WebSearch)
 async def web_search(query: str, max_results: int = 10, **kwargs) -> ToolResult:
-    provider = kwargs.get("_search_provider")
+    provider = kwargs.get("_search")
     if provider is None:
         from operator_use.providers.ddgs import DDGSSearch
         provider = DDGSSearch()
@@ -87,7 +87,7 @@ async def web_fetch(url: str, prompt: str | None = None, timeout: int = 10, **kw
     if not parsed_url.scheme or parsed_url.scheme not in ["http", "https"]:
         return ToolResult.error_result(f"Invalid URL: {url}. Must be http:// or https://")
 
-    provider = kwargs.get("_search_provider")
+    provider = kwargs.get("_search")
     if provider is None:
         from operator_use.providers.ddgs import DDGSSearch
         provider = DDGSSearch()