Update ImageReader file loading logic (run-llama#15848)

raspawar · Oct 7, 2024 · e7fd925 · e7fd925
1 parent 1523371
commit e7fd925
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/...ndex-integrations/readers/llama-index-readers-file/llama_index/readers/file/image/base.py b/...ndex-integrations/readers/llama-index-readers-file/llama_index/readers/file/image/base.py
@@ -82,7 +82,7 @@ def load_data(
         # load document image
         if fs:
             with fs.open(path=file) as f:
-                image = Image.open(f.read())
+                image = Image.open(BytesIO(f.read()))
         else:
             image = Image.open(file)
 

diff --git a/llama-index-integrations/readers/llama-index-readers-file/pyproject.toml b/llama-index-integrations/readers/llama-index-readers-file/pyproject.toml
@@ -51,7 +51,7 @@ license = "MIT"
 maintainers = ["FarisHijazi", "Haowjy", "ephe-meral", "hursh-desai", "iamarunbrahma", "jon-chuang", "mmaatouk", "ravi03071991", "sangwongenip", "thejessezhang"]
 name = "llama-index-readers-file"
 readme = "README.md"
-version = "0.2.0"
+version = "0.2.1"
 
 [tool.poetry.dependencies]
 python = ">=3.8.1,<4.0"