137 lines
4.4 KiB
JavaScript
137 lines
4.4 KiB
JavaScript
/**
|
|
* Copyright (c) 2024-2025 Тарабанов Александр Викторович
|
|
* All rights reserved.
|
|
*
|
|
* This software is proprietary and confidential.
|
|
* Unauthorized copying, modification, or distribution is prohibited.
|
|
*
|
|
* For licensing inquiries: info@hb3-accelerator.com
|
|
* Website: https://hb3-accelerator.com
|
|
* GitHub: https://github.com/HB3-ACCELERATOR
|
|
*/
|
|
|
|
const db = require('../db');
|
|
const OpenAI = require('openai');
|
|
const Anthropic = require('@anthropic-ai/sdk');
|
|
|
|
const TABLE = 'ai_providers_settings';
|
|
|
|
async function getProviderSettings(provider) {
|
|
const { rows } = await db.getQuery()(
|
|
`SELECT * FROM ${TABLE} WHERE provider = $1 LIMIT 1`,
|
|
[provider]
|
|
);
|
|
return rows[0] || null;
|
|
}
|
|
|
|
async function upsertProviderSettings({ provider, api_key, base_url, selected_model, embedding_model }) {
|
|
const { rows } = await db.getQuery()(
|
|
`INSERT INTO ${TABLE} (provider, api_key, base_url, selected_model, embedding_model, updated_at)
|
|
VALUES ($1, $2, $3, $4, $5, NOW())
|
|
ON CONFLICT (provider) DO UPDATE SET
|
|
api_key = EXCLUDED.api_key,
|
|
base_url = EXCLUDED.base_url,
|
|
selected_model = EXCLUDED.selected_model,
|
|
embedding_model = EXCLUDED.embedding_model,
|
|
updated_at = NOW()
|
|
RETURNING *`,
|
|
[provider, api_key, base_url, selected_model, embedding_model]
|
|
);
|
|
return rows[0];
|
|
}
|
|
|
|
async function deleteProviderSettings(provider) {
|
|
await db.getQuery()(
|
|
`DELETE FROM ${TABLE} WHERE provider = $1`,
|
|
[provider]
|
|
);
|
|
}
|
|
|
|
async function getProviderModels(provider, { api_key, base_url } = {}) {
|
|
try {
|
|
if (provider === 'openai') {
|
|
const client = new OpenAI({ apiKey: api_key, baseURL: base_url });
|
|
const res = await client.models.list();
|
|
return res.data ? res.data.map(m => ({ id: m.id, ...m })) : [];
|
|
}
|
|
if (provider === 'anthropic') {
|
|
const client = new Anthropic({ apiKey: api_key, baseURL: base_url });
|
|
const res = await client.models.list();
|
|
return res.data ? res.data.map(m => ({ id: m.id, ...m })) : [];
|
|
}
|
|
if (provider === 'google') {
|
|
const { GoogleGenAI } = await import('@google/genai');
|
|
const ai = new GoogleGenAI({ apiKey: api_key, baseUrl: base_url });
|
|
const pager = await ai.models.list();
|
|
const models = [];
|
|
for await (const model of pager) {
|
|
models.push(model);
|
|
}
|
|
return models;
|
|
}
|
|
if (provider === 'ollama') {
|
|
// Для Ollama — через ai-assistant.js
|
|
return [];
|
|
}
|
|
return [];
|
|
} catch (error) {
|
|
return [];
|
|
}
|
|
}
|
|
|
|
async function verifyProviderKey(provider, { api_key, base_url } = {}) {
|
|
try {
|
|
if (provider === 'openai') {
|
|
const client = new OpenAI({ apiKey: api_key, baseURL: base_url });
|
|
await client.models.list();
|
|
return { success: true };
|
|
}
|
|
if (provider === 'anthropic') {
|
|
const client = new Anthropic({ apiKey: api_key, baseURL: base_url });
|
|
await client.models.list();
|
|
return { success: true };
|
|
}
|
|
if (provider === 'google') {
|
|
const { GoogleGenAI } = await import('@google/genai');
|
|
const ai = new GoogleGenAI({ apiKey: api_key, baseUrl: base_url });
|
|
const pager = await ai.models.list();
|
|
for await (const _ of pager) {
|
|
break;
|
|
}
|
|
return { success: true };
|
|
}
|
|
if (provider === 'ollama') {
|
|
// Для Ollama — всегда true (локальный)
|
|
return { success: true };
|
|
}
|
|
return { success: false, error: 'Unknown provider' };
|
|
} catch (error) {
|
|
return { success: false, error: error.message };
|
|
}
|
|
}
|
|
|
|
async function getAllLLMModels() {
|
|
const { rows } = await db.getQuery()(
|
|
`SELECT provider, selected_model FROM ${TABLE} WHERE selected_model IS NOT NULL AND selected_model <> ''`
|
|
);
|
|
// Возвращаем массив объектов { id, provider }
|
|
return rows.map(r => ({ id: r.selected_model, provider: r.provider }));
|
|
}
|
|
|
|
async function getAllEmbeddingModels() {
|
|
const { rows } = await db.getQuery()(
|
|
`SELECT provider, embedding_model FROM ${TABLE} WHERE embedding_model IS NOT NULL AND embedding_model <> ''`
|
|
);
|
|
// Возвращаем массив объектов { id, provider }
|
|
return rows.map(r => ({ id: r.embedding_model, provider: r.provider }));
|
|
}
|
|
|
|
module.exports = {
|
|
getProviderSettings,
|
|
upsertProviderSettings,
|
|
deleteProviderSettings,
|
|
getProviderModels,
|
|
verifyProviderKey,
|
|
getAllLLMModels,
|
|
getAllEmbeddingModels,
|
|
};
|