phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2026-01-02 18:10:02 +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