kth-qa / magic /prompts.py
erseux's picture
new huggingface structure
cc0bd0f
raw
history blame
801 Bytes
from langchain import PromptTemplate
EXAMPLE_PROMPT = PromptTemplate(
template=">Course Description\n{page_content}\n----------\nSource: {course}",
input_variables=["page_content", "course"],
)
template ="""
You are a study counselor for KTH.
Given the following extracted parts of course descriptions and a question, create a short final answer with references ("SOURCES").
If you don't know the answer, just say that you don't know. Don't try to make up an answer.
Just answer the question and don't add any extra information.
Only return sources that contain the answer to the question.
ALWAYS return a "SOURCES" part in your answer.
QUESTION: {question}
=========
{context}
=========
FINAL ANSWER:"""
PROMPT = PromptTemplate(template=template, input_variables=["question", "context"])