diff --git a/packages/console/app/src/routes/zen/util/handler.ts b/packages/console/app/src/routes/zen/util/handler.ts index ee223498ca..4ef8b7a208 100644 --- a/packages/console/app/src/routes/zen/util/handler.ts +++ b/packages/console/app/src/routes/zen/util/handler.ts @@ -252,6 +252,8 @@ export async function handler( }) if (isNewInference) { headers.set("x-zen-model", model) + if (opts.modelList === "lite") + headers.set("x-zen-billing-source", billingSource === "lite" ? "go" : "credit") } headers.delete("host") headers.delete("content-length") @@ -261,6 +263,7 @@ export async function handler( headers.delete("x-opencode-client") headers.delete("x-opencode-request") headers.delete("x-zen-model") + headers.delete("x-zen-billing-source") } return headers })(),