Update app.py

app.py

@@ -29,7 +29,7 @@ class Agent1:
         rephrase_prompt = PromptTemplate(
             input_variables=["query"],
             template="""
-            Your task is to rephrase the given query into one or more concise, search-engine-friendly formats.
+            Rephrase the given query into one or more concise, search-engine-friendly formats.
             If the query contains multiple distinct questions, split them.
             Provide ONLY the rephrased queries without any additional text or explanations, one per line.
             
@@ -279,6 +279,7 @@ def ask_question(question, temperature, top_p, repetition_penalty, web_search):
 
     max_attempts = 3
     context_reduction_factor = 0.7
+    agent2_max_attempts = 2
 
     for attempt in range(max_attempts):
         try:
@@ -349,15 +350,19 @@ def ask_question(question, temperature, top_p, repetition_penalty, web_search):
             else:
                 answer = full_response.strip()
 
-            if not agent2.validate_response(question, answer):
-                follow_up_query = agent2.generate_follow_up_query(question, answer)
-                follow_up_results = agent1.process(follow_up_query)
-                follow_up_docs = [Document(page_content=result["text"], metadata={"source": result["link"], "query": follow_up_query}) for results in follow_up_results.values() for result in results if result["text"]]
-                database.add_documents(follow_up_docs)
-                context_str += "\n" + "\n".join([f"Follow-up Query: {doc.metadata['query']}\nSource: {doc.metadata['source']}\nContent: {doc.page_content}" for doc in follow_up_docs])
-                formatted_prompt = prompt_val.format(context=context_str, question=question)
-                full_response = generate_chunked_response(model, formatted_prompt)
-                answer = full_response.strip()
+            for agent2_attempt in range(agent2_max_attempts):
+                if agent2.validate_response(question, answer):
+                    break
+                
+                if agent2_attempt < agent2_max_attempts - 1:
+                    follow_up_query = agent2.generate_follow_up_query(question, answer)
+                    follow_up_results = agent1.process(follow_up_query)
+                    follow_up_docs = [Document(page_content=result["text"], metadata={"source": result["link"], "query": follow_up_query}) for results in follow_up_results.values() for result in results if result["text"]]
+                    database.add_documents(follow_up_docs)
+                    context_str += "\n" + "\n".join([f"Follow-up Query: {doc.metadata['query']}\nSource: {doc.metadata['source']}\nContent: {doc.page_content}" for doc in follow_up_docs])
+                    formatted_prompt = prompt_val.format(context=context_str, question=question)
+                    full_response = generate_chunked_response(model, formatted_prompt)
+                    answer = full_response.strip()
 
             if web_search:
                 sources = set(doc.metadata['source'] for doc in web_docs)