feat: supports for command return image base64

This commit is contained in:
Junyan Qin
2025-08-30 22:41:07 +08:00
parent 589f61931a
commit 11acd99c10

View File

@@ -74,7 +74,7 @@ class CommandHandler(handler.MessageHandler):
self.ap.logger.info(f'Command({query.query_id}) error: {self.cut_str(str(ret.error))}')
yield entities.StageProcessResult(result_type=entities.ResultType.CONTINUE, new_query=query)
elif ret.text is not None or ret.image_url is not None:
elif ret.text is not None or ret.image_url is not None or ret.image_base64 is not None:
content: list[provider_message.ContentElement] = []
if ret.text is not None:
@@ -83,6 +83,9 @@ class CommandHandler(handler.MessageHandler):
if ret.image_url is not None:
content.append(provider_message.ContentElement.from_image_url(ret.image_url))
if ret.image_base64 is not None:
content.append(provider_message.ContentElement.from_image_base64(ret.image_base64))
query.resp_messages.append(
provider_message.Message(
role='command',