Allow more flexible assistant and system response (#1256)

This commit is contained in:
Christopher Chou
2024-08-30 11:51:44 -07:00
committed by GitHub
parent 79ece2c51f
commit 51c554d812
4 changed files with 95 additions and 8 deletions

View File

@@ -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]]