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

fix provider lookup

Browse source
This commit is contained in:
Xi Yan 2024-12-30 15:04:10 -08:00
parent 0eb89bcb07
commit 0677d91699
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/providers/tests/agents/fixtures.py
View file

@ -86,7 +86,7 @@ async def agents_stack(request, inference_model, safety_shield):
provider_id = inference_provider.provider_id provider_id = inference_provider.provider_id
if inference_provider.config and "model" in inference_provider.config: if inference_provider.config and "model" in inference_provider.config:
model_to_provider_id = { model_to_provider_id = {
provider.config.model: provider.provider_id provider.config["model"]: provider.provider_id
for provider in providers["inference"] for provider in providers["inference"]
} }