use crate::models;
use serde::{Deserialize, Serialize};
#[derive(Clone, Default, Debug, PartialEq, Serialize, Deserialize)]
pub struct OpenAiMessage {
#[serde(rename = "content", skip_serializing_if = "Option::is_none")]
pub content: Option<String>,
#[serde(rename = "role")]
pub role: RoleTrue,
#[serde(rename = "tool_calls", skip_serializing_if = "Option::is_none")]
pub tool_calls: Option<Vec<serde_json::Value>>,
#[serde(rename = "tool_call_id", skip_serializing_if = "Option::is_none")]
pub tool_call_id: Option<String>,
}
impl OpenAiMessage {
pub fn new(content: Option<String>, role: RoleTrue) -> OpenAiMessage {
OpenAiMessage {
content,
role,
tool_calls: None,
tool_call_id: None,
}
}
}
#[derive(Clone, Copy, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Serialize, Deserialize)]
pub enum RoleTrue {
#[serde(rename = "assistant")]
Assistant,
#[serde(rename = "function")]
Function,
#[serde(rename = "user")]
User,
#[serde(rename = "system")]
System,
#[serde(rename = "tool")]
Tool,
}
impl Default for RoleTrue {
fn default() -> RoleTrue {
Self::Assistant
}
}