@mhalder/qdrant-mcp-server 1.6.0 → 2.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/.github/workflows/ci.yml +5 -16
- package/.nvmrc +1 -0
- package/CHANGELOG.md +30 -0
- package/README.md +86 -76
- package/build/embeddings/cohere.test.js +8 -6
- package/build/embeddings/cohere.test.js.map +1 -1
- package/build/embeddings/openai.test.js +12 -8
- package/build/embeddings/openai.test.js.map +1 -1
- package/build/index.js +26 -7
- package/build/index.js.map +1 -1
- package/build/qdrant/client.d.ts.map +1 -1
- package/build/qdrant/client.js.map +1 -1
- package/build/qdrant/client.test.js +64 -24
- package/build/qdrant/client.test.js.map +1 -1
- package/compose.yaml +57 -0
- package/examples/README.md +5 -3
- package/package.json +22 -19
- package/src/embeddings/cohere.test.ts +9 -8
- package/src/embeddings/openai.test.ts +13 -10
- package/src/index.ts +144 -51
- package/src/qdrant/client.test.ts +200 -79
- package/src/qdrant/client.ts +25 -14
- package/tsconfig.json +5 -4
- package/vitest.config.ts +1 -0
- package/docker-compose.yml +0 -22
package/.github/workflows/ci.yml
CHANGED
|
@@ -13,7 +13,7 @@ jobs:
|
|
|
13
13
|
|
|
14
14
|
strategy:
|
|
15
15
|
matrix:
|
|
16
|
-
node-version: [
|
|
16
|
+
node-version: [22.x]
|
|
17
17
|
|
|
18
18
|
steps:
|
|
19
19
|
- name: Checkout code
|
|
@@ -23,7 +23,7 @@ jobs:
|
|
|
23
23
|
uses: actions/setup-node@v4
|
|
24
24
|
with:
|
|
25
25
|
node-version: ${{ matrix.node-version }}
|
|
26
|
-
cache:
|
|
26
|
+
cache: "npm"
|
|
27
27
|
|
|
28
28
|
- name: Install dependencies
|
|
29
29
|
run: npm ci
|
|
@@ -34,24 +34,13 @@ jobs:
|
|
|
34
34
|
- name: Build project
|
|
35
35
|
run: npm run build
|
|
36
36
|
|
|
37
|
-
- name: Run
|
|
38
|
-
if: matrix.node-version != '22.x'
|
|
39
|
-
run: npm test -- --run
|
|
40
|
-
|
|
41
|
-
- name: Run provider verification tests
|
|
42
|
-
if: matrix.node-version != '22.x'
|
|
43
|
-
run: npm run test:providers
|
|
44
|
-
|
|
45
|
-
- name: Run tests with coverage (Node 22.x only)
|
|
46
|
-
if: matrix.node-version == '22.x'
|
|
37
|
+
- name: Run tests with coverage
|
|
47
38
|
run: npm run test:coverage
|
|
48
39
|
|
|
49
|
-
- name: Run provider verification tests
|
|
50
|
-
if: matrix.node-version == '22.x'
|
|
40
|
+
- name: Run provider verification tests
|
|
51
41
|
run: npm run test:providers
|
|
52
42
|
|
|
53
|
-
- name: Upload coverage to Codecov
|
|
54
|
-
if: matrix.node-version == '22.x'
|
|
43
|
+
- name: Upload coverage to Codecov
|
|
55
44
|
uses: codecov/codecov-action@v4
|
|
56
45
|
continue-on-error: true
|
|
57
46
|
with:
|
package/.nvmrc
ADDED
|
@@ -0,0 +1 @@
|
|
|
1
|
+
22
|
package/CHANGELOG.md
CHANGED
|
@@ -1,3 +1,33 @@
|
|
|
1
|
+
## 2.1.0 (2026-01-17)
|
|
2
|
+
|
|
3
|
+
* Merge pull request #42 from mhalder/feat/configurable-http-timeout ([15a5a5a](https://github.com/mhalder/qdrant-mcp-server/commit/15a5a5a)), closes [#42](https://github.com/mhalder/qdrant-mcp-server/issues/42)
|
|
4
|
+
* fix: add validation for HTTP_REQUEST_TIMEOUT_MS environment variable ([fdc169f](https://github.com/mhalder/qdrant-mcp-server/commit/fdc169f))
|
|
5
|
+
* feat: make HTTP request timeout configurable via environment variable ([09528e3](https://github.com/mhalder/qdrant-mcp-server/commit/09528e3))
|
|
6
|
+
|
|
7
|
+
## 2.0.0 (2026-01-17)
|
|
8
|
+
|
|
9
|
+
* build!: update dependencies and migrate to Podman ([bef9119](https://github.com/mhalder/qdrant-mcp-server/commit/bef9119))
|
|
10
|
+
* Merge pull request #41 from mhalder/chore/full-update-2026 ([ccfbd97](https://github.com/mhalder/qdrant-mcp-server/commit/ccfbd97)), closes [#41](https://github.com/mhalder/qdrant-mcp-server/issues/41)
|
|
11
|
+
* fix: add SELinux context labels to Podman volume mounts ([a9a0fb9](https://github.com/mhalder/qdrant-mcp-server/commit/a9a0fb9))
|
|
12
|
+
|
|
13
|
+
|
|
14
|
+
### BREAKING CHANGE
|
|
15
|
+
|
|
16
|
+
* Node.js minimum version changed from 20 to 22.
|
|
17
|
+
|
|
18
|
+
- Update all dependencies to latest versions:
|
|
19
|
+
- @modelcontextprotocol/sdk 1.0.4 → 1.25.2
|
|
20
|
+
- @qdrant/js-client-rest 1.12.0 → 1.16.2
|
|
21
|
+
- vitest 2.1.8 → 4.0.17
|
|
22
|
+
- semantic-release 24.2.9 → 25.0.2
|
|
23
|
+
- openai, cohere-ai, zod, and others
|
|
24
|
+
- Set Node.js minimum to 22.x (engines field + .nvmrc)
|
|
25
|
+
- Migrate docker-compose.yml to compose.yaml with Podman support
|
|
26
|
+
- Update tsconfig to ES2023/NodeNext
|
|
27
|
+
- Simplify CI workflow to Node 22 only
|
|
28
|
+
- Fix test mocking patterns for Vitest 4 compatibility
|
|
29
|
+
- Update documentation with Podman commands and correct test counts
|
|
30
|
+
|
|
1
31
|
## 1.6.0 (2026-01-17)
|
|
2
32
|
|
|
3
33
|
* Merge pull request #34 from No-Smoke/feature/add-qdrant-api-key-support ([c6af3ae](https://github.com/mhalder/qdrant-mcp-server/commit/c6af3ae)), closes [#34](https://github.com/mhalder/qdrant-mcp-server/issues/34)
|
package/README.md
CHANGED
|
@@ -24,8 +24,8 @@ A Model Context Protocol (MCP) server providing semantic search capabilities usi
|
|
|
24
24
|
|
|
25
25
|
### Prerequisites
|
|
26
26
|
|
|
27
|
-
- Node.js
|
|
28
|
-
-
|
|
27
|
+
- Node.js 22+
|
|
28
|
+
- Podman or Docker with Compose support
|
|
29
29
|
|
|
30
30
|
### Installation
|
|
31
31
|
|
|
@@ -35,9 +35,13 @@ git clone https://github.com/mhalder/qdrant-mcp-server.git
|
|
|
35
35
|
cd qdrant-mcp-server
|
|
36
36
|
npm install
|
|
37
37
|
|
|
38
|
-
# Start services
|
|
39
|
-
|
|
40
|
-
docker
|
|
38
|
+
# Start services (choose one)
|
|
39
|
+
podman compose up -d # Using Podman
|
|
40
|
+
docker compose up -d # Using Docker
|
|
41
|
+
|
|
42
|
+
# Pull the embedding model
|
|
43
|
+
podman exec ollama ollama pull nomic-embed-text # Podman
|
|
44
|
+
docker exec ollama ollama pull nomic-embed-text # Docker
|
|
41
45
|
|
|
42
46
|
# Build
|
|
43
47
|
npm run build
|
|
@@ -149,13 +153,13 @@ See [Advanced Configuration](#advanced-configuration) section below for all opti
|
|
|
149
153
|
|
|
150
154
|
### Code Vectorization
|
|
151
155
|
|
|
152
|
-
| Tool | Description
|
|
153
|
-
| ------------------ |
|
|
154
|
-
| `index_codebase` | Index a codebase for semantic code search with AST-aware chunking
|
|
155
|
-
| `search_code` | Search indexed codebase using natural language queries
|
|
156
|
-
| `reindex_changes` | Incrementally re-index only changed files (detects added/modified/deleted)
|
|
157
|
-
| `get_index_status` | Get indexing status and statistics for a codebase
|
|
158
|
-
| `clear_index` | Delete all indexed data for a codebase
|
|
156
|
+
| Tool | Description |
|
|
157
|
+
| ------------------ | -------------------------------------------------------------------------- |
|
|
158
|
+
| `index_codebase` | Index a codebase for semantic code search with AST-aware chunking |
|
|
159
|
+
| `search_code` | Search indexed codebase using natural language queries |
|
|
160
|
+
| `reindex_changes` | Incrementally re-index only changed files (detects added/modified/deleted) |
|
|
161
|
+
| `get_index_status` | Get indexing status and statistics for a codebase |
|
|
162
|
+
| `clear_index` | Delete all indexed data for a codebase |
|
|
159
163
|
|
|
160
164
|
### Resources
|
|
161
165
|
|
|
@@ -285,8 +289,8 @@ Intelligently index and search your codebase using semantic code search. Perfect
|
|
|
285
289
|
|
|
286
290
|
index_codebase({
|
|
287
291
|
path: "/workspace/my-app",
|
|
288
|
-
forceReindex: false
|
|
289
|
-
})
|
|
292
|
+
forceReindex: false, // Set to true to re-index from scratch
|
|
293
|
+
});
|
|
290
294
|
|
|
291
295
|
// Output:
|
|
292
296
|
// ✓ Indexed 247 files (1,823 chunks) in 45.2s
|
|
@@ -298,8 +302,8 @@ index_codebase({
|
|
|
298
302
|
search_code({
|
|
299
303
|
path: "/workspace/my-app",
|
|
300
304
|
query: "how does user authentication work?",
|
|
301
|
-
limit: 5
|
|
302
|
-
})
|
|
305
|
+
limit: 5,
|
|
306
|
+
});
|
|
303
307
|
|
|
304
308
|
// Results include file path, line numbers, and code snippets:
|
|
305
309
|
// [
|
|
@@ -323,16 +327,16 @@ search_code({
|
|
|
323
327
|
path: "/workspace/my-app",
|
|
324
328
|
query: "error handling patterns",
|
|
325
329
|
fileTypes: [".ts", ".tsx"],
|
|
326
|
-
limit: 10
|
|
327
|
-
})
|
|
330
|
+
limit: 10,
|
|
331
|
+
});
|
|
328
332
|
|
|
329
333
|
// Only search in specific directories
|
|
330
334
|
search_code({
|
|
331
335
|
path: "/workspace/my-app",
|
|
332
336
|
query: "API route handlers",
|
|
333
337
|
pathPattern: "src/api/**",
|
|
334
|
-
limit: 10
|
|
335
|
-
})
|
|
338
|
+
limit: 10,
|
|
339
|
+
});
|
|
336
340
|
```
|
|
337
341
|
|
|
338
342
|
#### Incremental Re-indexing
|
|
@@ -340,8 +344,8 @@ search_code({
|
|
|
340
344
|
```typescript
|
|
341
345
|
// After making changes to your codebase
|
|
342
346
|
reindex_changes({
|
|
343
|
-
path: "/workspace/my-app"
|
|
344
|
-
})
|
|
347
|
+
path: "/workspace/my-app",
|
|
348
|
+
});
|
|
345
349
|
|
|
346
350
|
// Output:
|
|
347
351
|
// ✓ Updated: +3 files added, ~5 files modified, -1 files deleted
|
|
@@ -352,8 +356,8 @@ reindex_changes({
|
|
|
352
356
|
|
|
353
357
|
```typescript
|
|
354
358
|
get_index_status({
|
|
355
|
-
path: "/workspace/my-app"
|
|
356
|
-
})
|
|
359
|
+
path: "/workspace/my-app",
|
|
360
|
+
});
|
|
357
361
|
|
|
358
362
|
// Output:
|
|
359
363
|
// {
|
|
@@ -369,6 +373,7 @@ get_index_status({
|
|
|
369
373
|
### Supported Languages
|
|
370
374
|
|
|
371
375
|
**Programming Languages** (35+ file types):
|
|
376
|
+
|
|
372
377
|
- **Web**: TypeScript, JavaScript, Vue, Svelte
|
|
373
378
|
- **Backend**: Python, Java, Go, Rust, Ruby, PHP
|
|
374
379
|
- **Systems**: C, C++, C#
|
|
@@ -406,11 +411,11 @@ Create a `.contextignore` file in your project root to specify additional patter
|
|
|
406
411
|
|
|
407
412
|
Typical performance on a modern laptop (Apple M1/M2 or similar):
|
|
408
413
|
|
|
409
|
-
| Codebase Size
|
|
410
|
-
|
|
411
|
-
| Small (10k LOC)
|
|
412
|
-
| Medium (100k LOC) | 500
|
|
413
|
-
| Large (500k LOC)
|
|
414
|
+
| Codebase Size | Files | Indexing Time | Search Latency |
|
|
415
|
+
| ----------------- | ----- | ------------- | -------------- |
|
|
416
|
+
| Small (10k LOC) | 50 | ~10s | <100ms |
|
|
417
|
+
| Medium (100k LOC) | 500 | ~2min | <200ms |
|
|
418
|
+
| Large (500k LOC) | 2,500 | ~10min | <500ms |
|
|
414
419
|
|
|
415
420
|
**Note**: Indexing time varies based on embedding provider. Ollama (local) is fastest for initial indexing.
|
|
416
421
|
|
|
@@ -419,6 +424,7 @@ Typical performance on a modern laptop (Apple M1/M2 or similar):
|
|
|
419
424
|
See [examples/](examples/) directory for detailed guides:
|
|
420
425
|
|
|
421
426
|
- **[Basic Usage](examples/basic/)** - Create collections, add documents, search
|
|
427
|
+
- **[Hybrid Search](examples/hybrid-search/)** - Combine semantic and keyword search
|
|
422
428
|
- **[Knowledge Base](examples/knowledge-base/)** - Structured documentation with metadata
|
|
423
429
|
- **[Advanced Filtering](examples/filters/)** - Complex boolean filters
|
|
424
430
|
- **[Rate Limiting](examples/rate-limiting/)** - Batch processing with cloud providers
|
|
@@ -430,39 +436,40 @@ See [examples/](examples/) directory for detailed guides:
|
|
|
430
436
|
|
|
431
437
|
#### Core Configuration
|
|
432
438
|
|
|
433
|
-
| Variable
|
|
434
|
-
|
|
|
435
|
-
| `TRANSPORT_MODE`
|
|
436
|
-
| `HTTP_PORT`
|
|
437
|
-
| `
|
|
438
|
-
| `
|
|
439
|
-
| `
|
|
440
|
-
| `
|
|
439
|
+
| Variable | Description | Default |
|
|
440
|
+
| ------------------------- | --------------------------------------- | --------------------- |
|
|
441
|
+
| `TRANSPORT_MODE` | "stdio" or "http" | stdio |
|
|
442
|
+
| `HTTP_PORT` | Port for HTTP transport | 3000 |
|
|
443
|
+
| `HTTP_REQUEST_TIMEOUT_MS` | Request timeout for HTTP transport (ms) | 300000 |
|
|
444
|
+
| `EMBEDDING_PROVIDER` | "ollama", "openai", "cohere", "voyage" | ollama |
|
|
445
|
+
| `QDRANT_URL` | Qdrant server URL | http://localhost:6333 |
|
|
446
|
+
| `QDRANT_API_KEY` | API key for Qdrant authentication | - |
|
|
447
|
+
| `PROMPTS_CONFIG_FILE` | Path to prompts configuration JSON | prompts.json |
|
|
441
448
|
|
|
442
449
|
#### Embedding Configuration
|
|
443
450
|
|
|
444
|
-
| Variable | Description
|
|
445
|
-
| ----------------------------------- |
|
|
446
|
-
| `EMBEDDING_MODEL` | Model name
|
|
447
|
-
| `EMBEDDING_BASE_URL` | Custom API URL
|
|
448
|
-
| `EMBEDDING_MAX_REQUESTS_PER_MINUTE` | Rate limit
|
|
449
|
-
| `EMBEDDING_RETRY_ATTEMPTS` | Retry count
|
|
450
|
-
| `EMBEDDING_RETRY_DELAY` | Initial retry delay (ms)
|
|
451
|
-
| `OPENAI_API_KEY` | OpenAI API key
|
|
452
|
-
| `COHERE_API_KEY` | Cohere API key
|
|
453
|
-
| `VOYAGE_API_KEY` | Voyage AI API key
|
|
451
|
+
| Variable | Description | Default |
|
|
452
|
+
| ----------------------------------- | ------------------------ | ----------------- |
|
|
453
|
+
| `EMBEDDING_MODEL` | Model name | Provider-specific |
|
|
454
|
+
| `EMBEDDING_BASE_URL` | Custom API URL | Provider-specific |
|
|
455
|
+
| `EMBEDDING_MAX_REQUESTS_PER_MINUTE` | Rate limit | Provider-specific |
|
|
456
|
+
| `EMBEDDING_RETRY_ATTEMPTS` | Retry count | 3 |
|
|
457
|
+
| `EMBEDDING_RETRY_DELAY` | Initial retry delay (ms) | 1000 |
|
|
458
|
+
| `OPENAI_API_KEY` | OpenAI API key | - |
|
|
459
|
+
| `COHERE_API_KEY` | Cohere API key | - |
|
|
460
|
+
| `VOYAGE_API_KEY` | Voyage AI API key | - |
|
|
454
461
|
|
|
455
462
|
#### Code Vectorization Configuration
|
|
456
463
|
|
|
457
|
-
| Variable
|
|
458
|
-
|
|
|
459
|
-
| `CODE_CHUNK_SIZE`
|
|
460
|
-
| `CODE_CHUNK_OVERLAP`
|
|
461
|
-
| `CODE_ENABLE_AST`
|
|
462
|
-
| `CODE_BATCH_SIZE`
|
|
463
|
-
| `CODE_CUSTOM_EXTENSIONS`
|
|
464
|
-
| `CODE_CUSTOM_IGNORE`
|
|
465
|
-
| `CODE_DEFAULT_LIMIT`
|
|
464
|
+
| Variable | Description | Default |
|
|
465
|
+
| ------------------------ | -------------------------------------------- | ------- |
|
|
466
|
+
| `CODE_CHUNK_SIZE` | Maximum chunk size in characters | 2500 |
|
|
467
|
+
| `CODE_CHUNK_OVERLAP` | Overlap between chunks in characters | 300 |
|
|
468
|
+
| `CODE_ENABLE_AST` | Enable AST-aware chunking (tree-sitter) | true |
|
|
469
|
+
| `CODE_BATCH_SIZE` | Number of chunks to embed in one batch | 100 |
|
|
470
|
+
| `CODE_CUSTOM_EXTENSIONS` | Additional file extensions (comma-separated) | - |
|
|
471
|
+
| `CODE_CUSTOM_IGNORE` | Additional ignore patterns (comma-separated) | - |
|
|
472
|
+
| `CODE_DEFAULT_LIMIT` | Default search result limit | 5 |
|
|
466
473
|
|
|
467
474
|
### Provider Comparison
|
|
468
475
|
|
|
@@ -473,25 +480,28 @@ See [examples/](examples/) directory for detailed guides:
|
|
|
473
480
|
| **Cohere** | `embed-english-v3.0` (default), `embed-multilingual-v3.0` | 1024 | 100/min | Multilingual support |
|
|
474
481
|
| **Voyage** | `voyage-2` (default), `voyage-large-2`, `voyage-code-2` | 1024, 1536 | 300/min | Code-specialized |
|
|
475
482
|
|
|
476
|
-
**Note:** Ollama models require
|
|
483
|
+
**Note:** Ollama models require pulling before use:
|
|
484
|
+
|
|
485
|
+
- Podman: `podman exec ollama ollama pull <model-name>`
|
|
486
|
+
- Docker: `docker exec ollama ollama pull <model-name>`
|
|
477
487
|
|
|
478
488
|
## Troubleshooting
|
|
479
489
|
|
|
480
|
-
| Issue
|
|
481
|
-
|
|
|
482
|
-
| **Qdrant not running**
|
|
483
|
-
| **Collection missing**
|
|
484
|
-
| **Ollama not running**
|
|
485
|
-
| **Model missing**
|
|
486
|
-
| **Rate limit errors**
|
|
487
|
-
| **API key errors**
|
|
488
|
-
| **Qdrant unauthorized**
|
|
489
|
-
| **Filter errors**
|
|
490
|
-
| **Codebase not indexed**
|
|
491
|
-
| **Slow indexing**
|
|
492
|
-
| **Files not found**
|
|
493
|
-
| **Search returns no results**
|
|
494
|
-
| **Out of memory during index**
|
|
490
|
+
| Issue | Solution |
|
|
491
|
+
| ------------------------------ | ----------------------------------------------------------------------------------------- |
|
|
492
|
+
| **Qdrant not running** | `podman compose up -d` or `docker compose up -d` |
|
|
493
|
+
| **Collection missing** | Create collection first before adding documents |
|
|
494
|
+
| **Ollama not running** | Verify with `curl http://localhost:11434`, start with `podman compose up -d` |
|
|
495
|
+
| **Model missing** | `podman exec ollama ollama pull nomic-embed-text` or `docker exec ollama ollama pull ...` |
|
|
496
|
+
| **Rate limit errors** | Adjust `EMBEDDING_MAX_REQUESTS_PER_MINUTE` to match your provider tier |
|
|
497
|
+
| **API key errors** | Verify correct API key in environment configuration |
|
|
498
|
+
| **Qdrant unauthorized** | Set `QDRANT_API_KEY` environment variable for secured instances |
|
|
499
|
+
| **Filter errors** | Ensure Qdrant filter format, check field names match metadata |
|
|
500
|
+
| **Codebase not indexed** | Run `index_codebase` before `search_code` |
|
|
501
|
+
| **Slow indexing** | Use Ollama (local) for faster indexing, or increase `CODE_BATCH_SIZE` |
|
|
502
|
+
| **Files not found** | Check `.gitignore` and `.contextignore` patterns |
|
|
503
|
+
| **Search returns no results** | Try broader queries, check if codebase is indexed with `get_index_status` |
|
|
504
|
+
| **Out of memory during index** | Reduce `CODE_CHUNK_SIZE` or `CODE_BATCH_SIZE` |
|
|
495
505
|
|
|
496
506
|
## Development
|
|
497
507
|
|
|
@@ -505,12 +515,12 @@ npm run test:coverage # Coverage report
|
|
|
505
515
|
|
|
506
516
|
### Testing
|
|
507
517
|
|
|
508
|
-
**
|
|
518
|
+
**548 tests** across 21 test files with **98%+ coverage**:
|
|
509
519
|
|
|
510
|
-
- **Unit Tests**: QdrantManager (
|
|
511
|
-
- **
|
|
520
|
+
- **Unit Tests**: QdrantManager (54), Ollama (41), OpenAI (25), Cohere (29), Voyage (31), Factory (43), Prompts (50), Transport (15), MCP Server (19)
|
|
521
|
+
- **Integration Tests**: Code indexer, scanner, chunker, synchronizer, merkle tree
|
|
512
522
|
|
|
513
|
-
**CI/CD**: GitHub Actions runs build, type-check, and tests on Node.js
|
|
523
|
+
**CI/CD**: GitHub Actions runs build, type-check, and tests on Node.js 22 LTS for every push/PR.
|
|
514
524
|
|
|
515
525
|
## Contributing
|
|
516
526
|
|
|
@@ -1,17 +1,19 @@
|
|
|
1
1
|
import { describe, it, expect, vi, beforeEach } from "vitest";
|
|
2
2
|
import { CohereEmbeddings } from "./cohere.js";
|
|
3
3
|
import { CohereClient } from "cohere-ai";
|
|
4
|
+
const mockClient = {
|
|
5
|
+
embed: vi.fn().mockResolvedValue({ embeddings: [[]] }),
|
|
6
|
+
};
|
|
4
7
|
vi.mock("cohere-ai", () => ({
|
|
5
|
-
CohereClient: vi.fn()
|
|
8
|
+
CohereClient: vi.fn().mockImplementation(function () {
|
|
9
|
+
return mockClient;
|
|
10
|
+
}),
|
|
6
11
|
}));
|
|
7
12
|
describe("CohereEmbeddings", () => {
|
|
8
13
|
let embeddings;
|
|
9
|
-
let mockClient;
|
|
10
14
|
beforeEach(() => {
|
|
11
|
-
mockClient
|
|
12
|
-
|
|
13
|
-
};
|
|
14
|
-
vi.mocked(CohereClient).mockImplementation(() => mockClient);
|
|
15
|
+
mockClient.embed.mockReset().mockResolvedValue({ embeddings: [[]] });
|
|
16
|
+
vi.mocked(CohereClient).mockClear();
|
|
15
17
|
embeddings = new CohereEmbeddings("test-api-key");
|
|
16
18
|
});
|
|
17
19
|
describe("constructor", () => {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"cohere.test.js","sourceRoot":"","sources":["../../src/embeddings/cohere.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAC/C,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AAEzC,EAAE,CAAC,IAAI,CAAC,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC;IAC1B,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;CACtB,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,IAAI,UAA4B,CAAC;IACjC,IAAI,UAAe,CAAC;IAEpB,UAAU,CAAC,GAAG,EAAE;QACd,UAAU,GAAG;YACX,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;SACf,CAAC;QAEF,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,UAAiB,CAAC,CAAC;QAEpE,UAAU,GAAG,IAAI,gBAAgB,CAAC,cAAc,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,MAAM,CAAC,UAAU,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;YACzD,MAAM,CAAC,UAAU,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yBAAyB,EAAE,GAAG,EAAE;YACjC,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,CAC3C,cAAc,EACd,yBAAyB,CAC1B,CAAC;YACF,MAAM,CAAC,gBAAgB,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;YACpE,MAAM,CAAC,gBAAgB,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;YACtC,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,CAC3C,cAAc,EACd,oBAAoB,EACpB,GAAG,CACJ,CAAC;YACF,MAAM,CAAC,gBAAgB,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,MAAM,eAAe,GAAG,IAAI,gBAAgB,CAC1C,cAAc,EACd,0BAA0B,CAC3B,CAAC;YACF,MAAM,CAAC,eAAe,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,iBAAiB,GAAG,IAAI,gBAAgB,CAC5C,cAAc,EACd,cAAc,CACf,CAAC;YACF,MAAM,CAAC,iBAAiB,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,qBAAqB,GAAG,IAAI,gBAAgB,CAChD,cAAc,EACd,oBAAoB,EACpB,SAAS,EACT,SAAS,EACT,cAAc,CACf,CAAC;YACF,MAAM,CAAC,qBAAqB,CAAC,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,OAAO,EAAE,GAAG,EAAE;QACrB,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC;iBAC9B,IAAI,CAAC,CAAC,CAAC;iBACP,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC;YAC5B,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC;gBACjC,UAAU,EAAE,CAAC,aAAa,CAAC;aAC5B,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAEnD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,SAAS,EAAE,aAAa;gBACxB,UAAU,EAAE,IAAI;aACjB,CAAC,CAAC;YACH,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAAC;gBAC5C,KAAK,EAAE,CAAC,WAAW,CAAC;gBACpB,KAAK,EAAE,oBAAoB;gBAC3B,SAAS,EAAE,iBAAiB;gBAC5B,cAAc,EAAE,CAAC,OAAO,CAAC;aAC1B,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;YACvC,MAAM,QAAQ,GAAG,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACtC,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC5C,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC;gBACjC,UAAU,EAAE,CAAC,aAAa,CAAC;aAC5B,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;YAEhD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YAChD,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAAC;gBAC5C,KAAK,EAAE,CAAC,QAAQ,CAAC;gBACjB,KAAK,EAAE,oBAAoB;gBAC3B,SAAS,EAAE,iBAAiB;gBAC5B,cAAc,EAAE,CAAC,OAAO,CAAC;aAC1B,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,CAC3C,cAAc,EACd,yBAAyB,EACzB,IAAI,CACL,CAAC;YACF,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC5C,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC;gBACjC,UAAU,EAAE,CAAC,aAAa,CAAC;aAC5B,CAAC,CAAC;YAEH,MAAM,gBAAgB,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAErC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAAC;gBAC5C,KAAK,EAAE,CAAC,MAAM,CAAC;gBACf,KAAK,EAAE,yBAAyB;gBAChC,SAAS,EAAE,iBAAiB;gBAC5B,cAAc,EAAE,CAAC,OAAO,CAAC;aAC1B,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC;gBACjC,UAAU,EAAE,EAAE;aACf,CAAC,CAAC;YAEH,MAAM,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACpD,uCAAuC,CACxC,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;YACvC,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC;YAE3D,MAAM,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;QACtE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,cAAc,GAAG;gBACrB,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;gBACrB,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;gBACrB,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;aACtB,CAAC;YACF,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC;gBACjC,UAAU,EAAE,cAAc;aAC3B,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;YAC1C,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YAEnD,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC;gBACtB,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE;gBAClD,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE;gBAClD,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE;aACnD,CAAC,CAAC;YACH,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAAC;gBAC5C,KAAK;gBACL,KAAK,EAAE,oBAAoB;gBAC3B,SAAS,EAAE,iBAAiB;gBAC5B,cAAc,EAAE,CAAC,OAAO,CAAC;aAC1B,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;YACzC,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC;gBACjC,UAAU,EAAE,EAAE;aACf,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;YAEhD,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC5C,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC;gBACjC,UAAU,EAAE,CAAC,aAAa,CAAC;aAC5B,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,UAAU,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC;YAE7D,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAChC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;YAC3C,MAAM,SAAS,GAAG,GAAG,CAAC;YACtB,MAAM,cAAc,GAAG,KAAK,CAAC,SAAS,CAAC;iBACpC,IAAI,CAAC,IAAI,CAAC;iBACV,GAAG,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;YAE9C,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC;gBACjC,UAAU,EAAE,cAAc;aAC3B,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,KAAK,CAAC,SAAS,CAAC;iBAC3B,IAAI,CAAC,IAAI,CAAC;iBACV,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YAC9B,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YAEnD,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC5D,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YAEvC,MAAM,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAC5D,wCAAwC,CACzC,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;YAChD,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC,CAAC;YAEjE,MAAM,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACrE,iBAAiB,CAClB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,MAAM,CAAC,UAAU,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,CAC3C,cAAc,EACd,oBAAoB,EACpB,GAAG,CACJ,CAAC;YACF,MAAM,CAAC,gBAAgB,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;QACxB,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,CAAC,UAAU,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;YACpC,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,CAC3C,cAAc,EACd,yBAAyB,CAC1B,CAAC;YACF,MAAM,CAAC,gBAAgB,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;QACtE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAE5C,UAAU,CAAC,KAAK;iBACb,qBAAqB,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,qBAAqB,EAAE,CAAC;iBACtE,qBAAqB,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,qBAAqB,EAAE,CAAC;iBACtE,iBAAiB,CAAC,EAAE,UAAU,EAAE,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;YAEtD,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAEnD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YAChD,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;YACjE,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAE5C,UAAU,CAAC,KAAK;iBACb,qBAAqB,CAAC;gBACrB,UAAU,EAAE,GAAG;gBACf,OAAO,EAAE,qBAAqB;aAC/B,CAAC;iBACD,iBAAiB,CAAC,EAAE,UAAU,EAAE,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;YAEtD,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAEnD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YAChD,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAE5C,UAAU,CAAC,KAAK;iBACb,qBAAqB,CAAC;gBACrB,OAAO,EAAE,kCAAkC;aAC5C,CAAC;iBACD,iBAAiB,CAAC,EAAE,UAAU,EAAE,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;YAEtD,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAEnD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YAChD,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAC9C,MAAM,mBAAmB,GAAG,IAAI,gBAAgB,CAC9C,cAAc,EACd,oBAAoB,EACpB,SAAS,EACT;gBACE,aAAa,EAAE,CAAC;gBAChB,YAAY,EAAE,GAAG;aAClB,CACF,CAAC;YAEF,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC5C,MAAM,cAAc,GAAG;gBACrB,MAAM,EAAE,GAAG;gBACX,OAAO,EAAE,qBAAqB;aAC/B,CAAC;YAEF,UAAU,CAAC,KAAK;iBACb,qBAAqB,CAAC,cAAc,CAAC;iBACrC,qBAAqB,CAAC,cAAc,CAAC;iBACrC,iBAAiB,CAAC,EAAE,UAAU,EAAE,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;YAEtD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC7B,MAAM,mBAAmB,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAC7C,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAExC,kEAAkE;YAClE,MAAM,CAAC,QAAQ,CAAC,CAAC,sBAAsB,CAAC,GAAG,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,mBAAmB,GAAG,IAAI,gBAAgB,CAC9C,cAAc,EACd,oBAAoB,EACpB,SAAS,EACT;gBACE,aAAa,EAAE,CAAC;gBAChB,YAAY,EAAE,GAAG;aAClB,CACF,CAAC;YAEF,MAAM,cAAc,GAAG;gBACrB,MAAM,EAAE,GAAG;gBACX,OAAO,EAAE,qBAAqB;aAC/B,CAAC;YAEF,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC;YAEnD,MAAM,MAAM,CAAC,mBAAmB,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAClE,uDAAuD,CACxD,CAAC;YAEF,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,MAAM,cAAc,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;YAEtE,UAAU,CAAC,KAAK;iBACb,qBAAqB,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,qBAAqB,EAAE,CAAC;iBACtE,iBAAiB,CAAC;gBACjB,UAAU,EAAE,cAAc;aAC3B,CAAC,CAAC;YAEL,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,UAAU,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;YAEhE,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAChC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,QAAQ,GAAG,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;YAC9C,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;YAE7C,MAAM,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACzD,iBAAiB,CAClB,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,CAC3C,cAAc,EACd,oBAAoB,EACpB,SAAS,EACT;gBACE,oBAAoB,EAAE,GAAG;gBACzB,aAAa,EAAE,CAAC;gBAChB,YAAY,EAAE,IAAI;aACnB,CACF,CAAC;YAEF,MAAM,CAAC,gBAAgB,CAAC,CAAC,WAAW,EAAE,CAAC;QACzC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
1
|
+
{"version":3,"file":"cohere.test.js","sourceRoot":"","sources":["../../src/embeddings/cohere.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAC/C,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AAEzC,MAAM,UAAU,GAAG;IACjB,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC;CACvD,CAAC;AAEF,EAAE,CAAC,IAAI,CAAC,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC;IAC1B,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC;QACvC,OAAO,UAAU,CAAC;IACpB,CAAC,CAAC;CACH,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,IAAI,UAA4B,CAAC;IAEjC,UAAU,CAAC,GAAG,EAAE;QACd,UAAU,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC,iBAAiB,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;QACrE,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,EAAE,CAAC;QACpC,UAAU,GAAG,IAAI,gBAAgB,CAAC,cAAc,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,MAAM,CAAC,UAAU,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;YACzD,MAAM,CAAC,UAAU,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yBAAyB,EAAE,GAAG,EAAE;YACjC,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,CAC3C,cAAc,EACd,yBAAyB,CAC1B,CAAC;YACF,MAAM,CAAC,gBAAgB,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;YACpE,MAAM,CAAC,gBAAgB,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;YACtC,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,CAC3C,cAAc,EACd,oBAAoB,EACpB,GAAG,CACJ,CAAC;YACF,MAAM,CAAC,gBAAgB,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,MAAM,eAAe,GAAG,IAAI,gBAAgB,CAC1C,cAAc,EACd,0BAA0B,CAC3B,CAAC;YACF,MAAM,CAAC,eAAe,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,iBAAiB,GAAG,IAAI,gBAAgB,CAC5C,cAAc,EACd,cAAc,CACf,CAAC;YACF,MAAM,CAAC,iBAAiB,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,qBAAqB,GAAG,IAAI,gBAAgB,CAChD,cAAc,EACd,oBAAoB,EACpB,SAAS,EACT,SAAS,EACT,cAAc,CACf,CAAC;YACF,MAAM,CAAC,qBAAqB,CAAC,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,OAAO,EAAE,GAAG,EAAE;QACrB,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC;iBAC9B,IAAI,CAAC,CAAC,CAAC;iBACP,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC;YAC5B,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC;gBACjC,UAAU,EAAE,CAAC,aAAa,CAAC;aAC5B,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAEnD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,SAAS,EAAE,aAAa;gBACxB,UAAU,EAAE,IAAI;aACjB,CAAC,CAAC;YACH,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAAC;gBAC5C,KAAK,EAAE,CAAC,WAAW,CAAC;gBACpB,KAAK,EAAE,oBAAoB;gBAC3B,SAAS,EAAE,iBAAiB;gBAC5B,cAAc,EAAE,CAAC,OAAO,CAAC;aAC1B,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;YACvC,MAAM,QAAQ,GAAG,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACtC,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC5C,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC;gBACjC,UAAU,EAAE,CAAC,aAAa,CAAC;aAC5B,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;YAEhD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YAChD,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAAC;gBAC5C,KAAK,EAAE,CAAC,QAAQ,CAAC;gBACjB,KAAK,EAAE,oBAAoB;gBAC3B,SAAS,EAAE,iBAAiB;gBAC5B,cAAc,EAAE,CAAC,OAAO,CAAC;aAC1B,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,CAC3C,cAAc,EACd,yBAAyB,EACzB,IAAI,CACL,CAAC;YACF,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC5C,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC;gBACjC,UAAU,EAAE,CAAC,aAAa,CAAC;aAC5B,CAAC,CAAC;YAEH,MAAM,gBAAgB,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAErC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAAC;gBAC5C,KAAK,EAAE,CAAC,MAAM,CAAC;gBACf,KAAK,EAAE,yBAAyB;gBAChC,SAAS,EAAE,iBAAiB;gBAC5B,cAAc,EAAE,CAAC,OAAO,CAAC;aAC1B,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC;gBACjC,UAAU,EAAE,EAAE;aACf,CAAC,CAAC;YAEH,MAAM,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACpD,uCAAuC,CACxC,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;YACvC,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC;YAE3D,MAAM,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;QACtE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,cAAc,GAAG;gBACrB,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;gBACrB,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;gBACrB,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;aACtB,CAAC;YACF,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC;gBACjC,UAAU,EAAE,cAAc;aAC3B,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;YAC1C,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YAEnD,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC;gBACtB,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE;gBAClD,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE;gBAClD,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE;aACnD,CAAC,CAAC;YACH,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAAC;gBAC5C,KAAK;gBACL,KAAK,EAAE,oBAAoB;gBAC3B,SAAS,EAAE,iBAAiB;gBAC5B,cAAc,EAAE,CAAC,OAAO,CAAC;aAC1B,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;YACzC,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC;gBACjC,UAAU,EAAE,EAAE;aACf,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;YAEhD,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC5C,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC;gBACjC,UAAU,EAAE,CAAC,aAAa,CAAC;aAC5B,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,UAAU,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC;YAE7D,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAChC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;YAC3C,MAAM,SAAS,GAAG,GAAG,CAAC;YACtB,MAAM,cAAc,GAAG,KAAK,CAAC,SAAS,CAAC;iBACpC,IAAI,CAAC,IAAI,CAAC;iBACV,GAAG,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;YAE9C,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC;gBACjC,UAAU,EAAE,cAAc;aAC3B,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,KAAK,CAAC,SAAS,CAAC;iBAC3B,IAAI,CAAC,IAAI,CAAC;iBACV,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YAC9B,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YAEnD,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC5D,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YAEvC,MAAM,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAC5D,wCAAwC,CACzC,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;YAChD,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC,CAAC;YAEjE,MAAM,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACrE,iBAAiB,CAClB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,MAAM,CAAC,UAAU,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,CAC3C,cAAc,EACd,oBAAoB,EACpB,GAAG,CACJ,CAAC;YACF,MAAM,CAAC,gBAAgB,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;QACxB,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,CAAC,UAAU,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;YACpC,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,CAC3C,cAAc,EACd,yBAAyB,CAC1B,CAAC;YACF,MAAM,CAAC,gBAAgB,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;QACtE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAE5C,UAAU,CAAC,KAAK;iBACb,qBAAqB,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,qBAAqB,EAAE,CAAC;iBACtE,qBAAqB,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,qBAAqB,EAAE,CAAC;iBACtE,iBAAiB,CAAC,EAAE,UAAU,EAAE,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;YAEtD,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAEnD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YAChD,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;YACjE,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAE5C,UAAU,CAAC,KAAK;iBACb,qBAAqB,CAAC;gBACrB,UAAU,EAAE,GAAG;gBACf,OAAO,EAAE,qBAAqB;aAC/B,CAAC;iBACD,iBAAiB,CAAC,EAAE,UAAU,EAAE,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;YAEtD,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAEnD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YAChD,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAE5C,UAAU,CAAC,KAAK;iBACb,qBAAqB,CAAC;gBACrB,OAAO,EAAE,kCAAkC;aAC5C,CAAC;iBACD,iBAAiB,CAAC,EAAE,UAAU,EAAE,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;YAEtD,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAEnD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YAChD,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAC9C,MAAM,mBAAmB,GAAG,IAAI,gBAAgB,CAC9C,cAAc,EACd,oBAAoB,EACpB,SAAS,EACT;gBACE,aAAa,EAAE,CAAC;gBAChB,YAAY,EAAE,GAAG;aAClB,CACF,CAAC;YAEF,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC5C,MAAM,cAAc,GAAG;gBACrB,MAAM,EAAE,GAAG;gBACX,OAAO,EAAE,qBAAqB;aAC/B,CAAC;YAEF,UAAU,CAAC,KAAK;iBACb,qBAAqB,CAAC,cAAc,CAAC;iBACrC,qBAAqB,CAAC,cAAc,CAAC;iBACrC,iBAAiB,CAAC,EAAE,UAAU,EAAE,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;YAEtD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC7B,MAAM,mBAAmB,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAC7C,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAExC,kEAAkE;YAClE,MAAM,CAAC,QAAQ,CAAC,CAAC,sBAAsB,CAAC,GAAG,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,mBAAmB,GAAG,IAAI,gBAAgB,CAC9C,cAAc,EACd,oBAAoB,EACpB,SAAS,EACT;gBACE,aAAa,EAAE,CAAC;gBAChB,YAAY,EAAE,GAAG;aAClB,CACF,CAAC;YAEF,MAAM,cAAc,GAAG;gBACrB,MAAM,EAAE,GAAG;gBACX,OAAO,EAAE,qBAAqB;aAC/B,CAAC;YAEF,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC;YAEnD,MAAM,MAAM,CAAC,mBAAmB,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAClE,uDAAuD,CACxD,CAAC;YAEF,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,MAAM,cAAc,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;YAEtE,UAAU,CAAC,KAAK;iBACb,qBAAqB,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,qBAAqB,EAAE,CAAC;iBACtE,iBAAiB,CAAC;gBACjB,UAAU,EAAE,cAAc;aAC3B,CAAC,CAAC;YAEL,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,UAAU,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;YAEhE,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAChC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,QAAQ,GAAG,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;YAC9C,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;YAE7C,MAAM,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACzD,iBAAiB,CAClB,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,CAC3C,cAAc,EACd,oBAAoB,EACpB,SAAS,EACT;gBACE,oBAAoB,EAAE,GAAG;gBACzB,aAAa,EAAE,CAAC;gBAChB,YAAY,EAAE,IAAI;aACnB,CACF,CAAC;YAEF,MAAM,CAAC,gBAAgB,CAAC,CAAC,WAAW,EAAE,CAAC;QACzC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -1,19 +1,23 @@
|
|
|
1
1
|
import { describe, it, expect, vi, beforeEach } from "vitest";
|
|
2
2
|
import { OpenAIEmbeddings } from "./openai.js";
|
|
3
3
|
import OpenAI from "openai";
|
|
4
|
+
const mockOpenAI = {
|
|
5
|
+
embeddings: {
|
|
6
|
+
create: vi.fn().mockResolvedValue({ data: [{ embedding: [] }] }),
|
|
7
|
+
},
|
|
8
|
+
};
|
|
4
9
|
vi.mock("openai", () => ({
|
|
5
|
-
default: vi.fn()
|
|
10
|
+
default: vi.fn().mockImplementation(function () {
|
|
11
|
+
return mockOpenAI;
|
|
12
|
+
}),
|
|
6
13
|
}));
|
|
7
14
|
describe("OpenAIEmbeddings", () => {
|
|
8
15
|
let embeddings;
|
|
9
|
-
let mockOpenAI;
|
|
10
16
|
beforeEach(() => {
|
|
11
|
-
mockOpenAI
|
|
12
|
-
|
|
13
|
-
|
|
14
|
-
|
|
15
|
-
};
|
|
16
|
-
vi.mocked(OpenAI).mockImplementation(() => mockOpenAI);
|
|
17
|
+
mockOpenAI.embeddings.create
|
|
18
|
+
.mockReset()
|
|
19
|
+
.mockResolvedValue({ data: [{ embedding: [] }] });
|
|
20
|
+
vi.mocked(OpenAI).mockClear();
|
|
17
21
|
embeddings = new OpenAIEmbeddings("test-api-key");
|
|
18
22
|
});
|
|
19
23
|
describe("constructor", () => {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"openai.test.js","sourceRoot":"","sources":["../../src/embeddings/openai.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAC/C,OAAO,MAAM,MAAM,QAAQ,CAAC;AAE5B,EAAE,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,EAAE,CAAC,CAAC;IACvB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;CACjB,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,IAAI,UAA4B,CAAC;IACjC,IAAI,UAAe,CAAC;IAEpB,UAAU,CAAC,GAAG,EAAE;QACd,UAAU,GAAG;YACX,UAAU,EAAE;gBACV,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;aAChB;SACF,CAAC;QAEF,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,UAAiB,CAAC,CAAC;QAE9D,UAAU,GAAG,IAAI,gBAAgB,CAAC,cAAc,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,MAAM,CAAC,UAAU,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;YAC7D,MAAM,CAAC,UAAU,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yBAAyB,EAAE,GAAG,EAAE;YACjC,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,CAC3C,cAAc,EACd,wBAAwB,CACzB,CAAC;YACF,MAAM,CAAC,gBAAgB,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;YACnE,MAAM,CAAC,gBAAgB,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;YACtC,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,CAC3C,cAAc,EACd,wBAAwB,EACxB,GAAG,CACJ,CAAC;YACF,MAAM,CAAC,gBAAgB,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;YAClE,MAAM,aAAa,GAAG,IAAI,gBAAgB,CACxC,cAAc,EACd,wBAAwB,CACzB,CAAC;YACF,MAAM,CAAC,aAAa,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,OAAO,EAAE,GAAG,EAAE;QACrB,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC;iBAC9B,IAAI,CAAC,CAAC,CAAC;iBACP,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC;YAC5B,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,iBAAiB,CAAC;gBAC7C,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC;aACrC,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAEnD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,SAAS,EAAE,aAAa;gBACxB,UAAU,EAAE,IAAI;aACjB,CAAC,CAAC;YACH,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC;gBACxD,KAAK,EAAE,wBAAwB;gBAC/B,KAAK,EAAE,WAAW;gBAClB,UAAU,EAAE,IAAI;aACjB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;YACvC,MAAM,QAAQ,GAAG,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACtC,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC5C,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,iBAAiB,CAAC;gBAC7C,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC;aACrC,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;YAEhD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YAChD,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC;gBACxD,KAAK,EAAE,wBAAwB;gBAC/B,KAAK,EAAE,QAAQ;gBACf,UAAU,EAAE,IAAI;aACjB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,CAC3C,cAAc,EACd,wBAAwB,EACxB,IAAI,CACL,CAAC;YACF,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC5C,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,iBAAiB,CAAC;gBAC7C,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC;aACrC,CAAC,CAAC;YAEH,MAAM,gBAAgB,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAErC,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC;gBACxD,KAAK,EAAE,wBAAwB;gBAC/B,KAAK,EAAE,MAAM;gBACb,UAAU,EAAE,IAAI;aACjB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;YACvC,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC;YAEvE,MAAM,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;QACtE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,cAAc,GAAG;gBACrB,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;gBACrB,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;gBACrB,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;aACtB,CAAC;YACF,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,iBAAiB,CAAC;gBAC7C,IAAI,EAAE;oBACJ,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE;oBAChC,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE;oBAChC,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE;iBACjC;aACF,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;YAC1C,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YAEnD,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC;gBACtB,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE;gBAClD,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE;gBAClD,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE;aACnD,CAAC,CAAC;YACH,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC;gBACxD,KAAK,EAAE,wBAAwB;gBAC/B,KAAK,EAAE,KAAK;gBACZ,UAAU,EAAE,IAAI;aACjB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;YACzC,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,iBAAiB,CAAC;gBAC7C,IAAI,EAAE,EAAE;aACT,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;YAEhD,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC5C,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,iBAAiB,CAAC;gBAC7C,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC;aACrC,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,UAAU,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC;YAE7D,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAChC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;YAC3C,MAAM,SAAS,GAAG,GAAG,CAAC;YACtB,MAAM,cAAc,GAAG,KAAK,CAAC,SAAS,CAAC;iBACpC,IAAI,CAAC,IAAI,CAAC;iBACV,GAAG,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;YAE9C,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,iBAAiB,CAAC;gBAC7C,IAAI,EAAE,cAAc,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC;aACzD,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,KAAK,CAAC,SAAS,CAAC;iBAC3B,IAAI,CAAC,IAAI,CAAC;iBACV,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YAC9B,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YAEnD,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;YAChD,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,iBAAiB,CAC5C,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAC7B,CAAC;YAEF,MAAM,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACrE,iBAAiB,CAClB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,MAAM,CAAC,UAAU,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,CAC3C,cAAc,EACd,wBAAwB,EACxB,GAAG,CACJ,CAAC;YACF,MAAM,CAAC,gBAAgB,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;QACxB,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,CAAC,UAAU,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;YACpC,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,CAC3C,cAAc,EACd,wBAAwB,CACzB,CAAC;YACF,MAAM,CAAC,gBAAgB,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;QACrE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAE5C,yDAAyD;YACzD,UAAU,CAAC,UAAU,CAAC,MAAM;iBACzB,qBAAqB,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,qBAAqB,EAAE,CAAC;iBACtE,qBAAqB,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,qBAAqB,EAAE,CAAC;iBACtE,iBAAiB,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC,EAAE,CAAC,CAAC;YAE/D,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAEnD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YAChD,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC5C,MAAM,cAAc,GAAG;gBACrB,MAAM,EAAE,GAAG;gBACX,OAAO,EAAE,qBAAqB;gBAC9B,OAAO,EAAE,EAAE,aAAa,EAAE,GAAG,EAAE;aAChC,CAAC;YAEF,UAAU,CAAC,UAAU,CAAC,MAAM;iBACzB,qBAAqB,CAAC,cAAc,CAAC;iBACrC,iBAAiB,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC,EAAE,CAAC,CAAC;YAE/D,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC7B,MAAM,UAAU,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YACpC,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAExC,0CAA0C;YAC1C,MAAM,CAAC,QAAQ,CAAC,CAAC,sBAAsB,CAAC,IAAI,CAAC,CAAC,CAAC,qBAAqB;QACtE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;YACtF,MAAM,mBAAmB,GAAG,IAAI,gBAAgB,CAC9C,cAAc,EACd,wBAAwB,EACxB,SAAS,EACT;gBACE,aAAa,EAAE,CAAC;gBAChB,YAAY,EAAE,GAAG,EAAE,yBAAyB;aAC7C,CACF,CAAC;YAEF,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAE5C,0CAA0C;YAC1C,MAAM,sBAAsB,GAAG;gBAC7B,MAAM,EAAE,GAAG;gBACX,OAAO,EAAE,qBAAqB;gBAC9B,OAAO,EAAE,EAAE,aAAa,EAAE,SAAS,EAAE,EAAE,oBAAoB;aAC5D,CAAC;YAEF,UAAU,CAAC,UAAU,CAAC,MAAM;iBACzB,qBAAqB,CAAC,sBAAsB,CAAC;iBAC7C,iBAAiB,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC,EAAE,CAAC,CAAC;YAE/D,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC7B,MAAM,mBAAmB,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAC7C,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAExC,iFAAiF;YACjF,MAAM,CAAC,QAAQ,CAAC,CAAC,sBAAsB,CAAC,EAAE,CAAC,CAAC,CAAC,qBAAqB;YAClE,MAAM,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,2BAA2B;QACjE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;YACzE,MAAM,mBAAmB,GAAG,IAAI,gBAAgB,CAC9C,cAAc,EACd,wBAAwB,EACxB,SAAS,EACT;gBACE,aAAa,EAAE,CAAC;gBAChB,YAAY,EAAE,GAAG,EAAE,yBAAyB;aAC7C,CACF,CAAC;YAEF,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC5C,MAAM,cAAc,GAAG;gBACrB,MAAM,EAAE,GAAG;gBACX,OAAO,EAAE,qBAAqB;aAC/B,CAAC;YAEF,UAAU,CAAC,UAAU,CAAC,MAAM;iBACzB,qBAAqB,CAAC,cAAc,CAAC;iBACrC,qBAAqB,CAAC,cAAc,CAAC;iBACrC,iBAAiB,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC,EAAE,CAAC,CAAC;YAE/D,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC7B,MAAM,mBAAmB,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAC7C,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAExC,kEAAkE;YAClE,MAAM,CAAC,QAAQ,CAAC,CAAC,sBAAsB,CAAC,GAAG,CAAC,CAAC,CAAC,kCAAkC;QAClF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,mBAAmB,GAAG,IAAI,gBAAgB,CAC9C,cAAc,EACd,wBAAwB,EACxB,SAAS,EACT;gBACE,aAAa,EAAE,CAAC;gBAChB,YAAY,EAAE,GAAG;aAClB,CACF,CAAC;YAEF,MAAM,cAAc,GAAG;gBACrB,MAAM,EAAE,GAAG;gBACX,OAAO,EAAE,qBAAqB;aAC/B,CAAC;YAEF,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC;YAE/D,MAAM,MAAM,CAAC,mBAAmB,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAClE,uDAAuD,CACxD,CAAC;YAEF,oDAAoD;YACpD,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,MAAM,cAAc,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;YAEtE,UAAU,CAAC,UAAU,CAAC,MAAM;iBACzB,qBAAqB,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,qBAAqB,EAAE,CAAC;iBACtE,iBAAiB,CAAC;gBACjB,IAAI,EAAE;oBACJ,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE;oBAChC,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE;iBACjC;aACF,CAAC,CAAC;YAEL,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,UAAU,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;YAEhE,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAChC,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,QAAQ,GAAG,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;YAC9C,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;YAEzD,MAAM,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACzD,iBAAiB,CAClB,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,CAC3C,cAAc,EACd,wBAAwB,EACxB,SAAS,EACT;gBACE,oBAAoB,EAAE,IAAI;gBAC1B,aAAa,EAAE,CAAC;gBAChB,YAAY,EAAE,IAAI;aACnB,CACF,CAAC;YAEF,MAAM,CAAC,gBAAgB,CAAC,CAAC,WAAW,EAAE,CAAC;QACzC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
1
|
+
{"version":3,"file":"openai.test.js","sourceRoot":"","sources":["../../src/embeddings/openai.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAC/C,OAAO,MAAM,MAAM,QAAQ,CAAC;AAE5B,MAAM,UAAU,GAAG;IACjB,UAAU,EAAE;QACV,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC;KACjE;CACF,CAAC;AAEF,EAAE,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,EAAE,CAAC,CAAC;IACvB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC;QAClC,OAAO,UAAU,CAAC;IACpB,CAAC,CAAC;CACH,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,IAAI,UAA4B,CAAC;IAEjC,UAAU,CAAC,GAAG,EAAE;QACd,UAAU,CAAC,UAAU,CAAC,MAAM;aACzB,SAAS,EAAE;aACX,iBAAiB,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC;QACpD,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,EAAE,CAAC;QAC9B,UAAU,GAAG,IAAI,gBAAgB,CAAC,cAAc,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,MAAM,CAAC,UAAU,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;YAC7D,MAAM,CAAC,UAAU,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yBAAyB,EAAE,GAAG,EAAE;YACjC,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,CAC3C,cAAc,EACd,wBAAwB,CACzB,CAAC;YACF,MAAM,CAAC,gBAAgB,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;YACnE,MAAM,CAAC,gBAAgB,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;YACtC,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,CAC3C,cAAc,EACd,wBAAwB,EACxB,GAAG,CACJ,CAAC;YACF,MAAM,CAAC,gBAAgB,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;YAClE,MAAM,aAAa,GAAG,IAAI,gBAAgB,CACxC,cAAc,EACd,wBAAwB,CACzB,CAAC;YACF,MAAM,CAAC,aAAa,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,OAAO,EAAE,GAAG,EAAE;QACrB,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC;iBAC9B,IAAI,CAAC,CAAC,CAAC;iBACP,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC;YAC5B,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,iBAAiB,CAAC;gBAC7C,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC;aACrC,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAEnD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,SAAS,EAAE,aAAa;gBACxB,UAAU,EAAE,IAAI;aACjB,CAAC,CAAC;YACH,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC;gBACxD,KAAK,EAAE,wBAAwB;gBAC/B,KAAK,EAAE,WAAW;gBAClB,UAAU,EAAE,IAAI;aACjB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;YACvC,MAAM,QAAQ,GAAG,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACtC,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC5C,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,iBAAiB,CAAC;gBAC7C,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC;aACrC,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;YAEhD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YAChD,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC;gBACxD,KAAK,EAAE,wBAAwB;gBAC/B,KAAK,EAAE,QAAQ;gBACf,UAAU,EAAE,IAAI;aACjB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,CAC3C,cAAc,EACd,wBAAwB,EACxB,IAAI,CACL,CAAC;YACF,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC5C,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,iBAAiB,CAAC;gBAC7C,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC;aACrC,CAAC,CAAC;YAEH,MAAM,gBAAgB,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAErC,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC;gBACxD,KAAK,EAAE,wBAAwB;gBAC/B,KAAK,EAAE,MAAM;gBACb,UAAU,EAAE,IAAI;aACjB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;YACvC,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC;YAEvE,MAAM,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;QACtE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,cAAc,GAAG;gBACrB,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;gBACrB,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;gBACrB,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;aACtB,CAAC;YACF,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,iBAAiB,CAAC;gBAC7C,IAAI,EAAE;oBACJ,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE;oBAChC,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE;oBAChC,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE;iBACjC;aACF,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;YAC1C,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YAEnD,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC;gBACtB,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE;gBAClD,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE;gBAClD,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE;aACnD,CAAC,CAAC;YACH,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC;gBACxD,KAAK,EAAE,wBAAwB;gBAC/B,KAAK,EAAE,KAAK;gBACZ,UAAU,EAAE,IAAI;aACjB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;YACzC,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,iBAAiB,CAAC;gBAC7C,IAAI,EAAE,EAAE;aACT,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;YAEhD,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC5C,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,iBAAiB,CAAC;gBAC7C,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC;aACrC,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,UAAU,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC;YAE7D,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAChC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;YAC3C,MAAM,SAAS,GAAG,GAAG,CAAC;YACtB,MAAM,cAAc,GAAG,KAAK,CAAC,SAAS,CAAC;iBACpC,IAAI,CAAC,IAAI,CAAC;iBACV,GAAG,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;YAE9C,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,iBAAiB,CAAC;gBAC7C,IAAI,EAAE,cAAc,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC;aACzD,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,KAAK,CAAC,SAAS,CAAC;iBAC3B,IAAI,CAAC,IAAI,CAAC;iBACV,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YAC9B,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YAEnD,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;YAChD,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,iBAAiB,CAC5C,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAC7B,CAAC;YAEF,MAAM,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACrE,iBAAiB,CAClB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,MAAM,CAAC,UAAU,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,CAC3C,cAAc,EACd,wBAAwB,EACxB,GAAG,CACJ,CAAC;YACF,MAAM,CAAC,gBAAgB,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;QACxB,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,CAAC,UAAU,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;YACpC,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,CAC3C,cAAc,EACd,wBAAwB,CACzB,CAAC;YACF,MAAM,CAAC,gBAAgB,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;QACrE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAE5C,yDAAyD;YACzD,UAAU,CAAC,UAAU,CAAC,MAAM;iBACzB,qBAAqB,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,qBAAqB,EAAE,CAAC;iBACtE,qBAAqB,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,qBAAqB,EAAE,CAAC;iBACtE,iBAAiB,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC,EAAE,CAAC,CAAC;YAE/D,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAEnD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YAChD,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC5C,MAAM,cAAc,GAAG;gBACrB,MAAM,EAAE,GAAG;gBACX,OAAO,EAAE,qBAAqB;gBAC9B,OAAO,EAAE,EAAE,aAAa,EAAE,GAAG,EAAE;aAChC,CAAC;YAEF,UAAU,CAAC,UAAU,CAAC,MAAM;iBACzB,qBAAqB,CAAC,cAAc,CAAC;iBACrC,iBAAiB,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC,EAAE,CAAC,CAAC;YAE/D,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC7B,MAAM,UAAU,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YACpC,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAExC,0CAA0C;YAC1C,MAAM,CAAC,QAAQ,CAAC,CAAC,sBAAsB,CAAC,IAAI,CAAC,CAAC,CAAC,qBAAqB;QACtE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;YACtF,MAAM,mBAAmB,GAAG,IAAI,gBAAgB,CAC9C,cAAc,EACd,wBAAwB,EACxB,SAAS,EACT;gBACE,aAAa,EAAE,CAAC;gBAChB,YAAY,EAAE,GAAG,EAAE,yBAAyB;aAC7C,CACF,CAAC;YAEF,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAE5C,0CAA0C;YAC1C,MAAM,sBAAsB,GAAG;gBAC7B,MAAM,EAAE,GAAG;gBACX,OAAO,EAAE,qBAAqB;gBAC9B,OAAO,EAAE,EAAE,aAAa,EAAE,SAAS,EAAE,EAAE,oBAAoB;aAC5D,CAAC;YAEF,UAAU,CAAC,UAAU,CAAC,MAAM;iBACzB,qBAAqB,CAAC,sBAAsB,CAAC;iBAC7C,iBAAiB,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC,EAAE,CAAC,CAAC;YAE/D,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC7B,MAAM,mBAAmB,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAC7C,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAExC,iFAAiF;YACjF,MAAM,CAAC,QAAQ,CAAC,CAAC,sBAAsB,CAAC,EAAE,CAAC,CAAC,CAAC,qBAAqB;YAClE,MAAM,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,2BAA2B;QACjE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;YACzE,MAAM,mBAAmB,GAAG,IAAI,gBAAgB,CAC9C,cAAc,EACd,wBAAwB,EACxB,SAAS,EACT;gBACE,aAAa,EAAE,CAAC;gBAChB,YAAY,EAAE,GAAG,EAAE,yBAAyB;aAC7C,CACF,CAAC;YAEF,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC5C,MAAM,cAAc,GAAG;gBACrB,MAAM,EAAE,GAAG;gBACX,OAAO,EAAE,qBAAqB;aAC/B,CAAC;YAEF,UAAU,CAAC,UAAU,CAAC,MAAM;iBACzB,qBAAqB,CAAC,cAAc,CAAC;iBACrC,qBAAqB,CAAC,cAAc,CAAC;iBACrC,iBAAiB,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC,EAAE,CAAC,CAAC;YAE/D,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC7B,MAAM,mBAAmB,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAC7C,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAExC,kEAAkE;YAClE,MAAM,CAAC,QAAQ,CAAC,CAAC,sBAAsB,CAAC,GAAG,CAAC,CAAC,CAAC,kCAAkC;QAClF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,mBAAmB,GAAG,IAAI,gBAAgB,CAC9C,cAAc,EACd,wBAAwB,EACxB,SAAS,EACT;gBACE,aAAa,EAAE,CAAC;gBAChB,YAAY,EAAE,GAAG;aAClB,CACF,CAAC;YAEF,MAAM,cAAc,GAAG;gBACrB,MAAM,EAAE,GAAG;gBACX,OAAO,EAAE,qBAAqB;aAC/B,CAAC;YAEF,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC;YAE/D,MAAM,MAAM,CAAC,mBAAmB,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAClE,uDAAuD,CACxD,CAAC;YAEF,oDAAoD;YACpD,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,MAAM,cAAc,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;YAEtE,UAAU,CAAC,UAAU,CAAC,MAAM;iBACzB,qBAAqB,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,qBAAqB,EAAE,CAAC;iBACtE,iBAAiB,CAAC;gBACjB,IAAI,EAAE;oBACJ,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE;oBAChC,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE;iBACjC;aACF,CAAC,CAAC;YAEL,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,UAAU,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;YAEhE,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAChC,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,QAAQ,GAAG,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;YAC9C,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;YAEzD,MAAM,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACzD,iBAAiB,CAClB,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,CAC3C,cAAc,EACd,wBAAwB,EACxB,SAAS,EACT;gBACE,oBAAoB,EAAE,IAAI;gBAC1B,aAAa,EAAE,CAAC;gBAChB,YAAY,EAAE,IAAI;aACnB,CACF,CAAC;YAEF,MAAM,CAAC,gBAAgB,CAAC,CAAC,WAAW,EAAE,CAAC;QACzC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
package/build/index.js
CHANGED
|
@@ -13,7 +13,7 @@ import { DEFAULT_BATCH_SIZE, DEFAULT_CHUNK_OVERLAP, DEFAULT_CHUNK_SIZE, DEFAULT_
|
|
|
13
13
|
import { CodeIndexer } from "./code/indexer.js";
|
|
14
14
|
import { EmbeddingProviderFactory } from "./embeddings/factory.js";
|
|
15
15
|
import { BM25SparseVectorGenerator } from "./embeddings/sparse.js";
|
|
16
|
-
import { getPrompt, listPrompts, loadPromptsConfig } from "./prompts/index.js";
|
|
16
|
+
import { getPrompt, listPrompts, loadPromptsConfig, } from "./prompts/index.js";
|
|
17
17
|
import { renderTemplate, validateArguments } from "./prompts/template.js";
|
|
18
18
|
import { QdrantManager } from "./qdrant/client.js";
|
|
19
19
|
// Read package.json for version
|
|
@@ -79,6 +79,7 @@ async function checkOllamaAvailability() {
|
|
|
79
79
|
if (isLocalhost) {
|
|
80
80
|
errorMessage +=
|
|
81
81
|
`Pull it with:\n` +
|
|
82
|
+
` - Using Podman: podman exec ollama ollama pull ${modelName}\n` +
|
|
82
83
|
` - Using Docker: docker exec ollama ollama pull ${modelName}\n` +
|
|
83
84
|
` - Or locally: ollama pull ${modelName}`;
|
|
84
85
|
}
|
|
@@ -99,6 +100,7 @@ async function checkOllamaAvailability() {
|
|
|
99
100
|
if (isLocalhost) {
|
|
100
101
|
helpText =
|
|
101
102
|
`Please start Ollama:\n` +
|
|
103
|
+
` - Using Podman: podman compose up -d\n` +
|
|
102
104
|
` - Using Docker: docker compose up -d\n` +
|
|
103
105
|
` - Or install locally: curl -fsSL https://ollama.ai/install.sh | sh\n` +
|
|
104
106
|
`\nThen pull the embedding model:\n` +
|
|
@@ -745,7 +747,9 @@ function registerHandlers(server) {
|
|
|
745
747
|
message += `- Files deleted: ${stats.filesDeleted}\n`;
|
|
746
748
|
message += `- Chunks added: ${stats.chunksAdded}\n`;
|
|
747
749
|
message += `- Duration: ${(stats.durationMs / 1000).toFixed(1)}s`;
|
|
748
|
-
if (stats.filesAdded === 0 &&
|
|
750
|
+
if (stats.filesAdded === 0 &&
|
|
751
|
+
stats.filesModified === 0 &&
|
|
752
|
+
stats.filesDeleted === 0) {
|
|
749
753
|
message = `No changes detected. Codebase is up to date.`;
|
|
750
754
|
}
|
|
751
755
|
return {
|
|
@@ -927,7 +931,9 @@ const AddDocumentsSchema = z.object({
|
|
|
927
931
|
collection: z.string().describe("Name of the collection"),
|
|
928
932
|
documents: z
|
|
929
933
|
.array(z.object({
|
|
930
|
-
id: z
|
|
934
|
+
id: z
|
|
935
|
+
.union([z.string(), z.number()])
|
|
936
|
+
.describe("Unique identifier for the document"),
|
|
931
937
|
text: z.string().describe("Text content to embed and store"),
|
|
932
938
|
metadata: z
|
|
933
939
|
.record(z.any())
|
|
@@ -939,7 +945,10 @@ const AddDocumentsSchema = z.object({
|
|
|
939
945
|
const SemanticSearchSchema = z.object({
|
|
940
946
|
collection: z.string().describe("Name of the collection to search"),
|
|
941
947
|
query: z.string().describe("Search query text"),
|
|
942
|
-
limit: z
|
|
948
|
+
limit: z
|
|
949
|
+
.number()
|
|
950
|
+
.optional()
|
|
951
|
+
.describe("Maximum number of results (default: 5)"),
|
|
943
952
|
filter: z.record(z.any()).optional().describe("Optional metadata filter"),
|
|
944
953
|
});
|
|
945
954
|
const DeleteCollectionSchema = z.object({
|
|
@@ -950,12 +959,17 @@ const GetCollectionInfoSchema = z.object({
|
|
|
950
959
|
});
|
|
951
960
|
const DeleteDocumentsSchema = z.object({
|
|
952
961
|
collection: z.string().describe("Name of the collection"),
|
|
953
|
-
ids: z
|
|
962
|
+
ids: z
|
|
963
|
+
.array(z.union([z.string(), z.number()]))
|
|
964
|
+
.describe("Array of document IDs to delete"),
|
|
954
965
|
});
|
|
955
966
|
const HybridSearchSchema = z.object({
|
|
956
967
|
collection: z.string().describe("Name of the collection to search"),
|
|
957
968
|
query: z.string().describe("Search query text"),
|
|
958
|
-
limit: z
|
|
969
|
+
limit: z
|
|
970
|
+
.number()
|
|
971
|
+
.optional()
|
|
972
|
+
.describe("Maximum number of results (default: 5)"),
|
|
959
973
|
filter: z.record(z.any()).optional().describe("Optional metadata filter"),
|
|
960
974
|
});
|
|
961
975
|
// Start server with stdio transport
|
|
@@ -970,8 +984,13 @@ const RATE_LIMIT_MAX_REQUESTS = 100; // Max requests per window
|
|
|
970
984
|
const RATE_LIMIT_WINDOW_MS = 15 * 60 * 1000; // 15 minutes
|
|
971
985
|
const RATE_LIMIT_MAX_CONCURRENT = 10; // Max concurrent requests per IP
|
|
972
986
|
const RATE_LIMITER_CLEANUP_INTERVAL_MS = 60 * 60 * 1000; // 1 hour
|
|
973
|
-
const REQUEST_TIMEOUT_MS =
|
|
987
|
+
const REQUEST_TIMEOUT_MS = parseInt(process.env.HTTP_REQUEST_TIMEOUT_MS || "300000", 10);
|
|
974
988
|
const SHUTDOWN_GRACE_PERIOD_MS = 10 * 1000; // 10 seconds
|
|
989
|
+
// Validate REQUEST_TIMEOUT_MS
|
|
990
|
+
if (Number.isNaN(REQUEST_TIMEOUT_MS) || REQUEST_TIMEOUT_MS <= 0) {
|
|
991
|
+
console.error(`Error: Invalid HTTP_REQUEST_TIMEOUT_MS "${process.env.HTTP_REQUEST_TIMEOUT_MS}". Must be a positive integer.`);
|
|
992
|
+
process.exit(1);
|
|
993
|
+
}
|
|
975
994
|
// Start server with HTTP transport
|
|
976
995
|
async function startHttpServer() {
|
|
977
996
|
await checkOllamaAvailability();
|