mod content safety endpoint

Azure-Samples · Oct 23, 2023 · fbe036f · fbe036f
1 parent db254e0
commit fbe036f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/backend/utilities/helpers/EnvHelper.py b/backend/utilities/helpers/EnvHelper.py
@@ -63,6 +63,8 @@ def __init__(self, **kwargs) -> None:
         self.APPINSIGHTS_CONNECTION_STRING = os.getenv('APPINSIGHTS_CONNECTION_STRING', '')
         # Azure AI Content Safety
         self.AZURE_CONTENT_SAFETY_ENDPOINT = os.getenv('AZURE_CONTENT_SAFETY_ENDPOINT', '')
+        if 'https' not in self.AZURE_CONTENT_SAFETY_ENDPOINT and 'cognitiveservices.azure.com' not in self.AZURE_CONTENT_SAFETY_ENDPOINT:
+            self.AZURE_CONTENT_SAFETY_ENDPOINT = f"https://{self.AZURE_CONTENT_SAFETY_ENDPOINT}.cognitiveservices.azure.com/"
         self.AZURE_CONTENT_SAFETY_KEY = os.getenv('AZURE_CONTENT_SAFETY_KEY', '')
         # Orchestration Settings
         self.ORCHESTRATION_STRATEGY = os.getenv('ORCHESTRATION_STRATEGY', 'openai_function')