Retry 2 additional times to avoid random time out error.

app.py

@@ -8,10 +8,10 @@ from utils.file_operations import hash_name
 #   2. update QQ group and Organization cards
 #   4. add auto_polishing function
 #   5. Use some simple method for simple tasks (including: writing abstract, conclusion, generate keywords, generate figures...)
-#       5.1 Use GPT 3.5 for abstract, conclusion, ...
+#       5.1 Use GPT 3.5 for abstract, conclusion, ... (or may not)
 #       5.2 Use local LLM to generate keywords, figures, ...
 #       5.3 Use embedding to find most related papers (find a paper dataset)
-#       5.4 Use Segmantic Scholar API instead of Arxiv API.
+#       5.4 Use Semantic Scholar API instead of Arxiv API.
 #   6. get logs when the procedure is not completed.
 
 openai_key = os.getenv("OPENAI_API_KEY")

auto_backgrounds.py

@@ -2,6 +2,7 @@ from utils.references import References
 from utils.file_operations import hash_name, make_archive, copy_templates
 from section_generator import section_generation_bg, keywords_generation, figures_generation, section_generation
 import logging
+import time
 
 TOTAL_TOKENS = 0
 TOTAL_PROMPTS_TOKENS = 0
@@ -104,6 +105,19 @@ def generate_draft(title, description="", template="ICLR2022", model="gpt-4"):
             message = f"Failed to generate {section}. {type(e).__name__} was raised:  {e}"
             print(message)
             logging.info(message)
+            max_attempts = 2
+            # re-try `max_attempts` time
+            for i in range(max_attempts):
+                time.sleep(20)
+                try:
+                    usage = section_generation(paper, section, destination_folder, model=model)
+                    log_usage(usage, section)
+                    e = None
+                except Exception as e:
+                    pass
+                if e is None:
+                    break
+
 
     input_dict = {"title": title, "description": description, "generator": "generate_draft"}
     filename = hash_name(input_dict) + ".zip"