🔄 Updated Azure API parameters

shroominic · Nov 8, 2023 · d6828ad · d6828ad
1 parent f0180f4
commit d6828ad
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/funcchain/utils/model_defaults.py b/funcchain/utils/model_defaults.py
@@ -114,8 +114,8 @@ def create_long_llm() -> RunnableWithFallbacks:
             {
                 "openai_api_type": "azure",
                 "openai_api_key": settings.AZURE_API_KEY,
-                "openai_api_base": settings.AZURE_API_BASE,
-                "openai_api_version": settings.AZURE_API_VERSION,
+                "azure_endpoint": settings.AZURE_API_BASE,
+                "api_version": settings.AZURE_API_VERSION,
             }
         )
         print("Model: AZURE")