| 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) | |