In [1]:
from piqard.utils.prompt_template import PromptTemplate
from piqard.language_models import CohereAPI


basic_prompt_template = """Answer the question.
Q: {{question}}
A:
"""
basic_prompt = PromptTemplate(template=basic_prompt_template)
cohere = CohereAPI(stop_token="\n")

cohere.query(basic_prompt.render(question="Who is the current CEO of twitter?"))
Out[1]:
'Jack Dorsey is the current CEO of twitter.'