@sprinterai/runtime 0.2.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/LICENSE +21 -0
- package/dist/adapters/a2a-adapter.d.ts +24 -0
- package/dist/adapters/a2a-adapter.d.ts.map +1 -0
- package/dist/adapters/a2a-adapter.js +13 -0
- package/dist/adapters/a2a-adapter.js.map +1 -0
- package/dist/adapters/http-agent-adapter.d.ts +22 -0
- package/dist/adapters/http-agent-adapter.d.ts.map +1 -0
- package/dist/adapters/http-agent-adapter.js +13 -0
- package/dist/adapters/http-agent-adapter.js.map +1 -0
- package/dist/adapters/index.d.ts +9 -0
- package/dist/adapters/index.d.ts.map +1 -0
- package/dist/adapters/index.js +5 -0
- package/dist/adapters/index.js.map +1 -0
- package/dist/adapters/mcp-adapter.d.ts +19 -0
- package/dist/adapters/mcp-adapter.d.ts.map +1 -0
- package/dist/adapters/mcp-adapter.js +13 -0
- package/dist/adapters/mcp-adapter.js.map +1 -0
- package/dist/adapters/openclaw-adapter.d.ts +20 -0
- package/dist/adapters/openclaw-adapter.d.ts.map +1 -0
- package/dist/adapters/openclaw-adapter.js +13 -0
- package/dist/adapters/openclaw-adapter.js.map +1 -0
- package/dist/agent/agent-registry.d.ts +25 -0
- package/dist/agent/agent-registry.d.ts.map +1 -0
- package/dist/agent/agent-registry.js +42 -0
- package/dist/agent/agent-registry.js.map +1 -0
- package/dist/agent/agent-registry.test.d.ts +2 -0
- package/dist/agent/agent-registry.test.d.ts.map +1 -0
- package/dist/agent/agent-registry.test.js +55 -0
- package/dist/agent/agent-registry.test.js.map +1 -0
- package/dist/agent/agent-resolver.d.ts +41 -0
- package/dist/agent/agent-resolver.d.ts.map +1 -0
- package/dist/agent/agent-resolver.js +76 -0
- package/dist/agent/agent-resolver.js.map +1 -0
- package/dist/agent/agent-resolver.test.d.ts +2 -0
- package/dist/agent/agent-resolver.test.d.ts.map +1 -0
- package/dist/agent/agent-resolver.test.js +116 -0
- package/dist/agent/agent-resolver.test.js.map +1 -0
- package/dist/agent/delegate.d.ts +14 -0
- package/dist/agent/delegate.d.ts.map +1 -0
- package/dist/agent/delegate.js +35 -0
- package/dist/agent/delegate.js.map +1 -0
- package/dist/agent/delegate.test.d.ts +2 -0
- package/dist/agent/delegate.test.d.ts.map +1 -0
- package/dist/agent/delegate.test.js +64 -0
- package/dist/agent/delegate.test.js.map +1 -0
- package/dist/agent/execute-agent.d.ts +43 -0
- package/dist/agent/execute-agent.d.ts.map +1 -0
- package/dist/agent/execute-agent.js +43 -0
- package/dist/agent/execute-agent.js.map +1 -0
- package/dist/agent/index.d.ts +10 -0
- package/dist/agent/index.d.ts.map +1 -0
- package/dist/agent/index.js +6 -0
- package/dist/agent/index.js.map +1 -0
- package/dist/agent/prompt-builder.d.ts +16 -0
- package/dist/agent/prompt-builder.d.ts.map +1 -0
- package/dist/agent/prompt-builder.js +17 -0
- package/dist/agent/prompt-builder.js.map +1 -0
- package/dist/agent/prompt-builder.test.d.ts +2 -0
- package/dist/agent/prompt-builder.test.d.ts.map +1 -0
- package/dist/agent/prompt-builder.test.js +60 -0
- package/dist/agent/prompt-builder.test.js.map +1 -0
- package/dist/chat/chat-handler.d.ts +63 -0
- package/dist/chat/chat-handler.d.ts.map +1 -0
- package/dist/chat/chat-handler.js +74 -0
- package/dist/chat/chat-handler.js.map +1 -0
- package/dist/chat/chat-handler.test.d.ts +2 -0
- package/dist/chat/chat-handler.test.d.ts.map +1 -0
- package/dist/chat/chat-handler.test.js +112 -0
- package/dist/chat/chat-handler.test.js.map +1 -0
- package/dist/chat/index.d.ts +4 -0
- package/dist/chat/index.d.ts.map +1 -0
- package/dist/chat/index.js +3 -0
- package/dist/chat/index.js.map +1 -0
- package/dist/chat/message-utils.d.ts +26 -0
- package/dist/chat/message-utils.d.ts.map +1 -0
- package/dist/chat/message-utils.js +65 -0
- package/dist/chat/message-utils.js.map +1 -0
- package/dist/chat/message-utils.test.d.ts +2 -0
- package/dist/chat/message-utils.test.d.ts.map +1 -0
- package/dist/chat/message-utils.test.js +85 -0
- package/dist/chat/message-utils.test.js.map +1 -0
- package/dist/eval/eval-runner.d.ts +22 -0
- package/dist/eval/eval-runner.d.ts.map +1 -0
- package/dist/eval/eval-runner.js +56 -0
- package/dist/eval/eval-runner.js.map +1 -0
- package/dist/eval/eval-runner.test.d.ts +2 -0
- package/dist/eval/eval-runner.test.d.ts.map +1 -0
- package/dist/eval/eval-runner.test.js +89 -0
- package/dist/eval/eval-runner.test.js.map +1 -0
- package/dist/eval/index.d.ts +4 -0
- package/dist/eval/index.d.ts.map +1 -0
- package/dist/eval/index.js +3 -0
- package/dist/eval/index.js.map +1 -0
- package/dist/eval/scorers.d.ts +26 -0
- package/dist/eval/scorers.d.ts.map +1 -0
- package/dist/eval/scorers.js +76 -0
- package/dist/eval/scorers.js.map +1 -0
- package/dist/eval/scorers.test.d.ts +2 -0
- package/dist/eval/scorers.test.d.ts.map +1 -0
- package/dist/eval/scorers.test.js +91 -0
- package/dist/eval/scorers.test.js.map +1 -0
- package/dist/guardrail/guardrail-pipeline.d.ts +37 -0
- package/dist/guardrail/guardrail-pipeline.d.ts.map +1 -0
- package/dist/guardrail/guardrail-pipeline.js +63 -0
- package/dist/guardrail/guardrail-pipeline.js.map +1 -0
- package/dist/guardrail/guardrail-pipeline.test.d.ts +2 -0
- package/dist/guardrail/guardrail-pipeline.test.d.ts.map +1 -0
- package/dist/guardrail/guardrail-pipeline.test.js +74 -0
- package/dist/guardrail/guardrail-pipeline.test.js.map +1 -0
- package/dist/guardrail/index.d.ts +3 -0
- package/dist/guardrail/index.d.ts.map +1 -0
- package/dist/guardrail/index.js +2 -0
- package/dist/guardrail/index.js.map +1 -0
- package/dist/index.d.ts +46 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +39 -0
- package/dist/index.js.map +1 -0
- package/dist/memory/index.d.ts +4 -0
- package/dist/memory/index.d.ts.map +1 -0
- package/dist/memory/index.js +3 -0
- package/dist/memory/index.js.map +1 -0
- package/dist/memory/memory-manager.d.ts +28 -0
- package/dist/memory/memory-manager.d.ts.map +1 -0
- package/dist/memory/memory-manager.js +34 -0
- package/dist/memory/memory-manager.js.map +1 -0
- package/dist/memory/memory-prompt.d.ts +10 -0
- package/dist/memory/memory-prompt.d.ts.map +1 -0
- package/dist/memory/memory-prompt.js +29 -0
- package/dist/memory/memory-prompt.js.map +1 -0
- package/dist/module/create-runtime.d.ts +65 -0
- package/dist/module/create-runtime.d.ts.map +1 -0
- package/dist/module/create-runtime.js +75 -0
- package/dist/module/create-runtime.js.map +1 -0
- package/dist/module/create-runtime.test.d.ts +2 -0
- package/dist/module/create-runtime.test.d.ts.map +1 -0
- package/dist/module/create-runtime.test.js +144 -0
- package/dist/module/create-runtime.test.js.map +1 -0
- package/dist/module/index.d.ts +5 -0
- package/dist/module/index.d.ts.map +1 -0
- package/dist/module/index.js +3 -0
- package/dist/module/index.js.map +1 -0
- package/dist/module/module-loader.d.ts +32 -0
- package/dist/module/module-loader.d.ts.map +1 -0
- package/dist/module/module-loader.js +123 -0
- package/dist/module/module-loader.js.map +1 -0
- package/dist/module/module-loader.test.d.ts +2 -0
- package/dist/module/module-loader.test.d.ts.map +1 -0
- package/dist/module/module-loader.test.js +96 -0
- package/dist/module/module-loader.test.js.map +1 -0
- package/dist/scoring/compute-score.d.ts +32 -0
- package/dist/scoring/compute-score.d.ts.map +1 -0
- package/dist/scoring/compute-score.js +54 -0
- package/dist/scoring/compute-score.js.map +1 -0
- package/dist/scoring/compute-score.test.d.ts +2 -0
- package/dist/scoring/compute-score.test.d.ts.map +1 -0
- package/dist/scoring/compute-score.test.js +99 -0
- package/dist/scoring/compute-score.test.js.map +1 -0
- package/dist/scoring/index.d.ts +3 -0
- package/dist/scoring/index.d.ts.map +1 -0
- package/dist/scoring/index.js +2 -0
- package/dist/scoring/index.js.map +1 -0
- package/dist/testing/in-memory-agent-store.d.ts +16 -0
- package/dist/testing/in-memory-agent-store.d.ts.map +1 -0
- package/dist/testing/in-memory-agent-store.js +37 -0
- package/dist/testing/in-memory-agent-store.js.map +1 -0
- package/dist/testing/in-memory-agent-store.test.d.ts +2 -0
- package/dist/testing/in-memory-agent-store.test.d.ts.map +1 -0
- package/dist/testing/in-memory-agent-store.test.js +111 -0
- package/dist/testing/in-memory-agent-store.test.js.map +1 -0
- package/dist/testing/in-memory-chat-store.d.ts +28 -0
- package/dist/testing/in-memory-chat-store.d.ts.map +1 -0
- package/dist/testing/in-memory-chat-store.js +87 -0
- package/dist/testing/in-memory-chat-store.js.map +1 -0
- package/dist/testing/in-memory-entity-store.d.ts +34 -0
- package/dist/testing/in-memory-entity-store.d.ts.map +1 -0
- package/dist/testing/in-memory-entity-store.js +134 -0
- package/dist/testing/in-memory-entity-store.js.map +1 -0
- package/dist/testing/in-memory-entity-store.test.d.ts +2 -0
- package/dist/testing/in-memory-entity-store.test.d.ts.map +1 -0
- package/dist/testing/in-memory-entity-store.test.js +341 -0
- package/dist/testing/in-memory-entity-store.test.js.map +1 -0
- package/dist/testing/in-memory-memory-store.d.ts +25 -0
- package/dist/testing/in-memory-memory-store.d.ts.map +1 -0
- package/dist/testing/in-memory-memory-store.js +59 -0
- package/dist/testing/in-memory-memory-store.js.map +1 -0
- package/dist/testing/in-memory-task-store.d.ts +20 -0
- package/dist/testing/in-memory-task-store.d.ts.map +1 -0
- package/dist/testing/in-memory-task-store.js +59 -0
- package/dist/testing/in-memory-task-store.js.map +1 -0
- package/dist/testing/in-memory-tenant-store.d.ts +16 -0
- package/dist/testing/in-memory-tenant-store.d.ts.map +1 -0
- package/dist/testing/in-memory-tenant-store.js +46 -0
- package/dist/testing/in-memory-tenant-store.js.map +1 -0
- package/dist/testing/in-memory-tool-store.d.ts +17 -0
- package/dist/testing/in-memory-tool-store.d.ts.map +1 -0
- package/dist/testing/in-memory-tool-store.js +39 -0
- package/dist/testing/in-memory-tool-store.js.map +1 -0
- package/dist/testing/index.d.ts +8 -0
- package/dist/testing/index.d.ts.map +1 -0
- package/dist/testing/index.js +8 -0
- package/dist/testing/index.js.map +1 -0
- package/dist/tool/ai-bridge.d.ts +13 -0
- package/dist/tool/ai-bridge.d.ts.map +1 -0
- package/dist/tool/ai-bridge.js +27 -0
- package/dist/tool/ai-bridge.js.map +1 -0
- package/dist/tool/ai-bridge.test.d.ts +2 -0
- package/dist/tool/ai-bridge.test.d.ts.map +1 -0
- package/dist/tool/ai-bridge.test.js +40 -0
- package/dist/tool/ai-bridge.test.js.map +1 -0
- package/dist/tool/entity-tools-factory.d.ts +7 -0
- package/dist/tool/entity-tools-factory.d.ts.map +1 -0
- package/dist/tool/entity-tools-factory.js +160 -0
- package/dist/tool/entity-tools-factory.js.map +1 -0
- package/dist/tool/entity-tools-factory.test.d.ts +2 -0
- package/dist/tool/entity-tools-factory.test.d.ts.map +1 -0
- package/dist/tool/entity-tools-factory.test.js +160 -0
- package/dist/tool/entity-tools-factory.test.js.map +1 -0
- package/dist/tool/index.d.ts +8 -0
- package/dist/tool/index.d.ts.map +1 -0
- package/dist/tool/index.js +6 -0
- package/dist/tool/index.js.map +1 -0
- package/dist/tool/resolve-tools.d.ts +28 -0
- package/dist/tool/resolve-tools.d.ts.map +1 -0
- package/dist/tool/resolve-tools.js +54 -0
- package/dist/tool/resolve-tools.js.map +1 -0
- package/dist/tool/resolve-tools.test.d.ts +2 -0
- package/dist/tool/resolve-tools.test.d.ts.map +1 -0
- package/dist/tool/resolve-tools.test.js +111 -0
- package/dist/tool/resolve-tools.test.js.map +1 -0
- package/dist/tool/tool-executor.d.ts +30 -0
- package/dist/tool/tool-executor.d.ts.map +1 -0
- package/dist/tool/tool-executor.js +64 -0
- package/dist/tool/tool-executor.js.map +1 -0
- package/dist/tool/tool-executor.test.d.ts +2 -0
- package/dist/tool/tool-executor.test.d.ts.map +1 -0
- package/dist/tool/tool-executor.test.js +105 -0
- package/dist/tool/tool-executor.test.js.map +1 -0
- package/dist/tool/tool-registry.d.ts +29 -0
- package/dist/tool/tool-registry.d.ts.map +1 -0
- package/dist/tool/tool-registry.js +59 -0
- package/dist/tool/tool-registry.js.map +1 -0
- package/dist/tool/tool-registry.test.d.ts +2 -0
- package/dist/tool/tool-registry.test.d.ts.map +1 -0
- package/dist/tool/tool-registry.test.js +76 -0
- package/dist/tool/tool-registry.test.js.map +1 -0
- package/dist/workflow/compile.d.ts +17 -0
- package/dist/workflow/compile.d.ts.map +1 -0
- package/dist/workflow/compile.js +83 -0
- package/dist/workflow/compile.js.map +1 -0
- package/dist/workflow/compile.test.d.ts +2 -0
- package/dist/workflow/compile.test.d.ts.map +1 -0
- package/dist/workflow/compile.test.js +154 -0
- package/dist/workflow/compile.test.js.map +1 -0
- package/dist/workflow/index.d.ts +4 -0
- package/dist/workflow/index.d.ts.map +1 -0
- package/dist/workflow/index.js +3 -0
- package/dist/workflow/index.js.map +1 -0
- package/dist/workflow/status.d.ts +21 -0
- package/dist/workflow/status.d.ts.map +1 -0
- package/dist/workflow/status.js +55 -0
- package/dist/workflow/status.js.map +1 -0
- package/dist/workflow/status.test.d.ts +2 -0
- package/dist/workflow/status.test.d.ts.map +1 -0
- package/dist/workflow/status.test.js +176 -0
- package/dist/workflow/status.test.js.map +1 -0
- package/package.json +100 -0
|
@@ -0,0 +1,112 @@
|
|
|
1
|
+
import { describe, it, expect, vi, beforeEach } from 'vitest';
|
|
2
|
+
import { AgentRegistry } from '../agent/agent-registry';
|
|
3
|
+
import { InMemoryAgentStore } from '../testing/in-memory-agent-store';
|
|
4
|
+
import { InMemoryChatStore } from '../testing/in-memory-chat-store';
|
|
5
|
+
import { InMemoryEntityStore } from '../testing/in-memory-entity-store';
|
|
6
|
+
import { InMemoryMemoryStore } from '../testing/in-memory-memory-store';
|
|
7
|
+
import { InMemoryTenantStore } from '../testing/in-memory-tenant-store';
|
|
8
|
+
import { createChatRouteHandler } from './chat-handler';
|
|
9
|
+
// Mock AI SDK to avoid actual model calls
|
|
10
|
+
vi.mock('ai', () => ({
|
|
11
|
+
streamText: vi.fn(() => ({
|
|
12
|
+
textStream: (async function* () {
|
|
13
|
+
yield 'Hello';
|
|
14
|
+
})(),
|
|
15
|
+
toDataStreamResponse: () => new Response('mocked'),
|
|
16
|
+
})),
|
|
17
|
+
generateText: vi.fn(),
|
|
18
|
+
stepCountIs: vi.fn((n) => ({ type: 'step-count', count: n })),
|
|
19
|
+
tool: vi.fn((config) => ({
|
|
20
|
+
type: 'function',
|
|
21
|
+
...config,
|
|
22
|
+
})),
|
|
23
|
+
}));
|
|
24
|
+
describe('createChatRouteHandler', () => {
|
|
25
|
+
let agentRegistry;
|
|
26
|
+
const mockModel = { modelId: 'test-model' };
|
|
27
|
+
beforeEach(() => {
|
|
28
|
+
agentRegistry = new AgentRegistry();
|
|
29
|
+
agentRegistry.register({
|
|
30
|
+
id: 'analyst',
|
|
31
|
+
name: 'Analyst',
|
|
32
|
+
description: 'Data analyst',
|
|
33
|
+
icon: 'chart',
|
|
34
|
+
systemPrompt: 'You analyze data',
|
|
35
|
+
toolGroups: [],
|
|
36
|
+
customTools: [],
|
|
37
|
+
});
|
|
38
|
+
});
|
|
39
|
+
function makeHandler() {
|
|
40
|
+
return createChatRouteHandler({
|
|
41
|
+
stores: {
|
|
42
|
+
agent: new InMemoryAgentStore(),
|
|
43
|
+
entity: new InMemoryEntityStore(),
|
|
44
|
+
chat: new InMemoryChatStore(),
|
|
45
|
+
tenant: new InMemoryTenantStore(),
|
|
46
|
+
memory: new InMemoryMemoryStore(),
|
|
47
|
+
},
|
|
48
|
+
defaultAgentSlug: 'analyst',
|
|
49
|
+
resolveModel: () => mockModel,
|
|
50
|
+
agentRegistry,
|
|
51
|
+
});
|
|
52
|
+
}
|
|
53
|
+
it('resolves agent and returns stream', async () => {
|
|
54
|
+
const handler = makeHandler();
|
|
55
|
+
const { agent } = await handler.handle({
|
|
56
|
+
messages: [{ role: 'user', content: 'hello' }],
|
|
57
|
+
});
|
|
58
|
+
expect(agent.id).toBe('analyst');
|
|
59
|
+
expect(agent.source).toBe('code');
|
|
60
|
+
});
|
|
61
|
+
it('uses agentSlug from request', async () => {
|
|
62
|
+
agentRegistry.register({
|
|
63
|
+
id: 'researcher',
|
|
64
|
+
name: 'Researcher',
|
|
65
|
+
description: 'Web researcher',
|
|
66
|
+
icon: 'search',
|
|
67
|
+
systemPrompt: 'Research the web',
|
|
68
|
+
});
|
|
69
|
+
const handler = makeHandler();
|
|
70
|
+
const { agent } = await handler.handle({
|
|
71
|
+
messages: [{ role: 'user', content: 'search' }],
|
|
72
|
+
agentSlug: 'researcher',
|
|
73
|
+
});
|
|
74
|
+
expect(agent.id).toBe('researcher');
|
|
75
|
+
});
|
|
76
|
+
it('throws on empty messages', async () => {
|
|
77
|
+
const handler = makeHandler();
|
|
78
|
+
await expect(handler.handle({ messages: [] })).rejects.toThrow('Messages are required');
|
|
79
|
+
});
|
|
80
|
+
it('throws on invalid message', async () => {
|
|
81
|
+
const handler = makeHandler();
|
|
82
|
+
await expect(handler.handle({
|
|
83
|
+
messages: [{ role: 'user', content: ' ' }],
|
|
84
|
+
})).rejects.toThrow('Invalid message');
|
|
85
|
+
});
|
|
86
|
+
it('calls buildExtraSections hook', async () => {
|
|
87
|
+
const buildExtraSections = vi
|
|
88
|
+
.fn()
|
|
89
|
+
.mockResolvedValue([{ label: 'Memory', content: 'user likes tables', priority: 5 }]);
|
|
90
|
+
const handler = createChatRouteHandler({
|
|
91
|
+
stores: {
|
|
92
|
+
agent: new InMemoryAgentStore(),
|
|
93
|
+
entity: new InMemoryEntityStore(),
|
|
94
|
+
chat: new InMemoryChatStore(),
|
|
95
|
+
tenant: new InMemoryTenantStore(),
|
|
96
|
+
memory: new InMemoryMemoryStore(),
|
|
97
|
+
},
|
|
98
|
+
defaultAgentSlug: 'analyst',
|
|
99
|
+
resolveModel: () => mockModel,
|
|
100
|
+
agentRegistry,
|
|
101
|
+
hooks: { buildExtraSections },
|
|
102
|
+
});
|
|
103
|
+
await handler.handle({
|
|
104
|
+
messages: [{ role: 'user', content: 'hello' }],
|
|
105
|
+
});
|
|
106
|
+
expect(buildExtraSections).toHaveBeenCalledTimes(1);
|
|
107
|
+
expect(buildExtraSections).toHaveBeenCalledWith(expect.objectContaining({
|
|
108
|
+
agent: expect.objectContaining({ id: 'analyst' }),
|
|
109
|
+
}));
|
|
110
|
+
});
|
|
111
|
+
});
|
|
112
|
+
//# sourceMappingURL=chat-handler.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"chat-handler.test.js","sourceRoot":"","sources":["../../src/chat/chat-handler.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAE9D,OAAO,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AACxD,OAAO,EAAE,kBAAkB,EAAE,MAAM,kCAAkC,CAAC;AACtE,OAAO,EAAE,iBAAiB,EAAE,MAAM,iCAAiC,CAAC;AACpE,OAAO,EAAE,mBAAmB,EAAE,MAAM,mCAAmC,CAAC;AACxE,OAAO,EAAE,mBAAmB,EAAE,MAAM,mCAAmC,CAAC;AACxE,OAAO,EAAE,mBAAmB,EAAE,MAAM,mCAAmC,CAAC;AAExE,OAAO,EAAE,sBAAsB,EAAE,MAAM,gBAAgB,CAAC;AAExD,0CAA0C;AAC1C,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,GAAG,EAAE,CAAC,CAAC;IACnB,UAAU,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC;QACvB,UAAU,EAAE,CAAC,KAAK,SAAS,CAAC;YAC1B,MAAM,OAAO,CAAC;QAChB,CAAC,CAAC,EAAE;QACJ,oBAAoB,EAAE,GAAG,EAAE,CAAC,IAAI,QAAQ,CAAC,QAAQ,CAAC;KACnD,CAAC,CAAC;IACH,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;IACrB,WAAW,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAS,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;IACrE,IAAI,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,MAA+B,EAAE,EAAE,CAAC,CAAC;QAChD,IAAI,EAAE,UAAU;QAChB,GAAG,MAAM;KACV,CAAC,CAAC;CACJ,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;IACtC,IAAI,aAA4B,CAAC;IACjC,MAAM,SAAS,GAAG,EAAE,OAAO,EAAE,YAAY,EAAW,CAAC;IAErD,UAAU,CAAC,GAAG,EAAE;QACd,aAAa,GAAG,IAAI,aAAa,EAAE,CAAC;QACpC,aAAa,CAAC,QAAQ,CAAC;YACrB,EAAE,EAAE,SAAS;YACb,IAAI,EAAE,SAAS;YACf,WAAW,EAAE,cAAc;YAC3B,IAAI,EAAE,OAAO;YACb,YAAY,EAAE,kBAAkB;YAChC,UAAU,EAAE,EAAE;YACd,WAAW,EAAE,EAAE;SAChB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,SAAS,WAAW;QAClB,OAAO,sBAAsB,CAAC;YAC5B,MAAM,EAAE;gBACN,KAAK,EAAE,IAAI,kBAAkB,EAAE;gBAC/B,MAAM,EAAE,IAAI,mBAAmB,EAAE;gBACjC,IAAI,EAAE,IAAI,iBAAiB,EAAE;gBAC7B,MAAM,EAAE,IAAI,mBAAmB,EAAE;gBACjC,MAAM,EAAE,IAAI,mBAAmB,EAAE;aAClC;YACD,gBAAgB,EAAE,SAAS;YAC3B,YAAY,EAAE,GAAG,EAAE,CAAC,SAAS;YAC7B,aAAa;SACd,CAAC,CAAC;IACL,CAAC;IAED,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;QACjD,MAAM,OAAO,GAAG,WAAW,EAAE,CAAC;QAC9B,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC;YACrC,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;SAC/C,CAAC,CAAC;QACH,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACjC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;QAC3C,aAAa,CAAC,QAAQ,CAAC;YACrB,EAAE,EAAE,YAAY;YAChB,IAAI,EAAE,YAAY;YAClB,WAAW,EAAE,gBAAgB;YAC7B,IAAI,EAAE,QAAQ;YACd,YAAY,EAAE,kBAAkB;SACjC,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,WAAW,EAAE,CAAC;QAC9B,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC;YACrC,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC;YAC/C,SAAS,EAAE,YAAY;SACxB,CAAC,CAAC;QACH,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;QACxC,MAAM,OAAO,GAAG,WAAW,EAAE,CAAC;QAC9B,MAAM,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,uBAAuB,CAAC,CAAC;IAC1F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;QACzC,MAAM,OAAO,GAAG,WAAW,EAAE,CAAC;QAC9B,MAAM,MAAM,CACV,OAAO,CAAC,MAAM,CAAC;YACb,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;SAC5C,CAAC,CACH,CAAC,OAAO,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;QAC7C,MAAM,kBAAkB,GAAG,EAAE;aAC1B,EAAE,EAAE;aACJ,iBAAiB,CAAC,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,mBAAmB,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QAEvF,MAAM,OAAO,GAAG,sBAAsB,CAAC;YACrC,MAAM,EAAE;gBACN,KAAK,EAAE,IAAI,kBAAkB,EAAE;gBAC/B,MAAM,EAAE,IAAI,mBAAmB,EAAE;gBACjC,IAAI,EAAE,IAAI,iBAAiB,EAAE;gBAC7B,MAAM,EAAE,IAAI,mBAAmB,EAAE;gBACjC,MAAM,EAAE,IAAI,mBAAmB,EAAE;aAClC;YACD,gBAAgB,EAAE,SAAS;YAC3B,YAAY,EAAE,GAAG,EAAE,CAAC,SAAS;YAC7B,aAAa;YACb,KAAK,EAAE,EAAE,kBAAkB,EAAE;SAC9B,CAAC,CAAC;QAEH,MAAM,OAAO,CAAC,MAAM,CAAC;YACnB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;SAC/C,CAAC,CAAC;QAEH,MAAM,CAAC,kBAAkB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACpD,MAAM,CAAC,kBAAkB,CAAC,CAAC,oBAAoB,CAC7C,MAAM,CAAC,gBAAgB,CAAC;YACtB,KAAK,EAAE,MAAM,CAAC,gBAAgB,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,CAAC;SAClD,CAAC,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,4 @@
|
|
|
1
|
+
export { createChatRouteHandler } from "./chat-handler";
|
|
2
|
+
export type { ChatHandlerConfig, ChatRequest, ChatHandler, ChatContext, } from "./chat-handler";
|
|
3
|
+
export { validateMessage, sanitizeContent, extractTextFromParts, toAIMessages, } from "./message-utils";
|
|
4
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/chat/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,sBAAsB,EAAE,MAAM,gBAAgB,CAAC;AACxD,YAAY,EACV,iBAAiB,EACjB,WAAW,EACX,WAAW,EACX,WAAW,GACZ,MAAM,gBAAgB,CAAC;AACxB,OAAO,EACL,eAAe,EACf,eAAe,EACf,oBAAoB,EACpB,YAAY,GACb,MAAM,iBAAiB,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/chat/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,sBAAsB,EAAE,MAAM,gBAAgB,CAAC;AAOxD,OAAO,EACL,eAAe,EACf,eAAe,EACf,oBAAoB,EACpB,YAAY,GACb,MAAM,iBAAiB,CAAC"}
|
|
@@ -0,0 +1,26 @@
|
|
|
1
|
+
/** Validate that a message object has the required shape. */
|
|
2
|
+
export declare function validateMessage(message: unknown): {
|
|
3
|
+
valid: boolean;
|
|
4
|
+
error?: string;
|
|
5
|
+
};
|
|
6
|
+
/** Sanitize message content by trimming and removing null bytes. */
|
|
7
|
+
export declare function sanitizeContent(content: string): string;
|
|
8
|
+
/**
|
|
9
|
+
* Extract plain text from AI SDK message parts.
|
|
10
|
+
* Handles text parts, tool calls, tool results, etc.
|
|
11
|
+
*/
|
|
12
|
+
export declare function extractTextFromParts(parts: unknown[] | null | undefined): string;
|
|
13
|
+
/**
|
|
14
|
+
* Convert messages to the format expected by AI SDK.
|
|
15
|
+
* Ensures each message has role and content.
|
|
16
|
+
*/
|
|
17
|
+
export declare function toAIMessages(messages: Array<{
|
|
18
|
+
role?: string;
|
|
19
|
+
content?: string;
|
|
20
|
+
sender_type?: string;
|
|
21
|
+
parts?: unknown[];
|
|
22
|
+
}>): Array<{
|
|
23
|
+
role: "user" | "assistant" | "system";
|
|
24
|
+
content: string;
|
|
25
|
+
}>;
|
|
26
|
+
//# sourceMappingURL=message-utils.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"message-utils.d.ts","sourceRoot":"","sources":["../../src/chat/message-utils.ts"],"names":[],"mappings":"AAAA,6DAA6D;AAC7D,wBAAgB,eAAe,CAAC,OAAO,EAAE,OAAO,GAAG;IACjD,KAAK,EAAE,OAAO,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB,CAmBA;AAED,oEAAoE;AACpE,wBAAgB,eAAe,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,CAEvD;AAED;;;GAGG;AACH,wBAAgB,oBAAoB,CAClC,KAAK,EAAE,OAAO,EAAE,GAAG,IAAI,GAAG,SAAS,GAClC,MAAM,CAkBR;AAED;;;GAGG;AACH,wBAAgB,YAAY,CAC1B,QAAQ,EAAE,KAAK,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,KAAK,CAAC,EAAE,OAAO,EAAE,CAAC;CACnB,CAAC,GACD,KAAK,CAAC;IAAE,IAAI,EAAE,MAAM,GAAG,WAAW,GAAG,QAAQ,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,CAAC,CAMnE"}
|
|
@@ -0,0 +1,65 @@
|
|
|
1
|
+
/** Validate that a message object has the required shape. */
|
|
2
|
+
export function validateMessage(message) {
|
|
3
|
+
if (!message || typeof message !== "object") {
|
|
4
|
+
return { valid: false, error: "Message must be an object" };
|
|
5
|
+
}
|
|
6
|
+
const msg = message;
|
|
7
|
+
if (typeof msg.content !== "string" && !Array.isArray(msg.parts)) {
|
|
8
|
+
return {
|
|
9
|
+
valid: false,
|
|
10
|
+
error: "Message must have content (string) or parts (array)",
|
|
11
|
+
};
|
|
12
|
+
}
|
|
13
|
+
if (typeof msg.content === "string" && msg.content.trim().length === 0) {
|
|
14
|
+
return { valid: false, error: "Message content cannot be empty" };
|
|
15
|
+
}
|
|
16
|
+
return { valid: true };
|
|
17
|
+
}
|
|
18
|
+
/** Sanitize message content by trimming and removing null bytes. */
|
|
19
|
+
export function sanitizeContent(content) {
|
|
20
|
+
return content.replace(/\0/g, "").trim();
|
|
21
|
+
}
|
|
22
|
+
/**
|
|
23
|
+
* Extract plain text from AI SDK message parts.
|
|
24
|
+
* Handles text parts, tool calls, tool results, etc.
|
|
25
|
+
*/
|
|
26
|
+
export function extractTextFromParts(parts) {
|
|
27
|
+
if (!parts || parts.length === 0)
|
|
28
|
+
return "";
|
|
29
|
+
const textParts = [];
|
|
30
|
+
for (const part of parts) {
|
|
31
|
+
if (!part || typeof part !== "object")
|
|
32
|
+
continue;
|
|
33
|
+
const p = part;
|
|
34
|
+
if (p.type === "text" && typeof p.text === "string") {
|
|
35
|
+
textParts.push(p.text);
|
|
36
|
+
}
|
|
37
|
+
else if (p.type === "tool-result" && typeof p.result === "string") {
|
|
38
|
+
textParts.push(p.result);
|
|
39
|
+
}
|
|
40
|
+
}
|
|
41
|
+
return textParts.join("\n");
|
|
42
|
+
}
|
|
43
|
+
/**
|
|
44
|
+
* Convert messages to the format expected by AI SDK.
|
|
45
|
+
* Ensures each message has role and content.
|
|
46
|
+
*/
|
|
47
|
+
export function toAIMessages(messages) {
|
|
48
|
+
return messages.map((msg) => {
|
|
49
|
+
const role = mapRole(msg.role ?? msg.sender_type ?? "user");
|
|
50
|
+
const content = msg.content ?? extractTextFromParts(msg.parts) ?? "";
|
|
51
|
+
return { role, content };
|
|
52
|
+
});
|
|
53
|
+
}
|
|
54
|
+
function mapRole(role) {
|
|
55
|
+
switch (role) {
|
|
56
|
+
case "assistant":
|
|
57
|
+
case "agent":
|
|
58
|
+
return "assistant";
|
|
59
|
+
case "system":
|
|
60
|
+
return "system";
|
|
61
|
+
default:
|
|
62
|
+
return "user";
|
|
63
|
+
}
|
|
64
|
+
}
|
|
65
|
+
//# sourceMappingURL=message-utils.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"message-utils.js","sourceRoot":"","sources":["../../src/chat/message-utils.ts"],"names":[],"mappings":"AAAA,6DAA6D;AAC7D,MAAM,UAAU,eAAe,CAAC,OAAgB;IAI9C,IAAI,CAAC,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;QAC5C,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,2BAA2B,EAAE,CAAC;IAC9D,CAAC;IAED,MAAM,GAAG,GAAG,OAAkC,CAAC;IAE/C,IAAI,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;QACjE,OAAO;YACL,KAAK,EAAE,KAAK;YACZ,KAAK,EAAE,qDAAqD;SAC7D,CAAC;IACJ,CAAC;IAED,IAAI,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,IAAI,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACvE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,iCAAiC,EAAE,CAAC;IACpE,CAAC;IAED,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;AACzB,CAAC;AAED,oEAAoE;AACpE,MAAM,UAAU,eAAe,CAAC,OAAe;IAC7C,OAAO,OAAO,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;AAC3C,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,oBAAoB,CAClC,KAAmC;IAEnC,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,EAAE,CAAC;IAE5C,MAAM,SAAS,GAAa,EAAE,CAAC;IAE/B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,IAAI,CAAC,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ;YAAE,SAAS;QAEhD,MAAM,CAAC,GAAG,IAA+B,CAAC;QAE1C,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,IAAI,OAAO,CAAC,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YACpD,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QACzB,CAAC;aAAM,IAAI,CAAC,CAAC,IAAI,KAAK,aAAa,IAAI,OAAO,CAAC,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;YACpE,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;QAC3B,CAAC;IACH,CAAC;IAED,OAAO,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC9B,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,YAAY,CAC1B,QAKE;IAEF,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;QAC1B,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,IAAI,IAAI,GAAG,CAAC,WAAW,IAAI,MAAM,CAAC,CAAC;QAC5D,MAAM,OAAO,GAAG,GAAG,CAAC,OAAO,IAAI,oBAAoB,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;QACrE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;IAC3B,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,OAAO,CAAC,IAAY;IAC3B,QAAQ,IAAI,EAAE,CAAC;QACb,KAAK,WAAW,CAAC;QACjB,KAAK,OAAO;YACV,OAAO,WAAW,CAAC;QACrB,KAAK,QAAQ;YACX,OAAO,QAAQ,CAAC;QAClB;YACE,OAAO,MAAM,CAAC;IAClB,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"message-utils.test.d.ts","sourceRoot":"","sources":["../../src/chat/message-utils.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,85 @@
|
|
|
1
|
+
import { describe, it, expect } from 'vitest';
|
|
2
|
+
import { validateMessage, sanitizeContent, extractTextFromParts, toAIMessages, } from './message-utils';
|
|
3
|
+
describe('validateMessage', () => {
|
|
4
|
+
it('validates a message with content', () => {
|
|
5
|
+
expect(validateMessage({ content: 'hello' })).toEqual({ valid: true });
|
|
6
|
+
});
|
|
7
|
+
it('validates a message with parts', () => {
|
|
8
|
+
expect(validateMessage({ parts: [{ type: 'text', text: 'hi' }] })).toEqual({
|
|
9
|
+
valid: true,
|
|
10
|
+
});
|
|
11
|
+
});
|
|
12
|
+
it('rejects non-object', () => {
|
|
13
|
+
expect(validateMessage(null)).toEqual({
|
|
14
|
+
valid: false,
|
|
15
|
+
error: 'Message must be an object',
|
|
16
|
+
});
|
|
17
|
+
expect(validateMessage('string')).toEqual({
|
|
18
|
+
valid: false,
|
|
19
|
+
error: 'Message must be an object',
|
|
20
|
+
});
|
|
21
|
+
});
|
|
22
|
+
it('rejects empty content', () => {
|
|
23
|
+
expect(validateMessage({ content: ' ' })).toEqual({
|
|
24
|
+
valid: false,
|
|
25
|
+
error: 'Message content cannot be empty',
|
|
26
|
+
});
|
|
27
|
+
});
|
|
28
|
+
it('rejects message without content or parts', () => {
|
|
29
|
+
expect(validateMessage({ role: 'user' })).toEqual({
|
|
30
|
+
valid: false,
|
|
31
|
+
error: 'Message must have content (string) or parts (array)',
|
|
32
|
+
});
|
|
33
|
+
});
|
|
34
|
+
});
|
|
35
|
+
describe('sanitizeContent', () => {
|
|
36
|
+
it('trims whitespace', () => {
|
|
37
|
+
expect(sanitizeContent(' hello ')).toBe('hello');
|
|
38
|
+
});
|
|
39
|
+
it('removes null bytes', () => {
|
|
40
|
+
expect(sanitizeContent('he\0llo')).toBe('hello');
|
|
41
|
+
});
|
|
42
|
+
});
|
|
43
|
+
describe('extractTextFromParts', () => {
|
|
44
|
+
it('returns empty for null/undefined', () => {
|
|
45
|
+
expect(extractTextFromParts(null)).toBe('');
|
|
46
|
+
expect(extractTextFromParts(undefined)).toBe('');
|
|
47
|
+
});
|
|
48
|
+
it('extracts text parts', () => {
|
|
49
|
+
const parts = [
|
|
50
|
+
{ type: 'text', text: 'Hello' },
|
|
51
|
+
{ type: 'text', text: 'World' },
|
|
52
|
+
];
|
|
53
|
+
expect(extractTextFromParts(parts)).toBe('Hello\nWorld');
|
|
54
|
+
});
|
|
55
|
+
it('extracts tool-result strings', () => {
|
|
56
|
+
const parts = [{ type: 'tool-result', result: 'Result here' }];
|
|
57
|
+
expect(extractTextFromParts(parts)).toBe('Result here');
|
|
58
|
+
});
|
|
59
|
+
it('skips non-text parts', () => {
|
|
60
|
+
const parts = [
|
|
61
|
+
{ type: 'text', text: 'hello' },
|
|
62
|
+
{ type: 'tool-call', toolName: 'search' },
|
|
63
|
+
];
|
|
64
|
+
expect(extractTextFromParts(parts)).toBe('hello');
|
|
65
|
+
});
|
|
66
|
+
});
|
|
67
|
+
describe('toAIMessages', () => {
|
|
68
|
+
it('maps user messages', () => {
|
|
69
|
+
const result = toAIMessages([{ role: 'user', content: 'hi' }]);
|
|
70
|
+
expect(result).toEqual([{ role: 'user', content: 'hi' }]);
|
|
71
|
+
});
|
|
72
|
+
it('maps agent to assistant', () => {
|
|
73
|
+
const result = toAIMessages([{ sender_type: 'agent', content: 'response' }]);
|
|
74
|
+
expect(result).toEqual([{ role: 'assistant', content: 'response' }]);
|
|
75
|
+
});
|
|
76
|
+
it('maps system messages', () => {
|
|
77
|
+
const result = toAIMessages([{ role: 'system', content: 'context' }]);
|
|
78
|
+
expect(result).toEqual([{ role: 'system', content: 'context' }]);
|
|
79
|
+
});
|
|
80
|
+
it('defaults unknown roles to user', () => {
|
|
81
|
+
const result = toAIMessages([{ role: 'unknown', content: 'test' }]);
|
|
82
|
+
expect(result[0].role).toBe('user');
|
|
83
|
+
});
|
|
84
|
+
});
|
|
85
|
+
//# sourceMappingURL=message-utils.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"message-utils.test.js","sourceRoot":"","sources":["../../src/chat/message-utils.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAE9C,OAAO,EACL,eAAe,EACf,eAAe,EACf,oBAAoB,EACpB,YAAY,GACb,MAAM,iBAAiB,CAAC;AAEzB,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAC1C,MAAM,CAAC,eAAe,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACzE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;QACxC,MAAM,CAAC,eAAe,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC;YACzE,KAAK,EAAE,IAAI;SACZ,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAC5B,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC;YACpC,KAAK,EAAE,KAAK;YACZ,KAAK,EAAE,2BAA2B;SACnC,CAAC,CAAC;QACH,MAAM,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC;YACxC,KAAK,EAAE,KAAK;YACZ,KAAK,EAAE,2BAA2B;SACnC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uBAAuB,EAAE,GAAG,EAAE;QAC/B,MAAM,CAAC,eAAe,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC;YACjD,KAAK,EAAE,KAAK;YACZ,KAAK,EAAE,iCAAiC;SACzC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,MAAM,CAAC,eAAe,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC;YAChD,KAAK,EAAE,KAAK;YACZ,KAAK,EAAE,qDAAqD;SAC7D,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,EAAE,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAC1B,MAAM,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAC5B,MAAM,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;IACpC,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAC1C,MAAM,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC5C,MAAM,CAAC,oBAAoB,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qBAAqB,EAAE,GAAG,EAAE;QAC7B,MAAM,KAAK,GAAG;YACZ,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE;YAC/B,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE;SAChC,CAAC;QACF,MAAM,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IAC3D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;QACtC,MAAM,KAAK,GAAG,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC,CAAC;QAC/D,MAAM,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sBAAsB,EAAE,GAAG,EAAE;QAC9B,MAAM,KAAK,GAAG;YACZ,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE;YAC/B,EAAE,IAAI,EAAE,WAAW,EAAE,QAAQ,EAAE,QAAQ,EAAE;SAC1C,CAAC;QACF,MAAM,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,EAAE,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAC5B,MAAM,MAAM,GAAG,YAAY,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;QAC/D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACjC,MAAM,MAAM,GAAG,YAAY,CAAC,CAAC,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC;QAC7E,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC;IACvE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sBAAsB,EAAE,GAAG,EAAE;QAC9B,MAAM,MAAM,GAAG,YAAY,CAAC,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;QACtE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;IACnE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;QACxC,MAAM,MAAM,GAAG,YAAY,CAAC,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;QACpE,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,22 @@
|
|
|
1
|
+
import type { EvalScorer, EvalDataset, EvalRun } from "@sprinterai/core";
|
|
2
|
+
/** Function that produces an output from an input. */
|
|
3
|
+
export type EvalTarget = (input: unknown) => Promise<unknown>;
|
|
4
|
+
/** Options for running an evaluation. */
|
|
5
|
+
export interface RunEvalOptions {
|
|
6
|
+
/** Scorers to apply to each output. */
|
|
7
|
+
scorers: EvalScorer[];
|
|
8
|
+
/** Dataset with inputs and expected outputs. */
|
|
9
|
+
dataset: EvalDataset;
|
|
10
|
+
/** The function/agent/tool to evaluate. */
|
|
11
|
+
target: EvalTarget;
|
|
12
|
+
/** Optional agent slug for metadata. */
|
|
13
|
+
agentSlug?: string;
|
|
14
|
+
/** Optional tool slug for metadata. */
|
|
15
|
+
toolSlug?: string;
|
|
16
|
+
}
|
|
17
|
+
/**
|
|
18
|
+
* Run an evaluation: execute the target on each dataset entry,
|
|
19
|
+
* score with all scorers, and aggregate results.
|
|
20
|
+
*/
|
|
21
|
+
export declare function runEval(options: RunEvalOptions): Promise<EvalRun>;
|
|
22
|
+
//# sourceMappingURL=eval-runner.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"eval-runner.d.ts","sourceRoot":"","sources":["../../src/eval/eval-runner.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,UAAU,EACV,WAAW,EAEX,OAAO,EACR,MAAM,kBAAkB,CAAC;AAE1B,sDAAsD;AACtD,MAAM,MAAM,UAAU,GAAG,CAAC,KAAK,EAAE,OAAO,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC;AAE9D,yCAAyC;AACzC,MAAM,WAAW,cAAc;IAC7B,uCAAuC;IACvC,OAAO,EAAE,UAAU,EAAE,CAAC;IACtB,gDAAgD;IAChD,OAAO,EAAE,WAAW,CAAC;IACrB,2CAA2C;IAC3C,MAAM,EAAE,UAAU,CAAC;IACnB,wCAAwC;IACxC,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,uCAAuC;IACvC,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED;;;GAGG;AACH,wBAAsB,OAAO,CAAC,OAAO,EAAE,cAAc,GAAG,OAAO,CAAC,OAAO,CAAC,CA0DvE"}
|
|
@@ -0,0 +1,56 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Run an evaluation: execute the target on each dataset entry,
|
|
3
|
+
* score with all scorers, and aggregate results.
|
|
4
|
+
*/
|
|
5
|
+
export async function runEval(options) {
|
|
6
|
+
const { scorers, dataset, target, agentSlug, toolSlug } = options;
|
|
7
|
+
const results = [];
|
|
8
|
+
for (let i = 0; i < dataset.entries.length; i++) {
|
|
9
|
+
const entry = dataset.entries[i];
|
|
10
|
+
const start = Date.now();
|
|
11
|
+
let output;
|
|
12
|
+
try {
|
|
13
|
+
output = await target(entry.input);
|
|
14
|
+
}
|
|
15
|
+
catch (err) {
|
|
16
|
+
output = { error: err instanceof Error ? err.message : String(err) };
|
|
17
|
+
}
|
|
18
|
+
const durationMs = Date.now() - start;
|
|
19
|
+
// Score with each scorer
|
|
20
|
+
const scores = {};
|
|
21
|
+
for (const scorer of scorers) {
|
|
22
|
+
scores[scorer.name] = await scorer.score({
|
|
23
|
+
input: entry.input,
|
|
24
|
+
output,
|
|
25
|
+
expected: entry.expected,
|
|
26
|
+
metadata: entry.metadata,
|
|
27
|
+
});
|
|
28
|
+
}
|
|
29
|
+
results.push({
|
|
30
|
+
entryIndex: i,
|
|
31
|
+
scores,
|
|
32
|
+
output,
|
|
33
|
+
durationMs,
|
|
34
|
+
});
|
|
35
|
+
}
|
|
36
|
+
// Compute summary statistics per scorer
|
|
37
|
+
const summary = {};
|
|
38
|
+
for (const scorer of scorers) {
|
|
39
|
+
const vals = results.map((r) => r.scores[scorer.name]);
|
|
40
|
+
summary[scorer.name] = {
|
|
41
|
+
mean: vals.reduce((a, b) => a + b, 0) / vals.length,
|
|
42
|
+
min: Math.min(...vals),
|
|
43
|
+
max: Math.max(...vals),
|
|
44
|
+
};
|
|
45
|
+
}
|
|
46
|
+
return {
|
|
47
|
+
id: `eval-${Date.now()}`,
|
|
48
|
+
datasetName: dataset.name,
|
|
49
|
+
agentSlug,
|
|
50
|
+
toolSlug,
|
|
51
|
+
results,
|
|
52
|
+
summary,
|
|
53
|
+
createdAt: new Date().toISOString(),
|
|
54
|
+
};
|
|
55
|
+
}
|
|
56
|
+
//# sourceMappingURL=eval-runner.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"eval-runner.js","sourceRoot":"","sources":["../../src/eval/eval-runner.ts"],"names":[],"mappings":"AAwBA;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,OAAO,CAAC,OAAuB;IACnD,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC;IAClE,MAAM,OAAO,GAAoB,EAAE,CAAC;IAEpC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAChD,MAAM,KAAK,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;QACjC,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEzB,IAAI,MAAe,CAAC;QACpB,IAAI,CAAC;YACH,MAAM,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QACrC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,GAAG,EAAE,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC;QACvE,CAAC;QAED,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC;QAEtC,yBAAyB;QACzB,MAAM,MAAM,GAA2B,EAAE,CAAC;QAC1C,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;YAC7B,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC;gBACvC,KAAK,EAAE,KAAK,CAAC,KAAK;gBAClB,MAAM;gBACN,QAAQ,EAAE,KAAK,CAAC,QAAQ;gBACxB,QAAQ,EAAE,KAAK,CAAC,QAAQ;aACzB,CAAC,CAAC;QACL,CAAC;QAED,OAAO,CAAC,IAAI,CAAC;YACX,UAAU,EAAE,CAAC;YACb,MAAM;YACN,MAAM;YACN,UAAU;SACX,CAAC,CAAC;IACL,CAAC;IAED,wCAAwC;IACxC,MAAM,OAAO,GACX,EAAE,CAAC;IAEL,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;QAC7B,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;QACvD,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG;YACrB,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,GAAG,IAAI,CAAC,MAAM;YACnD,GAAG,EAAE,IAAI,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC;YACtB,GAAG,EAAE,IAAI,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC;SACvB,CAAC;IACJ,CAAC;IAED,OAAO;QACL,EAAE,EAAE,QAAQ,IAAI,CAAC,GAAG,EAAE,EAAE;QACxB,WAAW,EAAE,OAAO,CAAC,IAAI;QACzB,SAAS;QACT,QAAQ;QACR,OAAO;QACP,OAAO;QACP,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;KACpC,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"eval-runner.test.d.ts","sourceRoot":"","sources":["../../src/eval/eval-runner.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,89 @@
|
|
|
1
|
+
import { describe, it, expect } from 'vitest';
|
|
2
|
+
import { runEval } from './eval-runner';
|
|
3
|
+
import { exactMatch, numericCloseness } from './scorers';
|
|
4
|
+
describe('runEval', () => {
|
|
5
|
+
it('runs evaluation and returns results', async () => {
|
|
6
|
+
const dataset = {
|
|
7
|
+
name: 'simple-test',
|
|
8
|
+
entries: [
|
|
9
|
+
{ input: 'hello', expected: 'hello' },
|
|
10
|
+
{ input: 'world', expected: 'world' },
|
|
11
|
+
{ input: 'foo', expected: 'bar' },
|
|
12
|
+
],
|
|
13
|
+
};
|
|
14
|
+
const run = await runEval({
|
|
15
|
+
scorers: [exactMatch],
|
|
16
|
+
dataset,
|
|
17
|
+
target: async (input) => input, // identity function
|
|
18
|
+
});
|
|
19
|
+
expect(run.datasetName).toBe('simple-test');
|
|
20
|
+
expect(run.results).toHaveLength(3);
|
|
21
|
+
// First two should be exact matches
|
|
22
|
+
expect(run.results[0].scores['exact-match']).toBe(1.0);
|
|
23
|
+
expect(run.results[1].scores['exact-match']).toBe(1.0);
|
|
24
|
+
// Third should not match
|
|
25
|
+
expect(run.results[2].scores['exact-match']).toBe(0.0);
|
|
26
|
+
// Summary
|
|
27
|
+
expect(run.summary['exact-match'].mean).toBeCloseTo(2 / 3, 2);
|
|
28
|
+
expect(run.summary['exact-match'].min).toBe(0.0);
|
|
29
|
+
expect(run.summary['exact-match'].max).toBe(1.0);
|
|
30
|
+
});
|
|
31
|
+
it('handles multiple scorers', async () => {
|
|
32
|
+
const dataset = {
|
|
33
|
+
name: 'multi-scorer',
|
|
34
|
+
entries: [{ input: 10, expected: 10 }],
|
|
35
|
+
};
|
|
36
|
+
const run = await runEval({
|
|
37
|
+
scorers: [exactMatch, numericCloseness],
|
|
38
|
+
dataset,
|
|
39
|
+
target: async (input) => input,
|
|
40
|
+
});
|
|
41
|
+
expect(Object.keys(run.results[0].scores)).toEqual(['exact-match', 'numeric-closeness']);
|
|
42
|
+
expect(run.results[0].scores['exact-match']).toBe(1.0);
|
|
43
|
+
expect(run.results[0].scores['numeric-closeness']).toBe(1.0);
|
|
44
|
+
});
|
|
45
|
+
it('captures errors from target function', async () => {
|
|
46
|
+
const dataset = {
|
|
47
|
+
name: 'error-test',
|
|
48
|
+
entries: [{ input: 'trigger-error', expected: 'ok' }],
|
|
49
|
+
};
|
|
50
|
+
const run = await runEval({
|
|
51
|
+
scorers: [exactMatch],
|
|
52
|
+
dataset,
|
|
53
|
+
target: async () => {
|
|
54
|
+
throw new Error('test error');
|
|
55
|
+
},
|
|
56
|
+
});
|
|
57
|
+
expect(run.results[0].output).toEqual({ error: 'test error' });
|
|
58
|
+
expect(run.results[0].scores['exact-match']).toBe(0.0);
|
|
59
|
+
});
|
|
60
|
+
it('includes duration for each entry', async () => {
|
|
61
|
+
const dataset = {
|
|
62
|
+
name: 'timing-test',
|
|
63
|
+
entries: [{ input: 1 }],
|
|
64
|
+
};
|
|
65
|
+
const run = await runEval({
|
|
66
|
+
scorers: [exactMatch],
|
|
67
|
+
dataset,
|
|
68
|
+
target: async (input) => input,
|
|
69
|
+
});
|
|
70
|
+
expect(run.results[0].durationMs).toBeGreaterThanOrEqual(0);
|
|
71
|
+
});
|
|
72
|
+
it('sets metadata fields', async () => {
|
|
73
|
+
const dataset = {
|
|
74
|
+
name: 'meta-test',
|
|
75
|
+
entries: [{ input: 1 }],
|
|
76
|
+
};
|
|
77
|
+
const run = await runEval({
|
|
78
|
+
scorers: [exactMatch],
|
|
79
|
+
dataset,
|
|
80
|
+
target: async (input) => input,
|
|
81
|
+
agentSlug: 'analyst',
|
|
82
|
+
toolSlug: 'search',
|
|
83
|
+
});
|
|
84
|
+
expect(run.agentSlug).toBe('analyst');
|
|
85
|
+
expect(run.toolSlug).toBe('search');
|
|
86
|
+
expect(run.createdAt).toBeDefined();
|
|
87
|
+
});
|
|
88
|
+
});
|
|
89
|
+
//# sourceMappingURL=eval-runner.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"eval-runner.test.js","sourceRoot":"","sources":["../../src/eval/eval-runner.test.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAE9C,OAAO,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AACxC,OAAO,EAAE,UAAU,EAAE,gBAAgB,EAAE,MAAM,WAAW,CAAC;AAEzD,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;IACvB,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;QACnD,MAAM,OAAO,GAAgB;YAC3B,IAAI,EAAE,aAAa;YACnB,OAAO,EAAE;gBACP,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE;gBACrC,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE;gBACrC,EAAE,KAAK,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE;aAClC;SACF,CAAC;QAEF,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC;YACxB,OAAO,EAAE,CAAC,UAAU,CAAC;YACrB,OAAO;YACP,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,KAAK,EAAE,oBAAoB;SACrD,CAAC,CAAC;QAEH,MAAM,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAC5C,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAEpC,oCAAoC;QACpC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACvD,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACvD,yBAAyB;QACzB,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAEvD,UAAU;QACV,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;QAC9D,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACjD,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;QACxC,MAAM,OAAO,GAAgB;YAC3B,IAAI,EAAE,cAAc;YACpB,OAAO,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC;SACvC,CAAC;QAEF,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC;YACxB,OAAO,EAAE,CAAC,UAAU,EAAE,gBAAgB,CAAC;YACvC,OAAO;YACP,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,KAAK;SAC/B,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,aAAa,EAAE,mBAAmB,CAAC,CAAC,CAAC;QACzF,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACvD,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QACpD,MAAM,OAAO,GAAgB;YAC3B,IAAI,EAAE,YAAY;YAClB,OAAO,EAAE,CAAC,EAAE,KAAK,EAAE,eAAe,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;SACtD,CAAC;QAEF,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC;YACxB,OAAO,EAAE,CAAC,UAAU,CAAC;YACrB,OAAO;YACP,MAAM,EAAE,KAAK,IAAI,EAAE;gBACjB,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC;YAChC,CAAC;SACF,CAAC,CAAC;QAEH,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC,CAAC;QAC/D,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACzD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;QAChD,MAAM,OAAO,GAAgB;YAC3B,IAAI,EAAE,aAAa;YACnB,OAAO,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;SACxB,CAAC;QAEF,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC;YACxB,OAAO,EAAE,CAAC,UAAU,CAAC;YACrB,OAAO;YACP,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,KAAK;SAC/B,CAAC,CAAC;QAEH,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE;QACpC,MAAM,OAAO,GAAgB;YAC3B,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;SACxB,CAAC;QAEF,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC;YACxB,OAAO,EAAE,CAAC,UAAU,CAAC;YACrB,OAAO;YACP,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,KAAK;YAC9B,SAAS,EAAE,SAAS;YACpB,QAAQ,EAAE,QAAQ;SACnB,CAAC,CAAC;QAEH,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACtC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACpC,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,WAAW,EAAE,CAAC;IACtC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/eval/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,UAAU,EACV,UAAU,EACV,gBAAgB,EAChB,QAAQ,EACR,gBAAgB,GACjB,MAAM,WAAW,CAAC;AACnB,OAAO,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AACxC,YAAY,EAAE,UAAU,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/eval/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,UAAU,EACV,UAAU,EACV,gBAAgB,EAChB,QAAQ,EACR,gBAAgB,GACjB,MAAM,WAAW,CAAC;AACnB,OAAO,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC"}
|
|
@@ -0,0 +1,26 @@
|
|
|
1
|
+
import type { EvalScorer } from "@sprinterai/core";
|
|
2
|
+
/**
|
|
3
|
+
* Exact match scorer: 1.0 if output === expected, 0.0 otherwise.
|
|
4
|
+
* Compares JSON stringified values for deep equality.
|
|
5
|
+
*/
|
|
6
|
+
export declare const exactMatch: EvalScorer;
|
|
7
|
+
/**
|
|
8
|
+
* Fuzzy text match scorer: compares strings case-insensitively after trimming.
|
|
9
|
+
* Returns a score based on substring containment.
|
|
10
|
+
* 1.0 = exact match, 0.5 = one contains the other, 0.0 = no match.
|
|
11
|
+
*/
|
|
12
|
+
export declare const fuzzyMatch: EvalScorer;
|
|
13
|
+
/**
|
|
14
|
+
* Numeric closeness scorer: returns 1.0 if values are equal,
|
|
15
|
+
* decreasing as the difference increases relative to expected.
|
|
16
|
+
* Uses formula: 1 - min(|output - expected| / |expected|, 1)
|
|
17
|
+
*/
|
|
18
|
+
export declare const numericCloseness: EvalScorer;
|
|
19
|
+
/**
|
|
20
|
+
* Contains scorer: checks if the output string contains the expected string.
|
|
21
|
+
* Useful for checking if an agent mentioned specific keywords.
|
|
22
|
+
*/
|
|
23
|
+
export declare const contains: EvalScorer;
|
|
24
|
+
/** All built-in scorers. */
|
|
25
|
+
export declare const BUILT_IN_SCORERS: EvalScorer[];
|
|
26
|
+
//# sourceMappingURL=scorers.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"scorers.d.ts","sourceRoot":"","sources":["../../src/eval/scorers.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAEnD;;;GAGG;AACH,eAAO,MAAM,UAAU,EAAE,UAOxB,CAAC;AAEF;;;;GAIG;AACH,eAAO,MAAM,UAAU,EAAE,UAcxB,CAAC;AAEF;;;;GAIG;AACH,eAAO,MAAM,gBAAgB,EAAE,UAa9B,CAAC;AAEF;;;GAGG;AACH,eAAO,MAAM,QAAQ,EAAE,UAStB,CAAC;AAEF,4BAA4B;AAC5B,eAAO,MAAM,gBAAgB,EAAE,UAAU,EAKxC,CAAC"}
|