text: """Your goal is to maximize your return, i.e. the sum of the rewards you receive.I will give you an observation, reward, terminiation flag, truncation flag, and the return so far, formatted as:Observation: <observation>Reward: <reward>Termination: <termination>Truncation: <truncation>Return: <sum_of_rewards>You will respond with an action, formatted as:Action: <action>where you replace <action> with your actual action.Do nothing else but return the action."""        self.action_parser = RegexParser(            regex=r"Action: (.*)", output_keys=["action"], default_output_key="action"        )        self.message_history = []        self.ret = 0    def random_action(self):        action = self.env.action_space.sample()        return action    def reset(self):        self.message_history = [            SystemMessage(content=self.docs),            SystemMessage(content=self.instructions),        ]    def observe(self, obs, rew=0, term=False, trunc=False, info=None):        self.ret += rew        obs_message = f"""Observation: {obs}Reward: {rew}Termination: {term}Truncation: {trunc}Return: {self.ret}        """        self.message_history.append(HumanMessage(content=obs_message))        return obs_message    def _act(self):