groundswell 0.0.1
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/.claude/settings.local.json +9 -0
- package/.claude/system_prompts/task-breakdown.md +100 -0
- package/PRPs/001-hierarchical-workflow-engine.md +2438 -0
- package/PRPs/PRDs/001-hierarchical-workflow-engine.md +543 -0
- package/PRPs/PRDs/002-agent-prompt.md +390 -0
- package/PRPs/PRDs/003-agent-prompt.md +943 -0
- package/PRPs/PRDs/004-agent-prompt.md +1136 -0
- package/PRPs/PRDs/tasks-001.json +492 -0
- package/PRPs/README.md +83 -0
- package/PRPs/templates/prp_base.md +222 -0
- package/README.md +218 -0
- package/docs/agent.md +422 -0
- package/docs/prompt.md +419 -0
- package/docs/workflow.md +600 -0
- package/examples/README.md +244 -0
- package/examples/examples/01-basic-workflow.ts +100 -0
- package/examples/examples/02-decorator-options.ts +217 -0
- package/examples/examples/03-parent-child.ts +241 -0
- package/examples/examples/04-observers-debugger.ts +340 -0
- package/examples/examples/05-error-handling.ts +387 -0
- package/examples/examples/06-concurrent-tasks.ts +352 -0
- package/examples/examples/07-agent-loops.ts +432 -0
- package/examples/examples/08-sdk-features.ts +667 -0
- package/examples/examples/09-reflection.ts +573 -0
- package/examples/examples/10-introspection.ts +550 -0
- package/examples/index.ts +143 -0
- package/examples/utils/helpers.ts +57 -0
- package/llms_full.txt +5890 -0
- package/package.json +63 -0
- package/plan/P1P2/PRP.md +527 -0
- package/plan/P1P2/research/LRU_CACHE_BEST_PRACTICES.md +1929 -0
- package/plan/P1P2/research/LRU_CACHE_CODE_PATTERNS.md +857 -0
- package/plan/P1P2/research/LRU_CACHE_INTEGRATION_GUIDE.md +738 -0
- package/plan/P1P2/research/LRU_CACHE_RESEARCH_INDEX.md +424 -0
- package/plan/P1P2/research/REFLECTION_INDEX.md +291 -0
- package/plan/P1P2/research/REFLECTION_RESEARCH_REPORT.md +1342 -0
- package/plan/P1P2/research/RESEARCH_SUMMARY.md +342 -0
- package/plan/P1P2/research/anthropic-sdk.md +174 -0
- package/plan/P1P2/research/async-local-storage.md +200 -0
- package/plan/P1P2/research/reflection-code-patterns.md +1205 -0
- package/plan/P1P2/research/reflection-decision-matrix.md +421 -0
- package/plan/P1P2/research/reflection-implementation-guide.md +1341 -0
- package/plan/P1P2/research/reflection-integration-guide.md +834 -0
- package/plan/P1P2/research/reflection-patterns.md +1468 -0
- package/plan/P1P2/research/reflection-quick-reference.md +558 -0
- package/plan/P1P2/research/zod-schema.md +152 -0
- package/plan/P3P4/PRP.md +1388 -0
- package/plan/P3P4/research/caching-lru.md +116 -0
- package/plan/P3P4/research/introspection-tools.md +177 -0
- package/plan/P3P4/research/reflection-patterns.md +117 -0
- package/plan/P4P5/PRP.md +1136 -0
- package/plan/P4P5/research/RESEARCH_SUMMARY.md +151 -0
- package/plan/architecture/external_deps.md +358 -0
- package/plan/architecture/system_context.md +242 -0
- package/plan/backlog.json +867 -0
- package/plan/research/INTROSPECTION_RESEARCH_SUMMARY.md +378 -0
- package/plan/research/README-INTROSPECTION.md +352 -0
- package/plan/research/agent-introspection-patterns.md +1085 -0
- package/plan/research/introspection-security-guide.md +928 -0
- package/plan/research/introspection-tool-examples.md +875 -0
- package/scripts/generate-llms-full.ts +206 -0
- package/src/__tests__/integration/agent-workflow.test.ts +256 -0
- package/src/__tests__/integration/tree-mirroring.test.ts +114 -0
- package/src/__tests__/unit/agent.test.ts +169 -0
- package/src/__tests__/unit/cache-key.test.ts +182 -0
- package/src/__tests__/unit/cache.test.ts +172 -0
- package/src/__tests__/unit/context.test.ts +138 -0
- package/src/__tests__/unit/decorators.test.ts +100 -0
- package/src/__tests__/unit/introspection-tools.test.ts +277 -0
- package/src/__tests__/unit/prompt.test.ts +135 -0
- package/src/__tests__/unit/reflection.test.ts +210 -0
- package/src/__tests__/unit/tree-debugger.test.ts +85 -0
- package/src/__tests__/unit/workflow.test.ts +81 -0
- package/src/cache/cache-key.ts +244 -0
- package/src/cache/cache.ts +236 -0
- package/src/cache/index.ts +8 -0
- package/src/core/agent.ts +573 -0
- package/src/core/context.ts +119 -0
- package/src/core/event-tree.ts +260 -0
- package/src/core/factory.ts +123 -0
- package/src/core/index.ts +17 -0
- package/src/core/logger.ts +87 -0
- package/src/core/mcp-handler.ts +184 -0
- package/src/core/prompt.ts +150 -0
- package/src/core/workflow-context.ts +349 -0
- package/src/core/workflow.ts +302 -0
- package/src/debugger/index.ts +1 -0
- package/src/debugger/tree-debugger.ts +210 -0
- package/src/decorators/index.ts +3 -0
- package/src/decorators/observed-state.ts +95 -0
- package/src/decorators/step.ts +139 -0
- package/src/decorators/task.ts +96 -0
- package/src/examples/index.ts +2 -0
- package/src/examples/tdd-orchestrator.ts +65 -0
- package/src/examples/test-cycle-workflow.ts +64 -0
- package/src/index.ts +140 -0
- package/src/reflection/index.ts +5 -0
- package/src/reflection/reflection.ts +407 -0
- package/src/tools/index.ts +36 -0
- package/src/tools/introspection.ts +464 -0
- package/src/types/agent.ts +90 -0
- package/src/types/decorators.ts +25 -0
- package/src/types/error-strategy.ts +13 -0
- package/src/types/error.ts +20 -0
- package/src/types/events.ts +74 -0
- package/src/types/index.ts +55 -0
- package/src/types/logging.ts +24 -0
- package/src/types/observer.ts +18 -0
- package/src/types/prompt.ts +40 -0
- package/src/types/reflection.ts +117 -0
- package/src/types/sdk-primitives.ts +128 -0
- package/src/types/snapshot.ts +14 -0
- package/src/types/workflow-context.ts +163 -0
- package/src/types/workflow.ts +37 -0
- package/src/utils/id.ts +11 -0
- package/src/utils/index.ts +3 -0
- package/src/utils/observable.ts +77 -0
- package/tasks.json +0 -0
- package/tsconfig.json +22 -0
- package/vitest.config.ts +16 -0
|
@@ -0,0 +1,210 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Unit tests for ReflectionManager
|
|
3
|
+
*/
|
|
4
|
+
|
|
5
|
+
import { describe, it, expect, beforeEach } from 'vitest';
|
|
6
|
+
import { ReflectionManager } from '../../reflection/reflection.js';
|
|
7
|
+
import type { ReflectionContext, WorkflowNode } from '../../types/index.js';
|
|
8
|
+
|
|
9
|
+
describe('ReflectionManager', () => {
|
|
10
|
+
const createTestNode = (): WorkflowNode => ({
|
|
11
|
+
id: 'test-node-1',
|
|
12
|
+
name: 'Test Node',
|
|
13
|
+
parent: null,
|
|
14
|
+
children: [],
|
|
15
|
+
status: 'failed',
|
|
16
|
+
logs: [],
|
|
17
|
+
events: [],
|
|
18
|
+
stateSnapshot: null,
|
|
19
|
+
});
|
|
20
|
+
|
|
21
|
+
const createContext = (
|
|
22
|
+
error: Error,
|
|
23
|
+
attemptNumber: number = 1
|
|
24
|
+
): ReflectionContext => ({
|
|
25
|
+
level: 'workflow',
|
|
26
|
+
failedNode: createTestNode(),
|
|
27
|
+
error,
|
|
28
|
+
attemptNumber,
|
|
29
|
+
previousAttempts: [],
|
|
30
|
+
});
|
|
31
|
+
|
|
32
|
+
describe('isEnabled', () => {
|
|
33
|
+
it('should return true when enabled', () => {
|
|
34
|
+
const manager = new ReflectionManager({ enabled: true, maxAttempts: 3 });
|
|
35
|
+
expect(manager.isEnabled()).toBe(true);
|
|
36
|
+
});
|
|
37
|
+
|
|
38
|
+
it('should return false when disabled', () => {
|
|
39
|
+
const manager = new ReflectionManager({ enabled: false, maxAttempts: 3 });
|
|
40
|
+
expect(manager.isEnabled()).toBe(false);
|
|
41
|
+
});
|
|
42
|
+
});
|
|
43
|
+
|
|
44
|
+
describe('getMaxAttempts', () => {
|
|
45
|
+
it('should return configured maxAttempts', () => {
|
|
46
|
+
const manager = new ReflectionManager({ enabled: true, maxAttempts: 5 });
|
|
47
|
+
expect(manager.getMaxAttempts()).toBe(5);
|
|
48
|
+
});
|
|
49
|
+
});
|
|
50
|
+
|
|
51
|
+
describe('getRetryDelayMs', () => {
|
|
52
|
+
it('should return 0 when not configured', () => {
|
|
53
|
+
const manager = new ReflectionManager({ enabled: true, maxAttempts: 3 });
|
|
54
|
+
expect(manager.getRetryDelayMs()).toBe(0);
|
|
55
|
+
});
|
|
56
|
+
|
|
57
|
+
it('should return configured delay', () => {
|
|
58
|
+
const manager = new ReflectionManager({
|
|
59
|
+
enabled: true,
|
|
60
|
+
maxAttempts: 3,
|
|
61
|
+
retryDelayMs: 1000,
|
|
62
|
+
});
|
|
63
|
+
expect(manager.getRetryDelayMs()).toBe(1000);
|
|
64
|
+
});
|
|
65
|
+
});
|
|
66
|
+
|
|
67
|
+
describe('triggerReflection', () => {
|
|
68
|
+
it('should throw when reflection is disabled', async () => {
|
|
69
|
+
const manager = new ReflectionManager({ enabled: false, maxAttempts: 3 });
|
|
70
|
+
|
|
71
|
+
await expect(manager.triggerReflection()).rejects.toThrow(
|
|
72
|
+
'Reflection is not enabled'
|
|
73
|
+
);
|
|
74
|
+
});
|
|
75
|
+
|
|
76
|
+
it('should record reflection in history', async () => {
|
|
77
|
+
const manager = new ReflectionManager({ enabled: true, maxAttempts: 3 });
|
|
78
|
+
|
|
79
|
+
await manager.triggerReflection('test reason');
|
|
80
|
+
|
|
81
|
+
const history = manager.getReflectionHistory();
|
|
82
|
+
expect(history).toHaveLength(1);
|
|
83
|
+
expect(history[0].reason).toBe('test reason');
|
|
84
|
+
});
|
|
85
|
+
});
|
|
86
|
+
|
|
87
|
+
describe('getReflectionHistory', () => {
|
|
88
|
+
it('should return empty array initially', () => {
|
|
89
|
+
const manager = new ReflectionManager({ enabled: true, maxAttempts: 3 });
|
|
90
|
+
expect(manager.getReflectionHistory()).toEqual([]);
|
|
91
|
+
});
|
|
92
|
+
|
|
93
|
+
it('should return copy of history (not reference)', () => {
|
|
94
|
+
const manager = new ReflectionManager({ enabled: true, maxAttempts: 3 });
|
|
95
|
+
const history1 = manager.getReflectionHistory();
|
|
96
|
+
const history2 = manager.getReflectionHistory();
|
|
97
|
+
|
|
98
|
+
expect(history1).not.toBe(history2);
|
|
99
|
+
});
|
|
100
|
+
});
|
|
101
|
+
|
|
102
|
+
describe('reflect (heuristic mode)', () => {
|
|
103
|
+
it('should not retry rate limit errors', async () => {
|
|
104
|
+
const manager = new ReflectionManager({ enabled: true, maxAttempts: 3 });
|
|
105
|
+
const context = createContext(new Error('rate limit exceeded'));
|
|
106
|
+
|
|
107
|
+
const result = await manager.reflect(context);
|
|
108
|
+
|
|
109
|
+
expect(result.shouldRetry).toBe(false);
|
|
110
|
+
expect(result.reason).toContain('Non-retryable error');
|
|
111
|
+
});
|
|
112
|
+
|
|
113
|
+
it('should not retry authentication errors', async () => {
|
|
114
|
+
const manager = new ReflectionManager({ enabled: true, maxAttempts: 3 });
|
|
115
|
+
const context = createContext(new Error('authentication failed'));
|
|
116
|
+
|
|
117
|
+
const result = await manager.reflect(context);
|
|
118
|
+
|
|
119
|
+
expect(result.shouldRetry).toBe(false);
|
|
120
|
+
});
|
|
121
|
+
|
|
122
|
+
it('should retry validation errors', async () => {
|
|
123
|
+
const manager = new ReflectionManager({ enabled: true, maxAttempts: 3 });
|
|
124
|
+
const context = createContext(new Error('validation failed: missing field'));
|
|
125
|
+
|
|
126
|
+
const result = await manager.reflect(context);
|
|
127
|
+
|
|
128
|
+
expect(result.shouldRetry).toBe(true);
|
|
129
|
+
expect(result.reason).toContain('Validation/parsing error');
|
|
130
|
+
});
|
|
131
|
+
|
|
132
|
+
it('should retry JSON parse errors', async () => {
|
|
133
|
+
const manager = new ReflectionManager({ enabled: true, maxAttempts: 3 });
|
|
134
|
+
const context = createContext(new Error('JSON parse error at position 10'));
|
|
135
|
+
|
|
136
|
+
const result = await manager.reflect(context);
|
|
137
|
+
|
|
138
|
+
expect(result.shouldRetry).toBe(true);
|
|
139
|
+
});
|
|
140
|
+
|
|
141
|
+
it('should retry timeout errors (up to limit)', async () => {
|
|
142
|
+
const manager = new ReflectionManager({ enabled: true, maxAttempts: 3 });
|
|
143
|
+
|
|
144
|
+
const context1 = createContext(new Error('timeout'), 1);
|
|
145
|
+
const result1 = await manager.reflect(context1);
|
|
146
|
+
expect(result1.shouldRetry).toBe(true);
|
|
147
|
+
|
|
148
|
+
const context2 = createContext(new Error('timeout'), 2);
|
|
149
|
+
const result2 = await manager.reflect(context2);
|
|
150
|
+
expect(result2.shouldRetry).toBe(false);
|
|
151
|
+
});
|
|
152
|
+
|
|
153
|
+
it('should stop retrying after max attempts', async () => {
|
|
154
|
+
const manager = new ReflectionManager({ enabled: true, maxAttempts: 2 });
|
|
155
|
+
const context = createContext(new Error('generic error'), 2);
|
|
156
|
+
|
|
157
|
+
const result = await manager.reflect(context);
|
|
158
|
+
|
|
159
|
+
expect(result.shouldRetry).toBe(false);
|
|
160
|
+
expect(result.reason).toContain('Max attempts');
|
|
161
|
+
});
|
|
162
|
+
|
|
163
|
+
it('should record reflection in history', async () => {
|
|
164
|
+
const manager = new ReflectionManager({ enabled: true, maxAttempts: 3 });
|
|
165
|
+
const context = createContext(new Error('test error'));
|
|
166
|
+
|
|
167
|
+
await manager.reflect(context);
|
|
168
|
+
|
|
169
|
+
const history = manager.getReflectionHistory();
|
|
170
|
+
expect(history).toHaveLength(1);
|
|
171
|
+
expect(history[0].level).toBe('workflow');
|
|
172
|
+
expect(history[0].error.message).toBe('test error');
|
|
173
|
+
});
|
|
174
|
+
});
|
|
175
|
+
|
|
176
|
+
describe('markLastReflectionSuccessful', () => {
|
|
177
|
+
it('should mark last entry as successful', async () => {
|
|
178
|
+
const manager = new ReflectionManager({ enabled: true, maxAttempts: 3 });
|
|
179
|
+
const context = createContext(new Error('test error'));
|
|
180
|
+
|
|
181
|
+
await manager.reflect(context);
|
|
182
|
+
manager.markLastReflectionSuccessful();
|
|
183
|
+
|
|
184
|
+
const history = manager.getReflectionHistory();
|
|
185
|
+
expect(history[0].success).toBe(true);
|
|
186
|
+
});
|
|
187
|
+
|
|
188
|
+
it('should do nothing when history is empty', () => {
|
|
189
|
+
const manager = new ReflectionManager({ enabled: true, maxAttempts: 3 });
|
|
190
|
+
|
|
191
|
+
// Should not throw
|
|
192
|
+
expect(() => manager.markLastReflectionSuccessful()).not.toThrow();
|
|
193
|
+
});
|
|
194
|
+
});
|
|
195
|
+
|
|
196
|
+
describe('event emission', () => {
|
|
197
|
+
it('should emit reflectionStart and reflectionEnd events', async () => {
|
|
198
|
+
const events: { type: string }[] = [];
|
|
199
|
+
const manager = new ReflectionManager({ enabled: true, maxAttempts: 3 });
|
|
200
|
+
manager.setEventEmitter((event) => events.push(event));
|
|
201
|
+
|
|
202
|
+
const context = createContext(new Error('test error'));
|
|
203
|
+
await manager.reflect(context);
|
|
204
|
+
|
|
205
|
+
expect(events).toHaveLength(2);
|
|
206
|
+
expect(events[0].type).toBe('reflectionStart');
|
|
207
|
+
expect(events[1].type).toBe('reflectionEnd');
|
|
208
|
+
});
|
|
209
|
+
});
|
|
210
|
+
});
|
|
@@ -0,0 +1,85 @@
|
|
|
1
|
+
import { describe, it, expect } from 'vitest';
|
|
2
|
+
import { Workflow, WorkflowTreeDebugger } from '../../index.js';
|
|
3
|
+
|
|
4
|
+
class DebugTestWorkflow extends Workflow {
|
|
5
|
+
async run(): Promise<void> {
|
|
6
|
+
this.setStatus('completed');
|
|
7
|
+
}
|
|
8
|
+
}
|
|
9
|
+
|
|
10
|
+
describe('WorkflowTreeDebugger', () => {
|
|
11
|
+
it('should render tree string', () => {
|
|
12
|
+
const wf = new DebugTestWorkflow('Root');
|
|
13
|
+
const debugger_ = new WorkflowTreeDebugger(wf);
|
|
14
|
+
|
|
15
|
+
const tree = debugger_.toTreeString();
|
|
16
|
+
expect(tree).toContain('Root');
|
|
17
|
+
expect(tree).toContain('[idle]');
|
|
18
|
+
});
|
|
19
|
+
|
|
20
|
+
it('should show child nodes in tree', () => {
|
|
21
|
+
const parent = new DebugTestWorkflow('Parent');
|
|
22
|
+
const child1 = new DebugTestWorkflow('Child1', parent);
|
|
23
|
+
const child2 = new DebugTestWorkflow('Child2', parent);
|
|
24
|
+
|
|
25
|
+
const debugger_ = new WorkflowTreeDebugger(parent);
|
|
26
|
+
const tree = debugger_.toTreeString();
|
|
27
|
+
|
|
28
|
+
expect(tree).toContain('Parent');
|
|
29
|
+
expect(tree).toContain('Child1');
|
|
30
|
+
expect(tree).toContain('Child2');
|
|
31
|
+
expect(tree).toContain('├──');
|
|
32
|
+
expect(tree).toContain('└──');
|
|
33
|
+
});
|
|
34
|
+
|
|
35
|
+
it('should find node by ID', () => {
|
|
36
|
+
const parent = new DebugTestWorkflow('Parent');
|
|
37
|
+
const child = new DebugTestWorkflow('Child', parent);
|
|
38
|
+
|
|
39
|
+
const debugger_ = new WorkflowTreeDebugger(parent);
|
|
40
|
+
|
|
41
|
+
expect(debugger_.getNode(parent.id)).toBe(parent.getNode());
|
|
42
|
+
expect(debugger_.getNode(child.id)).toBe(child.getNode());
|
|
43
|
+
expect(debugger_.getNode('nonexistent')).toBeUndefined();
|
|
44
|
+
});
|
|
45
|
+
|
|
46
|
+
it('should collect logs from all nodes', async () => {
|
|
47
|
+
const parent = new DebugTestWorkflow('Parent');
|
|
48
|
+
const child = new DebugTestWorkflow('Child', parent);
|
|
49
|
+
|
|
50
|
+
const debugger_ = new WorkflowTreeDebugger(parent);
|
|
51
|
+
|
|
52
|
+
// Add some logs manually
|
|
53
|
+
parent.getNode().logs.push({
|
|
54
|
+
id: '1',
|
|
55
|
+
workflowId: parent.id,
|
|
56
|
+
timestamp: Date.now(),
|
|
57
|
+
level: 'info',
|
|
58
|
+
message: 'Parent log',
|
|
59
|
+
});
|
|
60
|
+
|
|
61
|
+
child.getNode().logs.push({
|
|
62
|
+
id: '2',
|
|
63
|
+
workflowId: child.id,
|
|
64
|
+
timestamp: Date.now(),
|
|
65
|
+
level: 'info',
|
|
66
|
+
message: 'Child log',
|
|
67
|
+
});
|
|
68
|
+
|
|
69
|
+
const logString = debugger_.toLogString();
|
|
70
|
+
expect(logString).toContain('Parent log');
|
|
71
|
+
expect(logString).toContain('Child log');
|
|
72
|
+
});
|
|
73
|
+
|
|
74
|
+
it('should return stats', () => {
|
|
75
|
+
const parent = new DebugTestWorkflow('Parent');
|
|
76
|
+
new DebugTestWorkflow('Child1', parent);
|
|
77
|
+
new DebugTestWorkflow('Child2', parent);
|
|
78
|
+
|
|
79
|
+
const debugger_ = new WorkflowTreeDebugger(parent);
|
|
80
|
+
const stats = debugger_.getStats();
|
|
81
|
+
|
|
82
|
+
expect(stats.totalNodes).toBe(3);
|
|
83
|
+
expect(stats.byStatus.idle).toBe(3);
|
|
84
|
+
});
|
|
85
|
+
});
|
|
@@ -0,0 +1,81 @@
|
|
|
1
|
+
import { describe, it, expect } from 'vitest';
|
|
2
|
+
import { Workflow, WorkflowObserver, WorkflowNode, LogEntry, WorkflowEvent } from '../../index.js';
|
|
3
|
+
|
|
4
|
+
class SimpleWorkflow extends Workflow {
|
|
5
|
+
async run(): Promise<string> {
|
|
6
|
+
this.setStatus('running');
|
|
7
|
+
this.logger.info('Running simple workflow');
|
|
8
|
+
this.setStatus('completed');
|
|
9
|
+
return 'done';
|
|
10
|
+
}
|
|
11
|
+
}
|
|
12
|
+
|
|
13
|
+
describe('Workflow', () => {
|
|
14
|
+
it('should create with unique id', () => {
|
|
15
|
+
const wf1 = new SimpleWorkflow();
|
|
16
|
+
const wf2 = new SimpleWorkflow();
|
|
17
|
+
expect(wf1.id).not.toBe(wf2.id);
|
|
18
|
+
});
|
|
19
|
+
|
|
20
|
+
it('should use class name as default name', () => {
|
|
21
|
+
const wf = new SimpleWorkflow();
|
|
22
|
+
expect(wf.getNode().name).toBe('SimpleWorkflow');
|
|
23
|
+
});
|
|
24
|
+
|
|
25
|
+
it('should use custom name when provided', () => {
|
|
26
|
+
const wf = new SimpleWorkflow('CustomName');
|
|
27
|
+
expect(wf.getNode().name).toBe('CustomName');
|
|
28
|
+
});
|
|
29
|
+
|
|
30
|
+
it('should start with idle status', () => {
|
|
31
|
+
const wf = new SimpleWorkflow();
|
|
32
|
+
expect(wf.status).toBe('idle');
|
|
33
|
+
expect(wf.getNode().status).toBe('idle');
|
|
34
|
+
});
|
|
35
|
+
|
|
36
|
+
it('should attach child to parent', () => {
|
|
37
|
+
const parent = new SimpleWorkflow('Parent');
|
|
38
|
+
const child = new SimpleWorkflow('Child', parent);
|
|
39
|
+
|
|
40
|
+
expect(child.parent).toBe(parent);
|
|
41
|
+
expect(parent.children).toContain(child);
|
|
42
|
+
expect(parent.getNode().children).toContain(child.getNode());
|
|
43
|
+
});
|
|
44
|
+
|
|
45
|
+
it('should emit logs to observers', async () => {
|
|
46
|
+
const wf = new SimpleWorkflow();
|
|
47
|
+
const logs: LogEntry[] = [];
|
|
48
|
+
|
|
49
|
+
const observer: WorkflowObserver = {
|
|
50
|
+
onLog: (entry) => logs.push(entry),
|
|
51
|
+
onEvent: () => {},
|
|
52
|
+
onStateUpdated: () => {},
|
|
53
|
+
onTreeChanged: () => {},
|
|
54
|
+
};
|
|
55
|
+
|
|
56
|
+
wf.addObserver(observer);
|
|
57
|
+
await wf.run();
|
|
58
|
+
|
|
59
|
+
expect(logs.length).toBeGreaterThan(0);
|
|
60
|
+
expect(logs[0].message).toBe('Running simple workflow');
|
|
61
|
+
});
|
|
62
|
+
|
|
63
|
+
it('should emit childAttached event', () => {
|
|
64
|
+
const parent = new SimpleWorkflow('Parent');
|
|
65
|
+
const events: WorkflowEvent[] = [];
|
|
66
|
+
|
|
67
|
+
const observer: WorkflowObserver = {
|
|
68
|
+
onLog: () => {},
|
|
69
|
+
onEvent: (event) => events.push(event),
|
|
70
|
+
onStateUpdated: () => {},
|
|
71
|
+
onTreeChanged: () => {},
|
|
72
|
+
};
|
|
73
|
+
|
|
74
|
+
parent.addObserver(observer);
|
|
75
|
+
const child = new SimpleWorkflow('Child', parent);
|
|
76
|
+
|
|
77
|
+
const attachEvent = events.find((e) => e.type === 'childAttached');
|
|
78
|
+
expect(attachEvent).toBeDefined();
|
|
79
|
+
expect(attachEvent?.type === 'childAttached' && attachEvent.parentId).toBe(parent.id);
|
|
80
|
+
});
|
|
81
|
+
});
|
|
@@ -0,0 +1,244 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Cache Key Generation - Deterministic key generation for LLM response caching
|
|
3
|
+
*
|
|
4
|
+
* Uses SHA-256 hashing with deterministic JSON serialization to ensure
|
|
5
|
+
* identical inputs always produce identical cache keys.
|
|
6
|
+
*/
|
|
7
|
+
|
|
8
|
+
import { createHash } from 'node:crypto';
|
|
9
|
+
import type { Tool, MCPServer, Skill } from '../types/index.js';
|
|
10
|
+
import type { z } from 'zod';
|
|
11
|
+
|
|
12
|
+
/**
|
|
13
|
+
* Inputs used to generate a cache key
|
|
14
|
+
*/
|
|
15
|
+
export interface CacheKeyInputs {
|
|
16
|
+
/** User message content */
|
|
17
|
+
user: string;
|
|
18
|
+
/** Data passed to the prompt */
|
|
19
|
+
data?: Record<string, unknown>;
|
|
20
|
+
/** System prompt */
|
|
21
|
+
system?: string;
|
|
22
|
+
/** Model identifier */
|
|
23
|
+
model: string;
|
|
24
|
+
/** Temperature setting */
|
|
25
|
+
temperature?: number;
|
|
26
|
+
/** Maximum tokens */
|
|
27
|
+
maxTokens?: number;
|
|
28
|
+
/** Tools available to the agent */
|
|
29
|
+
tools?: Tool[];
|
|
30
|
+
/** MCP servers */
|
|
31
|
+
mcps?: MCPServer[];
|
|
32
|
+
/** Skills */
|
|
33
|
+
skills?: Skill[];
|
|
34
|
+
/** Zod response format schema */
|
|
35
|
+
responseFormat?: z.ZodType;
|
|
36
|
+
}
|
|
37
|
+
|
|
38
|
+
/**
|
|
39
|
+
* Deterministically stringify a value with sorted object keys
|
|
40
|
+
*
|
|
41
|
+
* Unlike JSON.stringify, this function guarantees consistent output
|
|
42
|
+
* regardless of key insertion order by sorting all object keys.
|
|
43
|
+
*
|
|
44
|
+
* @param value - Value to stringify
|
|
45
|
+
* @returns Deterministic JSON string
|
|
46
|
+
* @throws TypeError if circular reference detected
|
|
47
|
+
*/
|
|
48
|
+
export function deterministicStringify(value: unknown): string {
|
|
49
|
+
const seen = new WeakSet<object>();
|
|
50
|
+
|
|
51
|
+
function stringify(val: unknown): string {
|
|
52
|
+
if (val === null) return 'null';
|
|
53
|
+
if (val === undefined) return 'undefined';
|
|
54
|
+
if (typeof val === 'string') return JSON.stringify(val);
|
|
55
|
+
if (typeof val === 'number') {
|
|
56
|
+
if (Number.isNaN(val)) return 'NaN';
|
|
57
|
+
if (!Number.isFinite(val)) return val > 0 ? 'Infinity' : '-Infinity';
|
|
58
|
+
return String(val);
|
|
59
|
+
}
|
|
60
|
+
if (typeof val === 'boolean') return String(val);
|
|
61
|
+
if (typeof val === 'function') return 'function';
|
|
62
|
+
if (typeof val === 'symbol') return 'symbol';
|
|
63
|
+
if (typeof val === 'bigint') return `${val}n`;
|
|
64
|
+
|
|
65
|
+
if (typeof val === 'object') {
|
|
66
|
+
if (seen.has(val as object)) {
|
|
67
|
+
throw new TypeError('Converting circular structure to JSON');
|
|
68
|
+
}
|
|
69
|
+
seen.add(val as object);
|
|
70
|
+
|
|
71
|
+
let result: string;
|
|
72
|
+
|
|
73
|
+
if (Array.isArray(val)) {
|
|
74
|
+
result = '[' + val.map(stringify).join(',') + ']';
|
|
75
|
+
} else if (val instanceof Date) {
|
|
76
|
+
result = JSON.stringify(val.toISOString());
|
|
77
|
+
} else if (val instanceof Map) {
|
|
78
|
+
const entries = Array.from(val.entries())
|
|
79
|
+
.sort(([a], [b]) => String(a).localeCompare(String(b)))
|
|
80
|
+
.map(([k, v]) => `[${stringify(k)},${stringify(v)}]`);
|
|
81
|
+
result = `Map{${entries.join(',')}}`;
|
|
82
|
+
} else if (val instanceof Set) {
|
|
83
|
+
const values = Array.from(val)
|
|
84
|
+
.map(stringify)
|
|
85
|
+
.sort();
|
|
86
|
+
result = `Set{${values.join(',')}}`;
|
|
87
|
+
} else {
|
|
88
|
+
// Regular object - sort keys
|
|
89
|
+
const keys = Object.keys(val as Record<string, unknown>).sort();
|
|
90
|
+
const pairs = keys.map(
|
|
91
|
+
(k) =>
|
|
92
|
+
JSON.stringify(k) +
|
|
93
|
+
':' +
|
|
94
|
+
stringify((val as Record<string, unknown>)[k])
|
|
95
|
+
);
|
|
96
|
+
result = '{' + pairs.join(',') + '}';
|
|
97
|
+
}
|
|
98
|
+
|
|
99
|
+
seen.delete(val as object);
|
|
100
|
+
return result;
|
|
101
|
+
}
|
|
102
|
+
|
|
103
|
+
return String(val);
|
|
104
|
+
}
|
|
105
|
+
|
|
106
|
+
return stringify(value);
|
|
107
|
+
}
|
|
108
|
+
|
|
109
|
+
/**
|
|
110
|
+
* Generate a hash from a Zod schema's internal definition
|
|
111
|
+
*
|
|
112
|
+
* Zod schemas are functions that cannot be directly serialized.
|
|
113
|
+
* This extracts the schema's _def property which contains the
|
|
114
|
+
* schema structure and hashes it.
|
|
115
|
+
*
|
|
116
|
+
* @param schema - Zod schema to hash
|
|
117
|
+
* @returns SHA-256 hex digest of the schema definition
|
|
118
|
+
*/
|
|
119
|
+
export function getSchemaHash(schema: z.ZodType | undefined): string {
|
|
120
|
+
if (!schema) {
|
|
121
|
+
return 'no-schema';
|
|
122
|
+
}
|
|
123
|
+
|
|
124
|
+
try {
|
|
125
|
+
// Access Zod's internal _def property
|
|
126
|
+
const def = (schema as { _def?: unknown })._def;
|
|
127
|
+
if (!def) {
|
|
128
|
+
return 'unknown-schema';
|
|
129
|
+
}
|
|
130
|
+
|
|
131
|
+
// Create a simplified representation of the schema
|
|
132
|
+
const schemaRep = extractSchemaStructure(def);
|
|
133
|
+
const serialized = deterministicStringify(schemaRep);
|
|
134
|
+
|
|
135
|
+
return createHash('sha256').update(serialized, 'utf8').digest('hex');
|
|
136
|
+
} catch {
|
|
137
|
+
// Fallback for schemas that can't be introspected
|
|
138
|
+
return 'fallback-schema';
|
|
139
|
+
}
|
|
140
|
+
}
|
|
141
|
+
|
|
142
|
+
/**
|
|
143
|
+
* Extract a serializable structure from a Zod schema definition
|
|
144
|
+
*/
|
|
145
|
+
function extractSchemaStructure(def: unknown): unknown {
|
|
146
|
+
if (def === null || def === undefined) {
|
|
147
|
+
return def;
|
|
148
|
+
}
|
|
149
|
+
|
|
150
|
+
if (typeof def !== 'object') {
|
|
151
|
+
return def;
|
|
152
|
+
}
|
|
153
|
+
|
|
154
|
+
const typedDef = def as Record<string, unknown>;
|
|
155
|
+
const result: Record<string, unknown> = {};
|
|
156
|
+
|
|
157
|
+
// Extract key properties that define schema structure
|
|
158
|
+
if ('typeName' in typedDef) {
|
|
159
|
+
result.typeName = typedDef.typeName;
|
|
160
|
+
}
|
|
161
|
+
|
|
162
|
+
if ('shape' in typedDef && typeof typedDef.shape === 'function') {
|
|
163
|
+
// Object schema - extract shape
|
|
164
|
+
try {
|
|
165
|
+
const shape = typedDef.shape() as Record<string, { _def?: unknown }>;
|
|
166
|
+
result.shape = Object.fromEntries(
|
|
167
|
+
Object.entries(shape).map(([key, value]) => [
|
|
168
|
+
key,
|
|
169
|
+
extractSchemaStructure(value?._def),
|
|
170
|
+
])
|
|
171
|
+
);
|
|
172
|
+
} catch {
|
|
173
|
+
result.shape = 'complex-shape';
|
|
174
|
+
}
|
|
175
|
+
}
|
|
176
|
+
|
|
177
|
+
if ('type' in typedDef && typedDef.type && typeof typedDef.type === 'object') {
|
|
178
|
+
result.type = extractSchemaStructure((typedDef.type as { _def?: unknown })._def);
|
|
179
|
+
}
|
|
180
|
+
|
|
181
|
+
if ('values' in typedDef) {
|
|
182
|
+
result.values = typedDef.values;
|
|
183
|
+
}
|
|
184
|
+
|
|
185
|
+
if ('checks' in typedDef && Array.isArray(typedDef.checks)) {
|
|
186
|
+
result.checks = typedDef.checks.map((check: { kind?: string }) => check.kind);
|
|
187
|
+
}
|
|
188
|
+
|
|
189
|
+
return result;
|
|
190
|
+
}
|
|
191
|
+
|
|
192
|
+
/**
|
|
193
|
+
* Generate a deterministic cache key from prompt execution inputs
|
|
194
|
+
*
|
|
195
|
+
* The key is a 64-character SHA-256 hex digest that uniquely identifies
|
|
196
|
+
* a specific combination of prompt, data, configuration, and tools.
|
|
197
|
+
*
|
|
198
|
+
* @param inputs - Cache key inputs
|
|
199
|
+
* @returns 64-character hex SHA-256 digest
|
|
200
|
+
*/
|
|
201
|
+
export function generateCacheKey(inputs: CacheKeyInputs): string {
|
|
202
|
+
// Build normalized representation with sorted arrays
|
|
203
|
+
const normalized: Record<string, unknown> = {
|
|
204
|
+
user: inputs.user,
|
|
205
|
+
model: inputs.model,
|
|
206
|
+
};
|
|
207
|
+
|
|
208
|
+
// Include optional fields only if defined
|
|
209
|
+
if (inputs.data !== undefined) {
|
|
210
|
+
normalized.data = inputs.data;
|
|
211
|
+
}
|
|
212
|
+
|
|
213
|
+
if (inputs.system !== undefined) {
|
|
214
|
+
normalized.system = inputs.system;
|
|
215
|
+
}
|
|
216
|
+
|
|
217
|
+
if (inputs.temperature !== undefined) {
|
|
218
|
+
normalized.temperature = inputs.temperature;
|
|
219
|
+
}
|
|
220
|
+
|
|
221
|
+
if (inputs.maxTokens !== undefined) {
|
|
222
|
+
normalized.maxTokens = inputs.maxTokens;
|
|
223
|
+
}
|
|
224
|
+
|
|
225
|
+
// Sort tool/mcp/skill names for determinism
|
|
226
|
+
if (inputs.tools && inputs.tools.length > 0) {
|
|
227
|
+
normalized.tools = inputs.tools.map((t) => t.name).sort();
|
|
228
|
+
}
|
|
229
|
+
|
|
230
|
+
if (inputs.mcps && inputs.mcps.length > 0) {
|
|
231
|
+
normalized.mcps = inputs.mcps.map((m) => m.name).sort();
|
|
232
|
+
}
|
|
233
|
+
|
|
234
|
+
if (inputs.skills && inputs.skills.length > 0) {
|
|
235
|
+
normalized.skills = inputs.skills.map((s) => s.name).sort();
|
|
236
|
+
}
|
|
237
|
+
|
|
238
|
+
// Include schema hash
|
|
239
|
+
normalized.schemaHash = getSchemaHash(inputs.responseFormat);
|
|
240
|
+
|
|
241
|
+
// Generate SHA-256 hash
|
|
242
|
+
const serialized = deterministicStringify(normalized);
|
|
243
|
+
return createHash('sha256').update(serialized, 'utf8').digest('hex');
|
|
244
|
+
}
|