John
2e14873f2d
Track application source and tests; exclude local env, user workspaces, and runtime data via .gitignore. Co-authored-by: Cursor <cursoragent@cursor.com>
71 lines
2.2 KiB
JavaScript
71 lines
2.2 KiB
JavaScript
import fs from 'node:fs';
|
||
import path from 'node:path';
|
||
import { fileURLToPath } from 'node:url';
|
||
import { createDbPool } from '../db.mjs';
|
||
import { createLlmProviderService, testRelayConnection } from '../llm-providers.mjs';
|
||
|
||
const root = path.join(path.dirname(fileURLToPath(import.meta.url)), '..');
|
||
|
||
function loadEnvFile(filePath) {
|
||
if (!fs.existsSync(filePath)) return;
|
||
for (const line of fs.readFileSync(filePath, 'utf8').split('\n')) {
|
||
const trimmed = line.trim();
|
||
if (!trimmed || trimmed.startsWith('#')) continue;
|
||
const eq = trimmed.indexOf('=');
|
||
if (eq < 0) continue;
|
||
const key = trimmed.slice(0, eq).trim();
|
||
const value = trimmed.slice(eq + 1).trim();
|
||
if (!process.env[key]) process.env[key] = value;
|
||
}
|
||
}
|
||
|
||
loadEnvFile(path.join(root, '.env'));
|
||
loadEnvFile(path.join(root, '../tkmind_go/.env.local'));
|
||
|
||
const relayToken = process.env.CUSTOM_RELAY_BUYER_DEEPSEEK_API_KEY?.trim();
|
||
if (!relayToken) {
|
||
console.error('缺少 CUSTOM_RELAY_BUYER_DEEPSEEK_API_KEY（tkmind_go/.env.local）');
|
||
process.exit(1);
|
||
}
|
||
|
||
const pool = createDbPool();
|
||
const svc = createLlmProviderService(pool, {
|
||
apiTarget: process.env.TKMIND_API_TARGET ?? 'https://127.0.0.1:18006',
|
||
apiSecret: process.env.TKMIND_SERVER__SECRET_KEY ?? 'local-dev-secret',
|
||
});
|
||
|
||
const [rows] = await pool.query(
|
||
'SELECT id FROM h5_llm_provider_keys WHERE is_selected = 1 LIMIT 1',
|
||
);
|
||
const id = rows[0]?.id;
|
||
if (!id) {
|
||
console.error('未找到已启用的 LLM 配置');
|
||
process.exit(1);
|
||
}
|
||
|
||
const updated = await svc.updateKey(id, {
|
||
name: 'TKMind Relay DeepSeek',
|
||
apiUrl: 'https://andu.tkmind.cn/relay/buyer/v1/chat/completions',
|
||
apiKey: relayToken,
|
||
models: ['deepseek-chat', 'deepseek-reasoner'],
|
||
defaultModel: 'deepseek-chat',
|
||
relayProvider: 'deepseek',
|
||
});
|
||
|
||
if (!updated.ok) {
|
||
console.error('更新失败:', updated.message);
|
||
process.exit(1);
|
||
}
|
||
|
||
await svc.syncSelectedToGoosed();
|
||
const test = await testRelayConnection({
|
||
apiUrl: 'https://andu.tkmind.cn/relay/buyer/v1/chat/completions',
|
||
apiKey: relayToken,
|
||
model: 'deepseek-chat',
|
||
relayProvider: 'deepseek',
|
||
});
|
||
|
||
console.log('LLM 配置已更新为 andu DeepSeek relay');
|
||
console.log('连通测试:', test.ok ? `OK (${test.latencyMs}ms)` : test.message);
|
||
await pool.end();
|