timeout setting

app.py

@@ -142,7 +142,7 @@ def split_draft(draft, split_char = '\n\n'):
     # print(f"The draft answer has {len(draft_paragraphs)}")
     return draft_paragraphs
 
-def split_draft_openai(question, answer, NUM_PARAGRAPHS = 3):
+def split_draft_openai(question, answer, NUM_PARAGRAPHS = 4):
     split_prompt = f'''
 Split the answer of the question into multiple paragraphs with each paragraph containing a complete thought.
 The answer should be splited into less than {NUM_PARAGRAPHS} paragraphs.
@@ -349,7 +349,7 @@ def rat(question):
 
         # query = get_query(question, answer)
         print(f"{datetime.now()} [INFO] Generating query ...")
-        res = run_with_timeout(get_query_wrapper, 10, question, answer)
+        res = run_with_timeout(get_query_wrapper, 30, question, answer)
         if not res:
             print(f"{datetime.now()} [INFO] Generating query timeout, skipping...")
             continue
@@ -359,7 +359,7 @@ def rat(question):
 
         print(f"{datetime.now()} [INFO] Crawling network pages ...")
         # content = get_content(query)
-        res = run_with_timeout(get_content_wrapper, 10, query)
+        res = run_with_timeout(get_content_wrapper, 30, query)
         if not res:
             print(f"{datetime.now()} [INFO] Parsing network pages timeout, skipping ...")
             continue
@@ -372,7 +372,7 @@ def rat(question):
                 break
             print(f"{datetime.now()} [INFO] Revising answers with retrieved network pages...[{j}/{min(len(content),LIMIT)}]")
             # answer = get_revise_answer(question, answer, c)
-            res = run_with_timeout(get_revise_answer_wrapper, 15, question, answer, c)
+            res = run_with_timeout(get_revise_answer_wrapper, 30, question, answer, c)
             if not res:
                 print(f"{datetime.now()} [INFO] Revising answers timeout, skipping ...")
                 continue