Feat/6725 can not get image url from cogview tool (#6728)

langgenius · Jul 26, 2024 · 90d2c01 · 90d2c01
1 parent 83af503
commit 90d2c01
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/api/core/tools/provider/builtin/cogview/tools/cogview3.py b/api/core/tools/provider/builtin/cogview/tools/cogview3.py
@@ -30,7 +30,6 @@ def _invoke(self,
         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 @@ def _invoke(self,
         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