jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: update mismatch vector type (#9462)

Browse source
This commit is contained in:
ice yao 2024-10-18 08:21:41 +08:00 • committed by GitHub
commit 2155bba5b0
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
api/core/rag/datasource/vdb/pgvecto_rs/pgvecto_rs.py
View file

@ -216,7 +216,7 @@ class PGVectoRSFactory(AbstractVectorFactory):
else:
dataset_id = dataset.id
collection_name = Dataset.gen_collection_name_by_id(dataset_id).lower()
dataset.index_struct = json.dumps(self.gen_index_struct_dict(VectorType.WEAVIATE, collection_name))
dataset.index_struct = json.dumps(self.gen_index_struct_dict(VectorType.PGVECTO_RS, collection_name))
dim = len(embeddings.embed_query("pgvecto_rs"))
return PGVectoRS(