fix: enhance image URL handling in response processing for multiple formats

Author: wxg0103

apps/models_provider/impl/gemini_model_provider/model/tti.py

@@ -4,6 +4,7 @@
 from openai import OpenAI
 
 from common.config.tokenizer_manage_config import TokenizerManage
+from common.utils.logger import maxkb_logger
 from models_provider.base_model_provider import MaxKBBaseModel
 from models_provider.impl.base_tti import BaseTextToImage
 
@@ -78,7 +79,7 @@ def generate_image(self, prompt: str, negative_prompt: str = None):
 
             for part in response.parts:
                 if part.text is not None:
-                    print(part.text)
+                    maxkb_logger.info(part.text)
                 elif part.inline_data is not None:
                     image_bytes = part.inline_data.data
                     img_base64 = base64.b64encode(image_bytes).decode("utf-8")

apps/models_provider/impl/volcanic_engine_model_provider/model/tti.py

@@ -60,8 +60,20 @@ def generate_image(self, prompt: str, negative_prompt: str = None):
             prompt=prompt,
             **self.params
         )
-        if imagesResponse.data[0].url:
-            file_urls.append(imagesResponse.data[0].url)
-        elif imagesResponse.data[0].b64_json:
-            file_urls.append(imagesResponse.data[0].b64_json)
+        # 如果 data 是列表，遍历所有图片
+        if isinstance(imagesResponse.data, list):
+            for item in imagesResponse.data:
+                # 优先使用 URL，其次使用 base64
+                if hasattr(item, 'url') and item.url:
+                    file_urls.append(item.url)
+                elif hasattr(item, 'b64_json') and item.b64_json:
+                    file_urls.append(item.b64_json)
+        else:
+            # 如果 data 是单个对象
+            item = imagesResponse.data
+            if hasattr(item, 'url') and item.url:
+                file_urls.append(item.url)
+            elif hasattr(item, 'b64_json') and item.b64_json:
+                file_urls.append(item.b64_json)
+
         return file_urls