```python @tracker('main') def get_action(self, obs): obs = self.llm(obs) action = self.sub_agent.get_action(obs) # tracked by @tracker('subagent') ``` Ideally this would log the full cost ('main') and the cost of the subagent