Update main.py

main.py

@@ -103,17 +103,17 @@ async def generate_report(query: QueryModel):
             urls, optimized_search_query = search_brave(search_query, num_results=8)
             all_text_with_urls = fetch_and_extract_content(data_format, urls, query_str)
             reference = limit_tokens(str(all_text_with_urls))
-            prompt = user_prompt_final.format(description,reference)
+            user_prompt_final = user_prompt_final.format(description,reference)
         except Exception as e:
             print(e)
             query.internet = False
             print("failed to search/scrape results, falling back to LLM response")
 
     if not query.internet:
-        prompt = f"#### COMPLETE THE TASK: {description} #### IN THE CONTEXT OF ### CONTEXT: {query_str}"
+        user_prompt_final = prompt_user["offline"][query.output_format].format(description)
         system_prompt_final = prompt_system["offline"]
 
-    md_report = together_response(prompt, model=llm_default_medium, SysPrompt=system_prompt_final)
+    md_report = together_response(user_prompt_final, model=llm_default_medium, SysPrompt=system_prompt_final)
 
     if user_id != "test":
         insert_data(user_id, query_str, description, str(all_text_with_urls), md_report)