stop infos when done streaming answers

This commit is contained in:
Evan Lohn
2025-01-24 15:03:10 -08:00
parent eea6f2749a
commit 2d8486bac4
4 changed files with 22 additions and 1 deletions

View File

@@ -70,7 +70,7 @@ class StreamStopReason(Enum):
class StreamStopInfo(BaseModel):
stop_reason: StreamStopReason
stream_type: Literal["", "sub_questions", "sub_answer"] = ""
stream_type: Literal["", "sub_questions", "sub_answer", "main_answer"] = ""
# used to identify the stream that was stopped for agent search
level: int | None = None
level_question_nr: int | None = None