From 57340e7a5175a9de5e7d534bf5ff338c172aa65f Mon Sep 17 00:00:00 2001
From: Xi Yan <xiyan@meta.com>
Date: Mon, 30 Dec 2024 14:12:22 -0800
Subject: [PATCH] fix provider

---
 llama_stack/providers/tests/agents/fixtures.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/llama_stack/providers/tests/agents/fixtures.py b/llama_stack/providers/tests/agents/fixtures.py
index 733fd10a9..a2b927688 100644
--- a/llama_stack/providers/tests/agents/fixtures.py
+++ b/llama_stack/providers/tests/agents/fixtures.py
@@ -86,9 +86,9 @@ async def agents_stack(request, inference_model, safety_shield):
         ModelInput(
             model_id=model,
             model_type=ModelType.llm,
-            provider_id=providers["inference"][0].provider_id,
+            provider_id=providers["inference"][i].provider_id,
         )
-        for model in inference_models
+        for i, model in enumerate(inference_models)
     ]
     models.append(
         ModelInput(