fix: golangci-lint test assertion and gofmt
This commit is contained in:
@@ -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
|
|||||||
}
|
}
|
||||||
|
|
||||||
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
|
||||||
|
|||||||
Reference in New Issue
Block a user