| 1234567891011121314151617181920212223242526272829303132333435363738394041424344 |
- class ReActParser(object):
- def __init__(self):
- self.action = '\nAction:'
- self.action_input = '\nAction Input:'
- self.action_input_stop = '\nObservation:'
- self.observation = '\nObservation:'
- self.observation_stop = '\nThought:'
- def parse_latest_plugin_call(self, text):
- action = self.action
- action_input = self.action_input
- observation = self.action_input_stop
- plugin_name, plugin_args = '', ''
- i = text.rfind(action)
- j = text.rfind(action_input)
- k = text.rfind(observation)
- if 0 <= i < j: # If the text has `Action` and `Action input`,
- if k < j: # but does not contain `Observation`,
- # then it is likely that `Observation` is ommited by the LLM,
- # because the output text may have discarded the stop word.
- text = text.rstrip() + observation # Add it back.
- k = text.rfind(observation)
- plugin_name = text[i + len(action): j].strip()
- plugin_args = text[j + len(action_input): k].strip()
- text = text[:k]
- return plugin_name, plugin_args, text
- def _extract_first_target(self, text, start_flag, end_flag):
- target = ''
- i = text.find(start_flag)
- if i != -1:
- j = text.find(end_flag, i)
- if j != -1:
- target = text[i+len(start_flag):j].strip()
- else:
- target = text[i+len(start_flag):].strip()
- return target
- def get_first_observation(self, text):
- return self._extract_first_target(text, self.observation, self.observation_stop)
- def get_first_action_input(self, text):
- return self._extract_first_target(text, self.action_input, self.action_input_stop)
|
