From 343ecb3ad466159ee19cedb70e0b8f8ecba6fdac Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=8E=84=E5=87=A4=E7=A7=91=E6=8A=80?= Date: Tue, 4 Mar 2025 08:54:27 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96chat=E4=B8=AD=E5=BC=82?= =?UTF-8?q?=E6=AD=A5=E8=B0=83=E7=94=A8=E6=96=B9=E5=BC=8F?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- core/connection.py | 19 ++++--------------- 1 file changed, 4 insertions(+), 15 deletions(-) diff --git a/core/connection.py b/core/connection.py index 28be3218..2f3e1f76 100644 --- a/core/connection.py +++ b/core/connection.py @@ -203,23 +203,11 @@ class ConnectionHandler: return not self.is_device_verified - def async_run(self, coro): - loop = asyncio.new_event_loop() - asyncio.set_event_loop(loop) - try: - return loop.run_until_complete(coro) - finally: - loop.close() - def chat(self, query): if self.isNeedAuth(): self.llm_finish_task = True - loop = asyncio.new_event_loop() - asyncio.set_event_loop(loop) - try: - loop.run_until_complete(self._check_and_broadcast_auth_code()) - finally: - loop.close() + future = asyncio.run_coroutine_threadsafe(self._check_and_broadcast_auth_code(), self.loop) + future.result() return True self.dialogue.put(Message(role="user", content=query)) @@ -228,7 +216,8 @@ class ConnectionHandler: try: start_time = time.time() # 使用带记忆的对话 - memory_str = self.async_run(self.memory.query_memory(query)) + future = asyncio.run_coroutine_threadsafe(self.memory.query_memory(query), self.loop) + memory_str = future.result() self.logger.bind(tag=TAG).info(f"记忆内容: {memory_str}") llm_responses = self.llm.response(