Skip to content

Commit 57fa97d

Browse files
committed
fix tool role
1 parent 101042a commit 57fa97d

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/inferencesh/models/llm.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -255,7 +255,8 @@ def merge_tool_calls(messages: List[ContextMessage]) -> List[Dict[str, Any]]:
255255
user_input_image = input_data.image
256256
multipart = multipart or input_data.image is not None
257257

258-
user_msg = ContextMessage(role=ContextMessageRole.USER, text=user_input_text, image=user_input_image)
258+
input_role = input_data.role if hasattr(input_data, "role") else ContextMessageRole.USER
259+
user_msg = ContextMessage(role=input_role, text=user_input_text, image=user_input_image)
259260

260261
input_data.context.append(user_msg)
261262

0 commit comments

Comments
 (0)