From d16329c1af29a61864b8534fdf4499ca2688ec86 Mon Sep 17 00:00:00 2001
From: binary-husky <qingxu.fu@outlook.com>
Date: Mon, 14 Aug 2023 11:31:05 +0800
Subject: [PATCH] resolve sparkapi on_close error

---
 request_llm/com_sparkapi.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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