add wolfram alpha, bing search

llama_stack/providers/tests/tools/fixtures.py

@@ -33,6 +33,13 @@ def tool_runtime_memory_and_search() -> ProviderFixture:
                     "api_key": os.environ["TAVILY_SEARCH_API_KEY"],
                 },
             ),
+            Provider(
+                provider_id="wolfram-alpha",
+                provider_type="remote::wolfram-alpha",
+                config={
+                    "api_key": os.environ["WOLFRAM_ALPHA_API_KEY"],
+                },
+            ),
         ],
     )
 
@@ -53,12 +60,24 @@ def tool_group_input_tavily_search() -> ToolGroupInput:
     )
 
 
+@pytest.fixture(scope="session")
+def tool_group_input_wolfram_alpha() -> ToolGroupInput:
+    return ToolGroupInput(
+        toolgroup_id="builtin::wolfram_alpha",
+        provider_id="wolfram-alpha",
+    )
+
+
 TOOL_RUNTIME_FIXTURES = ["memory_and_search"]
 
 
 @pytest_asyncio.fixture(scope="session")
 async def tools_stack(
-    request, inference_model, tool_group_input_memory, tool_group_input_tavily_search
+    request,
+    inference_model,
+    tool_group_input_memory,
+    tool_group_input_tavily_search,
+    tool_group_input_wolfram_alpha,
 ):
     fixture_dict = request.param
 
@@ -104,6 +123,7 @@ async def tools_stack(
         models=models,
         tool_groups=[
             tool_group_input_tavily_search,
+            tool_group_input_wolfram_alpha,
             tool_group_input_memory,
         ],
     )

llama_stack/providers/tests/tools/test_tools.py

@@ -20,6 +20,11 @@ def sample_search_query():
     return "What are the latest developments in quantum computing?"
 
 
+@pytest.fixture
+def sample_wolfram_alpha_query():
+    return "What is the square root of 16?"
+
+
 @pytest.fixture
 def sample_documents():
     urls = [
@@ -61,6 +66,24 @@ class TestTools:
         assert len(response.content) > 0
         assert isinstance(response.content, str)
 
+    @pytest.mark.asyncio
+    async def test_wolfram_alpha_tool(self, tools_stack, sample_wolfram_alpha_query):
+        """Test the wolfram alpha tool functionality."""
+        if "WOLFRAM_ALPHA_API_KEY" not in os.environ:
+            pytest.skip("WOLFRAM_ALPHA_API_KEY not set, skipping test")
+
+        tools_impl = tools_stack.impls[Api.tool_runtime]
+
+        response = await tools_impl.invoke_tool(
+            tool_name="wolfram_alpha", args={"query": sample_wolfram_alpha_query}
+        )
+
+        # Verify the response
+        assert isinstance(response, ToolInvocationResult)
+        assert response.content is not None
+        assert len(response.content) > 0
+        assert isinstance(response.content, str)
+
     @pytest.mark.asyncio
     async def test_memory_tool(self, tools_stack, sample_documents):
         """Test the memory tool functionality."""