github/sim
1
0
Fork 0
mirror of https://github.com/simstudioai/sim.git synced 2026-01-10 23:48:09 -05:00
Code Issues Packages Projects Releases Wiki Activity

fix(copilot): consolidate usage limit validation (#1763)

Browse Source 
* Use helper for api key validation

* Lint
This commit is contained in:
Siddharth Ganesan
2025-10-29 13:14:28 -07:00
committed by GitHub
parent fcf947df22
commit b95ea9148f
1 changed files with 11 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
apps/sim/app/api/copilot/api-keys/validate/route.ts
View File

@@ -1,7 +1,5 @@
import { db } from '@sim/db'
import { userStats } from '@sim/db/schema'
import { eq } from 'drizzle-orm'
import { type NextRequest, NextResponse } from 'next/server'
import { checkServerSideUsageLimits } from '@/lib/billing/calculations/usage-monitor'
import { checkInternalApiKey } from '@/lib/copilot/utils'
import { createLogger } from '@/lib/logs/console/logger'
@@ -24,30 +22,18 @@ export async function POST(req: NextRequest) {
logger.info('[API VALIDATION] Validating usage limit', { userId })
const usage = await db
.select({
currentPeriodCost: userStats.currentPeriodCost,
totalCost: userStats.totalCost,
currentUsageLimit: userStats.currentUsageLimit,
})
.from(userStats)
.where(eq(userStats.userId, userId))
.limit(1)
const { isExceeded, currentUsage, limit } = await checkServerSideUsageLimits(userId)
logger.info('[API VALIDATION] Usage limit validated', { userId, usage })
logger.info('[API VALIDATION] Usage limit validated', {
userId,
currentUsage,
limit,
isExceeded,
})
if (usage.length > 0) {
const currentUsage = Number.parseFloat(
(usage[0].currentPeriodCost?.toString() as string) ||
(usage[0].totalCost as unknown as string) ||
'0'
)
const limit = Number.parseFloat((usage[0].currentUsageLimit as unknown as string) || '0')
if (!Number.isNaN(limit) && limit > 0 && currentUsage >= limit) {
logger.info('[API VALIDATION] Usage exceeded', { userId, currentUsage, limit })
return new NextResponse(null, { status: 402 })
}
if (isExceeded) {
logger.info('[API VALIDATION] Usage exceeded', { userId, currentUsage, limit })
return new NextResponse(null, { status: 402 })
}
return new NextResponse(null, { status: 200 })