add timeout loop

Lichang-Chen · Nov 13, 2023 · 08e51de · 08e51de
1 parent 952c99e
commit 08e51de
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/utils.py b/utils.py
@@ -43,6 +43,7 @@ def evaluate_by_chatgpt(data, output_entry, correctness_entry, gpt_model="gpt-4"
                     timeout=5)
                 break
             except:
+                print("Timeout, retrying...")
                 time.sleep(5)  # Wait for 5 seconds before retrying
 
         output_text = response['choices'][0]['message']['content']
@@ -97,6 +98,7 @@ def check_same_by_chatgpt(data, output_entry, gpt_model="gpt-4", load_json=False
 
                     break
                 except:
+                    print("Timeout, retrying...")
                     time.sleep(5)  # Wait for 5 seconds before retrying