mirror of
https://github.com/Youzini-afk/ST-Bionic-Memory-Ecology.git
synced 2026-06-14 02:40:45 +08:00
feat(vector): add vector space identity helpers
This commit is contained in:
67
tests/vector-space.mjs
Normal file
67
tests/vector-space.mjs
Normal file
@@ -0,0 +1,67 @@
|
||||
import assert from "node:assert/strict";
|
||||
|
||||
import {
|
||||
createVectorManifest,
|
||||
deriveVectorSpace,
|
||||
isVectorManifestCompatible,
|
||||
normalizeVectorApiUrl,
|
||||
summarizeVectorSpaceChange,
|
||||
} from "../vector/vector-space.js";
|
||||
|
||||
assert.equal(
|
||||
normalizeVectorApiUrl("https://example.com/v1/embeddings?key=secret"),
|
||||
"https://example.com/v1",
|
||||
);
|
||||
|
||||
const baseConfig = {
|
||||
mode: "direct",
|
||||
apiUrl: "https://example.com/v1/embeddings",
|
||||
apiKey: "sk-should-not-appear",
|
||||
model: "text-embedding-3-small",
|
||||
};
|
||||
|
||||
const space1536 = deriveVectorSpace(baseConfig, 1536, { probedAt: 1 });
|
||||
const sameSpace = deriveVectorSpace(
|
||||
{ ...baseConfig, apiKey: "sk-different" },
|
||||
1536,
|
||||
{ probedAt: 2 },
|
||||
);
|
||||
assert.equal(space1536.vectorSpaceId, sameSpace.vectorSpaceId);
|
||||
assert.equal(JSON.stringify(space1536).includes("sk-should-not-appear"), false);
|
||||
assert.equal(space1536.observedDim, 1536);
|
||||
|
||||
const space3072 = deriveVectorSpace(baseConfig, 3072, { probedAt: 3 });
|
||||
assert.notEqual(space1536.vectorSpaceId, space3072.vectorSpaceId);
|
||||
assert.equal(summarizeVectorSpaceChange(space1536, space3072), "dimension-changed");
|
||||
|
||||
const differentModel = deriveVectorSpace(
|
||||
{ ...baseConfig, model: "text-embedding-3-large" },
|
||||
1536,
|
||||
{ probedAt: 4 },
|
||||
);
|
||||
assert.notEqual(space1536.vectorSpaceId, differentModel.vectorSpaceId);
|
||||
assert.equal(summarizeVectorSpaceChange(space1536, differentModel), "model-changed");
|
||||
|
||||
const differentEndpoint = deriveVectorSpace(
|
||||
{ ...baseConfig, apiUrl: "https://other.example.com/v1/embeddings" },
|
||||
1536,
|
||||
{ probedAt: 5 },
|
||||
);
|
||||
assert.notEqual(space1536.vectorSpaceId, differentEndpoint.vectorSpaceId);
|
||||
assert.equal(summarizeVectorSpaceChange(space1536, differentEndpoint), "endpoint-changed");
|
||||
|
||||
const manifest = createVectorManifest({
|
||||
backend: "local",
|
||||
vectorSpace: space1536,
|
||||
status: "clean",
|
||||
nodeCount: 2,
|
||||
embeddedNodeCount: 2,
|
||||
});
|
||||
assert.equal(isVectorManifestCompatible(manifest, space1536), true);
|
||||
assert.equal(isVectorManifestCompatible(manifest, space3072), false);
|
||||
assert.equal(
|
||||
isVectorManifestCompatible({ ...manifest, status: "stale" }, space1536),
|
||||
false,
|
||||
);
|
||||
|
||||
console.log("vector-space tests passed");
|
||||
Reference in New Issue
Block a user