calvin.erfmann/pangolin
1
0
Fork 0
mirror of https://github.com/fosrl/pangolin.git synced 2026-09-08 12:13:16 +02:00
Code Issues Packages Projects Releases Wiki Activity

adjust conflict to be on unit<>period and add list endpoints for providers

Browse Source
This commit is contained in:
Owen
2026-08-10 11:39:34 -04:00
parent 0050fad70d
commit b180d064d1
12 changed files with 479 additions and 63 deletions
Download Patch File Download Diff File Expand all files Collapse all files
server/routers/aiBudget/createAiBudget.ts
+30 -17
View File
@@ -15,7 +15,7 @@ import createHttpError from "http-errors";
import logger from "@server/logger";
import { fromError } from "zod-validation-error";
import { OpenAPITags, registry } from "@server/openApi";
import { eq } from "drizzle-orm";
import { and, eq, isNull } from "drizzle-orm";
import type { CreateOrEditAiBudgetResponse } from "@server/routers/aiBudget/types";
import {
aiBudgetEnforcementSchema,
@@ -189,7 +189,7 @@ export async function createAiBudget(
}
}
const conflictCondition =
const scopeCondition =
providerId !== undefined
? eq(aiBudgets.providerId, providerId)
: modelId !== undefined
@@ -198,22 +198,35 @@ export async function createAiBudget(
? eq(aiBudgets.resourceId, resourceId)
: siteResourceId !== undefined
? eq(aiBudgets.siteResourceId, siteResourceId)
: undefined;
: roleId !== undefined
? eq(aiBudgets.roleId, roleId)
: and(
eq(aiBudgets.orgId, orgId),
isNull(aiBudgets.providerId),
isNull(aiBudgets.modelId),
isNull(aiBudgets.resourceId),
isNull(aiBudgets.siteResourceId),
isNull(aiBudgets.roleId)
);
if (conflictCondition) {
const [existing] = await db
.select({ budgetId: aiBudgets.budgetId })
.from(aiBudgets)
.where(conflictCondition)
.limit(1);
if (existing) {
return next(
createHttpError(
HttpCode.CONFLICT,
"A budget already exists for this scope"
)
);
}
const [existing] = await db
.select({ budgetId: aiBudgets.budgetId })
.from(aiBudgets)
.where(
and(
scopeCondition,
eq(aiBudgets.unit, unit),
eq(aiBudgets.period, period)
)
)
.limit(1);
if (existing) {
return next(
createHttpError(
HttpCode.CONFLICT,
`A ${period} ${unit} budget already exists for this scope`
)
);
}
const now = Date.now();