phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-23 04:32:25 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add wolfram alpha, bing search

Browse source
This commit is contained in:
Dinesh Yeduguru 2025-01-07 16:07:03 -08:00
parent f9a98c278a
commit 94cca7a72a
14 changed files with 411 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

22
llama_stack/providers/tests/tools/fixtures.py
View file

@ -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,
],
)