add wolfram alpha, bing search

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."""