Skip to content

Commit 880c3c1

Browse files
committed
Bugfix: Context can be empty in vector database search
1 parent 7dd21e3 commit 880c3c1

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

Source/Chatbook/PromptGenerators/VectorDatabases.wl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -847,13 +847,13 @@ vectorDBSearch[ names: $$dbNames, prompt_, prop: "Values"|"Results" ] := Enclose
847847

848848
results = ConfirmMatch[
849849
applyBias[ #, vectorDBSearch[ #, prompt, "Results" ] ] & /@ names,
850-
{ { KeyValuePattern[ "Distance" -> $$size ].. }... },
850+
{ { KeyValuePattern[ "Distance" -> $$size ]... }... },
851851
"Results"
852852
];
853853

854854
sorted = SortBy[ Flatten @ results, #Distance & ];
855855

856-
If[ prop === "Results",
856+
If[ prop === "Results" || sorted === { },
857857
sorted,
858858
ConfirmBy[ DeleteDuplicates @ Lookup[ sorted, "Value" ], ListQ, "Values" ]
859859
]

0 commit comments

Comments
 (0)