lmin/zmo-cli
1
0
Fork 0
Code Issues Pull Requests Actions 80 Packages Projects Releases Wiki Activity

sync

Browse Source
This commit is contained in:
Frank
2026-04-17 10:53:45 -04:00
parent 10c4ab9a3d
commit 20103eb97b
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/console/app/src/routes/zen/util/modelTpmLimiter.ts
+3 -1
View File
@@ -30,6 +30,8 @@ export function createModelTpmLimiter(providers: { id: string; model: string; tp
) )
}, },
track: async (id: string, model: string, usageInfo: UsageInfo) => { track: async (id: string, model: string, usageInfo: UsageInfo) => {
const key = `${id}/${model}`
if (!keys.includes(key)) return
const usage = const usage =
usageInfo.inputTokens + usageInfo.inputTokens +
usageInfo.outputTokens + usageInfo.outputTokens +
@@ -41,7 +43,7 @@ export function createModelTpmLimiter(providers: { id: string; model: string; tp
await Database.use((tx) => await Database.use((tx) =>
tx tx
.insert(ModelRateLimitTable) .insert(ModelRateLimitTable)
.values({ key: `${id}/${model}`, interval: yyyyMMddHHmm, count: usage }) .values({ key, interval: yyyyMMddHHmm, count: usage })
.onDuplicateKeyUpdate({ set: { count: sql`${ModelRateLimitTable.count} + ${usage}` } }), .onDuplicateKeyUpdate({ set: { count: sql`${ModelRateLimitTable.count} + ${usage}` } }),
) )
}, },