def test_missing_token_counts(self):
        """Ensures missing token counts default to zero."""
        raw_response = {"usage": {}}
        result = _get_response_token_counts(raw_response)
        assert result == {}

        raw_response = {"usage": {"prompt_tokens": 10}}
        expected_result = {
            "prompt_tokens": 10,
            "completion_tokens": 0,
            "total_tokens": 0,
        }
        result = _get_response_token_counts(raw_response)
        assert result == expected_result
Ensures missing token counts default to zero.
llama-index-integrations/llms/llama-index-llms-oci-data-science/tests/test_oci_data_science_utils.py