82 lines
4.4 KiB
JavaScript
82 lines
4.4 KiB
JavaScript
// ═══════════════════════════════════════════════════════════════════════════
|
|
// Story Summary - Embedder
|
|
// ═══════════════════════════════════════════════════════════════════════════
|
|
|
|
import { embed as sfEmbed } from '../llm/siliconflow.js';
|
|
// ═══════════════════════════════════════════════════════════════════════════
|
|
// 统一 embed 接口
|
|
// ═══════════════════════════════════════════════════════════════════════════
|
|
|
|
export async function embed(texts, config, options = {}) {
|
|
return await sfEmbed(texts, options);
|
|
}
|
|
|
|
// ═══════════════════════════════════════════════════════════════════════════
|
|
// 指纹（简化版）
|
|
// ═══════════════════════════════════════════════════════════════════════════
|
|
|
|
export function getEngineFingerprint(config) {
|
|
const api = config?.embeddingApi || {};
|
|
const provider = String(api.provider || 'siliconflow').toLowerCase();
|
|
const model = String(api.model || 'BAAI/bge-m3').trim() || 'BAAI/bge-m3';
|
|
return `${provider}:${model}:1024`;
|
|
}
|
|
|
|
// ═══════════════════════════════════════════════════════════════════════════
|
|
// 状态检查（简化版）
|
|
// ═══════════════════════════════════════════════════════════════════════════
|
|
|
|
export async function checkLocalModelStatus() {
|
|
// 不再支持本地模型
|
|
return { status: 'not_supported', message: '请使用在线服务' };
|
|
}
|
|
|
|
export function isLocalModelLoaded() {
|
|
return false;
|
|
}
|
|
|
|
export async function downloadLocalModel() {
|
|
throw new Error('本地模型已移除，请使用在线服务');
|
|
}
|
|
|
|
export function cancelDownload() { }
|
|
|
|
export async function deleteLocalModelCache() { }
|
|
|
|
// ═══════════════════════════════════════════════════════════════════════════
|
|
// 在线服务测试
|
|
// ═══════════════════════════════════════════════════════════════════════════
|
|
|
|
export async function testOnlineService(_provider, config = {}) {
|
|
if (!config?.key) {
|
|
throw new Error('请配置 Embedding API Key');
|
|
}
|
|
|
|
try {
|
|
const [vec] = await sfEmbed(['测试连接'], { apiConfig: config });
|
|
return { success: true, dims: vec?.length || 0 };
|
|
} catch (e) {
|
|
throw new Error(`连接失败: ${e.message}`);
|
|
}
|
|
}
|
|
|
|
export async function fetchOnlineModels() {
|
|
return ['BAAI/bge-m3'];
|
|
}
|
|
|
|
// ═══════════════════════════════════════════════════════════════════════════
|
|
// 兼容旧接口
|
|
// ═══════════════════════════════════════════════════════════════════════════
|
|
|
|
export const DEFAULT_LOCAL_MODEL = 'bge-m3';
|
|
|
|
export const LOCAL_MODELS = {};
|
|
|
|
export const ONLINE_PROVIDERS = {
|
|
siliconflow: {
|
|
id: 'siliconflow',
|
|
name: '硅基流动',
|
|
baseUrl: 'https://api.siliconflow.cn/v1',
|
|
},
|
|
};
|