perf: 控制台输出请求响应过程

This commit is contained in:
RockChinQ
2024-02-20 22:56:42 +08:00
parent 3258d5b255
commit 80858672b0
5 changed files with 37 additions and 4 deletions

View File

@@ -34,7 +34,12 @@ class Processor(stage.PipelineStage):
self.ap.logger.info(f"处理 {query.launcher_type.value}_{query.launcher_id} 的请求({query.query_id}): {message_text}")
if message_text.startswith('!') or message_text.startswith(''):
return self.cmd_handler.handle(query)
else:
return self.chat_handler.handle(query)
async def generator():
if message_text.startswith('!') or message_text.startswith(''):
async for result in self.cmd_handler.handle(query):
yield result
else:
async for result in self.chat_handler.handle(query):
yield result
return generator()