feat: add min-connection and max-connection for pgvector (#8841)

2024-09-27 18:16:20 +08:00 · 2024-09-27 18:16:20 +08:00 · 55e6123db9
commit 55e6123db9
parent c828a5dfdf
5 changed files with 28 additions and 2 deletions
--- a/api/configs/middleware/vdb/pgvector_config.py
+++ b/api/configs/middleware/vdb/pgvector_config.py
@ -33,3 +33,13 @@ class PGVectorConfig(BaseSettings):
        description="Name of the PostgreSQL database to connect to",
        default=None,
    )
+
+    PGVECTOR_MIN_CONNECTION: PositiveInt = Field(
+        description="Min connection of the PostgreSQL database",
+        default=1,
+    )
+
+    PGVECTOR_MAX_CONNECTION: PositiveInt = Field(
+        description="Max connection of the PostgreSQL database",
+        default=5,
+    )