LLM/opencode
1
0
Fork 0
mirror of https://github.com/anomalyco/opencode.git synced 2026-05-13 15:44:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

zen: update rate limiter

Browse Source
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
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 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()

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 { 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, "")