Update main.py

main.py

@@ -29,7 +29,7 @@ SysPromptJson = "You are now in the role of an expert AI who can extract structu
 SysPromptList = "You are now in the role of an expert AI who can extract structured information from user request. All elements must be in double quotes. You must respond ONLY with a valid python List. Do not add any additional comments."
 SysPromptDefault = "You are an expert AI, complete the given task. Do not add any additional comments."
 SysPromptMd = "You are an expert AI who can create a structured report using information provided in the context from user request.The report should be in markdown format consists of markdown tables structured into subtopics. Do not add any additional comments."
-
+SysPromptSearch = "Extract from the input text to form a concise Google search query, focusing only on the main topic and omitting additional redundant details in max 8 words, include year if necessory, 2024, Do not add any additional comments."
 sys_prompts = {
     "offline": {
         "Chat": "You are an expert AI, complete the given task. Do not add any additional comments.",
@@ -65,7 +65,8 @@ async def generate_report(request: Request, query: QueryModel):
 
     # Combine query with user keywords
     if query.internet:
-        search_query = query_str 
+        search_query = together_response(query_str, model=llm_default_medium, SysPrompt=sys_prompt_output_format, max_tokens = 25)
+        print(f"query:{query_str}, formatted query:{search_query}")
         
         # Search for relevant URLs
         urls = search_brave(search_query, num_results=4)