From ace1f2c1a44b6ef3ccdd64893fdc9b3e2d8918a5 Mon Sep 17 00:00:00 2001 From: guojialiang Date: Mon, 29 Jul 2024 22:03:04 +0800 Subject: [PATCH] =?UTF-8?q?fix:=E8=A7=A3=E5=86=B3=E6=B5=81=E5=BC=8F?= =?UTF-8?q?=E8=BE=93=E5=87=BA=E9=97=AE=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../app/callback_handler/agent_callback_handler.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/src/mindpilot/app/callback_handler/agent_callback_handler.py b/src/mindpilot/app/callback_handler/agent_callback_handler.py index 9fa216f..1bb13df 100644 --- a/src/mindpilot/app/callback_handler/agent_callback_handler.py +++ b/src/mindpilot/app/callback_handler/agent_callback_handler.py @@ -81,11 +81,12 @@ class AgentExecutorAsyncIteratorCallbackHandler(AsyncIteratorCallbackHandler): self.queue.put_nowait(dumps(data)) async def on_llm_end(self, response: LLMResult, **kwargs: Any) -> None: - data = { - "status": AgentStatus.llm_end, - "text": response.generations[0][0].message.content, - } - self.queue.put_nowait(dumps(data)) + pass + # data = { + # "status": AgentStatus.llm_end, + # "text": response.generations[0][0].message.content, + # } + # self.queue.put_nowait(dumps(data)) async def on_llm_error( self, error: Exception | KeyboardInterrupt, **kwargs: Any