phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-03 09:21:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

move brave and tavily to remote

Browse source
This commit is contained in:
Dinesh Yeduguru 2024-12-24 11:45:36 -08:00
parent 4dd2f4c363
commit c76f5f418f
8 changed files with 23 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -17,14 +17,6 @@ from llama_stack.providers.datatypes import (
def available_providers() -> List[ProviderSpec]:
return [
InlineProviderSpec(
api=Api.tool_runtime,
provider_type="inline::brave-search",
pip_packages=[],
module="llama_stack.providers.inline.tool_runtime.brave_search",
config_class="llama_stack.providers.inline.tool_runtime.brave_search.config.BraveSearchToolConfig",
provider_data_validator="llama_stack.providers.inline.tool_runtime.brave_search.BraveSearchToolProviderDataValidator",
),
InlineProviderSpec(
api=Api.tool_runtime,
provider_type="inline::memory-runtime",
@ -33,14 +25,6 @@ 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",
),
InlineProviderSpec(
api=Api.tool_runtime,
provider_type="inline::code-interpreter",
@ -48,6 +32,26 @@ def available_providers() -> List[ProviderSpec]:
module="llama_stack.providers.inline.tool_runtime.code_interpreter",
config_class="llama_stack.providers.inline.tool_runtime.code_interpreter.config.CodeInterpreterToolConfig",
),
remote_provider_spec(
api=Api.tool_runtime,
adapter=AdapterSpec(
adapter_type="brave-search",
module="llama_stack.providers.remote.tool_runtime.brave_search",
config_class="llama_stack.providers.remote.tool_runtime.brave_search.config.BraveSearchToolConfig",
pip_packages=["requests"],
provider_data_validator="llama_stack.providers.remote.tool_runtime.brave_search.BraveSearchToolProviderDataValidator",
),
),
remote_provider_spec(
api=Api.tool_runtime,
adapter=AdapterSpec(
adapter_type="tavily-search",
module="llama_stack.providers.remote.tool_runtime.tavily_search",
config_class="llama_stack.providers.remote.tool_runtime.tavily_search.config.TavilySearchToolConfig",
pip_packages=["requests"],
provider_data_validator="llama_stack.providers.remote.tool_runtime.tavily_search.TavilySearchToolProviderDataValidator",
),
),
remote_provider_spec(
api=Api.tool_runtime,
adapter=AdapterSpec(

2
llama_stack/providers/inline/tool_runtime/brave_search/__init__.py → llama_stack/providers/remote/tool_runtime/brave_search/__init__.py
View file

@ -14,7 +14,7 @@ class BraveSearchToolProviderDataValidator(BaseModel):
api_key: str
async def get_provider_impl(config: BraveSearchToolConfig, _deps):
async def get_adapter_impl(config: BraveSearchToolConfig, _deps):
impl = BraveSearchToolRuntimeImpl(config)
await impl.initialize()
return impl

0
llama_stack/providers/inline/tool_runtime/brave_search/brave_search.py → llama_stack/providers/remote/tool_runtime/brave_search/brave_search.py
View file

0
llama_stack/providers/inline/tool_runtime/brave_search/config.py → llama_stack/providers/remote/tool_runtime/brave_search/config.py
View file

2
llama_stack/providers/inline/tool_runtime/tavily_search/__init__.py → llama_stack/providers/remote/tool_runtime/tavily_search/__init__.py
View file

@ -14,7 +14,7 @@ class TavilySearchToolProviderDataValidator(BaseModel):
api_key: str
async def get_provider_impl(config: TavilySearchToolConfig, _deps):
async def get_adapter_impl(config: TavilySearchToolConfig, _deps):
impl = TavilySearchToolRuntimeImpl(config)
await impl.initialize()
return impl

0
llama_stack/providers/inline/tool_runtime/tavily_search/config.py → llama_stack/providers/remote/tool_runtime/tavily_search/config.py
View file

0
llama_stack/providers/inline/tool_runtime/tavily_search/tavily_search.py → llama_stack/providers/remote/tool_runtime/tavily_search/tavily_search.py
View file

2
tests/client-sdk/agents/test_agents.py
View file

@ -191,7 +191,7 @@ def test_builtin_tool_code_execution(llama_stack_client, agent_config):
messages=[
{
"role": "user",
"content": "Write code to answer the question: What is the 100th prime number?",
"content": "Write code and execute it to find the answer for: What is the 100th prime number?",
},
],
session_id=session_id,