Allow more flexible assistant and system response (#1256)
This commit is contained in:
@@ -200,11 +200,6 @@ class CompletionStreamResponse(BaseModel):
|
||||
usage: Optional[UsageInfo] = None
|
||||
|
||||
|
||||
class ChatCompletionMessageGenericParam(BaseModel):
|
||||
role: Literal["system", "assistant"]
|
||||
content: str
|
||||
|
||||
|
||||
class ChatCompletionMessageContentTextPart(BaseModel):
|
||||
type: Literal["text"]
|
||||
text: str
|
||||
@@ -225,6 +220,11 @@ ChatCompletionMessageContentPart = Union[
|
||||
]
|
||||
|
||||
|
||||
class ChatCompletionMessageGenericParam(BaseModel):
|
||||
role: Literal["system", "assistant"]
|
||||
content: Union[str, List[ChatCompletionMessageContentTextPart]]
|
||||
|
||||
|
||||
class ChatCompletionMessageUserParam(BaseModel):
|
||||
role: Literal["user"]
|
||||
content: Union[str, List[ChatCompletionMessageContentPart]]
|
||||
|
||||
Reference in New Issue
Block a user