feat(Vector): support pgvector (#1624)

This commit is contained in:
TQ 2024-02-20 22:29:26 +08:00 committed by GitHub
parent 066ea5bf28
commit cd40e3982b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
6 changed files with 323 additions and 59 deletions

View file

@ -16,6 +16,10 @@ llama-index = { extras = ["local_models"], version = "0.9.3" }
watchdog = "^3.0.0"
qdrant-client = "^1.6.9"
chromadb = {version = "^0.4.13", optional = true}
asyncpg = {version = "^0.29.0", optional = true}
pgvector = {version = "^0.2.5", optional = true}
psycopg2-binary = {version = "^2.9.9", optional = true}
sqlalchemy = {version = "^2.0.27", optional = true}
[tool.poetry.group.dev.dependencies]
black = "^22"
@ -45,6 +49,7 @@ transformers = "^4.34.0"
[tool.poetry.extras]
chroma = ["chromadb"]
pgvector = ["sqlalchemy", "pgvector", "psycopg2-binary", "asyncpg"]
[build-system]
requires = ["poetry-core>=1.0.0"]