query_rewrite_template = """You are an AI assistant tasked with reformulating user queries to improve retrieval in a RAG system. Given the original query, rewrite it to be more specific, detailed, and likely to retrieve relevant information.
Original query: {original_query}
Rewritten query:"""
query_rewrite_prompt = PromptTemplate(input_variables=["original_query"],template=query_rewrite_template)

# example query over the understanding climate change datasetoriginal_query = "气候变化对环境的影响是什么?"rewritten_query = rewrite_query(original_query)print("Original query:", original_query)print("\nRewritten query:", rewritten_query)

Original query: 气候变化对环境的影响是什么?
Rewritten query: 气候变化对各种生态系统的具体影响是什么，包括温度、降水模式、海平面和生物多样性的变化?

# Create a prompt template for step-back promptingstep_back_template = """You are an AI assistant tasked with generating broader, more general queries to improve context retrieval in a RAG system.Given the original query, generate a step-back query that is more general and can help retrieve relevant background information.
Original query: {original_query}
Step-back query:"""
step_back_prompt = PromptTemplate(input_variables=["original_query"],template=step_back_template)

original_query = "气候变化对环境的影响是什么?"step_back_query = generate_step_back_query(original_query)print("Original query:", original_query)print("\nStep-back query:", step_back_query)

Original query: 气候变化对环境的影响是什么?
Step-back query: 气候变化的一般影响是什么?

subquery_decomposition_template = """You are an AI assistant tasked with breaking down complex queries into simpler sub-queries for a RAG system.Given the original query, decompose it into 2-4 simpler sub-queries that, when answered together, would provide a comprehensive response to the original query.
Original query: {original_query}
example: What are the impacts of climate change on the environment?
Sub-queries:1. What are the impacts of climate change on biodiversity?2. How does climate change affect the oceans?3. What are the effects of climate change on agriculture?4. What are the impacts of climate change on human health?"""

subquery_decomposition_prompt = PromptTemplate(input_variables=["original_query"],template=subquery_decomposition_template)

original_query = "气候变化对环境的影响是什么?"sub_queries = decompose_query(original_query)print("\nSub-queries:")for i, sub_query in enumerate(sub_queries, 1):print(sub_query)

Sub-queries:Original query: 气候变化对环境的影响是什么?1. 气候变化如何影响生物多样性和生态系统?2. 气候变化对海洋环境和海洋生物有什么影响?3. 气候变化如何影响天气模式和极端天气事件?4. 气候变化对陆地环境，如森林和沙漠有什么影响?