mistral-openapi-client 0.1.0

/*
 * Mistral AI API
 *
 * Our Chat Completion and Embeddings APIs specification. Create your account on [La Plateforme](https://console.mistral.ai) to get access and read the [docs](https://docs.mistral.ai) to learn how to use it.
 *
 * The version of the OpenAPI document: 1.0.0
 * 
 * Generated by: https://openapi-generator.tech
 */

use crate::models;
use serde::{Deserialize, Serialize};

#[derive(Clone, Default, Debug, PartialEq, Serialize, Deserialize)]
pub struct Response {
    /// The ID of the job.
    #[serde(rename = "id")]
    pub id: uuid::Uuid,
    #[serde(rename = "auto_start")]
    pub auto_start: bool,
    #[serde(rename = "model")]
    pub model: String,
    /// The current status of the fine-tuning job.
    #[serde(rename = "status")]
    pub status: Status,
    /// The UNIX timestamp (in seconds) for when the fine-tuning job was created.
    #[serde(rename = "created_at")]
    pub created_at: i32,
    /// The UNIX timestamp (in seconds) for when the fine-tuning job was last modified.
    #[serde(rename = "modified_at")]
    pub modified_at: i32,
    /// A list containing the IDs of uploaded files that contain training data.
    #[serde(rename = "training_files")]
    pub training_files: Vec<uuid::Uuid>,
    #[serde(rename = "validation_files", skip_serializing_if = "Option::is_none")]
    pub validation_files: Option<Vec<uuid::Uuid>>,
    #[serde(rename = "object", skip_serializing_if = "Option::is_none")]
    pub object: Option<Object>,
    #[serde(rename = "fine_tuned_model", skip_serializing_if = "Option::is_none")]
    pub fine_tuned_model: Option<String>,
    #[serde(rename = "suffix", skip_serializing_if = "Option::is_none")]
    pub suffix: Option<String>,
    #[serde(rename = "integrations", skip_serializing_if = "Option::is_none")]
    pub integrations: Option<Vec<models::WandbIntegrationOut>>,
    #[serde(rename = "trained_tokens", skip_serializing_if = "Option::is_none")]
    pub trained_tokens: Option<i32>,
    #[serde(rename = "metadata", skip_serializing_if = "Option::is_none")]
    pub metadata: Option<Box<models::JobMetadataOut>>,
    /// The type of job (`FT` for fine-tuning).
    #[serde(rename = "job_type", skip_serializing_if = "Option::is_none")]
    pub job_type: Option<JobType>,
    #[serde(rename = "hyperparameters")]
    pub hyperparameters: Box<models::ClassifierTrainingParameters>,
    #[serde(rename = "repositories", skip_serializing_if = "Option::is_none")]
    pub repositories: Option<Vec<models::GithubRepositoryOut>>,
    #[serde(rename = "expected_duration_seconds", skip_serializing_if = "Option::is_none")]
    pub expected_duration_seconds: Option<i32>,
    #[serde(rename = "cost", skip_serializing_if = "Option::is_none")]
    pub cost: Option<f64>,
    #[serde(rename = "cost_currency", skip_serializing_if = "Option::is_none")]
    pub cost_currency: Option<String>,
    #[serde(rename = "train_tokens_per_step", skip_serializing_if = "Option::is_none")]
    pub train_tokens_per_step: Option<i32>,
    #[serde(rename = "train_tokens", skip_serializing_if = "Option::is_none")]
    pub train_tokens: Option<i32>,
    #[serde(rename = "data_tokens", skip_serializing_if = "Option::is_none")]
    pub data_tokens: Option<i32>,
    #[serde(rename = "estimated_start_time", skip_serializing_if = "Option::is_none")]
    pub estimated_start_time: Option<i32>,
    #[serde(rename = "deprecated", skip_serializing_if = "Option::is_none")]
    pub deprecated: Option<bool>,
    #[serde(rename = "details")]
    pub details: String,
    #[serde(rename = "epochs", skip_serializing_if = "Option::is_none")]
    pub epochs: Option<f64>,
    #[serde(rename = "training_steps", skip_serializing_if = "Option::is_none")]
    pub training_steps: Option<i32>,
}

impl Response {
    pub fn new(id: uuid::Uuid, auto_start: bool, model: String, status: Status, created_at: i32, modified_at: i32, training_files: Vec<uuid::Uuid>, hyperparameters: models::ClassifierTrainingParameters, details: String) -> Response {
        Response {
            id,
            auto_start,
            model,
            status,
            created_at,
            modified_at,
            training_files,
            validation_files: None,
            object: None,
            fine_tuned_model: None,
            suffix: None,
            integrations: None,
            trained_tokens: None,
            metadata: None,
            job_type: None,
            hyperparameters: Box::new(hyperparameters),
            repositories: None,
            expected_duration_seconds: None,
            cost: None,
            cost_currency: None,
            train_tokens_per_step: None,
            train_tokens: None,
            data_tokens: None,
            estimated_start_time: None,
            deprecated: None,
            details,
            epochs: None,
            training_steps: None,
        }
    }
}
/// The current status of the fine-tuning job.
#[derive(Clone, Copy, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Serialize, Deserialize)]
pub enum Status {
    #[serde(rename = "QUEUED")]
    Queued,
    #[serde(rename = "STARTED")]
    Started,
    #[serde(rename = "VALIDATING")]
    Validating,
    #[serde(rename = "VALIDATED")]
    Validated,
    #[serde(rename = "RUNNING")]
    Running,
    #[serde(rename = "FAILED_VALIDATION")]
    FailedValidation,
    #[serde(rename = "FAILED")]
    Failed,
    #[serde(rename = "SUCCESS")]
    Success,
    #[serde(rename = "CANCELLED")]
    Cancelled,
    #[serde(rename = "CANCELLATION_REQUESTED")]
    CancellationRequested,
}

impl Default for Status {
    fn default() -> Status {
        Self::Queued
    }
}
/// 
#[derive(Clone, Copy, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Serialize, Deserialize)]
pub enum Object {
    #[serde(rename = "job.metadata")]
    JobMetadata,
}

impl Default for Object {
    fn default() -> Object {
        Self::JobMetadata
    }
}
/// The type of job (`FT` for fine-tuning).
#[derive(Clone, Copy, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Serialize, Deserialize)]
pub enum JobType {
    #[serde(rename = "classifier")]
    Classifier,
}

impl Default for JobType {
    fn default() -> JobType {
        Self::Classifier
    }
}