693ed39de4
Since the only usage of prompt is within LLMs, it is reasonable to keep it within the LLM module. This way, it would be easier to discover module, and make the code base less complicated. Changes: * Move prompt components into llms * Bump version 0.3.1 * Make pip install dependencies in eager mode --------- Co-authored-by: ian <ian@cinnamon.is>
120 lines
3.4 KiB
Python
120 lines
3.4 KiB
Python
# flake8: noqa
|
|
|
|
from kotaemon.llms import PromptTemplate
|
|
|
|
zero_shot_planner_prompt = PromptTemplate(
|
|
template="""You are an AI agent who makes step-by-step plans to solve a problem under the help of external tools.
|
|
For each step, make one plan followed by one tool-call, which will be executed later to retrieve evidence for that step.
|
|
You should store each evidence into a distinct variable #E1, #E2, #E3 ... that can be referred to in later tool-call inputs.
|
|
|
|
##Available Tools##
|
|
{tool_description}
|
|
|
|
##Output Format (Replace '<...>')##
|
|
#Plan1: <describe your plan here>
|
|
#E1: <toolname>[<input here>] (eg. Search[What is Python])
|
|
#Plan2: <describe next plan>
|
|
#E2: <toolname>[<input here, you can use #E1 to represent its expected output>]
|
|
And so on...
|
|
|
|
##Your Task##
|
|
{task}
|
|
|
|
##Now Begin##
|
|
"""
|
|
)
|
|
|
|
one_shot_planner_prompt = PromptTemplate(
|
|
template="""You are an AI agent who makes step-by-step plans to solve a problem under the help of external tools.
|
|
For each step, make one plan followed by one tool-call, which will be executed later to retrieve evidence for that step.
|
|
You should store each evidence into a distinct variable #E1, #E2, #E3 ... that can be referred to in later tool-call inputs.
|
|
|
|
##Available Tools##
|
|
{tool_description}
|
|
|
|
##Output Format##
|
|
#Plan1: <describe your plan here>
|
|
#E1: <toolname>[<input here>]
|
|
#Plan2: <describe next plan>
|
|
#E2: <toolname>[<input here, you can use #E1 to represent its expected output>]
|
|
And so on...
|
|
|
|
##Example##
|
|
Task: What is the 4th root of 64 to the power of 3?
|
|
#Plan1: Find the 4th root of 64
|
|
#E1: Calculator[64^(1/4)]
|
|
#Plan2: Raise the result from #Plan1 to the power of 3
|
|
#E2: Calculator[#E1^3]
|
|
|
|
##Your Task##
|
|
{task}
|
|
|
|
##Now Begin##
|
|
"""
|
|
)
|
|
|
|
|
|
few_shot_planner_prompt = PromptTemplate(
|
|
template="""You are an AI agent who makes step-by-step plans to solve a problem under the help of external tools.
|
|
For each step, make one plan followed by one tool-call, which will be executed later to retrieve evidence for that step.
|
|
You should store each evidence into a distinct variable #E1, #E2, #E3 ... that can be referred to in later tool-call inputs.
|
|
|
|
##Available Tools##
|
|
{tool_description}
|
|
|
|
##Output Format (Replace '<...>')##
|
|
#Plan1: <describe your plan here>
|
|
#E1: <toolname>[<input>]
|
|
#Plan2: <describe next plan>
|
|
#E2: <toolname>[<input, you can use #E1 to represent its expected output>]
|
|
And so on...
|
|
|
|
##Examples##
|
|
{fewshot}
|
|
|
|
##Your Task##
|
|
{task}
|
|
|
|
##Now Begin##
|
|
"""
|
|
)
|
|
|
|
zero_shot_solver_prompt = PromptTemplate(
|
|
template="""You are an AI agent who solves a problem with my assistance. I will provide step-by-step plans(#Plan) and evidences(#E) that could be helpful.
|
|
Your task is to briefly summarize each step, then make a short final conclusion for your task.
|
|
|
|
##My Plans and Evidences##
|
|
{plan_evidence}
|
|
|
|
##Example Output##
|
|
First, I <did something> , and I think <...>; Second, I <...>, and I think <...>; ....
|
|
So, <your conclusion>.
|
|
|
|
##Your Task##
|
|
{task}
|
|
|
|
##Now Begin##
|
|
"""
|
|
)
|
|
|
|
few_shot_solver_prompt = PromptTemplate(
|
|
template="""You are an AI agent who solves a problem with my assistance. I will provide step-by-step plans and evidences that could be helpful.
|
|
Your task is to briefly summarize each step, then make a short final conclusion for your task.
|
|
|
|
##My Plans and Evidences##
|
|
{plan_evidence}
|
|
|
|
##Example Output##
|
|
First, I <did something> , and I think <...>; Second, I <...>, and I think <...>; ....
|
|
So, <your conclusion>.
|
|
|
|
##Example##
|
|
{fewshot}
|
|
|
|
##Your Task##
|
|
{task}
|
|
|
|
##Now Begin##
|
|
"""
|
|
)
|