From 98e9e0842747a0193da9c2d0b54250aada015483 Mon Sep 17 00:00:00 2001
From: =Harry Zhu <=>
Date: Mon, 13 Apr 2026 23:40:40 +0800
Subject: [PATCH] fix: use UserPromptMessage instead of AssistantPromptMessage
 for response_format hint

---
 plugins/bedrock/models/llm/llm.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/plugins/bedrock/models/llm/llm.py b/plugins/bedrock/models/llm/llm.py
index 5538d69..875a007 100644
--- a/plugins/bedrock/models/llm/llm.py
+++ b/plugins/bedrock/models/llm/llm.py
@@ -140,7 +140,7 @@ def _code_block_mode_wrapper(
                 prompt_messages[0] = format_prompt
             else:
                 prompt_messages.insert(0, format_prompt)
-            prompt_messages.append(AssistantPromptMessage(content=f"\n```{response_format}"))
+            prompt_messages.append(UserPromptMessage(content=f"Output ```{response_format} block only."))
         return self._invoke(model, credentials, prompt_messages, model_parameters, tools, stop, stream, user)
 
     def _invoke(
@@ -203,7 +203,7 @@ def _invoke(
                                 prompt_messages[0] = format_prompt
                             else:
                                 prompt_messages.insert(0, format_prompt)
-                            prompt_messages.append(AssistantPromptMessage(content=f"\n```{response_format}"))
+                            prompt_messages.append(UserPromptMessage(content=f"Output ```{response_format} block only."))
                         else:
                             # For non-Anthropic models, just remove response_format parameter
                             model_parameters.pop("response_format", None)