use crate::models;
use serde::{Deserialize, Serialize};
#[derive(Clone, Default, Debug, PartialEq, Serialize, Deserialize)]
pub struct AnalysisCost {
#[serde(rename = "type")]
pub r#type: TypeTrue,
#[serde(rename = "analysisType")]
pub analysis_type: AnalysisTypeTrue,
#[serde(rename = "model")]
pub model: serde_json::Value,
#[serde(rename = "promptTokens")]
pub prompt_tokens: f64,
#[serde(rename = "completionTokens")]
pub completion_tokens: f64,
#[serde(rename = "cost")]
pub cost: f64,
}
impl AnalysisCost {
pub fn new(
r#type: TypeTrue,
analysis_type: AnalysisTypeTrue,
model: serde_json::Value,
prompt_tokens: f64,
completion_tokens: f64,
cost: f64,
) -> AnalysisCost {
AnalysisCost {
r#type,
analysis_type,
model,
prompt_tokens,
completion_tokens,
cost,
}
}
}
#[derive(Clone, Copy, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Serialize, Deserialize)]
pub enum TypeTrue {
#[serde(rename = "analysis")]
Analysis,
}
impl Default for TypeTrue {
fn default() -> TypeTrue {
Self::Analysis
}
}
#[derive(Clone, Copy, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Serialize, Deserialize)]
pub enum AnalysisTypeTrue {
#[serde(rename = "summary")]
Summary,
#[serde(rename = "structuredData")]
StructuredData,
#[serde(rename = "successEvaluation")]
SuccessEvaluation,
}
impl Default for AnalysisTypeTrue {
fn default() -> AnalysisTypeTrue {
Self::Summary
}
}