Skip to content

Commit 637cc04

Browse files
committed
feat: fix
1 parent fb93827 commit 637cc04

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

aperag/service/agent_chat_service.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -264,9 +264,7 @@ async def _handle_single_message(
264264
)
265265
)
266266
# Message Consumer
267-
consumer_task = asyncio.create_task(
268-
self._consume_messages_from_queue(chat_id, message_id, trace_id, message_queue, websocket)
269-
)
267+
consumer_task = asyncio.create_task(self._consume_messages_from_queue(message_queue, websocket))
270268
process_result, consumer_result = await asyncio.gather(process_task, consumer_task, return_exceptions=True)
271269

272270
# Handle process_task exceptions with unified error formatting
@@ -360,7 +358,7 @@ async def _stream_message_content(
360358
await asyncio.sleep(delay)
361359

362360
async def _consume_messages_from_queue(
363-
self, chat_id: str, message_id: str, trace_id: str, message_queue: AgentMessageQueue, websocket: WebSocket
361+
self, message_queue: AgentMessageQueue, websocket: WebSocket
364362
) -> List[AgentToolCallResultResponse]:
365363
"""
366364
Consume messages from queue, send to WebSocket, and collect AgentToolCallResultResponse messages.

0 commit comments

Comments
 (0)