fix:score threshold (#24897)
This commit is contained in:
parent
9486715929
commit
2042353526
24 changed files with 25 additions and 25 deletions
|
|
@ -192,7 +192,7 @@ class VikingDBVector(BaseVector):
|
|||
metadata = result.fields.get(vdb_Field.METADATA_KEY.value)
|
||||
if metadata is not None:
|
||||
metadata = json.loads(metadata)
|
||||
if result.score > score_threshold:
|
||||
if result.score >= score_threshold:
|
||||
metadata["score"] = result.score
|
||||
doc = Document(page_content=result.fields.get(vdb_Field.CONTENT_KEY.value), metadata=metadata)
|
||||
docs.append(doc)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue