From 126f1db801f1179609d6d3fbf807f796cd61b218 Mon Sep 17 00:00:00 2001
From: bparees <bparees@redhat.com>
Date: Fri, 8 Aug 2025 15:44:04 -0400
Subject: [PATCH] Use the llamastack identifier for the model id, not the
 provider's model id

---
 src/app/endpoints/query.py | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

diff --git a/src/app/endpoints/query.py b/src/app/endpoints/query.py
index 86b1ef44..a628fd77 100644
--- a/src/app/endpoints/query.py
+++ b/src/app/endpoints/query.py
@@ -171,13 +171,14 @@ def select_model_and_provider_id(
                 },
             ) from e
 
-    llama_stack_model_id = f"{provider_id}/{model_id}"
     # Validate that the model_id and provider_id are in the available models
     logger.debug("Searching for model: %s, provider: %s", model_id, provider_id)
-    if not any(
-        m.identifier == llama_stack_model_id and m.provider_id == provider_id
-        for m in models
-    ):
+    
+    def check_model(m):
+        logger.debug("Available model - model_identifier: %s, provider_model_id: %s, provider_id: %s", m.identifier, m.provider_resource_id, m.provider_id)
+        return m.identifier == model_id and m.provider_id == provider_id
+    
+    if not any(check_model(m) for m in models):
         message = f"Model {model_id} from provider {provider_id} not found in available models"
         logger.error(message)
         raise HTTPException(
@@ -188,7 +189,7 @@ def select_model_and_provider_id(
             },
         )
 
-    return llama_stack_model_id, provider_id
+    return model_id, provider_id
 
 
 def _is_inout_shield(shield: Shield) -> bool: