Fix: return type mismatch (#664)

Co-authored-by: alcholiclg <ligongshengzju@foxmail.com>

Author: alcholiclg

ms_agent/tools/docling/doc_loader.py

@@ -41,7 +41,7 @@ def html_handle_figure(self, element: Tag, doc: DoclingDocument) -> None:
 
     if img_url:
         if img_url.startswith('data:'):
-            img_pil, ext = load_image_from_uri_to_pil(img_url)
+            img_pil = load_image_from_uri_to_pil(img_url)
         else:
             if not img_url.startswith('http'):
                 img_url = validate_url(img_url=img_url, backend=self)
@@ -98,7 +98,7 @@ def html_handle_image(self, element: Tag, doc: DoclingDocument) -> None:
 
     if img_url:
         if img_url.startswith('data:'):
-            img_pil, ext = load_image_from_uri_to_pil(img_url)
+            img_pil = load_image_from_uri_to_pil(img_url)
         else:
             if not img_url.startswith('http'):
                 img_url = validate_url(img_url=img_url, backend=self)

ms_agent/utils/download.py

@@ -334,7 +334,7 @@ def load_image_from_url_to_pil(url: str) -> 'Image.Image':
         return None
 
 
-def load_image_from_uri_to_pil(uri: str) -> tuple:
+def load_image_from_uri_to_pil(uri: str) -> 'Image.Image':
     """
     Load image from URI as a PIL Image object and extract its format extension.
     URI format: data:[<mime>][;base64],<encoded>
@@ -352,18 +352,16 @@ def load_image_from_uri_to_pil(uri: str) -> tuple:
             raw = base64.b64decode(encoded)
         else:
             raw = encoded.encode('utf-8')
-        m = re.match(r'data:(image/[^;]+)', header)
-        ext = m.group(1).split('/')[-1] if m else 'bin'
         img = Image.open(BytesIO(raw))
-        return img, ext
+        return img
     except ValueError as e:
         print(f'Error parsing URI format: {e}')
         return None
     except base64.binascii.Error as e:
         print(f'Error decoding base64 data: {e}')
         return None
     except IOError as e:
-        print(f'Error opening image: {e}')
+        print(f'Error opening image with PIL: {e}')
         return None
     except Exception as e:
         print(f'Unexpected error loading image from URI: {e}')