distilabel
distilabel copied to clipboard
[FEATURE] Allow `FormatTextGenerationSFT` to include tools/function calls in the formatted messages.
Is your feature request related to a problem? Please describe.
With the new StructuredGeneration
task we can generate datasets for function calling, and we could simplify preparing the dataset for training.
Describe the solution you'd like
Extend FormatTextGenerationSFT
(FormatChatGenerationSFT
too?) to include the available tools and the function calls
Describe alternatives you've considered Let the user do it on its own.
Additional context Use the format from Mistral function calling.