77 lines
3.1 KiB
JavaScript
77 lines
3.1 KiB
JavaScript
const DEFAULT_TIMEOUT_MS = 45_000;
|
||
|
||
function completionEndpoint(apiUrl) {
|
||
const value = String(apiUrl || "").replace(/\/+$/, "");
|
||
return /\/chat\/completions$/i.test(value) ? value : `${value}/chat/completions`;
|
||
}
|
||
|
||
function extractJson(text) {
|
||
const value = String(text || "").trim();
|
||
if (!value) throw new Error("model_response_empty");
|
||
try { return JSON.parse(value); } catch {}
|
||
const fenced = value.match(/```(?:json)?\s*([\s\S]*?)```/i);
|
||
if (fenced) return JSON.parse(fenced[1]);
|
||
const first = value.indexOf("{");
|
||
const last = value.lastIndexOf("}");
|
||
if (first >= 0 && last > first) return JSON.parse(value.slice(first, last + 1));
|
||
throw new Error("model_response_not_json");
|
||
}
|
||
|
||
export class DeepSeekJsonClient {
|
||
constructor({
|
||
apiKey = process.env.DEEPSEEK_API_KEY,
|
||
apiUrl = process.env.DEEPSEEK_API_URL || "https://api.deepseek.com/v1",
|
||
model = process.env.DEEPSEEK_MODEL || "deepseek-chat",
|
||
fetchImpl = globalThis.fetch,
|
||
timeoutMs = DEFAULT_TIMEOUT_MS,
|
||
} = {}) {
|
||
if (!apiKey) throw new Error("deepseek_api_key_missing");
|
||
if (!fetchImpl) throw new Error("fetch_unavailable");
|
||
this.apiKey = apiKey;
|
||
this.apiUrl = apiUrl;
|
||
this.model = model;
|
||
this.fetchImpl = fetchImpl;
|
||
this.timeoutMs = timeoutMs;
|
||
}
|
||
|
||
async interpret(input) {
|
||
const controller = new AbortController();
|
||
const timer = setTimeout(() => controller.abort(), this.timeoutMs);
|
||
try {
|
||
const response = await this.fetchImpl(completionEndpoint(this.apiUrl), {
|
||
method: "POST",
|
||
headers: {
|
||
authorization: `Bearer ${this.apiKey}`,
|
||
"content-type": "application/json",
|
||
},
|
||
body: JSON.stringify({
|
||
model: this.model,
|
||
stream: false,
|
||
temperature: 0.2,
|
||
response_format: { type: "json_object" },
|
||
messages: [
|
||
{
|
||
role: "system",
|
||
content: "你是TCS母体大脑的有界语言理解器官。只输出一个JSON对象，不输出隐藏思维过程。不得把候选理解写成事实、不得发明现实权限或冰朔的新意志。",
|
||
},
|
||
{
|
||
role: "user",
|
||
content: JSON.stringify({
|
||
instruction: "理解语言意图与因果关系。精确输出schema,event_id,event_sha256,summary,intent,causal_links,questions,novelty,correction_weight,evidence_weight,next_cognitive_action。causal_links与questions是字符串数组；三个weight是0到100整数；schema必须是guanghu.tcs-mother-interpretation/v1；id和sha必须原样复制。next_cognitive_action只能是COMPARE_WITH_MEMORY、REQUEST_EVIDENCE、REVIEW_CORRECTION、LINK_CROSS_DOMAIN、HOLD_CANDIDATE之一。",
|
||
runtime_input: input,
|
||
}),
|
||
},
|
||
],
|
||
}),
|
||
signal: controller.signal,
|
||
});
|
||
if (!response.ok) throw new Error(`model_http_${response.status}`);
|
||
const payload = await response.json();
|
||
return extractJson(payload?.choices?.[0]?.message?.content);
|
||
} finally {
|
||
clearTimeout(timer);
|
||
}
|
||
}
|
||
}
|
||
|
||
export { completionEndpoint, extractJson };
|