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