Update app.py

app.py

@@ -51,8 +51,13 @@ async def get_response_with_search(query, model, use_embeddings, num_calls=3, te
     else:
         context = "\n".join([f"{result['title']}\n{result['body']}\nSource: {result['href']}" for result in search_results])
 
-    prompt = f"""Using the following context from web search results:
+    system_message = """You are a highly knowledgeable AI assistant tasked with providing comprehensive and accurate information based on web search results. 
+    Your goal is to synthesize the given context into a coherent and detailed response that directly addresses the user's query. 
+    Please ensure that your response is well-structured, factual, and cites sources where appropriate."""
+
+    user_message = f"""Using the following context from web search results:
 {context}
+
 Write a detailed and complete research document that fulfills the following user request: '{query}'
 After writing the document, please provide a list of sources used in your response."""
 
@@ -63,7 +68,10 @@ After writing the document, please provide a list of sources used in your respon
     try:
         for _ in range(num_calls):
             for response in client.chat_completion(
-                messages=[{"role": "user", "content": prompt}],
+                messages=[
+                    {"role": "system", "content": system_message},
+                    {"role": "user", "content": user_message}
+                ],
                 max_tokens=6000,
                 temperature=temperature,
                 stream=True,