hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-24 18:23:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

dep: bump pgvecto-rs client from 0.1.x to 0.2.x (#6891)

Browse Source
This commit is contained in:
Bowen Liang
2024-08-02 15:51:23 +08:00
committed by GitHub
parent 7ab04e17e7
commit 6aa02f8c63
5 changed files with 19 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
api/core/rag/datasource/vdb/relyt/relyt_vector.py
View File

@@ -105,7 +105,7 @@ class RelytVector(BaseVector):
redis_client.set(collection_exist_cache_key, 1, ex=3600)
def add_texts(self, documents: list[Document], embeddings: list[list[float]], **kwargs):
from pgvecto_rs.sqlalchemy import Vector
from pgvecto_rs.sqlalchemy import VECTOR
ids = [str(uuid.uuid1()) for _ in documents]
metadatas = [d.metadata for d in documents]
@@ -118,7 +118,7 @@ class RelytVector(BaseVector):
self._collection_name,
Base.metadata,
Column("id", TEXT, primary_key=True),
Column("embedding", Vector(len(embeddings[0]))),
Column("embedding", VECTOR(len(embeddings[0]))),
Column("document", String, nullable=True),
Column("metadata", JSON, nullable=True),
extend_existing=True,
@@ -169,7 +169,7 @@ class RelytVector(BaseVector):
Args:
ids: List of ids to delete.
"""
from pgvecto_rs.sqlalchemy import Vector
from pgvecto_rs.sqlalchemy import VECTOR
if ids is None:
raise ValueError("No ids provided to delete.")
@@ -179,7 +179,7 @@ class RelytVector(BaseVector):
self._collection_name,
Base.metadata,
Column("id", TEXT, primary_key=True),
Column("embedding", Vector(self.embedding_dimension)),
Column("embedding", VECTOR(self.embedding_dimension)),
Column("document", String, nullable=True),
Column("metadata", JSON, nullable=True),
extend_existing=True,