Austin/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2024-11-16 11:42:29 +08:00
Code Issues Actions 4 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:
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