Ver código fonte

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

crazywoola 9 meses atrás
pai
commit
90d2c01218

+ 3 - 3
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