yujunjun/kotaemon
1
0
Fork 0
mirror of https://github.com/Cinnamon/kotaemon.git synced 2025-07-03 15:10:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(docstore): preserve retrieval ranking order in lancedb get() (#745)

Browse Source
This commit is contained in:
TommasoMoroHtx 2025-06-05 11:08:49 +02:00 committed by GitHub
parent ddb5187293
commit 833982ac81
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
libs/kotaemon/kotaemon/storages/docstores/lancedb.py
View File

@ -113,14 +113,17 @@ class LanceDBDocumentStore(BaseDocumentStore):
)
except (ValueError, FileNotFoundError):
docs = []
return [
Document(
id_=doc["id"],
# return the documents using the order of original ids (which were ordered by score)
doc_dict = {
doc["id"]: Document(
d_=doc["id"],
text=doc["text"] if doc["text"] else "<empty>",
metadata=json.loads(doc["attributes"]),
)
for doc in docs
]
}
return [doc_dict[_id] for _id in ids if _id in doc_dict]
def delete(self, ids: Union[List[str], str], refresh_indices: bool = True):
"""Delete document by id"""