openai_gateway_messages_usage_test.go 679 B

12345678910111213141516171819202122232425262728
  1. //go:build unit
  2. package service
  3. import (
  4. "testing"
  5. "github.com/Wei-Shaw/sub2api/internal/pkg/apicompat"
  6. "github.com/stretchr/testify/require"
  7. )
  8. func TestCopyOpenAIUsageFromResponsesUsageTrustsCanonicalCacheCreationValue(t *testing.T) {
  9. usage := &apicompat.ResponsesUsage{
  10. InputTokens: 20,
  11. OutputTokens: 2,
  12. CacheCreationInputTokens: 0,
  13. InputTokensDetails: &apicompat.ResponsesInputTokensDetails{
  14. CachedTokens: 3,
  15. CacheWriteTokens: 19,
  16. },
  17. }
  18. got := copyOpenAIUsageFromResponsesUsage(usage)
  19. require.Equal(t, 20, got.InputTokens)
  20. require.Equal(t, 3, got.CacheReadInputTokens)
  21. require.Zero(t, got.CacheCreationInputTokens)
  22. }