@timmeck/brain-core 2.36.62 → 2.36.63
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/dist/guardrails/guardrail-engine.d.ts +2 -0
- package/dist/guardrails/guardrail-engine.js +17 -9
- package/dist/guardrails/guardrail-engine.js.map +1 -1
- package/dist/index.d.ts +2 -0
- package/dist/index.js +1 -0
- package/dist/index.js.map +1 -1
- package/dist/utils/memory-watchdog.d.ts +33 -0
- package/dist/utils/memory-watchdog.js +88 -0
- package/dist/utils/memory-watchdog.js.map +1 -0
- package/package.json +1 -1
- package/dist/action/__tests__/action-bridge.test.d.ts +0 -1
- package/dist/action/__tests__/action-bridge.test.js +0 -175
- package/dist/action/__tests__/action-bridge.test.js.map +0 -1
- package/dist/action/__tests__/adjust-parameter-handler.test.d.ts +0 -1
- package/dist/action/__tests__/adjust-parameter-handler.test.js +0 -138
- package/dist/action/__tests__/adjust-parameter-handler.test.js.map +0 -1
- package/dist/action/__tests__/content-handler.test.d.ts +0 -1
- package/dist/action/__tests__/content-handler.test.js +0 -100
- package/dist/action/__tests__/content-handler.test.js.map +0 -1
- package/dist/action/__tests__/creative-seed-handler.test.d.ts +0 -1
- package/dist/action/__tests__/creative-seed-handler.test.js +0 -96
- package/dist/action/__tests__/creative-seed-handler.test.js.map +0 -1
- package/dist/action/__tests__/trade-handler.test.d.ts +0 -1
- package/dist/action/__tests__/trade-handler.test.js +0 -165
- package/dist/action/__tests__/trade-handler.test.js.map +0 -1
- package/dist/active-learning/__tests__/active-learning.test.d.ts +0 -1
- package/dist/active-learning/__tests__/active-learning.test.js +0 -132
- package/dist/active-learning/__tests__/active-learning.test.js.map +0 -1
- package/dist/agent-training/__tests__/agent-trainer.test.d.ts +0 -1
- package/dist/agent-training/__tests__/agent-trainer.test.js +0 -158
- package/dist/agent-training/__tests__/agent-trainer.test.js.map +0 -1
- package/dist/agent-training/__tests__/sub-agent-factory.test.d.ts +0 -1
- package/dist/agent-training/__tests__/sub-agent-factory.test.js +0 -100
- package/dist/agent-training/__tests__/sub-agent-factory.test.js.map +0 -1
- package/dist/agent-training/__tests__/sub-agent.test.d.ts +0 -1
- package/dist/agent-training/__tests__/sub-agent.test.js +0 -102
- package/dist/agent-training/__tests__/sub-agent.test.js.map +0 -1
- package/dist/chat/__tests__/chat-engine.test.d.ts +0 -1
- package/dist/chat/__tests__/chat-engine.test.js +0 -209
- package/dist/chat/__tests__/chat-engine.test.js.map +0 -1
- package/dist/checkpoint/__tests__/checkpoint-manager.test.d.ts +0 -1
- package/dist/checkpoint/__tests__/checkpoint-manager.test.js +0 -160
- package/dist/checkpoint/__tests__/checkpoint-manager.test.js.map +0 -1
- package/dist/code-health/__tests__/code-health.test.d.ts +0 -1
- package/dist/code-health/__tests__/code-health.test.js +0 -123
- package/dist/code-health/__tests__/code-health.test.js.map +0 -1
- package/dist/codegen/__tests__/code-forge.test.d.ts +0 -1
- package/dist/codegen/__tests__/code-forge.test.js +0 -105
- package/dist/codegen/__tests__/code-forge.test.js.map +0 -1
- package/dist/config/__tests__/loader.test.d.ts +0 -1
- package/dist/config/__tests__/loader.test.js +0 -86
- package/dist/config/__tests__/loader.test.js.map +0 -1
- package/dist/consciousness/__tests__/thought-stream.test.d.ts +0 -1
- package/dist/consciousness/__tests__/thought-stream.test.js +0 -62
- package/dist/consciousness/__tests__/thought-stream.test.js.map +0 -1
- package/dist/consensus/__tests__/consensus.test.d.ts +0 -1
- package/dist/consensus/__tests__/consensus.test.js +0 -159
- package/dist/consensus/__tests__/consensus.test.js.map +0 -1
- package/dist/content/__tests__/auto-publisher.test.d.ts +0 -1
- package/dist/content/__tests__/auto-publisher.test.js +0 -125
- package/dist/content/__tests__/auto-publisher.test.js.map +0 -1
- package/dist/content/__tests__/content-forge.test.d.ts +0 -1
- package/dist/content/__tests__/content-forge.test.js +0 -117
- package/dist/content/__tests__/content-forge.test.js.map +0 -1
- package/dist/creative/__tests__/creative-engine.test.d.ts +0 -1
- package/dist/creative/__tests__/creative-engine.test.js +0 -229
- package/dist/creative/__tests__/creative-engine.test.js.map +0 -1
- package/dist/cross-brain/__tests__/borg-sync-engine.test.d.ts +0 -1
- package/dist/cross-brain/__tests__/borg-sync-engine.test.js +0 -240
- package/dist/cross-brain/__tests__/borg-sync-engine.test.js.map +0 -1
- package/dist/cross-brain/__tests__/client.test.d.ts +0 -1
- package/dist/cross-brain/__tests__/client.test.js +0 -31
- package/dist/cross-brain/__tests__/client.test.js.map +0 -1
- package/dist/cross-brain/__tests__/notifications.test.d.ts +0 -1
- package/dist/cross-brain/__tests__/notifications.test.js +0 -52
- package/dist/cross-brain/__tests__/notifications.test.js.map +0 -1
- package/dist/cross-brain/__tests__/signal-router.test.d.ts +0 -1
- package/dist/cross-brain/__tests__/signal-router.test.js +0 -218
- package/dist/cross-brain/__tests__/signal-router.test.js.map +0 -1
- package/dist/cross-brain/__tests__/subscription.test.d.ts +0 -1
- package/dist/cross-brain/__tests__/subscription.test.js +0 -75
- package/dist/cross-brain/__tests__/subscription.test.js.map +0 -1
- package/dist/dashboard/__tests__/command-center-server.test.d.ts +0 -1
- package/dist/dashboard/__tests__/command-center-server.test.js +0 -522
- package/dist/dashboard/__tests__/command-center-server.test.js.map +0 -1
- package/dist/dashboard/__tests__/server.test.d.ts +0 -1
- package/dist/dashboard/__tests__/server.test.js +0 -105
- package/dist/dashboard/__tests__/server.test.js.map +0 -1
- package/dist/db/__tests__/connection.test.d.ts +0 -1
- package/dist/db/__tests__/connection.test.js +0 -135
- package/dist/db/__tests__/connection.test.js.map +0 -1
- package/dist/dream/__tests__/dream-engine.test.d.ts +0 -1
- package/dist/dream/__tests__/dream-engine.test.js +0 -184
- package/dist/dream/__tests__/dream-engine.test.js.map +0 -1
- package/dist/embeddings/__tests__/engine.test.d.ts +0 -1
- package/dist/embeddings/__tests__/engine.test.js +0 -82
- package/dist/embeddings/__tests__/engine.test.js.map +0 -1
- package/dist/feedback/__tests__/feedback-engine.test.d.ts +0 -1
- package/dist/feedback/__tests__/feedback-engine.test.js +0 -156
- package/dist/feedback/__tests__/feedback-engine.test.js.map +0 -1
- package/dist/feedback/__tests__/feedback-router.test.d.ts +0 -1
- package/dist/feedback/__tests__/feedback-router.test.js +0 -173
- package/dist/feedback/__tests__/feedback-router.test.js.map +0 -1
- package/dist/goals/__tests__/goal-engine.test.d.ts +0 -1
- package/dist/goals/__tests__/goal-engine.test.js +0 -203
- package/dist/goals/__tests__/goal-engine.test.js.map +0 -1
- package/dist/guardrails/__tests__/guardrail-engine.test.d.ts +0 -1
- package/dist/guardrails/__tests__/guardrail-engine.test.js +0 -343
- package/dist/guardrails/__tests__/guardrail-engine.test.js.map +0 -1
- package/dist/ipc/__tests__/integration.test.d.ts +0 -1
- package/dist/ipc/__tests__/integration.test.js +0 -352
- package/dist/ipc/__tests__/integration.test.js.map +0 -1
- package/dist/ipc/__tests__/protocol.test.d.ts +0 -1
- package/dist/ipc/__tests__/protocol.test.js +0 -69
- package/dist/ipc/__tests__/protocol.test.js.map +0 -1
- package/dist/knowledge-graph/__tests__/contradiction-resolver.test.d.ts +0 -1
- package/dist/knowledge-graph/__tests__/contradiction-resolver.test.js +0 -114
- package/dist/knowledge-graph/__tests__/contradiction-resolver.test.js.map +0 -1
- package/dist/knowledge-graph/__tests__/knowledge-graph.test.d.ts +0 -1
- package/dist/knowledge-graph/__tests__/knowledge-graph.test.js +0 -215
- package/dist/knowledge-graph/__tests__/knowledge-graph.test.js.map +0 -1
- package/dist/learning/__tests__/base-engine.test.d.ts +0 -1
- package/dist/learning/__tests__/base-engine.test.js +0 -49
- package/dist/learning/__tests__/base-engine.test.js.map +0 -1
- package/dist/llm/__tests__/anthropic-provider.test.d.ts +0 -1
- package/dist/llm/__tests__/anthropic-provider.test.js +0 -121
- package/dist/llm/__tests__/anthropic-provider.test.js.map +0 -1
- package/dist/llm/__tests__/llm-service.test.d.ts +0 -1
- package/dist/llm/__tests__/llm-service.test.js +0 -507
- package/dist/llm/__tests__/llm-service.test.js.map +0 -1
- package/dist/llm/__tests__/ollama-embedding.test.d.ts +0 -1
- package/dist/llm/__tests__/ollama-embedding.test.js +0 -128
- package/dist/llm/__tests__/ollama-embedding.test.js.map +0 -1
- package/dist/llm/__tests__/ollama-provider.test.d.ts +0 -1
- package/dist/llm/__tests__/ollama-provider.test.js +0 -213
- package/dist/llm/__tests__/ollama-provider.test.js.map +0 -1
- package/dist/llm/__tests__/provider.test.d.ts +0 -1
- package/dist/llm/__tests__/provider.test.js +0 -126
- package/dist/llm/__tests__/provider.test.js.map +0 -1
- package/dist/math/__tests__/time-decay.test.d.ts +0 -1
- package/dist/math/__tests__/time-decay.test.js +0 -37
- package/dist/math/__tests__/time-decay.test.js.map +0 -1
- package/dist/math/__tests__/wilson-score.test.d.ts +0 -1
- package/dist/math/__tests__/wilson-score.test.js +0 -43
- package/dist/math/__tests__/wilson-score.test.js.map +0 -1
- package/dist/memory/__tests__/base-memory-engine.test.d.ts +0 -1
- package/dist/memory/__tests__/base-memory-engine.test.js +0 -233
- package/dist/memory/__tests__/base-memory-engine.test.js.map +0 -1
- package/dist/metacognition/__tests__/auto-experiment-engine.test.d.ts +0 -1
- package/dist/metacognition/__tests__/auto-experiment-engine.test.js +0 -218
- package/dist/metacognition/__tests__/auto-experiment-engine.test.js.map +0 -1
- package/dist/metacognition/__tests__/evolution-engine.test.d.ts +0 -1
- package/dist/metacognition/__tests__/evolution-engine.test.js +0 -473
- package/dist/metacognition/__tests__/evolution-engine.test.js.map +0 -1
- package/dist/metacognition/__tests__/parameter-registry.test.d.ts +0 -1
- package/dist/metacognition/__tests__/parameter-registry.test.js +0 -223
- package/dist/metacognition/__tests__/parameter-registry.test.js.map +0 -1
- package/dist/missions/__tests__/mission-engine.test.d.ts +0 -1
- package/dist/missions/__tests__/mission-engine.test.js +0 -67
- package/dist/missions/__tests__/mission-engine.test.js.map +0 -1
- package/dist/notifications/__tests__/notification-service.test.d.ts +0 -1
- package/dist/notifications/__tests__/notification-service.test.js +0 -176
- package/dist/notifications/__tests__/notification-service.test.js.map +0 -1
- package/dist/plugin/__tests__/plugin-registry.test.d.ts +0 -1
- package/dist/plugin/__tests__/plugin-registry.test.js +0 -166
- package/dist/plugin/__tests__/plugin-registry.test.js.map +0 -1
- package/dist/prediction/__tests__/prediction-engine.test.d.ts +0 -1
- package/dist/prediction/__tests__/prediction-engine.test.js +0 -252
- package/dist/prediction/__tests__/prediction-engine.test.js.map +0 -1
- package/dist/proactive/__tests__/proactive-engine.test.d.ts +0 -1
- package/dist/proactive/__tests__/proactive-engine.test.js +0 -183
- package/dist/proactive/__tests__/proactive-engine.test.js.map +0 -1
- package/dist/rag/__tests__/rag-engine.test.d.ts +0 -1
- package/dist/rag/__tests__/rag-engine.test.js +0 -235
- package/dist/rag/__tests__/rag-engine.test.js.map +0 -1
- package/dist/research/__tests__/autonomy-features.test.d.ts +0 -1
- package/dist/research/__tests__/autonomy-features.test.js +0 -155
- package/dist/research/__tests__/autonomy-features.test.js.map +0 -1
- package/dist/research/__tests__/base-engine.test.d.ts +0 -1
- package/dist/research/__tests__/base-engine.test.js +0 -56
- package/dist/research/__tests__/base-engine.test.js.map +0 -1
- package/dist/research/__tests__/semantic-compressor.test.d.ts +0 -1
- package/dist/research/__tests__/semantic-compressor.test.js +0 -153
- package/dist/research/__tests__/semantic-compressor.test.js.map +0 -1
- package/dist/research/adapters/__tests__/playwright-adapter.test.d.ts +0 -1
- package/dist/research/adapters/__tests__/playwright-adapter.test.js +0 -64
- package/dist/research/adapters/__tests__/playwright-adapter.test.js.map +0 -1
- package/dist/research/adapters/__tests__/web-adapters.test.d.ts +0 -1
- package/dist/research/adapters/__tests__/web-adapters.test.js +0 -106
- package/dist/research/adapters/__tests__/web-adapters.test.js.map +0 -1
- package/dist/sandbox/__tests__/code-sandbox.test.d.ts +0 -1
- package/dist/sandbox/__tests__/code-sandbox.test.js +0 -182
- package/dist/sandbox/__tests__/code-sandbox.test.js.map +0 -1
- package/dist/scanner/__tests__/signal-scanner.test.d.ts +0 -1
- package/dist/scanner/__tests__/signal-scanner.test.js +0 -142
- package/dist/scanner/__tests__/signal-scanner.test.js.map +0 -1
- package/dist/self-modification/__tests__/self-modification-engine.test.d.ts +0 -1
- package/dist/self-modification/__tests__/self-modification-engine.test.js +0 -313
- package/dist/self-modification/__tests__/self-modification-engine.test.js.map +0 -1
- package/dist/strategy/__tests__/strategy-forge.test.d.ts +0 -1
- package/dist/strategy/__tests__/strategy-forge.test.js +0 -190
- package/dist/strategy/__tests__/strategy-forge.test.js.map +0 -1
- package/dist/strategy/__tests__/strategy-mutator.test.d.ts +0 -1
- package/dist/strategy/__tests__/strategy-mutator.test.js +0 -141
- package/dist/strategy/__tests__/strategy-mutator.test.js.map +0 -1
- package/dist/synapses/__tests__/activation.test.d.ts +0 -1
- package/dist/synapses/__tests__/activation.test.js +0 -87
- package/dist/synapses/__tests__/activation.test.js.map +0 -1
- package/dist/synapses/__tests__/decay.test.d.ts +0 -1
- package/dist/synapses/__tests__/decay.test.js +0 -73
- package/dist/synapses/__tests__/decay.test.js.map +0 -1
- package/dist/synapses/__tests__/hebbian.test.d.ts +0 -1
- package/dist/synapses/__tests__/hebbian.test.js +0 -95
- package/dist/synapses/__tests__/hebbian.test.js.map +0 -1
- package/dist/synapses/__tests__/pathfinder.test.d.ts +0 -1
- package/dist/synapses/__tests__/pathfinder.test.js +0 -74
- package/dist/synapses/__tests__/pathfinder.test.js.map +0 -1
- package/dist/synapses/__tests__/synapse-manager.test.d.ts +0 -1
- package/dist/synapses/__tests__/synapse-manager.test.js +0 -94
- package/dist/synapses/__tests__/synapse-manager.test.js.map +0 -1
- package/dist/teaching/__tests__/teaching.test.d.ts +0 -1
- package/dist/teaching/__tests__/teaching.test.js +0 -252
- package/dist/teaching/__tests__/teaching.test.js.map +0 -1
- package/dist/techradar/__tests__/techradar-engine.test.d.ts +0 -1
- package/dist/techradar/__tests__/techradar-engine.test.js +0 -246
- package/dist/techradar/__tests__/techradar-engine.test.js.map +0 -1
- package/dist/tool-learning/__tests__/tool-learning.test.d.ts +0 -1
- package/dist/tool-learning/__tests__/tool-learning.test.js +0 -187
- package/dist/tool-learning/__tests__/tool-learning.test.js.map +0 -1
- package/dist/user-model/__tests__/user-model.test.d.ts +0 -1
- package/dist/user-model/__tests__/user-model.test.js +0 -142
- package/dist/user-model/__tests__/user-model.test.js.map +0 -1
- package/dist/utils/__tests__/batch-queue.test.d.ts +0 -1
- package/dist/utils/__tests__/batch-queue.test.js +0 -85
- package/dist/utils/__tests__/batch-queue.test.js.map +0 -1
- package/dist/utils/__tests__/events.test.d.ts +0 -1
- package/dist/utils/__tests__/events.test.js +0 -38
- package/dist/utils/__tests__/events.test.js.map +0 -1
- package/dist/utils/__tests__/hash.test.d.ts +0 -1
- package/dist/utils/__tests__/hash.test.js +0 -25
- package/dist/utils/__tests__/hash.test.js.map +0 -1
- package/dist/utils/__tests__/logger.test.d.ts +0 -1
- package/dist/utils/__tests__/logger.test.js +0 -55
- package/dist/utils/__tests__/logger.test.js.map +0 -1
- package/dist/utils/__tests__/paths.test.d.ts +0 -1
- package/dist/utils/__tests__/paths.test.js +0 -50
- package/dist/utils/__tests__/paths.test.js.map +0 -1
- package/dist/utils/__tests__/retry.test.d.ts +0 -1
- package/dist/utils/__tests__/retry.test.js +0 -91
- package/dist/utils/__tests__/retry.test.js.map +0 -1
- package/dist/watchdog/__tests__/watchdog-service.test.d.ts +0 -1
- package/dist/watchdog/__tests__/watchdog-service.test.js +0 -113
- package/dist/watchdog/__tests__/watchdog-service.test.js.map +0 -1
|
@@ -1,128 +0,0 @@
|
|
|
1
|
-
import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
|
|
2
|
-
import { OllamaEmbeddingProvider } from '../ollama-embedding.js';
|
|
3
|
-
vi.mock('../../utils/logger.js', () => ({
|
|
4
|
-
getLogger: () => ({
|
|
5
|
-
info: vi.fn(),
|
|
6
|
-
warn: vi.fn(),
|
|
7
|
-
error: vi.fn(),
|
|
8
|
-
debug: vi.fn(),
|
|
9
|
-
}),
|
|
10
|
-
}));
|
|
11
|
-
describe('OllamaEmbeddingProvider', () => {
|
|
12
|
-
let fetchMock;
|
|
13
|
-
beforeEach(() => {
|
|
14
|
-
fetchMock = vi.fn();
|
|
15
|
-
vi.stubGlobal('fetch', fetchMock);
|
|
16
|
-
});
|
|
17
|
-
afterEach(() => {
|
|
18
|
-
vi.unstubAllGlobals();
|
|
19
|
-
});
|
|
20
|
-
describe('isAvailable', () => {
|
|
21
|
-
it('returns true when Ollama responds', async () => {
|
|
22
|
-
fetchMock.mockResolvedValueOnce({ ok: true });
|
|
23
|
-
const provider = new OllamaEmbeddingProvider();
|
|
24
|
-
expect(await provider.isAvailable()).toBe(true);
|
|
25
|
-
});
|
|
26
|
-
it('returns false when Ollama is down', async () => {
|
|
27
|
-
fetchMock.mockRejectedValueOnce(new Error('ECONNREFUSED'));
|
|
28
|
-
const provider = new OllamaEmbeddingProvider();
|
|
29
|
-
expect(await provider.isAvailable()).toBe(false);
|
|
30
|
-
});
|
|
31
|
-
it('caches availability result', async () => {
|
|
32
|
-
fetchMock.mockResolvedValueOnce({ ok: true });
|
|
33
|
-
const provider = new OllamaEmbeddingProvider();
|
|
34
|
-
await provider.isAvailable();
|
|
35
|
-
await provider.isAvailable();
|
|
36
|
-
expect(fetchMock).toHaveBeenCalledTimes(1);
|
|
37
|
-
});
|
|
38
|
-
it('resetAvailability clears cache', async () => {
|
|
39
|
-
fetchMock.mockResolvedValue({ ok: true });
|
|
40
|
-
const provider = new OllamaEmbeddingProvider();
|
|
41
|
-
await provider.isAvailable();
|
|
42
|
-
provider.resetAvailability();
|
|
43
|
-
await provider.isAvailable();
|
|
44
|
-
expect(fetchMock).toHaveBeenCalledTimes(2);
|
|
45
|
-
});
|
|
46
|
-
});
|
|
47
|
-
describe('embed', () => {
|
|
48
|
-
it('returns Float32Array embedding', async () => {
|
|
49
|
-
fetchMock.mockResolvedValueOnce({
|
|
50
|
-
ok: true,
|
|
51
|
-
json: async () => ({ embeddings: [[0.1, 0.2, 0.3, 0.4, 0.5]] }),
|
|
52
|
-
});
|
|
53
|
-
const provider = new OllamaEmbeddingProvider({ host: 'http://test:11434' });
|
|
54
|
-
const result = await provider.embed('test text');
|
|
55
|
-
expect(result).toBeInstanceOf(Float32Array);
|
|
56
|
-
expect(result.length).toBe(5);
|
|
57
|
-
expect(result[0]).toBeCloseTo(0.1);
|
|
58
|
-
const body = JSON.parse(fetchMock.mock.calls[0][1].body);
|
|
59
|
-
expect(body.model).toBe('qwen3-embedding:8b');
|
|
60
|
-
expect(body.input).toBe('test text');
|
|
61
|
-
});
|
|
62
|
-
it('uses custom model', async () => {
|
|
63
|
-
fetchMock.mockResolvedValueOnce({
|
|
64
|
-
ok: true,
|
|
65
|
-
json: async () => ({ embeddings: [[0.1]] }),
|
|
66
|
-
});
|
|
67
|
-
const provider = new OllamaEmbeddingProvider({ model: 'mxbai-embed-large' });
|
|
68
|
-
await provider.embed('test');
|
|
69
|
-
const body = JSON.parse(fetchMock.mock.calls[0][1].body);
|
|
70
|
-
expect(body.model).toBe('mxbai-embed-large');
|
|
71
|
-
});
|
|
72
|
-
it('throws on error response', async () => {
|
|
73
|
-
fetchMock.mockResolvedValueOnce({
|
|
74
|
-
ok: false,
|
|
75
|
-
status: 404,
|
|
76
|
-
text: async () => 'model not found',
|
|
77
|
-
});
|
|
78
|
-
const provider = new OllamaEmbeddingProvider();
|
|
79
|
-
await expect(provider.embed('test')).rejects.toThrow('Ollama embed error (404)');
|
|
80
|
-
});
|
|
81
|
-
it('returns empty Float32Array when no embeddings', async () => {
|
|
82
|
-
fetchMock.mockResolvedValueOnce({
|
|
83
|
-
ok: true,
|
|
84
|
-
json: async () => ({}),
|
|
85
|
-
});
|
|
86
|
-
const provider = new OllamaEmbeddingProvider();
|
|
87
|
-
const result = await provider.embed('test');
|
|
88
|
-
expect(result).toBeInstanceOf(Float32Array);
|
|
89
|
-
expect(result.length).toBe(0);
|
|
90
|
-
});
|
|
91
|
-
});
|
|
92
|
-
describe('embedBatch', () => {
|
|
93
|
-
it('processes multiple texts', async () => {
|
|
94
|
-
fetchMock
|
|
95
|
-
.mockResolvedValueOnce({ ok: true, json: async () => ({ embeddings: [[0.1, 0.2]] }) })
|
|
96
|
-
.mockResolvedValueOnce({ ok: true, json: async () => ({ embeddings: [[0.3, 0.4]] }) });
|
|
97
|
-
const provider = new OllamaEmbeddingProvider();
|
|
98
|
-
const results = await provider.embedBatch(['text1', 'text2']);
|
|
99
|
-
expect(results).toHaveLength(2);
|
|
100
|
-
expect(results[0]).toBeInstanceOf(Float32Array);
|
|
101
|
-
expect(results[1]).toBeInstanceOf(Float32Array);
|
|
102
|
-
});
|
|
103
|
-
it('returns null for failed items', async () => {
|
|
104
|
-
fetchMock
|
|
105
|
-
.mockResolvedValueOnce({ ok: true, json: async () => ({ embeddings: [[0.1]] }) })
|
|
106
|
-
.mockRejectedValueOnce(new Error('failed'));
|
|
107
|
-
const provider = new OllamaEmbeddingProvider();
|
|
108
|
-
const results = await provider.embedBatch(['ok', 'fail']);
|
|
109
|
-
expect(results[0]).toBeInstanceOf(Float32Array);
|
|
110
|
-
expect(results[1]).toBeNull();
|
|
111
|
-
});
|
|
112
|
-
});
|
|
113
|
-
describe('config', () => {
|
|
114
|
-
it('getModel returns configured model', () => {
|
|
115
|
-
const provider = new OllamaEmbeddingProvider({ model: 'custom-model' });
|
|
116
|
-
expect(provider.getModel()).toBe('custom-model');
|
|
117
|
-
});
|
|
118
|
-
it('getHost returns configured host', () => {
|
|
119
|
-
const provider = new OllamaEmbeddingProvider({ host: 'http://custom:1234' });
|
|
120
|
-
expect(provider.getHost()).toBe('http://custom:1234');
|
|
121
|
-
});
|
|
122
|
-
it('strips trailing slash from host', () => {
|
|
123
|
-
const provider = new OllamaEmbeddingProvider({ host: 'http://localhost:11434/' });
|
|
124
|
-
expect(provider.getHost()).toBe('http://localhost:11434');
|
|
125
|
-
});
|
|
126
|
-
});
|
|
127
|
-
});
|
|
128
|
-
//# sourceMappingURL=ollama-embedding.test.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"ollama-embedding.test.js","sourceRoot":"","sources":["../../../src/llm/__tests__/ollama-embedding.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,uBAAuB,EAAE,MAAM,wBAAwB,CAAC;AAEjE,EAAE,CAAC,IAAI,CAAC,uBAAuB,EAAE,GAAG,EAAE,CAAC,CAAC;IACtC,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC;QAChB,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;QACb,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;QACb,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;QACd,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;KACf,CAAC;CACH,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;IACvC,IAAI,SAAmC,CAAC;IAExC,UAAU,CAAC,GAAG,EAAE;QACd,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACpB,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,gBAAgB,EAAE,CAAC;IACxB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;YACjD,SAAS,CAAC,qBAAqB,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;YAC9C,MAAM,QAAQ,GAAG,IAAI,uBAAuB,EAAE,CAAC;YAC/C,MAAM,CAAC,MAAM,QAAQ,CAAC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;YACjD,SAAS,CAAC,qBAAqB,CAAC,IAAI,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC;YAC3D,MAAM,QAAQ,GAAG,IAAI,uBAAuB,EAAE,CAAC;YAC/C,MAAM,CAAC,MAAM,QAAQ,CAAC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;YAC1C,SAAS,CAAC,qBAAqB,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;YAC9C,MAAM,QAAQ,GAAG,IAAI,uBAAuB,EAAE,CAAC;YAC/C,MAAM,QAAQ,CAAC,WAAW,EAAE,CAAC;YAC7B,MAAM,QAAQ,CAAC,WAAW,EAAE,CAAC;YAC7B,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAC9C,SAAS,CAAC,iBAAiB,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;YAC1C,MAAM,QAAQ,GAAG,IAAI,uBAAuB,EAAE,CAAC;YAC/C,MAAM,QAAQ,CAAC,WAAW,EAAE,CAAC;YAC7B,QAAQ,CAAC,iBAAiB,EAAE,CAAC;YAC7B,MAAM,QAAQ,CAAC,WAAW,EAAE,CAAC;YAC7B,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,OAAO,EAAE,GAAG,EAAE;QACrB,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAC9C,SAAS,CAAC,qBAAqB,CAAC;gBAC9B,EAAE,EAAE,IAAI;gBACR,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,EAAE,CAAC;aAChE,CAAC,CAAC;YAEH,MAAM,QAAQ,GAAG,IAAI,uBAAuB,CAAC,EAAE,IAAI,EAAE,mBAAmB,EAAE,CAAC,CAAC;YAC5E,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAEjD,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;YAC5C,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC9B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;YAEnC,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YACzD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;YAC9C,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mBAAmB,EAAE,KAAK,IAAI,EAAE;YACjC,SAAS,CAAC,qBAAqB,CAAC;gBAC9B,EAAE,EAAE,IAAI;gBACR,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;aAC5C,CAAC,CAAC;YAEH,MAAM,QAAQ,GAAG,IAAI,uBAAuB,CAAC,EAAE,KAAK,EAAE,mBAAmB,EAAE,CAAC,CAAC;YAC7E,MAAM,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAE7B,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YACzD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;YACxC,SAAS,CAAC,qBAAqB,CAAC;gBAC9B,EAAE,EAAE,KAAK;gBACT,MAAM,EAAE,GAAG;gBACX,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,iBAAiB;aACpC,CAAC,CAAC;YAEH,MAAM,QAAQ,GAAG,IAAI,uBAAuB,EAAE,CAAC;YAC/C,MAAM,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,0BAA0B,CAAC,CAAC;QACnF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,SAAS,CAAC,qBAAqB,CAAC;gBAC9B,EAAE,EAAE,IAAI;gBACR,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;aACvB,CAAC,CAAC;YAEH,MAAM,QAAQ,GAAG,IAAI,uBAAuB,EAAE,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAC5C,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;YAC5C,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;YACxC,SAAS;iBACN,qBAAqB,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;iBACrF,qBAAqB,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;YAEzF,MAAM,QAAQ,GAAG,IAAI,uBAAuB,EAAE,CAAC;YAC/C,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,UAAU,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;YAE9D,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAChC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;YAChD,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;YAC7C,SAAS;iBACN,qBAAqB,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;iBAChF,qBAAqB,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC;YAE9C,MAAM,QAAQ,GAAG,IAAI,uBAAuB,EAAE,CAAC;YAC/C,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,UAAU,CAAC,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC;YAE1D,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;YAChD,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;QAChC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,QAAQ,EAAE,GAAG,EAAE;QACtB,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;YAC3C,MAAM,QAAQ,GAAG,IAAI,uBAAuB,CAAC,EAAE,KAAK,EAAE,cAAc,EAAE,CAAC,CAAC;YACxE,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,QAAQ,GAAG,IAAI,uBAAuB,CAAC,EAAE,IAAI,EAAE,oBAAoB,EAAE,CAAC,CAAC;YAC7E,MAAM,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QACxD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,QAAQ,GAAG,IAAI,uBAAuB,CAAC,EAAE,IAAI,EAAE,yBAAyB,EAAE,CAAC,CAAC;YAClF,MAAM,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
export {};
|
|
@@ -1,213 +0,0 @@
|
|
|
1
|
-
import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
|
|
2
|
-
import { OllamaProvider } from '../ollama-provider.js';
|
|
3
|
-
vi.mock('../../utils/logger.js', () => ({
|
|
4
|
-
getLogger: () => ({
|
|
5
|
-
info: vi.fn(),
|
|
6
|
-
warn: vi.fn(),
|
|
7
|
-
error: vi.fn(),
|
|
8
|
-
debug: vi.fn(),
|
|
9
|
-
}),
|
|
10
|
-
}));
|
|
11
|
-
describe('OllamaProvider', () => {
|
|
12
|
-
let fetchMock;
|
|
13
|
-
beforeEach(() => {
|
|
14
|
-
fetchMock = vi.fn();
|
|
15
|
-
vi.stubGlobal('fetch', fetchMock);
|
|
16
|
-
});
|
|
17
|
-
afterEach(() => {
|
|
18
|
-
vi.unstubAllGlobals();
|
|
19
|
-
});
|
|
20
|
-
describe('properties', () => {
|
|
21
|
-
it('has correct name and cost tier', () => {
|
|
22
|
-
const provider = new OllamaProvider();
|
|
23
|
-
expect(provider.name).toBe('ollama');
|
|
24
|
-
expect(provider.costTier).toBe('free');
|
|
25
|
-
});
|
|
26
|
-
it('has correct capabilities', () => {
|
|
27
|
-
const provider = new OllamaProvider();
|
|
28
|
-
expect(provider.capabilities.chat).toBe(true);
|
|
29
|
-
expect(provider.capabilities.generate).toBe(true);
|
|
30
|
-
expect(provider.capabilities.embed).toBe(true);
|
|
31
|
-
expect(provider.capabilities.reasoning).toBe(false);
|
|
32
|
-
});
|
|
33
|
-
it('uses default host and model', () => {
|
|
34
|
-
const provider = new OllamaProvider();
|
|
35
|
-
// Access via getStatus which includes these
|
|
36
|
-
expect(provider.name).toBe('ollama');
|
|
37
|
-
});
|
|
38
|
-
it('accepts custom config', () => {
|
|
39
|
-
const provider = new OllamaProvider({
|
|
40
|
-
host: 'http://custom:1234',
|
|
41
|
-
model: 'llama3:latest',
|
|
42
|
-
embedModel: 'mxbai-embed-large',
|
|
43
|
-
});
|
|
44
|
-
expect(provider.name).toBe('ollama');
|
|
45
|
-
});
|
|
46
|
-
});
|
|
47
|
-
describe('isAvailable', () => {
|
|
48
|
-
it('returns true when Ollama responds', async () => {
|
|
49
|
-
fetchMock.mockResolvedValueOnce({ ok: true });
|
|
50
|
-
const provider = new OllamaProvider();
|
|
51
|
-
expect(await provider.isAvailable()).toBe(true);
|
|
52
|
-
});
|
|
53
|
-
it('returns false when Ollama is down', async () => {
|
|
54
|
-
fetchMock.mockRejectedValueOnce(new Error('ECONNREFUSED'));
|
|
55
|
-
const provider = new OllamaProvider();
|
|
56
|
-
expect(await provider.isAvailable()).toBe(false);
|
|
57
|
-
});
|
|
58
|
-
it('caches availability for 30s', async () => {
|
|
59
|
-
fetchMock.mockResolvedValue({ ok: true });
|
|
60
|
-
const provider = new OllamaProvider();
|
|
61
|
-
await provider.isAvailable();
|
|
62
|
-
await provider.isAvailable();
|
|
63
|
-
// Only one fetch call (cached)
|
|
64
|
-
expect(fetchMock).toHaveBeenCalledTimes(1);
|
|
65
|
-
});
|
|
66
|
-
it('calls /api/tags endpoint', async () => {
|
|
67
|
-
fetchMock.mockResolvedValueOnce({ ok: true });
|
|
68
|
-
const provider = new OllamaProvider({ host: 'http://test:11434' });
|
|
69
|
-
await provider.isAvailable();
|
|
70
|
-
expect(fetchMock.mock.calls[0][0]).toBe('http://test:11434/api/tags');
|
|
71
|
-
});
|
|
72
|
-
});
|
|
73
|
-
describe('chat', () => {
|
|
74
|
-
it('sends correct request', async () => {
|
|
75
|
-
fetchMock.mockResolvedValueOnce({
|
|
76
|
-
ok: true,
|
|
77
|
-
json: async () => ({
|
|
78
|
-
message: { content: 'Ollama response' },
|
|
79
|
-
prompt_eval_count: 50,
|
|
80
|
-
eval_count: 30,
|
|
81
|
-
}),
|
|
82
|
-
});
|
|
83
|
-
const provider = new OllamaProvider({ host: 'http://test:11434', model: 'qwen3:14b' });
|
|
84
|
-
const result = await provider.chat([
|
|
85
|
-
{ role: 'system', content: 'Be helpful' },
|
|
86
|
-
{ role: 'user', content: 'Hello' },
|
|
87
|
-
]);
|
|
88
|
-
expect(result.text).toBe('Ollama response');
|
|
89
|
-
expect(result.inputTokens).toBe(50);
|
|
90
|
-
expect(result.outputTokens).toBe(30);
|
|
91
|
-
expect(result.model).toBe('qwen3:14b');
|
|
92
|
-
const body = JSON.parse(fetchMock.mock.calls[0][1].body);
|
|
93
|
-
expect(body.model).toBe('qwen3:14b');
|
|
94
|
-
expect(body.stream).toBe(false);
|
|
95
|
-
expect(body.messages).toHaveLength(2);
|
|
96
|
-
});
|
|
97
|
-
it('strips thinking tags from response', async () => {
|
|
98
|
-
fetchMock.mockResolvedValueOnce({
|
|
99
|
-
ok: true,
|
|
100
|
-
json: async () => ({
|
|
101
|
-
message: { content: '<think>Let me reason about this...</think>\n\nThe answer is 42' },
|
|
102
|
-
prompt_eval_count: 10,
|
|
103
|
-
eval_count: 20,
|
|
104
|
-
}),
|
|
105
|
-
});
|
|
106
|
-
const provider = new OllamaProvider();
|
|
107
|
-
const result = await provider.chat([{ role: 'user', content: 'test' }]);
|
|
108
|
-
expect(result.text).toBe('The answer is 42');
|
|
109
|
-
});
|
|
110
|
-
it('passes temperature and maxTokens', async () => {
|
|
111
|
-
fetchMock.mockResolvedValueOnce({
|
|
112
|
-
ok: true,
|
|
113
|
-
json: async () => ({
|
|
114
|
-
message: { content: 'ok' },
|
|
115
|
-
prompt_eval_count: 5,
|
|
116
|
-
eval_count: 5,
|
|
117
|
-
}),
|
|
118
|
-
});
|
|
119
|
-
const provider = new OllamaProvider();
|
|
120
|
-
await provider.chat([{ role: 'user', content: 'test' }], { temperature: 0.5, maxTokens: 256 });
|
|
121
|
-
const body = JSON.parse(fetchMock.mock.calls[0][1].body);
|
|
122
|
-
expect(body.options.temperature).toBe(0.5);
|
|
123
|
-
expect(body.options.num_predict).toBe(256);
|
|
124
|
-
});
|
|
125
|
-
it('throws on error response', async () => {
|
|
126
|
-
fetchMock.mockResolvedValueOnce({
|
|
127
|
-
ok: false,
|
|
128
|
-
status: 500,
|
|
129
|
-
text: async () => 'Internal error',
|
|
130
|
-
});
|
|
131
|
-
const provider = new OllamaProvider();
|
|
132
|
-
await expect(provider.chat([{ role: 'user', content: 'test' }]))
|
|
133
|
-
.rejects.toThrow('Ollama chat error (500)');
|
|
134
|
-
});
|
|
135
|
-
});
|
|
136
|
-
describe('generate', () => {
|
|
137
|
-
it('sends request to /api/generate', async () => {
|
|
138
|
-
fetchMock.mockResolvedValueOnce({
|
|
139
|
-
ok: true,
|
|
140
|
-
json: async () => ({ response: 'Generated text' }),
|
|
141
|
-
});
|
|
142
|
-
const provider = new OllamaProvider({ host: 'http://test:11434' });
|
|
143
|
-
const result = await provider.generate('Write a poem');
|
|
144
|
-
expect(result).toBe('Generated text');
|
|
145
|
-
expect(fetchMock.mock.calls[0][0]).toBe('http://test:11434/api/generate');
|
|
146
|
-
});
|
|
147
|
-
it('strips thinking tags', async () => {
|
|
148
|
-
fetchMock.mockResolvedValueOnce({
|
|
149
|
-
ok: true,
|
|
150
|
-
json: async () => ({ response: '<think>hmm</think>\nClean output' }),
|
|
151
|
-
});
|
|
152
|
-
const provider = new OllamaProvider();
|
|
153
|
-
const result = await provider.generate('test');
|
|
154
|
-
expect(result).toBe('Clean output');
|
|
155
|
-
});
|
|
156
|
-
});
|
|
157
|
-
describe('embed', () => {
|
|
158
|
-
it('sends request to /api/embed', async () => {
|
|
159
|
-
fetchMock.mockResolvedValueOnce({
|
|
160
|
-
ok: true,
|
|
161
|
-
json: async () => ({ embeddings: [[0.1, 0.2, 0.3, 0.4]] }),
|
|
162
|
-
});
|
|
163
|
-
const provider = new OllamaProvider({
|
|
164
|
-
host: 'http://test:11434',
|
|
165
|
-
embedModel: 'qwen3-embedding:8b',
|
|
166
|
-
});
|
|
167
|
-
const result = await provider.embed('test text');
|
|
168
|
-
expect(result).toEqual([0.1, 0.2, 0.3, 0.4]);
|
|
169
|
-
expect(fetchMock.mock.calls[0][0]).toBe('http://test:11434/api/embed');
|
|
170
|
-
const body = JSON.parse(fetchMock.mock.calls[0][1].body);
|
|
171
|
-
expect(body.model).toBe('qwen3-embedding:8b');
|
|
172
|
-
expect(body.input).toBe('test text');
|
|
173
|
-
});
|
|
174
|
-
it('returns empty array on missing embeddings', async () => {
|
|
175
|
-
fetchMock.mockResolvedValueOnce({
|
|
176
|
-
ok: true,
|
|
177
|
-
json: async () => ({}),
|
|
178
|
-
});
|
|
179
|
-
const provider = new OllamaProvider();
|
|
180
|
-
const result = await provider.embed('test');
|
|
181
|
-
expect(result).toEqual([]);
|
|
182
|
-
});
|
|
183
|
-
});
|
|
184
|
-
describe('getStatus', () => {
|
|
185
|
-
it('returns status with running models', async () => {
|
|
186
|
-
fetchMock
|
|
187
|
-
.mockResolvedValueOnce({
|
|
188
|
-
ok: true,
|
|
189
|
-
json: async () => ({
|
|
190
|
-
models: [{ name: 'qwen3:14b', size: 10_000_000, digest: 'abc', modified_at: '2024-01-01' }],
|
|
191
|
-
}),
|
|
192
|
-
})
|
|
193
|
-
.mockResolvedValueOnce({
|
|
194
|
-
ok: true,
|
|
195
|
-
json: async () => ({
|
|
196
|
-
models: [{ name: 'qwen3:14b', size: 10_000_000, size_vram: 9_000_000, expires_at: '2024-01-01' }],
|
|
197
|
-
}),
|
|
198
|
-
});
|
|
199
|
-
const provider = new OllamaProvider({ host: 'http://test:11434' });
|
|
200
|
-
const status = await provider.getStatus();
|
|
201
|
-
expect(status.available).toBe(true);
|
|
202
|
-
expect(status.installedModels).toHaveLength(1);
|
|
203
|
-
expect(status.runningModels).toHaveLength(1);
|
|
204
|
-
});
|
|
205
|
-
it('returns unavailable when server is down', async () => {
|
|
206
|
-
fetchMock.mockRejectedValue(new Error('ECONNREFUSED'));
|
|
207
|
-
const provider = new OllamaProvider();
|
|
208
|
-
const status = await provider.getStatus();
|
|
209
|
-
expect(status.available).toBe(false);
|
|
210
|
-
});
|
|
211
|
-
});
|
|
212
|
-
});
|
|
213
|
-
//# sourceMappingURL=ollama-provider.test.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"ollama-provider.test.js","sourceRoot":"","sources":["../../../src/llm/__tests__/ollama-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,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAEvD,EAAE,CAAC,IAAI,CAAC,uBAAuB,EAAE,GAAG,EAAE,CAAC,CAAC;IACtC,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC;QAChB,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;QACb,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;QACb,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;QACd,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;KACf,CAAC;CACH,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,IAAI,SAAmC,CAAC;IAExC,UAAU,CAAC,GAAG,EAAE;QACd,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACpB,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,gBAAgB,EAAE,CAAC;IACxB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,QAAQ,GAAG,IAAI,cAAc,EAAE,CAAC;YACtC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACrC,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0BAA0B,EAAE,GAAG,EAAE;YAClC,MAAM,QAAQ,GAAG,IAAI,cAAc,EAAE,CAAC;YACtC,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC9C,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClD,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC/C,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;YACrC,MAAM,QAAQ,GAAG,IAAI,cAAc,EAAE,CAAC;YACtC,4CAA4C;YAC5C,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uBAAuB,EAAE,GAAG,EAAE;YAC/B,MAAM,QAAQ,GAAG,IAAI,cAAc,CAAC;gBAClC,IAAI,EAAE,oBAAoB;gBAC1B,KAAK,EAAE,eAAe;gBACtB,UAAU,EAAE,mBAAmB;aAChC,CAAC,CAAC;YACH,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;YACjD,SAAS,CAAC,qBAAqB,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;YAC9C,MAAM,QAAQ,GAAG,IAAI,cAAc,EAAE,CAAC;YACtC,MAAM,CAAC,MAAM,QAAQ,CAAC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;YACjD,SAAS,CAAC,qBAAqB,CAAC,IAAI,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC;YAC3D,MAAM,QAAQ,GAAG,IAAI,cAAc,EAAE,CAAC;YACtC,MAAM,CAAC,MAAM,QAAQ,CAAC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;YAC3C,SAAS,CAAC,iBAAiB,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;YAC1C,MAAM,QAAQ,GAAG,IAAI,cAAc,EAAE,CAAC;YACtC,MAAM,QAAQ,CAAC,WAAW,EAAE,CAAC;YAC7B,MAAM,QAAQ,CAAC,WAAW,EAAE,CAAC;YAC7B,+BAA+B;YAC/B,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;YACxC,SAAS,CAAC,qBAAqB,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;YAC9C,MAAM,QAAQ,GAAG,IAAI,cAAc,CAAC,EAAE,IAAI,EAAE,mBAAmB,EAAE,CAAC,CAAC;YACnE,MAAM,QAAQ,CAAC,WAAW,EAAE,CAAC;YAC7B,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;QACxE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,MAAM,EAAE,GAAG,EAAE;QACpB,EAAE,CAAC,uBAAuB,EAAE,KAAK,IAAI,EAAE;YACrC,SAAS,CAAC,qBAAqB,CAAC;gBAC9B,EAAE,EAAE,IAAI;gBACR,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;oBACjB,OAAO,EAAE,EAAE,OAAO,EAAE,iBAAiB,EAAE;oBACvC,iBAAiB,EAAE,EAAE;oBACrB,UAAU,EAAE,EAAE;iBACf,CAAC;aACH,CAAC,CAAC;YAEH,MAAM,QAAQ,GAAG,IAAI,cAAc,CAAC,EAAE,IAAI,EAAE,mBAAmB,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC;YACvF,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC;gBACjC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,YAAY,EAAE;gBACzC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE;aACnC,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;YAC5C,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACpC,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YAEvC,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YACzD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YACrC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAChC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,SAAS,CAAC,qBAAqB,CAAC;gBAC9B,EAAE,EAAE,IAAI;gBACR,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;oBACjB,OAAO,EAAE,EAAE,OAAO,EAAE,gEAAgE,EAAE;oBACtF,iBAAiB,EAAE,EAAE;oBACrB,UAAU,EAAE,EAAE;iBACf,CAAC;aACH,CAAC,CAAC;YAEH,MAAM,QAAQ,GAAG,IAAI,cAAc,EAAE,CAAC;YACtC,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;YACxE,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;YAChD,SAAS,CAAC,qBAAqB,CAAC;gBAC9B,EAAE,EAAE,IAAI;gBACR,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;oBACjB,OAAO,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE;oBAC1B,iBAAiB,EAAE,CAAC;oBACpB,UAAU,EAAE,CAAC;iBACd,CAAC;aACH,CAAC,CAAC;YAEH,MAAM,QAAQ,GAAG,IAAI,cAAc,EAAE,CAAC;YACtC,MAAM,QAAQ,CAAC,IAAI,CACjB,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,EACnC,EAAE,WAAW,EAAE,GAAG,EAAE,SAAS,EAAE,GAAG,EAAE,CACrC,CAAC;YAEF,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YACzD,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC3C,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;YACxC,SAAS,CAAC,qBAAqB,CAAC;gBAC9B,EAAE,EAAE,KAAK;gBACT,MAAM,EAAE,GAAG;gBACX,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,gBAAgB;aACnC,CAAC,CAAC;YAEH,MAAM,QAAQ,GAAG,IAAI,cAAc,EAAE,CAAC;YACtC,MAAM,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;iBAC7D,OAAO,CAAC,OAAO,CAAC,yBAAyB,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;QACxB,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAC9C,SAAS,CAAC,qBAAqB,CAAC;gBAC9B,EAAE,EAAE,IAAI;gBACR,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,gBAAgB,EAAE,CAAC;aACnD,CAAC,CAAC;YAEH,MAAM,QAAQ,GAAG,IAAI,cAAc,CAAC,EAAE,IAAI,EAAE,mBAAmB,EAAE,CAAC,CAAC;YACnE,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC;YAEvD,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YACtC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,gCAAgC,CAAC,CAAC;QAC5E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE;YACpC,SAAS,CAAC,qBAAqB,CAAC;gBAC9B,EAAE,EAAE,IAAI;gBACR,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,kCAAkC,EAAE,CAAC;aACrE,CAAC,CAAC;YAEH,MAAM,QAAQ,GAAG,IAAI,cAAc,EAAE,CAAC;YACtC,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;YAC/C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,OAAO,EAAE,GAAG,EAAE;QACrB,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;YAC3C,SAAS,CAAC,qBAAqB,CAAC;gBAC9B,EAAE,EAAE,IAAI;gBACR,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,EAAE,CAAC;aAC3D,CAAC,CAAC;YAEH,MAAM,QAAQ,GAAG,IAAI,cAAc,CAAC;gBAClC,IAAI,EAAE,mBAAmB;gBACzB,UAAU,EAAE,oBAAoB;aACjC,CAAC,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAEjD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;YAC7C,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;YAEvE,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YACzD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;YAC9C,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,SAAS,CAAC,qBAAqB,CAAC;gBAC9B,EAAE,EAAE,IAAI;gBACR,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;aACvB,CAAC,CAAC;YAEH,MAAM,QAAQ,GAAG,IAAI,cAAc,EAAE,CAAC;YACtC,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAC5C,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;QACzB,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,SAAS;iBACN,qBAAqB,CAAC;gBACrB,EAAE,EAAE,IAAI;gBACR,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;oBACjB,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,CAAC;iBAC5F,CAAC;aACH,CAAC;iBACD,qBAAqB,CAAC;gBACrB,EAAE,EAAE,IAAI;gBACR,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;oBACjB,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,UAAU,EAAE,SAAS,EAAE,SAAS,EAAE,UAAU,EAAE,YAAY,EAAE,CAAC;iBAClG,CAAC;aACH,CAAC,CAAC;YAEL,MAAM,QAAQ,GAAG,IAAI,cAAc,CAAC,EAAE,IAAI,EAAE,mBAAmB,EAAE,CAAC,CAAC;YACnE,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,SAAS,EAAE,CAAC;YAE1C,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACpC,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC/C,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,SAAS,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC;YACvD,MAAM,QAAQ,GAAG,IAAI,cAAc,EAAE,CAAC;YACtC,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,SAAS,EAAE,CAAC;YAC1C,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
export {};
|
|
@@ -1,126 +0,0 @@
|
|
|
1
|
-
import { describe, it, expect } from 'vitest';
|
|
2
|
-
import { TaskRouter } from '../provider.js';
|
|
3
|
-
// ── Mock Provider Factory ────────────────────────────────
|
|
4
|
-
function createMockProvider(overrides = {}) {
|
|
5
|
-
return {
|
|
6
|
-
name: overrides.name ?? 'mock',
|
|
7
|
-
costTier: overrides.costTier ?? 'free',
|
|
8
|
-
capabilities: overrides.capabilities ?? { chat: true, generate: true, embed: true, reasoning: false },
|
|
9
|
-
isAvailable: overrides.isAvailable ?? (async () => true),
|
|
10
|
-
chat: overrides.chat ?? (async () => ({
|
|
11
|
-
text: 'mock response',
|
|
12
|
-
inputTokens: 10,
|
|
13
|
-
outputTokens: 10,
|
|
14
|
-
model: 'mock-model',
|
|
15
|
-
durationMs: 50,
|
|
16
|
-
})),
|
|
17
|
-
generate: overrides.generate ?? (async () => 'mock generated'),
|
|
18
|
-
embed: overrides.embed ?? (async () => [0.1, 0.2, 0.3]),
|
|
19
|
-
};
|
|
20
|
-
}
|
|
21
|
-
// ── TaskRouter Tests ─────────────────────────────────────
|
|
22
|
-
describe('TaskRouter', () => {
|
|
23
|
-
const freeProvider = createMockProvider({ name: 'local', costTier: 'free' });
|
|
24
|
-
const cheapProvider = createMockProvider({ name: 'cheap', costTier: 'cheap' });
|
|
25
|
-
const expensiveProvider = createMockProvider({ name: 'cloud', costTier: 'expensive' });
|
|
26
|
-
const allProviders = [freeProvider, cheapProvider, expensiveProvider];
|
|
27
|
-
describe('route', () => {
|
|
28
|
-
it('returns empty array for no providers', () => {
|
|
29
|
-
const router = new TaskRouter(true);
|
|
30
|
-
expect(router.route('explain', [])).toEqual([]);
|
|
31
|
-
});
|
|
32
|
-
it('routes simple tasks to free providers first when preferLocal', () => {
|
|
33
|
-
const router = new TaskRouter(true);
|
|
34
|
-
const result = router.route('summarize', allProviders);
|
|
35
|
-
expect(result[0].name).toBe('local');
|
|
36
|
-
expect(result[1].name).toBe('cheap');
|
|
37
|
-
expect(result[2].name).toBe('cloud');
|
|
38
|
-
});
|
|
39
|
-
it('routes complex tasks to expensive providers first', () => {
|
|
40
|
-
const router = new TaskRouter(true);
|
|
41
|
-
const result = router.route('synthesize_debate', allProviders);
|
|
42
|
-
expect(result[0].name).toBe('cloud');
|
|
43
|
-
expect(result[1].name).toBe('cheap');
|
|
44
|
-
expect(result[2].name).toBe('local');
|
|
45
|
-
});
|
|
46
|
-
it('routes "any" tasks to free providers when preferLocal', () => {
|
|
47
|
-
const router = new TaskRouter(true);
|
|
48
|
-
const result = router.route('research_question', allProviders);
|
|
49
|
-
expect(result[0].name).toBe('local');
|
|
50
|
-
});
|
|
51
|
-
it('routes "any" tasks to expensive providers when not preferLocal', () => {
|
|
52
|
-
const router = new TaskRouter(false);
|
|
53
|
-
const result = router.route('research_question', allProviders);
|
|
54
|
-
expect(result[0].name).toBe('cloud');
|
|
55
|
-
});
|
|
56
|
-
it('skips providers without chat capability', () => {
|
|
57
|
-
const embedOnly = createMockProvider({
|
|
58
|
-
name: 'embed-only',
|
|
59
|
-
capabilities: { chat: false, generate: false, embed: true, reasoning: false },
|
|
60
|
-
});
|
|
61
|
-
const router = new TaskRouter(true);
|
|
62
|
-
const result = router.route('summarize', [embedOnly, freeProvider]);
|
|
63
|
-
expect(result).toHaveLength(1);
|
|
64
|
-
expect(result[0].name).toBe('local');
|
|
65
|
-
});
|
|
66
|
-
});
|
|
67
|
-
describe('setPreferLocal', () => {
|
|
68
|
-
it('changes routing preference dynamically', () => {
|
|
69
|
-
const router = new TaskRouter(true);
|
|
70
|
-
expect(router.route('research_question', allProviders)[0].name).toBe('local');
|
|
71
|
-
router.setPreferLocal(false);
|
|
72
|
-
expect(router.route('research_question', allProviders)[0].name).toBe('cloud');
|
|
73
|
-
});
|
|
74
|
-
});
|
|
75
|
-
describe('getTier', () => {
|
|
76
|
-
it('returns correct tier for each template', () => {
|
|
77
|
-
const router = new TaskRouter();
|
|
78
|
-
expect(router.getTier('summarize')).toBe('local');
|
|
79
|
-
expect(router.getTier('synthesize_debate')).toBe('cloud');
|
|
80
|
-
expect(router.getTier('research_question')).toBe('any');
|
|
81
|
-
expect(router.getTier('explain')).toBe('cloud');
|
|
82
|
-
});
|
|
83
|
-
});
|
|
84
|
-
describe('getRoutingTable', () => {
|
|
85
|
-
it('returns all routing rules', () => {
|
|
86
|
-
const router = new TaskRouter();
|
|
87
|
-
const table = router.getRoutingTable();
|
|
88
|
-
expect(table.length).toBeGreaterThanOrEqual(8);
|
|
89
|
-
const templates = table.map(r => r.template);
|
|
90
|
-
expect(templates).toContain('explain');
|
|
91
|
-
expect(templates).toContain('summarize');
|
|
92
|
-
expect(templates).toContain('synthesize_debate');
|
|
93
|
-
});
|
|
94
|
-
});
|
|
95
|
-
});
|
|
96
|
-
// ── LLMProvider Interface Tests ──────────────────────────
|
|
97
|
-
describe('LLMProvider interface', () => {
|
|
98
|
-
it('mock provider satisfies interface', async () => {
|
|
99
|
-
const provider = createMockProvider();
|
|
100
|
-
expect(provider.name).toBe('mock');
|
|
101
|
-
expect(provider.costTier).toBe('free');
|
|
102
|
-
expect(provider.capabilities.chat).toBe(true);
|
|
103
|
-
expect(await provider.isAvailable()).toBe(true);
|
|
104
|
-
});
|
|
105
|
-
it('chat returns proper response', async () => {
|
|
106
|
-
const provider = createMockProvider();
|
|
107
|
-
const response = await provider.chat([{ role: 'user', content: 'hello' }]);
|
|
108
|
-
expect(response.text).toBe('mock response');
|
|
109
|
-
expect(response.inputTokens).toBe(10);
|
|
110
|
-
expect(response.outputTokens).toBe(10);
|
|
111
|
-
expect(response.model).toBe('mock-model');
|
|
112
|
-
expect(response.durationMs).toBeGreaterThanOrEqual(0);
|
|
113
|
-
});
|
|
114
|
-
it('generate returns string', async () => {
|
|
115
|
-
const provider = createMockProvider();
|
|
116
|
-
const result = await provider.generate('hello');
|
|
117
|
-
expect(typeof result).toBe('string');
|
|
118
|
-
});
|
|
119
|
-
it('embed returns number array', async () => {
|
|
120
|
-
const provider = createMockProvider();
|
|
121
|
-
const result = await provider.embed('hello');
|
|
122
|
-
expect(Array.isArray(result)).toBe(true);
|
|
123
|
-
expect(result.length).toBeGreaterThan(0);
|
|
124
|
-
});
|
|
125
|
-
});
|
|
126
|
-
//# sourceMappingURL=provider.test.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"provider.test.js","sourceRoot":"","sources":["../../../src/llm/__tests__/provider.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAI5C,4DAA4D;AAE5D,SAAS,kBAAkB,CAAC,YAAkC,EAAE;IAC9D,OAAO;QACL,IAAI,EAAE,SAAS,CAAC,IAAI,IAAI,MAAM;QAC9B,QAAQ,EAAE,SAAS,CAAC,QAAQ,IAAI,MAAM;QACtC,YAAY,EAAE,SAAS,CAAC,YAAY,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE;QACrG,WAAW,EAAE,SAAS,CAAC,WAAW,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC,IAAI,CAAC;QACxD,IAAI,EAAE,SAAS,CAAC,IAAI,IAAI,CAAC,KAAK,IAAkC,EAAE,CAAC,CAAC;YAClE,IAAI,EAAE,eAAe;YACrB,WAAW,EAAE,EAAE;YACf,YAAY,EAAE,EAAE;YAChB,KAAK,EAAE,YAAY;YACnB,UAAU,EAAE,EAAE;SACf,CAAC,CAAC;QACH,QAAQ,EAAE,SAAS,CAAC,QAAQ,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC,gBAAgB,CAAC;QAC9D,KAAK,EAAE,SAAS,CAAC,KAAK,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;KACxD,CAAC;AACJ,CAAC;AAED,4DAA4D;AAE5D,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IAC1B,MAAM,YAAY,GAAG,kBAAkB,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;IAC7E,MAAM,aAAa,GAAG,kBAAkB,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC;IAC/E,MAAM,iBAAiB,GAAG,kBAAkB,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,EAAE,CAAC,CAAC;IACvF,MAAM,YAAY,GAAG,CAAC,YAAY,EAAE,aAAa,EAAE,iBAAiB,CAAC,CAAC;IAEtE,QAAQ,CAAC,OAAO,EAAE,GAAG,EAAE;QACrB,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;YAC9C,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,IAAI,CAAC,CAAC;YACpC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8DAA8D,EAAE,GAAG,EAAE;YACtE,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,IAAI,CAAC,CAAC;YACpC,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,WAAW,EAAE,YAAY,CAAC,CAAC;YACvD,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;YAC3D,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,IAAI,CAAC,CAAC;YACpC,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,mBAAmB,EAAE,YAAY,CAAC,CAAC;YAC/D,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;YAC/D,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,IAAI,CAAC,CAAC;YACpC,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,mBAAmB,EAAE,YAAY,CAAC,CAAC;YAC/D,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gEAAgE,EAAE,GAAG,EAAE;YACxE,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,KAAK,CAAC,CAAC;YACrC,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,mBAAmB,EAAE,YAAY,CAAC,CAAC;YAC/D,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,MAAM,SAAS,GAAG,kBAAkB,CAAC;gBACnC,IAAI,EAAE,YAAY;gBAClB,YAAY,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE;aAC9E,CAAC,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,IAAI,CAAC,CAAC;YACpC,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC,CAAC;YACpE,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;YAChD,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,IAAI,CAAC,CAAC;YACpC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,mBAAmB,EAAE,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAE9E,MAAM,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;YAC7B,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,mBAAmB,EAAE,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAChF,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;QACvB,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;YAChD,MAAM,MAAM,GAAG,IAAI,UAAU,EAAE,CAAC;YAChC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAClD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC1D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACxD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,2BAA2B,EAAE,GAAG,EAAE;YACnC,MAAM,MAAM,GAAG,IAAI,UAAU,EAAE,CAAC;YAChC,MAAM,KAAK,GAAG,MAAM,CAAC,eAAe,EAAE,CAAC;YACvC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC;YAE/C,MAAM,SAAS,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;YAC7C,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;YACvC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;YACzC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,4DAA4D;AAE5D,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;IACrC,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;QACjD,MAAM,QAAQ,GAAG,kBAAkB,EAAE,CAAC;QACtC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACnC,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACvC,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC9C,MAAM,CAAC,MAAM,QAAQ,CAAC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;QAC5C,MAAM,QAAQ,GAAG,kBAAkB,EAAE,CAAC;QACtC,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC;QAC3E,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAC5C,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACtC,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACvC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAC1C,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;QACvC,MAAM,QAAQ,GAAG,kBAAkB,EAAE,CAAC;QACtC,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;QAChD,MAAM,CAAC,OAAO,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;QAC1C,MAAM,QAAQ,GAAG,kBAAkB,EAAE,CAAC;QACtC,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAC7C,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
export {};
|
|
@@ -1,37 +0,0 @@
|
|
|
1
|
-
import { describe, it, expect, vi, afterEach } from 'vitest';
|
|
2
|
-
import { timeDecayFactor } from '../time-decay.js';
|
|
3
|
-
describe('timeDecayFactor', () => {
|
|
4
|
-
afterEach(() => {
|
|
5
|
-
vi.restoreAllMocks();
|
|
6
|
-
});
|
|
7
|
-
it('returns ~1.0 for just-activated item', () => {
|
|
8
|
-
const now = new Date().toISOString();
|
|
9
|
-
const factor = timeDecayFactor(now, 30);
|
|
10
|
-
expect(factor).toBeGreaterThan(0.99);
|
|
11
|
-
expect(factor).toBeLessThanOrEqual(1.0);
|
|
12
|
-
});
|
|
13
|
-
it('returns ~0.5 at exactly one half-life', () => {
|
|
14
|
-
const halfLifeDays = 30;
|
|
15
|
-
const activatedAt = new Date(Date.now() - halfLifeDays * 24 * 60 * 60 * 1000).toISOString();
|
|
16
|
-
const factor = timeDecayFactor(activatedAt, halfLifeDays);
|
|
17
|
-
expect(factor).toBeCloseTo(0.5, 1);
|
|
18
|
-
});
|
|
19
|
-
it('returns ~0.25 at two half-lives', () => {
|
|
20
|
-
const halfLifeDays = 14;
|
|
21
|
-
const activatedAt = new Date(Date.now() - 28 * 24 * 60 * 60 * 1000).toISOString();
|
|
22
|
-
const factor = timeDecayFactor(activatedAt, halfLifeDays);
|
|
23
|
-
expect(factor).toBeCloseTo(0.25, 1);
|
|
24
|
-
});
|
|
25
|
-
it('returns very small value for very old items', () => {
|
|
26
|
-
const activatedAt = new Date(Date.now() - 365 * 24 * 60 * 60 * 1000).toISOString();
|
|
27
|
-
const factor = timeDecayFactor(activatedAt, 30);
|
|
28
|
-
expect(factor).toBeLessThan(0.01);
|
|
29
|
-
});
|
|
30
|
-
it('works with different half-life values', () => {
|
|
31
|
-
const activatedAt = new Date(Date.now() - 7 * 24 * 60 * 60 * 1000).toISOString();
|
|
32
|
-
const shortHalfLife = timeDecayFactor(activatedAt, 7);
|
|
33
|
-
const longHalfLife = timeDecayFactor(activatedAt, 45);
|
|
34
|
-
expect(shortHalfLife).toBeLessThan(longHalfLife);
|
|
35
|
-
});
|
|
36
|
-
});
|
|
37
|
-
//# sourceMappingURL=time-decay.test.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"time-decay.test.js","sourceRoot":"","sources":["../../../src/math/__tests__/time-decay.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AAC7D,OAAO,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAEnD,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,eAAe,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC9C,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QACrC,MAAM,MAAM,GAAG,eAAe,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QACxC,MAAM,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,CAAC,mBAAmB,CAAC,GAAG,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;QAC/C,MAAM,YAAY,GAAG,EAAE,CAAC;QACxB,MAAM,WAAW,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,YAAY,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,WAAW,EAAE,CAAC;QAC5F,MAAM,MAAM,GAAG,eAAe,CAAC,WAAW,EAAE,YAAY,CAAC,CAAC;QAC1D,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACzC,MAAM,YAAY,GAAG,EAAE,CAAC;QACxB,MAAM,WAAW,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,WAAW,EAAE,CAAC;QAClF,MAAM,MAAM,GAAG,eAAe,CAAC,WAAW,EAAE,YAAY,CAAC,CAAC;QAC1D,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACrD,MAAM,WAAW,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,GAAG,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,WAAW,EAAE,CAAC;QACnF,MAAM,MAAM,GAAG,eAAe,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;QAChD,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;QAC/C,MAAM,WAAW,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,WAAW,EAAE,CAAC;QACjF,MAAM,aAAa,GAAG,eAAe,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC;QACtD,MAAM,YAAY,GAAG,eAAe,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;QACtD,MAAM,CAAC,aAAa,CAAC,CAAC,YAAY,CAAC,YAAY,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
export {};
|
|
@@ -1,43 +0,0 @@
|
|
|
1
|
-
import { describe, it, expect } from 'vitest';
|
|
2
|
-
import { wilsonScore } from '../wilson-score.js';
|
|
3
|
-
describe('wilsonScore', () => {
|
|
4
|
-
it('returns 0 for zero trials', () => {
|
|
5
|
-
expect(wilsonScore(0, 0)).toBe(0);
|
|
6
|
-
});
|
|
7
|
-
it('returns low confidence for single success', () => {
|
|
8
|
-
const score = wilsonScore(1, 1);
|
|
9
|
-
expect(score).toBeGreaterThan(0);
|
|
10
|
-
expect(score).toBeLessThan(0.8);
|
|
11
|
-
});
|
|
12
|
-
it('converges toward actual rate with large samples', () => {
|
|
13
|
-
const score = wilsonScore(950, 1000);
|
|
14
|
-
expect(score).toBeGreaterThan(0.93);
|
|
15
|
-
expect(score).toBeLessThan(0.96);
|
|
16
|
-
});
|
|
17
|
-
it('penalizes small sample sizes', () => {
|
|
18
|
-
const small = wilsonScore(5, 5);
|
|
19
|
-
const large = wilsonScore(500, 500);
|
|
20
|
-
expect(large).toBeGreaterThan(small);
|
|
21
|
-
});
|
|
22
|
-
it('returns higher confidence with lower z-score', () => {
|
|
23
|
-
const z90 = wilsonScore(7, 10, 1.64);
|
|
24
|
-
const z95 = wilsonScore(7, 10, 1.96);
|
|
25
|
-
const z99 = wilsonScore(7, 10, 2.33);
|
|
26
|
-
expect(z90).toBeGreaterThan(z95);
|
|
27
|
-
expect(z95).toBeGreaterThan(z99);
|
|
28
|
-
});
|
|
29
|
-
it('handles 0 successes', () => {
|
|
30
|
-
const score = wilsonScore(0, 10);
|
|
31
|
-
expect(score).toBe(0);
|
|
32
|
-
});
|
|
33
|
-
it('handles 50/50 split', () => {
|
|
34
|
-
const score = wilsonScore(50, 100);
|
|
35
|
-
expect(score).toBeGreaterThan(0.39);
|
|
36
|
-
expect(score).toBeLessThan(0.51);
|
|
37
|
-
});
|
|
38
|
-
it('never returns negative', () => {
|
|
39
|
-
expect(wilsonScore(0, 1)).toBeGreaterThanOrEqual(0);
|
|
40
|
-
expect(wilsonScore(0, 100)).toBeGreaterThanOrEqual(0);
|
|
41
|
-
});
|
|
42
|
-
});
|
|
43
|
-
//# sourceMappingURL=wilson-score.test.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"wilson-score.test.js","sourceRoot":"","sources":["../../../src/math/__tests__/wilson-score.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAEjD,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,EAAE,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACnC,MAAM,CAAC,WAAW,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,MAAM,KAAK,GAAG,WAAW,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QAChC,MAAM,CAAC,KAAK,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QACjC,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;QACzD,MAAM,KAAK,GAAG,WAAW,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;QACrC,MAAM,CAAC,KAAK,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QACpC,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;QACtC,MAAM,KAAK,GAAG,WAAW,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QAChC,MAAM,KAAK,GAAG,WAAW,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QACpC,MAAM,CAAC,KAAK,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,GAAG,GAAG,WAAW,CAAC,CAAC,EAAE,EAAE,EAAE,IAAI,CAAC,CAAC;QACrC,MAAM,GAAG,GAAG,WAAW,CAAC,CAAC,EAAE,EAAE,EAAE,IAAI,CAAC,CAAC;QACrC,MAAM,GAAG,GAAG,WAAW,CAAC,CAAC,EAAE,EAAE,EAAE,IAAI,CAAC,CAAC;QACrC,MAAM,CAAC,GAAG,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;QACjC,MAAM,CAAC,GAAG,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qBAAqB,EAAE,GAAG,EAAE;QAC7B,MAAM,KAAK,GAAG,WAAW,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QACjC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACxB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qBAAqB,EAAE,GAAG,EAAE;QAC7B,MAAM,KAAK,GAAG,WAAW,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;QACnC,MAAM,CAAC,KAAK,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QACpC,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wBAAwB,EAAE,GAAG,EAAE;QAChC,MAAM,CAAC,WAAW,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC;QACpD,MAAM,CAAC,WAAW,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
export {};
|