Add AzureModelConfig to ModelConfigType for multi-provider support

ls1intum · Jan 31, 2025 · e060cd1 · e060cd1
1 parent 652c3d0
commit e060cd1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/llm_core/llm_core/models/__init__.py b/llm_core/llm_core/models/__init__.py
@@ -1,6 +1,7 @@
 from typing import Union
 
+from llm_core.models.providers.azure_model_config import AzureModelConfig
 from llm_core.models.providers.openai_model_config import OpenAIModelConfig
 
 
-ModelConfigType = Union[OpenAIModelConfig]
+ModelConfigType = Union[OpenAIModelConfig, AzureModelConfig]