From d832f279098c46955556446943ec7e860a57aa08 Mon Sep 17 00:00:00 2001 From: MrKan Date: Fri, 3 Apr 2026 15:06:00 +0300 Subject: [PATCH] =?UTF-8?q?REPL=20=D0=B2=20=D1=80=D1=83=D1=87=D0=BD=D0=BE?= =?UTF-8?q?=D0=BC=20=D1=82=D0=B5=D1=81=D1=82=D0=B5?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tests/manual.py | 20 ++++++++++++++------ 1 file changed, 14 insertions(+), 6 deletions(-) diff --git a/tests/manual.py b/tests/manual.py index 3b920e3..d59754c 100644 --- a/tests/manual.py +++ b/tests/manual.py @@ -1,4 +1,5 @@ import asyncio +import traceback from lambda_agent_api.agent_api import AgentApi from lambda_agent_api.server import MsgEventTextChunk @@ -12,13 +13,20 @@ async def main(): api = AgentApi("agent-1", "ws://localhost:8000/agent_ws/", callback=my_callback) await api.connect() - try: - async for chunk in api.send_message("Привет, агент!"): - if isinstance(chunk, MsgEventTextChunk): - print(chunk.text, end="", flush=True) + while True: + try: + prompt = await asyncio.get_event_loop().run_in_executor(None, input, ">>> ") + print("Agent: ", end="") + async for chunk in api.send_message(prompt): + if isinstance(chunk, MsgEventTextChunk): + print(chunk.text, end="", flush=True) + print("\n") + except KeyboardInterrupt: + break + except: + traceback.print_exc() - finally: - await api.close() + await api.close() asyncio.run(main()) \ No newline at end of file