fix(variables): user_message_text not provided

This commit is contained in:
Junyan Qin
2025-03-17 22:03:49 +08:00
parent 22306cb4ea
commit fa233e0a24

View File

@@ -60,11 +60,14 @@ class PreProcessor(stage.PipelineStage):
content_list = [] content_list = []
plain_text = ""
for me in query.message_chain: for me in query.message_chain:
if isinstance(me, platform_message.Plain): if isinstance(me, platform_message.Plain):
content_list.append( content_list.append(
llm_entities.ContentElement.from_text(me.text) llm_entities.ContentElement.from_text(me.text)
) )
plain_text += me.text
elif isinstance(me, platform_message.Image): elif isinstance(me, platform_message.Image):
if self.ap.provider_cfg.data['enable-vision'] and (self.ap.provider_cfg.data['runner'] != 'local-agent' or query.use_model.vision_supported): if self.ap.provider_cfg.data['enable-vision'] and (self.ap.provider_cfg.data['runner'] != 'local-agent' or query.use_model.vision_supported):
if me.base64 is not None: if me.base64 is not None:
@@ -72,6 +75,8 @@ class PreProcessor(stage.PipelineStage):
llm_entities.ContentElement.from_image_base64(me.base64) llm_entities.ContentElement.from_image_base64(me.base64)
) )
query.variables['user_message_text'] = plain_text
query.user_message = llm_entities.Message( query.user_message = llm_entities.Message(
role='user', role='user',
content=content_list content=content_list