From 90d2c0121843243c3130f81cd9057f971b423e78 Mon Sep 17 00:00:00 2001
From: crazywoola <100913391+crazywoola@users.noreply.github.com>
Date: Sat, 27 Jul 2024 00:07:31 +0800
Subject: [PATCH] Feat/6725 can not get image url from cogview tool (#6728)

---
 api/core/tools/provider/builtin/cogview/tools/cogview3.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/api/core/tools/provider/builtin/cogview/tools/cogview3.py b/api/core/tools/provider/builtin/cogview/tools/cogview3.py
index bb2720196f..89ffcf3347 100644
--- a/api/core/tools/provider/builtin/cogview/tools/cogview3.py
+++ b/api/core/tools/provider/builtin/cogview/tools/cogview3.py
@@ -30,7 +30,6 @@ class CogView3Tool(BuiltinTool):
         if not prompt:
             return self.create_text_message('Please input prompt')
         # get size
-        print(tool_parameters.get('prompt', 'square'))
         size = size_mapping[tool_parameters.get('size', 'square')]
         # get n
         n = tool_parameters.get('n', 1)
@@ -58,8 +57,9 @@ class CogView3Tool(BuiltinTool):
         result = []
         for image in response.data:
             result.append(self.create_image_message(image=image.url))
-        result.append(self.create_text_message(
-            f'\nGenerate image source to Seed ID: {seed_id}'))
+            result.append(self.create_json_message({
+                "url": image.url,
+            }))
         return result
 
     @staticmethod