@@ -334,7 +334,10 @@ def get_rag_context(
elif file.get("collection_name"):
collection_names.append(file["collection_name"])
elif file.get("id"):
- collection_names.append(f"file-{file['id']}")
+ if file.get("legacy"):
+ collection_names.append(f"{file['id']}")
+ else:
+ collection_names.append(f"file-{file['id']}")
collection_names = set(collection_names).difference(extracted_collections)
if not collection_names:
@@ -65,6 +65,7 @@ def upgrade():
user_id=doc.user_id,
description=doc.name,
meta={
+ "legacy": True,
"document": True,
"tags": json.loads(doc.content or "{}").get("tags", []),
},
@@ -101,7 +101,12 @@
]
: [];
- items = [...$knowledge, ...legacy_collections];
+ items = [...$knowledge, ...legacy_collections].map((item) => {
+ return {
+ ...item,
+ ...{ legacy: item?.legacy ?? item?.meta?.document ?? undefined }
+ };
+ });
fuse = new Fuse(items, {
keys: ['name', 'description']