Spaces:
Paused
Paused
Update main.py
Browse files
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=
|
| 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()
|