zen: update rate limiter
Some checks are pending
deploy / deploy (push) Waiting to run

This commit is contained in:
Frank 2026-05-06 15:32:08 -04:00
parent e41843eaf7
commit a4ab1408eb
2 changed files with 8 additions and 3 deletions

View file

@ -116,7 +116,7 @@ export async function handler(
const trialProviders = await trialLimiter?.check()
const rateLimiter = modelInfo.allowAnonymous
? 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()
const stickyTracker = createStickyTracker(modelInfo.stickyProvider, sessionId)
const stickyProvider = await stickyTracker?.get()

View file

@ -4,11 +4,16 @@ import { RateLimitError } from "./error"
import { i18n } from "~/i18n"
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
const dict = i18n(localeFromRequest(request))
const LIMIT = 300
const LIMIT = rateLimit ?? 300
const yyyyMMddHHmm = new Date(Date.now())
.toISOString()
.replace(/[^0-9]/g, "")