anyllm_translate 0.9.7

// Anthropic Messages API <-> Gemini generateContent API message mapping.
//
// Pure translation functions, no IO. Converts Anthropic request types into
// Gemini request types and Gemini response types back into Anthropic responses.

use std::collections::HashMap;

use crate::anthropic::messages as anthropic;
use crate::gemini::request as gemini;
use crate::gemini::response as gemini_resp;
use crate::mapping::tools_map::sanitize_schema_for_gemini;
use crate::util::ids::{generate_message_id, generate_tool_use_id};

// ---------------------------------------------------------------------------
// Request direction: Anthropic -> Gemini
// ---------------------------------------------------------------------------

/// Compute degradation warnings for a Gemini-bound request.
///
/// Call before translating to surface features that are silently dropped during
/// Anthropic → Gemini translation. Emit the result as an `x-anyllm-degradation`
/// response header so clients can detect lossy translations.
pub fn compute_gemini_request_warnings(
    req: &anthropic::MessageCreateRequest,
) -> crate::mapping::warnings::TranslationWarnings {
    use crate::mapping::warnings::TranslationWarnings;
    let mut w = TranslationWarnings::default();

    // Single pass: collect all per-block warning flags at once.
    let mut has_thinking = false;
    let mut has_document = false;
    let mut has_url_image = false;
    for msg in &req.messages {
        if let anthropic::Content::Blocks(blocks) = &msg.content {
            for b in blocks {
                match b {
                    // Thinking/RedactedThinking: no Gemini Content equivalent.
                    anthropic::ContentBlock::Thinking { .. }
                    | anthropic::ContentBlock::RedactedThinking { .. } => has_thinking = true,
                    // Document blocks have no Gemini equivalent.
                    anthropic::ContentBlock::Document { .. } => has_document = true,
                    // URL-type images: Gemini only accepts inline base64 data.
                    anthropic::ContentBlock::Image { source } if source.source_type != "base64" => {
                        has_url_image = true
                    }
                    _ => {}
                }
            }
        }
    }
    if has_thinking {
        w.add("thinking_blocks");
    }
    if has_document {
        w.add("document_blocks");
    }
    if has_url_image {
        w.add("url_images");
    }

    // cache_control on system blocks is dropped; Gemini has no prompt-caching API.
    if let Some(anthropic::System::Blocks(blocks)) = &req.system {
        if blocks.iter().any(|b| b.cache_control.is_some()) {
            w.add("cache_control");
        }
    }

    w
}

/// Convert an Anthropic `MessageCreateRequest` into a Gemini `GenerateContentRequest`.
///
/// Maps `thinking_config` to `generationConfig.thinkingConfig` for Gemini 2.5
/// thinking models. Drops unsupported features (thinking content blocks in prior
/// messages, document blocks, cache_control) and merges consecutive same-role
/// messages to satisfy Gemini's strict alternation requirement.
pub fn anthropic_to_gemini_request(
    req: &anthropic::MessageCreateRequest,
) -> gemini::GenerateContentRequest {
    let tool_id_map = build_tool_id_map(&req.messages);

    // System instruction
    let system_instruction = req.system.as_ref().map(|sys| {
        let text = match sys {
            anthropic::System::Text(s) => s.clone(),
            anthropic::System::Blocks(blocks) => blocks
                .iter()
                .map(|b| b.text.as_str())
                .collect::<Vec<_>>()
                .join("\n"),
        };
        gemini::Content {
            role: None,
            parts: vec![gemini::Part::text(text)],
        }
    });

    // Convert messages
    let mut contents: Vec<gemini::Content> = Vec::new();
    for msg in &req.messages {
        let role = match msg.role {
            anthropic::Role::User => "user",
            anthropic::Role::Assistant => "model",
        };
        let parts = content_blocks_to_parts(&msg.content, &tool_id_map);
        if !parts.is_empty() {
            contents.push(gemini::Content {
                role: Some(role.to_string()),
                parts,
            });
        }
    }
    contents = merge_consecutive_roles(contents);

    // Tools
    let tools = req.tools.as_ref().map(|tools| {
        vec![gemini::Tool {
            function_declarations: tools
                .iter()
                .map(|t| gemini::FunctionDeclaration {
                    name: t.name.clone(),
                    description: t.description.clone(),
                    parameters: Some(sanitize_schema_for_gemini(t.input_schema.clone())),
                })
                .collect(),
        }]
    });

    // Tool config
    let tool_config = req.tool_choice.as_ref().map(|tc| {
        if matches!(
            tc,
            anthropic::ToolChoice::Auto {
                disable_parallel_tool_use: Some(true)
            } | anthropic::ToolChoice::Any {
                disable_parallel_tool_use: Some(true)
            }
        ) {
            tracing::warn!(
                "disable_parallel_tool_use=true is not supported by Gemini; \
                 parallel tool calls may still occur"
            );
        }
        let (mode, allowed) = match tc {
            anthropic::ToolChoice::Auto { .. } => ("AUTO", None),
            anthropic::ToolChoice::Any { .. } => ("ANY", None),
            anthropic::ToolChoice::None => ("NONE", None),
            // Gemini ANY + allowedFunctionNames restricts to a specific tool.
            anthropic::ToolChoice::Tool { name, .. } => ("ANY", Some(vec![name.clone()])),
        };
        gemini::ToolConfig {
            function_calling_config: gemini::FunctionCallingConfig {
                mode: mode.to_string(),
                allowed_function_names: allowed,
            },
        }
    });

    // Generation config
    let generation_config = {
        let thinking_config =
            if let Some(anthropic::ThinkingConfig::Enabled { budget_tokens }) = &req.thinking {
                Some(gemini::ThinkingConfig {
                    thinking_budget: *budget_tokens,
                    include_thoughts: Some(true),
                })
            } else {
                None
            };
        let gc = gemini::GenerationConfig {
            max_output_tokens: Some(req.max_tokens),
            temperature: req.temperature,
            top_p: req.top_p,
            top_k: req.top_k,
            stop_sequences: req.stop_sequences.clone(),
            thinking_config,
            ..Default::default()
        };
        Some(gc)
    };

    gemini::GenerateContentRequest {
        contents,
        system_instruction,
        generation_config,
        tools,
        tool_config,
        safety_settings: None,
    }
}

