refac

backend/open_webui/apps/webui/routers/files.py

@@ -10,7 +10,7 @@ from open_webui.config import UPLOAD_DIR
 from open_webui.constants import ERROR_MESSAGES
 from open_webui.env import SRC_LOG_LEVELS
 from fastapi import APIRouter, Depends, File, HTTPException, UploadFile, status
-from fastapi.responses import FileResponse
+from fastapi.responses import FileResponse, StreamingResponse
 from open_webui.utils.utils import get_admin_user, get_verified_user
 
 log = logging.getLogger(__name__)
@@ -189,16 +189,32 @@ async def get_file_content_by_id(id: str, user=Depends(get_verified_user)):
     file = Files.get_file_by_id(id)
 
     if file and (file.user_id == user.id or user.role == "admin"):
-        file_path = Path(file.meta["path"])
-
-        # Check if the file already exists in the cache
-        if file_path.is_file():
-            print(f"file_path: {file_path}")
-            return FileResponse(file_path)
+        file_path = file.meta.get("path")
+        if file_path:
+            file_path = Path(file_path)
+
+            # Check if the file already exists in the cache
+            if file_path.is_file():
+                print(f"file_path: {file_path}")
+                return FileResponse(file_path)
+            else:
+                raise HTTPException(
+                    status_code=status.HTTP_404_NOT_FOUND,
+                    detail=ERROR_MESSAGES.NOT_FOUND,
+                )
         else:
-            raise HTTPException(
-                status_code=status.HTTP_404_NOT_FOUND,
-                detail=ERROR_MESSAGES.NOT_FOUND,
+            # File path doesn’t exist, return the content as .txt if possible
+            file_content = file.content.get("content", "")
+            file_name = file.filename
+
+            # Create a generator that encodes the file content
+            def generator():
+                yield file_content.encode("utf-8")
+
+            return StreamingResponse(
+                generator(),
+                media_type="text/plain",
+                headers={"Content-Disposition": f"attachment; filename={file_name}"},
             )
     else:
         raise HTTPException(

src/lib/components/chat/MessageInput/Commands/Knowledge.svelte

@@ -80,7 +80,7 @@
 							name: 'All Documents',
 							legacy: true,
 							type: 'collection',
-							description: 'Deprecated (legacy collection), please use the new knowledge base.',
+							description: 'Deprecated (legacy collection), please create a new knowledge base.',
 							title: $i18n.t('All Documents'),
 							collection_names: legacy_documents.map((item) => item.id)
 						},
@@ -93,7 +93,7 @@
 								name: tag,
 								legacy: true,
 								type: 'collection',
-								description: 'Deprecated (legacy collection), please use the new knowledge base.',
+								description: 'Deprecated (legacy collection), please create a new knowledge base.',
 								collection_names: legacy_documents
 									.filter((item) => (item?.meta?.tags ?? []).map((tag) => tag.name).includes(tag))
 									.map((item) => item.id)

src/lib/components/workspace/Models/Knowledge/Selector.svelte

@@ -21,7 +21,8 @@
 							name: 'All Documents',
 							legacy: true,
 							type: 'collection',
-							description: 'Deprecated (legacy collection), please use the new knowledge base.',
+							description: 'Deprecated (legacy collection), please create a new knowledge base.',
+
 							title: $i18n.t('All Documents'),
 							collection_names: legacy_documents.map((item) => item.id)
 						},
@@ -34,7 +35,8 @@
 								name: tag,
 								legacy: true,
 								type: 'collection',
-								description: 'Deprecated (legacy collection), please use the new knowledge base.',
+								description: 'Deprecated (legacy collection), please create a new knowledge base.',
+
 								collection_names: legacy_documents
 									.filter((item) => (item?.meta?.tags ?? []).map((tag) => tag.name).includes(tag))
 									.map((item) => item.id)