change Message init type annotation to support other roles (#11942)

2026-07-06 23:08:16 +00:00 · 2025-06-28 03:12:13 +02:00
parent 9ab7d97b85
commit 0e10ae1cf1
1 changed files with 1 additions and 1 deletions
@@ -623,7 +623,7 @@ class Message(OpenAIObject):
    def __init__(
        self,
        content: Optional[str] = None,
-        role: Literal["assistant"] = "assistant",
+        role: Literal["assistant", "user", "system", "tool", "function"] = "assistant",
        function_call=None,
        tool_calls: Optional[list] = None,
        audio: Optional[ChatCompletionAudioResponse] = None,