fix using aaaa-100 embedding when the prompt has aaaa-10000 and you have both aaaa-100 and aaaa-10000 in the directory with embeddings.
This commit is contained in:
parent
88ec0cf557
commit
71fe7fa49f
1 changed files with 2 additions and 1 deletions
|
@ -57,7 +57,8 @@ class EmbeddingDatabase:
|
|||
first_id = ids[0]
|
||||
if first_id not in self.ids_lookup:
|
||||
self.ids_lookup[first_id] = []
|
||||
self.ids_lookup[first_id].append((ids, embedding))
|
||||
|
||||
self.ids_lookup[first_id] = sorted(self.ids_lookup[first_id] + [(ids, embedding)], key=lambda x: len(x[0]), reverse=True)
|
||||
|
||||
return embedding
|
||||
|
||||
|
|
Loading…
Reference in a new issue