from inline_snapshot import snapshot

test_snapshot = snapshot(
    {
        "response1": "**ToolCall (star_wars_tool):** {}",
        "usage1": {
            "input_tokens": 3900,
            "output_tokens": 89,
            "cache_read_tokens": 0,
            "cache_write_tokens": 0,
            "reasoning_tokens": 77,
            "provider_tool_usage": None,
            "raw": """\
cache_tokens_details=None cached_content_token_count=None candidates_token_count=12 candidates_tokens_details=None prompt_token_count=3900 prompt_tokens_details=[ModalityTokenCount(
  modality=<MediaModality.TEXT: 'TEXT'>,
  token_count=3900
)] thoughts_token_count=77 tool_use_prompt_token_count=None tool_use_prompt_tokens_details=None total_token_count=3989 traffic_type=None\
""",
            "total_tokens": 3989,
        },
        "response2": """\
Ah, a query of sustenance! For breakfast, young Padawan, you might consider some nutrient paste, or perhaps a warm mug of Jawa Juice to awaken your internal systems.

However, my primary function is to navigate the hyperspace lanes of Star Wars lore and deliver responses with the precision of a thermal detonator! Would you like me to engage with you in the manner of a wise Jedi Master, a cunning scoundrel, or perhaps a stoic Mandalorian? What Star Wars wisdom or meme-tastic response can I offer you today?\
""",
        "usage2": {
            "input_tokens": 3904,
            "output_tokens": 295,
            "cache_read_tokens": 0,
            "cache_write_tokens": 0,
            "reasoning_tokens": 185,
            "provider_tool_usage": None,
            "raw": """\
cache_tokens_details=None cached_content_token_count=None candidates_token_count=110 candidates_tokens_details=None prompt_token_count=3904 prompt_tokens_details=[ModalityTokenCount(
  modality=<MediaModality.TEXT: 'TEXT'>,
  token_count=3904
)] thoughts_token_count=185 tool_use_prompt_token_count=None tool_use_prompt_tokens_details=None total_token_count=4199 traffic_type=None\
""",
            "total_tokens": 4199,
        },
    }
)
