Skip to content
Draft
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 9 additions & 0 deletions backend/onyx/chat/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -102,6 +102,14 @@ class MessageResponseIDInfo(BaseModel):
reserved_assistant_message_id: int


class MultiModelMessageResponseIDInfo(BaseModel):
"""Response info for multi-model chat: one user message, multiple assistant messages."""

user_message_id: int | None
reserved_assistant_message_ids: list[int] # One per model
model_names: list[str] # Display names for UI


class StreamingError(BaseModel):
error: str
stack_trace: str | None = None
Expand Down Expand Up @@ -200,6 +208,7 @@ class LLMMetricsContainer(BaseModel):
Packet
| StreamStopInfo
| MessageResponseIDInfo
| MultiModelMessageResponseIDInfo
| StreamingError
| UserKnowledgeFilePacket
| CreateChatSessionID
Expand Down
Loading
Loading