Update main.py

main.py

@@ -59,12 +59,12 @@ async def generate_report(request: Request, query: QueryModel):
     query_str = query.topic
     description = query.description
     user_id = query.user_id
-    internet = query.internet
+    internet = "online" if query.internet else "offline"
     sys_prompt_output_format = sys_prompts[internet][query.output_format]
     data_format = query.data_format
 
     # Combine query with user keywords
-    if internet:
+    if query.internet:
         search_query = description
         # Search for relevant URLs
         try:
@@ -76,10 +76,10 @@ async def generate_report(request: Request, query: QueryModel):
             prompt = f"#### COMPLETE THE TASK: {description} #### IN THE CONTEXT OF ### CONTEXT: {query_str} USING THE #### SCRAPED DATA:{additional_context}"
             
         except Exception as e:
-            internet = False
+            query.internet = False
             print("failed to search/scrape results, falling back to LLM response")
     
-    if not internet:
+    if not query.internet:
         prompt = f"#### COMPLETE THE TASK: {description} #### IN THE CONTEXT OF ### CONTEXT: {query_str}"
         all_text_with_urls = [("","")]