Method action

AgentBench.old/src/agent.py:36–48

(self, extend_messages: List[dict] = None)

        self.history.append(message)

    def action(self, extend_messages: List[dict] = None) -> str:
        extend = []
        if extend_messages:
            if isinstance(extend_messages, list):
                extend.extend(extend_messages)
            elif isinstance(extend_messages, dict):
                extend.append(extend_messages)
            else:
                raise Exception("Invalid extend_messages")
        result = self.model_inference(self.history + extend)
        self.history.extend(extend)
        self.history.append({"role": "agent", "content": result})
        return result
    
    def _calc_segments(self, msg: str):
        segments = 0