use crate::models;
use serde::{Deserialize, Serialize};
#[derive(Clone, Default, Debug, PartialEq, Serialize, Deserialize)]
pub struct CerebrasModel {
#[serde(rename = "messages", skip_serializing_if = "Option::is_none")]
pub messages: Option<Vec<models::OpenAiMessage>>,
#[serde(rename = "tools", skip_serializing_if = "Option::is_none")]
pub tools: Option<Vec<models::AnyscaleModelToolsInner>>,
#[serde(rename = "toolIds", skip_serializing_if = "Option::is_none")]
pub tool_ids: Option<Vec<String>>,
#[serde(rename = "knowledgeBase", skip_serializing_if = "Option::is_none")]
pub knowledge_base: Option<models::CreateCustomKnowledgeBaseDto>,
#[serde(rename = "knowledgeBaseId", skip_serializing_if = "Option::is_none")]
pub knowledge_base_id: Option<String>,
#[serde(rename = "model")]
pub model: ModelTrue,
#[serde(rename = "provider")]
pub provider: ProviderTrue,
#[serde(rename = "temperature", skip_serializing_if = "Option::is_none")]
pub temperature: Option<f64>,
#[serde(rename = "maxTokens", skip_serializing_if = "Option::is_none")]
pub max_tokens: Option<f64>,
#[serde(
rename = "emotionRecognitionEnabled",
skip_serializing_if = "Option::is_none"
)]
pub emotion_recognition_enabled: Option<bool>,
#[serde(rename = "numFastTurns", skip_serializing_if = "Option::is_none")]
pub num_fast_turns: Option<f64>,
}
impl CerebrasModel {
pub fn new(model: ModelTrue, provider: ProviderTrue) -> CerebrasModel {
CerebrasModel {
messages: None,
tools: None,
tool_ids: None,
knowledge_base: None,
knowledge_base_id: None,
model,
provider,
temperature: None,
max_tokens: None,
emotion_recognition_enabled: None,
num_fast_turns: None,
}
}
}
#[derive(Clone, Copy, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Serialize, Deserialize)]
pub enum ModelTrue {
#[serde(rename = "llama3.1-8b")]
Llama3Period18b,
#[serde(rename = "llama-3.3-70b")]
Llama3Period370b,
}
impl Default for ModelTrue {
fn default() -> ModelTrue {
Self::Llama3Period18b
}
}
#[derive(Clone, Copy, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Serialize, Deserialize)]
pub enum ProviderTrue {
#[serde(rename = "cerebras")]
Cerebras,
}
impl Default for ProviderTrue {
fn default() -> ProviderTrue {
Self::Cerebras
}
}