MahaNeta / utils /cot.py
ankush-003's picture
init
10757ec
raw
history blame contribute delete
829 Bytes
"""
This example is to show Chain Of Thought
"""
from langchain import PromptTemplate
from load_llm import load_llm
template = """Answer the question based on the context below. If the
question cannot be answered using the information provided answer
with "I don't know".
Q: Roger has 5 tennis balls. He buys 2 more cans of tennis balls. Each can contains 3 tennis balls. How many
tennis balls does he have now?
A: Roger started with 5 balls. 2 cans of 3 tennis balls is 6 tennis balls. 5+6 = 11.The answer is 11.
Q: The cafetaria has 23 apples. If they used 20 apples for lunch and bought 6 more, how many apples do they have?
"""
prompt_template = PromptTemplate(
input_variables=[],
template=template
)
prompt = prompt_template.format(
)
llm = load_llm()
print(llm(prompt))