Spaces:
Running
Running
File size: 1,575 Bytes
11fa0f1 22d0605 1d7579f 11fa0f1 1d7579f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 |
from typing import List
import langchain
from langchain.prompts import PromptTemplate
class InstructionGenerationTemplate(PromptTemplate):
"""A custom prompt template for generating instructions."""
input_variables: List[str] = ["num_questions", "context", "instruction_format", "lang", "additional_rules"]
template = """
You are a highly intelligent language model trained to assist with a variety of language tasks. Your task here is to generate {num_questions} diverse questions or instructions based on the context provided below:
Context:
{context}
Please follow these rules:
{additional_rules}
Please generate the instructions in the {instruction_format} format and in {lang} language. Remember to adhere to the rules mentioned above.
"""
template_format = "f-string"
def format(self, **kwargs):
"""Format the prompt."""
return self.template.format(**kwargs)
class AnswerGenerationTemplate(langchain.prompts.PromptTemplate):
"""A custom prompt template for generating answers to questions."""
input_variables: List[str] = ["questions", "additional_rules"]
template = """
You are a highly intelligent language model tasked with providing answers to the following questions :
Questions:
{questions}
Please follow these rules:
{additional_rules}
"""
template_format = "f-string"
def format(self, **kwargs):
"""Format the prompt."""
return self.template.format(**kwargs)
|