From fd1f1bfede2a7659c9b769db0600966415b4c9a7 Mon Sep 17 00:00:00 2001 From: Rajat Ahuja Date: Fri, 20 Jun 2025 12:01:04 -0400 Subject: [PATCH] CR comments --- src/routers/peers.py | 40 +++++++++++++++++++--------------------- 1 file changed, 19 insertions(+), 21 deletions(-) diff --git a/src/routers/peers.py b/src/routers/peers.py index 52817553..1bd67427 100644 --- a/src/routers/peers.py +++ b/src/routers/peers.py @@ -180,28 +180,26 @@ async def chat( workspace_id, peer_id, options.session_id, options.queries, options.stream ) - else: + async def parse_stream(): + try: + stream = await agent.chat( + workspace_id, + peer_id, + options.session_id, + options.queries, + stream=True, + ) + if isinstance(stream, AsyncMessageStreamManager): + async with stream as stream_manager: + async for text in stream_manager.text_stream: + yield text + except Exception as e: + logger.error(f"Error in stream: {str(e)}") + raise HTTPException(status_code=500, detail=str(e)) from e - async def parse_stream(): - try: - stream = await agent.chat( - workspace_id, - peer_id, - options.session_id, - options.queries, - stream=True, - ) - if type(stream) is AsyncMessageStreamManager: - async with stream as stream_manager: - async for text in stream_manager.text_stream: - yield text - except Exception as e: - logger.error(f"Error in stream: {str(e)}") - raise HTTPException(status_code=500, detail=str(e)) from e - - return StreamingResponse( - content=parse_stream(), media_type="text/event-stream", status_code=200 - ) + return StreamingResponse( + content=parse_stream(), media_type="text/event-stream", status_code=200 + ) @router.post(