Przeglądaj źródła

Merge pull request #12896 from gaby/fix-12893

fix: Optional content from files routes
Tim Jaeryang Baek 3 miesięcy temu
rodzic
commit
07347f518a

+ 3 - 1
backend/open_webui/retrieval/vector/dbs/qdrant.py

@@ -51,7 +51,9 @@ class QdrantClient:
         self.client.create_collection(
             collection_name=collection_name_with_prefix,
             vectors_config=models.VectorParams(
-                size=dimension, distance=models.Distance.COSINE, on_disk=self.QDRANT_ON_DISK
+                size=dimension,
+                distance=models.Distance.COSINE,
+                on_disk=self.QDRANT_ON_DISK,
             ),
         )
 

+ 4 - 2
backend/open_webui/routers/files.py

@@ -173,7 +173,8 @@ async def list_files(user=Depends(get_verified_user), content: bool = Query(True
 
     if not content:
         for file in files:
-            del file.data["content"]
+            if "content" in file.data:
+                del file.data["content"]
 
     return files
 
@@ -214,7 +215,8 @@ async def search_files(
 
     if not content:
         for file in matching_files:
-            del file.data["content"]
+            if "content" in file.data:
+                del file.data["content"]
 
     return matching_files