From 5ec0302e19ae9d9a1f94b7b90dec9ba1f44659ed Mon Sep 17 00:00:00 2001
From: manatlan <manatlan@gmail.com>
Date: Fri, 15 Nov 2024 10:10:04 +0100
Subject: FIX: fix the url in the markdown output (md compliance), and the
 chunk/ImageResponse to str ! (#2353)

add fix_url ( replace ' ' by '+' (to be markdown compliant))
---
 g4f/image.py                   | 11 ++++++++---
 g4f/providers/base_provider.py |  2 +-
 2 files changed, 9 insertions(+), 4 deletions(-)

(limited to 'g4f')

diff --git a/g4f/image.py b/g4f/image.py
index 6561b83a..556ec43d 100644
--- a/g4f/image.py
+++ b/g4f/image.py
@@ -23,6 +23,11 @@ EXTENSIONS_MAP: dict[str, str] = {
     "image/webp": "webp",
 }
 
+def fix_url(url:str) -> str:
+    """ replace ' ' by '+' (to be markdown compliant)"""
+    return url.replace(" ","+")
+    
+
 def to_image(image: ImageType, is_svg: bool = False) -> Image:
     """
     Converts the input image to a PIL Image object.
@@ -212,12 +217,12 @@ def format_images_markdown(images: Union[str, list], alt: str, preview: Union[st
         str: The formatted markdown string.
     """
     if isinstance(images, str):
-        result = f"[![{alt}]({preview.replace('{image}', images) if preview else images})]({images})"
+        result = f"[![{alt}]({fix_url(preview.replace('{image}', images) if preview else images)})]({fix_url(images)})"
     else:
         if not isinstance(preview, list):
             preview = [preview.replace('{image}', image) if preview else image for image in images]
         result = "\n".join(
-            f"[![#{idx+1} {alt}]({preview[idx]})]({image})"
+            f"[![#{idx+1} {alt}]({fix_url(preview[idx])})]({fix_url(image)})"
             #f'[<img src="{preview[idx]}" width="200" alt="#{idx+1} {alt}">]({image})'
             for idx, image in enumerate(images)
         )
@@ -302,4 +307,4 @@ class ImageRequest:
         self.options = options
 
     def get(self, key: str):
-        return self.options.get(key)
\ No newline at end of file
+        return self.options.get(key)
diff --git a/g4f/providers/base_provider.py b/g4f/providers/base_provider.py
index a03dcbba..5d48f2e0 100644
--- a/g4f/providers/base_provider.py
+++ b/g4f/providers/base_provider.py
@@ -248,7 +248,7 @@ class AsyncGeneratorProvider(AsyncProvider):
             str: The created result as a string.
         """
         return "".join([
-            chunk async for chunk in cls.create_async_generator(model, messages, stream=False, **kwargs) 
+            str(chunk) async for chunk in cls.create_async_generator(model, messages, stream=False, **kwargs) 
             if not isinstance(chunk, (Exception, FinishReason))
         ])
 
-- 
cgit v1.2.3