/// Build a map from Anthropic tool_use IDs to tool names.
///
/// Scans all messages for `ToolUse` blocks so that `ToolResult` translation can
/// look up the function name Gemini expects.
pub fn build_tool_id_map(messages: &[anthropic::InputMessage]) -> HashMap<String, String> {
    let mut map = HashMap::new();
    for msg in messages {
        let blocks = match &msg.content {
            anthropic::Content::Text(_) => continue,
            anthropic::Content::Blocks(b) => b,
        };
        for block in blocks {
            if let anthropic::ContentBlock::ToolUse { id, name, .. } = block {
                map.insert(id.clone(), name.clone());
            }
        }
    }
    map
}

/// Merge consecutive same-role `Content` entries by concatenating their parts.
///
/// Gemini requires strict user/model role alternation. When the Anthropic
/// conversation has two consecutive user (or model) turns, this merges them
/// into a single turn.
pub fn merge_consecutive_roles(contents: Vec<gemini::Content>) -> Vec<gemini::Content> {
    let mut merged: Vec<gemini::Content> = Vec::with_capacity(contents.len());
    for c in contents {
        if let Some(last) = merged.last_mut() {
            if last.role == c.role {
                last.parts.extend(c.parts);
                continue;
            }
        }
        merged.push(c);
    }

    // Gemini requires the first content turn to have role "user". An Anthropic
    // client may legally send an assistant-first conversation (for few-shot
    // prompting). Prepend a dummy user turn so Gemini does not return a 400.
    if merged.first().and_then(|c| c.role.as_deref()) == Some("model") {
        merged.insert(
            0,
            gemini::Content {
                role: Some("user".to_string()),
                parts: vec![gemini::Part::text(String::new())],
            },
        );
    }

    merged
}

/// Convert Anthropic message content into a vec of Gemini Parts.
fn content_blocks_to_parts(
    content: &anthropic::Content,
    tool_id_map: &HashMap<String, String>,
) -> Vec<gemini::Part> {
    match content {
        anthropic::Content::Text(s) => vec![gemini::Part::text(s.clone())],
        anthropic::Content::Blocks(blocks) => blocks
            .iter()
            .filter_map(|block| content_block_to_part(block, tool_id_map))
            .collect(),
    }
}

/// Convert a single Anthropic ContentBlock to a Gemini Part, or None if dropped.
fn content_block_to_part(
    block: &anthropic::ContentBlock,
    tool_id_map: &HashMap<String, String>,
) -> Option<gemini::Part> {
    match block {
        anthropic::ContentBlock::Text { text } => Some(gemini::Part::text(text.clone())),

        anthropic::ContentBlock::Image { source } => {
            // Gemini only supports inline base64 data, not URLs.
            if source.source_type == "base64" {
                let mime = source
                    .media_type
                    .clone()
                    .unwrap_or_else(|| "image/png".into());
                let data = source.data.clone().unwrap_or_default();
                Some(gemini::Part::inline_data(mime, data))
            } else {
                // URL-type images cannot be sent as inline_data; drop.
                None
            }
        }

        anthropic::ContentBlock::ToolUse { name, input, .. } => {
            // Strip the Anthropic tool_use id; Gemini uses name-based correlation.
            Some(gemini::Part::function_call(name.clone(), input.clone()))
        }

        anthropic::ContentBlock::ToolResult {
            tool_use_id,
            content,
            is_error,
        } => {
            let Some(name) = tool_id_map.get(tool_use_id).cloned() else {
                // Gemini requires the function name to match a declared FunctionDeclaration.
                // Emitting an unknown name causes a 400; drop the result instead.
                tracing::warn!(
                    tool_use_id,
                    "dropping ToolResult: tool_use_id not found in tool_id_map"
                );
                return None;
            };

            let response_value = tool_result_to_json(content, *is_error);
            Some(gemini::Part::function_response(name, response_value))
        }

        // Thinking, RedactedThinking, Document: not supported by Gemini, drop.
        anthropic::ContentBlock::Thinking { .. }
        | anthropic::ContentBlock::RedactedThinking { .. }
        | anthropic::ContentBlock::Document { .. } => None,
        _ => None,
    }
}

/// Convert Anthropic ToolResult content into a JSON value for Gemini FunctionResponse.
fn tool_result_to_json(
    content: &Option<anthropic::ToolResultContent>,
    is_error: Option<bool>,
) -> serde_json::Value {
    let text = match content {
        Some(anthropic::ToolResultContent::Text(s)) => s.clone(),
        Some(anthropic::ToolResultContent::Blocks(blocks)) => {
            // Concatenate text blocks; other block types (e.g., images) cannot be
            // represented in Gemini FunctionResponse and are replaced with a placeholder.
            blocks
                .iter()
                .map(|b| match b {
                    anthropic::ContentBlock::Text { text } => text.clone(),
                    _ => {
                        tracing::warn!(
                            "tool_result contains non-text block; \
                             replacing with \"[non-text]\" placeholder for Gemini"
                        );
                        "[non-text]".into()
                    }
                })
                .collect::<Vec<_>>()
                .join("\n")
        }
        None => String::new(),
    };

    if is_error == Some(true) {
        serde_json::json!({ "error": text })
    } else {
        serde_json::json!({ "result": text })
    }
}

// ---------------------------------------------------------------------------
// Response direction: Gemini -> Anthropic
// ---------------------------------------------------------------------------

/// Convert a Gemini `GenerateContentResponse` into an Anthropic `MessageResponse`.
///
/// Uses only the first candidate. Synthesizes Anthropic-format tool IDs for any
/// function calls.
pub fn gemini_to_anthropic_response(
    resp: &gemini_resp::GenerateContentResponse,
    model: &str,
) -> anthropic::MessageResponse {
    let candidate = resp.candidates.first();

    let content = candidate
        .map(|c| {
            c.content
                .parts
                .iter()
                .filter_map(gemini_part_to_content_block)
                .collect::<Vec<_>>()
        })
        .unwrap_or_default();

    let has_function_call = content
        .iter()
        .any(|b| matches!(b, anthropic::ContentBlock::ToolUse { .. }));

    let stop_reason = candidate
        .and_then(|c| c.finish_reason.as_ref())
        .map(|fr| match fr {
            gemini_resp::FinishReason::STOP if has_function_call => anthropic::StopReason::ToolUse,
            gemini_resp::FinishReason::STOP => anthropic::StopReason::EndTurn,
            gemini_resp::FinishReason::MAX_TOKENS => anthropic::StopReason::MaxTokens,
            // SAFETY, RECITATION, LANGUAGE, OTHER, Unknown all map to EndTurn.
            _ => anthropic::StopReason::EndTurn,
        })
        // No finish_reason at all (e.g. empty candidates) -> EndTurn.
        .or(if candidate.is_some() {
            Some(anthropic::StopReason::EndTurn)
        } else {
            None
        });

    let usage = resp
        .usage_metadata
        .as_ref()
        .map(|u| anthropic::Usage {
            input_tokens: u.prompt_token_count,
            output_tokens: u.candidates_token_count,
            cache_creation_input_tokens: None,
            cache_read_input_tokens: None,
            ..Default::default()
        })
        .unwrap_or_default();

    anthropic::MessageResponse {
        id: generate_message_id(),
        response_type: "message".into(),
        role: anthropic::Role::Assistant,
        content,
        model: model.to_string(),
        stop_reason,
        stop_sequence: None,
        usage,
        created: None,
    }
}

