distilabel icon indicating copy to clipboard operation
distilabel copied to clipboard

[FEATURE] Allow `FormatTextGenerationSFT` to include tools/function calls in the formatted messages.

Open plaguss opened this issue 8 months ago • 0 comments

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.

plaguss avatar Jun 17 '24 07:06 plaguss