Use azure endpoint

heyitsaamir · Sep 19, 2024 · 857fa44 · 857fa44
1 parent 03322b1
commit 857fa44
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/no_autogen_spec_solver.py b/src/no_autogen_spec_solver.py
@@ -38,7 +38,7 @@ def __init__(self, llm_config: Dict):
         elif llm_config.get("api_type") == "azure":
             logging.debug("Using Azure API key")
             self.client = instructor.from_openai(AzureOpenAI(
-                base_url=llm_config.get("base_url", ""),
+                azure_endpoint=llm_config.get("base_url", ""),
                 api_version=llm_config.get("api_version", ""),
                 azure_ad_token_provider=llm_config.get("azure_ad_token_provider"),
             ))