瀏覽代碼

fix: pass header to Tika if PDF_EXTRACT_IMAGES is true

ayan4m1 5 月之前
父節點
當前提交
039dec6820
共有 1 個文件被更改,包括 3 次插入0 次删除
  1. 3 0
      backend/open_webui/retrieval/loaders/main.py

+ 3 - 0
backend/open_webui/retrieval/loaders/main.py

@@ -99,6 +99,9 @@ class TikaLoader:
         else:
             headers = {}
 
+        if self.kwargs.get("PDF_EXTRACT_IMAGES") == True:
+            headers['X-Tika-PDFextractInlineImages'] = 'true'
+
         endpoint = self.url
         if not endpoint.endswith("/"):
             endpoint += "/"