add timeout=30s

app.py

@@ -89,7 +89,8 @@ def ask(query, max_attempt_times=3):
     answer = None
     attempt_times = 0
     while answer is None and attempt_times<max_attempt_times:
-        attempt_times += 1
+        wait_time = int(attempt_times*10)
+        time.sleep(wait_time)
         try:
             answer = openai.ChatCompletion.create(
                 model="gpt-3.5-turbo",
@@ -103,15 +104,17 @@ def ask(query, max_attempt_times=3):
                 idx = openai.api_keys.index(openai.api_key)
                 idx = (idx + 1) % len(openai.api_keys)
                 openai.api_key = openai.api_keys[idx]
-                attempt_times -= 0.7
+                attempt_times += 0
                 print(f"switch api_key")
-            wait_time = int(attempt_times*10)
-            print(f"retry in {wait_time} seconds...")
-            time.sleep(wait_time)
+            elif "Request timed out: HTTPSConnectionPool(host='api.openai.com', port=443): Read timed out. (read timeout=600)" in str(e):
+                attempt_times += 1000
+            else:
+                attempt_times += 1
+                print(f"retry in {attempt_times*10} seconds...")
     return answer
 
 
-def askingChatGPT(qs, qas, min_interval_seconds=3, max_interval_seconds=30):
+def askingChatGPT(qs, qas, min_interval_seconds=3, max_interval_seconds=15, max_retry_times=3):
     
     for i, q in enumerate(qs):
         ask_start_time = time.time()
@@ -127,7 +130,12 @@ def askingChatGPT(qs, qas, min_interval_seconds=3, max_interval_seconds=30):
                 print('ask call timed out after', timeout, 'seconds, retrying...')
             executor.shutdown(wait=False)
             return ask_(q, timeout*2)  # 当超时时，重新调用函数
-        a = ask_(q, max_interval_seconds)
+        
+        retry_times = 0
+        a = None
+        while a is None and retry_times<max_retry_times:
+            a = ask_(q, max_interval_seconds)
+            retry_times += 1
         
         qas.append({"q":q, "a":a})
         

dialogues_set/qas.json

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:687f82b17048923678a63782d0482c81ea21e33875c4bc70358483cf22814af4
-size 23202294
+oid sha256:5d9eb4ab3d7f781cc43111beaebc295c888b625f02f61d6407a3f6e610c8f2a4
+size 24089657