cai 0.13.0

User friendly CLI tool for AI tasks
Documentation
////////////////////////////////////////////////////////////////////////////////
////////////////////////////////// ANTHROPIC ///////////////////////////////////

// Pretty-printed string representation of the hashmap
pub const ANTHROPIC_MODEL_MAPPING: &[(&str, &str)] = &[
  // This will be replaced by build.rs:
  // {anthropic_model_hashmap}
];

pub fn get_anthropic_model(model_id: &str) -> &str {
  ANTHROPIC_MODEL_MAPPING
    .iter()
    .find(|(key, _)| key == &model_id)
    .map_or(model_id, |(_, value)| *value)
}

macro_rules! anthropic_models_pretty {
  ($prefix: expr) => {
    // This will be replaced by build.rs
    concat!($prefix, "\n", "{anthropic_models_pretty}")
  };
}

////////////////////////////////////////////////////////////////////////////////
/////////////////////////////////// CEREBRAS ///////////////////////////////////

// {const_assignments}

// Static mapping accessible from other files
pub const CEREBRAS_MODEL_MAPPING: &[(&str, &str)] = &[
  // This will be replaced by build.rs:
  // {cerebras_model_hashmap}
];

pub fn get_cerebras_model(model_id: &str) -> &str {
  CEREBRAS_MODEL_MAPPING
    .iter()
    .find(|(key, _)| key == &model_id)
    .map_or(model_id, |(_, value)| *value)
}

macro_rules! cerebras_models_pretty {
  ($prefix: expr) => {
    // This will be replaced by build.rs
    concat!($prefix, "\n", "{cerebras_models_pretty}")
  };
}

////////////////////////////////////////////////////////////////////////////////
/////////////////////////////////// DEEPSEEK ///////////////////////////////////

// {const_assignments}

// Static mapping accessible from other files
pub const DEEPSEEK_MODEL_MAPPING: &[(&str, &str)] = &[
  // This will be replaced by build.rs:
  // {deepseek_model_hashmap}
];

pub fn get_deepseek_model(model_id: &str) -> &str {
  DEEPSEEK_MODEL_MAPPING
    .iter()
    .find(|(key, _)| key == &model_id)
    .map_or(model_id, |(_, value)| *value)
}

macro_rules! deepseek_models_pretty {
  ($prefix: expr) => {
    // This will be replaced by build.rs
    concat!($prefix, "\n", "{deepseek_models_pretty}")
  };
}

////////////////////////////////////////////////////////////////////////////////
//////////////////////////////////// GOOGLE ////////////////////////////////////

// Pretty-printed string representation of the hashmap
pub const GOOGLE_MODEL_MAPPING: &[(&str, &str)] = &[
  // This will be replaced by build.rs:
  // {google_model_hashmap}
];

pub fn get_google_model(model_id: &str) -> &str {
  GOOGLE_MODEL_MAPPING
    .iter()
    .find(|(key, _)| key == &model_id)
    .map_or(model_id, |(_, value)| *value)
}

macro_rules! google_models_pretty {
  ($prefix: expr) => {
    // This will be replaced by build.rs
    concat!($prefix, "\n", "{google_models_pretty}")
  };
}

////////////////////////////////////////////////////////////////////////////////
///////////////////////////////////// GROQ /////////////////////////////////////

// {const_assignments}

// Static mapping accessible from other files
pub const GROQ_MODEL_MAPPING: &[(&str, &str)] = &[
  // This will be replaced by build.rs:
  // {groq_model_hashmap}
];

pub fn get_groq_model(model_id: &str) -> &str {
  GROQ_MODEL_MAPPING
    .iter()
    .find(|(key, _)| key == &model_id)
    .map_or(model_id, |(_, value)| *value)
}

macro_rules! groq_models_pretty {
  ($prefix: expr) => {
    // This will be replaced by build.rs
    concat!($prefix, "\n", "{groq_models_pretty}")
  };
}

////////////////////////////////////////////////////////////////////////////////
//////////////////////////////////// OLLAMA ////////////////////////////////////

// Pretty-printed string representation of the hashmap
pub const OLLAMA_MODEL_MAPPING: &[(&str, &str)] = &[
  // This will be replaced by build.rs:
  // {ollama_model_hashmap}
];

pub fn get_ollama_model(model_id: &str) -> &str {
  OLLAMA_MODEL_MAPPING
    .iter()
    .find(|(key, _)| key == &model_id)
    .map_or(model_id, |(_, value)| *value)
}

macro_rules! ollama_models_pretty {
  ($prefix: expr) => {
    // This will be replaced by build.rs
    concat!($prefix, "\n", "{ollama_models_pretty}")
  };
}

////////////////////////////////////////////////////////////////////////////////
//////////////////////////////////// OPENAI ////////////////////////////////////

// Pretty-printed string representation of the hashmap
pub const OPENAI_MODEL_MAPPING: &[(&str, &str)] = &[
  // This will be replaced by build.rs:
  // {openai_model_hashmap}
];

pub fn get_openai_model(model_id: &str) -> &str {
  OPENAI_MODEL_MAPPING
    .iter()
    .find(|(key, _)| key == &model_id)
    .map_or(model_id, |(_, value)| *value)
}

macro_rules! openai_models_pretty {
  ($prefix: expr) => {
    // This will be replaced by build.rs
    concat!($prefix, "\n", "{openai_models_pretty}")
  };
}

////////////////////////////////////////////////////////////////////////////////
////////////////////////////////////// XAI /////////////////////////////////////

// Pretty-printed string representation of the hashmap
pub const XAI_MODEL_MAPPING: &[(&str, &str)] = &[
  // This will be replaced by build.rs:
  // {xai_model_hashmap}
];

pub fn get_xai_model(model_id: &str) -> &str {
  XAI_MODEL_MAPPING
    .iter()
    .find(|(key, _)| key == &model_id)
    .map_or(model_id, |(_, value)| *value)
}

macro_rules! xai_models_pretty {
  ($prefix: expr) => {
    // This will be replaced by build.rs
    concat!($prefix, "\n", "{xai_models_pretty}")
  };
}

////////////////////////////////////////////////////////////////////////////////
/////////////////////////////// Perplexity /////////////////////////////////////

// Pretty-printed string representation of the hashmap
pub const PERPLEXITY_MODEL_MAPPING: &[(&str, &str)] = &[
  // This will be replaced by build.rs:
  // {perplexity_model_hashmap}
];

pub fn get_perplexity_model(model_id: &str) -> &str {
  PERPLEXITY_MODEL_MAPPING
    .iter()
    .find(|(key, _)| key == &model_id)
    .map_or(model_id, |(_, value)| *value)
}

macro_rules! perplexity_models_pretty {
  ($prefix: expr) => {
    // This will be replaced by build.rs
    concat!($prefix, "\n", "{perplexity_models_pretty}")
  };
}