86 lines
3.0 KiB
JavaScript
86 lines
3.0 KiB
JavaScript
// 伺服器端呼叫已設定的 AI Provider（OpenCode Go / Grok）
|
||
const AI_PROVIDERS = {
|
||
'opencode-go': {
|
||
label: 'OpenCode Go',
|
||
endpoint: 'https://opencode.ai/zen/go/v1/chat/completions',
|
||
keyEnv: 'OPENCODE_GO_API_KEY',
|
||
modelEnv: 'OPENCODE_GO_MODEL',
|
||
mode: 'chat',
|
||
},
|
||
grok: {
|
||
label: 'Grok',
|
||
endpoint: 'https://api.x.ai/v1/responses',
|
||
keyEnv: 'GROK_API_KEY',
|
||
modelEnv: 'GROK_MODEL',
|
||
mode: 'responses',
|
||
},
|
||
};
|
||
|
||
function normalizeAIText(data, mode) {
|
||
if (mode === 'responses') {
|
||
if (data?.output_text) return data.output_text;
|
||
const chunks = [];
|
||
for (const item of data?.output || []) {
|
||
for (const c of item?.content || []) {
|
||
if (typeof c?.text === 'string') chunks.push(c.text);
|
||
else if (typeof c?.content === 'string') chunks.push(c.content);
|
||
}
|
||
}
|
||
return chunks.join('\n').trim();
|
||
}
|
||
return data?.choices?.[0]?.message?.content || data?.choices?.[0]?.text || '';
|
||
}
|
||
|
||
export function getActiveAIConfig() {
|
||
const providerId = String(process.env.AI_ACTIVE_PROVIDER || 'grok').trim();
|
||
const provider = AI_PROVIDERS[providerId];
|
||
if (!provider) return null;
|
||
const apiKey = String(process.env[provider.keyEnv] || '').trim();
|
||
const model = String(process.env[provider.modelEnv] || '').trim();
|
||
if (!apiKey) return null;
|
||
return { providerId, provider, apiKey, model };
|
||
}
|
||
|
||
export function extractJSONObject(text) {
|
||
const raw = String(text || '').trim();
|
||
const fenced = raw.match(/```(?:json)?\s*([\s\S]*?)```/i)?.[1];
|
||
const candidate = fenced || raw;
|
||
const start = candidate.indexOf('{');
|
||
const end = candidate.lastIndexOf('}');
|
||
if (start < 0 || end <= start) return null;
|
||
try {
|
||
return JSON.parse(candidate.slice(start, end + 1));
|
||
} catch {
|
||
return null;
|
||
}
|
||
}
|
||
|
||
export async function callAI({ system, user, temperature = 0.15, timeoutMs = 90000 }) {
|
||
const cfg = getActiveAIConfig();
|
||
if (!cfg) return { ok: false, error: 'no_ai_key', text: null };
|
||
let { model, provider, apiKey, providerId } = cfg;
|
||
if (!model) model = 'grok-3-mini';
|
||
|
||
const ctrl = new AbortController();
|
||
const timer = setTimeout(() => ctrl.abort(), timeoutMs);
|
||
try {
|
||
const body = provider.mode === 'responses'
|
||
? { model, store: false, temperature, input: [{ role: 'system', content: system }, { role: 'user', content: user }] }
|
||
: { model, temperature, messages: [{ role: 'system', content: system }, { role: 'user', content: user }] };
|
||
const r = await fetch(provider.endpoint, {
|
||
method: 'POST',
|
||
headers: { 'Content-Type': 'application/json', Authorization: `Bearer ${apiKey}` },
|
||
body: JSON.stringify(body),
|
||
signal: ctrl.signal,
|
||
});
|
||
const data = await r.json().catch(() => ({}));
|
||
if (!r.ok) {
|
||
return { ok: false, error: data?.error?.message || data?.message || `HTTP ${r.status}`, text: null, providerId };
|
||
}
|
||
return { ok: true, text: normalizeAIText(data, provider.mode), providerId, model };
|
||
} catch (e) {
|
||
return { ok: false, error: String(e?.message || e), text: null, providerId };
|
||
} finally {
|
||
clearTimeout(timer);
|
||
}
|
||
} |