use serde_json::{json, Value};
use crate::canonical::{
CanonicalError, CanonicalRequest, Content, DocumentSource, ErrorKind, ImageSource, Message,
Role,
};
use crate::protocol::json::to_json_string;
fn slot_err(slot: &str) -> CanonicalError {
CanonicalError {
kind: ErrorKind::ParseInput,
message: format!("{slot} accepts only text content"),
provider_detail: None,
retry_after_seconds: None,
}
}
pub(super) fn instructions(req: &CanonicalRequest) -> Result<Option<String>, CanonicalError> {
let Some(system) = req.system.as_ref().filter(|s| !s.is_empty()) else {
return Ok(None);
};
let mut text = String::new();
for c in system {
match c {
Content::Text(t) => text.push_str(t),
_ => return Err(slot_err("instructions")),
}
}
Ok(Some(text))
}
pub(super) fn input_value(req: &CanonicalRequest) -> Result<Value, CanonicalError> {
let mut items = Vec::new();
for m in &req.messages {
message_items(m, &mut items)?;
}
Ok(Value::Array(items))
}
fn message_items(m: &Message, items: &mut Vec<Value>) -> Result<(), CanonicalError> {
let (role, text_type) = match m.role {
Role::User => ("user", "input_text"),
Role::System => ("system", "input_text"),
Role::Assistant => ("assistant", "output_text"),
Role::Tool => {
for c in &m.content {
items.push(function_call_output(c)?);
}
return Ok(());
}
};
let mut content = Vec::new();
let mut reasonings = Vec::new();
let mut calls = Vec::new();
for c in &m.content {
match c {
Content::Text(t) => content.push(json!({ "type": text_type, "text": t })),
Content::Image { source } if role == "user" => content.push(input_image(source)),
Content::Document { source } if role == "user" => content.push(input_file(source)),
Content::ToolUse {
id, name, input, ..
} => calls.push(json!({
"type": "function_call", "call_id": id, "name": name,
"arguments": to_json_string(input),
})),
Content::Thinking {
text,
id,
encrypted_content: Some(enc),
..
} => reasonings.push(reasoning_item(text, id.as_deref(), enc)),
Content::Thinking { .. } | Content::RedactedThinking { .. } => {} _ => return Err(slot_err(role)),
}
}
items.extend(reasonings); if !content.is_empty() {
items.push(json!({ "type": "message", "role": role, "content": content }));
}
items.extend(calls);
Ok(())
}
fn reasoning_item(text: &str, id: Option<&str>, encrypted_content: &str) -> Value {
let summary = if text.is_empty() {
json!([])
} else {
json!([{ "type": "summary_text", "text": text }])
};
let mut item = json!({
"type": "reasoning", "summary": summary, "encrypted_content": encrypted_content,
});
if let Some(id) = id {
item["id"] = json!(id);
}
item
}
fn function_call_output(c: &Content) -> Result<Value, CanonicalError> {
let Content::ToolResult {
tool_use_id,
content,
is_error,
} = c
else {
return Err(slot_err("tool"));
};
let mut text = String::new();
for part in content {
match part {
Content::Text(t) => text.push_str(t),
_ => return Err(slot_err("tool_result")),
}
}
if *is_error {
text = format!("[error] {text}");
}
Ok(json!({ "type": "function_call_output", "call_id": tool_use_id, "output": text }))
}
fn input_image(source: &ImageSource) -> Value {
let url = match source {
ImageSource::Base64 { media_type, data } => format!("data:{media_type};base64,{data}"),
ImageSource::Url { url } => url.clone(),
};
json!({ "type": "input_image", "image_url": url })
}
fn input_file(source: &DocumentSource) -> Value {
match source {
DocumentSource::Base64 { media_type, data } => json!({
"type": "input_file",
"filename": format!("document.{}", media_type.rsplit('/').next().unwrap_or("bin")),
"file_data": format!("data:{media_type};base64,{data}"),
}),
DocumentSource::Url { url } => json!({ "type": "input_file", "file_url": url }),
}
}