minor: 'role' must be system/assistant/tool, but case insensitive for now (#7499)
This commit is contained in:
@@ -313,6 +313,18 @@ class ChatCompletionMessageGenericParam(BaseModel):
|
|||||||
reasoning_content: Optional[str] = None
|
reasoning_content: Optional[str] = None
|
||||||
tool_calls: Optional[List[ToolCall]] = Field(default=None, examples=[None])
|
tool_calls: Optional[List[ToolCall]] = Field(default=None, examples=[None])
|
||||||
|
|
||||||
|
@field_validator("role", mode="before")
|
||||||
|
@classmethod
|
||||||
|
def _normalize_role(cls, v):
|
||||||
|
if isinstance(v, str):
|
||||||
|
v_lower = v.lower()
|
||||||
|
if v_lower not in {"system", "assistant", "tool"}:
|
||||||
|
raise ValueError(
|
||||||
|
"'role' must be one of 'system', 'assistant', or 'tool' (case-insensitive)."
|
||||||
|
)
|
||||||
|
return v_lower
|
||||||
|
raise ValueError("'role' must be a string")
|
||||||
|
|
||||||
|
|
||||||
class ChatCompletionMessageUserParam(BaseModel):
|
class ChatCompletionMessageUserParam(BaseModel):
|
||||||
role: Literal["user"]
|
role: Literal["user"]
|
||||||
|
|||||||
Reference in New Issue
Block a user