From 0677d91699e951836b4da4cae1036e9a92e17f73 Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Mon, 30 Dec 2024 15:04:10 -0800 Subject: [PATCH] fix provider lookup --- llama_stack/providers/tests/agents/fixtures.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llama_stack/providers/tests/agents/fixtures.py b/llama_stack/providers/tests/agents/fixtures.py index 6c41a41db..a90e5afc1 100644 --- a/llama_stack/providers/tests/agents/fixtures.py +++ b/llama_stack/providers/tests/agents/fixtures.py @@ -86,7 +86,7 @@ async def agents_stack(request, inference_model, safety_shield): provider_id = inference_provider.provider_id if inference_provider.config and "model" in inference_provider.config: model_to_provider_id = { - provider.config.model: provider.provider_id + provider.config["model"]: provider.provider_id for provider in providers["inference"] }