Respect requested query number

This commit is contained in:
BlipRanger
2023-06-25 01:01:03 -04:00
parent 7b733e7498
commit 5b4753b719

View File

@@ -864,9 +864,10 @@ def chromadb_multiquery():
except Exception as e:
print(e)
#remove duplicate msgs
#remove duplicate msgs, filter down to the right number
seen = set()
messages = [d for d in messages if not (d['content'] in seen or seen.add(d['content']))]
messages = sorted(messages, key=lambda x: x['distance'])[0:n_results]
return jsonify(messages)