Fix: accept array content format in ChatMessage + add extra=allow, update .gitignore
This commit is contained in:
@@ -9,9 +9,10 @@ from pydantic import BaseModel, Field
|
||||
# ============================================================================
|
||||
|
||||
class ChatMessage(BaseModel):
|
||||
model_config = {"extra": "allow"}
|
||||
"""A chat message in the conversation."""
|
||||
role: Literal["system", "user", "assistant", "function", "tool", "developer"]
|
||||
content: Optional[str] = None
|
||||
content: Optional[Union[str, List[Dict[str, Any]]]] = None
|
||||
name: Optional[str] = None
|
||||
function_call: Optional[Dict[str, Any]] = None
|
||||
tool_calls: Optional[List[Dict[str, Any]]] = None
|
||||
@@ -81,7 +82,7 @@ class ChatCompletionResponse(BaseModel):
|
||||
class ChatCompletionChunkDelta(BaseModel):
|
||||
"""Delta content in streaming response."""
|
||||
role: Optional[str] = None
|
||||
content: Optional[str] = None
|
||||
content: Optional[Union[str, List[Dict[str, Any]]]] = None
|
||||
function_call: Optional[Dict[str, Any]] = None
|
||||
tool_calls: Optional[List[Dict[str, Any]]] = None
|
||||
|
||||
|
||||
Reference in New Issue
Block a user