taozuqin/ST-Bionic-Memory-Ecology
1
0
Fork 0
mirror of https://github.com/Youzini-afk/ST-Bionic-Memory-Ecology.git synced 2026-06-13 18:31:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix plugin-only memory settings and UI flow

Browse Source
This commit is contained in:
Youzini-afk
2026-03-24 11:42:10 +08:00
parent 00ae535873
commit 7b68eebb9e
8 changed files with 276 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
embedding.js
View File

@@ -6,6 +6,13 @@
* 调用外部 API 获取文本向量，并提供暴力搜索 cosine 相似度
*/
function normalizeOpenAICompatibleBaseUrl(value) {
return String(value || '')
.trim()
.replace(/\/+(chat\/completions|embeddings)$/i, '')
.replace(/\/+$/, '');
}
/**
* 调用外部 Embedding API
*
@@ -17,19 +24,18 @@
* @returns {Promise<Float64Array|null>} 向量或 null
*/
export async function embedText(text, config) {
if (!text || !config.apiUrl || !config.apiKey || !config.model) {
const apiUrl = normalizeOpenAICompatibleBaseUrl(config?.apiUrl);
if (!text || !apiUrl || !config?.model) {
console.warn('[ST-BME] Embedding 配置不完整，跳过');
return null;
}
try {
const url = `${config.apiUrl.replace(/\/+$/, '')}/embeddings`;
const response = await fetch(url, {
const response = await fetch(`${apiUrl}/embeddings`, {
method: 'POST',
headers: {
'Content-Type': 'application/json',
'Authorization': `Bearer ${config.apiKey}`,
...(config.apiKey ? { Authorization: `Bearer ${config.apiKey}` } : {}),
},
body: JSON.stringify({
model: config.model,
@@ -66,18 +72,17 @@ export async function embedText(text, config) {
* @returns {Promise<(Float64Array|null)[]>}
*/
export async function embedBatch(texts, config) {
if (!texts.length || !config.apiUrl || !config.apiKey || !config.model) {
const apiUrl = normalizeOpenAICompatibleBaseUrl(config?.apiUrl);
if (!texts.length || !apiUrl || !config?.model) {
return texts.map(() => null);
}
try {
const url = `${config.apiUrl.replace(/\/+$/, '')}/embeddings`;
const response = await fetch(url, {
const response = await fetch(`${apiUrl}/embeddings`, {
method: 'POST',
headers: {
'Content-Type': 'application/json',
'Authorization': `Bearer ${config.apiKey}`,
...(config.apiKey ? { Authorization: `Bearer ${config.apiKey}` } : {}),
},
body: JSON.stringify({
model: config.model,