ElebobaandClaude Opus 5.5
d1f92407f9
CI / ruff · pytest (pull_request) Successful in 2m43s
- POST /dialog/{token}/events/stream: Server-Sent Events with the agent's steps («Смотрю вашу карту ЕМИАС…»,
«Ищу в клинических рекомендациях…»), the answer as the model writes it (OpenRouter stream=true), then the
final replies (same as /events, incl. the forbidden-output filter and sources). X-Accel-Buffering: no.
- Web: free-text messages use the stream — a live bubble with steps, typing dots and a cursor; falls back to
the plain request if the stream breaks.
- Phones: the bottom tab bar gets «Чат» (it had no way back to the chat).
Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
33 lines
1.4 KiB
Python
33 lines
1.4 KiB
Python
"""Live progress of the chat agent for streaming channels (web SSE).
|
|
|
|
The SSE endpoint sets a queue in the context; the agent pushes events while it works:
|
|
status — a tool step in plain Russian («Смотрю вашу карту ЕМИАС…»)
|
|
delta — a chunk of the answer as the model writes it
|
|
reset — the streamed text so far was a preface before tool calls: clear it
|
|
Channels without a queue (Telegram, tests) are unaffected. Deltas are the raw model text: the final reply
|
|
(forbidden-output filter, sources footer) always comes in the closing `done` event and replaces it.
|
|
"""
|
|
|
|
import asyncio
|
|
from contextvars import ContextVar
|
|
from typing import Any
|
|
|
|
STREAM: ContextVar[asyncio.Queue[tuple[str, Any]] | None] = ContextVar("agent_stream", default=None)
|
|
|
|
TOOL_STATUS = {
|
|
"get_patient_record": "Смотрю вашу карту ЕМИАС…",
|
|
"check_drug_for_patient": "Проверяю препарат по вашей карте: болезни, непереносимости, другие лекарства…",
|
|
"get_drug_info": "Открываю инструкцию к препарату (ГРЛС)…",
|
|
"search_guidelines": "Ищу в клинических рекомендациях Минздрава…",
|
|
}
|
|
|
|
|
|
def active() -> bool:
|
|
return STREAM.get() is not None
|
|
|
|
|
|
def emit(kind: str, payload: Any) -> None:
|
|
q = STREAM.get()
|
|
if q is not None:
|
|
q.put_nowait((kind, payload))
|