phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2026-01-01 08:04:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add provider data passing for library client

Browse source
This commit is contained in:
Dinesh Yeduguru 2025-01-13 13:25:36 -08:00
parent 78727aad26
commit 33b4e8df49
10 changed files with 49 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/providers/remote/tool_runtime/bing_search/__init__.py
View file

@ -12,7 +12,7 @@ from pydantic import BaseModel
class BingSearchToolProviderDataValidator(BaseModel):
api_key: str
bing_search_api_key: str
async def get_adapter_impl(config: BingSearchToolConfig, _deps):

6
llama_stack/providers/remote/tool_runtime/bing_search/bing_search.py
View file

@ -44,11 +44,11 @@ class BingSearchToolRuntimeImpl(
return self.config.api_key
provider_data = self.get_request_provider_data()
if provider_data is None or not provider_data.api_key:
if provider_data is None or not provider_data.bing_search_api_key:
raise ValueError(
'Pass Bing Search API Key in the header X-LlamaStack-Provider-Data as { "api_key": <your api key>}'
'Pass Bing Search API Key in the header X-LlamaStack-Provider-Data as { "bing_search_api_key": <your api key>}'
)
return provider_data.api_key
return provider_data.bing_search_api_key
async def list_runtime_tools(
self, tool_group_id: Optional[str] = None, mcp_endpoint: Optional[URL] = None

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

@ -11,7 +11,7 @@ from .config import BraveSearchToolConfig
class BraveSearchToolProviderDataValidator(BaseModel):
api_key: str
brave_search_api_key: str
async def get_adapter_impl(config: BraveSearchToolConfig, _deps):

6
llama_stack/providers/remote/tool_runtime/brave_search/brave_search.py
View file

@ -43,11 +43,11 @@ class BraveSearchToolRuntimeImpl(
return self.config.api_key
provider_data = self.get_request_provider_data()
if provider_data is None or not provider_data.api_key:
if provider_data is None or not provider_data.brave_search_api_key:
raise ValueError(
'Pass Search provider\'s API Key in the header X-LlamaStack-Provider-Data as { "api_key": <your api key>}'
'Pass Search provider\'s API Key in the header X-LlamaStack-Provider-Data as { "brave_search_api_key": <your api key>}'
)
return provider_data.api_key
return provider_data.brave_search_api_key
async def list_runtime_tools(
self, tool_group_id: Optional[str] = None, mcp_endpoint: Optional[URL] = None

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

@ -11,7 +11,7 @@ from .tavily_search import TavilySearchToolRuntimeImpl
class TavilySearchToolProviderDataValidator(BaseModel):
api_key: str
tavily_search_api_key: str
async def get_adapter_impl(config: TavilySearchToolConfig, _deps):

6
llama_stack/providers/remote/tool_runtime/tavily_search/tavily_search.py
View file

@ -43,11 +43,11 @@ class TavilySearchToolRuntimeImpl(
return self.config.api_key
provider_data = self.get_request_provider_data()
if provider_data is None or not provider_data.api_key:
if provider_data is None or not provider_data.tavily_search_api_key:
raise ValueError(
'Pass Search provider\'s API Key in the header X-LlamaStack-Provider-Data as { "api_key": <your api key>}'
'Pass Search provider\'s API Key in the header X-LlamaStack-Provider-Data as { "tavily_search_api_key": <your api key>}'
)
return provider_data.api_key
return provider_data.tavily_search_api_key
async def list_runtime_tools(
self, tool_group_id: Optional[str] = None, mcp_endpoint: Optional[URL] = None

2
llama_stack/providers/remote/tool_runtime/wolfram_alpha/__init__.py
View file

@ -13,7 +13,7 @@ __all__ = ["WolframAlphaToolConfig", "WolframAlphaToolRuntimeImpl"]
class WolframAlphaToolProviderDataValidator(BaseModel):
api_key: str
wolfram_alpha_api_key: str
async def get_adapter_impl(config: WolframAlphaToolConfig, _deps):

6
llama_stack/providers/remote/tool_runtime/wolfram_alpha/wolfram_alpha.py
View file

@ -44,11 +44,11 @@ class WolframAlphaToolRuntimeImpl(
return self.config.api_key
provider_data = self.get_request_provider_data()
if provider_data is None or not provider_data.api_key:
if provider_data is None or not provider_data.wolfram_alpha_api_key:
raise ValueError(
'Pass WolframAlpha API Key in the header X-LlamaStack-Provider-Data as { "api_key": <your api key>}'
'Pass WolframAlpha API Key in the header X-LlamaStack-Provider-Data as { "wolfram_alpha_api_key": <your api key>}'
)
return provider_data.api_key
return provider_data.wolfram_alpha_api_key
async def list_runtime_tools(
self, tool_group_id: Optional[str] = None, mcp_endpoint: Optional[URL] = None