Merge pull request #155 from pauldotyu/main

feat: adding additional env vars for oai endpoint in ai-service
Azure-Samples · Jul 15, 2024 · 10c18c5 · 10c18c5
2 parents a7dffd3 + 1398ed4
commit 10c18c5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/ai-service/main.py b/src/ai-service/main.py
@@ -19,7 +19,7 @@ async def get_health():
     capabilities = ["description"]
 
     # Check if the environment variable is set
-    if os.environ.get("AZURE_OPENAI_DALLE_ENDPOINT") and os.environ.get("AZURE_OPENAI_DALLE_DEPLOYMENT_NAME"):
+    if (os.environ.get("AZURE_OPENAI_DALLE_ENDPOINT") or os.environ.get("AZURE_OPENAI_ENDPOINT")) and os.environ.get("AZURE_OPENAI_DALLE_DEPLOYMENT_NAME"):
         # If it is, add "image" to the array
         capabilities.append("image")
 

diff --git a/src/ai-service/routers/image_generator.py b/src/ai-service/routers/image_generator.py
@@ -28,7 +28,7 @@ async def post_image(request: Request) -> JSONResponse:
         print("Calling OpenAI")
 
         api_version = os.environ.get("AZURE_OPENAI_API_VERSION")
-        endpoint = os.environ.get("AZURE_OPENAI_DALLE_ENDPOINT")
+        endpoint = os.environ.get("AZURE_OPENAI_DALLE_ENDPOINT") or os.environ.get("AZURE_OPENAI_ENDPOINT")
         model_deployment_name = os.environ.get("AZURE_OPENAI_DALLE_DEPLOYMENT_NAME")
 
         token_provider = get_bearer_token_provider(DefaultAzureCredential(), "https://cognitiveservices.azure.com/.default")