import {z} from 'zod'
import {createSuccessResponse, withErrorHandling} from '../../utils/response.js'
import {BaseToolSchema, createToolClient} from '../../utils/tools.js'
export const CreateDatasetToolParams = BaseToolSchema.extend({
name: z
.string()
.describe('The name of the dataset (will be automatically formatted to match requirements)'),
aclMode: z.enum(['private', 'public']).optional().describe('The ACL mode for the dataset'),
})
type Params = z.infer<typeof CreateDatasetToolParams>
async function _tool(args: Params) {
const client = createToolClient(args)
// Only lowercase letters and numbers are allowed
const datasetName = args.name.toLowerCase().replace(/[^a-z0-9]/g, '')
const newDataset = await client.datasets.create(datasetName, {
aclMode: args.aclMode,
})
return createSuccessResponse('Dataset created successfully', {newDataset})
}
export const createDatasetTool = withErrorHandling(_tool, 'Error creating dataset')
