feat: 处理下stop的请求，防止报错

carduo · Jun 26, 2024 · 7e56844 · 7e56844
1 parent c9a065c
commit 7e56844
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/backend/bisheng/chat/client.py b/src/backend/bisheng/chat/client.py
@@ -179,6 +179,9 @@ async def handle_gpts_message(self, message: Dict[any, any]):
         if not message:
             return
         logger.debug(f'receive client message, client_key: {self.client_key} message: {message}')
+        if message.get('action') == 'stop':
+            logger.info(f'need stop agent, client_key: {self.client_key}, message: {message}')
+            return
 
         inputs = message.get('inputs', {})
         input_msg = inputs.get('input')