use serde::{Deserialize, Serialize};
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
pub struct AgentMessage {
pub role: MessageRole,
pub content: MessageContent,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub metadata: Option<MessageMetadata>,
}
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
pub enum MessageRole {
User,
Agent,
Tool,
System,
}
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
#[serde(tag = "type", content = "value")]
pub enum MessageContent {
Text(String),
Json(serde_json::Value),
}
impl MessageContent {
pub fn as_text(&self) -> Option<&str> {
match self {
MessageContent::Text(text) => Some(text.as_str()),
_ => None,
}
}
pub fn as_json(&self) -> Option<&serde_json::Value> {
match self {
MessageContent::Json(value) => Some(value),
_ => None,
}
}
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct ToolInvocation {
pub tool_name: String,
pub args: serde_json::Value,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub tool_call_id: Option<String>,
}
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
pub struct MessageMetadata {
#[serde(default, skip_serializing_if = "Option::is_none")]
pub tool_call_id: Option<String>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub cache_control: Option<CacheControl>,
}
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
pub struct CacheControl {
#[serde(rename = "type")]
pub cache_type: String,
}