ai_api / src /shared /validation /schemas /payloadRules.ts
Yogesh
initial deploy
cd8bd0a
Raw
History Blame Contribute Delete
2.15 kB
import { z } from "zod";
import {
ACCOUNT_FALLBACK_STRATEGY_VALUES,
ROUTING_STRATEGY_VALUES,
} from "@/shared/constants/routingStrategies";
import { SUPPORTED_BATCH_ENDPOINTS } from "@/shared/constants/batchEndpoints";
import { MAX_REQUEST_BODY_LIMIT_MB, MIN_REQUEST_BODY_LIMIT_MB } from "@/shared/constants/bodySize";
import { COMBO_CONFIG_MODES } from "@/shared/constants/comboConfigMode";
import { providerAllowsOptionalApiKey } from "@/shared/constants/providers";
import { HIDEABLE_SIDEBAR_ITEM_IDS } from "@/shared/constants/sidebarVisibility";
import {
isForbiddenUpstreamHeaderName,
isForbiddenCustomHeaderName,
} from "@/shared/constants/upstreamHeaders";
import { MAX_TIMER_TIMEOUT_MS } from "@/shared/utils/runtimeTimeouts";
export const payloadRuleModelSpecSchema = z
.object({
name: z.string().trim().min(1),
protocol: z.string().trim().min(1).optional(),
})
.strict();
export const payloadMutationRuleSchema = z
.object({
models: z.array(payloadRuleModelSpecSchema).min(1),
params: z
.record(z.string().trim().min(1), z.unknown())
.refine((value) => Object.keys(value).length > 0, "params must contain at least one path"),
})
.strict();
export const payloadFilterRuleSchema = z
.object({
models: z.array(payloadRuleModelSpecSchema).min(1),
params: z.array(z.string().trim().min(1)).min(1),
})
.strict();
export const updatePayloadRulesSchema = z
.object({
default: z.array(payloadMutationRuleSchema).optional(),
override: z.array(payloadMutationRuleSchema).optional(),
filter: z.array(payloadFilterRuleSchema).optional(),
defaultRaw: z.array(payloadMutationRuleSchema).optional(),
"default-raw": z.array(payloadMutationRuleSchema).optional(),
})
.strict()
.superRefine((value, ctx) => {
if (
value.default === undefined &&
value.override === undefined &&
value.filter === undefined &&
value.defaultRaw === undefined &&
value["default-raw"] === undefined
) {
ctx.addIssue({
code: z.ZodIssueCode.custom,
message: "No valid fields to update",
path: [],
});
}
});