tikendraw commited on
Commit
b03c265
·
1 Parent(s): b1fb492

retries if fails to generate answer

Browse files
Files changed (1) hide show
  1. app/utils.py +2 -2
app/utils.py CHANGED
@@ -20,12 +20,12 @@ from core.llms.utils import user_message_with_images
20
  from PIL import Image
21
  from streamlit.runtime.uploaded_file_manager import UploadedFile
22
  from core.prompts.decision_prompt import PLAN_SYSTEM_PROMPT, COTorDAPromptOutput, Decision
23
- from tenacity import retry, stop_after_attempt, wait_random_exponential
24
 
25
 
26
 
27
 
28
- @retry(stop=stop_after_attempt(3))
29
  def cot_or_da_func(problem: str, llm: BaseLLM = None, **kwargs) -> COTorDAPromptOutput:
30
 
31
  cot_decision_message = [
 
20
  from PIL import Image
21
  from streamlit.runtime.uploaded_file_manager import UploadedFile
22
  from core.prompts.decision_prompt import PLAN_SYSTEM_PROMPT, COTorDAPromptOutput, Decision
23
+ from tenacity import retry, stop_after_attempt, wait_incrementing
24
 
25
 
26
 
27
 
28
+ @retry(stop=stop_after_attempt(3), wait=wait_incrementing())
29
  def cot_or_da_func(problem: str, llm: BaseLLM = None, **kwargs) -> COTorDAPromptOutput:
30
 
31
  cot_decision_message = [