From 24fff62d324e30a8f660817311657046cae057ff Mon Sep 17 00:00:00 2001 From: Ceceliachenen Date: Thu, 15 Aug 2024 14:16:15 +0800 Subject: [PATCH] fix multi_modal image (#156) --- src/pai_rag/integrations/readers/pai_image_reader.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/pai_rag/integrations/readers/pai_image_reader.py b/src/pai_rag/integrations/readers/pai_image_reader.py index 06a7814e..f9252621 100644 --- a/src/pai_rag/integrations/readers/pai_image_reader.py +++ b/src/pai_rag/integrations/readers/pai_image_reader.py @@ -88,7 +88,7 @@ def load_data( extra_info["file_path"] = file_path extra_info["file_name"] = os.path.basename(file_path) extra_info["image_url"] = image_url - image_doc = ImageDocument(image=image_url, extra_info=extra_info) + image_doc = ImageDocument(image_url=image_url, extra_info=extra_info) docs = [image_doc] # docs = self.load_image_urls([image_url], extra_info=extra_info) return docs