phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-23 02:09:40 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add X-LlamaStack-Client-Version, rename ProviderData -> Provider-Data

Browse source
This commit is contained in:
Ashwin Bharambe 2025-01-08 21:21:51 -08:00
parent a5c57cd381
commit 9b45a2f0a0
13 changed files with 1281 additions and 152 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/providers/inline/scoring/braintrust/braintrust.py
View file

@ -156,7 +156,7 @@ class BraintrustScoringImpl(
provider_data = self.get_request_provider_data()
if provider_data is None or not provider_data.openai_api_key:
raise ValueError(
'Pass OpenAI API Key in the header X-LlamaStack-ProviderData as { "openai_api_key": <your api key>}'
'Pass OpenAI API Key in the header X-LlamaStack-Provider-Data as { "openai_api_key": <your api key>}'
)
self.config.openai_api_key = provider_data.openai_api_key

2
llama_stack/providers/remote/inference/fireworks/fireworks.py
View file

@ -118,7 +118,7 @@ class FireworksInferenceAdapter(
provider_data = self.get_request_provider_data()
if provider_data is None or not provider_data.fireworks_api_key:
raise ValueError(
'Pass Fireworks API Key in the header X-LlamaStack-ProviderData as { "fireworks_api_key": <your api key>}'
'Pass Fireworks API Key in the header X-LlamaStack-Provider-Data as { "fireworks_api_key": <your api key>}'
)
return provider_data.fireworks_api_key

2
llama_stack/providers/remote/inference/groq/groq.py
View file

@ -145,6 +145,6 @@ class GroqInferenceAdapter(Inference, ModelRegistryHelper, NeedsRequestProviderD
provider_data = self.get_request_provider_data()
if provider_data is None or not provider_data.groq_api_key:
raise ValueError(
'Pass Groq API Key in the header X-LlamaStack-ProviderData as { "groq_api_key": "<your api key>" }'
'Pass Groq API Key in the header X-LlamaStack-Provider-Data as { "groq_api_key": "<your api key>" }'
)
return Groq(api_key=provider_data.groq_api_key)

2
llama_stack/providers/remote/inference/together/together.py
View file

@ -135,7 +135,7 @@ class TogetherInferenceAdapter(
provider_data = self.get_request_provider_data()
if provider_data is None or not provider_data.together_api_key:
raise ValueError(
'Pass Together API Key in the header X-LlamaStack-ProviderData as { "together_api_key": <your api key>}'
'Pass Together API Key in the header X-LlamaStack-Provider-Data as { "together_api_key": <your api key>}'
)
together_api_key = provider_data.together_api_key
return Together(api_key=together_api_key)

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

@ -46,7 +46,7 @@ class BingSearchToolRuntimeImpl(
provider_data = self.get_request_provider_data()
if provider_data is None or not provider_data.api_key:
raise ValueError(
'Pass Bing Search API Key in the header X-LlamaStack-ProviderData as { "api_key": <your api key>}'
'Pass Bing Search API Key in the header X-LlamaStack-Provider-Data as { "api_key": <your api key>}'
)
return provider_data.api_key

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

@ -45,7 +45,7 @@ class BraveSearchToolRuntimeImpl(
provider_data = self.get_request_provider_data()
if provider_data is None or not provider_data.api_key:
raise ValueError(
'Pass Search provider\'s API Key in the header X-LlamaStack-ProviderData as { "api_key": <your api key>}'
'Pass Search provider\'s API Key in the header X-LlamaStack-Provider-Data as { "api_key": <your api key>}'
)
return provider_data.api_key

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

@ -45,7 +45,7 @@ class TavilySearchToolRuntimeImpl(
provider_data = self.get_request_provider_data()
if provider_data is None or not provider_data.api_key:
raise ValueError(
'Pass Search provider\'s API Key in the header X-LlamaStack-ProviderData as { "api_key": <your api key>}'
'Pass Search provider\'s API Key in the header X-LlamaStack-Provider-Data as { "api_key": <your api key>}'
)
return provider_data.api_key

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

@ -46,7 +46,7 @@ class WolframAlphaToolRuntimeImpl(
provider_data = self.get_request_provider_data()
if provider_data is None or not provider_data.api_key:
raise ValueError(
'Pass WolframAlpha API Key in the header X-LlamaStack-ProviderData as { "api_key": <your api key>}'
'Pass WolframAlpha API Key in the header X-LlamaStack-Provider-Data as { "api_key": <your api key>}'
)
return provider_data.api_key

2
llama_stack/providers/tests/resolver.py
View file

@ -79,7 +79,7 @@ async def construct_stack_for_test(
if provider_data:
set_request_provider_data(
{"X-LlamaStack-ProviderData": json.dumps(provider_data)}
{"X-LlamaStack-Provider-Data": json.dumps(provider_data)}
)
return test_stack