use serde_json::{json, Map, Value};
use crate::canonical::{
CanonicalError, CanonicalRequest, ErrorKind, OutputFormat, Tool, ToolChoice,
};
use crate::protocol::json::finish_body;
use crate::protocol::{ProviderCtx, WireRequest};
mod input;
pub(super) const REQUEST_PATH: &str = "/responses";
pub(super) fn encode(
req: &CanonicalRequest,
ctx: &ProviderCtx,
) -> Result<WireRequest, CanonicalError> {
let mut body = Map::new();
body.insert("model".into(), json!(ctx.model));
if let Some(text) = input::instructions(req)? {
body.insert("instructions".into(), json!(text));
}
body.insert("input".into(), input::input_value(req)?);
if !req.tools.is_empty() {
body.insert("tools".into(), tools_value(&req.tools)?); }
if let Some(tc) = tool_choice_value(&req.tool_choice) {
body.insert("tool_choice".into(), tc); }
if let Some(p) = req.parallel_tool_calls {
body.insert("parallel_tool_calls".into(), json!(p)); }
if let Some(n) = req.max_tokens {
body.insert("max_output_tokens".into(), json!(n)); }
if req.reasoning.is_none() {
if let Some(t) = req.temperature {
body.insert("temperature".into(), json!(t));
}
if let Some(p) = req.top_p {
body.insert("top_p".into(), json!(p));
}
}
if let Some(r) = req.reasoning {
body.insert(
"reasoning".into(),
json!({"effort": r.as_str(), "summary": "auto"}),
);
body.insert("include".into(), json!(["reasoning.encrypted_content"]));
}
body.insert("stream".into(), json!(req.stream.unwrap_or(false))); if let Some(fmt) = text_format(&req.output) {
body.insert("text".into(), json!({ "format": fmt }));
}
for (k, v) in &req.extra {
body.entry(k.clone()).or_insert_with(|| v.clone()); }
Ok(finish_body(body, format!("{}{REQUEST_PATH}", ctx.base_url)))
}
fn text_format(output: &Option<OutputFormat>) -> Option<Value> {
Some(match output.as_ref()? {
OutputFormat::Json => json!({ "type": "json_object" }),
OutputFormat::JsonSchema {
name,
schema,
strict,
} => {
let mut f = json!({
"type": "json_schema",
"name": name.as_deref().unwrap_or("response"),
"schema": schema,
});
if let Some(s) = strict {
f["strict"] = json!(s);
}
f
}
})
}
fn tools_value(tools: &[Tool]) -> Result<Value, CanonicalError> {
let mut out = Vec::new();
for t in tools {
let Tool::Custom {
name,
description,
input_schema,
strict,
} = t
else {
return Err(CanonicalError {
kind: ErrorKind::ParseInput,
message: "provider-typed tools are not projected for this dialect".into(),
provider_detail: None,
retry_after_seconds: None,
});
};
let mut f = json!({ "type": "function", "name": name, "parameters": input_schema });
if let Some(d) = description {
f["description"] = json!(d);
}
if let Some(s) = strict {
f["strict"] = json!(s);
}
out.push(f);
}
Ok(Value::Array(out))
}
fn tool_choice_value(tc: &ToolChoice) -> Option<Value> {
Some(match tc {
ToolChoice::Auto => return None,
ToolChoice::Any => json!("required"),
ToolChoice::None => json!("none"),
ToolChoice::Tool { name } => json!({ "type": "function", "name": name }),
})
}