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(vector): harden runtime embedding sync

Browse Source
This commit is contained in:
opencode
2026-05-15 09:55:00 +00:00
parent e22f9e4e37
commit be38a2c0d2
9 changed files with 652 additions and 143 deletions
Download Patch File Download Diff File Expand all files Collapse all files

100
vector/vector-index.js
View File

@@ -64,6 +64,8 @@ function getConfiguredTimeoutMs(config = {}) {
})();
}
const VECTOR_CONNECTION_PROBE_TEXTS = ["test connection", "runtime batch probe"];
const BACKEND_STATUS_MODEL_SOURCES = {
openai: "openai",
cohere: "cohere",
@@ -1457,15 +1459,69 @@ export async function testVectorConnection(config, chatId = "connection-test") {
return { success: false, dimensions: 0, error: validation.error };
}
if (isDirectVectorConfig(config)) {
if (isDirectVectorConfig(config) || isBackendVectorConfig(config)) {
try {
const vec = await embedText("test connection", config);
if (vec) {
return { success: true, dimensions: vec.length, error: "" };
const vectors = await embedBatch(VECTOR_CONNECTION_PROBE_TEXTS, config, {
isQuery: true,
});
const firstVector = vectors.find((vector) => vector && vector.length > 0);
if (firstVector) {
if (isBackendVectorConfig(config)) {
const response = await fetchWithTimeout(
"/api/vector/query",
{
method: "POST",
headers: getRequestHeaders(),
body: JSON.stringify({
collectionId: buildVectorCollectionId(chatId),
searchText: VECTOR_CONNECTION_PROBE_TEXTS[0],
topK: 1,
threshold: 0,
...buildBackendSourceRequest(config),
}),
},
getConfiguredTimeoutMs(config),
);
const payload = await response.text().catch(() => "");
if (!response.ok) {
return {
success: false,
dimensions: firstVector.length,
error: payload || response.statusText,
batchCapable: true,
vectorStoreCapable: false,
mode: config.mode,
source: config.source || "backend",
};
}
}
return {
success: true,
dimensions: firstVector.length,
error: "",
batchCapable: true,
vectorStoreCapable: isBackendVectorConfig(config) ? true : undefined,
mode: config.mode,
source: config.source || "direct",
};
}
return { success: false, dimensions: 0, error: "API 返回空结果" };
return {
success: false,
dimensions: 0,
error: "批量 Embedding API 返回空结果",
batchCapable: false,
mode: config.mode,
source: config.source || "direct",
};
} catch (error) {
return { success: false, dimensions: 0, error: String(error) };
return {
success: false,
dimensions: 0,
error: String(error),
batchCapable: false,
mode: config.mode,
source: config.source || "direct",
};
}
}
@@ -1480,38 +1536,8 @@ export async function testVectorConnection(config, chatId = "connection-test") {
}
}
try {
const response = await fetchWithTimeout(
"/api/vector/query",
{
method: "POST",
headers: getRequestHeaders(),
body: JSON.stringify({
collectionId: buildVectorCollectionId(chatId),
searchText: "test connection",
topK: 1,
threshold: 0,
...buildBackendSourceRequest(config),
}),
},
getConfiguredTimeoutMs(config),
);
const payload = await response.text().catch(() => "");
if (!response.ok) {
return {
success: false,
dimensions: 0,
error: payload || response.statusText,
};
}
return { success: true, dimensions: 0, error: "" };
} catch (error) {
return { success: false, dimensions: 0, error: String(error) };
}
return { success: false, dimensions: 0, error: "未知向量配置" };
}
export function getVectorIndexStats(graph) {
const state = graph?.vectorIndexState;
if (!state) {