Building a prompt that can accept multiple keys

[johnhu33]

How to build a query pipeline that can take multiple keys in the prompt template?

prompt_str = "answer the query according to the context. ##context：{context}，##query：{query}"
prompt_tmpl = PromptTemplate(prompt_str)

[dosubot[bot]]

To build a query pipeline that can take multiple keys in the prompt template, you can use the PromptTemplate class to define your template and then format it with the necessary data. Here's an example based on your provided prompt_str and prompt_tmpl:

from llama_index.core.prompts.base import PromptTemplate
from llama_index.core.prompts.prompt_type import PromptType

# Define the prompt template
prompt_str = "answer the query according to the context. ##context：{context}，##query：{query}"
prompt_tmpl = PromptTemplate(template=prompt_str, prompt_type=PromptType.CUSTOM)

# Example data to fill the template
data = {
    "context": "This is the context information.",
    "query": "What is the answer to the query?"
}

# Build the query pipeline
formatted_prompt = prompt_tmpl.template.format(**data)
print(formatted_prompt)

This will output:

answer the query according to the context. ##context：This is the context information.，##query：What is the answer to the query?

This approach allows you to dynamically insert multiple keys into your prompt template using the PromptTemplate class ^[1].

To continue talking to Dosu, mention @dosu.