github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-11 13:55:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

community[patch]: fix bugs in baiduvectordb as vectorstore (#19380)

Browse Source 
fix small bugs in vectorstore/baiduvectordb
This commit is contained in:
fengjial 2024-03-23 08:03:59 +08:00 committed by GitHub
parent 5402aef32e
commit 3b52ee05d1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/docs/integrations/vectorstores/baiduvectordb.ipynb
View File

@ -68,7 +68,7 @@
")\n", ")\n",
"\n", "\n",
"vector_db = BaiduVectorDB.from_documents(\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",
")" ")"
] ]
}, },

3
libs/community/langchain_community/vectorstores/baiduvectordb.py
View File

@ -163,6 +163,7 @@ class BaiduVectorDB(VectorStore):
self.field_vector, self.field_vector,
self.mochowenum.FieldType.FLOAT_VECTOR, self.mochowenum.FieldType.FLOAT_VECTOR,
dimension=self.table_params.dimension, dimension=self.table_params.dimension,
not_null=True,
) )
) )
fields.append(schema.Field(self.field_text, self.mochowenum.FieldType.STRING)) fields.append(schema.Field(self.field_text, self.mochowenum.FieldType.STRING))
@ -358,7 +359,7 @@ class BaiduVectorDB(VectorStore):
doc = Document( doc = Document(
page_content=row_data.get(self.field_text), metadata=meta 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) ret.append(pair)
return ret return ret