Commit c9145ad4 authored by erio's avatar erio
Browse files

fix: golangci-lint test assertion and gofmt

parent 3851628a
...@@ -493,7 +493,7 @@ func TestValidatePricingBillingMode(t *testing.T) { ...@@ -493,7 +493,7 @@ func TestValidatePricingBillingMode(t *testing.T) {
err := validatePricingBillingMode(tt.pricing) err := validatePricingBillingMode(tt.pricing)
if tt.wantErr { if tt.wantErr {
require.Error(t, err) require.Error(t, err)
require.Contains(t, err.Error(), "Per-request price or intervals required") require.Contains(t, err.Error(), "per-request price or intervals required")
} else { } else {
require.NoError(t, err) require.NoError(t, err)
} }
......
...@@ -4245,22 +4245,22 @@ func (s *OpenAIGatewayService) RecordUsage(ctx context.Context, input *OpenAIRec ...@@ -4245,22 +4245,22 @@ func (s *OpenAIGatewayService) RecordUsage(ctx context.Context, input *OpenAIRec
} }
usageLog := &UsageLog{ usageLog := &UsageLog{
UserID: user.ID, UserID: user.ID,
APIKeyID: apiKey.ID, APIKeyID: apiKey.ID,
AccountID: account.ID, AccountID: account.ID,
RequestID: requestID, RequestID: requestID,
Model: result.Model, Model: result.Model,
RequestedModel: requestedModel, RequestedModel: requestedModel,
UpstreamModel: optionalNonEqualStringPtr(result.UpstreamModel, result.Model), UpstreamModel: optionalNonEqualStringPtr(result.UpstreamModel, result.Model),
ServiceTier: result.ServiceTier, ServiceTier: result.ServiceTier,
ReasoningEffort: result.ReasoningEffort, ReasoningEffort: result.ReasoningEffort,
InboundEndpoint: optionalTrimmedStringPtr(input.InboundEndpoint), InboundEndpoint: optionalTrimmedStringPtr(input.InboundEndpoint),
UpstreamEndpoint: optionalTrimmedStringPtr(input.UpstreamEndpoint), UpstreamEndpoint: optionalTrimmedStringPtr(input.UpstreamEndpoint),
InputTokens: actualInputTokens, InputTokens: actualInputTokens,
OutputTokens: result.Usage.OutputTokens, OutputTokens: result.Usage.OutputTokens,
CacheCreationTokens: result.Usage.CacheCreationInputTokens, CacheCreationTokens: result.Usage.CacheCreationInputTokens,
CacheReadTokens: result.Usage.CacheReadInputTokens, CacheReadTokens: result.Usage.CacheReadInputTokens,
ImageOutputTokens: result.Usage.ImageOutputTokens, ImageOutputTokens: result.Usage.ImageOutputTokens,
} }
if cost != nil { if cost != nil {
usageLog.InputCost = cost.InputCost usageLog.InputCost = cost.InputCost
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment