t0ng7u
d146e45e2f
Add a Bun script to apply and normalize AGPL copyright headers across the default frontend source files. The script keeps headers idempotent, upgrades existing headers to the 2023-2026 QuantumNous range, and is exposed through `bun run copyright` for future maintenance.
108 lines
2.9 KiB
TypeScript
Vendored
108 lines
2.9 KiB
TypeScript
Vendored
/*
|
|
Copyright (C) 2023-2026 QuantumNous
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU Affero General Public License as
|
|
published by the Free Software Foundation, either version 3 of the
|
|
License, or (at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU Affero General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Affero General Public License
|
|
along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
For commercial licensing, please contact support@quantumnous.com
|
|
*/
|
|
import type { AffinityRule } from './types'
|
|
|
|
const CODEX_CLI_HEADER_PASSTHROUGH_HEADERS = [
|
|
'Originator',
|
|
'Session_id',
|
|
'User-Agent',
|
|
'X-Codex-Beta-Features',
|
|
'X-Codex-Turn-Metadata',
|
|
]
|
|
|
|
const CLAUDE_CLI_HEADER_PASSTHROUGH_HEADERS = [
|
|
'X-Stainless-Arch',
|
|
'X-Stainless-Lang',
|
|
'X-Stainless-Os',
|
|
'X-Stainless-Package-Version',
|
|
'X-Stainless-Retry-Count',
|
|
'X-Stainless-Runtime',
|
|
'X-Stainless-Runtime-Version',
|
|
'X-Stainless-Timeout',
|
|
'User-Agent',
|
|
'X-App',
|
|
'Anthropic-Beta',
|
|
'Anthropic-Dangerous-Direct-Browser-Access',
|
|
'Anthropic-Version',
|
|
]
|
|
|
|
function buildPassHeadersTemplate(headers: string[]) {
|
|
return {
|
|
operations: [
|
|
{
|
|
mode: 'pass_headers',
|
|
value: [...headers],
|
|
keep_origin: true,
|
|
},
|
|
],
|
|
}
|
|
}
|
|
|
|
export type RuleTemplate = Omit<AffinityRule, 'id'>
|
|
|
|
export const RULE_TEMPLATES: Record<string, RuleTemplate> = {
|
|
codexCli: {
|
|
name: 'codex cli trace',
|
|
model_regex: ['^gpt-.*$'],
|
|
path_regex: ['/v1/responses'],
|
|
key_sources: [{ type: 'gjson', path: 'prompt_cache_key' }],
|
|
param_override_template: buildPassHeadersTemplate(
|
|
CODEX_CLI_HEADER_PASSTHROUGH_HEADERS
|
|
),
|
|
value_regex: '',
|
|
ttl_seconds: 0,
|
|
skip_retry_on_failure: true,
|
|
include_using_group: true,
|
|
include_model_name: false,
|
|
include_rule_name: true,
|
|
},
|
|
claudeCli: {
|
|
name: 'claude cli trace',
|
|
model_regex: ['^claude-.*$'],
|
|
path_regex: ['/v1/messages'],
|
|
key_sources: [{ type: 'gjson', path: 'metadata.user_id' }],
|
|
param_override_template: buildPassHeadersTemplate(
|
|
CLAUDE_CLI_HEADER_PASSTHROUGH_HEADERS
|
|
),
|
|
value_regex: '',
|
|
ttl_seconds: 0,
|
|
skip_retry_on_failure: true,
|
|
include_using_group: true,
|
|
include_model_name: false,
|
|
include_rule_name: true,
|
|
},
|
|
}
|
|
|
|
export function makeUniqueName(
|
|
existingNames: Set<string>,
|
|
baseName: string
|
|
): string {
|
|
const base = (baseName || '').trim() || 'rule'
|
|
if (!existingNames.has(base)) return base
|
|
for (let i = 2; i < 1000; i++) {
|
|
const n = `${base}-${i}`
|
|
if (!existingNames.has(n)) return n
|
|
}
|
|
return `${base}-${Date.now()}`
|
|
}
|
|
|
|
export function cloneTemplate<T>(template: T): T {
|
|
return JSON.parse(JSON.stringify(template))
|
|
}
|