Update app.py

app.py

@@ -135,7 +135,7 @@ Search Query: transformer architectures
 User Question: What is the history of the transformer model?
 Search Query: transformer model history
 """
-def to_query(provider,message):
+def to_query(provider,question):
     try:
         agent = LlamaCppAgent(
                 provider,
@@ -143,7 +143,12 @@ def to_query(provider,message):
                 predefined_messages_formatter_type=MessagesFormatterType.GEMMA_2,
                 debug_output=True,
             )
-    
+
+            message="""
+Now, rewrite the following question:
+User Question: %s
+Search Query:
+    """%question
         
         settings = provider.get_provider_default_settings()
         messages = BasicChatHistory()
@@ -207,6 +212,7 @@ def respond(
         provider = LlamaCppPythonProvider(llm)
 
         query = to_query(provider,message)
+        print("<query>")
         print(f"from {message} to {query}")
         text = retriever_tool(query=f"{query}")