@mcp-abap-adt/llm-agent 8.5.0 → 9.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/README.md +1 -0
- package/dist/generated/version.d.ts +1 -1
- package/dist/generated/version.js +1 -1
- package/dist/index.d.ts +13 -2
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +14 -1
- package/dist/index.js.map +1 -1
- package/dist/smart-agent/agent.d.ts +11 -1
- package/dist/smart-agent/agent.d.ts.map +1 -1
- package/dist/smart-agent/agent.js +40 -3
- package/dist/smart-agent/agent.js.map +1 -1
- package/dist/smart-agent/builder.d.ts +30 -1
- package/dist/smart-agent/builder.d.ts.map +1 -1
- package/dist/smart-agent/builder.js +126 -23
- package/dist/smart-agent/builder.js.map +1 -1
- package/dist/smart-agent/interfaces/pipeline.d.ts +5 -1
- package/dist/smart-agent/interfaces/pipeline.d.ts.map +1 -1
- package/dist/smart-agent/interfaces/rag.d.ts +75 -13
- package/dist/smart-agent/interfaces/rag.d.ts.map +1 -1
- package/dist/smart-agent/interfaces/rag.js +0 -3
- package/dist/smart-agent/interfaces/rag.js.map +1 -1
- package/dist/smart-agent/pipeline/context.d.ts +3 -1
- package/dist/smart-agent/pipeline/context.d.ts.map +1 -1
- package/dist/smart-agent/pipeline/default-pipeline.d.ts.map +1 -1
- package/dist/smart-agent/pipeline/default-pipeline.js +2 -0
- package/dist/smart-agent/pipeline/default-pipeline.js.map +1 -1
- package/dist/smart-agent/pipeline/handlers/history-upsert.d.ts.map +1 -1
- package/dist/smart-agent/pipeline/handlers/history-upsert.js +9 -3
- package/dist/smart-agent/pipeline/handlers/history-upsert.js.map +1 -1
- package/dist/smart-agent/rag/corrections/active-filtering-rag.d.ts +11 -0
- package/dist/smart-agent/rag/corrections/active-filtering-rag.d.ts.map +1 -0
- package/dist/smart-agent/rag/corrections/active-filtering-rag.js +33 -0
- package/dist/smart-agent/rag/corrections/active-filtering-rag.js.map +1 -0
- package/dist/smart-agent/rag/corrections/errors.d.ts +23 -0
- package/dist/smart-agent/rag/corrections/errors.d.ts.map +1 -0
- package/dist/smart-agent/rag/corrections/errors.js +44 -0
- package/dist/smart-agent/rag/corrections/errors.js.map +1 -0
- package/dist/smart-agent/rag/corrections/index.d.ts +4 -0
- package/dist/smart-agent/rag/corrections/index.d.ts.map +1 -0
- package/dist/smart-agent/rag/corrections/index.js +4 -0
- package/dist/smart-agent/rag/corrections/index.js.map +1 -0
- package/dist/smart-agent/rag/corrections/metadata.d.ts +24 -0
- package/dist/smart-agent/rag/corrections/metadata.d.ts.map +1 -0
- package/dist/smart-agent/rag/corrections/metadata.js +42 -0
- package/dist/smart-agent/rag/corrections/metadata.js.map +1 -0
- package/dist/smart-agent/rag/in-memory-rag.d.ts +3 -1
- package/dist/smart-agent/rag/in-memory-rag.d.ts.map +1 -1
- package/dist/smart-agent/rag/in-memory-rag.js +28 -0
- package/dist/smart-agent/rag/in-memory-rag.js.map +1 -1
- package/dist/smart-agent/rag/index.d.ts +7 -0
- package/dist/smart-agent/rag/index.d.ts.map +1 -1
- package/dist/smart-agent/rag/index.js +7 -0
- package/dist/smart-agent/rag/index.js.map +1 -1
- package/dist/smart-agent/rag/inverted-index.d.ts +2 -0
- package/dist/smart-agent/rag/inverted-index.d.ts.map +1 -1
- package/dist/smart-agent/rag/inverted-index.js +16 -0
- package/dist/smart-agent/rag/inverted-index.js.map +1 -1
- package/dist/smart-agent/rag/mcp-tools/index.d.ts +2 -0
- package/dist/smart-agent/rag/mcp-tools/index.d.ts.map +1 -0
- package/dist/smart-agent/rag/mcp-tools/index.js +2 -0
- package/dist/smart-agent/rag/mcp-tools/index.js.map +1 -0
- package/dist/smart-agent/rag/mcp-tools/rag-collection-tools.d.ts +20 -0
- package/dist/smart-agent/rag/mcp-tools/rag-collection-tools.d.ts.map +1 -0
- package/dist/smart-agent/rag/mcp-tools/rag-collection-tools.js +234 -0
- package/dist/smart-agent/rag/mcp-tools/rag-collection-tools.js.map +1 -0
- package/dist/smart-agent/rag/overlays/index.d.ts +3 -0
- package/dist/smart-agent/rag/overlays/index.d.ts.map +1 -0
- package/dist/smart-agent/rag/overlays/index.js +3 -0
- package/dist/smart-agent/rag/overlays/index.js.map +1 -0
- package/dist/smart-agent/rag/overlays/overlay-rag.d.ts +15 -0
- package/dist/smart-agent/rag/overlays/overlay-rag.d.ts.map +1 -0
- package/dist/smart-agent/rag/overlays/overlay-rag.js +55 -0
- package/dist/smart-agent/rag/overlays/overlay-rag.js.map +1 -0
- package/dist/smart-agent/rag/overlays/session-scoped-rag.d.ts +12 -0
- package/dist/smart-agent/rag/overlays/session-scoped-rag.d.ts.map +1 -0
- package/dist/smart-agent/rag/overlays/session-scoped-rag.js +31 -0
- package/dist/smart-agent/rag/overlays/session-scoped-rag.js.map +1 -0
- package/dist/smart-agent/rag/providers/base-provider.d.ts +29 -0
- package/dist/smart-agent/rag/providers/base-provider.d.ts.map +1 -0
- package/dist/smart-agent/rag/providers/base-provider.js +30 -0
- package/dist/smart-agent/rag/providers/base-provider.js.map +1 -0
- package/dist/smart-agent/rag/providers/in-memory-rag-provider.d.ts +31 -0
- package/dist/smart-agent/rag/providers/in-memory-rag-provider.d.ts.map +1 -0
- package/dist/smart-agent/rag/providers/in-memory-rag-provider.js +26 -0
- package/dist/smart-agent/rag/providers/in-memory-rag-provider.js.map +1 -0
- package/dist/smart-agent/rag/providers/index.d.ts +6 -0
- package/dist/smart-agent/rag/providers/index.d.ts.map +1 -0
- package/dist/smart-agent/rag/providers/index.js +6 -0
- package/dist/smart-agent/rag/providers/index.js.map +1 -0
- package/dist/smart-agent/rag/providers/qdrant-rag-provider.d.ts +39 -0
- package/dist/smart-agent/rag/providers/qdrant-rag-provider.d.ts.map +1 -0
- package/dist/smart-agent/rag/providers/qdrant-rag-provider.js +93 -0
- package/dist/smart-agent/rag/providers/qdrant-rag-provider.js.map +1 -0
- package/dist/smart-agent/rag/providers/simple-provider-registry.d.ts +8 -0
- package/dist/smart-agent/rag/providers/simple-provider-registry.d.ts.map +1 -0
- package/dist/smart-agent/rag/providers/simple-provider-registry.js +16 -0
- package/dist/smart-agent/rag/providers/simple-provider-registry.js.map +1 -0
- package/dist/smart-agent/rag/providers/vector-rag-provider.d.ts +33 -0
- package/dist/smart-agent/rag/providers/vector-rag-provider.d.ts.map +1 -0
- package/dist/smart-agent/rag/providers/vector-rag-provider.js +28 -0
- package/dist/smart-agent/rag/providers/vector-rag-provider.js.map +1 -0
- package/dist/smart-agent/rag/qdrant-rag.d.ts +4 -2
- package/dist/smart-agent/rag/qdrant-rag.d.ts.map +1 -1
- package/dist/smart-agent/rag/qdrant-rag.js +94 -0
- package/dist/smart-agent/rag/qdrant-rag.js.map +1 -1
- package/dist/smart-agent/rag/registry/index.d.ts +2 -0
- package/dist/smart-agent/rag/registry/index.d.ts.map +1 -0
- package/dist/smart-agent/rag/registry/index.js +2 -0
- package/dist/smart-agent/rag/registry/index.js.map +1 -0
- package/dist/smart-agent/rag/registry/simple-rag-registry.d.ts +34 -0
- package/dist/smart-agent/rag/registry/simple-rag-registry.d.ts.map +1 -0
- package/dist/smart-agent/rag/registry/simple-rag-registry.js +143 -0
- package/dist/smart-agent/rag/registry/simple-rag-registry.js.map +1 -0
- package/dist/smart-agent/rag/strategies/edit/direct.d.ts +13 -0
- package/dist/smart-agent/rag/strategies/edit/direct.d.ts.map +1 -0
- package/dist/smart-agent/rag/strategies/edit/direct.js +31 -0
- package/dist/smart-agent/rag/strategies/edit/direct.js.map +1 -0
- package/dist/smart-agent/rag/strategies/edit/immutable.d.ts +11 -0
- package/dist/smart-agent/rag/strategies/edit/immutable.d.ts.map +1 -0
- package/dist/smart-agent/rag/strategies/edit/immutable.js +14 -0
- package/dist/smart-agent/rag/strategies/edit/immutable.js.map +1 -0
- package/dist/smart-agent/rag/strategies/edit/index.d.ts +5 -0
- package/dist/smart-agent/rag/strategies/edit/index.d.ts.map +1 -0
- package/dist/smart-agent/rag/strategies/edit/index.js +5 -0
- package/dist/smart-agent/rag/strategies/edit/index.js.map +1 -0
- package/dist/smart-agent/rag/strategies/edit/overlay.d.ts +8 -0
- package/dist/smart-agent/rag/strategies/edit/overlay.d.ts.map +1 -0
- package/dist/smart-agent/rag/strategies/edit/overlay.js +8 -0
- package/dist/smart-agent/rag/strategies/edit/overlay.js.map +1 -0
- package/dist/smart-agent/rag/strategies/edit/session-scoped.d.ts +19 -0
- package/dist/smart-agent/rag/strategies/edit/session-scoped.d.ts.map +1 -0
- package/dist/smart-agent/rag/strategies/edit/session-scoped.js +46 -0
- package/dist/smart-agent/rag/strategies/edit/session-scoped.js.map +1 -0
- package/dist/smart-agent/rag/strategies/id/caller-provided.d.ts +6 -0
- package/dist/smart-agent/rag/strategies/id/caller-provided.d.ts.map +1 -0
- package/dist/smart-agent/rag/strategies/id/caller-provided.js +10 -0
- package/dist/smart-agent/rag/strategies/id/caller-provided.js.map +1 -0
- package/dist/smart-agent/rag/strategies/id/canonical-key.d.ts +6 -0
- package/dist/smart-agent/rag/strategies/id/canonical-key.d.ts.map +1 -0
- package/dist/smart-agent/rag/strategies/id/canonical-key.js +14 -0
- package/dist/smart-agent/rag/strategies/id/canonical-key.js.map +1 -0
- package/dist/smart-agent/rag/strategies/id/global-unique.d.ts +6 -0
- package/dist/smart-agent/rag/strategies/id/global-unique.d.ts.map +1 -0
- package/dist/smart-agent/rag/strategies/id/global-unique.js +9 -0
- package/dist/smart-agent/rag/strategies/id/global-unique.js.map +1 -0
- package/dist/smart-agent/rag/strategies/id/index.d.ts +5 -0
- package/dist/smart-agent/rag/strategies/id/index.d.ts.map +1 -0
- package/dist/smart-agent/rag/strategies/id/index.js +5 -0
- package/dist/smart-agent/rag/strategies/id/index.js.map +1 -0
- package/dist/smart-agent/rag/strategies/id/session-scoped.d.ts +8 -0
- package/dist/smart-agent/rag/strategies/id/session-scoped.d.ts.map +1 -0
- package/dist/smart-agent/rag/strategies/id/session-scoped.js +14 -0
- package/dist/smart-agent/rag/strategies/id/session-scoped.js.map +1 -0
- package/dist/smart-agent/rag/vector-rag.d.ts +4 -2
- package/dist/smart-agent/rag/vector-rag.d.ts.map +1 -1
- package/dist/smart-agent/rag/vector-rag.js +71 -28
- package/dist/smart-agent/rag/vector-rag.js.map +1 -1
- package/dist/smart-agent/resilience/fallback-rag.d.ts +5 -5
- package/dist/smart-agent/resilience/fallback-rag.d.ts.map +1 -1
- package/dist/smart-agent/resilience/fallback-rag.js +52 -23
- package/dist/smart-agent/resilience/fallback-rag.js.map +1 -1
- package/dist/smart-agent/testing/index.d.ts +4 -2
- package/dist/smart-agent/testing/index.d.ts.map +1 -1
- package/dist/smart-agent/testing/index.js +41 -17
- package/dist/smart-agent/testing/index.js.map +1 -1
- package/package.json +1 -1
- package/dist/agents/sap-ai-core-direct-agent.d.ts +0 -24
- package/dist/agents/sap-ai-core-direct-agent.d.ts.map +0 -1
- package/dist/agents/sap-ai-core-direct-agent.js +0 -117
- package/dist/agents/sap-ai-core-direct-agent.js.map +0 -1
- package/dist/llm-providers/sap-ai-core-direct.d.ts +0 -30
- package/dist/llm-providers/sap-ai-core-direct.d.ts.map +0 -1
- package/dist/llm-providers/sap-ai-core-direct.js +0 -178
- package/dist/llm-providers/sap-ai-core-direct.js.map +0 -1
- package/dist/smart-agent/interfaces/tool-result-compactor.d.ts +0 -11
- package/dist/smart-agent/interfaces/tool-result-compactor.d.ts.map +0 -1
- package/dist/smart-agent/interfaces/tool-result-compactor.js +0 -2
- package/dist/smart-agent/interfaces/tool-result-compactor.js.map +0 -1
- package/dist/smart-agent/pipeline/handlers/rag-upsert.d.ts +0 -18
- package/dist/smart-agent/pipeline/handlers/rag-upsert.d.ts.map +0 -1
- package/dist/smart-agent/pipeline/handlers/rag-upsert.js +0 -45
- package/dist/smart-agent/pipeline/handlers/rag-upsert.js.map +0 -1
- package/dist/smart-agent/policy/llm-tool-result-compactor.d.ts +0 -23
- package/dist/smart-agent/policy/llm-tool-result-compactor.d.ts.map +0 -1
- package/dist/smart-agent/policy/llm-tool-result-compactor.js +0 -57
- package/dist/smart-agent/policy/llm-tool-result-compactor.js.map +0 -1
- package/dist/smart-agent/policy/rag-only-tool-result-compactor.d.ts +0 -19
- package/dist/smart-agent/policy/rag-only-tool-result-compactor.d.ts.map +0 -1
- package/dist/smart-agent/policy/rag-only-tool-result-compactor.js +0 -33
- package/dist/smart-agent/policy/rag-only-tool-result-compactor.js.map +0 -1
- package/dist/smart-agent/policy/truncating-tool-result-compactor.d.ts +0 -19
- package/dist/smart-agent/policy/truncating-tool-result-compactor.d.ts.map +0 -1
- package/dist/smart-agent/policy/truncating-tool-result-compactor.js +0 -37
- package/dist/smart-agent/policy/truncating-tool-result-compactor.js.map +0 -1
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/smart-agent/rag/mcp-tools/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,6BAA6B,GAG9B,MAAM,2BAA2B,CAAC"}
|
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
import { z } from 'zod';
|
|
2
|
+
import type { IRagProviderRegistry, IRagRegistry } from '../../interfaces/rag.js';
|
|
3
|
+
export interface RagToolContext {
|
|
4
|
+
sessionId?: string;
|
|
5
|
+
userId?: string;
|
|
6
|
+
[key: string]: unknown;
|
|
7
|
+
}
|
|
8
|
+
export interface RagToolEntry {
|
|
9
|
+
toolDefinition: {
|
|
10
|
+
name: string;
|
|
11
|
+
description: string;
|
|
12
|
+
inputSchema: z.ZodRawShape;
|
|
13
|
+
};
|
|
14
|
+
handler: (context: RagToolContext, args: Record<string, unknown>) => Promise<unknown>;
|
|
15
|
+
}
|
|
16
|
+
export declare function buildRagCollectionToolEntries(opts: {
|
|
17
|
+
registry: IRagRegistry;
|
|
18
|
+
providerRegistry?: IRagProviderRegistry;
|
|
19
|
+
}): RagToolEntry[];
|
|
20
|
+
//# sourceMappingURL=rag-collection-tools.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"rag-collection-tools.d.ts","sourceRoot":"","sources":["../../../../src/smart-agent/rag/mcp-tools/rag-collection-tools.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,KAAK,EACV,oBAAoB,EACpB,YAAY,EACb,MAAM,yBAAyB,CAAC;AAMjC,MAAM,WAAW,cAAc;IAC7B,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;CACxB;AAED,MAAM,WAAW,YAAY;IAC3B,cAAc,EAAE;QACd,IAAI,EAAE,MAAM,CAAC;QACb,WAAW,EAAE,MAAM,CAAC;QACpB,WAAW,EAAE,CAAC,CAAC,WAAW,CAAC;KAC5B,CAAC;IACF,OAAO,EAAE,CACP,OAAO,EAAE,cAAc,EACvB,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAC1B,OAAO,CAAC,OAAO,CAAC,CAAC;CACvB;AAED,wBAAgB,6BAA6B,CAAC,IAAI,EAAE;IAClD,QAAQ,EAAE,YAAY,CAAC;IACvB,gBAAgB,CAAC,EAAE,oBAAoB,CAAC;CACzC,GAAG,YAAY,EAAE,CA+OjB"}
|
|
@@ -0,0 +1,234 @@
|
|
|
1
|
+
import { z } from 'zod';
|
|
2
|
+
import { buildCorrectionMetadata, deprecateMetadata, } from '../corrections/metadata.js';
|
|
3
|
+
export function buildRagCollectionToolEntries(opts) {
|
|
4
|
+
const { registry } = opts;
|
|
5
|
+
const resolveEditor = (name) => {
|
|
6
|
+
if (typeof name !== 'string') {
|
|
7
|
+
return { ok: false, error: 'collection is required' };
|
|
8
|
+
}
|
|
9
|
+
const editor = registry.getEditor(name);
|
|
10
|
+
if (!editor) {
|
|
11
|
+
return {
|
|
12
|
+
ok: false,
|
|
13
|
+
error: `Collection '${name}' is read-only or unknown`,
|
|
14
|
+
};
|
|
15
|
+
}
|
|
16
|
+
return { ok: true, editor };
|
|
17
|
+
};
|
|
18
|
+
const addTool = {
|
|
19
|
+
toolDefinition: {
|
|
20
|
+
name: 'rag_add',
|
|
21
|
+
description: 'Add a new document to a RAG collection.',
|
|
22
|
+
inputSchema: {
|
|
23
|
+
collection: z.string(),
|
|
24
|
+
text: z.string(),
|
|
25
|
+
canonicalKey: z.string(),
|
|
26
|
+
tags: z.array(z.string()).optional(),
|
|
27
|
+
},
|
|
28
|
+
},
|
|
29
|
+
handler: async (_ctx, args) => {
|
|
30
|
+
const r = resolveEditor(args.collection);
|
|
31
|
+
if (!r.ok)
|
|
32
|
+
return r;
|
|
33
|
+
const res = await r.editor.upsert(String(args.text), {
|
|
34
|
+
canonicalKey: String(args.canonicalKey),
|
|
35
|
+
tags: args.tags,
|
|
36
|
+
});
|
|
37
|
+
return res.ok
|
|
38
|
+
? { ok: true, id: res.value.id }
|
|
39
|
+
: { ok: false, error: res.error.message };
|
|
40
|
+
},
|
|
41
|
+
};
|
|
42
|
+
const correctTool = {
|
|
43
|
+
toolDefinition: {
|
|
44
|
+
name: 'rag_correct',
|
|
45
|
+
description: 'Supersede a document with a new corrected version. Marks the predecessor as superseded.',
|
|
46
|
+
inputSchema: {
|
|
47
|
+
collection: z.string(),
|
|
48
|
+
predecessorId: z.string(),
|
|
49
|
+
predecessorCanonicalKey: z.string(),
|
|
50
|
+
newText: z.string(),
|
|
51
|
+
reason: z.string(),
|
|
52
|
+
},
|
|
53
|
+
},
|
|
54
|
+
handler: async (_ctx, args) => {
|
|
55
|
+
const r = resolveEditor(args.collection);
|
|
56
|
+
if (!r.ok)
|
|
57
|
+
return r;
|
|
58
|
+
const predecessorMeta = {
|
|
59
|
+
canonicalKey: String(args.predecessorCanonicalKey),
|
|
60
|
+
};
|
|
61
|
+
const newRes = await r.editor.upsert(String(args.newText), {
|
|
62
|
+
canonicalKey: predecessorMeta.canonicalKey,
|
|
63
|
+
});
|
|
64
|
+
if (!newRes.ok)
|
|
65
|
+
return { ok: false, error: newRes.error.message };
|
|
66
|
+
const { predecessor } = buildCorrectionMetadata({
|
|
67
|
+
predecessor: predecessorMeta,
|
|
68
|
+
predecessorId: String(args.predecessorId),
|
|
69
|
+
newEntryId: newRes.value.id,
|
|
70
|
+
reason: String(args.reason),
|
|
71
|
+
});
|
|
72
|
+
const supRes = await r.editor.upsert('', {
|
|
73
|
+
...predecessor,
|
|
74
|
+
id: String(args.predecessorId),
|
|
75
|
+
});
|
|
76
|
+
if (!supRes.ok)
|
|
77
|
+
return { ok: false, error: supRes.error.message };
|
|
78
|
+
return {
|
|
79
|
+
ok: true,
|
|
80
|
+
predecessorId: String(args.predecessorId),
|
|
81
|
+
newId: newRes.value.id,
|
|
82
|
+
};
|
|
83
|
+
},
|
|
84
|
+
};
|
|
85
|
+
const deprecateTool = {
|
|
86
|
+
toolDefinition: {
|
|
87
|
+
name: 'rag_deprecate',
|
|
88
|
+
description: 'Mark a document as deprecated (idempotent).',
|
|
89
|
+
inputSchema: {
|
|
90
|
+
collection: z.string(),
|
|
91
|
+
id: z.string(),
|
|
92
|
+
canonicalKey: z.string(),
|
|
93
|
+
reason: z.string(),
|
|
94
|
+
},
|
|
95
|
+
},
|
|
96
|
+
handler: async (_ctx, args) => {
|
|
97
|
+
const r = resolveEditor(args.collection);
|
|
98
|
+
if (!r.ok)
|
|
99
|
+
return r;
|
|
100
|
+
const meta = deprecateMetadata({ canonicalKey: String(args.canonicalKey) }, String(args.reason));
|
|
101
|
+
const res = await r.editor.upsert('', {
|
|
102
|
+
...meta,
|
|
103
|
+
id: String(args.id),
|
|
104
|
+
});
|
|
105
|
+
return res.ok
|
|
106
|
+
? { ok: true, id: res.value.id }
|
|
107
|
+
: { ok: false, error: res.error.message };
|
|
108
|
+
},
|
|
109
|
+
};
|
|
110
|
+
const listTool = {
|
|
111
|
+
toolDefinition: {
|
|
112
|
+
name: 'rag_list_collections',
|
|
113
|
+
description: 'List known RAG collections with optional scope/provider filters.',
|
|
114
|
+
inputSchema: {
|
|
115
|
+
scope: z.enum(['session', 'user', 'global']).optional(),
|
|
116
|
+
provider: z.string().optional(),
|
|
117
|
+
},
|
|
118
|
+
},
|
|
119
|
+
handler: async (_ctx, args) => {
|
|
120
|
+
const metas = registry.list().filter((m) => {
|
|
121
|
+
if (args.scope && m.scope !== args.scope)
|
|
122
|
+
return false;
|
|
123
|
+
if (args.provider && m.providerName !== args.provider)
|
|
124
|
+
return false;
|
|
125
|
+
return true;
|
|
126
|
+
});
|
|
127
|
+
return { ok: true, collections: metas };
|
|
128
|
+
},
|
|
129
|
+
};
|
|
130
|
+
const describeTool = {
|
|
131
|
+
toolDefinition: {
|
|
132
|
+
name: 'rag_describe_collection',
|
|
133
|
+
description: 'Return the metadata of a RAG collection by name.',
|
|
134
|
+
inputSchema: { name: z.string() },
|
|
135
|
+
},
|
|
136
|
+
handler: async (_ctx, args) => {
|
|
137
|
+
const name = String(args.name);
|
|
138
|
+
const meta = registry.list().find((m) => m.name === name);
|
|
139
|
+
if (!meta) {
|
|
140
|
+
return { ok: false, error: `Collection '${name}' not found` };
|
|
141
|
+
}
|
|
142
|
+
return { ok: true, meta };
|
|
143
|
+
},
|
|
144
|
+
};
|
|
145
|
+
const deleteTool = {
|
|
146
|
+
toolDefinition: {
|
|
147
|
+
name: 'rag_delete_collection',
|
|
148
|
+
description: 'Delete a RAG collection you own (session or user scope).',
|
|
149
|
+
inputSchema: { name: z.string() },
|
|
150
|
+
},
|
|
151
|
+
handler: async (ctx, args) => {
|
|
152
|
+
const name = String(args.name);
|
|
153
|
+
const meta = registry.list().find((m) => m.name === name);
|
|
154
|
+
if (!meta) {
|
|
155
|
+
return { ok: false, error: `Collection '${name}' not found` };
|
|
156
|
+
}
|
|
157
|
+
if (meta.scope === 'global' || !meta.scope) {
|
|
158
|
+
return {
|
|
159
|
+
ok: false,
|
|
160
|
+
error: `Global collections cannot be deleted via MCP`,
|
|
161
|
+
};
|
|
162
|
+
}
|
|
163
|
+
if (meta.scope === 'session') {
|
|
164
|
+
if (!ctx.sessionId || ctx.sessionId !== meta.sessionId) {
|
|
165
|
+
return {
|
|
166
|
+
ok: false,
|
|
167
|
+
error: `sessionId mismatch for collection '${name}'`,
|
|
168
|
+
};
|
|
169
|
+
}
|
|
170
|
+
}
|
|
171
|
+
if (meta.scope === 'user') {
|
|
172
|
+
if (!ctx.userId || ctx.userId !== meta.userId) {
|
|
173
|
+
return {
|
|
174
|
+
ok: false,
|
|
175
|
+
error: `userId mismatch for collection '${name}'`,
|
|
176
|
+
};
|
|
177
|
+
}
|
|
178
|
+
}
|
|
179
|
+
const res = await registry.deleteCollection(name);
|
|
180
|
+
return res.ok ? { ok: true } : { ok: false, error: res.error.message };
|
|
181
|
+
},
|
|
182
|
+
};
|
|
183
|
+
const tools = [
|
|
184
|
+
addTool,
|
|
185
|
+
correctTool,
|
|
186
|
+
deprecateTool,
|
|
187
|
+
listTool,
|
|
188
|
+
describeTool,
|
|
189
|
+
deleteTool,
|
|
190
|
+
];
|
|
191
|
+
if (opts.providerRegistry) {
|
|
192
|
+
const providerRegistry = opts.providerRegistry;
|
|
193
|
+
const createTool = {
|
|
194
|
+
toolDefinition: {
|
|
195
|
+
name: 'rag_create_collection',
|
|
196
|
+
description: 'Create a new RAG collection via a provider.',
|
|
197
|
+
inputSchema: {
|
|
198
|
+
provider: z.string(),
|
|
199
|
+
name: z.string(),
|
|
200
|
+
scope: z.enum(['session', 'user', 'global']),
|
|
201
|
+
displayName: z.string().optional(),
|
|
202
|
+
description: z.string().optional(),
|
|
203
|
+
tags: z.array(z.string()).optional(),
|
|
204
|
+
},
|
|
205
|
+
},
|
|
206
|
+
handler: async (ctx, args) => {
|
|
207
|
+
const providerName = String(args.provider);
|
|
208
|
+
const provider = providerRegistry.getProvider(providerName);
|
|
209
|
+
if (!provider) {
|
|
210
|
+
return {
|
|
211
|
+
ok: false,
|
|
212
|
+
error: `RAG provider '${providerName}' is not registered`,
|
|
213
|
+
};
|
|
214
|
+
}
|
|
215
|
+
const res = await registry.createCollection({
|
|
216
|
+
providerName,
|
|
217
|
+
collectionName: String(args.name),
|
|
218
|
+
scope: args.scope,
|
|
219
|
+
sessionId: ctx.sessionId,
|
|
220
|
+
userId: ctx.userId,
|
|
221
|
+
displayName: args.displayName,
|
|
222
|
+
description: args.description,
|
|
223
|
+
tags: args.tags,
|
|
224
|
+
});
|
|
225
|
+
return res.ok
|
|
226
|
+
? { ok: true, meta: res.value }
|
|
227
|
+
: { ok: false, error: res.error.message };
|
|
228
|
+
},
|
|
229
|
+
};
|
|
230
|
+
tools.push(createTool);
|
|
231
|
+
}
|
|
232
|
+
return tools;
|
|
233
|
+
}
|
|
234
|
+
//# sourceMappingURL=rag-collection-tools.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"rag-collection-tools.js","sourceRoot":"","sources":["../../../../src/smart-agent/rag/mcp-tools/rag-collection-tools.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAKxB,OAAO,EACL,uBAAuB,EACvB,iBAAiB,GAClB,MAAM,4BAA4B,CAAC;AAoBpC,MAAM,UAAU,6BAA6B,CAAC,IAG7C;IACC,MAAM,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC;IAE1B,MAAM,aAAa,GAAG,CAAC,IAAa,EAAE,EAAE;QACtC,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC7B,OAAO,EAAE,EAAE,EAAE,KAAc,EAAE,KAAK,EAAE,wBAAwB,EAAE,CAAC;QACjE,CAAC;QACD,MAAM,MAAM,GAAG,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;QACxC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO;gBACL,EAAE,EAAE,KAAc;gBAClB,KAAK,EAAE,eAAe,IAAI,2BAA2B;aACtD,CAAC;QACJ,CAAC;QACD,OAAO,EAAE,EAAE,EAAE,IAAa,EAAE,MAAM,EAAE,CAAC;IACvC,CAAC,CAAC;IAEF,MAAM,OAAO,GAAiB;QAC5B,cAAc,EAAE;YACd,IAAI,EAAE,SAAS;YACf,WAAW,EAAE,yCAAyC;YACtD,WAAW,EAAE;gBACX,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE;gBACtB,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;gBAChB,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE;gBACxB,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;aACrC;SACF;QACD,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,EAAE;YAC5B,MAAM,CAAC,GAAG,aAAa,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YACzC,IAAI,CAAC,CAAC,CAAC,EAAE;gBAAE,OAAO,CAAC,CAAC;YACpB,MAAM,GAAG,GAAG,MAAM,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;gBACnD,YAAY,EAAE,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC;gBACvC,IAAI,EAAE,IAAI,CAAC,IAA4B;aACxC,CAAC,CAAC;YACH,OAAO,GAAG,CAAC,EAAE;gBACX,CAAC,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE;gBAChC,CAAC,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;QAC9C,CAAC;KACF,CAAC;IAEF,MAAM,WAAW,GAAiB;QAChC,cAAc,EAAE;YACd,IAAI,EAAE,aAAa;YACnB,WAAW,EACT,yFAAyF;YAC3F,WAAW,EAAE;gBACX,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE;gBACtB,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE;gBACzB,uBAAuB,EAAE,CAAC,CAAC,MAAM,EAAE;gBACnC,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE;gBACnB,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;aACnB;SACF;QACD,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,EAAE;YAC5B,MAAM,CAAC,GAAG,aAAa,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YACzC,IAAI,CAAC,CAAC,CAAC,EAAE;gBAAE,OAAO,CAAC,CAAC;YACpB,MAAM,eAAe,GAAG;gBACtB,YAAY,EAAE,MAAM,CAAC,IAAI,CAAC,uBAAuB,CAAC;aACnD,CAAC;YACF,MAAM,MAAM,GAAG,MAAM,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE;gBACzD,YAAY,EAAE,eAAe,CAAC,YAAY;aAC3C,CAAC,CAAC;YACH,IAAI,CAAC,MAAM,CAAC,EAAE;gBAAE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;YAElE,MAAM,EAAE,WAAW,EAAE,GAAG,uBAAuB,CAAC;gBAC9C,WAAW,EAAE,eAAe;gBAC5B,aAAa,EAAE,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC;gBACzC,UAAU,EAAE,MAAM,CAAC,KAAK,CAAC,EAAE;gBAC3B,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC;aAC5B,CAAC,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,EAAE;gBACvC,GAAG,WAAW;gBACd,EAAE,EAAE,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC;aAC/B,CAAC,CAAC;YACH,IAAI,CAAC,MAAM,CAAC,EAAE;gBAAE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;YAClE,OAAO;gBACL,EAAE,EAAE,IAAI;gBACR,aAAa,EAAE,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC;gBACzC,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,EAAE;aACvB,CAAC;QACJ,CAAC;KACF,CAAC;IAEF,MAAM,aAAa,GAAiB;QAClC,cAAc,EAAE;YACd,IAAI,EAAE,eAAe;YACrB,WAAW,EAAE,6CAA6C;YAC1D,WAAW,EAAE;gBACX,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE;gBACtB,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE;gBACd,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE;gBACxB,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;aACnB;SACF;QACD,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,EAAE;YAC5B,MAAM,CAAC,GAAG,aAAa,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YACzC,IAAI,CAAC,CAAC,CAAC,EAAE;gBAAE,OAAO,CAAC,CAAC;YACpB,MAAM,IAAI,GAAG,iBAAiB,CAC5B,EAAE,YAAY,EAAE,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE,EAC3C,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CACpB,CAAC;YACF,MAAM,GAAG,GAAG,MAAM,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,EAAE;gBACpC,GAAG,IAAI;gBACP,EAAE,EAAE,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;aACpB,CAAC,CAAC;YACH,OAAO,GAAG,CAAC,EAAE;gBACX,CAAC,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE;gBAChC,CAAC,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;QAC9C,CAAC;KACF,CAAC;IAEF,MAAM,QAAQ,GAAiB;QAC7B,cAAc,EAAE;YACd,IAAI,EAAE,sBAAsB;YAC5B,WAAW,EACT,kEAAkE;YACpE,WAAW,EAAE;gBACX,KAAK,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC,CAAC,QAAQ,EAAE;gBACvD,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;aAChC;SACF;QACD,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,EAAE;YAC5B,MAAM,KAAK,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;gBACzC,IAAI,IAAI,CAAC,KAAK,IAAI,CAAC,CAAC,KAAK,KAAK,IAAI,CAAC,KAAK;oBAAE,OAAO,KAAK,CAAC;gBACvD,IAAI,IAAI,CAAC,QAAQ,IAAI,CAAC,CAAC,YAAY,KAAK,IAAI,CAAC,QAAQ;oBAAE,OAAO,KAAK,CAAC;gBACpE,OAAO,IAAI,CAAC;YACd,CAAC,CAAC,CAAC;YACH,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;QAC1C,CAAC;KACF,CAAC;IAEF,MAAM,YAAY,GAAiB;QACjC,cAAc,EAAE;YACd,IAAI,EAAE,yBAAyB;YAC/B,WAAW,EAAE,kDAAkD;YAC/D,WAAW,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE;SAClC;QACD,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,EAAE;YAC5B,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC/B,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;YAC1D,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,eAAe,IAAI,aAAa,EAAE,CAAC;YAChE,CAAC;YACD,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;QAC5B,CAAC;KACF,CAAC;IAEF,MAAM,UAAU,GAAiB;QAC/B,cAAc,EAAE;YACd,IAAI,EAAE,uBAAuB;YAC7B,WAAW,EAAE,0DAA0D;YACvE,WAAW,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE;SAClC;QACD,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;YAC3B,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC/B,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;YAC1D,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,eAAe,IAAI,aAAa,EAAE,CAAC;YAChE,CAAC;YACD,IAAI,IAAI,CAAC,KAAK,KAAK,QAAQ,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;gBAC3C,OAAO;oBACL,EAAE,EAAE,KAAK;oBACT,KAAK,EAAE,8CAA8C;iBACtD,CAAC;YACJ,CAAC;YACD,IAAI,IAAI,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;gBAC7B,IAAI,CAAC,GAAG,CAAC,SAAS,IAAI,GAAG,CAAC,SAAS,KAAK,IAAI,CAAC,SAAS,EAAE,CAAC;oBACvD,OAAO;wBACL,EAAE,EAAE,KAAK;wBACT,KAAK,EAAE,sCAAsC,IAAI,GAAG;qBACrD,CAAC;gBACJ,CAAC;YACH,CAAC;YACD,IAAI,IAAI,CAAC,KAAK,KAAK,MAAM,EAAE,CAAC;gBAC1B,IAAI,CAAC,GAAG,CAAC,MAAM,IAAI,GAAG,CAAC,MAAM,KAAK,IAAI,CAAC,MAAM,EAAE,CAAC;oBAC9C,OAAO;wBACL,EAAE,EAAE,KAAK;wBACT,KAAK,EAAE,mCAAmC,IAAI,GAAG;qBAClD,CAAC;gBACJ,CAAC;YACH,CAAC;YACD,MAAM,GAAG,GAAG,MAAM,QAAQ,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;YAClD,OAAO,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;QACzE,CAAC;KACF,CAAC;IAEF,MAAM,KAAK,GAAmB;QAC5B,OAAO;QACP,WAAW;QACX,aAAa;QACb,QAAQ;QACR,YAAY;QACZ,UAAU;KACX,CAAC;IAEF,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAC1B,MAAM,gBAAgB,GAAG,IAAI,CAAC,gBAAgB,CAAC;QAC/C,MAAM,UAAU,GAAiB;YAC/B,cAAc,EAAE;gBACd,IAAI,EAAE,uBAAuB;gBAC7B,WAAW,EAAE,6CAA6C;gBAC1D,WAAW,EAAE;oBACX,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE;oBACpB,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;oBAChB,KAAK,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;oBAC5C,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;oBAClC,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;oBAClC,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;iBACrC;aACF;YACD,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;gBAC3B,MAAM,YAAY,GAAG,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;gBAC3C,MAAM,QAAQ,GAAG,gBAAgB,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC;gBAC5D,IAAI,CAAC,QAAQ,EAAE,CAAC;oBACd,OAAO;wBACL,EAAE,EAAE,KAAK;wBACT,KAAK,EAAE,iBAAiB,YAAY,qBAAqB;qBAC1D,CAAC;gBACJ,CAAC;gBACD,MAAM,GAAG,GAAG,MAAM,QAAQ,CAAC,gBAAgB,CAAC;oBAC1C,YAAY;oBACZ,cAAc,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC;oBACjC,KAAK,EAAE,IAAI,CAAC,KAAsC;oBAClD,SAAS,EAAE,GAAG,CAAC,SAAS;oBACxB,MAAM,EAAE,GAAG,CAAC,MAAM;oBAClB,WAAW,EAAE,IAAI,CAAC,WAAiC;oBACnD,WAAW,EAAE,IAAI,CAAC,WAAiC;oBACnD,IAAI,EAAE,IAAI,CAAC,IAA4B;iBACxC,CAAC,CAAC;gBACH,OAAO,GAAG,CAAC,EAAE;oBACX,CAAC,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,CAAC,KAAK,EAAE;oBAC/B,CAAC,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;YAC9C,CAAC;SACF,CAAC;QACF,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IACzB,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/smart-agent/rag/overlays/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAC9C,OAAO,EAAE,gBAAgB,EAAE,MAAM,yBAAyB,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/smart-agent/rag/overlays/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAC9C,OAAO,EAAE,gBAAgB,EAAE,MAAM,yBAAyB,CAAC"}
|
|
@@ -0,0 +1,15 @@
|
|
|
1
|
+
import type { IQueryEmbedding } from '../../interfaces/query-embedding.js';
|
|
2
|
+
import type { IRag } from '../../interfaces/rag.js';
|
|
3
|
+
import type { CallOptions, RagError, RagResult, Result } from '../../interfaces/types.js';
|
|
4
|
+
export declare class OverlayRag implements IRag {
|
|
5
|
+
protected readonly base: IRag;
|
|
6
|
+
protected readonly overlay: IRag;
|
|
7
|
+
constructor(base: IRag, overlay: IRag);
|
|
8
|
+
query(embedding: IQueryEmbedding, k: number, options?: CallOptions): Promise<Result<RagResult[], RagError>>;
|
|
9
|
+
getById(id: string, options?: CallOptions): Promise<Result<RagResult | null, RagError>>;
|
|
10
|
+
healthCheck(options?: CallOptions): Promise<Result<void, RagError>>;
|
|
11
|
+
/** Hook for subclasses to drop overlay rows (e.g. by sessionId). */
|
|
12
|
+
protected filterOverlay(results: RagResult[]): RagResult[];
|
|
13
|
+
protected overlayAllows(_result: RagResult): boolean;
|
|
14
|
+
}
|
|
15
|
+
//# sourceMappingURL=overlay-rag.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"overlay-rag.d.ts","sourceRoot":"","sources":["../../../../src/smart-agent/rag/overlays/overlay-rag.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,qCAAqC,CAAC;AAC3E,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,yBAAyB,CAAC;AACpD,OAAO,KAAK,EACV,WAAW,EACX,QAAQ,EACR,SAAS,EACT,MAAM,EACP,MAAM,2BAA2B,CAAC;AAEnC,qBAAa,UAAW,YAAW,IAAI;IAEnC,SAAS,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI;IAC7B,SAAS,CAAC,QAAQ,CAAC,OAAO,EAAE,IAAI;gBADb,IAAI,EAAE,IAAI,EACV,OAAO,EAAE,IAAI;IAG5B,KAAK,CACT,SAAS,EAAE,eAAe,EAC1B,CAAC,EAAE,MAAM,EACT,OAAO,CAAC,EAAE,WAAW,GACpB,OAAO,CAAC,MAAM,CAAC,SAAS,EAAE,EAAE,QAAQ,CAAC,CAAC;IAwBnC,OAAO,CACX,EAAE,EAAE,MAAM,EACV,OAAO,CAAC,EAAE,WAAW,GACpB,OAAO,CAAC,MAAM,CAAC,SAAS,GAAG,IAAI,EAAE,QAAQ,CAAC,CAAC;IAOxC,WAAW,CAAC,OAAO,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,MAAM,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;IASzE,oEAAoE;IACpE,SAAS,CAAC,aAAa,CAAC,OAAO,EAAE,SAAS,EAAE,GAAG,SAAS,EAAE;IAI1D,SAAS,CAAC,aAAa,CAAC,OAAO,EAAE,SAAS,GAAG,OAAO;CAGrD"}
|
|
@@ -0,0 +1,55 @@
|
|
|
1
|
+
export class OverlayRag {
|
|
2
|
+
base;
|
|
3
|
+
overlay;
|
|
4
|
+
constructor(base, overlay) {
|
|
5
|
+
this.base = base;
|
|
6
|
+
this.overlay = overlay;
|
|
7
|
+
}
|
|
8
|
+
async query(embedding, k, options) {
|
|
9
|
+
const [baseRes, overlayRes] = await Promise.all([
|
|
10
|
+
this.base.query(embedding, k, options),
|
|
11
|
+
this.overlay.query(embedding, k, options),
|
|
12
|
+
]);
|
|
13
|
+
if (!baseRes.ok)
|
|
14
|
+
return baseRes;
|
|
15
|
+
if (!overlayRes.ok)
|
|
16
|
+
return overlayRes;
|
|
17
|
+
const overlayList = this.filterOverlay(overlayRes.value);
|
|
18
|
+
const overlayKeys = new Set(overlayList
|
|
19
|
+
.map((r) => r.metadata.canonicalKey)
|
|
20
|
+
.filter((key) => typeof key === 'string'));
|
|
21
|
+
const baseKept = baseRes.value.filter((r) => {
|
|
22
|
+
const key = r.metadata.canonicalKey;
|
|
23
|
+
return typeof key !== 'string' || !overlayKeys.has(key);
|
|
24
|
+
});
|
|
25
|
+
const merged = [...overlayList, ...baseKept]
|
|
26
|
+
.sort((a, b) => b.score - a.score)
|
|
27
|
+
.slice(0, k);
|
|
28
|
+
return { ok: true, value: merged };
|
|
29
|
+
}
|
|
30
|
+
async getById(id, options) {
|
|
31
|
+
const o = await this.overlay.getById(id, options);
|
|
32
|
+
if (!o.ok)
|
|
33
|
+
return o;
|
|
34
|
+
if (o.value !== null && this.overlayAllows(o.value))
|
|
35
|
+
return o;
|
|
36
|
+
return this.base.getById(id, options);
|
|
37
|
+
}
|
|
38
|
+
async healthCheck(options) {
|
|
39
|
+
const [a, b] = await Promise.all([
|
|
40
|
+
this.base.healthCheck(options),
|
|
41
|
+
this.overlay.healthCheck(options),
|
|
42
|
+
]);
|
|
43
|
+
if (!a.ok)
|
|
44
|
+
return a;
|
|
45
|
+
return b;
|
|
46
|
+
}
|
|
47
|
+
/** Hook for subclasses to drop overlay rows (e.g. by sessionId). */
|
|
48
|
+
filterOverlay(results) {
|
|
49
|
+
return results;
|
|
50
|
+
}
|
|
51
|
+
overlayAllows(_result) {
|
|
52
|
+
return true;
|
|
53
|
+
}
|
|
54
|
+
}
|
|
55
|
+
//# sourceMappingURL=overlay-rag.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"overlay-rag.js","sourceRoot":"","sources":["../../../../src/smart-agent/rag/overlays/overlay-rag.ts"],"names":[],"mappings":"AASA,MAAM,OAAO,UAAU;IAEA;IACA;IAFrB,YACqB,IAAU,EACV,OAAa;QADb,SAAI,GAAJ,IAAI,CAAM;QACV,YAAO,GAAP,OAAO,CAAM;IAC/B,CAAC;IAEJ,KAAK,CAAC,KAAK,CACT,SAA0B,EAC1B,CAAS,EACT,OAAqB;QAErB,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;YAC9C,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC,EAAE,OAAO,CAAC;YACtC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC,EAAE,OAAO,CAAC;SAC1C,CAAC,CAAC;QACH,IAAI,CAAC,OAAO,CAAC,EAAE;YAAE,OAAO,OAAO,CAAC;QAChC,IAAI,CAAC,UAAU,CAAC,EAAE;YAAE,OAAO,UAAU,CAAC;QAEtC,MAAM,WAAW,GAAG,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QACzD,MAAM,WAAW,GAAG,IAAI,GAAG,CACzB,WAAW;aACR,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,YAAY,CAAC;aACnC,MAAM,CAAC,CAAC,GAAG,EAAiB,EAAE,CAAC,OAAO,GAAG,KAAK,QAAQ,CAAC,CAC3D,CAAC;QACF,MAAM,QAAQ,GAAG,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;YAC1C,MAAM,GAAG,GAAG,CAAC,CAAC,QAAQ,CAAC,YAAY,CAAC;YACpC,OAAO,OAAO,GAAG,KAAK,QAAQ,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,CAAC,GAAG,WAAW,EAAE,GAAG,QAAQ,CAAC;aACzC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC;aACjC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QACf,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;IACrC,CAAC;IAED,KAAK,CAAC,OAAO,CACX,EAAU,EACV,OAAqB;QAErB,MAAM,CAAC,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QAClD,IAAI,CAAC,CAAC,CAAC,EAAE;YAAE,OAAO,CAAC,CAAC;QACpB,IAAI,CAAC,CAAC,KAAK,KAAK,IAAI,IAAI,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,KAAK,CAAC;YAAE,OAAO,CAAC,CAAC;QAC9D,OAAO,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;IACxC,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,OAAqB;QACrC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;YAC/B,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC;YAC9B,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC;SAClC,CAAC,CAAC;QACH,IAAI,CAAC,CAAC,CAAC,EAAE;YAAE,OAAO,CAAC,CAAC;QACpB,OAAO,CAAC,CAAC;IACX,CAAC;IAED,oEAAoE;IAC1D,aAAa,CAAC,OAAoB;QAC1C,OAAO,OAAO,CAAC;IACjB,CAAC;IAES,aAAa,CAAC,OAAkB;QACxC,OAAO,IAAI,CAAC;IACd,CAAC;CACF"}
|
|
@@ -0,0 +1,12 @@
|
|
|
1
|
+
import type { IRag } from '../../interfaces/rag.js';
|
|
2
|
+
import type { RagResult } from '../../interfaces/types.js';
|
|
3
|
+
import { OverlayRag } from './overlay-rag.js';
|
|
4
|
+
export declare class SessionScopedRag extends OverlayRag {
|
|
5
|
+
private readonly sessionId;
|
|
6
|
+
private readonly ttlMs?;
|
|
7
|
+
constructor(base: IRag, overlay: IRag, sessionId: string, ttlMs?: number | undefined);
|
|
8
|
+
protected filterOverlay(results: RagResult[]): RagResult[];
|
|
9
|
+
protected overlayAllows(result: RagResult): boolean;
|
|
10
|
+
private matches;
|
|
11
|
+
}
|
|
12
|
+
//# sourceMappingURL=session-scoped-rag.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"session-scoped-rag.d.ts","sourceRoot":"","sources":["../../../../src/smart-agent/rag/overlays/session-scoped-rag.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,yBAAyB,CAAC;AACpD,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,2BAA2B,CAAC;AAC3D,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAE9C,qBAAa,gBAAiB,SAAQ,UAAU;IAI5C,OAAO,CAAC,QAAQ,CAAC,SAAS;IAC1B,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC;gBAHvB,IAAI,EAAE,IAAI,EACV,OAAO,EAAE,IAAI,EACI,SAAS,EAAE,MAAM,EACjB,KAAK,CAAC,EAAE,MAAM,YAAA;cAKd,aAAa,CAAC,OAAO,EAAE,SAAS,EAAE,GAAG,SAAS,EAAE;cAMhD,aAAa,CAAC,MAAM,EAAE,SAAS,GAAG,OAAO;IAM5D,OAAO,CAAC,OAAO;CAWhB"}
|
|
@@ -0,0 +1,31 @@
|
|
|
1
|
+
import { OverlayRag } from './overlay-rag.js';
|
|
2
|
+
export class SessionScopedRag extends OverlayRag {
|
|
3
|
+
sessionId;
|
|
4
|
+
ttlMs;
|
|
5
|
+
constructor(base, overlay, sessionId, ttlMs) {
|
|
6
|
+
super(base, overlay);
|
|
7
|
+
this.sessionId = sessionId;
|
|
8
|
+
this.ttlMs = ttlMs;
|
|
9
|
+
}
|
|
10
|
+
filterOverlay(results) {
|
|
11
|
+
const cutoff = this.ttlMs !== undefined ? Date.now() - this.ttlMs : undefined;
|
|
12
|
+
return results.filter((r) => this.matches(r, cutoff));
|
|
13
|
+
}
|
|
14
|
+
overlayAllows(result) {
|
|
15
|
+
const cutoff = this.ttlMs !== undefined ? Date.now() - this.ttlMs : undefined;
|
|
16
|
+
return this.matches(result, cutoff);
|
|
17
|
+
}
|
|
18
|
+
matches(result, cutoffMs) {
|
|
19
|
+
if (result.metadata.sessionId !== this.sessionId)
|
|
20
|
+
return false;
|
|
21
|
+
if (cutoffMs !== undefined) {
|
|
22
|
+
const createdMs = typeof result.metadata.createdAt === 'number'
|
|
23
|
+
? result.metadata.createdAt
|
|
24
|
+
: undefined;
|
|
25
|
+
if (createdMs !== undefined && createdMs < cutoffMs)
|
|
26
|
+
return false;
|
|
27
|
+
}
|
|
28
|
+
return true;
|
|
29
|
+
}
|
|
30
|
+
}
|
|
31
|
+
//# sourceMappingURL=session-scoped-rag.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"session-scoped-rag.js","sourceRoot":"","sources":["../../../../src/smart-agent/rag/overlays/session-scoped-rag.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAE9C,MAAM,OAAO,gBAAiB,SAAQ,UAAU;IAI3B;IACA;IAJnB,YACE,IAAU,EACV,OAAa,EACI,SAAiB,EACjB,KAAc;QAE/B,KAAK,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QAHJ,cAAS,GAAT,SAAS,CAAQ;QACjB,UAAK,GAAL,KAAK,CAAS;IAGjC,CAAC;IAEkB,aAAa,CAAC,OAAoB;QACnD,MAAM,MAAM,GACV,IAAI,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;QACjE,OAAO,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC;IACxD,CAAC;IAEkB,aAAa,CAAC,MAAiB;QAChD,MAAM,MAAM,GACV,IAAI,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;QACjE,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACtC,CAAC;IAEO,OAAO,CAAC,MAAiB,EAAE,QAA4B;QAC7D,IAAI,MAAM,CAAC,QAAQ,CAAC,SAAS,KAAK,IAAI,CAAC,SAAS;YAAE,OAAO,KAAK,CAAC;QAC/D,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;YAC3B,MAAM,SAAS,GACb,OAAO,MAAM,CAAC,QAAQ,CAAC,SAAS,KAAK,QAAQ;gBAC3C,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,SAAS;gBAC3B,CAAC,CAAC,SAAS,CAAC;YAChB,IAAI,SAAS,KAAK,SAAS,IAAI,SAAS,GAAG,QAAQ;gBAAE,OAAO,KAAK,CAAC;QACpE,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;CACF"}
|
|
@@ -0,0 +1,29 @@
|
|
|
1
|
+
import type { IIdStrategy, IRag, IRagEditor, IRagProvider, RagCollectionScope } from '../../interfaces/rag.js';
|
|
2
|
+
import type { RagError, Result } from '../../interfaces/types.js';
|
|
3
|
+
export declare abstract class AbstractRagProvider implements IRagProvider {
|
|
4
|
+
abstract readonly name: string;
|
|
5
|
+
abstract readonly kind: string;
|
|
6
|
+
abstract readonly editable: boolean;
|
|
7
|
+
abstract readonly supportedScopes: readonly RagCollectionScope[];
|
|
8
|
+
protected idStrategyFactory?: (opts: {
|
|
9
|
+
scope: RagCollectionScope;
|
|
10
|
+
sessionId?: string;
|
|
11
|
+
userId?: string;
|
|
12
|
+
}) => IIdStrategy;
|
|
13
|
+
abstract createCollection(name: string, opts: {
|
|
14
|
+
scope: RagCollectionScope;
|
|
15
|
+
sessionId?: string;
|
|
16
|
+
userId?: string;
|
|
17
|
+
}): Promise<Result<{
|
|
18
|
+
rag: IRag;
|
|
19
|
+
editor: IRagEditor;
|
|
20
|
+
}, RagError>>;
|
|
21
|
+
protected checkScope(scope: RagCollectionScope): Result<void, RagError>;
|
|
22
|
+
protected pickIdStrategy(opts: {
|
|
23
|
+
scope: RagCollectionScope;
|
|
24
|
+
sessionId?: string;
|
|
25
|
+
userId?: string;
|
|
26
|
+
}): IIdStrategy;
|
|
27
|
+
protected buildEditor(rag: IRag, idStrategy: IIdStrategy): IRagEditor;
|
|
28
|
+
}
|
|
29
|
+
//# sourceMappingURL=base-provider.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"base-provider.d.ts","sourceRoot":"","sources":["../../../../src/smart-agent/rag/providers/base-provider.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,WAAW,EACX,IAAI,EACJ,UAAU,EACV,YAAY,EACZ,kBAAkB,EACnB,MAAM,yBAAyB,CAAC;AACjC,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,2BAA2B,CAAC;AAWlE,8BAAsB,mBAAoB,YAAW,YAAY;IAC/D,QAAQ,CAAC,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IAC/B,QAAQ,CAAC,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IAC/B,QAAQ,CAAC,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC;IACpC,QAAQ,CAAC,QAAQ,CAAC,eAAe,EAAE,SAAS,kBAAkB,EAAE,CAAC;IAEjE,SAAS,CAAC,iBAAiB,CAAC,EAAE,CAAC,IAAI,EAAE;QACnC,KAAK,EAAE,kBAAkB,CAAC;QAC1B,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,MAAM,CAAC,EAAE,MAAM,CAAC;KACjB,KAAK,WAAW,CAAC;IAElB,QAAQ,CAAC,gBAAgB,CACvB,IAAI,EAAE,MAAM,EACZ,IAAI,EAAE;QACJ,KAAK,EAAE,kBAAkB,CAAC;QAC1B,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,MAAM,CAAC,EAAE,MAAM,CAAC;KACjB,GACA,OAAO,CAAC,MAAM,CAAC;QAAE,GAAG,EAAE,IAAI,CAAC;QAAC,MAAM,EAAE,UAAU,CAAA;KAAE,EAAE,QAAQ,CAAC,CAAC;IAE/D,SAAS,CAAC,UAAU,CAAC,KAAK,EAAE,kBAAkB,GAAG,MAAM,CAAC,IAAI,EAAE,QAAQ,CAAC;IAOvE,SAAS,CAAC,cAAc,CAAC,IAAI,EAAE;QAC7B,KAAK,EAAE,kBAAkB,CAAC;QAC1B,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,MAAM,CAAC,EAAE,MAAM,CAAC;KACjB,GAAG,WAAW;IAQf,SAAS,CAAC,WAAW,CAAC,GAAG,EAAE,IAAI,EAAE,UAAU,EAAE,WAAW,GAAG,UAAU;CAUtE"}
|
|
@@ -0,0 +1,30 @@
|
|
|
1
|
+
import { UnsupportedScopeError } from '../corrections/errors.js';
|
|
2
|
+
import { DirectEditStrategy, ImmutableEditStrategy, } from '../strategies/edit/index.js';
|
|
3
|
+
import { GlobalUniqueIdStrategy, SessionScopedIdStrategy, } from '../strategies/id/index.js';
|
|
4
|
+
export class AbstractRagProvider {
|
|
5
|
+
idStrategyFactory;
|
|
6
|
+
checkScope(scope) {
|
|
7
|
+
if (!this.supportedScopes.includes(scope)) {
|
|
8
|
+
return { ok: false, error: new UnsupportedScopeError(this.name, scope) };
|
|
9
|
+
}
|
|
10
|
+
return { ok: true, value: undefined };
|
|
11
|
+
}
|
|
12
|
+
pickIdStrategy(opts) {
|
|
13
|
+
if (this.idStrategyFactory)
|
|
14
|
+
return this.idStrategyFactory(opts);
|
|
15
|
+
if (opts.scope === 'session' && opts.sessionId) {
|
|
16
|
+
return new SessionScopedIdStrategy(opts.sessionId);
|
|
17
|
+
}
|
|
18
|
+
return new GlobalUniqueIdStrategy();
|
|
19
|
+
}
|
|
20
|
+
buildEditor(rag, idStrategy) {
|
|
21
|
+
if (!this.editable)
|
|
22
|
+
return new ImmutableEditStrategy(this.name);
|
|
23
|
+
const writer = rag.writer?.();
|
|
24
|
+
if (!writer) {
|
|
25
|
+
throw new Error(`Provider '${this.name}' requires an IRag with writer() support for editable mode`);
|
|
26
|
+
}
|
|
27
|
+
return new DirectEditStrategy(writer, idStrategy);
|
|
28
|
+
}
|
|
29
|
+
}
|
|
30
|
+
//# sourceMappingURL=base-provider.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"base-provider.js","sourceRoot":"","sources":["../../../../src/smart-agent/rag/providers/base-provider.ts"],"names":[],"mappings":"AAQA,OAAO,EAAE,qBAAqB,EAAE,MAAM,0BAA0B,CAAC;AACjE,OAAO,EACL,kBAAkB,EAClB,qBAAqB,GACtB,MAAM,6BAA6B,CAAC;AACrC,OAAO,EACL,sBAAsB,EACtB,uBAAuB,GACxB,MAAM,2BAA2B,CAAC;AAEnC,MAAM,OAAgB,mBAAmB;IAM7B,iBAAiB,CAIT;IAWR,UAAU,CAAC,KAAyB;QAC5C,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YAC1C,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,qBAAqB,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,CAAC,EAAE,CAAC;QAC3E,CAAC;QACD,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;IACxC,CAAC;IAES,cAAc,CAAC,IAIxB;QACC,IAAI,IAAI,CAAC,iBAAiB;YAAE,OAAO,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QAChE,IAAI,IAAI,CAAC,KAAK,KAAK,SAAS,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YAC/C,OAAO,IAAI,uBAAuB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACrD,CAAC;QACD,OAAO,IAAI,sBAAsB,EAAE,CAAC;IACtC,CAAC;IAES,WAAW,CAAC,GAAS,EAAE,UAAuB;QACtD,IAAI,CAAC,IAAI,CAAC,QAAQ;YAAE,OAAO,IAAI,qBAAqB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChE,MAAM,MAAM,GAAG,GAAG,CAAC,MAAM,EAAE,EAAE,CAAC;QAC9B,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CACb,aAAa,IAAI,CAAC,IAAI,4DAA4D,CACnF,CAAC;QACJ,CAAC;QACD,OAAO,IAAI,kBAAkB,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;IACpD,CAAC;CACF"}
|
|
@@ -0,0 +1,31 @@
|
|
|
1
|
+
import type { IIdStrategy, IRag, IRagEditor, RagCollectionScope } from '../../interfaces/rag.js';
|
|
2
|
+
import type { RagError, Result } from '../../interfaces/types.js';
|
|
3
|
+
import { type InMemoryRagConfig } from '../in-memory-rag.js';
|
|
4
|
+
import { AbstractRagProvider } from './base-provider.js';
|
|
5
|
+
export interface InMemoryRagProviderConfig {
|
|
6
|
+
name: string;
|
|
7
|
+
editable?: boolean;
|
|
8
|
+
inMemoryRagConfig?: InMemoryRagConfig;
|
|
9
|
+
idStrategyFactory?: (opts: {
|
|
10
|
+
scope: RagCollectionScope;
|
|
11
|
+
sessionId?: string;
|
|
12
|
+
userId?: string;
|
|
13
|
+
}) => IIdStrategy;
|
|
14
|
+
}
|
|
15
|
+
export declare class InMemoryRagProvider extends AbstractRagProvider {
|
|
16
|
+
readonly name: string;
|
|
17
|
+
readonly kind = "vector";
|
|
18
|
+
readonly editable: boolean;
|
|
19
|
+
readonly supportedScopes: readonly ["session"];
|
|
20
|
+
private readonly inMemoryCfg?;
|
|
21
|
+
constructor(cfg: InMemoryRagProviderConfig);
|
|
22
|
+
createCollection(_name: string, opts: {
|
|
23
|
+
scope: RagCollectionScope;
|
|
24
|
+
sessionId?: string;
|
|
25
|
+
userId?: string;
|
|
26
|
+
}): Promise<Result<{
|
|
27
|
+
rag: IRag;
|
|
28
|
+
editor: IRagEditor;
|
|
29
|
+
}, RagError>>;
|
|
30
|
+
}
|
|
31
|
+
//# sourceMappingURL=in-memory-rag-provider.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"in-memory-rag-provider.d.ts","sourceRoot":"","sources":["../../../../src/smart-agent/rag/providers/in-memory-rag-provider.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,WAAW,EACX,IAAI,EACJ,UAAU,EACV,kBAAkB,EACnB,MAAM,yBAAyB,CAAC;AACjC,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,2BAA2B,CAAC;AAClE,OAAO,EAAe,KAAK,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAC1E,OAAO,EAAE,mBAAmB,EAAE,MAAM,oBAAoB,CAAC;AAEzD,MAAM,WAAW,yBAAyB;IACxC,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB,iBAAiB,CAAC,EAAE,iBAAiB,CAAC;IACtC,iBAAiB,CAAC,EAAE,CAAC,IAAI,EAAE;QACzB,KAAK,EAAE,kBAAkB,CAAC;QAC1B,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,MAAM,CAAC,EAAE,MAAM,CAAC;KACjB,KAAK,WAAW,CAAC;CACnB;AAED,qBAAa,mBAAoB,SAAQ,mBAAmB;IAC1D,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,IAAI,YAAY;IACzB,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC;IAC3B,QAAQ,CAAC,eAAe,uBAAwB;IAEhD,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAoB;gBAErC,GAAG,EAAE,yBAAyB;IAQpC,gBAAgB,CACpB,KAAK,EAAE,MAAM,EACb,IAAI,EAAE;QACJ,KAAK,EAAE,kBAAkB,CAAC;QAC1B,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,MAAM,CAAC,EAAE,MAAM,CAAC;KACjB,GACA,OAAO,CAAC,MAAM,CAAC;QAAE,GAAG,EAAE,IAAI,CAAC;QAAC,MAAM,EAAE,UAAU,CAAA;KAAE,EAAE,QAAQ,CAAC,CAAC;CAOhE"}
|
|
@@ -0,0 +1,26 @@
|
|
|
1
|
+
import { InMemoryRag } from '../in-memory-rag.js';
|
|
2
|
+
import { AbstractRagProvider } from './base-provider.js';
|
|
3
|
+
export class InMemoryRagProvider extends AbstractRagProvider {
|
|
4
|
+
name;
|
|
5
|
+
kind = 'vector';
|
|
6
|
+
editable;
|
|
7
|
+
supportedScopes = ['session'];
|
|
8
|
+
inMemoryCfg;
|
|
9
|
+
constructor(cfg) {
|
|
10
|
+
super();
|
|
11
|
+
this.name = cfg.name;
|
|
12
|
+
this.editable = cfg.editable ?? true;
|
|
13
|
+
this.inMemoryCfg = cfg.inMemoryRagConfig;
|
|
14
|
+
if (cfg.idStrategyFactory)
|
|
15
|
+
this.idStrategyFactory = cfg.idStrategyFactory;
|
|
16
|
+
}
|
|
17
|
+
async createCollection(_name, opts) {
|
|
18
|
+
const scopeCheck = this.checkScope(opts.scope);
|
|
19
|
+
if (!scopeCheck.ok)
|
|
20
|
+
return scopeCheck;
|
|
21
|
+
const rag = new InMemoryRag(this.inMemoryCfg);
|
|
22
|
+
const editor = this.buildEditor(rag, this.pickIdStrategy(opts));
|
|
23
|
+
return { ok: true, value: { rag, editor } };
|
|
24
|
+
}
|
|
25
|
+
}
|
|
26
|
+
//# sourceMappingURL=in-memory-rag-provider.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"in-memory-rag-provider.js","sourceRoot":"","sources":["../../../../src/smart-agent/rag/providers/in-memory-rag-provider.ts"],"names":[],"mappings":"AAOA,OAAO,EAAE,WAAW,EAA0B,MAAM,qBAAqB,CAAC;AAC1E,OAAO,EAAE,mBAAmB,EAAE,MAAM,oBAAoB,CAAC;AAazD,MAAM,OAAO,mBAAoB,SAAQ,mBAAmB;IACjD,IAAI,CAAS;IACb,IAAI,GAAG,QAAQ,CAAC;IAChB,QAAQ,CAAU;IAClB,eAAe,GAAG,CAAC,SAAS,CAAU,CAAC;IAE/B,WAAW,CAAqB;IAEjD,YAAY,GAA8B;QACxC,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,IAAI,GAAG,GAAG,CAAC,IAAI,CAAC;QACrB,IAAI,CAAC,QAAQ,GAAG,GAAG,CAAC,QAAQ,IAAI,IAAI,CAAC;QACrC,IAAI,CAAC,WAAW,GAAG,GAAG,CAAC,iBAAiB,CAAC;QACzC,IAAI,GAAG,CAAC,iBAAiB;YAAE,IAAI,CAAC,iBAAiB,GAAG,GAAG,CAAC,iBAAiB,CAAC;IAC5E,CAAC;IAED,KAAK,CAAC,gBAAgB,CACpB,KAAa,EACb,IAIC;QAED,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC/C,IAAI,CAAC,UAAU,CAAC,EAAE;YAAE,OAAO,UAAU,CAAC;QACtC,MAAM,GAAG,GAAG,IAAI,WAAW,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC9C,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,EAAE,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,CAAC;QAChE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC;IAC9C,CAAC;CACF"}
|
|
@@ -0,0 +1,6 @@
|
|
|
1
|
+
export { AbstractRagProvider } from './base-provider.js';
|
|
2
|
+
export { InMemoryRagProvider, type InMemoryRagProviderConfig, } from './in-memory-rag-provider.js';
|
|
3
|
+
export { QdrantRagProvider, type QdrantRagProviderConfig, } from './qdrant-rag-provider.js';
|
|
4
|
+
export { SimpleRagProviderRegistry } from './simple-provider-registry.js';
|
|
5
|
+
export { VectorRagProvider, type VectorRagProviderConfig, } from './vector-rag-provider.js';
|
|
6
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/smart-agent/rag/providers/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAE,MAAM,oBAAoB,CAAC;AACzD,OAAO,EACL,mBAAmB,EACnB,KAAK,yBAAyB,GAC/B,MAAM,6BAA6B,CAAC;AACrC,OAAO,EACL,iBAAiB,EACjB,KAAK,uBAAuB,GAC7B,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,yBAAyB,EAAE,MAAM,+BAA+B,CAAC;AAC1E,OAAO,EACL,iBAAiB,EACjB,KAAK,uBAAuB,GAC7B,MAAM,0BAA0B,CAAC"}
|
|
@@ -0,0 +1,6 @@
|
|
|
1
|
+
export { AbstractRagProvider } from './base-provider.js';
|
|
2
|
+
export { InMemoryRagProvider, } from './in-memory-rag-provider.js';
|
|
3
|
+
export { QdrantRagProvider, } from './qdrant-rag-provider.js';
|
|
4
|
+
export { SimpleRagProviderRegistry } from './simple-provider-registry.js';
|
|
5
|
+
export { VectorRagProvider, } from './vector-rag-provider.js';
|
|
6
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/smart-agent/rag/providers/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAE,MAAM,oBAAoB,CAAC;AACzD,OAAO,EACL,mBAAmB,GAEpB,MAAM,6BAA6B,CAAC;AACrC,OAAO,EACL,iBAAiB,GAElB,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,yBAAyB,EAAE,MAAM,+BAA+B,CAAC;AAC1E,OAAO,EACL,iBAAiB,GAElB,MAAM,0BAA0B,CAAC"}
|
|
@@ -0,0 +1,39 @@
|
|
|
1
|
+
import type { IEmbedder, IIdStrategy, IRag, IRagEditor, RagCollectionScope } from '../../interfaces/rag.js';
|
|
2
|
+
import { RagError, type Result } from '../../interfaces/types.js';
|
|
3
|
+
import { AbstractRagProvider } from './base-provider.js';
|
|
4
|
+
export interface QdrantRagProviderConfig {
|
|
5
|
+
name: string;
|
|
6
|
+
url: string;
|
|
7
|
+
apiKey?: string;
|
|
8
|
+
embedder: IEmbedder;
|
|
9
|
+
editable?: boolean;
|
|
10
|
+
timeoutMs?: number;
|
|
11
|
+
supportedScopes?: readonly RagCollectionScope[];
|
|
12
|
+
idStrategyFactory?: (opts: {
|
|
13
|
+
scope: RagCollectionScope;
|
|
14
|
+
sessionId?: string;
|
|
15
|
+
userId?: string;
|
|
16
|
+
}) => IIdStrategy;
|
|
17
|
+
}
|
|
18
|
+
export declare class QdrantRagProvider extends AbstractRagProvider {
|
|
19
|
+
readonly name: string;
|
|
20
|
+
readonly kind = "vector";
|
|
21
|
+
readonly editable: boolean;
|
|
22
|
+
readonly supportedScopes: readonly RagCollectionScope[];
|
|
23
|
+
private readonly url;
|
|
24
|
+
private readonly apiKey?;
|
|
25
|
+
private readonly embedder;
|
|
26
|
+
private readonly timeoutMs?;
|
|
27
|
+
constructor(cfg: QdrantRagProviderConfig);
|
|
28
|
+
createCollection(name: string, opts: {
|
|
29
|
+
scope: RagCollectionScope;
|
|
30
|
+
sessionId?: string;
|
|
31
|
+
userId?: string;
|
|
32
|
+
}): Promise<Result<{
|
|
33
|
+
rag: IRag;
|
|
34
|
+
editor: IRagEditor;
|
|
35
|
+
}, RagError>>;
|
|
36
|
+
deleteCollection(name: string): Promise<Result<void, RagError>>;
|
|
37
|
+
listCollections(): Promise<Result<string[], RagError>>;
|
|
38
|
+
}
|
|
39
|
+
//# sourceMappingURL=qdrant-rag-provider.d.ts.map
|