feature: add Azure Identity for Azure OpenAI and Azure AI Search authentication (#3262)

mem0/embeddings/azure_openai.py

@@ -1,11 +1,14 @@
 import os
 from typing import Literal, Optional
 
+from azure.identity import DefaultAzureCredential, get_bearer_token_provider
 from openai import AzureOpenAI
 
 from mem0.configs.embeddings.base import BaseEmbedderConfig
 from mem0.embeddings.base import EmbeddingBase
 
+SCOPE = "https://cognitiveservices.azure.com/.default"
+
 
 class AzureOpenAIEmbedding(EmbeddingBase):
     def __init__(self, config: Optional[BaseEmbedderConfig] = None):
@@ -17,9 +20,21 @@ class AzureOpenAIEmbedding(EmbeddingBase):
         api_version = self.config.azure_kwargs.api_version or os.getenv("EMBEDDING_AZURE_API_VERSION")
         default_headers = self.config.azure_kwargs.default_headers
 
+        # If the API key is not provided or is a placeholder, use DefaultAzureCredential.
+        if api_key is None or api_key == "" or api_key == "your-api-key":
+            self.credential = DefaultAzureCredential()
+            azure_ad_token_provider = get_bearer_token_provider(
+                self.credential,
+                SCOPE,
+            )
+            api_key = None
+        else:
+            azure_ad_token_provider = None
+
         self.client = AzureOpenAI(
             azure_deployment=azure_deployment,
             azure_endpoint=azure_endpoint,
+            azure_ad_token_provider=azure_ad_token_provider,
             api_version=api_version,
             api_key=api_key,
             http_client=self.config.http_client,