fix properly the mapping to azuregpt

alan-turing-institute · Jun 12, 2024 · 34255bb · 34255bb
1 parent cfc5240
commit 34255bb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/reginald/models/__init__.py b/reginald/models/__init__.py
@@ -48,7 +48,7 @@ def get_model(model_name: str):
                     LlamaIndexGPTAzure,
                 )
 
-                return LlamaIndexGPTOpenAI
+                return LlamaIndexGPTAzure
             case "llama-index-gpt-openai":
                 from reginald.models.llama_index.llama_index_openai import (
                     LlamaIndexGPTOpenAI,