/// Convert a single Gemini Part to an Anthropic ContentBlock, or None if not mappable.
fn gemini_part_to_content_block(part: &gemini::Part) -> Option<anthropic::ContentBlock> {
    // Thought parts from thinking models map to Anthropic thinking blocks.
    if part.thought == Some(true) {
        return part
            .text
            .as_ref()
            .map(|text| anthropic::ContentBlock::Thinking {
                thinking: text.clone(),
                signature: None,
            });
    }
    if let Some(text) = &part.text {
        return Some(anthropic::ContentBlock::Text { text: text.clone() });
    }
    if let Some(fc) = &part.function_call {
        return Some(anthropic::ContentBlock::ToolUse {
            id: generate_tool_use_id(),
            name: fc.name.clone(),
            input: fc.args.clone(),
        });
    }
    // inline_data, file_data, function_response: not expected in model output,
    // or have no Anthropic equivalent. Drop.
    None
}

// ---------------------------------------------------------------------------
// Input direction: Gemini CLI -> Anthropic (for accepting Gemini-format input)
// ---------------------------------------------------------------------------

/// Convert a Gemini CLI `GenerateContentRequest` into an Anthropic `MessageCreateRequest`.
///
/// `model` is the model name extracted from the URL path (e.g. `gemini-2.5-pro` from
/// `POST /v1beta/models/gemini-2.5-pro:generateContent`). All generation config fields
/// map directly; unsupported Gemini features (safety settings, response_schema) are dropped.
pub fn gemini_to_anthropic_request(
    req: &gemini::GenerateContentRequest,
    model: &str,
) -> anthropic::MessageCreateRequest {
    // Build name->id map so that function_response parts reference the same
    // synthetic tool_use_id as the corresponding function_call part.
    let mut name_to_id: HashMap<String, String> = HashMap::new();
    for content in &req.contents {
        for part in &content.parts {
            if let Some(ref fc) = part.function_call {
                name_to_id
                    .entry(fc.name.clone())
                    .or_insert_with(generate_tool_use_id);
            }
        }
    }

    // System instruction -> system
    let system = req.system_instruction.as_ref().map(|si| {
        let text = si
            .parts
            .iter()
            .filter_map(|p| p.text.as_deref())
            .collect::<Vec<_>>()
            .join("\n");
        anthropic::System::Text(text)
    });

    // Contents -> messages
    let messages: Vec<anthropic::InputMessage> = req
        .contents
        .iter()
        .filter_map(|c| gemini_content_to_input_message(c, &name_to_id))
        .collect();

    // Generation config
    let gc = req.generation_config.as_ref();
    let max_tokens = gc.and_then(|g| g.max_output_tokens).unwrap_or(8192);
    let temperature = gc.and_then(|g| g.temperature);
    let top_p = gc.and_then(|g| g.top_p);
    let top_k = gc.and_then(|g| g.top_k);
    let stop_sequences = gc
        .and_then(|g| g.stop_sequences.clone())
        .filter(|v| !v.is_empty());

    // Tools
    let tools = req.tools.as_ref().map(|ts| {
        ts.iter()
            .flat_map(|t| t.function_declarations.iter())
            .map(|fd| anthropic::Tool {
                name: fd.name.clone(),
                description: fd.description.clone(),
                input_schema: fd
                    .parameters
                    .clone()
                    .unwrap_or_else(|| serde_json::json!({"type": "object"})),
            })
            .collect::<Vec<_>>()
    });

    // Tool choice
    let tool_choice = req.tool_config.as_ref().map(|tc| {
        match tc.function_calling_config.mode.as_str() {
            "NONE" => anthropic::ToolChoice::None,
            "ANY" => match tc.function_calling_config.allowed_function_names.as_deref() {
                Some([name]) => anthropic::ToolChoice::Tool { name: name.clone() },
                _ => anthropic::ToolChoice::Any {
                    disable_parallel_tool_use: None,
                },
            },
            // AUTO or anything else
            _ => anthropic::ToolChoice::Auto {
                disable_parallel_tool_use: None,
            },
        }
    });

    anthropic::MessageCreateRequest {
        model: model.to_string(),
        max_tokens,
        messages,
        system,
        temperature,
        top_p,
        top_k,
        stop_sequences,
        tools,
        tool_choice,
        metadata: None,
        thinking: None,
        stream: None,
        extra: Default::default(),
    }
}

/// Convert a single Gemini `Content` turn into an Anthropic `InputMessage`.
/// Returns `None` if the turn produces no content blocks (e.g. all parts were dropped).
fn gemini_content_to_input_message(
    content: &gemini::Content,
    name_to_id: &HashMap<String, String>,
) -> Option<anthropic::InputMessage> {
    let role = match content.role.as_deref() {
        Some("model") => anthropic::Role::Assistant,
        // "user", None, or anything unrecognised -> user.
        _ => anthropic::Role::User,
    };
    let blocks: Vec<anthropic::ContentBlock> = content
        .parts
        .iter()
        .filter_map(|p| gemini_input_part_to_block(p, name_to_id))
        .collect();
    if blocks.is_empty() {
        return None;
    }
    Some(anthropic::InputMessage {
        role,
        content: anthropic::Content::Blocks(blocks),
    })
}

