mirror of
https://github.com/DayuanJiang/next-ai-draw-io.git
synced 2026-10-07 02:07:47 +08:00
dayuan.jiang
eb2fa69f58
The unauthenticated /api/validate-model endpoint should not let callers raise the token budget; the env var / admin setting alone fixes #883. Also import the shared constants in the settings registry instead of duplicating them, and move the setting to the Features group alongside the other validation settings.
45 lines
1.5 KiB
TypeScript
45 lines
1.5 KiB
TypeScript
import { describe, expect, it } from "vitest"
|
|
import {
|
|
DEFAULT_MODEL_VALIDATION_MAX_TOKENS,
|
|
MAX_MODEL_VALIDATION_MAX_TOKENS,
|
|
parseModelValidationMaxTokens,
|
|
resolveModelValidationMaxTokens,
|
|
} from "@/lib/model-validation"
|
|
|
|
describe("model validation token budget", () => {
|
|
it("defaults to a larger reasoning-friendly budget", () => {
|
|
expect(resolveModelValidationMaxTokens(undefined)).toBe(
|
|
DEFAULT_MODEL_VALIDATION_MAX_TOKENS,
|
|
)
|
|
})
|
|
|
|
it("parses valid env values", () => {
|
|
expect(resolveModelValidationMaxTokens("2000")).toBe(2000)
|
|
expect(resolveModelValidationMaxTokens(" 300 ")).toBe(300)
|
|
})
|
|
|
|
it("falls back to the default on invalid values", () => {
|
|
expect(resolveModelValidationMaxTokens("abc")).toBe(
|
|
DEFAULT_MODEL_VALIDATION_MAX_TOKENS,
|
|
)
|
|
expect(resolveModelValidationMaxTokens("")).toBe(
|
|
DEFAULT_MODEL_VALIDATION_MAX_TOKENS,
|
|
)
|
|
})
|
|
|
|
it("rejects invalid values", () => {
|
|
expect(parseModelValidationMaxTokens(0)).toBeNull()
|
|
expect(parseModelValidationMaxTokens(-1)).toBeNull()
|
|
expect(parseModelValidationMaxTokens(1.5)).toBeNull()
|
|
expect(parseModelValidationMaxTokens("abc")).toBeNull()
|
|
})
|
|
|
|
it("caps very large values", () => {
|
|
expect(
|
|
resolveModelValidationMaxTokens(
|
|
MAX_MODEL_VALIDATION_MAX_TOKENS + 1,
|
|
),
|
|
).toBe(MAX_MODEL_VALIDATION_MAX_TOKENS)
|
|
})
|
|
})
|