From 11acd99c103c338a20cc7ca8f4099f5aad0ddee0 Mon Sep 17 00:00:00 2001 From: Junyan Qin Date: Sat, 30 Aug 2025 22:41:07 +0800 Subject: [PATCH] feat: supports for command return image base64 --- pkg/pipeline/process/handlers/command.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/pkg/pipeline/process/handlers/command.py b/pkg/pipeline/process/handlers/command.py index db2187e3..92cebe02 100644 --- a/pkg/pipeline/process/handlers/command.py +++ b/pkg/pipeline/process/handlers/command.py @@ -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',