/// Convert a single Gemini `Part` from a user/model message into an Anthropic `ContentBlock`.
fn gemini_input_part_to_block(
    part: &gemini::Part,
    name_to_id: &HashMap<String, String>,
) -> Option<anthropic::ContentBlock> {
    if let Some(ref text) = part.text {
        return Some(anthropic::ContentBlock::Text { text: text.clone() });
    }
    if let Some(ref fc) = part.function_call {
        let id = name_to_id
            .get(&fc.name)
            .cloned()
            .unwrap_or_else(generate_tool_use_id);
        return Some(anthropic::ContentBlock::ToolUse {
            id,
            name: fc.name.clone(),
            input: fc.args.clone(),
        });
    }
    if let Some(ref fr) = part.function_response {
        let tool_use_id = name_to_id
            .get(&fr.name)
            .cloned()
            .unwrap_or_else(generate_tool_use_id);
        return Some(anthropic::ContentBlock::ToolResult {
            tool_use_id,
            content: Some(anthropic::ToolResultContent::Text(
                serde_json::to_string(&fr.response).unwrap_or_default(),
            )),
            is_error: None,
        });
    }
    if let Some(ref data) = part.inline_data {
        if data.mime_type.starts_with("image/") {
            return Some(anthropic::ContentBlock::Image {
                source: anthropic::ImageSource {
                    source_type: "base64".to_string(),
                    media_type: Some(data.mime_type.clone()),
                    data: Some(data.data.clone()),
                    url: None,
                },
            });
        }
        // Non-image inline data (audio, video, etc.) has no Anthropic equivalent — drop.
        tracing::warn!(
            mime_type = %data.mime_type,
            "dropping inline_data part with non-image mime type (no Anthropic equivalent)"
        );
    }
    None
}

