package com.gzzm.lobster.llm.adapter;

import org.junit.Test;

import java.util.LinkedHashMap;
import java.util.Map;

import static org.junit.Assert.assertEquals;

public class OpenAiCompatibleAdapterCacheUsageTest {

    @Test
    public void promptTokenCountAcceptsInputTokensAlias() {
        Map<String, Object> usage = new LinkedHashMap<>();
        usage.put("input_tokens", 1000);

        assertEquals(1000, OpenAiCompatibleAdapter.promptTokenCount(usage));
    }

    @Test
    public void promptTokensWinOverInputTokensWhenBothPresent() {
        Map<String, Object> usage = new LinkedHashMap<>();
        usage.put("prompt_tokens", 900);
        usage.put("input_tokens", 1000);

        assertEquals(900, OpenAiCompatibleAdapter.promptTokenCount(usage));
    }
}