feat: Allow setting the initial vector length on pgvector document_chunk table

This commit is contained in:
Jason Kidd
2025-01-03 09:11:09 -08:00
parent 996ade9090
commit 70b74b5217
2 changed files with 41 additions and 2 deletions

View File

@@ -1211,6 +1211,9 @@ if VECTOR_DB == "pgvector" and not PGVECTOR_DB_URL.startswith("postgres"):
raise ValueError(
"Pgvector requires setting PGVECTOR_DB_URL or using Postgres with vector extension as the primary database."
)
PGVECTOR_INITIALIZE_MAX_VECTOR_LENGTH = int(
os.environ.get("PGVECTOR_INITIALIZE_MAX_VECTOR_LENGTH", "1536")
)
####################################
# Information Retrieval (RAG)