/// Convert an Anthropic `MessageResponse` into a Gemini `GenerateContentResponse`.
///
/// Used when the proxy accepts Gemini CLI input and must return Gemini-format output.
/// This is the inverse of `gemini_to_anthropic_response`.
pub fn anthropic_to_gemini_response(
    resp: &anthropic::MessageResponse,
) -> gemini_resp::GenerateContentResponse {
    let parts: Vec<gemini::Part> = resp
        .content
        .iter()
        .filter_map(|block| match block {
            anthropic::ContentBlock::Text { text } => Some(gemini::Part::text(text.clone())),
            anthropic::ContentBlock::ToolUse { name, input, .. } => {
                Some(gemini::Part::function_call(name.clone(), input.clone()))
            }
            anthropic::ContentBlock::Thinking { thinking, .. } => Some(gemini::Part {
                thought: Some(true),
                text: Some(thinking.clone()),
                ..Default::default()
            }),
            // ToolResult, Image, Document, RedactedThinking: not expected in model output.
            _ => None,
        })
        .collect();

    let finish_reason = resp.stop_reason.as_ref().map(|sr| match sr {
        anthropic::StopReason::EndTurn | anthropic::StopReason::ToolUse => {
            gemini_resp::FinishReason::STOP
        }
        anthropic::StopReason::MaxTokens => gemini_resp::FinishReason::MAX_TOKENS,
        anthropic::StopReason::StopSequence => gemini_resp::FinishReason::STOP,
        _ => gemini_resp::FinishReason::STOP,
    });

    let candidate = gemini_resp::Candidate {
        content: gemini::Content {
            role: Some("model".to_string()),
            parts,
        },
        finish_reason,
        safety_ratings: None,
    };

    gemini_resp::GenerateContentResponse {
        candidates: vec![candidate],
        usage_metadata: Some(gemini_resp::UsageMetadata {
            prompt_token_count: resp.usage.input_tokens,
            candidates_token_count: resp.usage.output_tokens,
            total_token_count: resp.usage.input_tokens + resp.usage.output_tokens,
            cached_content_token_count: 0,
        }),
        model_version: Some(resp.model.clone()),
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use serde_json::json;

    // Helper: build a minimal Anthropic request for testing.
    fn make_request(messages: Vec<anthropic::InputMessage>) -> anthropic::MessageCreateRequest {
        anthropic::MessageCreateRequest {
            model: "claude-3-5-sonnet-20241022".into(),
            max_tokens: 1024,
            messages,
            system: None,
            temperature: None,
            top_p: None,
            top_k: None,
            stop_sequences: None,
            tools: None,
            tool_choice: None,
            metadata: None,
            thinking: None,
            stream: None,
            extra: serde_json::Map::new(),
        }
    }

    fn user_text(text: &str) -> anthropic::InputMessage {
        anthropic::InputMessage {
            role: anthropic::Role::User,
            content: anthropic::Content::Text(text.into()),
        }
    }

    fn user_blocks(blocks: Vec<anthropic::ContentBlock>) -> anthropic::InputMessage {
        anthropic::InputMessage {
            role: anthropic::Role::User,
            content: anthropic::Content::Blocks(blocks),
        }
    }

    fn assistant_blocks(blocks: Vec<anthropic::ContentBlock>) -> anthropic::InputMessage {
        anthropic::InputMessage {
            role: anthropic::Role::Assistant,
            content: anthropic::Content::Blocks(blocks),
        }
    }

    // -----------------------------------------------------------------------
    // Request mapping tests
    // -----------------------------------------------------------------------

    #[test]
    fn simple_text_message_maps_correctly() {
        let req = make_request(vec![user_text("Hello")]);
        let gem = anthropic_to_gemini_request(&req);
        assert_eq!(gem.contents.len(), 1);
        assert_eq!(gem.contents[0].role.as_deref(), Some("user"));
        assert_eq!(gem.contents[0].parts[0].text.as_deref(), Some("Hello"));
    }

    #[test]
    fn system_prompt_text_extracted_to_system_instruction() {
        let mut req = make_request(vec![user_text("Hi")]);
        req.system = Some(anthropic::System::Text("Be helpful.".into()));
        let gem = anthropic_to_gemini_request(&req);
        let si = gem.system_instruction.unwrap();
        assert!(si.role.is_none(), "systemInstruction should have no role");
        assert_eq!(si.parts[0].text.as_deref(), Some("Be helpful."));
    }

    #[test]
    fn system_blocks_concatenated() {
        let mut req = make_request(vec![user_text("Hi")]);
        req.system = Some(anthropic::System::Blocks(vec![
            anthropic::SystemBlock {
                block_type: "text".into(),
                text: "First.".into(),
                cache_control: None,
            },
            anthropic::SystemBlock {
                block_type: "text".into(),
                text: "Second.".into(),
                cache_control: Some(anthropic::CacheControl {
                    cache_type: "ephemeral".into(),
                }),
            },
        ]));
        let gem = anthropic_to_gemini_request(&req);
        let si = gem.system_instruction.unwrap();
        assert_eq!(si.parts[0].text.as_deref(), Some("First.\nSecond."));
    }

    #[test]
    fn assistant_role_maps_to_model() {
        let req = make_request(vec![
            user_text("Hello"),
            assistant_blocks(vec![anthropic::ContentBlock::Text {
                text: "Hi there".into(),
            }]),
        ]);
        let gem = anthropic_to_gemini_request(&req);
        assert_eq!(gem.contents[1].role.as_deref(), Some("model"));
    }

    #[test]
    fn image_content_maps_to_inline_data() {
        let req = make_request(vec![user_blocks(vec![anthropic::ContentBlock::Image {
            source: anthropic::ImageSource {
                source_type: "base64".into(),
                media_type: Some("image/jpeg".into()),
                data: Some("abc123==".into()),
                url: None,
            },
        }])]);
        let gem = anthropic_to_gemini_request(&req);
        let part = &gem.contents[0].parts[0];
        let id = part.inline_data.as_ref().unwrap();
        assert_eq!(id.mime_type, "image/jpeg");
        assert_eq!(id.data, "abc123==");
    }

    #[test]
    fn url_image_dropped() {
        let req = make_request(vec![user_blocks(vec![anthropic::ContentBlock::Image {
            source: anthropic::ImageSource {
                source_type: "url".into(),
                media_type: None,
                data: None,
                url: Some("https://example.com/img.png".into()),
            },
        }])]);
        let gem = anthropic_to_gemini_request(&req);
        // The URL image is dropped; no parts remain, so the content entry is empty
        // and filtered out.
        assert!(gem.contents.is_empty() || gem.contents[0].parts.is_empty());
    }

    #[test]
    fn tool_use_maps_to_function_call_id_stripped() {
        let req = make_request(vec![assistant_blocks(vec![
            anthropic::ContentBlock::ToolUse {
                id: "toolu_abc123".into(),
                name: "get_weather".into(),
                input: json!({"city": "London"}),
            },
        ])]);
        let gem = anthropic_to_gemini_request(&req);
        // A dummy user turn is prepended because Gemini requires user-first; model is at [1].
        let model_content = gem
            .contents
            .iter()
            .find(|c| c.role.as_deref() == Some("model"))
            .expect("should have a model turn");
        let fc = model_content.parts[0].function_call.as_ref().unwrap();
        assert_eq!(fc.name, "get_weather");
        assert_eq!(fc.args, json!({"city": "London"}));
        // No id field on Gemini FunctionCallData.
    }

    #[test]
    fn tool_result_maps_to_function_response_with_name_lookup() {
        let req = make_request(vec![
            assistant_blocks(vec![anthropic::ContentBlock::ToolUse {
                id: "toolu_abc".into(),
                name: "get_weather".into(),
                input: json!({}),
            }]),
            user_blocks(vec![anthropic::ContentBlock::ToolResult {
                tool_use_id: "toolu_abc".into(),
                content: Some(anthropic::ToolResultContent::Text("72F sunny".into())),
                is_error: None,
            }]),
        ]);
        let gem = anthropic_to_gemini_request(&req);
        // Find the user content that contains the function_response (not the dummy
        // empty user turn that was prepended for Gemini's user-first requirement).
        let user_content = gem
            .contents
            .iter()
            .find(|c| {
                c.role.as_deref() == Some("user")
                    && c.parts
                        .first()
                        .and_then(|p| p.function_response.as_ref())
                        .is_some()
            })
            .unwrap();
        let fr = user_content.parts[0].function_response.as_ref().unwrap();
        assert_eq!(fr.name, "get_weather");
        assert_eq!(fr.response, json!({"result": "72F sunny"}));
    }

    #[test]
    fn tool_result_error_wraps_in_error_key() {
        let req = make_request(vec![
            assistant_blocks(vec![anthropic::ContentBlock::ToolUse {
                id: "toolu_err".into(),
                name: "broken_tool".into(),
                input: json!({}),
            }]),
            user_blocks(vec![anthropic::ContentBlock::ToolResult {
                tool_use_id: "toolu_err".into(),
                content: Some(anthropic::ToolResultContent::Text("timeout".into())),
                is_error: Some(true),
            }]),
        ]);
        let gem = anthropic_to_gemini_request(&req);
        let user_content = gem
            .contents
            .iter()
            .find(|c| {
                c.role.as_deref() == Some("user")
                    && c.parts
                        .first()
                        .and_then(|p| p.function_response.as_ref())
                        .is_some()
            })
            .unwrap();
        let fr = user_content.parts[0].function_response.as_ref().unwrap();
        assert_eq!(fr.response, json!({"error": "timeout"}));
    }

    #[test]
    fn tool_result_unknown_id_is_dropped() {
        // A ToolResult whose tool_use_id is not in the tool_id_map must be dropped
        // rather than emitting "unknown_tool", which would cause a Gemini 400.
        // When the only block in a message is dropped, the whole message is omitted.
        let req = make_request(vec![user_blocks(vec![
            anthropic::ContentBlock::ToolResult {
                tool_use_id: "toolu_missing".into(),
                content: Some(anthropic::ToolResultContent::Text("data".into())),
                is_error: None,
            },
        ])]);
        let gem = anthropic_to_gemini_request(&req);
        // The message has no valid parts, so it is omitted entirely from contents.
        assert!(
            gem.contents.is_empty(),
            "unknown ToolResult should be dropped; resulting empty message should be omitted"
        );
    }

    #[test]
    fn thinking_blocks_dropped() {
        let req = make_request(vec![assistant_blocks(vec![
            anthropic::ContentBlock::Thinking {
                thinking: "Let me think...".into(),
                signature: None,
            },
            anthropic::ContentBlock::Text {
                text: "Answer".into(),
            },
        ])]);
        let gem = anthropic_to_gemini_request(&req);
        // A dummy user turn is prepended; find the model turn by role.
        let model_content = gem
            .contents
            .iter()
            .find(|c| c.role.as_deref() == Some("model"))
            .expect("should have a model turn");
        assert_eq!(model_content.parts.len(), 1);
        assert_eq!(model_content.parts[0].text.as_deref(), Some("Answer"));
    }

    #[test]
    fn redacted_thinking_blocks_dropped() {
        let req = make_request(vec![assistant_blocks(vec![
            anthropic::ContentBlock::RedactedThinking {
                data: "encrypted".into(),
            },
            anthropic::ContentBlock::Text {
                text: "Visible".into(),
            },
        ])]);
        let gem = anthropic_to_gemini_request(&req);
        // A dummy user turn is prepended; find the model turn by role.
        let model_content = gem
            .contents
            .iter()
            .find(|c| c.role.as_deref() == Some("model"))
            .expect("should have a model turn");
        assert_eq!(model_content.parts.len(), 1);
        assert_eq!(model_content.parts[0].text.as_deref(), Some("Visible"));
    }

    #[test]
    fn document_blocks_dropped() {
        let req = make_request(vec![user_blocks(vec![
            anthropic::ContentBlock::Document {
                source: anthropic::DocumentSource {
                    source_type: "base64".into(),
                    media_type: "application/pdf".into(),
                    data: "JVBER...".into(),
                },
                title: Some("doc.pdf".into()),
            },
            anthropic::ContentBlock::Text {
                text: "Summarize this".into(),
            },
        ])]);
        let gem = anthropic_to_gemini_request(&req);
        assert_eq!(gem.contents[0].parts.len(), 1);
        assert_eq!(
            gem.contents[0].parts[0].text.as_deref(),
            Some("Summarize this")
        );
    }

    #[test]
    fn tools_mapped_to_function_declarations() {
        let mut req = make_request(vec![user_text("weather?")]);
        req.tools = Some(vec![anthropic::Tool {
            name: "get_weather".into(),
            description: Some("Get weather info".into()),
            input_schema: json!({
                "type": "object",
                "properties": {"city": {"type": "string"}},
                "required": ["city"]
            }),
        }]);
        let gem = anthropic_to_gemini_request(&req);
        let decls = &gem.tools.unwrap()[0].function_declarations;
        assert_eq!(decls.len(), 1);
        assert_eq!(decls[0].name, "get_weather");
        assert_eq!(decls[0].description.as_deref(), Some("Get weather info"));
        assert!(decls[0].parameters.is_some());
    }

    #[test]
    fn tool_schemas_sanitized() {
        let mut req = make_request(vec![user_text("test")]);
        req.tools = Some(vec![anthropic::Tool {
            name: "t".into(),
            description: None,
            input_schema: json!({
                "$schema": "http://json-schema.org/draft-07/schema#",
                "type": "object",
                "additionalProperties": false,
                "properties": {"x": {"type": "string"}}
            }),
        }]);
        let gem = anthropic_to_gemini_request(&req);
        let tools = gem.tools.unwrap();
        let params = tools[0].function_declarations[0]
            .parameters
            .as_ref()
            .unwrap();
        // sanitize_schema_for_gemini strips $schema and additionalProperties
        assert!(params.get("$schema").is_none());
        assert!(params.get("additionalProperties").is_none());
        assert_eq!(params["type"], "object");
    }

    #[test]
    fn tool_choice_auto_maps() {
        let mut req = make_request(vec![user_text("test")]);
        req.tool_choice = Some(anthropic::ToolChoice::Auto {
            disable_parallel_tool_use: None,
        });
        let gem = anthropic_to_gemini_request(&req);
        assert_eq!(
            gem.tool_config.unwrap().function_calling_config.mode,
            "AUTO"
        );
    }

    #[test]
    fn tool_choice_any_maps() {
        let mut req = make_request(vec![user_text("test")]);
        req.tool_choice = Some(anthropic::ToolChoice::Any {
            disable_parallel_tool_use: None,
        });
        let gem = anthropic_to_gemini_request(&req);
        assert_eq!(gem.tool_config.unwrap().function_calling_config.mode, "ANY");
    }

    #[test]
    fn tool_choice_none_maps() {
        let mut req = make_request(vec![user_text("test")]);
        req.tool_choice = Some(anthropic::ToolChoice::None);
        let gem = anthropic_to_gemini_request(&req);
        assert_eq!(
            gem.tool_config.unwrap().function_calling_config.mode,
            "NONE"
        );
    }

    #[test]
    fn tool_choice_specific_tool_maps_to_any_with_allowed_names() {
        let mut req = make_request(vec![user_text("test")]);
        req.tool_choice = Some(anthropic::ToolChoice::Tool {
            name: "get_weather".into(),
        });
        let gem = anthropic_to_gemini_request(&req);
        let fc = gem.tool_config.unwrap().function_calling_config;
        assert_eq!(fc.mode, "ANY");
        assert_eq!(
            fc.allowed_function_names,
            Some(vec!["get_weather".to_string()])
        );
    }

    #[test]
    fn generation_config_fields_mapped() {
        let mut req = make_request(vec![user_text("test")]);
        req.max_tokens = 2048;
        req.temperature = Some(0.7);
        req.top_p = Some(0.9);
        req.top_k = Some(40);
        req.stop_sequences = Some(vec!["STOP".into()]);
        let gem = anthropic_to_gemini_request(&req);
        let gc = gem.generation_config.unwrap();
        assert_eq!(gc.max_output_tokens, Some(2048));
        let temp = gc.temperature.unwrap();
        assert!((temp - 0.7).abs() < 0.001);
        let top_p = gc.top_p.unwrap();
        assert!((top_p - 0.9).abs() < 0.001);
        assert_eq!(gc.top_k, Some(40));
        assert_eq!(gc.stop_sequences, Some(vec!["STOP".into()]));
    }

    #[test]
    fn consecutive_user_messages_merged() {
        let req = make_request(vec![user_text("first"), user_text("second")]);
        let gem = anthropic_to_gemini_request(&req);
        assert_eq!(gem.contents.len(), 1, "should merge into one content");
        assert_eq!(gem.contents[0].parts.len(), 2);
        assert_eq!(gem.contents[0].parts[0].text.as_deref(), Some("first"));
        assert_eq!(gem.contents[0].parts[1].text.as_deref(), Some("second"));
    }

    #[test]
    fn user_user_model_becomes_user_model() {
        let req = make_request(vec![
            user_text("a"),
            user_text("b"),
            assistant_blocks(vec![anthropic::ContentBlock::Text {
                text: "reply".into(),
            }]),
        ]);
        let gem = anthropic_to_gemini_request(&req);
        assert_eq!(gem.contents.len(), 2);
        assert_eq!(gem.contents[0].role.as_deref(), Some("user"));
        assert_eq!(gem.contents[0].parts.len(), 2);
        assert_eq!(gem.contents[1].role.as_deref(), Some("model"));
    }

    #[test]
    fn empty_messages_list() {
        let req = make_request(vec![]);
        let gem = anthropic_to_gemini_request(&req);
        assert!(gem.contents.is_empty());
    }

    #[test]
    fn content_text_shorthand_works() {
        // Content::Text(string) is a shorthand accepted by Anthropic API
        let req = make_request(vec![anthropic::InputMessage {
            role: anthropic::Role::User,
            content: anthropic::Content::Text("shorthand".into()),
        }]);
        let gem = anthropic_to_gemini_request(&req);
        assert_eq!(gem.contents[0].parts[0].text.as_deref(), Some("shorthand"));
    }

    // -----------------------------------------------------------------------
    // build_tool_id_map tests
    // -----------------------------------------------------------------------

    #[test]
    fn build_tool_id_map_finds_tool_uses() {
        let messages = vec![
            assistant_blocks(vec![
                anthropic::ContentBlock::ToolUse {
                    id: "toolu_1".into(),
                    name: "calc".into(),
                    input: json!({}),
                },
                anthropic::ContentBlock::ToolUse {
                    id: "toolu_2".into(),
                    name: "search".into(),
                    input: json!({}),
                },
            ]),
            user_blocks(vec![anthropic::ContentBlock::ToolResult {
                tool_use_id: "toolu_1".into(),
                content: None,
                is_error: None,
            }]),
        ];
        let map = build_tool_id_map(&messages);
        assert_eq!(map.get("toolu_1").unwrap(), "calc");
        assert_eq!(map.get("toolu_2").unwrap(), "search");
    }

    #[test]
    fn build_tool_id_map_empty_on_no_tool_use() {
        let messages = vec![user_text("no tools here")];
        let map = build_tool_id_map(&messages);
        assert!(map.is_empty());
    }

    // -----------------------------------------------------------------------
    // merge_consecutive_roles tests
    // -----------------------------------------------------------------------

    #[test]
    fn merge_consecutive_roles_no_op_for_alternating() {
        let contents = vec![
            gemini::Content {
                role: Some("user".into()),
                parts: vec![gemini::Part::text("a")],
            },
            gemini::Content {
                role: Some("model".into()),
                parts: vec![gemini::Part::text("b")],
            },
        ];
        let merged = merge_consecutive_roles(contents);
        assert_eq!(merged.len(), 2);
    }

    #[test]
    fn merge_consecutive_roles_merges_same_role() {
        let contents = vec![
            gemini::Content {
                role: Some("user".into()),
                parts: vec![gemini::Part::text("a")],
            },
            gemini::Content {
                role: Some("user".into()),
                parts: vec![gemini::Part::text("b")],
            },
            gemini::Content {
                role: Some("model".into()),
                parts: vec![gemini::Part::text("c")],
            },
        ];
        let merged = merge_consecutive_roles(contents);
        assert_eq!(merged.len(), 2);
        assert_eq!(merged[0].parts.len(), 2);
    }

    // -----------------------------------------------------------------------
    // Response mapping tests
    // -----------------------------------------------------------------------

    fn make_gemini_response(
        parts: Vec<gemini::Part>,
        finish_reason: Option<gemini_resp::FinishReason>,
    ) -> gemini_resp::GenerateContentResponse {
        gemini_resp::GenerateContentResponse {
            candidates: vec![gemini_resp::Candidate {
                content: gemini::Content {
                    role: Some("model".into()),
                    parts,
                },
                finish_reason,
                safety_ratings: None,
            }],
            usage_metadata: Some(gemini_resp::UsageMetadata {
                prompt_token_count: 10,
                candidates_token_count: 20,
                total_token_count: 30,
                cached_content_token_count: 0,
            }),
            model_version: None,
        }
    }

    #[test]
    fn simple_text_response() {
        let resp = make_gemini_response(
            vec![gemini::Part::text("Hello!")],
            Some(gemini_resp::FinishReason::STOP),
        );
        let msg = gemini_to_anthropic_response(&resp, "gemini-2.5-flash");
        assert_eq!(msg.content.len(), 1);
        match &msg.content[0] {
            anthropic::ContentBlock::Text { text } => assert_eq!(text, "Hello!"),
            _ => panic!("expected Text block"),
        }
        assert_eq!(msg.model, "gemini-2.5-flash");
        assert_eq!(msg.stop_reason, Some(anthropic::StopReason::EndTurn));
    }

    #[test]
    fn function_call_maps_to_tool_use_with_synthesized_id() {
        let resp = make_gemini_response(
            vec![gemini::Part::function_call(
                "get_weather",
                json!({"city": "NYC"}),
            )],
            Some(gemini_resp::FinishReason::STOP),
        );
        let msg = gemini_to_anthropic_response(&resp, "gemini-2.5-flash");
        assert_eq!(msg.content.len(), 1);
        match &msg.content[0] {
            anthropic::ContentBlock::ToolUse { id, name, input } => {
                assert!(
                    id.starts_with("toolu_"),
                    "synthesized ID should have toolu_ prefix"
                );
                assert_eq!(name, "get_weather");
                assert_eq!(input, &json!({"city": "NYC"}));
            }
            _ => panic!("expected ToolUse block"),
        }
    }

    #[test]
    fn mixed_text_and_function_call() {
        let resp = make_gemini_response(
            vec![
                gemini::Part::text("Let me check the weather."),
                gemini::Part::function_call("get_weather", json!({"city": "London"})),
            ],
            Some(gemini_resp::FinishReason::STOP),
        );
        let msg = gemini_to_anthropic_response(&resp, "gemini-2.5-flash");
        assert_eq!(msg.content.len(), 2);
        assert!(matches!(
            &msg.content[0],
            anthropic::ContentBlock::Text { .. }
        ));
        assert!(matches!(
            &msg.content[1],
            anthropic::ContentBlock::ToolUse { .. }
        ));
        // Has function call + STOP -> ToolUse stop reason.
        assert_eq!(msg.stop_reason, Some(anthropic::StopReason::ToolUse));
    }

    #[test]
    fn finish_reason_stop_without_tools_maps_to_end_turn() {
        let resp = make_gemini_response(
            vec![gemini::Part::text("done")],
            Some(gemini_resp::FinishReason::STOP),
        );
        let msg = gemini_to_anthropic_response(&resp, "test");
        assert_eq!(msg.stop_reason, Some(anthropic::StopReason::EndTurn));
    }

    #[test]
    fn finish_reason_stop_with_function_call_maps_to_tool_use() {
        let resp = make_gemini_response(
            vec![gemini::Part::function_call("f", json!({}))],
            Some(gemini_resp::FinishReason::STOP),
        );
        let msg = gemini_to_anthropic_response(&resp, "test");
        assert_eq!(msg.stop_reason, Some(anthropic::StopReason::ToolUse));
    }

    #[test]
    fn finish_reason_max_tokens_maps_to_max_tokens() {
        let resp = make_gemini_response(
            vec![gemini::Part::text("trunc")],
            Some(gemini_resp::FinishReason::MAX_TOKENS),
        );
        let msg = gemini_to_anthropic_response(&resp, "test");
        assert_eq!(msg.stop_reason, Some(anthropic::StopReason::MaxTokens));
    }

    #[test]
    fn finish_reason_safety_maps_to_end_turn() {
        let resp = make_gemini_response(vec![], Some(gemini_resp::FinishReason::SAFETY));
        let msg = gemini_to_anthropic_response(&resp, "test");
        assert_eq!(msg.stop_reason, Some(anthropic::StopReason::EndTurn));
    }

    #[test]
    fn usage_metadata_mapped() {
        let resp = make_gemini_response(
            vec![gemini::Part::text("ok")],
            Some(gemini_resp::FinishReason::STOP),
        );
        let msg = gemini_to_anthropic_response(&resp, "test");
        assert_eq!(msg.usage.input_tokens, 10);
        assert_eq!(msg.usage.output_tokens, 20);
    }

    #[test]
    fn empty_candidates_gives_empty_content() {
        let resp = gemini_resp::GenerateContentResponse {
            candidates: vec![],
            usage_metadata: None,
            model_version: None,
        };
        let msg = gemini_to_anthropic_response(&resp, "test");
        assert!(msg.content.is_empty());
        assert!(msg.stop_reason.is_none());
        assert_eq!(msg.usage, anthropic::Usage::default());
    }

    #[test]
    fn no_finish_reason_defaults_to_end_turn() {
        let resp = make_gemini_response(vec![gemini::Part::text("partial")], None);
        let msg = gemini_to_anthropic_response(&resp, "test");
        assert_eq!(msg.stop_reason, Some(anthropic::StopReason::EndTurn));
    }

    #[test]
    fn multiple_text_parts_become_separate_blocks() {
        let resp = make_gemini_response(
            vec![gemini::Part::text("one"), gemini::Part::text("two")],
            Some(gemini_resp::FinishReason::STOP),
        );
        let msg = gemini_to_anthropic_response(&resp, "test");
        assert_eq!(msg.content.len(), 2);
        match (&msg.content[0], &msg.content[1]) {
            (
                anthropic::ContentBlock::Text { text: a },
                anthropic::ContentBlock::Text { text: b },
            ) => {
                assert_eq!(a, "one");
                assert_eq!(b, "two");
            }
            _ => panic!("expected two Text blocks"),
        }
    }

    #[test]
    fn model_name_passed_through() {
        let resp = make_gemini_response(
            vec![gemini::Part::text("x")],
            Some(gemini_resp::FinishReason::STOP),
        );
        let msg = gemini_to_anthropic_response(&resp, "gemini-2.5-pro");
        assert_eq!(msg.model, "gemini-2.5-pro");
    }

    #[test]
    fn message_id_has_correct_format() {
        let resp = make_gemini_response(
            vec![gemini::Part::text("x")],
            Some(gemini_resp::FinishReason::STOP),
        );
        let msg = gemini_to_anthropic_response(&resp, "test");
        assert!(msg.id.starts_with("msg_"));
        assert_eq!(msg.response_type, "message");
        assert_eq!(msg.role, anthropic::Role::Assistant);
    }

    #[test]
    fn response_without_usage_metadata_gives_zero_usage() {
        let mut resp = make_gemini_response(
            vec![gemini::Part::text("x")],
            Some(gemini_resp::FinishReason::STOP),
        );
        resp.usage_metadata = None;
        let msg = gemini_to_anthropic_response(&resp, "test");
        assert_eq!(msg.usage.input_tokens, 0);
        assert_eq!(msg.usage.output_tokens, 0);
    }

    // -----------------------------------------------------------------------
    // Thinking config tests
    // -----------------------------------------------------------------------

    #[test]
    fn thinking_config_enabled_sets_gemini_thinking_config() {
        let mut req = make_request(vec![user_text("think hard")]);
        req.thinking = Some(anthropic::ThinkingConfig::Enabled {
            budget_tokens: 8192,
        });
        let gem = anthropic_to_gemini_request(&req);
        let gc = gem.generation_config.unwrap();
        let tc = gc.thinking_config.expect("thinkingConfig should be set");
        assert_eq!(tc.thinking_budget, 8192);
        assert_eq!(tc.include_thoughts, Some(true));
    }

    #[test]
    fn thinking_config_disabled_no_thinking_config() {
        let mut req = make_request(vec![user_text("hi")]);
        req.thinking = Some(anthropic::ThinkingConfig::Disabled);
        let gem = anthropic_to_gemini_request(&req);
        let gc = gem.generation_config.unwrap();
        assert!(
            gc.thinking_config.is_none(),
            "disabled should not set thinkingConfig"
        );
    }

    #[test]
    fn thinking_config_absent_no_thinking_config() {
        let req = make_request(vec![user_text("hi")]);
        let gem = anthropic_to_gemini_request(&req);
        let gc = gem.generation_config.unwrap();
        assert!(gc.thinking_config.is_none());
    }

    #[test]
    fn gemini_thought_parts_become_thinking_blocks() {
        let thought_part = gemini::Part {
            thought: Some(true),
            text: Some("Let me reason...".into()),
            ..Default::default()
        };
        let resp = make_gemini_response(
            vec![thought_part, gemini::Part::text("Answer")],
            Some(gemini_resp::FinishReason::STOP),
        );
        let msg = gemini_to_anthropic_response(&resp, "gemini-2.5-pro");
        assert_eq!(msg.content.len(), 2);
        match &msg.content[0] {
            anthropic::ContentBlock::Thinking { thinking, .. } => {
                assert_eq!(thinking, "Let me reason...")
            }
            _ => panic!("expected Thinking block first"),
        }
        match &msg.content[1] {
            anthropic::ContentBlock::Text { text } => assert_eq!(text, "Answer"),
            _ => panic!("expected Text block second"),
        }
    }

    #[test]
    fn gemini_thought_only_no_text() {
        let thought_part = gemini::Part {
            thought: Some(true),
            text: Some("Only thinking".into()),
            ..Default::default()
        };
        let resp = make_gemini_response(vec![thought_part], Some(gemini_resp::FinishReason::STOP));
        let msg = gemini_to_anthropic_response(&resp, "gemini-2.5-pro");
        assert_eq!(msg.content.len(), 1);
        assert!(matches!(
            &msg.content[0],
            anthropic::ContentBlock::Thinking { .. }
        ));
    }
}