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

add tavily

Browse source
This commit is contained in:
Dinesh Yeduguru 2024-12-20 21:01:41 -08:00
parent dcdf9da6ef
commit 9192a9bbb4
6 changed files with 163 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

37
llama_stack/providers/tests/agents/fixtures.py
View file

@ -77,6 +77,13 @@ def tool_runtime_memory() -> ProviderFixture:
"api_key": os.environ["BRAVE_SEARCH_API_KEY"],
},
),
Provider(
provider_id="tavily-search",
provider_type="inline::tavily-search",
config={
"api_key": os.environ["TAVILY_SEARCH_API_KEY"],
},
),
],
)
@ -146,13 +153,41 @@ async def agents_stack(request, inference_model, safety_shield):
ToolDef(
name="brave_search",
description="brave_search",
parameters=[],
parameters=[
ToolParameter(
name="query",
description="query",
parameter_type="string",
required=True,
),
],
metadata={},
),
],
),
provider_id="brave-search",
),
ToolGroupInput(
tool_group_id="tavily_search_group",
tool_group=UserDefinedToolGroupDef(
tools=[
ToolDef(
name="tavily_search",
description="tavily_search",
parameters=[
ToolParameter(
name="query",
description="query",
parameter_type="string",
required=True,
),
],
metadata={},
),
],
),
provider_id="tavily-search",
),
ToolGroupInput(
tool_group_id="memory_group",
tool_group=UserDefinedToolGroupDef(