Browse Source

fix:解决流式输出问题

main
guojialiang 1 year ago
parent
commit
ace1f2c1a4
1 changed files with 6 additions and 5 deletions
  1. +6
    -5
      src/mindpilot/app/callback_handler/agent_callback_handler.py

+ 6
- 5
src/mindpilot/app/callback_handler/agent_callback_handler.py View File

@@ -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


Loading…
Cancel
Save