Update embedding dependency to support system assigned MI

microsoft · Aug 5, 2024 · a474b71 · a474b71
1 parent 4003f18
commit a474b71
Showing 1 changed file with 17 additions and 9 deletions.
diff --git a/backend/settings.py b/backend/settings.py
@@ -175,16 +175,24 @@ def extract_embedding_dependency(self) -> Optional[dict]:
                 "type": "deployment_name",
                 "deployment_name": self.embedding_name
             }
-
-        elif self.embedding_endpoint and self.embedding_key:
-            return {
-                "type": "endpoint",
-                "endpoint": self.embedding_endpoint,
-                "authentication": {
-                    "type": "api_key",
-                    "api_key": self.embedding_key
+        elif self.embedding_endpoint:
+            if self.embedding_key:
+                return {
+                    "type": "endpoint",
+                    "endpoint": self.embedding_endpoint,
+                    "authentication": {
+                        "type": "api_key",
+                        "api_key": self.embedding_key
+                    }
+                }
+            else:
+                return {
+                    "type": "endpoint",
+                    "endpoint": self.embedding_endpoint,
+                    "authentication": {
+                        "type": "system_assigned_managed_identity"
+                    }
                 }
-            }
         else:   
             return None