cloison-runtime 0.1.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/LICENSE +21 -0
- package/README.md +313 -0
- package/dist/cli.d.ts +3 -0
- package/dist/cli.d.ts.map +1 -0
- package/dist/cli.js +47 -0
- package/dist/cli.js.map +1 -0
- package/dist/config/index.d.ts +57 -0
- package/dist/config/index.d.ts.map +1 -0
- package/dist/config/index.js +27 -0
- package/dist/config/index.js.map +1 -0
- package/dist/credentials/index.d.ts +4 -0
- package/dist/credentials/index.d.ts.map +1 -0
- package/dist/credentials/index.js +3 -0
- package/dist/credentials/index.js.map +1 -0
- package/dist/credentials/proxy.d.ts +3 -0
- package/dist/credentials/proxy.d.ts.map +1 -0
- package/dist/credentials/proxy.js +11 -0
- package/dist/credentials/proxy.js.map +1 -0
- package/dist/credentials/store.d.ts +7 -0
- package/dist/credentials/store.d.ts.map +1 -0
- package/dist/credentials/store.js +115 -0
- package/dist/credentials/store.js.map +1 -0
- package/dist/credentials/types.d.ts +14 -0
- package/dist/credentials/types.d.ts.map +1 -0
- package/dist/credentials/types.js +2 -0
- package/dist/credentials/types.js.map +1 -0
- package/dist/hooks/index.d.ts +3 -0
- package/dist/hooks/index.d.ts.map +1 -0
- package/dist/hooks/index.js +2 -0
- package/dist/hooks/index.js.map +1 -0
- package/dist/hooks/runner.d.ts +7 -0
- package/dist/hooks/runner.d.ts.map +1 -0
- package/dist/hooks/runner.js +20 -0
- package/dist/hooks/runner.js.map +1 -0
- package/dist/hooks/types.d.ts +39 -0
- package/dist/hooks/types.d.ts.map +1 -0
- package/dist/hooks/types.js +2 -0
- package/dist/hooks/types.js.map +1 -0
- package/dist/index.d.ts +32 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +58 -0
- package/dist/index.js.map +1 -0
- package/dist/infra/env.d.ts +2 -0
- package/dist/infra/env.d.ts.map +1 -0
- package/dist/infra/env.js +6 -0
- package/dist/infra/env.js.map +1 -0
- package/dist/infra/warning-filter.d.ts +8 -0
- package/dist/infra/warning-filter.d.ts.map +1 -0
- package/dist/infra/warning-filter.js +66 -0
- package/dist/infra/warning-filter.js.map +1 -0
- package/dist/logging/subsystem.d.ts +29 -0
- package/dist/logging/subsystem.d.ts.map +1 -0
- package/dist/logging/subsystem.js +322 -0
- package/dist/logging/subsystem.js.map +1 -0
- package/dist/memory/embedding-batch.d.ts +38 -0
- package/dist/memory/embedding-batch.d.ts.map +1 -0
- package/dist/memory/embedding-batch.js +253 -0
- package/dist/memory/embedding-batch.js.map +1 -0
- package/dist/memory/embedding-cache.d.ts +16 -0
- package/dist/memory/embedding-cache.d.ts.map +1 -0
- package/dist/memory/embedding-cache.js +113 -0
- package/dist/memory/embedding-cache.js.map +1 -0
- package/dist/memory/embeddings-debug.d.ts +2 -0
- package/dist/memory/embeddings-debug.d.ts.map +1 -0
- package/dist/memory/embeddings-debug.js +12 -0
- package/dist/memory/embeddings-debug.js.map +1 -0
- package/dist/memory/embeddings.d.ts +17 -0
- package/dist/memory/embeddings.d.ts.map +1 -0
- package/dist/memory/embeddings.js +203 -0
- package/dist/memory/embeddings.js.map +1 -0
- package/dist/memory/file-indexer.d.ts +26 -0
- package/dist/memory/file-indexer.d.ts.map +1 -0
- package/dist/memory/file-indexer.js +260 -0
- package/dist/memory/file-indexer.js.map +1 -0
- package/dist/memory/fs-utils.d.ts +12 -0
- package/dist/memory/fs-utils.d.ts.map +1 -0
- package/dist/memory/fs-utils.js +24 -0
- package/dist/memory/fs-utils.js.map +1 -0
- package/dist/memory/hybrid.d.ts +46 -0
- package/dist/memory/hybrid.d.ts.map +1 -0
- package/dist/memory/hybrid.js +85 -0
- package/dist/memory/hybrid.js.map +1 -0
- package/dist/memory/index.d.ts +17 -0
- package/dist/memory/index.d.ts.map +1 -0
- package/dist/memory/index.js +15 -0
- package/dist/memory/index.js.map +1 -0
- package/dist/memory/internal.d.ts +39 -0
- package/dist/memory/internal.d.ts.map +1 -0
- package/dist/memory/internal.js +292 -0
- package/dist/memory/internal.js.map +1 -0
- package/dist/memory/manager-search.d.ts +61 -0
- package/dist/memory/manager-search.d.ts.map +1 -0
- package/dist/memory/manager-search.js +102 -0
- package/dist/memory/manager-search.js.map +1 -0
- package/dist/memory/mmr.d.ts +63 -0
- package/dist/memory/mmr.d.ts.map +1 -0
- package/dist/memory/mmr.js +165 -0
- package/dist/memory/mmr.js.map +1 -0
- package/dist/memory/query-expansion.d.ts +42 -0
- package/dist/memory/query-expansion.d.ts.map +1 -0
- package/dist/memory/query-expansion.js +776 -0
- package/dist/memory/query-expansion.js.map +1 -0
- package/dist/memory/session-indexer.d.ts +41 -0
- package/dist/memory/session-indexer.d.ts.map +1 -0
- package/dist/memory/session-indexer.js +367 -0
- package/dist/memory/session-indexer.js.map +1 -0
- package/dist/memory/simple-manager.d.ts +29 -0
- package/dist/memory/simple-manager.d.ts.map +1 -0
- package/dist/memory/simple-manager.js +216 -0
- package/dist/memory/simple-manager.js.map +1 -0
- package/dist/memory/sqlite.d.ts +2 -0
- package/dist/memory/sqlite.d.ts.map +1 -0
- package/dist/memory/sqlite.js +16 -0
- package/dist/memory/sqlite.js.map +1 -0
- package/dist/memory/ssrf.d.ts +18 -0
- package/dist/memory/ssrf.d.ts.map +1 -0
- package/dist/memory/ssrf.js +396 -0
- package/dist/memory/ssrf.js.map +1 -0
- package/dist/memory/temporal-decay.d.ts +26 -0
- package/dist/memory/temporal-decay.d.ts.map +1 -0
- package/dist/memory/temporal-decay.js +120 -0
- package/dist/memory/temporal-decay.js.map +1 -0
- package/dist/memory/types.d.ts +95 -0
- package/dist/memory/types.d.ts.map +1 -0
- package/dist/memory/types.js +2 -0
- package/dist/memory/types.js.map +1 -0
- package/dist/package.json +68 -0
- package/dist/platform/index.d.ts +3 -0
- package/dist/platform/index.d.ts.map +1 -0
- package/dist/platform/index.js +2 -0
- package/dist/platform/index.js.map +1 -0
- package/dist/platform/platform.d.ts +3 -0
- package/dist/platform/platform.d.ts.map +1 -0
- package/dist/platform/platform.js +91 -0
- package/dist/platform/platform.js.map +1 -0
- package/dist/platform/types.d.ts +18 -0
- package/dist/platform/types.d.ts.map +1 -0
- package/dist/platform/types.js +2 -0
- package/dist/platform/types.js.map +1 -0
- package/dist/runtime/agent.d.ts +36 -0
- package/dist/runtime/agent.d.ts.map +1 -0
- package/dist/runtime/agent.js +250 -0
- package/dist/runtime/agent.js.map +1 -0
- package/dist/runtime/api-key-rotation.d.ts +26 -0
- package/dist/runtime/api-key-rotation.d.ts.map +1 -0
- package/dist/runtime/api-key-rotation.js +174 -0
- package/dist/runtime/api-key-rotation.js.map +1 -0
- package/dist/runtime/context-guard.d.ts +32 -0
- package/dist/runtime/context-guard.d.ts.map +1 -0
- package/dist/runtime/context-guard.js +61 -0
- package/dist/runtime/context-guard.js.map +1 -0
- package/dist/runtime/failover-error.d.ts +62 -0
- package/dist/runtime/failover-error.d.ts.map +1 -0
- package/dist/runtime/failover-error.js +733 -0
- package/dist/runtime/failover-error.js.map +1 -0
- package/dist/runtime/failover-policy.d.ts +5 -0
- package/dist/runtime/failover-policy.d.ts.map +1 -0
- package/dist/runtime/failover-policy.js +18 -0
- package/dist/runtime/failover-policy.js.map +1 -0
- package/dist/runtime/index.d.ts +13 -0
- package/dist/runtime/index.d.ts.map +1 -0
- package/dist/runtime/index.js +13 -0
- package/dist/runtime/index.js.map +1 -0
- package/dist/runtime/memory-flush.d.ts +24 -0
- package/dist/runtime/memory-flush.d.ts.map +1 -0
- package/dist/runtime/memory-flush.js +64 -0
- package/dist/runtime/memory-flush.js.map +1 -0
- package/dist/runtime/memory-tools.d.ts +14 -0
- package/dist/runtime/memory-tools.d.ts.map +1 -0
- package/dist/runtime/memory-tools.js +58 -0
- package/dist/runtime/memory-tools.js.map +1 -0
- package/dist/runtime/model-fallback.d.ts +56 -0
- package/dist/runtime/model-fallback.d.ts.map +1 -0
- package/dist/runtime/model-fallback.js +301 -0
- package/dist/runtime/model-fallback.js.map +1 -0
- package/dist/runtime/model-fallback.types.d.ts +14 -0
- package/dist/runtime/model-fallback.types.d.ts.map +1 -0
- package/dist/runtime/model-fallback.types.js +3 -0
- package/dist/runtime/model-fallback.types.js.map +1 -0
- package/dist/runtime/retry.d.ts +24 -0
- package/dist/runtime/retry.d.ts.map +1 -0
- package/dist/runtime/retry.js +100 -0
- package/dist/runtime/retry.js.map +1 -0
- package/dist/runtime/session-pruning.d.ts +22 -0
- package/dist/runtime/session-pruning.d.ts.map +1 -0
- package/dist/runtime/session-pruning.js +118 -0
- package/dist/runtime/session-pruning.js.map +1 -0
- package/dist/runtime/stream-adapters.d.ts +11 -0
- package/dist/runtime/stream-adapters.d.ts.map +1 -0
- package/dist/runtime/stream-adapters.js +46 -0
- package/dist/runtime/stream-adapters.js.map +1 -0
- package/dist/runtime/subagent.d.ts +83 -0
- package/dist/runtime/subagent.d.ts.map +1 -0
- package/dist/runtime/subagent.js +190 -0
- package/dist/runtime/subagent.js.map +1 -0
- package/dist/runtime/tool-result-truncation.d.ts +25 -0
- package/dist/runtime/tool-result-truncation.d.ts.map +1 -0
- package/dist/runtime/tool-result-truncation.js +115 -0
- package/dist/runtime/tool-result-truncation.js.map +1 -0
- package/dist/sandbox/cgroup.d.ts +20 -0
- package/dist/sandbox/cgroup.d.ts.map +1 -0
- package/dist/sandbox/cgroup.js +82 -0
- package/dist/sandbox/cgroup.js.map +1 -0
- package/dist/sandbox/index.d.ts +12 -0
- package/dist/sandbox/index.d.ts.map +1 -0
- package/dist/sandbox/index.js +10 -0
- package/dist/sandbox/index.js.map +1 -0
- package/dist/sandbox/ipc.d.ts +26 -0
- package/dist/sandbox/ipc.d.ts.map +1 -0
- package/dist/sandbox/ipc.js +154 -0
- package/dist/sandbox/ipc.js.map +1 -0
- package/dist/sandbox/manager.d.ts +4 -0
- package/dist/sandbox/manager.d.ts.map +1 -0
- package/dist/sandbox/manager.js +251 -0
- package/dist/sandbox/manager.js.map +1 -0
- package/dist/sandbox/namespace.d.ts +12 -0
- package/dist/sandbox/namespace.d.ts.map +1 -0
- package/dist/sandbox/namespace.js +119 -0
- package/dist/sandbox/namespace.js.map +1 -0
- package/dist/sandbox/proxy-tools.d.ts +14 -0
- package/dist/sandbox/proxy-tools.d.ts.map +1 -0
- package/dist/sandbox/proxy-tools.js +63 -0
- package/dist/sandbox/proxy-tools.js.map +1 -0
- package/dist/sandbox/rootfs.d.ts +20 -0
- package/dist/sandbox/rootfs.d.ts.map +1 -0
- package/dist/sandbox/rootfs.js +247 -0
- package/dist/sandbox/rootfs.js.map +1 -0
- package/dist/sandbox/seccomp-apply.d.ts +9 -0
- package/dist/sandbox/seccomp-apply.d.ts.map +1 -0
- package/dist/sandbox/seccomp-apply.js +227 -0
- package/dist/sandbox/seccomp-apply.js.map +1 -0
- package/dist/sandbox/seccomp.d.ts +13 -0
- package/dist/sandbox/seccomp.d.ts.map +1 -0
- package/dist/sandbox/seccomp.js +120 -0
- package/dist/sandbox/seccomp.js.map +1 -0
- package/dist/sandbox/types.d.ts +66 -0
- package/dist/sandbox/types.d.ts.map +1 -0
- package/dist/sandbox/types.js +8 -0
- package/dist/sandbox/types.js.map +1 -0
- package/dist/sandbox/worker.d.ts +15 -0
- package/dist/sandbox/worker.d.ts.map +1 -0
- package/dist/sandbox/worker.js +151 -0
- package/dist/sandbox/worker.js.map +1 -0
- package/dist/sessions/index.d.ts +3 -0
- package/dist/sessions/index.d.ts.map +1 -0
- package/dist/sessions/index.js +3 -0
- package/dist/sessions/index.js.map +1 -0
- package/dist/sessions/store.d.ts +17 -0
- package/dist/sessions/store.d.ts.map +1 -0
- package/dist/sessions/store.js +70 -0
- package/dist/sessions/store.js.map +1 -0
- package/dist/sessions/transcript-events.d.ts +11 -0
- package/dist/sessions/transcript-events.d.ts.map +1 -0
- package/dist/sessions/transcript-events.js +40 -0
- package/dist/sessions/transcript-events.js.map +1 -0
- package/dist/shared/agent-session.d.ts +10 -0
- package/dist/shared/agent-session.d.ts.map +1 -0
- package/dist/shared/agent-session.js +33 -0
- package/dist/shared/agent-session.js.map +1 -0
- package/dist/shared/constants.d.ts +6 -0
- package/dist/shared/constants.d.ts.map +1 -0
- package/dist/shared/constants.js +17 -0
- package/dist/shared/constants.js.map +1 -0
- package/dist/shared/fs.d.ts +7 -0
- package/dist/shared/fs.d.ts.map +1 -0
- package/dist/shared/fs.js +14 -0
- package/dist/shared/fs.js.map +1 -0
- package/dist/shared/index.d.ts +4 -0
- package/dist/shared/index.d.ts.map +1 -0
- package/dist/shared/index.js +4 -0
- package/dist/shared/index.js.map +1 -0
- package/dist/skills/enablement.d.ts +10 -0
- package/dist/skills/enablement.d.ts.map +1 -0
- package/dist/skills/enablement.js +52 -0
- package/dist/skills/enablement.js.map +1 -0
- package/dist/skills/index.d.ts +4 -0
- package/dist/skills/index.d.ts.map +1 -0
- package/dist/skills/index.js +4 -0
- package/dist/skills/index.js.map +1 -0
- package/dist/skills/loader.d.ts +8 -0
- package/dist/skills/loader.d.ts.map +1 -0
- package/dist/skills/loader.js +8 -0
- package/dist/skills/loader.js.map +1 -0
- package/dist/skills/registry.d.ts +19 -0
- package/dist/skills/registry.d.ts.map +1 -0
- package/dist/skills/registry.js +106 -0
- package/dist/skills/registry.js.map +1 -0
- package/dist/utils/boolean.d.ts +6 -0
- package/dist/utils/boolean.d.ts.map +1 -0
- package/dist/utils/boolean.js +28 -0
- package/dist/utils/boolean.js.map +1 -0
- package/dist/utils/run-with-concurrency.d.ts +12 -0
- package/dist/utils/run-with-concurrency.d.ts.map +1 -0
- package/dist/utils/run-with-concurrency.js +40 -0
- package/dist/utils/run-with-concurrency.js.map +1 -0
- package/dist/utils.d.ts +3 -0
- package/dist/utils.d.ts.map +1 -0
- package/dist/utils.js +38 -0
- package/dist/utils.js.map +1 -0
- package/dist/workspace/index.d.ts +3 -0
- package/dist/workspace/index.d.ts.map +1 -0
- package/dist/workspace/index.js +2 -0
- package/dist/workspace/index.js.map +1 -0
- package/dist/workspace/runner.d.ts +19 -0
- package/dist/workspace/runner.d.ts.map +1 -0
- package/dist/workspace/runner.js +491 -0
- package/dist/workspace/runner.js.map +1 -0
- package/dist/workspace/types.d.ts +37 -0
- package/dist/workspace/types.d.ts.map +1 -0
- package/dist/workspace/types.js +2 -0
- package/dist/workspace/types.js.map +1 -0
- package/dist/workspace/workspace.d.ts +12 -0
- package/dist/workspace/workspace.d.ts.map +1 -0
- package/dist/workspace/workspace.js +85 -0
- package/dist/workspace/workspace.js.map +1 -0
- package/package.json +82 -0
|
@@ -0,0 +1,16 @@
|
|
|
1
|
+
import type { DatabaseSync } from "node:sqlite";
|
|
2
|
+
export interface EmbeddingCache {
|
|
3
|
+
get(provider: string, model: string, contentHash: string): number[] | null;
|
|
4
|
+
loadBatch(provider: string, model: string, providerKey: string, hashes: string[]): Map<string, number[]>;
|
|
5
|
+
set(provider: string, model: string, providerKey: string, contentHash: string, embedding: number[], dims: number): void;
|
|
6
|
+
upsertBatch(provider: string, model: string, providerKey: string, entries: Array<{
|
|
7
|
+
hash: string;
|
|
8
|
+
embedding: number[];
|
|
9
|
+
}>): void;
|
|
10
|
+
prune(maxEntries: number): number;
|
|
11
|
+
count(): number;
|
|
12
|
+
}
|
|
13
|
+
export declare function createEmbeddingCacheTable(db: DatabaseSync): void;
|
|
14
|
+
export declare function createEmbeddingCache(db: DatabaseSync): EmbeddingCache;
|
|
15
|
+
export declare function hashContent(content: string): string;
|
|
16
|
+
//# sourceMappingURL=embedding-cache.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"embedding-cache.d.ts","sourceRoot":"","sources":["../../src/memory/embedding-cache.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAShD,MAAM,WAAW,cAAc;IAC7B,GAAG,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,GAAG,MAAM,EAAE,GAAG,IAAI,CAAC;IAC3E,SAAS,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,GAAG,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;IACzG,GAAG,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,MAAM,GAAG,IAAI,CAAC;IACxH,WAAW,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,EAAE,CAAA;KAAE,CAAC,GAAG,IAAI,CAAC;IAC/H,KAAK,CAAC,UAAU,EAAE,MAAM,GAAG,MAAM,CAAC;IAClC,KAAK,IAAI,MAAM,CAAC;CACjB;AAED,wBAAgB,yBAAyB,CAAC,EAAE,EAAE,YAAY,GAAG,IAAI,CAgBhE;AAED,wBAAgB,oBAAoB,CAAC,EAAE,EAAE,YAAY,GAAG,cAAc,CA8FrE;AAED,wBAAgB,WAAW,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,CAEnD"}
|
|
@@ -0,0 +1,113 @@
|
|
|
1
|
+
// Ported from OpenClaw extensions/memory-core/src/memory/manager-embedding-ops.ts
|
|
2
|
+
// Cache operations: loadEmbeddingCache, upsertEmbeddingCache, collectCachedEmbeddings, pruneEmbeddingCacheIfNeeded
|
|
3
|
+
import * as crypto from "node:crypto";
|
|
4
|
+
import { createSubsystemLogger } from "../logging/subsystem.js";
|
|
5
|
+
import { parseEmbedding } from "./internal.js";
|
|
6
|
+
const log = createSubsystemLogger("embedding-cache");
|
|
7
|
+
const EMBEDDING_CACHE_TABLE = "embedding_cache";
|
|
8
|
+
const CACHE_BATCH_SIZE = 400;
|
|
9
|
+
export function createEmbeddingCacheTable(db) {
|
|
10
|
+
db.exec(`
|
|
11
|
+
CREATE TABLE IF NOT EXISTS ${EMBEDDING_CACHE_TABLE} (
|
|
12
|
+
provider TEXT NOT NULL,
|
|
13
|
+
model TEXT NOT NULL,
|
|
14
|
+
provider_key TEXT NOT NULL,
|
|
15
|
+
hash TEXT NOT NULL,
|
|
16
|
+
embedding TEXT NOT NULL,
|
|
17
|
+
dims INTEGER,
|
|
18
|
+
updated_at INTEGER NOT NULL,
|
|
19
|
+
PRIMARY KEY (provider, model, provider_key, hash)
|
|
20
|
+
)
|
|
21
|
+
`);
|
|
22
|
+
db.exec(`CREATE INDEX IF NOT EXISTS idx_embedding_cache_updated_at ON ${EMBEDDING_CACHE_TABLE}(updated_at)`);
|
|
23
|
+
}
|
|
24
|
+
export function createEmbeddingCache(db) {
|
|
25
|
+
createEmbeddingCacheTable(db);
|
|
26
|
+
const stmtGet = db.prepare(`SELECT embedding FROM ${EMBEDDING_CACHE_TABLE} WHERE provider = ? AND model = ? AND provider_key = ? AND hash = ?`);
|
|
27
|
+
const stmtCount = db.prepare(`SELECT COUNT(*) as c FROM ${EMBEDDING_CACHE_TABLE}`);
|
|
28
|
+
const stmtPrune = db.prepare(`DELETE FROM ${EMBEDDING_CACHE_TABLE} WHERE rowid IN (SELECT rowid FROM ${EMBEDDING_CACHE_TABLE} ORDER BY updated_at ASC LIMIT ?)`);
|
|
29
|
+
return {
|
|
30
|
+
// Simple single-item get (backwards compat)
|
|
31
|
+
get(provider, model, contentHash) {
|
|
32
|
+
try {
|
|
33
|
+
const row = stmtGet.get(provider, model, "", contentHash);
|
|
34
|
+
if (!row)
|
|
35
|
+
return null;
|
|
36
|
+
return parseEmbedding(row.embedding);
|
|
37
|
+
}
|
|
38
|
+
catch {
|
|
39
|
+
return null;
|
|
40
|
+
}
|
|
41
|
+
},
|
|
42
|
+
// Batch lookup (OpenClaw's loadEmbeddingCache with 400-item SQL batching)
|
|
43
|
+
loadBatch(provider, model, providerKey, hashes) {
|
|
44
|
+
const out = new Map();
|
|
45
|
+
if (hashes.length === 0)
|
|
46
|
+
return out;
|
|
47
|
+
const unique = [];
|
|
48
|
+
const seen = new Set();
|
|
49
|
+
for (const hash of hashes) {
|
|
50
|
+
if (!hash || seen.has(hash))
|
|
51
|
+
continue;
|
|
52
|
+
seen.add(hash);
|
|
53
|
+
unique.push(hash);
|
|
54
|
+
}
|
|
55
|
+
if (unique.length === 0)
|
|
56
|
+
return out;
|
|
57
|
+
const baseParams = [provider, model, providerKey];
|
|
58
|
+
for (let start = 0; start < unique.length; start += CACHE_BATCH_SIZE) {
|
|
59
|
+
const batch = unique.slice(start, start + CACHE_BATCH_SIZE);
|
|
60
|
+
const placeholders = batch.map(() => "?").join(", ");
|
|
61
|
+
try {
|
|
62
|
+
const rows = db.prepare(`SELECT hash, embedding FROM ${EMBEDDING_CACHE_TABLE} WHERE provider = ? AND model = ? AND provider_key = ? AND hash IN (${placeholders})`).all(...baseParams, ...batch);
|
|
63
|
+
for (const row of rows) {
|
|
64
|
+
out.set(row.hash, parseEmbedding(row.embedding));
|
|
65
|
+
}
|
|
66
|
+
}
|
|
67
|
+
catch {
|
|
68
|
+
// continue with partial results
|
|
69
|
+
}
|
|
70
|
+
}
|
|
71
|
+
return out;
|
|
72
|
+
},
|
|
73
|
+
// Single-item insert
|
|
74
|
+
set(provider, model, providerKey, contentHash, embedding, dims) {
|
|
75
|
+
try {
|
|
76
|
+
db.prepare(`INSERT INTO ${EMBEDDING_CACHE_TABLE} (provider, model, provider_key, hash, embedding, dims, updated_at) VALUES (?, ?, ?, ?, ?, ?, ?) ON CONFLICT(provider, model, provider_key, hash) DO UPDATE SET embedding=excluded.embedding, dims=excluded.dims, updated_at=excluded.updated_at`).run(provider, model, providerKey, contentHash, JSON.stringify(embedding), dims, Date.now());
|
|
77
|
+
}
|
|
78
|
+
catch (err) {
|
|
79
|
+
log.warn("cache write failed", { error: String(err) });
|
|
80
|
+
}
|
|
81
|
+
},
|
|
82
|
+
// Bulk upsert (OpenClaw's upsertEmbeddingCache)
|
|
83
|
+
upsertBatch(provider, model, providerKey, entries) {
|
|
84
|
+
if (entries.length === 0)
|
|
85
|
+
return;
|
|
86
|
+
const now = Date.now();
|
|
87
|
+
const stmt = db.prepare(`INSERT INTO ${EMBEDDING_CACHE_TABLE} (provider, model, provider_key, hash, embedding, dims, updated_at) VALUES (?, ?, ?, ?, ?, ?, ?) ON CONFLICT(provider, model, provider_key, hash) DO UPDATE SET embedding=excluded.embedding, dims=excluded.dims, updated_at=excluded.updated_at`);
|
|
88
|
+
for (const entry of entries) {
|
|
89
|
+
const embedding = entry.embedding ?? [];
|
|
90
|
+
stmt.run(provider, model, providerKey, entry.hash, JSON.stringify(embedding), embedding.length, now);
|
|
91
|
+
}
|
|
92
|
+
},
|
|
93
|
+
// LRU pruning (OpenClaw's pruneEmbeddingCacheIfNeeded)
|
|
94
|
+
prune(maxEntries) {
|
|
95
|
+
if (maxEntries <= 0)
|
|
96
|
+
return 0;
|
|
97
|
+
const row = stmtCount.get();
|
|
98
|
+
const count = row?.c ?? 0;
|
|
99
|
+
if (count <= maxEntries)
|
|
100
|
+
return 0;
|
|
101
|
+
const excess = count - maxEntries;
|
|
102
|
+
stmtPrune.run(excess);
|
|
103
|
+
return excess;
|
|
104
|
+
},
|
|
105
|
+
count() {
|
|
106
|
+
return stmtCount.get().c;
|
|
107
|
+
},
|
|
108
|
+
};
|
|
109
|
+
}
|
|
110
|
+
export function hashContent(content) {
|
|
111
|
+
return crypto.createHash("sha256").update(content).digest("hex");
|
|
112
|
+
}
|
|
113
|
+
//# sourceMappingURL=embedding-cache.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"embedding-cache.js","sourceRoot":"","sources":["../../src/memory/embedding-cache.ts"],"names":[],"mappings":"AAAA,kFAAkF;AAClF,mHAAmH;AAEnH,OAAO,KAAK,MAAM,MAAM,aAAa,CAAC;AAEtC,OAAO,EAAE,qBAAqB,EAAE,MAAM,yBAAyB,CAAC;AAChE,OAAO,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAE/C,MAAM,GAAG,GAAG,qBAAqB,CAAC,iBAAiB,CAAC,CAAC;AAErD,MAAM,qBAAqB,GAAG,iBAAiB,CAAC;AAChD,MAAM,gBAAgB,GAAG,GAAG,CAAC;AAW7B,MAAM,UAAU,yBAAyB,CAAC,EAAgB;IACxD,EAAE,CAAC,IAAI,CAAC;iCACuB,qBAAqB;;;;;;;;;;GAUnD,CAAC,CAAC;IACH,EAAE,CAAC,IAAI,CACL,gEAAgE,qBAAqB,cAAc,CACpG,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,EAAgB;IACnD,yBAAyB,CAAC,EAAE,CAAC,CAAC;IAE9B,MAAM,OAAO,GAAG,EAAE,CAAC,OAAO,CACxB,yBAAyB,qBAAqB,qEAAqE,CACpH,CAAC;IACF,MAAM,SAAS,GAAG,EAAE,CAAC,OAAO,CAAC,6BAA6B,qBAAqB,EAAE,CAAC,CAAC;IACnF,MAAM,SAAS,GAAG,EAAE,CAAC,OAAO,CAC1B,eAAe,qBAAqB,sCAAsC,qBAAqB,mCAAmC,CACnI,CAAC;IAEF,OAAO;QACL,4CAA4C;QAC5C,GAAG,CAAC,QAAQ,EAAE,KAAK,EAAE,WAAW;YAC9B,IAAI,CAAC;gBACH,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,KAAK,EAAE,EAAE,EAAE,WAAW,CAAsC,CAAC;gBAC/F,IAAI,CAAC,GAAG;oBAAE,OAAO,IAAI,CAAC;gBACtB,OAAO,cAAc,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YACvC,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO,IAAI,CAAC;YACd,CAAC;QACH,CAAC;QAED,0EAA0E;QAC1E,SAAS,CAAC,QAAQ,EAAE,KAAK,EAAE,WAAW,EAAE,MAAM;YAC5C,MAAM,GAAG,GAAG,IAAI,GAAG,EAAoB,CAAC;YACxC,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;gBAAE,OAAO,GAAG,CAAC;YAEpC,MAAM,MAAM,GAAa,EAAE,CAAC;YAC5B,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;YAC/B,KAAK,MAAM,IAAI,IAAI,MAAM,EAAE,CAAC;gBAC1B,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC;oBAAE,SAAS;gBACtC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;gBACf,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACpB,CAAC;YACD,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;gBAAE,OAAO,GAAG,CAAC;YAEpC,MAAM,UAAU,GAAG,CAAC,QAAQ,EAAE,KAAK,EAAE,WAAW,CAAC,CAAC;YAClD,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,MAAM,CAAC,MAAM,EAAE,KAAK,IAAI,gBAAgB,EAAE,CAAC;gBACrE,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,GAAG,gBAAgB,CAAC,CAAC;gBAC5D,MAAM,YAAY,GAAG,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACrD,IAAI,CAAC;oBACH,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CACrB,+BAA+B,qBAAqB,uEAAuE,YAAY,GAAG,CAC3I,CAAC,GAAG,CAAC,GAAG,UAAU,EAAE,GAAG,KAAK,CAA+C,CAAC;oBAC7E,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;wBACvB,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,cAAc,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC;oBACnD,CAAC;gBACH,CAAC;gBAAC,MAAM,CAAC;oBACP,gCAAgC;gBAClC,CAAC;YACH,CAAC;YACD,OAAO,GAAG,CAAC;QACb,CAAC;QAED,qBAAqB;QACrB,GAAG,CAAC,QAAQ,EAAE,KAAK,EAAE,WAAW,EAAE,WAAW,EAAE,SAAS,EAAE,IAAI;YAC5D,IAAI,CAAC;gBACH,EAAE,CAAC,OAAO,CACR,eAAe,qBAAqB,kPAAkP,CACvR,CAAC,GAAG,CAAC,QAAQ,EAAE,KAAK,EAAE,WAAW,EAAE,WAAW,EAAE,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;YAChG,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,GAAG,CAAC,IAAI,CAAC,oBAAoB,EAAE,EAAE,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YACzD,CAAC;QACH,CAAC;QAED,gDAAgD;QAChD,WAAW,CAAC,QAAQ,EAAE,KAAK,EAAE,WAAW,EAAE,OAAO;YAC/C,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;gBAAE,OAAO;YACjC,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YACvB,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CACrB,eAAe,qBAAqB,kPAAkP,CACvR,CAAC;YACF,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;gBAC5B,MAAM,SAAS,GAAG,KAAK,CAAC,SAAS,IAAI,EAAE,CAAC;gBACxC,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,KAAK,EAAE,WAAW,EAAE,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,EAAE,SAAS,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;YACvG,CAAC;QACH,CAAC;QAED,uDAAuD;QACvD,KAAK,CAAC,UAAU;YACd,IAAI,UAAU,IAAI,CAAC;gBAAE,OAAO,CAAC,CAAC;YAC9B,MAAM,GAAG,GAAG,SAAS,CAAC,GAAG,EAA+B,CAAC;YACzD,MAAM,KAAK,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;YAC1B,IAAI,KAAK,IAAI,UAAU;gBAAE,OAAO,CAAC,CAAC;YAClC,MAAM,MAAM,GAAG,KAAK,GAAG,UAAU,CAAC;YAClC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YACtB,OAAO,MAAM,CAAC;QAChB,CAAC;QAED,KAAK;YACH,OAAQ,SAAS,CAAC,GAAG,EAAoB,CAAC,CAAC,CAAC;QAC9C,CAAC;KACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,WAAW,CAAC,OAAe;IACzC,OAAO,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AACnE,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"embeddings-debug.d.ts","sourceRoot":"","sources":["../../src/memory/embeddings-debug.ts"],"names":[],"mappings":"AAMA,wBAAgB,kBAAkB,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAMxF"}
|
|
@@ -0,0 +1,12 @@
|
|
|
1
|
+
import { isTruthyEnvValue } from "../infra/env.js";
|
|
2
|
+
import { createSubsystemLogger } from "../logging/subsystem.js";
|
|
3
|
+
const debugEmbeddings = isTruthyEnvValue(process.env.CLOISON_DEBUG_MEMORY_EMBEDDINGS);
|
|
4
|
+
const log = createSubsystemLogger("memory/embeddings");
|
|
5
|
+
export function debugEmbeddingsLog(message, meta) {
|
|
6
|
+
if (!debugEmbeddings) {
|
|
7
|
+
return;
|
|
8
|
+
}
|
|
9
|
+
const suffix = meta ? ` ${JSON.stringify(meta)}` : "";
|
|
10
|
+
log.raw(`${message}${suffix}`);
|
|
11
|
+
}
|
|
12
|
+
//# sourceMappingURL=embeddings-debug.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"embeddings-debug.js","sourceRoot":"","sources":["../../src/memory/embeddings-debug.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,iBAAiB,CAAC;AACnD,OAAO,EAAE,qBAAqB,EAAE,MAAM,yBAAyB,CAAC;AAEhE,MAAM,eAAe,GAAG,gBAAgB,CAAC,OAAO,CAAC,GAAG,CAAC,+BAA+B,CAAC,CAAC;AACtF,MAAM,GAAG,GAAG,qBAAqB,CAAC,mBAAmB,CAAC,CAAC;AAEvD,MAAM,UAAU,kBAAkB,CAAC,OAAe,EAAE,IAA8B;IAChF,IAAI,CAAC,eAAe,EAAE,CAAC;QACrB,OAAO;IACT,CAAC;IACD,MAAM,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;IACtD,GAAG,CAAC,GAAG,CAAC,GAAG,OAAO,GAAG,MAAM,EAAE,CAAC,CAAC;AACjC,CAAC"}
|
|
@@ -0,0 +1,17 @@
|
|
|
1
|
+
export type EmbeddingProvider = {
|
|
2
|
+
id: string;
|
|
3
|
+
model: string;
|
|
4
|
+
maxInputTokens?: number;
|
|
5
|
+
embedQuery: (text: string) => Promise<number[]>;
|
|
6
|
+
embedBatch: (texts: string[]) => Promise<number[][]>;
|
|
7
|
+
};
|
|
8
|
+
export type EmbeddingProviderId = "openai" | "gemini" | "voyage" | "mistral" | "ollama";
|
|
9
|
+
export type CreateEmbeddingProviderOptions = {
|
|
10
|
+
provider: EmbeddingProviderId;
|
|
11
|
+
apiKey?: string;
|
|
12
|
+
model?: string;
|
|
13
|
+
baseUrl?: string;
|
|
14
|
+
enableSsrf?: boolean;
|
|
15
|
+
};
|
|
16
|
+
export declare function createEmbeddingProvider(options: CreateEmbeddingProviderOptions): EmbeddingProvider;
|
|
17
|
+
//# sourceMappingURL=embeddings.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"embeddings.d.ts","sourceRoot":"","sources":["../../src/memory/embeddings.ts"],"names":[],"mappings":"AAEA,MAAM,MAAM,iBAAiB,GAAG;IAC9B,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,UAAU,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;IAChD,UAAU,EAAE,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;CACtD,CAAC;AAEF,MAAM,MAAM,mBAAmB,GAAG,QAAQ,GAAG,QAAQ,GAAG,QAAQ,GAAG,SAAS,GAAG,QAAQ,CAAC;AA4PxF,MAAM,MAAM,8BAA8B,GAAG;IAC3C,QAAQ,EAAE,mBAAmB,CAAC;IAC9B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,UAAU,CAAC,EAAE,OAAO,CAAC;CACtB,CAAC;AAEF,wBAAgB,uBAAuB,CACrC,OAAO,EAAE,8BAA8B,GACtC,iBAAiB,CAqBnB"}
|
|
@@ -0,0 +1,203 @@
|
|
|
1
|
+
import { buildBaseUrlPolicy, fetchWithSsrfGuard } from "./ssrf.js";
|
|
2
|
+
function sanitizeErrorBody(body) {
|
|
3
|
+
const truncated = body.slice(0, 500);
|
|
4
|
+
return truncated
|
|
5
|
+
.replace(/(?:sk-|sk-ant-|key-|pa-|AIza|Bearer\s+)[A-Za-z0-9_-]{10,}/g, "[REDACTED]")
|
|
6
|
+
.replace(/[?&]key=[^&\s"']{10,}/g, "?key=[REDACTED]")
|
|
7
|
+
.replace(/"(api_key|apiKey|secret|token|password)"\s*:\s*"[^"]{6,}"/gi, '"$1":"[REDACTED]"');
|
|
8
|
+
}
|
|
9
|
+
const APPROX_CHARS_PER_TOKEN = 4;
|
|
10
|
+
function validateInputLength(text, maxTokens) {
|
|
11
|
+
if (maxTokens && text.length > maxTokens * APPROX_CHARS_PER_TOKEN) {
|
|
12
|
+
throw new Error(`Input text (~${Math.ceil(text.length / APPROX_CHARS_PER_TOKEN)} tokens) exceeds provider limit of ${maxTokens} tokens`);
|
|
13
|
+
}
|
|
14
|
+
}
|
|
15
|
+
function sanitizeAndNormalizeEmbedding(vec) {
|
|
16
|
+
const sanitized = vec.map((value) => (Number.isFinite(value) ? value : 0));
|
|
17
|
+
const magnitude = Math.sqrt(sanitized.reduce((sum, value) => sum + value * value, 0));
|
|
18
|
+
if (magnitude < 1e-10)
|
|
19
|
+
return sanitized;
|
|
20
|
+
return sanitized.map((value) => value / magnitude);
|
|
21
|
+
}
|
|
22
|
+
async function ssrfFetch(url, init, ssrfPolicy) {
|
|
23
|
+
if (ssrfPolicy) {
|
|
24
|
+
return fetchWithSsrfGuard(url, init, ssrfPolicy);
|
|
25
|
+
}
|
|
26
|
+
return fetch(url, init);
|
|
27
|
+
}
|
|
28
|
+
function createBearerEmbeddingProvider(config) {
|
|
29
|
+
const { id, apiKey, model, baseUrl, maxInputTokens, ssrfPolicy } = config;
|
|
30
|
+
async function call(input) {
|
|
31
|
+
const res = await ssrfFetch(`${baseUrl}/embeddings`, {
|
|
32
|
+
method: "POST",
|
|
33
|
+
headers: {
|
|
34
|
+
"Content-Type": "application/json",
|
|
35
|
+
Authorization: `Bearer ${apiKey}`,
|
|
36
|
+
},
|
|
37
|
+
body: JSON.stringify({ input, model }),
|
|
38
|
+
}, ssrfPolicy);
|
|
39
|
+
if (!res.ok) {
|
|
40
|
+
throw new Error(`${id} embeddings failed (${res.status}): ${sanitizeErrorBody(await res.text())}`);
|
|
41
|
+
}
|
|
42
|
+
const body = (await res.json());
|
|
43
|
+
return body.data.map((d) => sanitizeAndNormalizeEmbedding(d.embedding));
|
|
44
|
+
}
|
|
45
|
+
return {
|
|
46
|
+
id,
|
|
47
|
+
model,
|
|
48
|
+
maxInputTokens,
|
|
49
|
+
embedQuery: async (text) => {
|
|
50
|
+
validateInputLength(text, maxInputTokens);
|
|
51
|
+
const results = await call([text]);
|
|
52
|
+
if (!results || results.length === 0) {
|
|
53
|
+
throw new Error(`${id} returned empty embedding response`);
|
|
54
|
+
}
|
|
55
|
+
return results[0];
|
|
56
|
+
},
|
|
57
|
+
embedBatch: async (texts) => {
|
|
58
|
+
for (const t of texts)
|
|
59
|
+
validateInputLength(t, maxInputTokens);
|
|
60
|
+
return call(texts);
|
|
61
|
+
},
|
|
62
|
+
};
|
|
63
|
+
}
|
|
64
|
+
// --- OpenAI ---
|
|
65
|
+
const OPENAI_MAX_INPUT_TOKENS = {
|
|
66
|
+
"text-embedding-3-small": 8192,
|
|
67
|
+
"text-embedding-3-large": 8192,
|
|
68
|
+
"text-embedding-ada-002": 8191,
|
|
69
|
+
};
|
|
70
|
+
function createOpenAiProvider(params) {
|
|
71
|
+
const model = params.model ?? "text-embedding-3-small";
|
|
72
|
+
const baseUrl = (params.baseUrl ?? "https://api.openai.com/v1").replace(/\/+$/, "");
|
|
73
|
+
return createBearerEmbeddingProvider({
|
|
74
|
+
id: "openai",
|
|
75
|
+
apiKey: params.apiKey,
|
|
76
|
+
model,
|
|
77
|
+
baseUrl,
|
|
78
|
+
maxInputTokens: OPENAI_MAX_INPUT_TOKENS[model],
|
|
79
|
+
ssrfPolicy: params.enableSsrf !== false ? buildBaseUrlPolicy(baseUrl) ?? undefined : undefined,
|
|
80
|
+
});
|
|
81
|
+
}
|
|
82
|
+
// --- Gemini (different API shape — not Bearer-based) ---
|
|
83
|
+
const GEMINI_MAX_INPUT_TOKENS = {
|
|
84
|
+
"gemini-embedding-001": 2048,
|
|
85
|
+
"text-embedding-004": 2048,
|
|
86
|
+
};
|
|
87
|
+
function createGeminiProvider(params) {
|
|
88
|
+
const model = params.model ?? "gemini-embedding-001";
|
|
89
|
+
const baseUrl = (params.baseUrl ?? "https://generativelanguage.googleapis.com/v1beta").replace(/\/+$/, "");
|
|
90
|
+
const ssrfPolicy = params.enableSsrf !== false ? buildBaseUrlPolicy(baseUrl) ?? undefined : undefined;
|
|
91
|
+
async function call(texts) {
|
|
92
|
+
const requests = texts.map((text) => ({
|
|
93
|
+
model: `models/${model}`,
|
|
94
|
+
content: { parts: [{ text }] },
|
|
95
|
+
}));
|
|
96
|
+
const res = await ssrfFetch(`${baseUrl}/models/${model}:batchEmbedContents`, {
|
|
97
|
+
method: "POST",
|
|
98
|
+
headers: {
|
|
99
|
+
"Content-Type": "application/json",
|
|
100
|
+
"x-goog-api-key": params.apiKey,
|
|
101
|
+
},
|
|
102
|
+
body: JSON.stringify({ requests }),
|
|
103
|
+
}, ssrfPolicy);
|
|
104
|
+
if (!res.ok)
|
|
105
|
+
throw new Error(`gemini embeddings failed (${res.status}): ${sanitizeErrorBody(await res.text())}`);
|
|
106
|
+
const body = (await res.json());
|
|
107
|
+
return body.embeddings.map((e) => sanitizeAndNormalizeEmbedding(e.values));
|
|
108
|
+
}
|
|
109
|
+
const maxTokens = GEMINI_MAX_INPUT_TOKENS[model];
|
|
110
|
+
return {
|
|
111
|
+
id: "gemini",
|
|
112
|
+
model,
|
|
113
|
+
maxInputTokens: maxTokens,
|
|
114
|
+
embedQuery: async (text) => {
|
|
115
|
+
validateInputLength(text, maxTokens);
|
|
116
|
+
const results = await call([text]);
|
|
117
|
+
if (!results || results.length === 0)
|
|
118
|
+
throw new Error("gemini returned empty embedding response");
|
|
119
|
+
return results[0];
|
|
120
|
+
},
|
|
121
|
+
embedBatch: async (texts) => {
|
|
122
|
+
for (const t of texts)
|
|
123
|
+
validateInputLength(t, maxTokens);
|
|
124
|
+
return call(texts);
|
|
125
|
+
},
|
|
126
|
+
};
|
|
127
|
+
}
|
|
128
|
+
// --- Voyage ---
|
|
129
|
+
function createVoyageProvider(params) {
|
|
130
|
+
const baseUrl = (params.baseUrl ?? "https://api.voyageai.com/v1").replace(/\/+$/, "");
|
|
131
|
+
return createBearerEmbeddingProvider({
|
|
132
|
+
id: "voyage",
|
|
133
|
+
apiKey: params.apiKey,
|
|
134
|
+
model: params.model ?? "voyage-3-lite",
|
|
135
|
+
baseUrl,
|
|
136
|
+
ssrfPolicy: params.enableSsrf !== false ? buildBaseUrlPolicy(baseUrl) ?? undefined : undefined,
|
|
137
|
+
});
|
|
138
|
+
}
|
|
139
|
+
// --- Mistral ---
|
|
140
|
+
function createMistralProvider(params) {
|
|
141
|
+
const baseUrl = (params.baseUrl ?? "https://api.mistral.ai/v1").replace(/\/+$/, "");
|
|
142
|
+
return createBearerEmbeddingProvider({
|
|
143
|
+
id: "mistral",
|
|
144
|
+
apiKey: params.apiKey,
|
|
145
|
+
model: params.model ?? "mistral-embed",
|
|
146
|
+
baseUrl,
|
|
147
|
+
ssrfPolicy: params.enableSsrf !== false ? buildBaseUrlPolicy(baseUrl) ?? undefined : undefined,
|
|
148
|
+
});
|
|
149
|
+
}
|
|
150
|
+
// --- Ollama (different API shape — no auth, different response) ---
|
|
151
|
+
function createOllamaProvider(params) {
|
|
152
|
+
const model = params.model ?? "nomic-embed-text";
|
|
153
|
+
const baseUrl = (params.baseUrl ?? "http://localhost:11434").replace(/\/+$/, "");
|
|
154
|
+
const ssrfPolicy = params.enableSsrf !== false ? buildBaseUrlPolicy(baseUrl) ?? undefined : undefined;
|
|
155
|
+
async function call(input) {
|
|
156
|
+
const res = await ssrfFetch(`${baseUrl}/api/embed`, {
|
|
157
|
+
method: "POST",
|
|
158
|
+
headers: { "Content-Type": "application/json" },
|
|
159
|
+
body: JSON.stringify({ model, input }),
|
|
160
|
+
}, ssrfPolicy);
|
|
161
|
+
if (!res.ok)
|
|
162
|
+
throw new Error(`ollama embeddings failed (${res.status}): ${sanitizeErrorBody(await res.text())}`);
|
|
163
|
+
const body = (await res.json());
|
|
164
|
+
return body.embeddings.map((v) => sanitizeAndNormalizeEmbedding(v));
|
|
165
|
+
}
|
|
166
|
+
return {
|
|
167
|
+
id: "ollama",
|
|
168
|
+
model,
|
|
169
|
+
embedQuery: async (text) => {
|
|
170
|
+
const results = await call([text]);
|
|
171
|
+
if (!results || results.length === 0)
|
|
172
|
+
throw new Error("ollama returned empty embedding response");
|
|
173
|
+
return results[0];
|
|
174
|
+
},
|
|
175
|
+
embedBatch: call,
|
|
176
|
+
};
|
|
177
|
+
}
|
|
178
|
+
export function createEmbeddingProvider(options) {
|
|
179
|
+
const { provider, apiKey, model, baseUrl, enableSsrf } = options;
|
|
180
|
+
switch (provider) {
|
|
181
|
+
case "openai":
|
|
182
|
+
if (!apiKey)
|
|
183
|
+
throw new Error("OpenAI embedding provider requires an API key");
|
|
184
|
+
return createOpenAiProvider({ apiKey, model, baseUrl, enableSsrf });
|
|
185
|
+
case "gemini":
|
|
186
|
+
if (!apiKey)
|
|
187
|
+
throw new Error("Gemini embedding provider requires an API key");
|
|
188
|
+
return createGeminiProvider({ apiKey, model, baseUrl, enableSsrf });
|
|
189
|
+
case "voyage":
|
|
190
|
+
if (!apiKey)
|
|
191
|
+
throw new Error("Voyage embedding provider requires an API key");
|
|
192
|
+
return createVoyageProvider({ apiKey, model, baseUrl, enableSsrf });
|
|
193
|
+
case "mistral":
|
|
194
|
+
if (!apiKey)
|
|
195
|
+
throw new Error("Mistral embedding provider requires an API key");
|
|
196
|
+
return createMistralProvider({ apiKey, model, baseUrl, enableSsrf });
|
|
197
|
+
case "ollama":
|
|
198
|
+
return createOllamaProvider({ model, baseUrl, enableSsrf });
|
|
199
|
+
default:
|
|
200
|
+
throw new Error(`Unknown embedding provider: ${provider}`);
|
|
201
|
+
}
|
|
202
|
+
}
|
|
203
|
+
//# sourceMappingURL=embeddings.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"embeddings.js","sourceRoot":"","sources":["../../src/memory/embeddings.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,kBAAkB,EAAmB,MAAM,WAAW,CAAC;AAYpF,SAAS,iBAAiB,CAAC,IAAY;IACrC,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;IACrC,OAAO,SAAS;SACb,OAAO,CAAC,4DAA4D,EAAE,YAAY,CAAC;SACnF,OAAO,CAAC,wBAAwB,EAAE,iBAAiB,CAAC;SACpD,OAAO,CAAC,6DAA6D,EAAE,mBAAmB,CAAC,CAAC;AACjG,CAAC;AAED,MAAM,sBAAsB,GAAG,CAAC,CAAC;AAEjC,SAAS,mBAAmB,CAAC,IAAY,EAAE,SAAkB;IAC3D,IAAI,SAAS,IAAI,IAAI,CAAC,MAAM,GAAG,SAAS,GAAG,sBAAsB,EAAE,CAAC;QAClE,MAAM,IAAI,KAAK,CACb,gBAAgB,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,sBAAsB,CAAC,sCAAsC,SAAS,SAAS,CACxH,CAAC;IACJ,CAAC;AACH,CAAC;AAED,SAAS,6BAA6B,CAAC,GAAa;IAClD,MAAM,SAAS,GAAG,GAAG,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAC3E,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,GAAG,GAAG,KAAK,GAAG,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC;IACtF,IAAI,SAAS,GAAG,KAAK;QAAE,OAAO,SAAS,CAAC;IACxC,OAAO,SAAS,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,GAAG,SAAS,CAAC,CAAC;AACrD,CAAC;AAED,KAAK,UAAU,SAAS,CACtB,GAAW,EACX,IAAiB,EACjB,UAAuB;IAEvB,IAAI,UAAU,EAAE,CAAC;QACf,OAAO,kBAAkB,CAAC,GAAG,EAAE,IAAI,EAAE,UAAU,CAAC,CAAC;IACnD,CAAC;IACD,OAAO,KAAK,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;AAC1B,CAAC;AAWD,SAAS,6BAA6B,CAAC,MAA4B;IACjE,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,GAAG,MAAM,CAAC;IAE1E,KAAK,UAAU,IAAI,CAAC,KAAe;QACjC,MAAM,GAAG,GAAG,MAAM,SAAS,CACzB,GAAG,OAAO,aAAa,EACvB;YACE,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,aAAa,EAAE,UAAU,MAAM,EAAE;aAClC;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;SACvC,EACD,UAAU,CACX,CAAC;QACF,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,GAAG,EAAE,uBAAuB,GAAG,CAAC,MAAM,MAAM,iBAAiB,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;QACrG,CAAC;QACD,MAAM,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAA6C,CAAC;QAC5E,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,6BAA6B,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;IAC1E,CAAC;IAED,OAAO;QACL,EAAE;QACF,KAAK;QACL,cAAc;QACd,UAAU,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YACzB,mBAAmB,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC;YAC1C,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;YACnC,IAAI,CAAC,OAAO,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACrC,MAAM,IAAI,KAAK,CAAC,GAAG,EAAE,oCAAoC,CAAC,CAAC;YAC7D,CAAC;YACD,OAAO,OAAO,CAAC,CAAC,CAAC,CAAC;QACpB,CAAC;QACD,UAAU,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;YAC1B,KAAK,MAAM,CAAC,IAAI,KAAK;gBAAE,mBAAmB,CAAC,CAAC,EAAE,cAAc,CAAC,CAAC;YAC9D,OAAO,IAAI,CAAC,KAAK,CAAC,CAAC;QACrB,CAAC;KACF,CAAC;AACJ,CAAC;AAED,iBAAiB;AAEjB,MAAM,uBAAuB,GAA2B;IACtD,wBAAwB,EAAE,IAAI;IAC9B,wBAAwB,EAAE,IAAI;IAC9B,wBAAwB,EAAE,IAAI;CAC/B,CAAC;AAEF,SAAS,oBAAoB,CAAC,MAK7B;IACC,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,wBAAwB,CAAC;IACvD,MAAM,OAAO,GAAG,CAAC,MAAM,CAAC,OAAO,IAAI,2BAA2B,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;IACpF,OAAO,6BAA6B,CAAC;QACnC,EAAE,EAAE,QAAQ;QACZ,MAAM,EAAE,MAAM,CAAC,MAAM;QACrB,KAAK;QACL,OAAO;QACP,cAAc,EAAE,uBAAuB,CAAC,KAAK,CAAC;QAC9C,UAAU,EAAE,MAAM,CAAC,UAAU,KAAK,KAAK,CAAC,CAAC,CAAC,kBAAkB,CAAC,OAAO,CAAC,IAAI,SAAS,CAAC,CAAC,CAAC,SAAS;KAC/F,CAAC,CAAC;AACL,CAAC;AAED,0DAA0D;AAE1D,MAAM,uBAAuB,GAA2B;IACtD,sBAAsB,EAAE,IAAI;IAC5B,oBAAoB,EAAE,IAAI;CAC3B,CAAC;AAEF,SAAS,oBAAoB,CAAC,MAK7B;IACC,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,sBAAsB,CAAC;IACrD,MAAM,OAAO,GAAG,CAAC,MAAM,CAAC,OAAO,IAAI,kDAAkD,CAAC,CAAC,OAAO,CAC5F,MAAM,EACN,EAAE,CACH,CAAC;IACF,MAAM,UAAU,GAAG,MAAM,CAAC,UAAU,KAAK,KAAK,CAAC,CAAC,CAAC,kBAAkB,CAAC,OAAO,CAAC,IAAI,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC;IAEtG,KAAK,UAAU,IAAI,CAAC,KAAe;QACjC,MAAM,QAAQ,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YACpC,KAAK,EAAE,UAAU,KAAK,EAAE;YACxB,OAAO,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE;SAC/B,CAAC,CAAC,CAAC;QACJ,MAAM,GAAG,GAAG,MAAM,SAAS,CACzB,GAAG,OAAO,WAAW,KAAK,qBAAqB,EAC/C;YACE,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,gBAAgB,EAAE,MAAM,CAAC,MAAM;aAChC;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,QAAQ,EAAE,CAAC;SACnC,EACD,UAAU,CACX,CAAC;QACF,IAAI,CAAC,GAAG,CAAC,EAAE;YAAE,MAAM,IAAI,KAAK,CAAC,6BAA6B,GAAG,CAAC,MAAM,MAAM,iBAAiB,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;QACjH,MAAM,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAgD,CAAC;QAC/E,OAAO,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,6BAA6B,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;IAC7E,CAAC;IAED,MAAM,SAAS,GAAG,uBAAuB,CAAC,KAAK,CAAC,CAAC;IAEjD,OAAO;QACL,EAAE,EAAE,QAAQ;QACZ,KAAK;QACL,cAAc,EAAE,SAAS;QACzB,UAAU,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YACzB,mBAAmB,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;YACrC,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;YACnC,IAAI,CAAC,OAAO,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;gBAAE,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAC;YAClG,OAAO,OAAO,CAAC,CAAC,CAAC,CAAC;QACpB,CAAC;QACD,UAAU,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;YAC1B,KAAK,MAAM,CAAC,IAAI,KAAK;gBAAE,mBAAmB,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;YACzD,OAAO,IAAI,CAAC,KAAK,CAAC,CAAC;QACrB,CAAC;KACF,CAAC;AACJ,CAAC;AAED,iBAAiB;AAEjB,SAAS,oBAAoB,CAAC,MAK7B;IACC,MAAM,OAAO,GAAG,CAAC,MAAM,CAAC,OAAO,IAAI,6BAA6B,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;IACtF,OAAO,6BAA6B,CAAC;QACnC,EAAE,EAAE,QAAQ;QACZ,MAAM,EAAE,MAAM,CAAC,MAAM;QACrB,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,eAAe;QACtC,OAAO;QACP,UAAU,EAAE,MAAM,CAAC,UAAU,KAAK,KAAK,CAAC,CAAC,CAAC,kBAAkB,CAAC,OAAO,CAAC,IAAI,SAAS,CAAC,CAAC,CAAC,SAAS;KAC/F,CAAC,CAAC;AACL,CAAC;AAED,kBAAkB;AAElB,SAAS,qBAAqB,CAAC,MAK9B;IACC,MAAM,OAAO,GAAG,CAAC,MAAM,CAAC,OAAO,IAAI,2BAA2B,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;IACpF,OAAO,6BAA6B,CAAC;QACnC,EAAE,EAAE,SAAS;QACb,MAAM,EAAE,MAAM,CAAC,MAAM;QACrB,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,eAAe;QACtC,OAAO;QACP,UAAU,EAAE,MAAM,CAAC,UAAU,KAAK,KAAK,CAAC,CAAC,CAAC,kBAAkB,CAAC,OAAO,CAAC,IAAI,SAAS,CAAC,CAAC,CAAC,SAAS;KAC/F,CAAC,CAAC;AACL,CAAC;AAED,qEAAqE;AAErE,SAAS,oBAAoB,CAAC,MAI7B;IACC,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,kBAAkB,CAAC;IACjD,MAAM,OAAO,GAAG,CAAC,MAAM,CAAC,OAAO,IAAI,wBAAwB,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;IACjF,MAAM,UAAU,GAAG,MAAM,CAAC,UAAU,KAAK,KAAK,CAAC,CAAC,CAAC,kBAAkB,CAAC,OAAO,CAAC,IAAI,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC;IAEtG,KAAK,UAAU,IAAI,CAAC,KAAe;QACjC,MAAM,GAAG,GAAG,MAAM,SAAS,CACzB,GAAG,OAAO,YAAY,EACtB;YACE,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;YAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;SACvC,EACD,UAAU,CACX,CAAC;QACF,IAAI,CAAC,GAAG,CAAC,EAAE;YAAE,MAAM,IAAI,KAAK,CAAC,6BAA6B,GAAG,CAAC,MAAM,MAAM,iBAAiB,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;QACjH,MAAM,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAA+B,CAAC;QAC9D,OAAO,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,6BAA6B,CAAC,CAAC,CAAC,CAAC,CAAC;IACtE,CAAC;IAED,OAAO;QACL,EAAE,EAAE,QAAQ;QACZ,KAAK;QACL,UAAU,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YACzB,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;YACnC,IAAI,CAAC,OAAO,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;gBAAE,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAC;YAClG,OAAO,OAAO,CAAC,CAAC,CAAC,CAAC;QACpB,CAAC;QACD,UAAU,EAAE,IAAI;KACjB,CAAC;AACJ,CAAC;AAYD,MAAM,UAAU,uBAAuB,CACrC,OAAuC;IAEvC,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,GAAG,OAAO,CAAC;IAEjE,QAAQ,QAAQ,EAAE,CAAC;QACjB,KAAK,QAAQ;YACX,IAAI,CAAC,MAAM;gBAAE,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;YAC9E,OAAO,oBAAoB,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QACtE,KAAK,QAAQ;YACX,IAAI,CAAC,MAAM;gBAAE,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;YAC9E,OAAO,oBAAoB,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QACtE,KAAK,QAAQ;YACX,IAAI,CAAC,MAAM;gBAAE,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;YAC9E,OAAO,oBAAoB,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QACtE,KAAK,SAAS;YACZ,IAAI,CAAC,MAAM;gBAAE,MAAM,IAAI,KAAK,CAAC,gDAAgD,CAAC,CAAC;YAC/E,OAAO,qBAAqB,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QACvE,KAAK,QAAQ;YACX,OAAO,oBAAoB,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QAC9D;YACE,MAAM,IAAI,KAAK,CAAC,+BAA+B,QAAQ,EAAE,CAAC,CAAC;IAC/D,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1,26 @@
|
|
|
1
|
+
import type { SimpleMemoryManager } from "./simple-manager.js";
|
|
2
|
+
export interface FileIndexerOptions {
|
|
3
|
+
workspaceDir: string;
|
|
4
|
+
memory: SimpleMemoryManager;
|
|
5
|
+
watchPaths?: string[];
|
|
6
|
+
debounceMs?: number;
|
|
7
|
+
extraPaths?: string[];
|
|
8
|
+
chunking?: {
|
|
9
|
+
tokens: number;
|
|
10
|
+
overlap: number;
|
|
11
|
+
};
|
|
12
|
+
}
|
|
13
|
+
export interface FileIndexer {
|
|
14
|
+
start(): void;
|
|
15
|
+
stop(): void;
|
|
16
|
+
sync(): Promise<FileIndexSyncResult>;
|
|
17
|
+
readonly watching: boolean;
|
|
18
|
+
}
|
|
19
|
+
export interface FileIndexSyncResult {
|
|
20
|
+
filesProcessed: number;
|
|
21
|
+
chunksStored: number;
|
|
22
|
+
chunksRemoved: number;
|
|
23
|
+
errors: number;
|
|
24
|
+
}
|
|
25
|
+
export declare function createFileIndexer(options: FileIndexerOptions): FileIndexer;
|
|
26
|
+
//# sourceMappingURL=file-indexer.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"file-indexer.d.ts","sourceRoot":"","sources":["../../src/memory/file-indexer.ts"],"names":[],"mappings":"AAQA,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,qBAAqB,CAAC;AAiB/D,MAAM,WAAW,kBAAkB;IACjC,YAAY,EAAE,MAAM,CAAC;IACrB,MAAM,EAAE,mBAAmB,CAAC;IAC5B,UAAU,CAAC,EAAE,MAAM,EAAE,CAAC;IACtB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,UAAU,CAAC,EAAE,MAAM,EAAE,CAAC;IACtB,QAAQ,CAAC,EAAE;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,CAAC;CAChD;AAED,MAAM,WAAW,WAAW;IAC1B,KAAK,IAAI,IAAI,CAAC;IACd,IAAI,IAAI,IAAI,CAAC;IACb,IAAI,IAAI,OAAO,CAAC,mBAAmB,CAAC,CAAC;IACrC,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC;CAC5B;AAED,MAAM,WAAW,mBAAmB;IAClC,cAAc,EAAE,MAAM,CAAC;IACvB,YAAY,EAAE,MAAM,CAAC;IACrB,aAAa,EAAE,MAAM,CAAC;IACtB,MAAM,EAAE,MAAM,CAAC;CAChB;AAcD,wBAAgB,iBAAiB,CAAC,OAAO,EAAE,kBAAkB,GAAG,WAAW,CA8N1E"}
|