Python: postgres memory store

.github/workflows/python-integration-tests.yml

@@ -63,6 +63,7 @@ jobs:
           OpenAI__ApiKey: ${{ secrets.OPENAI__APIKEY }}
           Pinecone__ApiKey: ${{ secrets.PINECONE__APIKEY }}
           Pinecone__Environment: ${{ secrets.PINECONE__ENVIRONMENT }}
+          Postgres__Connectionstr: ${{secrets.POSTGRES__CONNECTIONSTR}}
         run: |
           cd python
           poetry run pytest ./tests/integration

FEATURE_MATRIX.md

@@ -44,7 +44,7 @@
 | Math Skill                        | ✅  | ✅     | ✅   |       |
 | Text Skill                        | ✅  | ✅     | 🔄   |       |
 | Time Skill                        | ✅  | ✅     | ✅   |       |
-| Wait Skill                        | ✅  | ❌     | ✅   |       |
+| Wait Skill                        | ✅  | ✅     | ✅   |       |
 
 ## Planning
 
@@ -63,10 +63,10 @@
 | Weaviate      | ✅  | ✅     | ❌   | Currently supported on Python 3.9+, 3.8 coming soon                             |
 | Chroma        | ✅  | ✅     | ❌   |                                                                                 |
 | Qdrant        | ✅  | ❌     | ❌   |                                                                                 |
-| Pinecone      | ✅  | ❌     | ❌   |                                                                                 |
+| Pinecone      | ✅  | ✅     | ❌   |                                                                                 |
 | Milvus        | ❌  | ❌     | ❌   | Coming soon                                                                     |
 | Sqlite        | ✅  | ❌     | ❌   | Vector optimization requires [sqlite-vss](https://github.com/asg017/sqlite-vss) |
-| Postgres      | ✅  | ❌     | ❌   | Vector optimization requires [pgvector](https://github.com/pgvector/pgvector)   |
+| Postgres      | ✅  | ✅     | ❌   | Vector optimization requires [pgvector](https://github.com/pgvector/pgvector)   |
 | CosmosDB      | ✅  | ❌     | ❌   | CosmosDB is not optimized for vector storage                                    |
 | Redis         | ✅  | ❌     | ❌   | Vector optimization requires [RediSearch](https://redis.io/docs/stack/search)   |
 

python/.env.example

@@ -5,3 +5,4 @@ AZURE_OPENAI_ENDPOINT=""
 AZURE_OPENAI_API_KEY=""
 PINECONE_API_KEY=""
 PINECONE_ENVIRONMENT=""
+POSTGRES_CONNECTION_STRING=""

python/pyproject.toml

@@ -37,6 +37,10 @@ weaviate-client = "^3.18.0"
 [tool.poetry.group.pinecone.dependencies]
 pinecone-client = "^2.2.2"
 
+[tool.poetry.group.postgres.dependencies]
+psycopg-pool = "^3.1.7"
+psycopg = "^3.1.9"
+
 [tool.isort]
 profile = "black"
 

python/semantic_kernel/__init__.py

@@ -18,13 +18,15 @@
     azure_openai_settings_from_dot_env,
     openai_settings_from_dot_env,
     pinecone_settings_from_dot_env,
+    postgres_settings_from_dot_env,
 )
 
 __all__ = [
     "Kernel",
     "NullLogger",
     "openai_settings_from_dot_env",
     "azure_openai_settings_from_dot_env",
+    "postgres_settings_from_dot_env",
     "pinecone_settings_from_dot_env",
     "PromptTemplateConfig",
     "PromptTemplate",

python/semantic_kernel/connectors/memory/pinecone/pinecone_memory_store.py

@@ -25,6 +25,8 @@
 
 
 class PineconeMemoryStore(MemoryStoreBase):
+    """A memory store that uses Pinecone as the backend."""
+
     _logger: Logger
     _pinecone_api_key: str
     _pinecone_environment: str
@@ -42,8 +44,8 @@ def __init__(
         Arguments:
             pinecone_api_key {str} -- The Pinecone API key.
             pinecone_environment {str} -- The Pinecone environment.
-            logger {Optional[Logger]} -- The logger to use. (default: {None})
             default_dimensionality {int} -- The default dimensionality to use for new collections.
+            logger {Optional[Logger]} -- The logger to use. (default: {None})
         """
         if default_dimensionality > MAX_DIMENSIONALITY:
             raise ValueError(

python/semantic_kernel/connectors/memory/postgres/__init__.py

@@ -0,0 +1,7 @@
+# Copyright (c) Microsoft. All rights reserved.
+
+from semantic_kernel.connectors.memory.postgres.postgres_memory_store import (
+    PostgresMemoryStore,
+)
+
+__all__ = ["PostgresMemoryStore"]