[Fix] Rename the return parameter from 'azure_model_map' to 'azure_mo…

…del_to_deployment_id_map'
Significant-Gravitas · Jul 3, 2023 · cd1462b · cd1462b
1 parent 418ba5a
commit cd1462b
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/autogpt/config/config.py b/autogpt/config/config.py
@@ -253,7 +253,7 @@ def build_config_from_env(cls):
             config_dict["openai_api_base"] = azure_config["openai_api_base"]
             config_dict["openai_api_version"] = azure_config["openai_api_version"]
             config_dict["azure_model_to_deployment_id_map"] = azure_config[
-                "azure_model_map"
+                "azure_model_to_deployment_id_map"
             ]
 
             openai.api_type = azure_config["openai_api_type"]
@@ -293,7 +293,9 @@ def load_azure_config(cls, config_file: str = AZURE_CONFIG_FILE) -> Dict[str, st
             "openai_api_base": config_params.get("azure_api_base") or "",
             "openai_api_version": config_params.get("azure_api_version")
             or "2023-03-15-preview",
-            "azure_model_map": config_params.get("azure_model_map", {}),
+            "azure_model_to_deployment_id_map": config_params.get(
+                "azure_model_map", {}
+            ),
         }