Timothy Jaeryang Baek 5 months ago
parent
commit
2e7af7bdbf

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

@@ -893,7 +893,7 @@ def process_file(
             # Update the content in the file
             # Update the content in the file
             # Usage: /files/{file_id}/data/content/update
             # Usage: /files/{file_id}/data/content/update
 
 
-            VECTOR_DB_CLIENT.delete(collection_name=f"file-{file.id}")
+            VECTOR_DB_CLIENT.delete_collection(collection_name=f"file-{file.id}")
 
 
             docs = [
             docs = [
                 Document(
                 Document(
@@ -1015,7 +1015,7 @@ def process_file(
                 return {
                 return {
                     "status": True,
                     "status": True,
                     "collection_name": collection_name,
                     "collection_name": collection_name,
-                    "filename": file.meta.get("name", file.filename),
+                    "filename": file.filename,
                     "content": text_content,
                     "content": text_content,
                 }
                 }
         except Exception as e:
         except Exception as e:

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

@@ -483,7 +483,6 @@ def generate_ollama_batch_embeddings(
         r.raise_for_status()
         r.raise_for_status()
         data = r.json()
         data = r.json()
 
 
-        print(data)
         if "embeddings" in data:
         if "embeddings" in data:
             return data["embeddings"]
             return data["embeddings"]
         else:
         else: