AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-14 19:35:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
crazywoola 2024-07-27 00:07:31 +08:00 committed by GitHub
parent 83af50368f
commit 90d2c01218
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api/core/tools/provider/builtin/cogview/tools/cogview3.py
View File

@ -30,7 +30,6 @@ class CogView3Tool(BuiltinTool):
if not prompt: if not prompt:
return self.create_text_message('Please input prompt') return self.create_text_message('Please input prompt')
# get size # get size
print(tool_parameters.get('prompt', 'square'))
size = size_mapping[tool_parameters.get('size', 'square')] size = size_mapping[tool_parameters.get('size', 'square')]
# get n # get n
n = tool_parameters.get('n', 1) n = tool_parameters.get('n', 1)
@ -58,8 +57,9 @@ class CogView3Tool(BuiltinTool):
result = [] result = []
for image in response.data: for image in response.data:
result.append(self.create_image_message(image=image.url)) result.append(self.create_image_message(image=image.url))
result.append(self.create_text_message( result.append(self.create_json_message({
f'\nGenerate image source to Seed ID: {seed_id}')) "url": image.url,
}))
return result return result
@staticmethod @staticmethod