Add dynamic Entra credentials to AsyncConnectionPool

microsoft · Oct 18, 2024 · c758b12 · c758b12
1 parent bec6917
commit c758b12
Show file tree

Hide file tree

Showing 6 changed files with 230 additions and 151 deletions.
diff --git a/python/samples/concepts/memory/new_memory.py b/python/samples/concepts/memory/new_memory.py
@@ -12,6 +12,9 @@
 from semantic_kernel.connectors.ai.open_ai import OpenAIEmbeddingPromptExecutionSettings, OpenAITextEmbedding
 from semantic_kernel.connectors.ai.open_ai.services.azure_text_embedding import AzureTextEmbedding
 from semantic_kernel.connectors.memory.azure_ai_search import AzureAISearchCollection
+from semantic_kernel.connectors.memory.azure_db_for_postgres.azure_db_for_postgres_collection import (
+    AzureDBForPostgresCollection,
+)
 from semantic_kernel.connectors.memory.postgres.postgres_collection import PostgresCollection
 from semantic_kernel.connectors.memory.qdrant import QdrantCollection
 from semantic_kernel.connectors.memory.redis import RedisHashsetCollection, RedisJsonCollection
@@ -88,6 +91,10 @@ class MyDataModelList:
     "ai_search": lambda: AzureAISearchCollection[MyDataModel](
         data_model_type=MyDataModel,
     ),
+    "azure_db_for_postgres": lambda: AzureDBForPostgresCollection[str, MyDataModel](
+        data_model_type=MyDataModel,
+        collection_name=collection_name,
+    ),
     "postgres": lambda: PostgresCollection[str, MyDataModel](
         data_model_type=MyDataModel,
         collection_name=collection_name,