From 9ac4ef1ba165289c7b3d5b5350622617385b5386 Mon Sep 17 00:00:00 2001 From: MrKan Date: Sun, 19 Apr 2026 11:58:45 +0300 Subject: [PATCH] =?UTF-8?q?=D0=BF=D0=BE=D0=B4=D0=B4=D0=B5=D1=80=D0=B6?= =?UTF-8?q?=D0=BA=D0=B0=20chat=5Fid?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- lambda_agent_api/agent_api.py | 7 +++++-- tests/manual.py | 3 ++- 2 files changed, 7 insertions(+), 3 deletions(-) diff --git a/lambda_agent_api/agent_api.py b/lambda_agent_api/agent_api.py index d130bcf..e2e5346 100644 --- a/lambda_agent_api/agent_api.py +++ b/lambda_agent_api/agent_api.py @@ -2,6 +2,7 @@ import logging from typing import Callable, Optional, AsyncIterator import aiohttp import asyncio +from urllib.parse import urljoin from lambda_agent_api.server import * from lambda_agent_api.client import * @@ -27,10 +28,12 @@ class AgentApi: agent_id: str, url: str, callback: Optional[Callable[[ServerMessage], None]] = None, - on_disconnect: Optional[Callable[['AgentApi'], None]] = None + on_disconnect: Optional[Callable[['AgentApi'], None]] = None, + chat_id: int = 0 # значение по умолчанию для обратной совместимости ): self.id = agent_id # ID агента для словаря - self.url = url + self.chat_id = chat_id + self.url = urljoin(url, f"{chat_id}/") self.callback = callback self.on_disconnect = on_disconnect diff --git a/tests/manual.py b/tests/manual.py index a39802b..b801a2d 100644 --- a/tests/manual.py +++ b/tests/manual.py @@ -7,7 +7,8 @@ from lambda_agent_api.server import MsgEventTextChunk, MsgEventToolCallChunk, Ms async def main(): - api = AgentApi("agent-1", "ws://localhost:8000/agent_ws/") + chat_id = input("Chat id: ") or 0 + api = AgentApi("agent-1", "ws://localhost:8000/agent_ws/", chat_id=chat_id) await api.connect() while True: