models.rs

use bson::DateTime;
use mongodb::bson::{doc, oid::ObjectId};
use serde::{Deserialize, Serialize};
use serde_json::Value;

#[derive(Serialize, Deserialize, Clone, Debug)]
pub struct DatasourceConnectionSettings {
    pub syncCatalog: Value,
    pub scheduleType: String,
    pub namespaceDefinition: Option<String>,
    pub namespaceFormat: Option<String>,
    pub nonBreakingSchemaUpdatesBehavior: String,
    pub prefix: Option<String>,
    pub name: String,
    pub sourceId: String,
    pub destinationId: String,
    pub status: String,
}

#[derive(Serialize, Deserialize, Clone, Debug)]
pub struct DataSources {
    pub _id: ObjectId,
    pub orgId: ObjectId,
    pub teamId: ObjectId,
    pub modelId: Option<ObjectId>,
    pub name: String,
    pub description: Option<String>,
    pub filename: Option<String>,
    pub originalName: String,
    pub sourceType: String,
    pub sourceId: Option<String>,
    pub syncedCount: Option<i32>,
    pub embeddedCount: Option<i32>,
    pub destinationId: Option<String>,
    pub workspaceId: Option<String>,
    pub connectionId: Option<String>,
    pub chunkStrategy: Option<String>,
    pub chunkCharacter: Option<String>,
    pub lastSyncedDate: Option<DateTime>,
    pub embeddingField: Option<String>,
    pub createdDate: Option<DateTime>,
    pub status: String,
}

#[derive(Serialize, Deserialize)]
pub enum ChunkingStrategy {
    SEMANTIC_CHUNKING,
    CHARACTER_CHUNKING,
    CODE_SPLIT,
    UNKNOWN,
}

impl From<String> for ChunkingStrategy {
    fn from(value: String) -> Self {
        match value.as_str() {
            "semantic" => ChunkingStrategy::SEMANTIC_CHUNKING,
            "character" => ChunkingStrategy::CHARACTER_CHUNKING,
            "code" => ChunkingStrategy::CODE_SPLIT,
            _ => ChunkingStrategy::UNKNOWN,
        }
    }
}

#[derive(Serialize, Deserialize, Clone)]
pub struct Model {
    pub _id: ObjectId,
    pub orgId: ObjectId,
    pub teamId: ObjectId,
    pub credentialId: Option<ObjectId>,
    pub name: String,
    pub model: String,
    pub embeddingLength: i32,
    pub modelType: String,
}


#[derive(Serialize, Deserialize, Clone, Debug)]
pub struct CredentialsObj {
    pub key: Option<String>,
    pub endpoint: Option<String>,
    pub org: Option<String>,
}

#[derive(Serialize, Deserialize, Clone, Debug)]
pub struct Credentials {
    pub _id: ObjectId,
    pub orgId: ObjectId,
    pub teamId: ObjectId,
    pub name: String,
    pub createdDate: Option<DateTime>,
    pub credentials: Option<CredentialsObj>,
}