Fix token tracking in Agent class to use token_process instead of _token_process

This commit is contained in:
Brandon Hancock
2025-02-28 11:44:51 -05:00
parent 88d8079dcd
commit ef48cbe971

View File

@@ -327,7 +327,7 @@ class Agent(BaseAgent):
request_within_rpm_limit=(
self._rpm_controller.check_or_wait if self._rpm_controller else None
),
callbacks=[TokenCalcHandler(self._token_process)],
callbacks=[TokenCalcHandler(self.token_process)],
)
def get_delegation_tools(self, agents: List[BaseAgent]):