Skip to content

Commit 4c14874

Browse files
committed
fix: IsCustom() -> IsCustomModel
1 parent 9c92a9a commit 4c14874

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

internal/services/toolkit/client/completion_v2.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,7 @@ func (a *AIClientV2) ChatCompletionStreamV2(ctx context.Context, callbackStream
7979
// Track usage on all exit paths (success or error) to prevent abuse
8080
// Only track if userID is provided and user is not using their own API key (BYOK)
8181
defer func() {
82-
if !userID.IsZero() && !llmProvider.IsCustom() && usage.Cost > 0 {
82+
if !userID.IsZero() && !llmProvider.IsCustomModel && usage.Cost > 0 {
8383
// Use a detached context since the request context may be canceled
8484
trackCtx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
8585
defer cancel()

0 commit comments

Comments
 (0)