@code-rag/core 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 +19 -0
- package/dist/auth/audit-log.d.ts +35 -0
- package/dist/auth/audit-log.js +110 -0
- package/dist/auth/audit-log.js.map +1 -0
- package/dist/auth/audit-log.test.d.ts +1 -0
- package/dist/auth/audit-log.test.js +261 -0
- package/dist/auth/audit-log.test.js.map +1 -0
- package/dist/auth/index.d.ts +6 -0
- package/dist/auth/index.js +5 -0
- package/dist/auth/index.js.map +1 -0
- package/dist/auth/oidc-provider.d.ts +49 -0
- package/dist/auth/oidc-provider.js +358 -0
- package/dist/auth/oidc-provider.js.map +1 -0
- package/dist/auth/oidc-provider.test.d.ts +1 -0
- package/dist/auth/oidc-provider.test.js +520 -0
- package/dist/auth/oidc-provider.test.js.map +1 -0
- package/dist/auth/rbac.d.ts +29 -0
- package/dist/auth/rbac.js +75 -0
- package/dist/auth/rbac.js.map +1 -0
- package/dist/auth/rbac.test.d.ts +1 -0
- package/dist/auth/rbac.test.js +224 -0
- package/dist/auth/rbac.test.js.map +1 -0
- package/dist/auth/saml-provider.d.ts +51 -0
- package/dist/auth/saml-provider.js +355 -0
- package/dist/auth/saml-provider.js.map +1 -0
- package/dist/auth/saml-provider.test.d.ts +1 -0
- package/dist/auth/saml-provider.test.js +422 -0
- package/dist/auth/saml-provider.test.js.map +1 -0
- package/dist/auth/types.d.ts +81 -0
- package/dist/auth/types.js +11 -0
- package/dist/auth/types.js.map +1 -0
- package/dist/auth/types.test.d.ts +1 -0
- package/dist/auth/types.test.js +147 -0
- package/dist/auth/types.test.js.map +1 -0
- package/dist/backlog/ab-reference-scanner.d.ts +10 -0
- package/dist/backlog/ab-reference-scanner.js +22 -0
- package/dist/backlog/ab-reference-scanner.js.map +1 -0
- package/dist/backlog/ab-reference-scanner.test.d.ts +1 -0
- package/dist/backlog/ab-reference-scanner.test.js +83 -0
- package/dist/backlog/ab-reference-scanner.test.js.map +1 -0
- package/dist/backlog/azure-devops-provider.d.ts +59 -0
- package/dist/backlog/azure-devops-provider.js +283 -0
- package/dist/backlog/azure-devops-provider.js.map +1 -0
- package/dist/backlog/backlog-provider.d.ts +13 -0
- package/dist/backlog/backlog-provider.js +6 -0
- package/dist/backlog/backlog-provider.js.map +1 -0
- package/dist/backlog/backlog-provider.test.d.ts +1 -0
- package/dist/backlog/backlog-provider.test.js +426 -0
- package/dist/backlog/backlog-provider.test.js.map +1 -0
- package/dist/backlog/clickup-provider.d.ts +55 -0
- package/dist/backlog/clickup-provider.js +301 -0
- package/dist/backlog/clickup-provider.js.map +1 -0
- package/dist/backlog/clickup-provider.test.d.ts +1 -0
- package/dist/backlog/clickup-provider.test.js +426 -0
- package/dist/backlog/clickup-provider.test.js.map +1 -0
- package/dist/backlog/clickup-reference-scanner.d.ts +10 -0
- package/dist/backlog/clickup-reference-scanner.js +32 -0
- package/dist/backlog/clickup-reference-scanner.js.map +1 -0
- package/dist/backlog/clickup-reference-scanner.test.d.ts +1 -0
- package/dist/backlog/clickup-reference-scanner.test.js +92 -0
- package/dist/backlog/clickup-reference-scanner.test.js.map +1 -0
- package/dist/backlog/code-linker.d.ts +63 -0
- package/dist/backlog/code-linker.js +90 -0
- package/dist/backlog/code-linker.js.map +1 -0
- package/dist/backlog/code-linker.test.d.ts +1 -0
- package/dist/backlog/code-linker.test.js +325 -0
- package/dist/backlog/code-linker.test.js.map +1 -0
- package/dist/backlog/index.d.ts +14 -0
- package/dist/backlog/index.js +8 -0
- package/dist/backlog/index.js.map +1 -0
- package/dist/backlog/jira-provider.d.ts +60 -0
- package/dist/backlog/jira-provider.js +272 -0
- package/dist/backlog/jira-provider.js.map +1 -0
- package/dist/backlog/jira-provider.test.d.ts +1 -0
- package/dist/backlog/jira-provider.test.js +449 -0
- package/dist/backlog/jira-provider.test.js.map +1 -0
- package/dist/backlog/jira-reference-scanner.d.ts +11 -0
- package/dist/backlog/jira-reference-scanner.js +26 -0
- package/dist/backlog/jira-reference-scanner.js.map +1 -0
- package/dist/backlog/jira-reference-scanner.test.d.ts +1 -0
- package/dist/backlog/jira-reference-scanner.test.js +127 -0
- package/dist/backlog/jira-reference-scanner.test.js.map +1 -0
- package/dist/backlog/types.d.ts +22 -0
- package/dist/backlog/types.js +1 -0
- package/dist/backlog/types.js.map +1 -0
- package/dist/chunker/ast-chunker.d.ts +45 -0
- package/dist/chunker/ast-chunker.js +292 -0
- package/dist/chunker/ast-chunker.js.map +1 -0
- package/dist/chunker/ast-chunker.test.d.ts +1 -0
- package/dist/chunker/ast-chunker.test.js +391 -0
- package/dist/chunker/ast-chunker.test.js.map +1 -0
- package/dist/chunker/chunker.d.ts +8 -0
- package/dist/chunker/chunker.js +1 -0
- package/dist/chunker/chunker.js.map +1 -0
- package/dist/chunker/index.d.ts +3 -0
- package/dist/chunker/index.js +2 -0
- package/dist/chunker/index.js.map +1 -0
- package/dist/config/config-parser.d.ts +15 -0
- package/dist/config/config-parser.js +283 -0
- package/dist/config/config-parser.js.map +1 -0
- package/dist/config/config-parser.test.d.ts +1 -0
- package/dist/config/config-parser.test.js +699 -0
- package/dist/config/config-parser.test.js.map +1 -0
- package/dist/docs/confluence-provider.d.ts +121 -0
- package/dist/docs/confluence-provider.js +459 -0
- package/dist/docs/confluence-provider.js.map +1 -0
- package/dist/docs/confluence-provider.test.d.ts +1 -0
- package/dist/docs/confluence-provider.test.js +765 -0
- package/dist/docs/confluence-provider.test.js.map +1 -0
- package/dist/docs/index.d.ts +4 -0
- package/dist/docs/index.js +2 -0
- package/dist/docs/index.js.map +1 -0
- package/dist/docs/sharepoint-provider.d.ts +150 -0
- package/dist/docs/sharepoint-provider.js +637 -0
- package/dist/docs/sharepoint-provider.js.map +1 -0
- package/dist/docs/sharepoint-provider.test.d.ts +1 -0
- package/dist/docs/sharepoint-provider.test.js +873 -0
- package/dist/docs/sharepoint-provider.test.js.map +1 -0
- package/dist/embedding/bm25-index.d.ts +12 -0
- package/dist/embedding/bm25-index.js +89 -0
- package/dist/embedding/bm25-index.js.map +1 -0
- package/dist/embedding/bm25-index.test.d.ts +1 -0
- package/dist/embedding/bm25-index.test.js +289 -0
- package/dist/embedding/bm25-index.test.js.map +1 -0
- package/dist/embedding/hybrid-search.d.ts +13 -0
- package/dist/embedding/hybrid-search.js +124 -0
- package/dist/embedding/hybrid-search.js.map +1 -0
- package/dist/embedding/hybrid-search.test.d.ts +1 -0
- package/dist/embedding/hybrid-search.test.js +266 -0
- package/dist/embedding/hybrid-search.test.js.map +1 -0
- package/dist/embedding/index.d.ts +11 -0
- package/dist/embedding/index.js +7 -0
- package/dist/embedding/index.js.map +1 -0
- package/dist/embedding/lancedb-store.d.ts +21 -0
- package/dist/embedding/lancedb-store.js +172 -0
- package/dist/embedding/lancedb-store.js.map +1 -0
- package/dist/embedding/lancedb-store.test.d.ts +1 -0
- package/dist/embedding/lancedb-store.test.js +268 -0
- package/dist/embedding/lancedb-store.test.js.map +1 -0
- package/dist/embedding/model-lifecycle-manager.d.ts +83 -0
- package/dist/embedding/model-lifecycle-manager.js +419 -0
- package/dist/embedding/model-lifecycle-manager.js.map +1 -0
- package/dist/embedding/model-lifecycle-manager.test.d.ts +1 -0
- package/dist/embedding/model-lifecycle-manager.test.js +642 -0
- package/dist/embedding/model-lifecycle-manager.test.js.map +1 -0
- package/dist/embedding/ollama-embedding-provider.d.ts +16 -0
- package/dist/embedding/ollama-embedding-provider.js +74 -0
- package/dist/embedding/ollama-embedding-provider.js.map +1 -0
- package/dist/embedding/ollama-embedding-provider.test.d.ts +1 -0
- package/dist/embedding/ollama-embedding-provider.test.js +198 -0
- package/dist/embedding/ollama-embedding-provider.test.js.map +1 -0
- package/dist/embedding/openai-compatible-embedding-provider.d.ts +19 -0
- package/dist/embedding/openai-compatible-embedding-provider.js +108 -0
- package/dist/embedding/openai-compatible-embedding-provider.js.map +1 -0
- package/dist/embedding/openai-compatible-embedding-provider.test.d.ts +1 -0
- package/dist/embedding/openai-compatible-embedding-provider.test.js +456 -0
- package/dist/embedding/openai-compatible-embedding-provider.test.js.map +1 -0
- package/dist/embedding/qdrant-store.d.ts +28 -0
- package/dist/embedding/qdrant-store.js +174 -0
- package/dist/embedding/qdrant-store.js.map +1 -0
- package/dist/embedding/qdrant-store.test.d.ts +1 -0
- package/dist/embedding/qdrant-store.test.js +359 -0
- package/dist/embedding/qdrant-store.test.js.map +1 -0
- package/dist/enrichment/index.d.ts +4 -0
- package/dist/enrichment/index.js +2 -0
- package/dist/enrichment/index.js.map +1 -0
- package/dist/enrichment/nl-enricher.d.ts +16 -0
- package/dist/enrichment/nl-enricher.js +47 -0
- package/dist/enrichment/nl-enricher.js.map +1 -0
- package/dist/enrichment/nl-enricher.test.d.ts +1 -0
- package/dist/enrichment/nl-enricher.test.js +154 -0
- package/dist/enrichment/nl-enricher.test.js.map +1 -0
- package/dist/enrichment/ollama-client.d.ts +18 -0
- package/dist/enrichment/ollama-client.js +55 -0
- package/dist/enrichment/ollama-client.js.map +1 -0
- package/dist/enrichment/ollama-client.test.d.ts +1 -0
- package/dist/enrichment/ollama-client.test.js +129 -0
- package/dist/enrichment/ollama-client.test.js.map +1 -0
- package/dist/git/git-client.d.ts +22 -0
- package/dist/git/git-client.js +6 -0
- package/dist/git/git-client.js.map +1 -0
- package/dist/git/git-client.test.d.ts +1 -0
- package/dist/git/git-client.test.js +200 -0
- package/dist/git/git-client.test.js.map +1 -0
- package/dist/git/ignore-filter.d.ts +2 -0
- package/dist/git/ignore-filter.js +31 -0
- package/dist/git/ignore-filter.js.map +1 -0
- package/dist/git/ignore-filter.test.d.ts +1 -0
- package/dist/git/ignore-filter.test.js +87 -0
- package/dist/git/ignore-filter.test.js.map +1 -0
- package/dist/git/index.d.ts +4 -0
- package/dist/git/index.js +3 -0
- package/dist/git/index.js.map +1 -0
- package/dist/git/simple-git-client.d.ts +12 -0
- package/dist/git/simple-git-client.js +138 -0
- package/dist/git/simple-git-client.js.map +1 -0
- package/dist/graph/cross-repo-resolver.d.ts +50 -0
- package/dist/graph/cross-repo-resolver.js +315 -0
- package/dist/graph/cross-repo-resolver.js.map +1 -0
- package/dist/graph/cross-repo-resolver.test.d.ts +1 -0
- package/dist/graph/cross-repo-resolver.test.js +548 -0
- package/dist/graph/cross-repo-resolver.test.js.map +1 -0
- package/dist/graph/dependency-graph.d.ts +44 -0
- package/dist/graph/dependency-graph.js +108 -0
- package/dist/graph/dependency-graph.js.map +1 -0
- package/dist/graph/dependency-graph.test.d.ts +1 -0
- package/dist/graph/dependency-graph.test.js +276 -0
- package/dist/graph/dependency-graph.test.js.map +1 -0
- package/dist/graph/graph-builder.d.ts +11 -0
- package/dist/graph/graph-builder.js +113 -0
- package/dist/graph/graph-builder.js.map +1 -0
- package/dist/graph/graph-builder.test.d.ts +1 -0
- package/dist/graph/graph-builder.test.js +178 -0
- package/dist/graph/graph-builder.test.js.map +1 -0
- package/dist/graph/import-resolver.d.ts +11 -0
- package/dist/graph/import-resolver.js +199 -0
- package/dist/graph/import-resolver.js.map +1 -0
- package/dist/graph/import-resolver.test.d.ts +1 -0
- package/dist/graph/import-resolver.test.js +282 -0
- package/dist/graph/import-resolver.test.js.map +1 -0
- package/dist/graph/index.d.ts +7 -0
- package/dist/graph/index.js +4 -0
- package/dist/graph/index.js.map +1 -0
- package/dist/index.d.ts +31 -0
- package/dist/index.js +15 -0
- package/dist/index.js.map +1 -0
- package/dist/indexer/file-scanner.d.ts +34 -0
- package/dist/indexer/file-scanner.js +69 -0
- package/dist/indexer/file-scanner.js.map +1 -0
- package/dist/indexer/file-scanner.test.d.ts +1 -0
- package/dist/indexer/file-scanner.test.js +110 -0
- package/dist/indexer/file-scanner.test.js.map +1 -0
- package/dist/indexer/file-watcher.d.ts +79 -0
- package/dist/indexer/file-watcher.js +148 -0
- package/dist/indexer/incremental-indexer.d.ts +67 -0
- package/dist/indexer/incremental-indexer.js +142 -0
- package/dist/indexer/incremental-indexer.js.map +1 -0
- package/dist/indexer/incremental-indexer.test.d.ts +1 -0
- package/dist/indexer/incremental-indexer.test.js +266 -0
- package/dist/indexer/incremental-indexer.test.js.map +1 -0
- package/dist/indexer/index-check.d.ts +22 -0
- package/dist/indexer/index-check.js +74 -0
- package/dist/indexer/index-check.js.map +1 -0
- package/dist/indexer/index-check.test.d.ts +1 -0
- package/dist/indexer/index-check.test.js +100 -0
- package/dist/indexer/index-check.test.js.map +1 -0
- package/dist/indexer/index-state.d.ts +61 -0
- package/dist/indexer/index-state.js +82 -0
- package/dist/indexer/index-state.js.map +1 -0
- package/dist/indexer/index-state.test.d.ts +1 -0
- package/dist/indexer/index-state.test.js +140 -0
- package/dist/indexer/index-state.test.js.map +1 -0
- package/dist/indexer/index.d.ts +12 -0
- package/dist/indexer/index.js +6 -0
- package/dist/indexer/index.js.map +1 -0
- package/dist/indexer/multi-repo-indexer.d.ts +63 -0
- package/dist/indexer/multi-repo-indexer.js +144 -0
- package/dist/indexer/multi-repo-indexer.js.map +1 -0
- package/dist/indexer/multi-repo-indexer.test.d.ts +1 -0
- package/dist/indexer/multi-repo-indexer.test.js +238 -0
- package/dist/indexer/multi-repo-indexer.test.js.map +1 -0
- package/dist/parser/index.d.ts +4 -0
- package/dist/parser/index.js +3 -0
- package/dist/parser/index.js.map +1 -0
- package/dist/parser/language-registry.d.ts +46 -0
- package/dist/parser/language-registry.js +219 -0
- package/dist/parser/language-registry.js.map +1 -0
- package/dist/parser/language-registry.test.d.ts +1 -0
- package/dist/parser/language-registry.test.js +225 -0
- package/dist/parser/language-registry.test.js.map +1 -0
- package/dist/parser/markdown-parser.d.ts +124 -0
- package/dist/parser/markdown-parser.js +487 -0
- package/dist/parser/markdown-parser.js.map +1 -0
- package/dist/parser/markdown-parser.test.d.ts +1 -0
- package/dist/parser/markdown-parser.test.js +600 -0
- package/dist/parser/markdown-parser.test.js.map +1 -0
- package/dist/parser/tree-sitter-parser.d.ts +32 -0
- package/dist/parser/tree-sitter-parser.js +146 -0
- package/dist/parser/tree-sitter-parser.js.map +1 -0
- package/dist/retrieval/context-expander.d.ts +51 -0
- package/dist/retrieval/context-expander.js +218 -0
- package/dist/retrieval/context-expander.js.map +1 -0
- package/dist/retrieval/context-expander.test.d.ts +1 -0
- package/dist/retrieval/context-expander.test.js +339 -0
- package/dist/retrieval/context-expander.test.js.map +1 -0
- package/dist/retrieval/cross-encoder-reranker.d.ts +16 -0
- package/dist/retrieval/cross-encoder-reranker.js +90 -0
- package/dist/retrieval/cross-encoder-reranker.js.map +1 -0
- package/dist/retrieval/cross-encoder-reranker.test.d.ts +1 -0
- package/dist/retrieval/cross-encoder-reranker.test.js +305 -0
- package/dist/retrieval/cross-encoder-reranker.test.js.map +1 -0
- package/dist/retrieval/index.d.ts +8 -0
- package/dist/retrieval/index.js +4 -0
- package/dist/retrieval/index.js.map +1 -0
- package/dist/retrieval/query-analyzer.d.ts +29 -0
- package/dist/retrieval/query-analyzer.js +238 -0
- package/dist/retrieval/query-analyzer.js.map +1 -0
- package/dist/retrieval/query-analyzer.test.d.ts +1 -0
- package/dist/retrieval/query-analyzer.test.js +236 -0
- package/dist/retrieval/query-analyzer.test.js.map +1 -0
- package/dist/retrieval/token-budget.d.ts +51 -0
- package/dist/retrieval/token-budget.js +141 -0
- package/dist/retrieval/token-budget.js.map +1 -0
- package/dist/retrieval/token-budget.test.d.ts +1 -0
- package/dist/retrieval/token-budget.test.js +404 -0
- package/dist/retrieval/token-budget.test.js.map +1 -0
- package/dist/storage/azure-blob-provider.d.ts +19 -0
- package/dist/storage/azure-blob-provider.js +199 -0
- package/dist/storage/azure-blob-provider.js.map +1 -0
- package/dist/storage/azure-blob-provider.test.d.ts +1 -0
- package/dist/storage/azure-blob-provider.test.js +250 -0
- package/dist/storage/azure-blob-provider.test.js.map +1 -0
- package/dist/storage/gcs-provider.d.ts +22 -0
- package/dist/storage/gcs-provider.js +241 -0
- package/dist/storage/gcs-provider.js.map +1 -0
- package/dist/storage/gcs-provider.test.d.ts +1 -0
- package/dist/storage/gcs-provider.test.js +299 -0
- package/dist/storage/gcs-provider.test.js.map +1 -0
- package/dist/storage/index.d.ts +5 -0
- package/dist/storage/index.js +4 -0
- package/dist/storage/index.js.map +1 -0
- package/dist/storage/s3-provider.d.ts +21 -0
- package/dist/storage/s3-provider.js +220 -0
- package/dist/storage/s3-provider.js.map +1 -0
- package/dist/storage/s3-provider.test.d.ts +1 -0
- package/dist/storage/s3-provider.test.js +329 -0
- package/dist/storage/s3-provider.test.js.map +1 -0
- package/dist/storage/types.d.ts +65 -0
- package/dist/storage/types.js +12 -0
- package/dist/storage/types.js.map +1 -0
- package/dist/types/chunk.d.ts +32 -0
- package/dist/types/chunk.js +1 -0
- package/dist/types/chunk.js.map +1 -0
- package/dist/types/config.d.ts +71 -0
- package/dist/types/config.js +1 -0
- package/dist/types/config.js.map +1 -0
- package/dist/types/index.d.ts +5 -0
- package/dist/types/index.js +1 -0
- package/dist/types/index.js.map +1 -0
- package/dist/types/provider.d.ts +54 -0
- package/dist/types/provider.js +36 -0
- package/dist/types/provider.js.map +1 -0
- package/dist/types/search.d.ts +27 -0
- package/dist/types/search.js +1 -0
- package/dist/types/search.js.map +1 -0
- package/package.json +70 -0
|
@@ -0,0 +1,456 @@
|
|
|
1
|
+
import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
|
|
2
|
+
import { OpenAICompatibleEmbeddingProvider } from './openai-compatible-embedding-provider.js';
|
|
3
|
+
import { EmbedError } from '../types/provider.js';
|
|
4
|
+
describe('OpenAICompatibleEmbeddingProvider', () => {
|
|
5
|
+
const originalFetch = globalThis.fetch;
|
|
6
|
+
beforeEach(() => {
|
|
7
|
+
globalThis.fetch = vi.fn();
|
|
8
|
+
});
|
|
9
|
+
afterEach(() => {
|
|
10
|
+
globalThis.fetch = originalFetch;
|
|
11
|
+
});
|
|
12
|
+
// ---------- Constructor / Config ----------
|
|
13
|
+
describe('constructor / config', () => {
|
|
14
|
+
it('should use default config values when none provided', () => {
|
|
15
|
+
const provider = new OpenAICompatibleEmbeddingProvider();
|
|
16
|
+
expect(provider.dimensions).toBe(768);
|
|
17
|
+
});
|
|
18
|
+
it('should accept custom config values', () => {
|
|
19
|
+
const provider = new OpenAICompatibleEmbeddingProvider({
|
|
20
|
+
baseUrl: 'http://remote:9999/v1',
|
|
21
|
+
model: 'custom-model',
|
|
22
|
+
dimensions: 384,
|
|
23
|
+
apiKey: 'sk-test-key',
|
|
24
|
+
maxBatchSize: 50,
|
|
25
|
+
});
|
|
26
|
+
expect(provider.dimensions).toBe(384);
|
|
27
|
+
});
|
|
28
|
+
it('should merge partial config with defaults', () => {
|
|
29
|
+
const provider = new OpenAICompatibleEmbeddingProvider({ dimensions: 512 });
|
|
30
|
+
expect(provider.dimensions).toBe(512);
|
|
31
|
+
});
|
|
32
|
+
it('should strip trailing slashes from baseUrl', async () => {
|
|
33
|
+
const mockResponse = {
|
|
34
|
+
ok: true,
|
|
35
|
+
json: vi.fn().mockResolvedValue({
|
|
36
|
+
data: [{ index: 0, embedding: [0.1, 0.2] }],
|
|
37
|
+
}),
|
|
38
|
+
};
|
|
39
|
+
vi.mocked(globalThis.fetch).mockResolvedValue(mockResponse);
|
|
40
|
+
const provider = new OpenAICompatibleEmbeddingProvider({
|
|
41
|
+
baseUrl: 'http://localhost:1234/v1///',
|
|
42
|
+
});
|
|
43
|
+
await provider.embed(['test']);
|
|
44
|
+
expect(globalThis.fetch).toHaveBeenCalledWith('http://localhost:1234/v1/embeddings', expect.objectContaining({ method: 'POST' }));
|
|
45
|
+
});
|
|
46
|
+
});
|
|
47
|
+
// ---------- Successful Embedding ----------
|
|
48
|
+
describe('embed — success', () => {
|
|
49
|
+
it('should return embeddings on success', async () => {
|
|
50
|
+
const mockResponse = {
|
|
51
|
+
ok: true,
|
|
52
|
+
json: vi.fn().mockResolvedValue({
|
|
53
|
+
object: 'list',
|
|
54
|
+
data: [
|
|
55
|
+
{ object: 'embedding', index: 0, embedding: [0.1, 0.2, 0.3] },
|
|
56
|
+
{ object: 'embedding', index: 1, embedding: [0.4, 0.5, 0.6] },
|
|
57
|
+
],
|
|
58
|
+
model: 'nomic-embed-text',
|
|
59
|
+
usage: { prompt_tokens: 10, total_tokens: 10 },
|
|
60
|
+
}),
|
|
61
|
+
};
|
|
62
|
+
vi.mocked(globalThis.fetch).mockResolvedValue(mockResponse);
|
|
63
|
+
const provider = new OpenAICompatibleEmbeddingProvider();
|
|
64
|
+
const result = await provider.embed(['hello', 'world']);
|
|
65
|
+
expect(result.isOk()).toBe(true);
|
|
66
|
+
if (result.isOk()) {
|
|
67
|
+
expect(result.value).toEqual([
|
|
68
|
+
[0.1, 0.2, 0.3],
|
|
69
|
+
[0.4, 0.5, 0.6],
|
|
70
|
+
]);
|
|
71
|
+
}
|
|
72
|
+
});
|
|
73
|
+
it('should return empty array for empty input', async () => {
|
|
74
|
+
const provider = new OpenAICompatibleEmbeddingProvider();
|
|
75
|
+
const result = await provider.embed([]);
|
|
76
|
+
expect(result.isOk()).toBe(true);
|
|
77
|
+
if (result.isOk()) {
|
|
78
|
+
expect(result.value).toEqual([]);
|
|
79
|
+
}
|
|
80
|
+
expect(globalThis.fetch).not.toHaveBeenCalled();
|
|
81
|
+
});
|
|
82
|
+
it('should sort response data by index for correct ordering', async () => {
|
|
83
|
+
// Simulate out-of-order response
|
|
84
|
+
const mockResponse = {
|
|
85
|
+
ok: true,
|
|
86
|
+
json: vi.fn().mockResolvedValue({
|
|
87
|
+
data: [
|
|
88
|
+
{ index: 2, embedding: [0.7, 0.8] },
|
|
89
|
+
{ index: 0, embedding: [0.1, 0.2] },
|
|
90
|
+
{ index: 1, embedding: [0.4, 0.5] },
|
|
91
|
+
],
|
|
92
|
+
}),
|
|
93
|
+
};
|
|
94
|
+
vi.mocked(globalThis.fetch).mockResolvedValue(mockResponse);
|
|
95
|
+
const provider = new OpenAICompatibleEmbeddingProvider();
|
|
96
|
+
const result = await provider.embed(['a', 'b', 'c']);
|
|
97
|
+
expect(result.isOk()).toBe(true);
|
|
98
|
+
if (result.isOk()) {
|
|
99
|
+
expect(result.value).toEqual([
|
|
100
|
+
[0.1, 0.2],
|
|
101
|
+
[0.4, 0.5],
|
|
102
|
+
[0.7, 0.8],
|
|
103
|
+
]);
|
|
104
|
+
}
|
|
105
|
+
});
|
|
106
|
+
});
|
|
107
|
+
// ---------- Request format ----------
|
|
108
|
+
describe('embed — request format', () => {
|
|
109
|
+
it('should POST to {baseUrl}/embeddings with correct body', async () => {
|
|
110
|
+
const mockResponse = {
|
|
111
|
+
ok: true,
|
|
112
|
+
json: vi.fn().mockResolvedValue({
|
|
113
|
+
data: [{ index: 0, embedding: [1, 2, 3] }],
|
|
114
|
+
}),
|
|
115
|
+
};
|
|
116
|
+
vi.mocked(globalThis.fetch).mockResolvedValue(mockResponse);
|
|
117
|
+
const provider = new OpenAICompatibleEmbeddingProvider({
|
|
118
|
+
baseUrl: 'http://localhost:11434/v1',
|
|
119
|
+
model: 'nomic-embed-text',
|
|
120
|
+
});
|
|
121
|
+
await provider.embed(['test']);
|
|
122
|
+
expect(globalThis.fetch).toHaveBeenCalledWith('http://localhost:11434/v1/embeddings', expect.objectContaining({
|
|
123
|
+
method: 'POST',
|
|
124
|
+
headers: { 'Content-Type': 'application/json' },
|
|
125
|
+
body: JSON.stringify({
|
|
126
|
+
input: ['test'],
|
|
127
|
+
model: 'nomic-embed-text',
|
|
128
|
+
}),
|
|
129
|
+
}));
|
|
130
|
+
});
|
|
131
|
+
it('should include Authorization header when apiKey is provided', async () => {
|
|
132
|
+
const mockResponse = {
|
|
133
|
+
ok: true,
|
|
134
|
+
json: vi.fn().mockResolvedValue({
|
|
135
|
+
data: [{ index: 0, embedding: [1, 2, 3] }],
|
|
136
|
+
}),
|
|
137
|
+
};
|
|
138
|
+
vi.mocked(globalThis.fetch).mockResolvedValue(mockResponse);
|
|
139
|
+
const provider = new OpenAICompatibleEmbeddingProvider({
|
|
140
|
+
baseUrl: 'https://api.openai.com/v1',
|
|
141
|
+
apiKey: 'sk-test-key-123',
|
|
142
|
+
model: 'text-embedding-3-small',
|
|
143
|
+
});
|
|
144
|
+
await provider.embed(['test']);
|
|
145
|
+
expect(globalThis.fetch).toHaveBeenCalledWith('https://api.openai.com/v1/embeddings', expect.objectContaining({
|
|
146
|
+
headers: {
|
|
147
|
+
'Content-Type': 'application/json',
|
|
148
|
+
'Authorization': 'Bearer sk-test-key-123',
|
|
149
|
+
},
|
|
150
|
+
}));
|
|
151
|
+
});
|
|
152
|
+
it('should NOT include Authorization header when apiKey is not provided', async () => {
|
|
153
|
+
const mockResponse = {
|
|
154
|
+
ok: true,
|
|
155
|
+
json: vi.fn().mockResolvedValue({
|
|
156
|
+
data: [{ index: 0, embedding: [1, 2, 3] }],
|
|
157
|
+
}),
|
|
158
|
+
};
|
|
159
|
+
vi.mocked(globalThis.fetch).mockResolvedValue(mockResponse);
|
|
160
|
+
const provider = new OpenAICompatibleEmbeddingProvider({
|
|
161
|
+
baseUrl: 'http://localhost:1234/v1',
|
|
162
|
+
});
|
|
163
|
+
await provider.embed(['test']);
|
|
164
|
+
expect(globalThis.fetch).toHaveBeenCalledWith('http://localhost:1234/v1/embeddings', expect.objectContaining({
|
|
165
|
+
headers: { 'Content-Type': 'application/json' },
|
|
166
|
+
}));
|
|
167
|
+
});
|
|
168
|
+
it('should NOT include Authorization header when apiKey is empty string', async () => {
|
|
169
|
+
const mockResponse = {
|
|
170
|
+
ok: true,
|
|
171
|
+
json: vi.fn().mockResolvedValue({
|
|
172
|
+
data: [{ index: 0, embedding: [1, 2, 3] }],
|
|
173
|
+
}),
|
|
174
|
+
};
|
|
175
|
+
vi.mocked(globalThis.fetch).mockResolvedValue(mockResponse);
|
|
176
|
+
const provider = new OpenAICompatibleEmbeddingProvider({
|
|
177
|
+
apiKey: '',
|
|
178
|
+
});
|
|
179
|
+
await provider.embed(['test']);
|
|
180
|
+
expect(globalThis.fetch).toHaveBeenCalledWith(expect.stringContaining('/embeddings'), expect.objectContaining({
|
|
181
|
+
headers: { 'Content-Type': 'application/json' },
|
|
182
|
+
}));
|
|
183
|
+
});
|
|
184
|
+
});
|
|
185
|
+
// ---------- Batch splitting ----------
|
|
186
|
+
describe('embed — batch splitting', () => {
|
|
187
|
+
it('should split texts into batches based on maxBatchSize', async () => {
|
|
188
|
+
const texts = Array.from({ length: 250 }, (_, i) => `text_${i}`);
|
|
189
|
+
const batch1 = Array.from({ length: 100 }, (_, i) => ({
|
|
190
|
+
index: i,
|
|
191
|
+
embedding: [0.1],
|
|
192
|
+
}));
|
|
193
|
+
const batch2 = Array.from({ length: 100 }, (_, i) => ({
|
|
194
|
+
index: i,
|
|
195
|
+
embedding: [0.2],
|
|
196
|
+
}));
|
|
197
|
+
const batch3 = Array.from({ length: 50 }, (_, i) => ({
|
|
198
|
+
index: i,
|
|
199
|
+
embedding: [0.3],
|
|
200
|
+
}));
|
|
201
|
+
const fetchMock = vi.mocked(globalThis.fetch);
|
|
202
|
+
fetchMock
|
|
203
|
+
.mockResolvedValueOnce({
|
|
204
|
+
ok: true,
|
|
205
|
+
json: vi.fn().mockResolvedValue({ data: batch1 }),
|
|
206
|
+
})
|
|
207
|
+
.mockResolvedValueOnce({
|
|
208
|
+
ok: true,
|
|
209
|
+
json: vi.fn().mockResolvedValue({ data: batch2 }),
|
|
210
|
+
})
|
|
211
|
+
.mockResolvedValueOnce({
|
|
212
|
+
ok: true,
|
|
213
|
+
json: vi.fn().mockResolvedValue({ data: batch3 }),
|
|
214
|
+
});
|
|
215
|
+
const provider = new OpenAICompatibleEmbeddingProvider({ maxBatchSize: 100 });
|
|
216
|
+
const result = await provider.embed(texts);
|
|
217
|
+
expect(result.isOk()).toBe(true);
|
|
218
|
+
if (result.isOk()) {
|
|
219
|
+
expect(result.value).toHaveLength(250);
|
|
220
|
+
}
|
|
221
|
+
expect(fetchMock).toHaveBeenCalledTimes(3);
|
|
222
|
+
// Verify batch sizes
|
|
223
|
+
const firstCallBody = JSON.parse(fetchMock.mock.calls[0][1].body);
|
|
224
|
+
const secondCallBody = JSON.parse(fetchMock.mock.calls[1][1].body);
|
|
225
|
+
const thirdCallBody = JSON.parse(fetchMock.mock.calls[2][1].body);
|
|
226
|
+
expect(firstCallBody.input).toHaveLength(100);
|
|
227
|
+
expect(secondCallBody.input).toHaveLength(100);
|
|
228
|
+
expect(thirdCallBody.input).toHaveLength(50);
|
|
229
|
+
});
|
|
230
|
+
it('should respect custom maxBatchSize', async () => {
|
|
231
|
+
const texts = Array.from({ length: 15 }, (_, i) => `text_${i}`);
|
|
232
|
+
const fetchMock = vi.mocked(globalThis.fetch);
|
|
233
|
+
// 3 batches of 5
|
|
234
|
+
for (let b = 0; b < 3; b++) {
|
|
235
|
+
fetchMock.mockResolvedValueOnce({
|
|
236
|
+
ok: true,
|
|
237
|
+
json: vi.fn().mockResolvedValue({
|
|
238
|
+
data: Array.from({ length: 5 }, (_, i) => ({
|
|
239
|
+
index: i,
|
|
240
|
+
embedding: [0.1],
|
|
241
|
+
})),
|
|
242
|
+
}),
|
|
243
|
+
});
|
|
244
|
+
}
|
|
245
|
+
const provider = new OpenAICompatibleEmbeddingProvider({ maxBatchSize: 5 });
|
|
246
|
+
const result = await provider.embed(texts);
|
|
247
|
+
expect(result.isOk()).toBe(true);
|
|
248
|
+
if (result.isOk()) {
|
|
249
|
+
expect(result.value).toHaveLength(15);
|
|
250
|
+
}
|
|
251
|
+
expect(fetchMock).toHaveBeenCalledTimes(3);
|
|
252
|
+
});
|
|
253
|
+
it('should not batch when texts fit in single batch', async () => {
|
|
254
|
+
const texts = Array.from({ length: 50 }, (_, i) => `text_${i}`);
|
|
255
|
+
const batchData = Array.from({ length: 50 }, (_, i) => ({
|
|
256
|
+
index: i,
|
|
257
|
+
embedding: [0.1, 0.2],
|
|
258
|
+
}));
|
|
259
|
+
vi.mocked(globalThis.fetch).mockResolvedValueOnce({
|
|
260
|
+
ok: true,
|
|
261
|
+
json: vi.fn().mockResolvedValue({ data: batchData }),
|
|
262
|
+
});
|
|
263
|
+
const provider = new OpenAICompatibleEmbeddingProvider({ maxBatchSize: 100 });
|
|
264
|
+
const result = await provider.embed(texts);
|
|
265
|
+
expect(result.isOk()).toBe(true);
|
|
266
|
+
expect(globalThis.fetch).toHaveBeenCalledTimes(1);
|
|
267
|
+
});
|
|
268
|
+
it('should stop batching on error in any batch', async () => {
|
|
269
|
+
const texts = Array.from({ length: 200 }, (_, i) => `text_${i}`);
|
|
270
|
+
const fetchMock = vi.mocked(globalThis.fetch);
|
|
271
|
+
fetchMock
|
|
272
|
+
.mockResolvedValueOnce({
|
|
273
|
+
ok: true,
|
|
274
|
+
json: vi.fn().mockResolvedValue({
|
|
275
|
+
data: Array.from({ length: 100 }, (_, i) => ({
|
|
276
|
+
index: i,
|
|
277
|
+
embedding: [0.1],
|
|
278
|
+
})),
|
|
279
|
+
}),
|
|
280
|
+
})
|
|
281
|
+
.mockResolvedValueOnce({
|
|
282
|
+
ok: false,
|
|
283
|
+
status: 503,
|
|
284
|
+
statusText: 'Service Unavailable',
|
|
285
|
+
json: vi.fn().mockRejectedValue(new Error('not json')),
|
|
286
|
+
});
|
|
287
|
+
const provider = new OpenAICompatibleEmbeddingProvider({ maxBatchSize: 100 });
|
|
288
|
+
const result = await provider.embed(texts);
|
|
289
|
+
expect(result.isErr()).toBe(true);
|
|
290
|
+
if (result.isErr()) {
|
|
291
|
+
expect(result.error).toBeInstanceOf(EmbedError);
|
|
292
|
+
expect(result.error.message).toContain('503');
|
|
293
|
+
}
|
|
294
|
+
});
|
|
295
|
+
});
|
|
296
|
+
// ---------- Error handling ----------
|
|
297
|
+
describe('embed — error handling', () => {
|
|
298
|
+
it('should return EmbedError on non-200 status', async () => {
|
|
299
|
+
const mockResponse = {
|
|
300
|
+
ok: false,
|
|
301
|
+
status: 500,
|
|
302
|
+
statusText: 'Internal Server Error',
|
|
303
|
+
json: vi.fn().mockRejectedValue(new Error('no json')),
|
|
304
|
+
};
|
|
305
|
+
vi.mocked(globalThis.fetch).mockResolvedValue(mockResponse);
|
|
306
|
+
const provider = new OpenAICompatibleEmbeddingProvider();
|
|
307
|
+
const result = await provider.embed(['hello']);
|
|
308
|
+
expect(result.isErr()).toBe(true);
|
|
309
|
+
if (result.isErr()) {
|
|
310
|
+
expect(result.error).toBeInstanceOf(EmbedError);
|
|
311
|
+
expect(result.error.message).toContain('status 500');
|
|
312
|
+
}
|
|
313
|
+
});
|
|
314
|
+
it('should extract error message from JSON error response', async () => {
|
|
315
|
+
const mockResponse = {
|
|
316
|
+
ok: false,
|
|
317
|
+
status: 401,
|
|
318
|
+
statusText: 'Unauthorized',
|
|
319
|
+
json: vi.fn().mockResolvedValue({
|
|
320
|
+
error: {
|
|
321
|
+
message: 'Invalid API key provided',
|
|
322
|
+
type: 'invalid_request_error',
|
|
323
|
+
},
|
|
324
|
+
}),
|
|
325
|
+
};
|
|
326
|
+
vi.mocked(globalThis.fetch).mockResolvedValue(mockResponse);
|
|
327
|
+
const provider = new OpenAICompatibleEmbeddingProvider({
|
|
328
|
+
apiKey: 'bad-key',
|
|
329
|
+
});
|
|
330
|
+
const result = await provider.embed(['hello']);
|
|
331
|
+
expect(result.isErr()).toBe(true);
|
|
332
|
+
if (result.isErr()) {
|
|
333
|
+
expect(result.error).toBeInstanceOf(EmbedError);
|
|
334
|
+
expect(result.error.message).toContain('Invalid API key provided');
|
|
335
|
+
expect(result.error.message).toContain('401');
|
|
336
|
+
}
|
|
337
|
+
});
|
|
338
|
+
it('should return EmbedError on network error (ECONNREFUSED)', async () => {
|
|
339
|
+
vi.mocked(globalThis.fetch).mockRejectedValue(new Error('fetch failed: ECONNREFUSED'));
|
|
340
|
+
const provider = new OpenAICompatibleEmbeddingProvider({
|
|
341
|
+
baseUrl: 'http://localhost:9999/v1',
|
|
342
|
+
});
|
|
343
|
+
const result = await provider.embed(['hello']);
|
|
344
|
+
expect(result.isErr()).toBe(true);
|
|
345
|
+
if (result.isErr()) {
|
|
346
|
+
expect(result.error).toBeInstanceOf(EmbedError);
|
|
347
|
+
expect(result.error.message).toContain('Cannot connect');
|
|
348
|
+
expect(result.error.message).toContain('localhost:9999');
|
|
349
|
+
}
|
|
350
|
+
});
|
|
351
|
+
it('should return EmbedError on timeout', async () => {
|
|
352
|
+
vi.mocked(globalThis.fetch).mockRejectedValue(new Error('TimeoutError: The operation timed out'));
|
|
353
|
+
const provider = new OpenAICompatibleEmbeddingProvider({
|
|
354
|
+
timeout: 5000,
|
|
355
|
+
});
|
|
356
|
+
const result = await provider.embed(['hello']);
|
|
357
|
+
expect(result.isErr()).toBe(true);
|
|
358
|
+
if (result.isErr()) {
|
|
359
|
+
expect(result.error).toBeInstanceOf(EmbedError);
|
|
360
|
+
expect(result.error.message).toContain('timed out');
|
|
361
|
+
}
|
|
362
|
+
});
|
|
363
|
+
it('should return EmbedError when data is not an array', async () => {
|
|
364
|
+
const mockResponse = {
|
|
365
|
+
ok: true,
|
|
366
|
+
json: vi.fn().mockResolvedValue({
|
|
367
|
+
data: 'not-an-array',
|
|
368
|
+
}),
|
|
369
|
+
};
|
|
370
|
+
vi.mocked(globalThis.fetch).mockResolvedValue(mockResponse);
|
|
371
|
+
const provider = new OpenAICompatibleEmbeddingProvider();
|
|
372
|
+
const result = await provider.embed(['hello']);
|
|
373
|
+
expect(result.isErr()).toBe(true);
|
|
374
|
+
if (result.isErr()) {
|
|
375
|
+
expect(result.error).toBeInstanceOf(EmbedError);
|
|
376
|
+
expect(result.error.message).toContain('data is not an array');
|
|
377
|
+
}
|
|
378
|
+
});
|
|
379
|
+
it('should handle non-Error throw gracefully', async () => {
|
|
380
|
+
vi.mocked(globalThis.fetch).mockRejectedValue('string error');
|
|
381
|
+
const provider = new OpenAICompatibleEmbeddingProvider();
|
|
382
|
+
const result = await provider.embed(['hello']);
|
|
383
|
+
expect(result.isErr()).toBe(true);
|
|
384
|
+
if (result.isErr()) {
|
|
385
|
+
expect(result.error).toBeInstanceOf(EmbedError);
|
|
386
|
+
expect(result.error.message).toContain('Unknown error');
|
|
387
|
+
}
|
|
388
|
+
});
|
|
389
|
+
});
|
|
390
|
+
// ---------- Compatibility with different servers ----------
|
|
391
|
+
describe('embed — server compatibility', () => {
|
|
392
|
+
it('should work with LM Studio on localhost:1234 (default config)', async () => {
|
|
393
|
+
const mockResponse = {
|
|
394
|
+
ok: true,
|
|
395
|
+
json: vi.fn().mockResolvedValue({
|
|
396
|
+
data: [{ index: 0, embedding: [0.1, 0.2, 0.3] }],
|
|
397
|
+
model: 'nomic-embed-text',
|
|
398
|
+
usage: { prompt_tokens: 5, total_tokens: 5 },
|
|
399
|
+
}),
|
|
400
|
+
};
|
|
401
|
+
vi.mocked(globalThis.fetch).mockResolvedValue(mockResponse);
|
|
402
|
+
const provider = new OpenAICompatibleEmbeddingProvider();
|
|
403
|
+
const result = await provider.embed(['hello from LM Studio']);
|
|
404
|
+
expect(result.isOk()).toBe(true);
|
|
405
|
+
expect(globalThis.fetch).toHaveBeenCalledWith('http://localhost:1234/v1/embeddings', expect.objectContaining({ method: 'POST' }));
|
|
406
|
+
});
|
|
407
|
+
it('should work with Ollama OpenAI-compatible endpoint', async () => {
|
|
408
|
+
const mockResponse = {
|
|
409
|
+
ok: true,
|
|
410
|
+
json: vi.fn().mockResolvedValue({
|
|
411
|
+
data: [{ index: 0, embedding: [0.1, 0.2, 0.3] }],
|
|
412
|
+
}),
|
|
413
|
+
};
|
|
414
|
+
vi.mocked(globalThis.fetch).mockResolvedValue(mockResponse);
|
|
415
|
+
const provider = new OpenAICompatibleEmbeddingProvider({
|
|
416
|
+
baseUrl: 'http://localhost:11434/v1',
|
|
417
|
+
model: 'nomic-embed-text',
|
|
418
|
+
});
|
|
419
|
+
const result = await provider.embed(['hello from Ollama']);
|
|
420
|
+
expect(result.isOk()).toBe(true);
|
|
421
|
+
expect(globalThis.fetch).toHaveBeenCalledWith('http://localhost:11434/v1/embeddings', expect.objectContaining({ method: 'POST' }));
|
|
422
|
+
});
|
|
423
|
+
it('should work with OpenAI API (with API key)', async () => {
|
|
424
|
+
const mockResponse = {
|
|
425
|
+
ok: true,
|
|
426
|
+
json: vi.fn().mockResolvedValue({
|
|
427
|
+
object: 'list',
|
|
428
|
+
data: [
|
|
429
|
+
{ object: 'embedding', index: 0, embedding: Array.from({ length: 1536 }, () => 0.01) },
|
|
430
|
+
],
|
|
431
|
+
model: 'text-embedding-3-small',
|
|
432
|
+
usage: { prompt_tokens: 5, total_tokens: 5 },
|
|
433
|
+
}),
|
|
434
|
+
};
|
|
435
|
+
vi.mocked(globalThis.fetch).mockResolvedValue(mockResponse);
|
|
436
|
+
const provider = new OpenAICompatibleEmbeddingProvider({
|
|
437
|
+
baseUrl: 'https://api.openai.com/v1',
|
|
438
|
+
apiKey: 'sk-proj-xxxxx',
|
|
439
|
+
model: 'text-embedding-3-small',
|
|
440
|
+
dimensions: 1536,
|
|
441
|
+
});
|
|
442
|
+
const result = await provider.embed(['hello from OpenAI']);
|
|
443
|
+
expect(result.isOk()).toBe(true);
|
|
444
|
+
if (result.isOk()) {
|
|
445
|
+
expect(result.value[0]).toHaveLength(1536);
|
|
446
|
+
}
|
|
447
|
+
expect(globalThis.fetch).toHaveBeenCalledWith('https://api.openai.com/v1/embeddings', expect.objectContaining({
|
|
448
|
+
headers: {
|
|
449
|
+
'Content-Type': 'application/json',
|
|
450
|
+
'Authorization': 'Bearer sk-proj-xxxxx',
|
|
451
|
+
},
|
|
452
|
+
}));
|
|
453
|
+
});
|
|
454
|
+
});
|
|
455
|
+
});
|
|
456
|
+
//# sourceMappingURL=openai-compatible-embedding-provider.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"openai-compatible-embedding-provider.test.js","sourceRoot":"","sources":["../../src/embedding/openai-compatible-embedding-provider.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,iCAAiC,EAAE,MAAM,2CAA2C,CAAC;AAC9F,OAAO,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAElD,QAAQ,CAAC,mCAAmC,EAAE,GAAG,EAAE;IACjD,MAAM,aAAa,GAAG,UAAU,CAAC,KAAK,CAAC;IAEvC,UAAU,CAAC,GAAG,EAAE;QACd,UAAU,CAAC,KAAK,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,UAAU,CAAC,KAAK,GAAG,aAAa,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,6CAA6C;IAE7C,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,MAAM,QAAQ,GAAG,IAAI,iCAAiC,EAAE,CAAC;YACzD,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;YAC5C,MAAM,QAAQ,GAAG,IAAI,iCAAiC,CAAC;gBACrD,OAAO,EAAE,uBAAuB;gBAChC,KAAK,EAAE,cAAc;gBACrB,UAAU,EAAE,GAAG;gBACf,MAAM,EAAE,aAAa;gBACrB,YAAY,EAAE,EAAE;aACjB,CAAC,CAAC;YACH,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,QAAQ,GAAG,IAAI,iCAAiC,CAAC,EAAE,UAAU,EAAE,GAAG,EAAE,CAAC,CAAC;YAC5E,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,YAAY,GAAG;gBACnB,EAAE,EAAE,IAAI;gBACR,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;oBAC9B,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,CAAC;iBAC5C,CAAC;aACH,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,iBAAiB,CAC3C,YAAmC,CACpC,CAAC;YAEF,MAAM,QAAQ,GAAG,IAAI,iCAAiC,CAAC;gBACrD,OAAO,EAAE,6BAA6B;aACvC,CAAC,CAAC;YACH,MAAM,QAAQ,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;YAE/B,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAC3C,qCAAqC,EACrC,MAAM,CAAC,gBAAgB,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAC5C,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,6CAA6C;IAE7C,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;YACnD,MAAM,YAAY,GAAG;gBACnB,EAAE,EAAE,IAAI;gBACR,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;oBAC9B,MAAM,EAAE,MAAM;oBACd,IAAI,EAAE;wBACJ,EAAE,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,EAAE;wBAC7D,EAAE,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,EAAE;qBAC9D;oBACD,KAAK,EAAE,kBAAkB;oBACzB,KAAK,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE;iBAC/C,CAAC;aACH,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,iBAAiB,CAC3C,YAAmC,CACpC,CAAC;YAEF,MAAM,QAAQ,GAAG,IAAI,iCAAiC,EAAE,CAAC;YACzD,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;YAExD,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACjC,IAAI,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;gBAClB,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC;oBAC3B,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC;oBACf,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC;iBAChB,CAAC,CAAC;YACL,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,QAAQ,GAAG,IAAI,iCAAiC,EAAE,CAAC;YACzD,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YAExC,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACjC,IAAI,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;gBAClB,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YACnC,CAAC;YAED,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;YACvE,iCAAiC;YACjC,MAAM,YAAY,GAAG;gBACnB,EAAE,EAAE,IAAI;gBACR,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;oBAC9B,IAAI,EAAE;wBACJ,EAAE,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE;wBACnC,EAAE,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE;wBACnC,EAAE,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE;qBACpC;iBACF,CAAC;aACH,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,iBAAiB,CAC3C,YAAmC,CACpC,CAAC;YAEF,MAAM,QAAQ,GAAG,IAAI,iCAAiC,EAAE,CAAC;YACzD,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;YAErD,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACjC,IAAI,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;gBAClB,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC;oBAC3B,CAAC,GAAG,EAAE,GAAG,CAAC;oBACV,CAAC,GAAG,EAAE,GAAG,CAAC;oBACV,CAAC,GAAG,EAAE,GAAG,CAAC;iBACX,CAAC,CAAC;YACL,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,uCAAuC;IAEvC,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;QACtC,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,MAAM,YAAY,GAAG;gBACnB,EAAE,EAAE,IAAI;gBACR,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;oBAC9B,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;iBAC3C,CAAC;aACH,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,iBAAiB,CAC3C,YAAmC,CACpC,CAAC;YAEF,MAAM,QAAQ,GAAG,IAAI,iCAAiC,CAAC;gBACrD,OAAO,EAAE,2BAA2B;gBACpC,KAAK,EAAE,kBAAkB;aAC1B,CAAC,CAAC;YACH,MAAM,QAAQ,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;YAE/B,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAC3C,sCAAsC,EACtC,MAAM,CAAC,gBAAgB,CAAC;gBACtB,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;gBAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,KAAK,EAAE,CAAC,MAAM,CAAC;oBACf,KAAK,EAAE,kBAAkB;iBAC1B,CAAC;aACH,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;YAC3E,MAAM,YAAY,GAAG;gBACnB,EAAE,EAAE,IAAI;gBACR,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;oBAC9B,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;iBAC3C,CAAC;aACH,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,iBAAiB,CAC3C,YAAmC,CACpC,CAAC;YAEF,MAAM,QAAQ,GAAG,IAAI,iCAAiC,CAAC;gBACrD,OAAO,EAAE,2BAA2B;gBACpC,MAAM,EAAE,iBAAiB;gBACzB,KAAK,EAAE,wBAAwB;aAChC,CAAC,CAAC;YACH,MAAM,QAAQ,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;YAE/B,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAC3C,sCAAsC,EACtC,MAAM,CAAC,gBAAgB,CAAC;gBACtB,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,eAAe,EAAE,wBAAwB;iBAC1C;aACF,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;YACnF,MAAM,YAAY,GAAG;gBACnB,EAAE,EAAE,IAAI;gBACR,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;oBAC9B,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;iBAC3C,CAAC;aACH,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,iBAAiB,CAC3C,YAAmC,CACpC,CAAC;YAEF,MAAM,QAAQ,GAAG,IAAI,iCAAiC,CAAC;gBACrD,OAAO,EAAE,0BAA0B;aACpC,CAAC,CAAC;YACH,MAAM,QAAQ,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;YAE/B,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAC3C,qCAAqC,EACrC,MAAM,CAAC,gBAAgB,CAAC;gBACtB,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;aAChD,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;YACnF,MAAM,YAAY,GAAG;gBACnB,EAAE,EAAE,IAAI;gBACR,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;oBAC9B,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;iBAC3C,CAAC;aACH,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,iBAAiB,CAC3C,YAAmC,CACpC,CAAC;YAEF,MAAM,QAAQ,GAAG,IAAI,iCAAiC,CAAC;gBACrD,MAAM,EAAE,EAAE;aACX,CAAC,CAAC;YACH,MAAM,QAAQ,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;YAE/B,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAC3C,MAAM,CAAC,gBAAgB,CAAC,aAAa,CAAC,EACtC,MAAM,CAAC,gBAAgB,CAAC;gBACtB,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;aAChD,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,wCAAwC;IAExC,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACvC,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YACjE,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;gBACpD,KAAK,EAAE,CAAC;gBACR,SAAS,EAAE,CAAC,GAAG,CAAC;aACjB,CAAC,CAAC,CAAC;YACJ,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;gBACpD,KAAK,EAAE,CAAC;gBACR,SAAS,EAAE,CAAC,GAAG,CAAC;aACjB,CAAC,CAAC,CAAC;YACJ,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;gBACnD,KAAK,EAAE,CAAC;gBACR,SAAS,EAAE,CAAC,GAAG,CAAC;aACjB,CAAC,CAAC,CAAC;YAEJ,MAAM,SAAS,GAAG,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YAC9C,SAAS;iBACN,qBAAqB,CAAC;gBACrB,EAAE,EAAE,IAAI;gBACR,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;aAC3B,CAAC;iBACxB,qBAAqB,CAAC;gBACrB,EAAE,EAAE,IAAI;gBACR,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;aAC3B,CAAC;iBACxB,qBAAqB,CAAC;gBACrB,EAAE,EAAE,IAAI;gBACR,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;aAC3B,CAAC,CAAC;YAE5B,MAAM,QAAQ,GAAG,IAAI,iCAAiC,CAAC,EAAE,YAAY,EAAE,GAAG,EAAE,CAAC,CAAC;YAC9E,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YAE3C,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACjC,IAAI,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;gBAClB,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC;YACzC,CAAC;YAED,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAE3C,qBAAqB;YACrB,MAAM,aAAa,GAAG,IAAI,CAAC,KAAK,CAC9B,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAE,CAAC,CAAC,CAAE,CAAC,IAAc,CACrB,CAAC;YACzB,MAAM,cAAc,GAAG,IAAI,CAAC,KAAK,CAC/B,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAE,CAAC,CAAC,CAAE,CAAC,IAAc,CACrB,CAAC;YACzB,MAAM,aAAa,GAAG,IAAI,CAAC,KAAK,CAC9B,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAE,CAAC,CAAC,CAAE,CAAC,IAAc,CACrB,CAAC;YAEzB,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC;YAC9C,MAAM,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC;YAC/C,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YAEhE,MAAM,SAAS,GAAG,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YAC9C,iBAAiB;YACjB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC3B,SAAS,CAAC,qBAAqB,CAAC;oBAC9B,EAAE,EAAE,IAAI;oBACR,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;wBAC9B,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;4BACzC,KAAK,EAAE,CAAC;4BACR,SAAS,EAAE,CAAC,GAAG,CAAC;yBACjB,CAAC,CAAC;qBACJ,CAAC;iBACoB,CAAC,CAAC;YAC5B,CAAC;YAED,MAAM,QAAQ,GAAG,IAAI,iCAAiC,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC,CAAC;YAC5E,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YAE3C,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACjC,IAAI,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;gBAClB,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;YACxC,CAAC;YACD,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;YAC/D,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YAChE,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;gBACtD,KAAK,EAAE,CAAC;gBACR,SAAS,EAAE,CAAC,GAAG,EAAE,GAAG,CAAC;aACtB,CAAC,CAAC,CAAC;YAEJ,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,qBAAqB,CAAC;gBAChD,EAAE,EAAE,IAAI;gBACR,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;aAC9B,CAAC,CAAC;YAE1B,MAAM,QAAQ,GAAG,IAAI,iCAAiC,CAAC,EAAE,YAAY,EAAE,GAAG,EAAE,CAAC,CAAC;YAC9E,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YAE3C,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACjC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YAEjE,MAAM,SAAS,GAAG,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YAC9C,SAAS;iBACN,qBAAqB,CAAC;gBACrB,EAAE,EAAE,IAAI;gBACR,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;oBAC9B,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;wBAC3C,KAAK,EAAE,CAAC;wBACR,SAAS,EAAE,CAAC,GAAG,CAAC;qBACjB,CAAC,CAAC;iBACJ,CAAC;aACoB,CAAC;iBACxB,qBAAqB,CAAC;gBACrB,EAAE,EAAE,KAAK;gBACT,MAAM,EAAE,GAAG;gBACX,UAAU,EAAE,qBAAqB;gBACjC,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,UAAU,CAAC,CAAC;aAChC,CAAC,CAAC;YAE5B,MAAM,QAAQ,GAAG,IAAI,iCAAiC,CAAC,EAAE,YAAY,EAAE,GAAG,EAAE,CAAC,CAAC;YAC9E,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YAE3C,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClC,IAAI,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;gBACnB,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC;gBAChD,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;YAChD,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,uCAAuC;IAEvC,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;QACtC,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,YAAY,GAAG;gBACnB,EAAE,EAAE,KAAK;gBACT,MAAM,EAAE,GAAG;gBACX,UAAU,EAAE,uBAAuB;gBACnC,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC;aACtD,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,iBAAiB,CAC3C,YAAmC,CACpC,CAAC;YAEF,MAAM,QAAQ,GAAG,IAAI,iCAAiC,EAAE,CAAC;YACzD,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;YAE/C,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClC,IAAI,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;gBACnB,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC;gBAChD,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;YACvD,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,MAAM,YAAY,GAAG;gBACnB,EAAE,EAAE,KAAK;gBACT,MAAM,EAAE,GAAG;gBACX,UAAU,EAAE,cAAc;gBAC1B,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;oBAC9B,KAAK,EAAE;wBACL,OAAO,EAAE,0BAA0B;wBACnC,IAAI,EAAE,uBAAuB;qBAC9B;iBACF,CAAC;aACH,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,iBAAiB,CAC3C,YAAmC,CACpC,CAAC;YAEF,MAAM,QAAQ,GAAG,IAAI,iCAAiC,CAAC;gBACrD,MAAM,EAAE,SAAS;aAClB,CAAC,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;YAE/C,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClC,IAAI,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;gBACnB,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC;gBAChD,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,0BAA0B,CAAC,CAAC;gBACnE,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;YAChD,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;YACxE,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,iBAAiB,CAC3C,IAAI,KAAK,CAAC,4BAA4B,CAAC,CACxC,CAAC;YAEF,MAAM,QAAQ,GAAG,IAAI,iCAAiC,CAAC;gBACrD,OAAO,EAAE,0BAA0B;aACpC,CAAC,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;YAE/C,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClC,IAAI,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;gBACnB,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC;gBAChD,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;gBACzD,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;YAC3D,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;YACnD,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,iBAAiB,CAC3C,IAAI,KAAK,CAAC,uCAAuC,CAAC,CACnD,CAAC;YAEF,MAAM,QAAQ,GAAG,IAAI,iCAAiC,CAAC;gBACrD,OAAO,EAAE,IAAI;aACd,CAAC,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;YAE/C,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClC,IAAI,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;gBACnB,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC;gBAChD,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;YACtD,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;YAClE,MAAM,YAAY,GAAG;gBACnB,EAAE,EAAE,IAAI;gBACR,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;oBAC9B,IAAI,EAAE,cAAc;iBACrB,CAAC;aACH,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,iBAAiB,CAC3C,YAAmC,CACpC,CAAC;YAEF,MAAM,QAAQ,GAAG,IAAI,iCAAiC,EAAE,CAAC;YACzD,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;YAE/C,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClC,IAAI,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;gBACnB,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC;gBAChD,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,sBAAsB,CAAC,CAAC;YACjE,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC;YAE9D,MAAM,QAAQ,GAAG,IAAI,iCAAiC,EAAE,CAAC;YACzD,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;YAE/C,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClC,IAAI,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;gBACnB,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC;gBAChD,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;YAC1D,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,6DAA6D;IAE7D,QAAQ,CAAC,8BAA8B,EAAE,GAAG,EAAE;QAC5C,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;YAC7E,MAAM,YAAY,GAAG;gBACnB,EAAE,EAAE,IAAI;gBACR,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;oBAC9B,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,EAAE,CAAC;oBAChD,KAAK,EAAE,kBAAkB;oBACzB,KAAK,EAAE,EAAE,aAAa,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE;iBAC7C,CAAC;aACH,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,iBAAiB,CAC3C,YAAmC,CACpC,CAAC;YAEF,MAAM,QAAQ,GAAG,IAAI,iCAAiC,EAAE,CAAC;YACzD,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC;YAE9D,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACjC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAC3C,qCAAqC,EACrC,MAAM,CAAC,gBAAgB,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAC5C,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;YAClE,MAAM,YAAY,GAAG;gBACnB,EAAE,EAAE,IAAI;gBACR,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;oBAC9B,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,EAAE,CAAC;iBACjD,CAAC;aACH,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,iBAAiB,CAC3C,YAAmC,CACpC,CAAC;YAEF,MAAM,QAAQ,GAAG,IAAI,iCAAiC,CAAC;gBACrD,OAAO,EAAE,2BAA2B;gBACpC,KAAK,EAAE,kBAAkB;aAC1B,CAAC,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC;YAE3D,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACjC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAC3C,sCAAsC,EACtC,MAAM,CAAC,gBAAgB,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAC5C,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,YAAY,GAAG;gBACnB,EAAE,EAAE,IAAI;gBACR,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;oBAC9B,MAAM,EAAE,MAAM;oBACd,IAAI,EAAE;wBACJ,EAAE,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,EAAE;qBACvF;oBACD,KAAK,EAAE,wBAAwB;oBAC/B,KAAK,EAAE,EAAE,aAAa,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE;iBAC7C,CAAC;aACH,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,iBAAiB,CAC3C,YAAmC,CACpC,CAAC;YAEF,MAAM,QAAQ,GAAG,IAAI,iCAAiC,CAAC;gBACrD,OAAO,EAAE,2BAA2B;gBACpC,MAAM,EAAE,eAAe;gBACvB,KAAK,EAAE,wBAAwB;gBAC/B,UAAU,EAAE,IAAI;aACjB,CAAC,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC;YAE3D,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACjC,IAAI,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;gBAClB,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;YAC7C,CAAC;YAED,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAC3C,sCAAsC,EACtC,MAAM,CAAC,gBAAgB,CAAC;gBACtB,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,eAAe,EAAE,sBAAsB;iBACxC;aACF,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,28 @@
|
|
|
1
|
+
import { type Result } from 'neverthrow';
|
|
2
|
+
import { StoreError, type VectorStore } from '../types/provider.js';
|
|
3
|
+
export interface QdrantConfig {
|
|
4
|
+
url?: string;
|
|
5
|
+
collectionName?: string;
|
|
6
|
+
apiKey?: string;
|
|
7
|
+
}
|
|
8
|
+
export declare class QdrantVectorStore implements VectorStore {
|
|
9
|
+
private readonly url;
|
|
10
|
+
private readonly collectionName;
|
|
11
|
+
private readonly _dimensions;
|
|
12
|
+
private readonly apiKey?;
|
|
13
|
+
private client;
|
|
14
|
+
private collectionReady;
|
|
15
|
+
constructor(dimensions: number, config?: QdrantConfig);
|
|
16
|
+
get dimensions(): number;
|
|
17
|
+
connect(): Promise<void>;
|
|
18
|
+
private ensureConnected;
|
|
19
|
+
upsert(ids: string[], embeddings: number[][], metadata: Record<string, unknown>[]): Promise<Result<void, StoreError>>;
|
|
20
|
+
query(embedding: number[], topK: number): Promise<Result<{
|
|
21
|
+
id: string;
|
|
22
|
+
score: number;
|
|
23
|
+
metadata?: Record<string, unknown>;
|
|
24
|
+
}[], StoreError>>;
|
|
25
|
+
delete(ids: string[]): Promise<Result<void, StoreError>>;
|
|
26
|
+
count(): Promise<Result<number, StoreError>>;
|
|
27
|
+
close(): void;
|
|
28
|
+
}
|