use core::fmt;
use std::fmt::{Display, Formatter};
use serde::{Deserialize, Serialize};
use utoipa::ToSchema;
#[derive(Deserialize, Serialize, Debug, Clone)]
pub struct ChatMessage {
pub role: ChatMessageRole,
pub content: String,
#[serde(skip_serializing_if = "Option::is_none")]
pub name: Option<String>,
}
#[derive(Deserialize, Serialize, Debug, Clone, Copy, ToSchema)]
#[serde(rename_all = "lowercase")]
pub enum ChatMessageRole {
System,
User,
Assistant,
Function,
}
impl From<String> for ChatMessageRole {
fn from(role: String) -> Self {
match role.as_str() {
"system" => ChatMessageRole::System,
"user" => ChatMessageRole::User,
"assistant" => ChatMessageRole::Assistant,
"function" => ChatMessageRole::Function,
_ => ChatMessageRole::User,
}
}
}
impl Display for ChatMessageRole {
fn fmt(&self, f: &mut Formatter<'_>) -> fmt::Result {
write!(f, "{:?}", self)
}
}