Przeglądaj źródła

Fix usage & billing for custom model aliases and cached/reasoning tokens (#4222)

Co-authored-by: Melih Mucuk <[email protected]>
Co-authored-by: Aiden Cline <[email protected]>
Melih Mucuk 3 miesięcy temu
rodzic
commit
b63b6d04c6
1 zmienionych plików z 3 dodań i 0 usunięć
  1. 3 0
      packages/opencode/src/session/index.ts

+ 3 - 0
packages/opencode/src/session/index.ts

@@ -402,6 +402,9 @@ export namespace Session {
           .add(new Decimal(tokens.output).mul(input.model.cost?.output ?? 0).div(1_000_000))
           .add(new Decimal(tokens.cache.read).mul(input.model.cost?.cache_read ?? 0).div(1_000_000))
           .add(new Decimal(tokens.cache.write).mul(input.model.cost?.cache_write ?? 0).div(1_000_000))
+          // TODO: update models.dev to have better pricing model, for now:
+          // charge reasoning tokens at the same rate as output tokens
+          .add(new Decimal(tokens.reasoning).mul(input.model.cost?.output ?? 0).div(1_000_000))
           .toNumber(),
         tokens,
       }