phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-22 22:19:49 +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

8
llama_stack/providers/registry/tool_runtime.py
View file

@ -33,6 +33,14 @@ def available_providers() -> List[ProviderSpec]:
config_class="llama_stack.providers.inline.tool_runtime.memory.config.MemoryToolConfig",
api_dependencies=[Api.memory, Api.memory_banks, Api.inference],
),
InlineProviderSpec(
api=Api.tool_runtime,
provider_type="inline::tavily-search",
pip_packages=[],
module="llama_stack.providers.inline.tool_runtime.tavily_search",
config_class="llama_stack.providers.inline.tool_runtime.tavily_search.config.TavilySearchToolConfig",
provider_data_validator="llama_stack.providers.inline.tool_runtime.tavily_search.TavilySearchToolProviderDataValidator",
),
remote_provider_spec(
api=Api.tool_runtime,
adapter=AdapterSpec(