local
140 lines
3.9 KiB
TypeScript
140 lines
3.9 KiB
TypeScript
import { sdk } from '../sdk'
|
|
import { configFile } from '../file-models/config.json'
|
|
|
|
const { InputSpec, Value } = sdk
|
|
|
|
// Operator-facing knob for tier-quota tuning without a code change or
|
|
// redeploy. The schema is { core: TierConfig, pro: TierConfig,
|
|
// max: TierConfig } where TierConfig is
|
|
// { lifetime: number|null, monthly: number|null, geminiCapMonthly: number|null }
|
|
// null means "no cap on this dimension." The relay reads this on every
|
|
// request via configFile's live-reload.
|
|
const inputSpec = InputSpec.of({
|
|
// Core tier knobs.
|
|
core_lifetime: Value.number({
|
|
name: 'Core — Lifetime Credits',
|
|
description:
|
|
'Total credits a Core (unlicensed) install can ever spend across its lifetime. Default 10 (5 served via Gemini + 5 via operator hardware).',
|
|
required: true,
|
|
default: 10,
|
|
min: 0,
|
|
max: 1_000_000,
|
|
integer: true,
|
|
step: 1,
|
|
units: 'credits',
|
|
placeholder: null,
|
|
}),
|
|
core_gemini_cap: Value.number({
|
|
name: 'Core — Gemini Cap (lifetime)',
|
|
description:
|
|
'Within the Core lifetime allowance, how many credits may be served via Gemini (the rest spill to the operator-hardware fallback). Default 5.',
|
|
required: true,
|
|
default: 5,
|
|
min: 0,
|
|
max: 1_000_000,
|
|
integer: true,
|
|
step: 1,
|
|
units: 'credits',
|
|
placeholder: null,
|
|
}),
|
|
// Pro tier knobs.
|
|
pro_monthly: Value.number({
|
|
name: 'Pro — Monthly Credits',
|
|
description:
|
|
'Total credits a Pro user gets each calendar month. Resets on the 1st. Default 50.',
|
|
required: true,
|
|
default: 50,
|
|
min: 0,
|
|
max: 1_000_000,
|
|
integer: true,
|
|
step: 1,
|
|
units: 'credits',
|
|
placeholder: null,
|
|
}),
|
|
pro_gemini_cap: Value.number({
|
|
name: 'Pro — Gemini Cap (monthly)',
|
|
description:
|
|
'Within the Pro monthly allowance, how many credits may be served via Gemini (the rest spill to the operator-hardware fallback). Default 25.',
|
|
required: true,
|
|
default: 25,
|
|
min: 0,
|
|
max: 1_000_000,
|
|
integer: true,
|
|
step: 1,
|
|
units: 'credits',
|
|
placeholder: null,
|
|
}),
|
|
// Max tier knobs.
|
|
max_gemini_cap: Value.number({
|
|
name: 'Max — Gemini Cap (monthly)',
|
|
description:
|
|
'Max-tier users get unlimited total credits but a capped slice goes via Gemini. Default 50.',
|
|
required: true,
|
|
default: 50,
|
|
min: 0,
|
|
max: 1_000_000,
|
|
integer: true,
|
|
step: 1,
|
|
units: 'credits',
|
|
placeholder: null,
|
|
}),
|
|
})
|
|
|
|
export const adjustTierQuotas = sdk.Action.withInput(
|
|
'adjust-tier-quotas',
|
|
|
|
async ({ effects }) => ({
|
|
name: 'Adjust Tier Quotas',
|
|
description:
|
|
'Tune the per-tier monthly credit caps and Gemini exposure without redeploying. Changes apply to the next request — no restart needed.',
|
|
warning: null,
|
|
allowedStatuses: 'any',
|
|
group: null,
|
|
visibility: 'enabled',
|
|
}),
|
|
|
|
inputSpec,
|
|
|
|
async ({ effects }) => {
|
|
const config = await configFile.read().once()
|
|
let parsed: any = {}
|
|
try {
|
|
parsed = JSON.parse(config?.relay_tier_quotas_json || '{}')
|
|
} catch {
|
|
parsed = {}
|
|
}
|
|
return {
|
|
core_lifetime: parsed?.core?.lifetime ?? 10,
|
|
core_gemini_cap: parsed?.core?.geminiCapLifetime ?? 5,
|
|
pro_monthly: parsed?.pro?.monthly ?? 50,
|
|
pro_gemini_cap: parsed?.pro?.geminiCapMonthly ?? 25,
|
|
max_gemini_cap: parsed?.max?.geminiCapMonthly ?? 50,
|
|
}
|
|
},
|
|
|
|
async ({ effects, input }) => {
|
|
const quotas = {
|
|
core: {
|
|
lifetime: input.core_lifetime ?? 10,
|
|
geminiCapLifetime: input.core_gemini_cap ?? 5,
|
|
monthly: null,
|
|
geminiCapMonthly: null,
|
|
},
|
|
pro: {
|
|
lifetime: null,
|
|
monthly: input.pro_monthly ?? 50,
|
|
geminiCapMonthly: input.pro_gemini_cap ?? 25,
|
|
},
|
|
max: {
|
|
lifetime: null,
|
|
monthly: null,
|
|
geminiCapMonthly: input.max_gemini_cap ?? 50,
|
|
},
|
|
}
|
|
await configFile.merge(effects, {
|
|
relay_tier_quotas_json: JSON.stringify(quotas),
|
|
})
|
|
return null
|
|
},
|
|
)
|