From f300b4c6fb0f51b62a7b0701cac53b3d71465480 Mon Sep 17 00:00:00 2001 From: collhoun <2904yr@mail.ru> Date: Tue, 7 Apr 2026 10:37:15 +0300 Subject: [PATCH] =?UTF-8?q?#7=20=D0=94=D0=BE=D0=B1=D0=B0=D0=B2=D0=B8=D1=82?= =?UTF-8?q?=D1=8C=20=D0=BE=D1=81=D0=BD=D0=BE=D0=B2=D0=BD=D1=8B=D0=B5=20?= =?UTF-8?q?=D0=BC=D0=BE=D0=B4=D0=B5=D0=BB=D0=B8=20=D0=B8=D0=B2=D0=B5=D0=BD?= =?UTF-8?q?=D1=82=D0=BE=D0=B2=20=D0=B0=D0=B3=D0=B5=D0=BD=D1=82=D0=B0.=20?= =?UTF-8?q?=D0=94=D0=BE=D0=B1=D0=B0=D0=B2=D0=BB=D0=B5=D0=BD=D1=8B=20=D0=BE?= =?UTF-8?q?=D1=81=D0=BD=D0=BE=D0=B2=D0=BD=D1=8B=D0=B5=20=D0=B8=D0=BD=D0=B2?= =?UTF-8?q?=D0=B5=D1=82=D1=8B?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- lambda_agent_api/server.py | 124 +++++++++++++++++++++++++------------ 1 file changed, 86 insertions(+), 38 deletions(-) diff --git a/lambda_agent_api/server.py b/lambda_agent_api/server.py index dbeaaa5..e5882f7 100644 --- a/lambda_agent_api/server.py +++ b/lambda_agent_api/server.py @@ -1,72 +1,120 @@ from pydantic import BaseModel, Field, TypeAdapter from enum import Enum -from typing import Literal, Annotated, Union +from typing import Literal, Annotated, Union, Any, Dict, Optional -__all__ = ['EServerMessage', 'MsgStatus', 'MsgError', 'MsgEventTextChunk', 'MsgEventEnd', 'AgentEventUnion', 'ServerMessage'] +__all__ = [ + 'EServerMessage', 'MsgStatus', 'MsgError', 'MsgGracefulDisconnect', + 'MsgEventTextChunk', 'MsgEventToolCallChunk', 'MsgEventToolResult', + 'MsgEventCustomUpdate', 'MsgEventEnd', + 'AgentEventUnion', 'ServerMessage' +] class EServerMessage(str, Enum): STATUS = "STATUS" ERROR = "ERROR" GRACEFUL_DISCONNECT = "GRACEFUL_DISCONNECT" + + # Ивенты агента AGENT_EVENT_TEXT_CHUNK = "AGENT_EVENT_TEXT_CHUNK" + AGENT_EVENT_TOOL_CALL_CHUNK = "AGENT_EVENT_TOOL_CALL_CHUNK" # Новое + AGENT_EVENT_TOOL_RESULT = "AGENT_EVENT_TOOL_RESULT" # Новоеы + AGENT_EVENT_CUSTOM_UPDATE = "AGENT_EVENT_CUSTOM_UPDATE" # Новое AGENT_EVENT_END = "AGENT_EVENT_END" class MsgStatus(BaseModel): - """ - Отправляется сервером при открытии соединения с клиентом. - Будет дополнен информацией о готовности агента принимать сообщения. - """ + """Отправляется сервером при открытии соединения с клиентом.""" type: Literal[EServerMessage.STATUS] = EServerMessage.STATUS -class MsgEventTextChunk(BaseModel): - """ - Чанк текста ответа агента. - """ - type: Literal[EServerMessage.AGENT_EVENT_TEXT_CHUNK] = EServerMessage.AGENT_EVENT_TEXT_CHUNK - text: str - - -class MsgEventEnd(BaseModel): - """ - Агент закончил генерацию ответа. - """ - type: Literal[EServerMessage.AGENT_EVENT_END] = EServerMessage.AGENT_EVENT_END - tokens_used: int - - class MsgError(BaseModel): - """ - Неопределенная ошибка в работе агента. - """ + """Неопределенная ошибка в работе агента.""" type: Literal[EServerMessage.ERROR] = EServerMessage.ERROR code: str details: str class MsgGracefulDisconnect(BaseModel): - """ - Отправляется перед завершением работы контейнера с агентом. Например, при долгом бездействии. - Нужно, чтобы отделять обрыв соединения из-за ошибки с необходимостью повторного подключения. - Приход этого сообщения означает, что агент осознанно завершает работу с клиентом по какой-то причине. - Для дальнейшего взаимодействия нужно снова обратиться к мастеру. - """ + """Отправляется перед завершением работы контейнера с агентом.""" type: Literal[EServerMessage.GRACEFUL_DISCONNECT] = EServerMessage.GRACEFUL_DISCONNECT -AgentEventUnion = Union[MsgEventTextChunk, MsgEventEnd] +# ------------------------------------------------------------------ +# AGENT EVENTS (События генерации) +# ------------------------------------------------------------------ + +class MsgEventTextChunk(BaseModel): + """Чанк текста ответа агента.""" + type: Literal[EServerMessage.AGENT_EVENT_TEXT_CHUNK] = EServerMessage.AGENT_EVENT_TEXT_CHUNK + text: str + # Новое: "main" (главный агент) или "tools:..." (субагент, если будем использовать) + source: str = "main" # пока везде будет main +class MsgEventToolCallChunk(BaseModel): + """Агент решил использовать инструмент и генерирует аргументы.""" + type: Literal[EServerMessage.AGENT_EVENT_TOOL_CALL_CHUNK] = EServerMessage.AGENT_EVENT_TOOL_CALL_CHUNK + tool_name: Optional[str] = Field( + None, description="Имя инструмента (приходит обычно в первом чанке)") + args_chunk: Optional[str] = Field( + None, description="Кусок JSON-аргументов") + source: str = "main" + + +class MsgEventToolResult(BaseModel): + """Инструмент отработал и вернул результат.""" + type: Literal[EServerMessage.AGENT_EVENT_TOOL_RESULT] = EServerMessage.AGENT_EVENT_TOOL_RESULT + tool_name: str + result: Any # Может быть строкой, словарем или списком + source: str = "main" + + +class MsgEventCustomUpdate(BaseModel): + """Кастомный прогресс (например, скачивание файла) изнутри инструмента.""" + type: Literal[EServerMessage.AGENT_EVENT_CUSTOM_UPDATE] = EServerMessage.AGENT_EVENT_CUSTOM_UPDATE + payload: Dict[str, Any] = Field( + ..., description="Любые данные о прогрессе (status, progress и т.д.)") + source: str = "main" + + +class MsgEventEnd(BaseModel): + """Агент закончил генерацию ответа.""" + type: Literal[EServerMessage.AGENT_EVENT_END] = EServerMessage.AGENT_EVENT_END + tokens_used: int + + +# ------------------------------------------------------------------ +# UNIONS & ADAPTERS +# ------------------------------------------------------------------ + +# Обновлено: добавили новые модели в Union +AgentEventUnion = Union[ + MsgEventTextChunk, + MsgEventToolCallChunk, + MsgEventToolResult, + MsgEventCustomUpdate, + MsgEventEnd +] + +# Обновлено: добавили новые модели в Union адаптера ServerMessage = TypeAdapter(Annotated[ - Union[MsgStatus, MsgEventTextChunk, MsgEventEnd, MsgError, MsgGracefulDisconnect], + Union[ + MsgStatus, + MsgError, + MsgGracefulDisconnect, + MsgEventTextChunk, + MsgEventToolCallChunk, + MsgEventToolResult, + MsgEventCustomUpdate, + MsgEventEnd + ], Field(discriminator="type") ]) """ -Объединяет все типы исходящих сообщений в одно для удобной автоматической десериализации.\n -Pydantic сам определит нужный тип в зависимости от поля ``type``.\n -Использование:\n -msg = ServerMessage.model_validate_json(json) -""" \ No newline at end of file +Объединяет все типы исходящих сообщений в одно для удобной автоматической десериализации. +Pydantic сам определит нужный тип в зависимости от поля `type`. +Использование: +msg = ServerMessage.validate_json(json_str) +"""