Skip to content

Commit 961e25f

Browse files
authored
fix: better bedrock message handler close #10976 (#11317)
Signed-off-by: yihong0618 <[email protected]>
1 parent 138bf69 commit 961e25f

File tree

1 file changed

+7
-0
lines changed
  • api/core/model_runtime/model_providers/bedrock/llm

1 file changed

+7
-0
lines changed

api/core/model_runtime/model_providers/bedrock/llm/llm.py

+7
Original file line numberDiff line numberDiff line change
@@ -196,6 +196,13 @@ def _generate_with_converse(
196196
if model_info["support_tool_use"] and tools:
197197
parameters["toolConfig"] = self._convert_converse_tool_config(tools=tools)
198198
try:
199+
# for issue #10976
200+
conversations_list = parameters["messages"]
201+
# if two consecutive user messages found, combine them into one message
202+
for i in range(len(conversations_list) - 2, -1, -1):
203+
if conversations_list[i]["role"] == conversations_list[i + 1]["role"]:
204+
conversations_list[i]["content"].extend(conversations_list.pop(i + 1)["content"])
205+
199206
if stream:
200207
response = bedrock_client.converse_stream(**parameters)
201208
return self._handle_converse_stream_response(

0 commit comments

Comments
 (0)