Browse Source

refac: legacy support

Timothy J. Baek 7 months ago
parent
commit
1fe1c27220

+ 4 - 1
backend/open_webui/apps/retrieval/utils.py

@@ -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:

+ 1 - 0
backend/open_webui/migrations/versions/6a39f3d8e55c_add_knowledge_table.py

@@ -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", []),
                 },

+ 6 - 1
src/lib/components/chat/MessageInput/Commands/Knowledge.svelte

@@ -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']