浏览代码

fix: web attachment issue

Timothy J. Baek 6 月之前
父节点
当前提交
6336d34b59
共有 2 个文件被更改,包括 4 次插入3 次删除
  1. 2 2
      backend/open_webui/apps/retrieval/main.py
  2. 2 1
      backend/open_webui/apps/retrieval/utils.py

+ 2 - 2
backend/open_webui/apps/retrieval/main.py

@@ -709,8 +709,8 @@ def save_docs_to_vector_db(
             if overwrite:
                 VECTOR_DB_CLIENT.delete_collection(collection_name=collection_name)
                 log.info(f"deleting existing collection {collection_name}")
-
-            if add is False:
+            elif add is False:
+                log.info(f"collection {collection_name} already exists, overwrite is False and add is False")
                 return True
 
         log.info(f"adding to collection {collection_name}")

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

@@ -385,6 +385,8 @@ def get_rag_context(
             extracted_collections.extend(collection_names)
 
         if context:
+            if "data" in file:
+                del file["data"]
             relevant_contexts.append({**context, "file": file})
 
     contexts = []
@@ -401,7 +403,6 @@ def get_rag_context(
                         ]
                     )
                 )
-
                 contexts.append(
                     ((", ".join(file_names) + ":\n\n") if file_names else "")
                     + "\n\n".join(