Support Tool calls datasets with thinking traces

### Reminder

- [x] I have read the above rules and searched the existing issues.

### Description

Is there a way to add the thinking traces before the tool calls when training with the function role?
As far as I see, the tool calls are placed in the content with a special role tag `function`. This prevents us from training models to think before tool calls.

### Pull Request

_No response_