167 lines
6.3 KiB
JavaScript
167 lines
6.3 KiB
JavaScript
"use strict";
|
||
var __importDefault = (this && this.__importDefault) || function (mod) {
|
||
return (mod && mod.__esModule) ? mod : { "default": mod };
|
||
};
|
||
Object.defineProperty(exports, "__esModule", { value: true });
|
||
exports.OpenAIResponsesClient = void 0;
|
||
const fs_1 = __importDefault(require("fs"));
|
||
const path_1 = __importDefault(require("path"));
|
||
const config_1 = require("../config");
|
||
const http_1 = require("../utils/http");
|
||
function extractUsage(raw) {
|
||
const usage = (raw.usage ?? {});
|
||
const input = Number(usage.input_tokens ?? usage.prompt_tokens ?? 0);
|
||
const output = Number(usage.output_tokens ?? usage.completion_tokens ?? 0);
|
||
const total = Number(usage.total_tokens ?? input + output);
|
||
return {
|
||
input_tokens: Number.isFinite(input) ? input : 0,
|
||
output_tokens: Number.isFinite(output) ? output : 0,
|
||
total_tokens: Number.isFinite(total) ? total : 0
|
||
};
|
||
}
|
||
function extractOutputText(raw) {
|
||
if (typeof raw.output_text === "string" && raw.output_text.trim().length > 0) {
|
||
return raw.output_text;
|
||
}
|
||
const output = raw.output;
|
||
if (Array.isArray(output)) {
|
||
for (const item of output) {
|
||
if (!item || typeof item !== "object") {
|
||
continue;
|
||
}
|
||
const content = item.content;
|
||
if (!Array.isArray(content)) {
|
||
continue;
|
||
}
|
||
for (const c of content) {
|
||
if (!c || typeof c !== "object") {
|
||
continue;
|
||
}
|
||
const block = c;
|
||
if (typeof block.text === "string" && block.text.trim()) {
|
||
return block.text;
|
||
}
|
||
}
|
||
}
|
||
}
|
||
const response = raw.response;
|
||
if (response && typeof response === "object") {
|
||
const nested = response;
|
||
if (typeof nested.output_text === "string" && nested.output_text.trim().length > 0) {
|
||
return nested.output_text;
|
||
}
|
||
}
|
||
throw new http_1.ApiError("OPENAI_OUTPUT_PARSE_FAILED", "Не удалось извлечь output_text из Responses API ответа.", 502, raw);
|
||
}
|
||
function loadSchemaForTransport(schemaVersion) {
|
||
const schemaFile = schemaVersion === "v1"
|
||
? "normalized_query_v1.json"
|
||
: schemaVersion === "v2_0_1"
|
||
? "normalized_query_v2_0_1.json"
|
||
: schemaVersion === "v2_0_2"
|
||
? "normalized_query_v2_0_2.json"
|
||
: "normalized_query_v2.json";
|
||
const schemaPath = path_1.default.resolve(config_1.SCHEMAS_DIR, schemaFile);
|
||
return JSON.parse(fs_1.default.readFileSync(schemaPath, "utf-8"));
|
||
}
|
||
class OpenAIResponsesClient {
|
||
async testConnection(config) {
|
||
const payload = {
|
||
model: config.model,
|
||
input: [
|
||
{
|
||
role: "user",
|
||
content: [{ type: "input_text", text: "ping" }]
|
||
}
|
||
],
|
||
max_output_tokens: 16
|
||
};
|
||
await this.post(config, payload);
|
||
return { ok: true, model: config.model };
|
||
}
|
||
async normalize(config, prompt) {
|
||
const schema = loadSchemaForTransport(prompt.schemaVersion);
|
||
const schemaName = prompt.schemaVersion === "v1"
|
||
? "normalized_query_v1"
|
||
: prompt.schemaVersion === "v2_0_1"
|
||
? "normalized_query_v2_0_1"
|
||
: prompt.schemaVersion === "v2_0_2"
|
||
? "normalized_query_v2_0_2"
|
||
: "normalized_query_v2";
|
||
const developerPrompt = prompt.controlledRetryInstruction
|
||
? `${prompt.developerPrompt}\n\n${prompt.controlledRetryInstruction}`
|
||
: prompt.developerPrompt;
|
||
const payload = {
|
||
model: config.model,
|
||
temperature: config.temperature ?? 0,
|
||
max_output_tokens: config.maxOutputTokens ?? 700,
|
||
input: [
|
||
{
|
||
role: "system",
|
||
content: [{ type: "input_text", text: prompt.systemPrompt }]
|
||
},
|
||
{
|
||
role: "developer",
|
||
content: [{ type: "input_text", text: developerPrompt }]
|
||
},
|
||
{
|
||
role: "user",
|
||
content: [
|
||
{
|
||
type: "input_text",
|
||
text: `${prompt.domainPrompt}\n\nПользовательский вопрос:\n${prompt.userQuestion}`
|
||
}
|
||
]
|
||
}
|
||
],
|
||
text: {
|
||
format: {
|
||
type: "json_schema",
|
||
name: schemaName,
|
||
strict: true,
|
||
schema
|
||
}
|
||
}
|
||
};
|
||
const raw = await this.post(config, payload);
|
||
const outputText = extractOutputText(raw);
|
||
return {
|
||
raw,
|
||
outputText,
|
||
usage: extractUsage(raw)
|
||
};
|
||
}
|
||
async post(config, payload) {
|
||
if (!config.apiKey || config.apiKey.trim().length < 10) {
|
||
throw new http_1.ApiError("OPENAI_API_KEY_MISSING", "API ключ OpenAI не задан или слишком короткий.", 400);
|
||
}
|
||
const url = `${(config.baseUrl ?? config_1.DEFAULT_OPENAI_BASE_URL).replace(/\/$/, "")}/responses`;
|
||
const response = await fetch(url, {
|
||
method: "POST",
|
||
headers: {
|
||
Authorization: `Bearer ${config.apiKey}`,
|
||
"Content-Type": "application/json"
|
||
},
|
||
body: JSON.stringify(payload)
|
||
});
|
||
const text = await response.text();
|
||
let data;
|
||
try {
|
||
data = JSON.parse(text);
|
||
}
|
||
catch {
|
||
throw new http_1.ApiError("OPENAI_NON_JSON_RESPONSE", "OpenAI вернул не-JSON ответ.", 502, { status: response.status, body: text.slice(0, 500) });
|
||
}
|
||
if (!response.ok) {
|
||
const errorObj = (data.error ?? {});
|
||
throw new http_1.ApiError("OPENAI_REQUEST_FAILED", String(errorObj.message ?? `OpenAI request failed with status ${response.status}`), response.status, {
|
||
status: response.status,
|
||
type: errorObj.type ?? null,
|
||
code: errorObj.code ?? null
|
||
});
|
||
}
|
||
return data;
|
||
}
|
||
}
|
||
exports.OpenAIResponsesClient = OpenAIResponsesClient;
|