From 3b52ee05d1fc01ec7a997a97bf4d8dc4c851310e Mon Sep 17 00:00:00 2001 From: fengjial Date: Sat, 23 Mar 2024 08:03:59 +0800 Subject: [PATCH] community[patch]: fix bugs in baiduvectordb as vectorstore (#19380) fix small bugs in vectorstore/baiduvectordb --- docs/docs/integrations/vectorstores/baiduvectordb.ipynb | 2 +- .../langchain_community/vectorstores/baiduvectordb.py | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/docs/docs/integrations/vectorstores/baiduvectordb.ipynb b/docs/docs/integrations/vectorstores/baiduvectordb.ipynb index 3667b9097b6..e42c8d68830 100644 --- a/docs/docs/integrations/vectorstores/baiduvectordb.ipynb +++ b/docs/docs/integrations/vectorstores/baiduvectordb.ipynb @@ -68,7 +68,7 @@ ")\n", "\n", "vector_db = BaiduVectorDB.from_documents(\n", - " docs, embeddings, connection_params=conn_params, drop=True\n", + " docs, embeddings, connection_params=conn_params, drop_old=True\n", ")" ] }, diff --git a/libs/community/langchain_community/vectorstores/baiduvectordb.py b/libs/community/langchain_community/vectorstores/baiduvectordb.py index 8f622997535..f378df7b5eb 100644 --- a/libs/community/langchain_community/vectorstores/baiduvectordb.py +++ b/libs/community/langchain_community/vectorstores/baiduvectordb.py @@ -163,6 +163,7 @@ class BaiduVectorDB(VectorStore): self.field_vector, self.mochowenum.FieldType.FLOAT_VECTOR, dimension=self.table_params.dimension, + not_null=True, ) ) fields.append(schema.Field(self.field_text, self.mochowenum.FieldType.STRING)) @@ -358,7 +359,7 @@ class BaiduVectorDB(VectorStore): doc = Document( page_content=row_data.get(self.field_text), metadata=meta ) - pair = (doc, result.get("distance", 0.0)) + pair = (doc, result.get("score", 0.0)) ret.append(pair) return ret