use crate::models;
use serde::{Deserialize, Serialize};
#[derive(Clone, Default, Debug, PartialEq, Serialize, Deserialize)]
pub struct KnowledgeBase {
#[serde(rename = "name")]
pub name: String,
#[serde(rename = "provider")]
pub provider: ProviderTrue,
#[serde(rename = "model", skip_serializing_if = "Option::is_none")]
pub model: Option<ModelTrue>,
#[serde(rename = "description")]
pub description: String,
#[serde(rename = "fileIds")]
pub file_ids: Vec<String>,
}
impl KnowledgeBase {
pub fn new(
name: String,
provider: ProviderTrue,
description: String,
file_ids: Vec<String>,
) -> KnowledgeBase {
KnowledgeBase {
name,
provider,
model: None,
description,
file_ids,
}
}
}
#[derive(Clone, Copy, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Serialize, Deserialize)]
pub enum ProviderTrue {
#[serde(rename = "google")]
Google,
}
impl Default for ProviderTrue {
fn default() -> ProviderTrue {
Self::Google
}
}
#[derive(Clone, Copy, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Serialize, Deserialize)]
pub enum ModelTrue {
#[serde(rename = "gemini-2.5-pro-preview-05-06")]
Gemini2Period5ProPreview0506,
#[serde(rename = "gemini-2.5-flash-preview-05-20")]
Gemini2Period5FlashPreview0520,
#[serde(rename = "gemini-2.5-flash-preview-04-17")]
Gemini2Period5FlashPreview0417,
#[serde(rename = "gemini-2.0-flash-thinking-exp")]
Gemini2Period0FlashThinkingExp,
#[serde(rename = "gemini-2.0-pro-exp-02-05")]
Gemini2Period0ProExp0205,
#[serde(rename = "gemini-2.0-flash")]
Gemini2Period0Flash,
#[serde(rename = "gemini-2.0-flash-lite")]
Gemini2Period0FlashLite,
#[serde(rename = "gemini-2.0-flash-lite-preview-02-05")]
Gemini2Period0FlashLitePreview0205,
#[serde(rename = "gemini-2.0-flash-exp")]
Gemini2Period0FlashExp,
#[serde(rename = "gemini-2.0-flash-realtime-exp")]
Gemini2Period0FlashRealtimeExp,
#[serde(rename = "gemini-1.5-flash")]
Gemini1Period5Flash,
#[serde(rename = "gemini-1.5-flash-002")]
Gemini1Period5Flash002,
#[serde(rename = "gemini-1.5-pro")]
Gemini1Period5Pro,
#[serde(rename = "gemini-1.5-pro-002")]
Gemini1Period5Pro002,
#[serde(rename = "gemini-1.0-pro")]
Gemini1Period0Pro,
}
impl Default for ModelTrue {
fn default() -> ModelTrue {
Self::Gemini2Period5ProPreview0506
}
}