use crate::{ValidationError, ValidationResult};
use serde::{Deserialize, Serialize};
use std::fmt::Display;
#[derive(Debug, Clone, Copy, PartialEq, Default, Serialize, Deserialize)]
#[serde(transparent)]
pub struct TopLogprobs {
value: u32,
}
impl Display for TopLogprobs {
fn fmt(
&self,
f: &mut std::fmt::Formatter<'_>,
) -> std::fmt::Result {
write!(f, "{}", self.value)
}
}
impl TopLogprobs {
pub fn new(value: u32) -> ValidationResult<Self, u32> {
if value > 5 {
Err(ValidationError {
type_name: "TopLogprobs".to_string(),
reason: "The top logprobs must be between 0 and 5.".to_string(),
value,
})
} else {
Ok(Self {
value,
})
}
}
}