ceos/opencode
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

zen: update rate limiter

Browse source
This commit is contained in:
Frank 2026-05-06 15:32:08 -04:00
commit a4ab1408eb
2 changed files with 8 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
packages/console/app/src/routes/zen/util/handler.ts
View file

@ -116,7 +116,7 @@ export async function handler(
const trialProviders = await trialLimiter?.check() const trialProviders = await trialLimiter?.check()
const rateLimiter = modelInfo.allowAnonymous const rateLimiter = modelInfo.allowAnonymous
? createIpRateLimiter(modelInfo.id, modelInfo.rateLimit, ip, input.request) ? createIpRateLimiter(modelInfo.id, modelInfo.rateLimit, ip, input.request)
: createKeyRateLimiter(modelInfo.id, zenApiKey, input.request) : createKeyRateLimiter(modelInfo.id, modelInfo.rateLimit, zenApiKey, input.request)
await rateLimiter?.check() await rateLimiter?.check()
const stickyTracker = createStickyTracker(modelInfo.stickyProvider, sessionId) const stickyTracker = createStickyTracker(modelInfo.stickyProvider, sessionId)
const stickyProvider = await stickyTracker?.get() const stickyProvider = await stickyTracker?.get()

9
packages/console/app/src/routes/zen/util/keyRateLimiter.ts
View file

@ -4,11 +4,16 @@ import { RateLimitError } from "./error"
import { i18n } from "~/i18n" import { i18n } from "~/i18n"
import { localeFromRequest } from "~/lib/language" import { localeFromRequest } from "~/lib/language"
export function createRateLimiter(modelId: string, zenApiKey: string | undefined, request: Request) { export function createRateLimiter(
modelId: string,
rateLimit: number | undefined,
zenApiKey: string | undefined,
request: Request,
) {
if (!zenApiKey) return if (!zenApiKey) return
const dict = i18n(localeFromRequest(request)) const dict = i18n(localeFromRequest(request))
const LIMIT = 300 const LIMIT = rateLimit ?? 300
const yyyyMMddHHmm = new Date(Date.now()) const yyyyMMddHHmm = new Date(Date.now())
.toISOString() .toISOString()
.replace(/[^0-9]/g, "") .replace(/[^0-9]/g, "")