import {
  CreateFineTuneInputModelEnum,
  FineTuneCohereInputModelEnum,
  FineTuneDto,
  FineTuneOpenaiInputModelEnum,
} from "~/browser-client";

export type StartFineTuneAction = {
  finetune: FineTuneDto;
};

export type StartOpenAIFineTuneBody = {
  model: FineTuneOpenaiInputModelEnum;
};

export type StartMonteloFineTuneBody = {
  model: CreateFineTuneInputModelEnum;
};

export type StartCohereFineTuneBody = {
  model: FineTuneCohereInputModelEnum;
};
