Merge pull request #51 from sofyanajridi/fix-azure-openai-api-key

Fix Azure OpenAI bug with API key

Author: pamelafox

scripts/service_setup.py

@@ -57,7 +57,8 @@ def get_openai_client(oai_config: dict):
         return openai.AzureOpenAI(
             api_version=oai_config["api_version"],
             azure_endpoint=oai_config["api_base"],
-            azure_ad_token=oai_config["api_key"],
+            api_key=oai_config["api_key"] if os.environ.get("AZURE_OPENAI_KEY") else None,
+            azure_ad_token=None if os.environ.get("AZURE_OPENAI_KEY") else oai_config["api_key"],
             azure_deployment=oai_config["deployment_id"],
         )
     else: