49 lines
No EOL
1.8 KiB
TypeScript
49 lines
No EOL
1.8 KiB
TypeScript
import { NextResponse } from 'next/server';
|
|
import fs from 'fs';
|
|
|
|
const userConfigPath = process.env.USER_CONFIG_PATH!;
|
|
const canChangeKeys = process.env.CAN_CHANGE_KEYS !== 'false';
|
|
|
|
export async function GET() {
|
|
if (!canChangeKeys) {
|
|
return NextResponse.json({
|
|
error: 'You are not allowed to access this resource',
|
|
})
|
|
}
|
|
|
|
if (!fs.existsSync(userConfigPath)) {
|
|
return NextResponse.json({})
|
|
}
|
|
const configData = fs.readFileSync(userConfigPath, 'utf-8')
|
|
return NextResponse.json(JSON.parse(configData))
|
|
}
|
|
|
|
export async function POST(request: Request) {
|
|
if (!canChangeKeys) {
|
|
return NextResponse.json({
|
|
error: 'You are not allowed to access this resource',
|
|
})
|
|
}
|
|
|
|
const userConfig = await request.json()
|
|
|
|
let existingConfig: LLMConfig = {}
|
|
if (fs.existsSync(userConfigPath)) {
|
|
const configData = fs.readFileSync(userConfigPath, 'utf-8')
|
|
existingConfig = JSON.parse(configData)
|
|
}
|
|
const mergedConfig: LLMConfig = {
|
|
LLM: userConfig.LLM || existingConfig.LLM,
|
|
OPENAI_API_KEY: userConfig.OPENAI_API_KEY || existingConfig.OPENAI_API_KEY,
|
|
GOOGLE_API_KEY: userConfig.GOOGLE_API_KEY || existingConfig.GOOGLE_API_KEY,
|
|
OLLAMA_URL: userConfig.OLLAMA_URL || existingConfig.OLLAMA_URL,
|
|
OLLAMA_MODEL: userConfig.OLLAMA_MODEL || existingConfig.OLLAMA_MODEL,
|
|
CUSTOM_LLM_URL: userConfig.CUSTOM_LLM_URL || existingConfig.CUSTOM_LLM_URL,
|
|
CUSTOM_LLM_API_KEY: userConfig.CUSTOM_LLM_API_KEY || existingConfig.CUSTOM_LLM_API_KEY,
|
|
CUSTOM_MODEL: userConfig.CUSTOM_MODEL || existingConfig.CUSTOM_MODEL,
|
|
PEXELS_API_KEY: userConfig.PEXELS_API_KEY || existingConfig.PEXELS_API_KEY,
|
|
USE_CUSTOM_URL: userConfig.USE_CUSTOM_URL === undefined ? existingConfig.USE_CUSTOM_URL : userConfig.USE_CUSTOM_URL,
|
|
}
|
|
fs.writeFileSync(userConfigPath, JSON.stringify(mergedConfig))
|
|
return NextResponse.json(mergedConfig)
|
|
}
|