enh(agents/openai): Reuse smolagents tools. (#14)

* enh(agents/openai): Reuse smolagents tools.

- Use wrapped DuckDuckGoSearchTool and VisitWebpageTool by default.

Generates an actually usable trace compared to openai's `WebSearchTool`.

* fix(test_unit_openai): Use `ANY` in `tools` assertion.

src/surf_spot_finder/agents/openai.py

@@ -47,8 +47,29 @@ def run_openai_agent(
         AsyncOpenAI,
         OpenAIChatCompletionsModel,
         Runner,
-        WebSearchTool,
+        function_tool,
     )
+    from smolagents import DuckDuckGoSearchTool, VisitWebpageTool
+
+    @function_tool
+    def search_web(query: str) -> str:
+        """Performs a duckduckgo web search based on your query (think a Google search) then returns the top search results.
+
+        Args:
+            query: The search query to perform.
+        """
+        search_tool = DuckDuckGoSearchTool()
+        return search_tool.forward(query)
+
+    @function_tool
+    def visit_webpage(url: str) -> str:
+        """Visits a webpage at the given url and reads its content as a markdown string. Use this to browse webpages.
+
+        Args:
+            url: The url of the webpage to visit.
+        """
+        visit_tool = VisitWebpageTool()
+        return visit_tool.forward(url)
 
     if api_key_var and base_url:
         external_client = AsyncOpenAI(
@@ -62,14 +83,14 @@ def run_openai_agent(
                 model=model_id,
                 openai_client=external_client,
             ),
-            tools=[WebSearchTool()],
+            tools=[search_web, visit_webpage],
         )
     else:
         agent = Agent(
             model=model_id,
             instructions=instructions,
             name=name,
-            tools=[WebSearchTool()],
+            tools=[search_web, visit_webpage],
         )
     result = Runner.run_sync(agent, prompt)
     logger.info(result.final_output)

tests/unit/agents/test_unit_openai.py

@@ -1,6 +1,6 @@
 import os
 import pytest
-from unittest.mock import patch, MagicMock
+from unittest.mock import patch, MagicMock, ANY
 
 from surf_spot_finder.agents.openai import run_openai_agent
 
@@ -32,9 +32,7 @@ def test_run_openai_agent_default(mock_agents_module):
         model="gpt-4o",
         instructions=None,
         name="surf-spot-finder",
-        tools=[
-            mock_agents_module["WebSearchTool"].return_value,
-        ],
+        tools=ANY,
     )