community[patch]: fix bugs in baiduvectordb as vectorstore (langchain…

…-ai#19380) fix small bugs in vectorstore/baiduvectordb
chrispy-snps · Mar 30, 2024 · 1212ab7 · 1212ab7
1 parent 56a8852
commit 1212ab7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git 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
@@ -163,6 +163,7 @@ def _create_table(self, table_name: str) -> None:
                 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 @@ def _similarity_search_with_score(
                 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