File size: 529 Bytes
77320e4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
from ..prompt import FINAL_ANSWER_KEY, OBSERVATION_KEY, THOUGHT_KEY, PromptTemplate
class SimpleReactPrompt(PromptTemplate):
_input_variables = ["instruction", "agent_scratchpad"]
_template = "{instruction} \n{agent_scratchpad}"
_keywords = {
OBSERVATION_KEY: "[EOS]Observation:",
THOUGHT_KEY: "[SEP]",
FINAL_ANSWER_KEY: "[END]"
}
_name = 'SimpleReactPrompt'
_validate_template = True
_skip_on_failure = True
def __init__(self, **data):
super().__init__(**data)
|