diff --git a/request_llm/com_sparkapi.py b/request_llm/com_sparkapi.py
index 07f2853..c83710b 100644
--- a/request_llm/com_sparkapi.py
+++ b/request_llm/com_sparkapi.py
@@ -121,7 +121,7 @@ class SparkRequestInstance():
             self.time_to_exit_event.set()
 
         # 收到websocket关闭的处理
-        def on_close(ws):
+        def on_close(ws, *args):
             self.time_to_exit_event.set()
 
         # websocket