fix(word_extractor): Fix type error and remove stream in ssrf_proxy (#11241)

Signed-off-by: -LAN- <laipz8200@outlook.com>
2024-12-02 10:24:03 +08:00 · 2024-12-02 10:24:03 +08:00 · 9601102885
commit 9601102885
parent 56c2d1cc55
2 changed files with 1 additions and 3 deletions
--- a/api/core/rag/extractor/word_extractor.py
+++ b/api/core/rag/extractor/word_extractor.py
@ -86,7 +86,7 @@ class WordExtractor(BaseExtractor):
                image_count += 1
                if rel.is_external:
                    url = rel.reltype
-                    response = ssrf_proxy.get(url, stream=True)
+                    response = ssrf_proxy.get(url)
                    if response.status_code == 200:
                        image_ext = mimetypes.guess_extension(response.headers["Content-Type"])
                        file_uuid = str(uuid.uuid4())