소스 검색

fix: web/youtube attachements

Timothy Jaeryang Baek 1 개월 전
부모
커밋
487979859a
2개의 변경된 파일7개의 추가작업 그리고 4개의 파일을 삭제
  1. 5 2
      backend/open_webui/retrieval/utils.py
  2. 2 2
      src/lib/components/chat/Chat.svelte

+ 5 - 2
backend/open_webui/retrieval/utils.py

@@ -487,9 +487,12 @@ def get_sources_from_items(
 
         if item.get("type") == "text":
             # Raw Text
-            # Used during temporary chat file uploads
+            # Used during temporary chat file uploads or web page & youtube attachements
 
-            if item.get("file"):
+            if item.get("collection_name"):
+                # If item has a collection name, use it
+                collection_names.append(item.get("collection_name"))
+            elif item.get("file"):
                 # if item has file data, use it
                 query_result = {
                     "documents": [

+ 2 - 2
src/lib/components/chat/Chat.svelte

@@ -671,7 +671,7 @@
 		console.log(url);
 
 		const fileItem = {
-			type: 'doc',
+			type: 'text',
 			name: url,
 			collection_name: '',
 			status: 'uploading',
@@ -704,7 +704,7 @@
 		console.log(url);
 
 		const fileItem = {
-			type: 'doc',
+			type: 'text',
 			name: url,
 			collection_name: '',
 			status: 'uploading',