Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Sign in / Register
Toggle navigation
Menu
Open sidebar
陈曦
sub2api
Commits
c6fd8811
Unverified
Commit
c6fd8811
authored
Mar 30, 2026
by
Wesley Liddick
Committed by
GitHub
Mar 30, 2026
Browse files
Merge pull request #1354 from wucm667/fix/billing-use-requested-model
fix(billing): 计费始终使用用户请求的原始模型,而非映射后的上游模型
parents
8f0dbdea
f5764d8d
Changes
3
Hide whitespace changes
Inline
Side-by-side
backend/internal/service/openai_gateway_record_usage_test.go
View file @
c6fd8811
...
@@ -895,14 +895,16 @@ func TestOpenAIGatewayServiceRecordUsage_UsesRequestedModelAndUpstreamModelMetad
...
@@ -895,14 +895,16 @@ func TestOpenAIGatewayServiceRecordUsage_UsesRequestedModelAndUpstreamModelMetad
require
.
Equal
(
t
,
1
,
userRepo
.
deductCalls
)
require
.
Equal
(
t
,
1
,
userRepo
.
deductCalls
)
}
}
func
TestOpenAIGatewayServiceRecordUsage_BillsMappedRequestsUsing
UpstreamModelFallback
(
t
*
testing
.
T
)
{
func
TestOpenAIGatewayServiceRecordUsage_BillsMappedRequestsUsing
RequestedModel
(
t
*
testing
.
T
)
{
usageRepo
:=
&
openAIRecordUsageLogRepoStub
{
inserted
:
true
}
usageRepo
:=
&
openAIRecordUsageLogRepoStub
{
inserted
:
true
}
userRepo
:=
&
openAIRecordUsageUserRepoStub
{}
userRepo
:=
&
openAIRecordUsageUserRepoStub
{}
subRepo
:=
&
openAIRecordUsageSubRepoStub
{}
subRepo
:=
&
openAIRecordUsageSubRepoStub
{}
svc
:=
newOpenAIRecordUsageServiceForTest
(
usageRepo
,
userRepo
,
subRepo
,
nil
)
svc
:=
newOpenAIRecordUsageServiceForTest
(
usageRepo
,
userRepo
,
subRepo
,
nil
)
usage
:=
OpenAIUsage
{
InputTokens
:
20
,
OutputTokens
:
10
}
usage
:=
OpenAIUsage
{
InputTokens
:
20
,
OutputTokens
:
10
}
expectedCost
,
err
:=
svc
.
billingService
.
CalculateCost
(
"gpt-5.1-codex"
,
UsageTokens
{
// Billing should use the requested model ("gpt-5.1"), not the upstream mapped model ("gpt-5.1-codex").
// This ensures pricing is always based on the model the user requested.
expectedCost
,
err
:=
svc
.
billingService
.
CalculateCost
(
"gpt-5.1"
,
UsageTokens
{
InputTokens
:
20
,
InputTokens
:
20
,
OutputTokens
:
10
,
OutputTokens
:
10
,
},
1.1
)
},
1.1
)
...
...
backend/internal/service/openai_gateway_service.go
View file @
c6fd8811
...
@@ -4153,9 +4153,6 @@ func (s *OpenAIGatewayService) RecordUsage(ctx context.Context, input *OpenAIRec
...
@@ -4153,9 +4153,6 @@ func (s *OpenAIGatewayService) RecordUsage(ctx context.Context, input *OpenAIRec
}
}
billingModel
:=
forwardResultBillingModel
(
result
.
Model
,
result
.
UpstreamModel
)
billingModel
:=
forwardResultBillingModel
(
result
.
Model
,
result
.
UpstreamModel
)
if
result
.
BillingModel
!=
""
{
billingModel
=
strings
.
TrimSpace
(
result
.
BillingModel
)
}
serviceTier
:=
""
serviceTier
:=
""
if
result
.
ServiceTier
!=
nil
{
if
result
.
ServiceTier
!=
nil
{
serviceTier
=
strings
.
TrimSpace
(
*
result
.
ServiceTier
)
serviceTier
=
strings
.
TrimSpace
(
*
result
.
ServiceTier
)
...
...
backend/internal/service/usage_log_helpers.go
View file @
c6fd8811
...
@@ -21,8 +21,8 @@ func optionalNonEqualStringPtr(value, compare string) *string {
...
@@ -21,8 +21,8 @@ func optionalNonEqualStringPtr(value, compare string) *string {
}
}
func
forwardResultBillingModel
(
requestedModel
,
upstreamModel
string
)
string
{
func
forwardResultBillingModel
(
requestedModel
,
upstreamModel
string
)
string
{
if
trimmed
Upstream
:=
strings
.
TrimSpace
(
upstream
Model
);
trimmed
Upstream
!=
""
{
if
trimmed
:=
strings
.
TrimSpace
(
requested
Model
);
trimmed
!=
""
{
return
trimmed
Upstream
return
trimmed
}
}
return
strings
.
TrimSpace
(
requested
Model
)
return
strings
.
TrimSpace
(
upstream
Model
)
}
}
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment