harn-vm 0.10.120

use super::parse_llm_response;
use crate::llm::capabilities::WireDialect;

#[test]
fn deepinfra_estimated_cost_becomes_the_public_usage_cost() {
    let response = serde_json::json!({
        "id": "chatcmpl-deepinfra-receipt",
        "model": "moonshotai/Kimi-K3",
        "choices": [{
            "index": 0,
            "finish_reason": "stop",
            "message": { "role": "assistant", "content": "ok" }
        }],
        "usage": {
            "prompt_tokens": 91,
            "completion_tokens": 1,
            "total_tokens": 92,
            "estimated_cost": 0.0002736
        }
    });

    let result = parse_llm_response(
        &response,
        "deepinfra",
        "moonshotai/Kimi-K3",
        WireDialect::OpenAiCompat,
        false,
    )
    .expect("DeepInfra response parses");

    assert_eq!(result.telemetry.provider_cost_usd, Some(0.0002736));
    let usage = result.usage();
    assert_eq!(usage.input_tokens, 91);
    assert_eq!(usage.output_tokens, 1);
    assert_eq!(usage.cost_usd, Some(0.0002736));
    assert_eq!(usage.known_cost_usd, 0.0002736);
    assert_eq!(usage.unpriced_calls, 0);
}

#[test]
fn xai_cost_ticks_become_the_public_usage_cost() {
    let response = serde_json::json!({
        "id": "chatcmpl-xai-receipt",
        "model": "grok-build-0.1",
        "choices": [{
            "index": 0,
            "finish_reason": "stop",
            "message": { "role": "assistant", "content": "ok" }
        }],
        "usage": {
            "prompt_tokens": 191,
            "completion_tokens": 1,
            "total_tokens": 324,
            "cost_in_usd_ticks": 3_546_000
        }
    });

    let result = parse_llm_response(
        &response,
        "xai",
        "grok-build-0.1",
        WireDialect::OpenAiCompat,
        false,
    )
    .expect("xAI response parses");

    assert_eq!(result.telemetry.provider_cost_usd, Some(0.0003546));
    let usage = result.usage();
    assert_eq!(usage.input_tokens, 191);
    assert_eq!(usage.output_tokens, 1);
    assert_eq!(usage.cost_usd, Some(0.0003546));
    assert_eq!(usage.known_cost_usd, 0.0003546);
    assert_eq!(usage.unpriced_calls, 0);
}

#[test]
fn openai_parser_preserves_harmony_wrapper_before_normalizing() {
    let response = serde_json::json!({
        "choices": [{
            "message": {
                "content": "",
                "tool_calls": [{
                    "id": "chatcmpl-tool-1",
                    "type": "function",
                    "function": {
                        "name": "tool",
                        "arguments": "{\"name\":\"look\",\"args\":{\"intent\":\"read\",\"file\":\"src/lib.rs\"}}"
                    }
                }]
            },
            "finish_reason": "tool_calls"
        }],
        "usage": {"prompt_tokens": 10, "completion_tokens": 20}
    });

    let result = parse_llm_response(
        &response,
        "fireworks",
        "accounts/fireworks/models/gpt-oss-120b",
        WireDialect::OpenAiCompat,
        false,
    )
    .expect("parser succeeds");

    assert_eq!(result.tool_calls.len(), 1);
    assert_eq!(result.tool_calls[0]["name"], "look");
    assert_eq!(result.tool_calls[0]["arguments"]["intent"], "read");
    assert_eq!(result.tool_calls[0]["arguments"]["file"], "src/lib.rs");
    assert_eq!(result.raw_tool_calls.len(), 1);
    assert_eq!(result.raw_tool_calls[0]["function"]["name"], "tool");
    assert_eq!(
        result.raw_tool_calls[0]["function"]["arguments"],
        "{\"name\":\"look\",\"args\":{\"intent\":\"read\",\"file\":\"src/lib.rs\"}}"
    );
}

#[test]
fn openai_parser_recovers_look_from_harmony_recipient_to_wrapper() {
    // Fireworks Harmony demux can lift the recipient token `to` into
    // function.name while leaving clean look args and no real tool name
    // (burin-code#4809 Face 2).
    let response = serde_json::json!({
        "choices": [{
            "message": {
                "content": "",
                "tool_calls": [{
                    "id": "call_to_wrapper",
                    "type": "function",
                    "function": {
                        "name": "to",
                        "arguments": "{\"file\": \"Sources/App.swift\", \"intent\": \"read\"}"
                    }
                }]
            },
            "finish_reason": "tool_calls"
        }],
        "usage": {"prompt_tokens": 10, "completion_tokens": 12}
    });

    let result = parse_llm_response(
        &response,
        "fireworks",
        "gpt-oss-120b",
        WireDialect::OpenAiCompat,
        false,
    )
    .expect("parser succeeds");

    assert_eq!(result.stop_reason.as_deref(), Some("tool_calls"));
    assert_eq!(result.tool_calls.len(), 1);
    assert_eq!(result.tool_calls[0]["name"], "look");
    assert_eq!(
        result.tool_calls[0]["arguments"]["file"],
        "Sources/App.swift"
    );
    assert_eq!(result.tool_calls[0]["arguments"]["intent"], "read");
    assert_eq!(result.raw_tool_calls.len(), 1);
    assert_eq!(result.raw_tool_calls[0]["function"]["name"], "to");
}

#[test]
fn openai_parser_preserves_channel_suffix_before_stripping() {
    let response = serde_json::json!({
        "choices": [{
            "message": {
                "content": "",
                "tool_calls": [{
                    "id": "chatcmpl-tool-1",
                    "type": "function",
                    "function": {
                        "name": "run<|channel|>commentary",
                        "arguments": "{\"command\":\"cargo test\"}"
                    }
                }]
            },
            "finish_reason": "tool_calls"
        }],
        "usage": {"prompt_tokens": 10, "completion_tokens": 20}
    });

    let result = parse_llm_response(
        &response,
        "fireworks",
        "accounts/fireworks/models/gpt-oss-120b",
        WireDialect::OpenAiCompat,
        false,
    )
    .expect("parser succeeds");

    assert_eq!(result.tool_calls.len(), 1);
    assert_eq!(result.tool_calls[0]["name"], "run");
    assert_eq!(result.tool_calls[0]["arguments"]["command"], "cargo test");
    assert_eq!(result.raw_tool_calls.len(), 1);
    assert_eq!(
        result.raw_tool_calls[0]["function"]["name"],
        "run<|channel|>commentary"
    );
}