pvanand commited on
Commit
da118f4
·
verified ·
1 Parent(s): 2062c62

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +3 -4
main.py CHANGED
@@ -146,17 +146,15 @@ def json_from_text(text):
146
  # Use Fuzzy JSON loading
147
  return loads(json_out)
148
 
149
- @retry(tries=3, delay=10)
150
  def generate_topics(user_input, num_topics, previous_queries):
151
  prompt = prompt_topics.format(user_input=user_input, num_topics=num_topics)
152
  response_topics = together_response(prompt, model=llm_default_medium, SysPrompt=SysPromptList, temperature=1)
153
  subtopics = json_from_text(response_topics)
154
  return subtopics
155
 
156
- @retry(tries=3, delay=5)
157
  def generate_subtopics(main_task,user_input,num_topics,excluded_topics):
158
- print("trying")
159
- raise ValueError("Intentional Exception for testing retry decorator")
160
  excluded_topics = ",".join(excluded_topics)
161
  prompt = prompt_subtopics.format(main_task = main_task,user_input=user_input, num_topics=num_topics, excluded_topics=excluded_topics)
162
  response_topics = together_response(prompt, model=llm_default_medium, SysPrompt=SysPromptList, temperature=1)
@@ -169,6 +167,7 @@ def generate_report(topic, description):
169
  md_report = together_response(prompt, model = llm_default_medium, SysPrompt = SysPromptMdOffline)
170
  return md_to_html(md_report)
171
 
 
172
  def get_images(query, num_results):
173
  time.sleep(0.5)
174
  ddgs = DDGS()
 
146
  # Use Fuzzy JSON loading
147
  return loads(json_out)
148
 
149
+ @retry(tries=3, delay=0.5)
150
  def generate_topics(user_input, num_topics, previous_queries):
151
  prompt = prompt_topics.format(user_input=user_input, num_topics=num_topics)
152
  response_topics = together_response(prompt, model=llm_default_medium, SysPrompt=SysPromptList, temperature=1)
153
  subtopics = json_from_text(response_topics)
154
  return subtopics
155
 
156
+ @retry(tries=3, delay=0.5)
157
  def generate_subtopics(main_task,user_input,num_topics,excluded_topics):
 
 
158
  excluded_topics = ",".join(excluded_topics)
159
  prompt = prompt_subtopics.format(main_task = main_task,user_input=user_input, num_topics=num_topics, excluded_topics=excluded_topics)
160
  response_topics = together_response(prompt, model=llm_default_medium, SysPrompt=SysPromptList, temperature=1)
 
167
  md_report = together_response(prompt, model = llm_default_medium, SysPrompt = SysPromptMdOffline)
168
  return md_to_html(md_report)
169
 
170
+ @retry(tries=3, delay=0.5)
171
  def get_images(query, num_results):
172
  time.sleep(0.5)
173
  ddgs = DDGS()