github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-27 00:48:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

community[patch]: Adding validation when vector does not exist (#19698)

Browse Source 
Adding validation when vector does not exist

Co-authored-by: gaoyuan <gaoyuan.20001218@bytedance.com>
This commit is contained in:
高远 2024-03-29 04:58:23 +08:00 committed by GitHub
parent f55b11fb73
commit 688ca48019
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
libs/community/langchain_community/vectorstores/vikingdb.py
View File

@ -289,9 +289,13 @@ class VikingDB(VectorStore):
ret = []
for item in res:
item.fields.pop("primary_key")
item.fields.pop("vector")
page_content = item.fields.pop("text")
if "primary_key" in item.fields:
item.fields.pop("primary_key")
if "vector" in item.fields:
item.fields.pop("vector")
page_content = ""
if "text" in item.fields:
page_content = item.fields.pop("text")
doc = Document(page_content=page_content, metadata=item.fields)
pair = (doc, item.score)
ret.append(pair)
@ -351,6 +355,12 @@ class VikingDB(VectorStore):
documents = []
ordered_result_embeddings = []
for item in res:
if (
"vector" not in item.fields
or "primary_key" not in item.fields
or "text" not in item.fields
):
continue
ordered_result_embeddings.append(item.fields.pop("vector"))
item.fields.pop("primary_key")
page_content = item.fields.pop("text")