typing

2025-08-01 03:42:30 +08:00
parent 25cb8d41bb
commit 70e12122b6
1 changed files with 1 additions and 1 deletions
--- a/src/llm_models/payload_content/message.py
+++ b/src/llm_models/payload_content/message.py
@@ -34,7 +34,7 @@ class Message:
 class MessageBuilder:
    def __init__(self):
        self.__role: RoleType = RoleType.User
-        self.__content: list[tuple[str, str] | str] = []
+        self.__content: list[tuple[str, str] | str | tuple[str, BytesIO]] = []
        self.__tool_call_id: str | None = None

    def set_role(self, role: RoleType = RoleType.User) -> "MessageBuilder":