fix multi_modal image (#156)

aigc-apps · Aug 15, 2024 · 24fff62 · 24fff62
1 parent 684b896
commit 24fff62
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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