template = """Context:
<<context>>
Answer the following question:
Q: <<query>>
A:
"""
prompt = template.replace('<<context>>', context).replace('<<query>>', query)