add timeout=180s

app.py

@@ -1,6 +1,7 @@
 import openai
 import tiktoken
 
+import concurrent
 import collections
 import threading
 import datetime
@@ -110,12 +111,23 @@ def ask(query, max_attempt_times=3):
     return answer
 
 
-def askingChatGPT(qs, qas, min_interval_seconds=10):
+def askingChatGPT(qs, qas, min_interval_seconds=10, max_interval_seconds=120):
     
     for i, q in enumerate(qs):
         ask_start_time = time.time()
         
-        a = ask(q)
+        #a = ask(q)
+        def ask_(q):
+            with concurrent.futures.ThreadPoolExecutor() as executor:
+                future = executor.submit(ask, q)  # 提交函数调用任务
+                try:
+                    a = future.result(timeout=max_interval_seconds)  # 等待函数调用任务完成，超时时间为5秒
+                    return a
+                except concurrent.futures.TimeoutError:
+                    print('ask call timed out after', max_interval_seconds, 'seconds, retrying...')
+                    return ask_(q)  # 当超时时，重新调用函数
+        a = ask_(q)
+        
         qas.append({"q":q, "a":a})
         
         ask_end_time = time.time()