platform/agent_api
5
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

manual тест для проверки новых моделей

Browse source
This commit is contained in:
Егор Кандрушин 2026-04-15 02:09:41 +03:00
parent cbba336049
commit 507268b2c1
1 changed files with 15 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

18
tests/manual.py
View file

@ -2,7 +2,7 @@ import asyncio
import traceback import traceback
from lambda_agent_api.agent_api import AgentApi from lambda_agent_api.agent_api import AgentApi
from lambda_agent_api.server import MsgEventTextChunk from lambda_agent_api.server import MsgEventTextChunk, MsgEventToolCallChunk, MsgEventToolResult
def my_callback(message): def my_callback(message):
@ -17,9 +17,21 @@ async def main():
try: try:
prompt = await asyncio.get_event_loop().run_in_executor(None, input, ">>> ") prompt = await asyncio.get_event_loop().run_in_executor(None, input, ">>> ")
print("Agent: ", end="") print("Agent: ", end="")
is_tool = False
async for chunk in api.send_message(prompt): async for chunk in api.send_message(prompt):
if isinstance(chunk, MsgEventTextChunk): match chunk:
print(chunk.text, end="", flush=True) case MsgEventTextChunk():
is_tool = True
print(chunk.text, end="", flush=True)
case MsgEventToolCallChunk():
if not is_tool:
print(f"\n\n### TOOL CALL: ({chunk.tool_name}) ", end="", flush=True)
is_tool = True
print(chunk.args_chunk, end="", flush=True)
case MsgEventToolResult():
is_tool = False
print(f"\nResult: {chunk.result}\n\n", end="", flush=True)
print("\n") print("\n")
except KeyboardInterrupt: except KeyboardInterrupt:
break break