cognitive-core 0.0.2 → 0.1.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +302 -116
- package/SKILL.md +193 -0
- package/dist/agents/index.d.ts +3 -0
- package/dist/agents/index.d.ts.map +1 -0
- package/dist/agents/index.js +5 -0
- package/dist/agents/index.js.map +1 -0
- package/dist/agents/mock-provider.d.ts +23 -0
- package/dist/agents/mock-provider.d.ts.map +1 -0
- package/dist/agents/mock-provider.js +71 -0
- package/dist/agents/mock-provider.js.map +1 -0
- package/dist/agents/types.d.ts +98 -0
- package/dist/agents/types.d.ts.map +1 -0
- package/dist/agents/types.js +44 -0
- package/dist/agents/types.js.map +1 -0
- package/dist/atlas.d.ts +196 -0
- package/dist/atlas.d.ts.map +1 -0
- package/dist/atlas.js +373 -0
- package/dist/atlas.js.map +1 -0
- package/dist/bin/cognitive-core.d.ts +18 -0
- package/dist/bin/cognitive-core.d.ts.map +1 -0
- package/dist/bin/cognitive-core.js +419 -0
- package/dist/bin/cognitive-core.js.map +1 -0
- package/dist/embeddings/bm25.d.ts +104 -0
- package/dist/embeddings/bm25.d.ts.map +1 -0
- package/dist/embeddings/bm25.js +264 -0
- package/dist/embeddings/bm25.js.map +1 -0
- package/dist/embeddings/index.d.ts +12 -0
- package/dist/embeddings/index.d.ts.map +1 -0
- package/dist/embeddings/index.js +16 -0
- package/dist/embeddings/index.js.map +1 -0
- package/dist/embeddings/manager.d.ts +112 -0
- package/dist/embeddings/manager.d.ts.map +1 -0
- package/dist/embeddings/manager.js +215 -0
- package/dist/embeddings/manager.js.map +1 -0
- package/dist/embeddings/provider.d.ts +101 -0
- package/dist/embeddings/provider.d.ts.map +1 -0
- package/dist/embeddings/provider.js +232 -0
- package/dist/embeddings/provider.js.map +1 -0
- package/dist/embeddings/vector-store.d.ts +101 -0
- package/dist/embeddings/vector-store.d.ts.map +1 -0
- package/dist/embeddings/vector-store.js +256 -0
- package/dist/embeddings/vector-store.js.map +1 -0
- package/dist/factory.d.ts +193 -0
- package/dist/factory.d.ts.map +1 -0
- package/dist/factory.js +109 -0
- package/dist/factory.js.map +1 -0
- package/dist/index.d.ts +30 -453
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +84 -509
- package/dist/index.js.map +1 -0
- package/dist/learning/analyzer.d.ts +110 -0
- package/dist/learning/analyzer.d.ts.map +1 -0
- package/dist/learning/analyzer.js +213 -0
- package/dist/learning/analyzer.js.map +1 -0
- package/dist/learning/effectiveness.d.ts +158 -0
- package/dist/learning/effectiveness.d.ts.map +1 -0
- package/dist/learning/effectiveness.js +251 -0
- package/dist/learning/effectiveness.js.map +1 -0
- package/dist/learning/index.d.ts +8 -0
- package/dist/learning/index.d.ts.map +1 -0
- package/dist/learning/index.js +11 -0
- package/dist/learning/index.js.map +1 -0
- package/dist/learning/llm-extractor.d.ts +88 -0
- package/dist/learning/llm-extractor.d.ts.map +1 -0
- package/dist/learning/llm-extractor.js +372 -0
- package/dist/learning/llm-extractor.js.map +1 -0
- package/dist/learning/meta-learner.d.ts +80 -0
- package/dist/learning/meta-learner.d.ts.map +1 -0
- package/dist/learning/meta-learner.js +355 -0
- package/dist/learning/meta-learner.js.map +1 -0
- package/dist/learning/pipeline.d.ts +65 -0
- package/dist/learning/pipeline.d.ts.map +1 -0
- package/dist/learning/pipeline.js +170 -0
- package/dist/learning/pipeline.js.map +1 -0
- package/dist/learning/playbook-extractor.d.ts +113 -0
- package/dist/learning/playbook-extractor.d.ts.map +1 -0
- package/dist/learning/playbook-extractor.js +523 -0
- package/dist/learning/playbook-extractor.js.map +1 -0
- package/dist/learning/usage-inference.d.ts +82 -0
- package/dist/learning/usage-inference.d.ts.map +1 -0
- package/dist/learning/usage-inference.js +261 -0
- package/dist/learning/usage-inference.js.map +1 -0
- package/dist/mcp/index.d.ts +6 -0
- package/dist/mcp/index.d.ts.map +1 -0
- package/dist/mcp/index.js +6 -0
- package/dist/mcp/index.js.map +1 -0
- package/dist/mcp/playbook-server.d.ts +120 -0
- package/dist/mcp/playbook-server.d.ts.map +1 -0
- package/dist/mcp/playbook-server.js +427 -0
- package/dist/mcp/playbook-server.js.map +1 -0
- package/dist/memory/curated-loader.d.ts +62 -0
- package/dist/memory/curated-loader.d.ts.map +1 -0
- package/dist/memory/curated-loader.js +106 -0
- package/dist/memory/curated-loader.js.map +1 -0
- package/dist/memory/experience.d.ts +122 -0
- package/dist/memory/experience.d.ts.map +1 -0
- package/dist/memory/experience.js +392 -0
- package/dist/memory/experience.js.map +1 -0
- package/dist/memory/index.d.ts +6 -0
- package/dist/memory/index.d.ts.map +1 -0
- package/dist/memory/index.js +9 -0
- package/dist/memory/index.js.map +1 -0
- package/dist/memory/meta.d.ts +90 -0
- package/dist/memory/meta.d.ts.map +1 -0
- package/dist/memory/meta.js +362 -0
- package/dist/memory/meta.js.map +1 -0
- package/dist/memory/playbook.d.ts +133 -0
- package/dist/memory/playbook.d.ts.map +1 -0
- package/dist/memory/playbook.js +357 -0
- package/dist/memory/playbook.js.map +1 -0
- package/dist/memory/system.d.ts +167 -0
- package/dist/memory/system.d.ts.map +1 -0
- package/dist/memory/system.js +383 -0
- package/dist/memory/system.js.map +1 -0
- package/dist/runtime/backends/acp.d.ts +67 -0
- package/dist/runtime/backends/acp.d.ts.map +1 -0
- package/dist/runtime/backends/acp.js +290 -0
- package/dist/runtime/backends/acp.js.map +1 -0
- package/dist/runtime/backends/index.d.ts +5 -0
- package/dist/runtime/backends/index.d.ts.map +1 -0
- package/dist/runtime/backends/index.js +6 -0
- package/dist/runtime/backends/index.js.map +1 -0
- package/dist/runtime/backends/mock.d.ts +67 -0
- package/dist/runtime/backends/mock.d.ts.map +1 -0
- package/dist/runtime/backends/mock.js +153 -0
- package/dist/runtime/backends/mock.js.map +1 -0
- package/dist/runtime/backends/subprocess.d.ts +56 -0
- package/dist/runtime/backends/subprocess.d.ts.map +1 -0
- package/dist/runtime/backends/subprocess.js +260 -0
- package/dist/runtime/backends/subprocess.js.map +1 -0
- package/dist/runtime/flows/learning.d.ts +73 -0
- package/dist/runtime/flows/learning.d.ts.map +1 -0
- package/dist/runtime/flows/learning.js +116 -0
- package/dist/runtime/flows/learning.js.map +1 -0
- package/dist/runtime/flows/validation.d.ts +122 -0
- package/dist/runtime/flows/validation.d.ts.map +1 -0
- package/dist/runtime/flows/validation.js +223 -0
- package/dist/runtime/flows/validation.js.map +1 -0
- package/dist/runtime/index.d.ts +6 -0
- package/dist/runtime/index.d.ts.map +1 -0
- package/dist/runtime/index.js +8 -0
- package/dist/runtime/index.js.map +1 -0
- package/dist/runtime/manager.d.ts +116 -0
- package/dist/runtime/manager.d.ts.map +1 -0
- package/dist/runtime/manager.js +416 -0
- package/dist/runtime/manager.js.map +1 -0
- package/dist/runtime/types.d.ts +138 -0
- package/dist/runtime/types.d.ts.map +1 -0
- package/dist/runtime/types.js +2 -0
- package/dist/runtime/types.js.map +1 -0
- package/dist/search/evaluator.d.ts +102 -0
- package/dist/search/evaluator.d.ts.map +1 -0
- package/dist/search/evaluator.js +352 -0
- package/dist/search/evaluator.js.map +1 -0
- package/dist/search/index.d.ts +7 -0
- package/dist/search/index.d.ts.map +1 -0
- package/dist/search/index.js +11 -0
- package/dist/search/index.js.map +1 -0
- package/dist/search/refinement-loop.d.ts +73 -0
- package/dist/search/refinement-loop.d.ts.map +1 -0
- package/dist/search/refinement-loop.js +245 -0
- package/dist/search/refinement-loop.js.map +1 -0
- package/dist/search/refinement-types.d.ts +154 -0
- package/dist/search/refinement-types.d.ts.map +1 -0
- package/dist/search/refinement-types.js +99 -0
- package/dist/search/refinement-types.js.map +1 -0
- package/dist/search/router.d.ts +61 -0
- package/dist/search/router.d.ts.map +1 -0
- package/dist/search/router.js +197 -0
- package/dist/search/router.js.map +1 -0
- package/dist/search/solver.d.ts +75 -0
- package/dist/search/solver.d.ts.map +1 -0
- package/dist/search/solver.js +216 -0
- package/dist/search/solver.js.map +1 -0
- package/dist/search/verification-runner.d.ts +125 -0
- package/dist/search/verification-runner.d.ts.map +1 -0
- package/dist/search/verification-runner.js +440 -0
- package/dist/search/verification-runner.js.map +1 -0
- package/dist/surfacing/index.d.ts +2 -0
- package/dist/surfacing/index.d.ts.map +1 -0
- package/dist/surfacing/index.js +2 -0
- package/dist/surfacing/index.js.map +1 -0
- package/dist/surfacing/skill-library.d.ts +158 -0
- package/dist/surfacing/skill-library.d.ts.map +1 -0
- package/dist/surfacing/skill-library.js +429 -0
- package/dist/surfacing/skill-library.js.map +1 -0
- package/dist/types/config.d.ts +1113 -0
- package/dist/types/config.d.ts.map +1 -0
- package/dist/types/config.js +274 -0
- package/dist/types/config.js.map +1 -0
- package/dist/types/index.d.ts +9 -0
- package/dist/types/index.d.ts.map +1 -0
- package/dist/types/index.js +14 -0
- package/dist/types/index.js.map +1 -0
- package/dist/types/memory.d.ts +339 -0
- package/dist/types/memory.d.ts.map +1 -0
- package/dist/types/memory.js +207 -0
- package/dist/types/memory.js.map +1 -0
- package/dist/types/meta.d.ts +146 -0
- package/dist/types/meta.d.ts.map +1 -0
- package/dist/types/meta.js +51 -0
- package/dist/types/meta.js.map +1 -0
- package/dist/types/outcome.d.ts +42 -0
- package/dist/types/outcome.d.ts.map +1 -0
- package/dist/types/outcome.js +50 -0
- package/dist/types/outcome.js.map +1 -0
- package/dist/types/playbook.d.ts +119 -0
- package/dist/types/playbook.d.ts.map +1 -0
- package/dist/types/playbook.js +71 -0
- package/dist/types/playbook.js.map +1 -0
- package/dist/types/step.d.ts +44 -0
- package/dist/types/step.d.ts.map +1 -0
- package/dist/types/step.js +32 -0
- package/dist/types/step.js.map +1 -0
- package/dist/types/task.d.ts +91 -0
- package/dist/types/task.d.ts.map +1 -0
- package/dist/types/task.js +39 -0
- package/dist/types/task.js.map +1 -0
- package/dist/types/trajectory.d.ts +221 -0
- package/dist/types/trajectory.d.ts.map +1 -0
- package/dist/types/trajectory.js +60 -0
- package/dist/types/trajectory.js.map +1 -0
- package/dist/utils/index.d.ts +4 -0
- package/dist/utils/index.d.ts.map +1 -0
- package/dist/utils/index.js +4 -0
- package/dist/utils/index.js.map +1 -0
- package/dist/utils/similarity.d.ts +31 -0
- package/dist/utils/similarity.d.ts.map +1 -0
- package/dist/utils/similarity.js +107 -0
- package/dist/utils/similarity.js.map +1 -0
- package/dist/utils/storage.d.ts +106 -0
- package/dist/utils/storage.d.ts.map +1 -0
- package/dist/utils/storage.js +203 -0
- package/dist/utils/storage.js.map +1 -0
- package/dist/utils/validation.d.ts +129 -0
- package/dist/utils/validation.d.ts.map +1 -0
- package/dist/utils/validation.js +171 -0
- package/dist/utils/validation.js.map +1 -0
- package/package.json +50 -34
- package/scripts/migrate-to-playbooks.ts +307 -0
- package/src/agents/index.ts +14 -0
- package/src/agents/mock-provider.ts +93 -0
- package/src/agents/types.ts +137 -0
- package/src/atlas.ts +560 -0
- package/src/bin/cognitive-core.ts +470 -0
- package/src/embeddings/bm25.ts +337 -0
- package/src/embeddings/index.ts +39 -0
- package/src/embeddings/manager.ts +288 -0
- package/src/embeddings/provider.ts +311 -0
- package/src/embeddings/vector-store.ts +353 -0
- package/src/factory.ts +263 -0
- package/src/index.ts +246 -0
- package/src/learning/analyzer.ts +335 -0
- package/src/learning/effectiveness.ts +428 -0
- package/src/learning/index.ts +58 -0
- package/src/learning/llm-extractor.ts +542 -0
- package/src/learning/meta-learner.ts +516 -0
- package/src/learning/pipeline.ts +244 -0
- package/src/learning/playbook-extractor.ts +702 -0
- package/src/learning/usage-inference.ts +372 -0
- package/src/mcp/index.ts +12 -0
- package/src/mcp/playbook-server.ts +565 -0
- package/src/memory/curated-loader.ts +160 -0
- package/src/memory/experience.ts +515 -0
- package/src/memory/index.ts +27 -0
- package/src/memory/meta.ts +506 -0
- package/src/memory/playbook.ts +493 -0
- package/src/memory/system.ts +551 -0
- package/src/runtime/backends/acp.ts +378 -0
- package/src/runtime/backends/index.ts +24 -0
- package/src/runtime/backends/mock.ts +218 -0
- package/src/runtime/backends/subprocess.ts +356 -0
- package/src/runtime/flows/learning.ts +183 -0
- package/src/runtime/flows/validation.ts +381 -0
- package/src/runtime/index.ts +53 -0
- package/src/runtime/manager.ts +541 -0
- package/src/runtime/types.ts +157 -0
- package/src/search/evaluator.ts +474 -0
- package/src/search/index.ts +59 -0
- package/src/search/refinement-loop.ts +363 -0
- package/src/search/refinement-types.ts +159 -0
- package/src/search/router.ts +261 -0
- package/src/search/solver.ts +303 -0
- package/src/search/verification-runner.ts +570 -0
- package/src/surfacing/index.ts +6 -0
- package/src/surfacing/skill-library.ts +594 -0
- package/src/types/config.ts +333 -0
- package/src/types/index.ts +130 -0
- package/src/types/memory.ts +270 -0
- package/src/types/meta.ts +218 -0
- package/src/types/outcome.ts +66 -0
- package/src/types/playbook.ts +196 -0
- package/src/types/step.ts +40 -0
- package/src/types/task.ts +52 -0
- package/src/types/trajectory.ts +80 -0
- package/src/utils/index.ts +38 -0
- package/src/utils/similarity.ts +139 -0
- package/src/utils/storage.ts +249 -0
- package/src/utils/validation.ts +286 -0
- package/tests/embeddings/bm25.test.ts +130 -0
- package/tests/embeddings/manager.test.ts +205 -0
- package/tests/integration/atlas.test.ts +266 -0
- package/tests/integration/e2e.test.ts +929 -0
- package/tests/learning/analyzer.test.ts +426 -0
- package/tests/learning/effectiveness.test.ts +542 -0
- package/tests/learning/pipeline.test.ts +176 -0
- package/tests/learning/playbook-extractor-provenance.test.ts +114 -0
- package/tests/learning/usage-inference.test.ts +254 -0
- package/tests/mcp/playbook-server.test.ts +252 -0
- package/tests/memory/experience.test.ts +198 -0
- package/tests/memory/playbook.test.ts +338 -0
- package/tests/memory/provenance.test.ts +639 -0
- package/tests/memory/system.test.ts +325 -0
- package/tests/runtime/agent-manager.test.ts +512 -0
- package/tests/runtime/mock-backend.test.ts +248 -0
- package/tests/search/refinement-loop.test.ts +468 -0
- package/tests/search/refinement.test.ts +267 -0
- package/tests/search/router.test.ts +427 -0
- package/tests/surfacing/skill-library.test.ts +292 -0
- package/tests/types/outcome.test.ts +147 -0
- package/tests/types/step.test.ts +133 -0
- package/tests/types/task.test.ts +158 -0
- package/tests/types/trajectory.test.ts +253 -0
- package/tests/utils/similarity.test.ts +188 -0
- package/tests/utils/validation.test.ts +252 -0
- package/tsconfig.json +25 -0
- package/vitest.config.ts +22 -0
- package/dist/index.d.mts +0 -466
- package/dist/index.mjs +0 -478
|
@@ -0,0 +1,416 @@
|
|
|
1
|
+
import { createTrajectory, createStep, successOutcome, failureOutcome, } from '../types/index.js';
|
|
2
|
+
/**
|
|
3
|
+
* Default knowledge injector - formats memory for context injection
|
|
4
|
+
*/
|
|
5
|
+
export class DefaultKnowledgeInjector {
|
|
6
|
+
format(knowledge, _agentType, options) {
|
|
7
|
+
const format = options?.format ?? 'markdown';
|
|
8
|
+
if (format === 'json') {
|
|
9
|
+
return JSON.stringify(knowledge, null, 2);
|
|
10
|
+
}
|
|
11
|
+
if (format === 'xml') {
|
|
12
|
+
return this.formatXml(knowledge);
|
|
13
|
+
}
|
|
14
|
+
return this.formatMarkdown(knowledge);
|
|
15
|
+
}
|
|
16
|
+
formatMarkdown(knowledge) {
|
|
17
|
+
const sections = [];
|
|
18
|
+
if (knowledge.playbooks.length > 0) {
|
|
19
|
+
sections.push('## Applicable Playbooks\n');
|
|
20
|
+
for (const { playbook, score, matchType } of knowledge.playbooks) {
|
|
21
|
+
sections.push(`### ${playbook.name} (${(score * 100).toFixed(0)}% match via ${matchType})`);
|
|
22
|
+
sections.push(`**Strategy:** ${playbook.guidance.strategy}`);
|
|
23
|
+
if (playbook.guidance.tactics.length > 0) {
|
|
24
|
+
sections.push(`**Tactics:** ${playbook.guidance.tactics.join(', ')}`);
|
|
25
|
+
}
|
|
26
|
+
if (playbook.guidance.steps && playbook.guidance.steps.length > 0) {
|
|
27
|
+
sections.push('**Steps:**');
|
|
28
|
+
playbook.guidance.steps.forEach((step, i) => {
|
|
29
|
+
sections.push(`${i + 1}. ${step}`);
|
|
30
|
+
});
|
|
31
|
+
}
|
|
32
|
+
sections.push(`*Confidence: ${(playbook.confidence * 100).toFixed(0)}%*\n`);
|
|
33
|
+
}
|
|
34
|
+
}
|
|
35
|
+
if (knowledge.experiences.length > 0) {
|
|
36
|
+
sections.push('## Similar Past Experiences\n');
|
|
37
|
+
for (const { experience, score } of knowledge.experiences) {
|
|
38
|
+
sections.push(`### Experience (relevance: ${(score * 100).toFixed(0)}%)`);
|
|
39
|
+
sections.push(`**Task:** ${experience.taskInput.slice(0, 200)}...`);
|
|
40
|
+
sections.push(`**Outcome:** ${experience.success ? 'Success' : 'Failed'}`);
|
|
41
|
+
sections.push(`**Approach:** ${experience.solutionOutput.slice(0, 300)}...`);
|
|
42
|
+
sections.push('');
|
|
43
|
+
}
|
|
44
|
+
}
|
|
45
|
+
return sections.join('\n');
|
|
46
|
+
}
|
|
47
|
+
formatXml(knowledge) {
|
|
48
|
+
const parts = ['<atlas-knowledge>'];
|
|
49
|
+
if (knowledge.playbooks.length > 0) {
|
|
50
|
+
parts.push(' <playbooks>');
|
|
51
|
+
for (const { playbook, score } of knowledge.playbooks) {
|
|
52
|
+
parts.push(` <playbook name="${this.escapeXml(playbook.name)}" relevance="${score.toFixed(2)}">`);
|
|
53
|
+
parts.push(` <strategy>${this.escapeXml(playbook.guidance.strategy)}</strategy>`);
|
|
54
|
+
if (playbook.guidance.steps && playbook.guidance.steps.length > 0) {
|
|
55
|
+
parts.push(' <steps>');
|
|
56
|
+
playbook.guidance.steps.forEach((step, i) => {
|
|
57
|
+
parts.push(` <step n="${i + 1}">${this.escapeXml(step)}</step>`);
|
|
58
|
+
});
|
|
59
|
+
parts.push(' </steps>');
|
|
60
|
+
}
|
|
61
|
+
parts.push(` <confidence>${playbook.confidence}</confidence>`);
|
|
62
|
+
parts.push(' </playbook>');
|
|
63
|
+
}
|
|
64
|
+
parts.push(' </playbooks>');
|
|
65
|
+
}
|
|
66
|
+
if (knowledge.experiences.length > 0) {
|
|
67
|
+
parts.push(' <experiences>');
|
|
68
|
+
for (const { experience, score } of knowledge.experiences) {
|
|
69
|
+
parts.push(` <experience relevance="${score.toFixed(2)}">`);
|
|
70
|
+
parts.push(` <task>${this.escapeXml(experience.taskInput.slice(0, 200))}</task>`);
|
|
71
|
+
parts.push(` <outcome>${experience.success ? 'success' : 'failure'}</outcome>`);
|
|
72
|
+
parts.push(' </experience>');
|
|
73
|
+
}
|
|
74
|
+
parts.push(' </experiences>');
|
|
75
|
+
}
|
|
76
|
+
parts.push('</atlas-knowledge>');
|
|
77
|
+
return parts.join('\n');
|
|
78
|
+
}
|
|
79
|
+
escapeXml(str) {
|
|
80
|
+
return str
|
|
81
|
+
.replace(/&/g, '&')
|
|
82
|
+
.replace(/</g, '<')
|
|
83
|
+
.replace(/>/g, '>')
|
|
84
|
+
.replace(/"/g, '"');
|
|
85
|
+
}
|
|
86
|
+
}
|
|
87
|
+
/**
|
|
88
|
+
* Default trajectory extractor - converts agent sessions to trajectories
|
|
89
|
+
*/
|
|
90
|
+
export class DefaultTrajectoryExtractor {
|
|
91
|
+
extract(session, task) {
|
|
92
|
+
const steps = [];
|
|
93
|
+
// Convert tool calls to steps
|
|
94
|
+
for (const toolCall of session.toolCalls) {
|
|
95
|
+
steps.push(createStep({
|
|
96
|
+
thought: this.extractThought(session.messages, toolCall),
|
|
97
|
+
action: `${toolCall.name}: ${this.summarizeInput(toolCall.input)}`,
|
|
98
|
+
observation: toolCall.error
|
|
99
|
+
? `Error: ${toolCall.error}`
|
|
100
|
+
: this.summarizeOutput(toolCall.output),
|
|
101
|
+
metadata: {
|
|
102
|
+
toolCallId: toolCall.id,
|
|
103
|
+
toolName: toolCall.name,
|
|
104
|
+
duration: toolCall.endTime
|
|
105
|
+
? toolCall.endTime.getTime() - toolCall.startTime.getTime()
|
|
106
|
+
: undefined,
|
|
107
|
+
},
|
|
108
|
+
}));
|
|
109
|
+
}
|
|
110
|
+
// Determine outcome
|
|
111
|
+
const outcome = session.state === 'completed'
|
|
112
|
+
? successOutcome(session.result)
|
|
113
|
+
: failureOutcome(session.error ?? 'Session did not complete successfully');
|
|
114
|
+
return createTrajectory({
|
|
115
|
+
task,
|
|
116
|
+
steps,
|
|
117
|
+
outcome,
|
|
118
|
+
agentId: `${session.agentType}:${session.id}`,
|
|
119
|
+
wallTimeSeconds: session.endTime
|
|
120
|
+
? (session.endTime.getTime() - session.startTime.getTime()) / 1000
|
|
121
|
+
: 0,
|
|
122
|
+
metadata: {
|
|
123
|
+
sessionId: session.id,
|
|
124
|
+
agentType: session.agentType,
|
|
125
|
+
messageCount: session.messages.length,
|
|
126
|
+
...session.metadata,
|
|
127
|
+
},
|
|
128
|
+
});
|
|
129
|
+
}
|
|
130
|
+
extractThought(messages, toolCall) {
|
|
131
|
+
// Find the assistant message just before this tool call
|
|
132
|
+
const toolCallTime = toolCall.startTime.getTime();
|
|
133
|
+
const precedingMessages = messages.filter((m) => m.role === 'assistant' &&
|
|
134
|
+
m.timestamp.getTime() < toolCallTime &&
|
|
135
|
+
m.timestamp.getTime() > toolCallTime - 60000 // Within 1 minute
|
|
136
|
+
);
|
|
137
|
+
if (precedingMessages.length > 0) {
|
|
138
|
+
const lastMessage = precedingMessages[precedingMessages.length - 1];
|
|
139
|
+
// Extract reasoning/thought from the message
|
|
140
|
+
return lastMessage.content.slice(0, 500);
|
|
141
|
+
}
|
|
142
|
+
return undefined;
|
|
143
|
+
}
|
|
144
|
+
summarizeInput(input) {
|
|
145
|
+
if (typeof input === 'string') {
|
|
146
|
+
return input.slice(0, 100);
|
|
147
|
+
}
|
|
148
|
+
if (typeof input === 'object' && input !== null) {
|
|
149
|
+
const str = JSON.stringify(input);
|
|
150
|
+
return str.slice(0, 100) + (str.length > 100 ? '...' : '');
|
|
151
|
+
}
|
|
152
|
+
return String(input);
|
|
153
|
+
}
|
|
154
|
+
summarizeOutput(output) {
|
|
155
|
+
if (output === undefined)
|
|
156
|
+
return 'No output';
|
|
157
|
+
if (typeof output === 'string') {
|
|
158
|
+
return output.slice(0, 200) + (output.length > 200 ? '...' : '');
|
|
159
|
+
}
|
|
160
|
+
if (typeof output === 'object' && output !== null) {
|
|
161
|
+
const str = JSON.stringify(output);
|
|
162
|
+
return str.slice(0, 200) + (str.length > 200 ? '...' : '');
|
|
163
|
+
}
|
|
164
|
+
return String(output);
|
|
165
|
+
}
|
|
166
|
+
}
|
|
167
|
+
/**
|
|
168
|
+
* Agent Manager - orchestrates agent spawning, observation, and learning
|
|
169
|
+
*/
|
|
170
|
+
export class AgentManager {
|
|
171
|
+
backends = new Map();
|
|
172
|
+
memory;
|
|
173
|
+
injector;
|
|
174
|
+
extractor;
|
|
175
|
+
observers = [];
|
|
176
|
+
activeSessions = new Map();
|
|
177
|
+
constructor(memory, options) {
|
|
178
|
+
this.memory = memory;
|
|
179
|
+
this.injector = options?.injector ?? new DefaultKnowledgeInjector();
|
|
180
|
+
this.extractor = options?.extractor ?? new DefaultTrajectoryExtractor();
|
|
181
|
+
}
|
|
182
|
+
/**
|
|
183
|
+
* Register an agent backend
|
|
184
|
+
*/
|
|
185
|
+
registerBackend(backend) {
|
|
186
|
+
this.backends.set(backend.name, backend);
|
|
187
|
+
}
|
|
188
|
+
/**
|
|
189
|
+
* Get a registered backend
|
|
190
|
+
*/
|
|
191
|
+
getBackend(name) {
|
|
192
|
+
return this.backends.get(name);
|
|
193
|
+
}
|
|
194
|
+
/**
|
|
195
|
+
* List available backends
|
|
196
|
+
*/
|
|
197
|
+
listBackends() {
|
|
198
|
+
return Array.from(this.backends.keys());
|
|
199
|
+
}
|
|
200
|
+
/**
|
|
201
|
+
* Add an observer for agent events
|
|
202
|
+
*/
|
|
203
|
+
addObserver(callbacks) {
|
|
204
|
+
this.observers.push(callbacks);
|
|
205
|
+
}
|
|
206
|
+
/**
|
|
207
|
+
* Remove an observer
|
|
208
|
+
*/
|
|
209
|
+
removeObserver(callbacks) {
|
|
210
|
+
const index = this.observers.indexOf(callbacks);
|
|
211
|
+
if (index >= 0) {
|
|
212
|
+
this.observers.splice(index, 1);
|
|
213
|
+
}
|
|
214
|
+
}
|
|
215
|
+
/**
|
|
216
|
+
* Spawn an agent to execute a task
|
|
217
|
+
* This is the main entry point for running agents
|
|
218
|
+
*/
|
|
219
|
+
async spawn(config) {
|
|
220
|
+
// Find appropriate backend
|
|
221
|
+
const backend = this.findBackend(config.agentType);
|
|
222
|
+
if (!backend) {
|
|
223
|
+
throw new Error(`No backend found for agent type: ${config.agentType}. ` +
|
|
224
|
+
`Available backends: ${this.listBackends().join(', ')}`);
|
|
225
|
+
}
|
|
226
|
+
// Query memory for relevant knowledge if not provided
|
|
227
|
+
let knowledge = config.injectedKnowledge;
|
|
228
|
+
if (!knowledge) {
|
|
229
|
+
knowledge = await this.memory.queryV2(config.task.description, {
|
|
230
|
+
domains: config.task.domain ? [config.task.domain] : undefined,
|
|
231
|
+
});
|
|
232
|
+
}
|
|
233
|
+
// Format knowledge for injection
|
|
234
|
+
const formattedKnowledge = this.injector.format(knowledge, config.agentType);
|
|
235
|
+
// Use a session holder to allow callbacks to reference the session
|
|
236
|
+
// before it's assigned (closures capture the holder object, not the value)
|
|
237
|
+
const sessionHolder = { current: null };
|
|
238
|
+
// Build full config with knowledge injection
|
|
239
|
+
const fullConfig = {
|
|
240
|
+
...config,
|
|
241
|
+
systemPromptAdditions: [
|
|
242
|
+
config.systemPromptAdditions,
|
|
243
|
+
formattedKnowledge
|
|
244
|
+
? `\n## Atlas Knowledge\n${formattedKnowledge}`
|
|
245
|
+
: undefined,
|
|
246
|
+
]
|
|
247
|
+
.filter(Boolean)
|
|
248
|
+
.join('\n\n'),
|
|
249
|
+
onMessage: (message) => {
|
|
250
|
+
config.onMessage?.(message);
|
|
251
|
+
if (sessionHolder.current) {
|
|
252
|
+
this.notifyMessage(sessionHolder.current, message);
|
|
253
|
+
}
|
|
254
|
+
},
|
|
255
|
+
onToolCall: (toolCall) => {
|
|
256
|
+
config.onToolCall?.(toolCall);
|
|
257
|
+
if (sessionHolder.current) {
|
|
258
|
+
this.notifyToolCall(sessionHolder.current, toolCall);
|
|
259
|
+
}
|
|
260
|
+
},
|
|
261
|
+
};
|
|
262
|
+
// Spawn the agent
|
|
263
|
+
const session = await backend.spawn(fullConfig);
|
|
264
|
+
sessionHolder.current = session;
|
|
265
|
+
this.activeSessions.set(session.id, session);
|
|
266
|
+
this.notifySessionStart(session);
|
|
267
|
+
// Wait for completion and extract trajectory
|
|
268
|
+
const finalSession = await this.waitForCompletion(backend, session.id);
|
|
269
|
+
this.activeSessions.delete(session.id);
|
|
270
|
+
const trajectory = this.extractor.extract(finalSession, config.task);
|
|
271
|
+
this.notifySessionEnd(finalSession, trajectory);
|
|
272
|
+
return {
|
|
273
|
+
session: finalSession,
|
|
274
|
+
trajectory,
|
|
275
|
+
success: finalSession.state === 'completed',
|
|
276
|
+
metrics: {
|
|
277
|
+
totalTime: finalSession.endTime
|
|
278
|
+
? finalSession.endTime.getTime() - finalSession.startTime.getTime()
|
|
279
|
+
: 0,
|
|
280
|
+
toolCallCount: finalSession.toolCalls.length,
|
|
281
|
+
messageCount: finalSession.messages.length,
|
|
282
|
+
},
|
|
283
|
+
};
|
|
284
|
+
}
|
|
285
|
+
/**
|
|
286
|
+
* Spawn an agent WITHOUT knowledge injection (for baseline/control)
|
|
287
|
+
*/
|
|
288
|
+
async spawnBaseline(config) {
|
|
289
|
+
return this.spawn({
|
|
290
|
+
...config,
|
|
291
|
+
injectedKnowledge: {
|
|
292
|
+
experiences: [],
|
|
293
|
+
playbooks: [],
|
|
294
|
+
},
|
|
295
|
+
});
|
|
296
|
+
}
|
|
297
|
+
/**
|
|
298
|
+
* Run an A/B comparison: with and without knowledge injection
|
|
299
|
+
*/
|
|
300
|
+
async compareWithKnowledge(config) {
|
|
301
|
+
// Run baseline first
|
|
302
|
+
const baseline = await this.spawnBaseline(config);
|
|
303
|
+
// Run with knowledge
|
|
304
|
+
const withKnowledge = await this.spawn(config);
|
|
305
|
+
return {
|
|
306
|
+
withKnowledge,
|
|
307
|
+
baseline,
|
|
308
|
+
improvement: {
|
|
309
|
+
timeDelta: baseline.metrics.totalTime - withKnowledge.metrics.totalTime,
|
|
310
|
+
toolCallDelta: baseline.metrics.toolCallCount - withKnowledge.metrics.toolCallCount,
|
|
311
|
+
successDelta: (withKnowledge.success ? 1 : 0) - (baseline.success ? 1 : 0),
|
|
312
|
+
},
|
|
313
|
+
};
|
|
314
|
+
}
|
|
315
|
+
/**
|
|
316
|
+
* Observe an external agent session and learn from it
|
|
317
|
+
* Use this when you have access to another agent's execution
|
|
318
|
+
*/
|
|
319
|
+
async observeAndLearn(session, task) {
|
|
320
|
+
const trajectory = this.extractor.extract(session, task);
|
|
321
|
+
// Store in memory for learning
|
|
322
|
+
await this.memory.storeTrajectory(trajectory);
|
|
323
|
+
return trajectory;
|
|
324
|
+
}
|
|
325
|
+
/**
|
|
326
|
+
* Get active sessions
|
|
327
|
+
*/
|
|
328
|
+
getActiveSessions() {
|
|
329
|
+
return Array.from(this.activeSessions.values());
|
|
330
|
+
}
|
|
331
|
+
/**
|
|
332
|
+
* Terminate a session
|
|
333
|
+
*/
|
|
334
|
+
async terminate(sessionId) {
|
|
335
|
+
const session = this.activeSessions.get(sessionId);
|
|
336
|
+
if (!session)
|
|
337
|
+
return;
|
|
338
|
+
const backend = this.findBackend(session.agentType);
|
|
339
|
+
if (backend?.terminate) {
|
|
340
|
+
await backend.terminate(sessionId);
|
|
341
|
+
}
|
|
342
|
+
this.activeSessions.delete(sessionId);
|
|
343
|
+
}
|
|
344
|
+
/**
|
|
345
|
+
* Terminate all active sessions
|
|
346
|
+
*/
|
|
347
|
+
async terminateAll() {
|
|
348
|
+
const sessionIds = Array.from(this.activeSessions.keys());
|
|
349
|
+
await Promise.all(sessionIds.map((id) => this.terminate(id)));
|
|
350
|
+
}
|
|
351
|
+
/**
|
|
352
|
+
* Close the manager and clean up resources
|
|
353
|
+
* Terminates all active sessions and clears all state
|
|
354
|
+
*/
|
|
355
|
+
async close() {
|
|
356
|
+
// Terminate all active sessions
|
|
357
|
+
await this.terminateAll();
|
|
358
|
+
// Clear backends
|
|
359
|
+
this.backends.clear();
|
|
360
|
+
// Clear observers
|
|
361
|
+
this.observers = [];
|
|
362
|
+
}
|
|
363
|
+
// Private helpers
|
|
364
|
+
findBackend(agentType) {
|
|
365
|
+
// First try exact match
|
|
366
|
+
for (const backend of this.backends.values()) {
|
|
367
|
+
if (backend.supportedTypes.includes(agentType)) {
|
|
368
|
+
return backend;
|
|
369
|
+
}
|
|
370
|
+
}
|
|
371
|
+
// Then try by backend name
|
|
372
|
+
return this.backends.get(agentType);
|
|
373
|
+
}
|
|
374
|
+
async waitForCompletion(backend, sessionId) {
|
|
375
|
+
// Poll for completion
|
|
376
|
+
while (true) {
|
|
377
|
+
const session = await backend.getSession(sessionId);
|
|
378
|
+
if (!session) {
|
|
379
|
+
throw new Error(`Session ${sessionId} not found`);
|
|
380
|
+
}
|
|
381
|
+
if (session.state === 'completed' ||
|
|
382
|
+
session.state === 'failed') {
|
|
383
|
+
return session;
|
|
384
|
+
}
|
|
385
|
+
// Wait before polling again
|
|
386
|
+
await new Promise((resolve) => setTimeout(resolve, 1000));
|
|
387
|
+
}
|
|
388
|
+
}
|
|
389
|
+
notifySessionStart(session) {
|
|
390
|
+
for (const observer of this.observers) {
|
|
391
|
+
observer.onSessionStart?.(session);
|
|
392
|
+
}
|
|
393
|
+
}
|
|
394
|
+
notifyMessage(session, message) {
|
|
395
|
+
for (const observer of this.observers) {
|
|
396
|
+
observer.onMessage?.(session, message);
|
|
397
|
+
}
|
|
398
|
+
}
|
|
399
|
+
notifyToolCall(session, toolCall) {
|
|
400
|
+
for (const observer of this.observers) {
|
|
401
|
+
observer.onToolCall?.(session, toolCall);
|
|
402
|
+
}
|
|
403
|
+
}
|
|
404
|
+
notifySessionEnd(session, trajectory) {
|
|
405
|
+
for (const observer of this.observers) {
|
|
406
|
+
observer.onSessionEnd?.(session, trajectory);
|
|
407
|
+
}
|
|
408
|
+
}
|
|
409
|
+
}
|
|
410
|
+
/**
|
|
411
|
+
* Create an agent manager
|
|
412
|
+
*/
|
|
413
|
+
export function createAgentManager(memory, options) {
|
|
414
|
+
return new AgentManager(memory, options);
|
|
415
|
+
}
|
|
416
|
+
//# sourceMappingURL=manager.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"manager.js","sourceRoot":"","sources":["../../src/runtime/manager.ts"],"names":[],"mappings":"AACA,OAAO,EACL,gBAAgB,EAChB,UAAU,EACV,cAAc,EACd,cAAc,GACf,MAAM,mBAAmB,CAAC;AAc3B;;GAEG;AACH,MAAM,OAAO,wBAAwB;IACnC,MAAM,CACJ,SAA8B,EAC9B,UAAkB,EAClB,OAAsE;QAEtE,MAAM,MAAM,GAAG,OAAO,EAAE,MAAM,IAAI,UAAU,CAAC;QAE7C,IAAI,MAAM,KAAK,MAAM,EAAE,CAAC;YACtB,OAAO,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;QAC5C,CAAC;QAED,IAAI,MAAM,KAAK,KAAK,EAAE,CAAC;YACrB,OAAO,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QACnC,CAAC;QAED,OAAO,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;IACxC,CAAC;IAEO,cAAc,CAAC,SAA8B;QACnD,MAAM,QAAQ,GAAa,EAAE,CAAC;QAE9B,IAAI,SAAS,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACnC,QAAQ,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC;YAC3C,KAAK,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,SAAS,CAAC,SAAS,EAAE,CAAC;gBACjE,QAAQ,CAAC,IAAI,CAAC,OAAO,QAAQ,CAAC,IAAI,KAAK,CAAC,KAAK,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,SAAS,GAAG,CAAC,CAAC;gBAC5F,QAAQ,CAAC,IAAI,CAAC,iBAAiB,QAAQ,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC;gBAC7D,IAAI,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACzC,QAAQ,CAAC,IAAI,CAAC,gBAAgB,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBACxE,CAAC;gBACD,IAAI,QAAQ,CAAC,QAAQ,CAAC,KAAK,IAAI,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAClE,QAAQ,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;oBAC5B,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE;wBAC1C,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;oBACrC,CAAC,CAAC,CAAC;gBACL,CAAC;gBACD,QAAQ,CAAC,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,UAAU,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;YAC9E,CAAC;QACH,CAAC;QAED,IAAI,SAAS,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACrC,QAAQ,CAAC,IAAI,CAAC,+BAA+B,CAAC,CAAC;YAC/C,KAAK,MAAM,EAAE,UAAU,EAAE,KAAK,EAAE,IAAI,SAAS,CAAC,WAAW,EAAE,CAAC;gBAC1D,QAAQ,CAAC,IAAI,CAAC,8BAA8B,CAAC,KAAK,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;gBAC1E,QAAQ,CAAC,IAAI,CAAC,aAAa,UAAU,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC;gBACpE,QAAQ,CAAC,IAAI,CAAC,gBAAgB,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC;gBAC3E,QAAQ,CAAC,IAAI,CAAC,iBAAiB,UAAU,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC;gBAC7E,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACpB,CAAC;QACH,CAAC;QAED,OAAO,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC7B,CAAC;IAEO,SAAS,CAAC,SAA8B;QAC9C,MAAM,KAAK,GAAa,CAAC,mBAAmB,CAAC,CAAC;QAE9C,IAAI,SAAS,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACnC,KAAK,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YAC5B,KAAK,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,IAAI,SAAS,CAAC,SAAS,EAAE,CAAC;gBACtD,KAAK,CAAC,IAAI,CAAC,uBAAuB,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,gBAAgB,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;gBACrG,KAAK,CAAC,IAAI,CAAC,mBAAmB,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;gBACvF,IAAI,QAAQ,CAAC,QAAQ,CAAC,KAAK,IAAI,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAClE,KAAK,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;oBAC5B,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE;wBAC1C,KAAK,CAAC,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;oBAC1E,CAAC,CAAC,CAAC;oBACH,KAAK,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;gBAC/B,CAAC;gBACD,KAAK,CAAC,IAAI,CAAC,qBAAqB,QAAQ,CAAC,UAAU,eAAe,CAAC,CAAC;gBACpE,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;YAChC,CAAC;YACD,KAAK,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAC/B,CAAC;QAED,IAAI,SAAS,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACrC,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;YAC9B,KAAK,MAAM,EAAE,UAAU,EAAE,KAAK,EAAE,IAAI,SAAS,CAAC,WAAW,EAAE,CAAC;gBAC1D,KAAK,CAAC,IAAI,CAAC,8BAA8B,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;gBAC/D,KAAK,CAAC,IAAI,CAAC,eAAe,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC;gBACvF,KAAK,CAAC,IAAI,CAAC,kBAAkB,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,YAAY,CAAC,CAAC;gBACrF,KAAK,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;YAClC,CAAC;YACD,KAAK,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;QACjC,CAAC;QAED,KAAK,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QACjC,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;IAEO,SAAS,CAAC,GAAW;QAC3B,OAAO,GAAG;aACP,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC;aACtB,OAAO,CAAC,IAAI,EAAE,MAAM,CAAC;aACrB,OAAO,CAAC,IAAI,EAAE,MAAM,CAAC;aACrB,OAAO,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;IAC7B,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,0BAA0B;IACrC,OAAO,CAAC,OAAqB,EAAE,IAAU;QACvC,MAAM,KAAK,GAAW,EAAE,CAAC;QAEzB,8BAA8B;QAC9B,KAAK,MAAM,QAAQ,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;YACzC,KAAK,CAAC,IAAI,CACR,UAAU,CAAC;gBACT,OAAO,EAAE,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,QAAQ,EAAE,QAAQ,CAAC;gBACxD,MAAM,EAAE,GAAG,QAAQ,CAAC,IAAI,KAAK,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE;gBAClE,WAAW,EAAE,QAAQ,CAAC,KAAK;oBACzB,CAAC,CAAC,UAAU,QAAQ,CAAC,KAAK,EAAE;oBAC5B,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,MAAM,CAAC;gBACzC,QAAQ,EAAE;oBACR,UAAU,EAAE,QAAQ,CAAC,EAAE;oBACvB,QAAQ,EAAE,QAAQ,CAAC,IAAI;oBACvB,QAAQ,EAAE,QAAQ,CAAC,OAAO;wBACxB,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,OAAO,EAAE,GAAG,QAAQ,CAAC,SAAS,CAAC,OAAO,EAAE;wBAC3D,CAAC,CAAC,SAAS;iBACd;aACF,CAAC,CACH,CAAC;QACJ,CAAC;QAED,oBAAoB;QACpB,MAAM,OAAO,GACX,OAAO,CAAC,KAAK,KAAK,WAAW;YAC3B,CAAC,CAAC,cAAc,CAAC,OAAO,CAAC,MAAM,CAAC;YAChC,CAAC,CAAC,cAAc,CACZ,OAAO,CAAC,KAAK,IAAI,uCAAuC,CACzD,CAAC;QAER,OAAO,gBAAgB,CAAC;YACtB,IAAI;YACJ,KAAK;YACL,OAAO;YACP,OAAO,EAAE,GAAG,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC,EAAE,EAAE;YAC7C,eAAe,EAAE,OAAO,CAAC,OAAO;gBAC9B,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,EAAE,GAAG,OAAO,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC,GAAG,IAAI;gBAClE,CAAC,CAAC,CAAC;YACL,QAAQ,EAAE;gBACR,SAAS,EAAE,OAAO,CAAC,EAAE;gBACrB,SAAS,EAAE,OAAO,CAAC,SAAS;gBAC5B,YAAY,EAAE,OAAO,CAAC,QAAQ,CAAC,MAAM;gBACrC,GAAG,OAAO,CAAC,QAAQ;aACpB;SACF,CAAC,CAAC;IACL,CAAC;IAEO,cAAc,CACpB,QAAwB,EACxB,QAAkB;QAElB,wDAAwD;QACxD,MAAM,YAAY,GAAG,QAAQ,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC;QAClD,MAAM,iBAAiB,GAAG,QAAQ,CAAC,MAAM,CACvC,CAAC,CAAC,EAAE,EAAE,CACJ,CAAC,CAAC,IAAI,KAAK,WAAW;YACtB,CAAC,CAAC,SAAS,CAAC,OAAO,EAAE,GAAG,YAAY;YACpC,CAAC,CAAC,SAAS,CAAC,OAAO,EAAE,GAAG,YAAY,GAAG,KAAK,CAAC,kBAAkB;SAClE,CAAC;QAEF,IAAI,iBAAiB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACjC,MAAM,WAAW,GAAG,iBAAiB,CAAC,iBAAiB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YACpE,6CAA6C;YAC7C,OAAO,WAAW,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;QAC3C,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,cAAc,CAAC,KAAc;QACnC,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC9B,OAAO,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;QAC7B,CAAC;QACD,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;YAChD,MAAM,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;YAClC,OAAO,GAAG,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAC7D,CAAC;QACD,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;IACvB,CAAC;IAEO,eAAe,CAAC,MAAe;QACrC,IAAI,MAAM,KAAK,SAAS;YAAE,OAAO,WAAW,CAAC;QAC7C,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;YAC/B,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QACnE,CAAC;QACD,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,KAAK,IAAI,EAAE,CAAC;YAClD,MAAM,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YACnC,OAAO,GAAG,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAC7D,CAAC;QACD,OAAO,MAAM,CAAC,MAAM,CAAC,CAAC;IACxB,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,YAAY;IACf,QAAQ,GAA8B,IAAI,GAAG,EAAE,CAAC;IAChD,MAAM,CAAe;IACrB,QAAQ,CAAoB;IAC5B,SAAS,CAAsB;IAC/B,SAAS,GAA6B,EAAE,CAAC;IACzC,cAAc,GAA8B,IAAI,GAAG,EAAE,CAAC;IAE9D,YACE,MAAoB,EACpB,OAGC;QAED,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,QAAQ,GAAG,OAAO,EAAE,QAAQ,IAAI,IAAI,wBAAwB,EAAE,CAAC;QACpE,IAAI,CAAC,SAAS,GAAG,OAAO,EAAE,SAAS,IAAI,IAAI,0BAA0B,EAAE,CAAC;IAC1E,CAAC;IAED;;OAEG;IACH,eAAe,CAAC,OAAqB;QACnC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IAC3C,CAAC;IAED;;OAEG;IACH,UAAU,CAAC,IAAY;QACrB,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACjC,CAAC;IAED;;OAEG;IACH,YAAY;QACV,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC;IAC1C,CAAC;IAED;;OAEG;IACH,WAAW,CAAC,SAAiC;QAC3C,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IACjC,CAAC;IAED;;OAEG;IACH,cAAc,CAAC,SAAiC;QAC9C,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAChD,IAAI,KAAK,IAAI,CAAC,EAAE,CAAC;YACf,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QAClC,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,KAAK,CAAC,MAAwB;QAClC,2BAA2B;QAC3B,MAAM,OAAO,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QACnD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CACb,oCAAoC,MAAM,CAAC,SAAS,IAAI;gBACtD,uBAAuB,IAAI,CAAC,YAAY,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAC1D,CAAC;QACJ,CAAC;QAED,sDAAsD;QACtD,IAAI,SAAS,GAAG,MAAM,CAAC,iBAAiB,CAAC;QACzC,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,SAAS,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,EAAE;gBAC7D,OAAO,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS;aAC/D,CAAC,CAAC;QACL,CAAC;QAED,iCAAiC;QACjC,MAAM,kBAAkB,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,SAAS,CAAC,CAAC;QAE7E,mEAAmE;QACnE,2EAA2E;QAC3E,MAAM,aAAa,GAAqC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;QAE1E,6CAA6C;QAC7C,MAAM,UAAU,GAAqB;YACnC,GAAG,MAAM;YACT,qBAAqB,EAAE;gBACrB,MAAM,CAAC,qBAAqB;gBAC5B,kBAAkB;oBAChB,CAAC,CAAC,yBAAyB,kBAAkB,EAAE;oBAC/C,CAAC,CAAC,SAAS;aACd;iBACE,MAAM,CAAC,OAAO,CAAC;iBACf,IAAI,CAAC,MAAM,CAAC;YACf,SAAS,EAAE,CAAC,OAAO,EAAE,EAAE;gBACrB,MAAM,CAAC,SAAS,EAAE,CAAC,OAAO,CAAC,CAAC;gBAC5B,IAAI,aAAa,CAAC,OAAO,EAAE,CAAC;oBAC1B,IAAI,CAAC,aAAa,CAAC,aAAa,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;gBACrD,CAAC;YACH,CAAC;YACD,UAAU,EAAE,CAAC,QAAQ,EAAE,EAAE;gBACvB,MAAM,CAAC,UAAU,EAAE,CAAC,QAAQ,CAAC,CAAC;gBAC9B,IAAI,aAAa,CAAC,OAAO,EAAE,CAAC;oBAC1B,IAAI,CAAC,cAAc,CAAC,aAAa,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;gBACvD,CAAC;YACH,CAAC;SACF,CAAC;QAEF,kBAAkB;QAClB,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;QAChD,aAAa,CAAC,OAAO,GAAG,OAAO,CAAC;QAChC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QAC7C,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC;QAEjC,6CAA6C;QAC7C,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,OAAO,EAAE,OAAO,CAAC,EAAE,CAAC,CAAC;QACvE,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAEvC,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,YAAY,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;QACrE,IAAI,CAAC,gBAAgB,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC;QAEhD,OAAO;YACL,OAAO,EAAE,YAAY;YACrB,UAAU;YACV,OAAO,EAAE,YAAY,CAAC,KAAK,KAAK,WAAW;YAC3C,OAAO,EAAE;gBACP,SAAS,EAAE,YAAY,CAAC,OAAO;oBAC7B,CAAC,CAAC,YAAY,CAAC,OAAO,CAAC,OAAO,EAAE,GAAG,YAAY,CAAC,SAAS,CAAC,OAAO,EAAE;oBACnE,CAAC,CAAC,CAAC;gBACL,aAAa,EAAE,YAAY,CAAC,SAAS,CAAC,MAAM;gBAC5C,YAAY,EAAE,YAAY,CAAC,QAAQ,CAAC,MAAM;aAC3C;SACF,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,aAAa,CACjB,MAAmD;QAEnD,OAAO,IAAI,CAAC,KAAK,CAAC;YAChB,GAAG,MAAM;YACT,iBAAiB,EAAE;gBACjB,WAAW,EAAE,EAAE;gBACf,SAAS,EAAE,EAAE;aACd;SACF,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,oBAAoB,CACxB,MAAmD;QAUnD,qBAAqB;QACrB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;QAElD,qBAAqB;QACrB,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QAE/C,OAAO;YACL,aAAa;YACb,QAAQ;YACR,WAAW,EAAE;gBACX,SAAS,EAAE,QAAQ,CAAC,OAAO,CAAC,SAAS,GAAG,aAAa,CAAC,OAAO,CAAC,SAAS;gBACvE,aAAa,EACX,QAAQ,CAAC,OAAO,CAAC,aAAa,GAAG,aAAa,CAAC,OAAO,CAAC,aAAa;gBACtE,YAAY,EACV,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;aAC/D;SACF,CAAC;IACJ,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,eAAe,CAAC,OAAqB,EAAE,IAAU;QACrD,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QAEzD,+BAA+B;QAC/B,MAAM,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;QAE9C,OAAO,UAAU,CAAC;IACpB,CAAC;IAED;;OAEG;IACH,iBAAiB;QACf,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,MAAM,EAAE,CAAC,CAAC;IAClD,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,SAAS,CAAC,SAAiB;QAC/B,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACnD,IAAI,CAAC,OAAO;YAAE,OAAO;QAErB,MAAM,OAAO,GAAG,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACpD,IAAI,OAAO,EAAE,SAAS,EAAE,CAAC;YACvB,MAAM,OAAO,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QACrC,CAAC;QAED,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;IACxC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,YAAY;QAChB,MAAM,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC,CAAC;QAC1D,MAAM,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IAChE,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,KAAK;QACT,gCAAgC;QAChC,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;QAE1B,iBAAiB;QACjB,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;QAEtB,kBAAkB;QAClB,IAAI,CAAC,SAAS,GAAG,EAAE,CAAC;IACtB,CAAC;IAED,kBAAkB;IAEV,WAAW,CAAC,SAAiB;QACnC,wBAAwB;QACxB,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC;YAC7C,IAAI,OAAO,CAAC,cAAc,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;gBAC/C,OAAO,OAAO,CAAC;YACjB,CAAC;QACH,CAAC;QACD,2BAA2B;QAC3B,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IACtC,CAAC;IAEO,KAAK,CAAC,iBAAiB,CAC7B,OAAqB,EACrB,SAAiB;QAEjB,sBAAsB;QACtB,OAAO,IAAI,EAAE,CAAC;YACZ,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;YACpD,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,MAAM,IAAI,KAAK,CAAC,WAAW,SAAS,YAAY,CAAC,CAAC;YACpD,CAAC;YAED,IACE,OAAO,CAAC,KAAK,KAAK,WAAW;gBAC7B,OAAO,CAAC,KAAK,KAAK,QAAQ,EAC1B,CAAC;gBACD,OAAO,OAAO,CAAC;YACjB,CAAC;YAED,4BAA4B;YAC5B,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC;QAC5D,CAAC;IACH,CAAC;IAEO,kBAAkB,CAAC,OAAqB;QAC9C,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACtC,QAAQ,CAAC,cAAc,EAAE,CAAC,OAAO,CAAC,CAAC;QACrC,CAAC;IACH,CAAC;IAEO,aAAa,CAAC,OAAqB,EAAE,OAAqB;QAChE,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACtC,QAAQ,CAAC,SAAS,EAAE,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QACzC,CAAC;IACH,CAAC;IAEO,cAAc,CAAC,OAAqB,EAAE,QAAkB;QAC9D,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACtC,QAAQ,CAAC,UAAU,EAAE,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;QAC3C,CAAC;IACH,CAAC;IAEO,gBAAgB,CAAC,OAAqB,EAAE,UAAsB;QACpE,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACtC,QAAQ,CAAC,YAAY,EAAE,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;QAC/C,CAAC;IACH,CAAC;CACF;AAED;;GAEG;AACH,MAAM,UAAU,kBAAkB,CAChC,MAAoB,EACpB,OAGC;IAED,OAAO,IAAI,YAAY,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;AAC3C,CAAC"}
|
|
@@ -0,0 +1,138 @@
|
|
|
1
|
+
import type { Task, Trajectory } from '../types/index.js';
|
|
2
|
+
import type { MemoryQueryResultV2 } from '../memory/system.js';
|
|
3
|
+
/**
|
|
4
|
+
* Message in an agent conversation
|
|
5
|
+
*/
|
|
6
|
+
export interface AgentMessage {
|
|
7
|
+
role: 'user' | 'assistant' | 'system' | 'tool_use' | 'tool_result';
|
|
8
|
+
content: string;
|
|
9
|
+
toolName?: string;
|
|
10
|
+
toolInput?: unknown;
|
|
11
|
+
toolOutput?: unknown;
|
|
12
|
+
timestamp: Date;
|
|
13
|
+
}
|
|
14
|
+
/**
|
|
15
|
+
* Tool call made by an agent
|
|
16
|
+
*/
|
|
17
|
+
export interface ToolCall {
|
|
18
|
+
id: string;
|
|
19
|
+
name: string;
|
|
20
|
+
input: unknown;
|
|
21
|
+
output?: unknown;
|
|
22
|
+
error?: string;
|
|
23
|
+
startTime: Date;
|
|
24
|
+
endTime?: Date;
|
|
25
|
+
}
|
|
26
|
+
/**
|
|
27
|
+
* Agent execution state
|
|
28
|
+
*/
|
|
29
|
+
export type AgentState = 'idle' | 'running' | 'paused' | 'completed' | 'failed';
|
|
30
|
+
/**
|
|
31
|
+
* Agent session - represents a running or completed agent execution
|
|
32
|
+
*/
|
|
33
|
+
export interface AgentSession {
|
|
34
|
+
id: string;
|
|
35
|
+
agentType: string;
|
|
36
|
+
task: Task;
|
|
37
|
+
state: AgentState;
|
|
38
|
+
messages: AgentMessage[];
|
|
39
|
+
toolCalls: ToolCall[];
|
|
40
|
+
startTime: Date;
|
|
41
|
+
endTime?: Date;
|
|
42
|
+
result?: unknown;
|
|
43
|
+
error?: string;
|
|
44
|
+
metadata: Record<string, unknown>;
|
|
45
|
+
}
|
|
46
|
+
/**
|
|
47
|
+
* Configuration for spawning an agent
|
|
48
|
+
*/
|
|
49
|
+
export interface AgentSpawnConfig {
|
|
50
|
+
/** Type of agent to spawn (e.g., 'claude-code', 'custom', etc.) */
|
|
51
|
+
agentType: string;
|
|
52
|
+
/** Task for the agent to work on */
|
|
53
|
+
task: Task;
|
|
54
|
+
/** Optional knowledge to inject into the agent's context */
|
|
55
|
+
injectedKnowledge?: MemoryQueryResultV2;
|
|
56
|
+
/** Custom system prompt additions */
|
|
57
|
+
systemPromptAdditions?: string;
|
|
58
|
+
/** Environment variables for the agent */
|
|
59
|
+
env?: Record<string, string>;
|
|
60
|
+
/** Working directory */
|
|
61
|
+
cwd?: string;
|
|
62
|
+
/** Timeout in milliseconds */
|
|
63
|
+
timeout?: number;
|
|
64
|
+
/** Whether to capture detailed tool calls */
|
|
65
|
+
captureToolCalls?: boolean;
|
|
66
|
+
/** Callback for real-time observation */
|
|
67
|
+
onMessage?: (message: AgentMessage) => void;
|
|
68
|
+
onToolCall?: (toolCall: ToolCall) => void;
|
|
69
|
+
/** Additional backend-specific options */
|
|
70
|
+
backendOptions?: Record<string, unknown>;
|
|
71
|
+
}
|
|
72
|
+
/**
|
|
73
|
+
* Result of agent execution
|
|
74
|
+
*/
|
|
75
|
+
export interface AgentResult {
|
|
76
|
+
session: AgentSession;
|
|
77
|
+
trajectory: Trajectory;
|
|
78
|
+
success: boolean;
|
|
79
|
+
metrics: {
|
|
80
|
+
totalTime: number;
|
|
81
|
+
toolCallCount: number;
|
|
82
|
+
messageCount: number;
|
|
83
|
+
tokenEstimate?: number;
|
|
84
|
+
};
|
|
85
|
+
}
|
|
86
|
+
/**
|
|
87
|
+
* Agent backend protocol - implement this for different agent systems
|
|
88
|
+
*/
|
|
89
|
+
export interface AgentBackend {
|
|
90
|
+
/** Unique name for this backend */
|
|
91
|
+
readonly name: string;
|
|
92
|
+
/** Supported agent types */
|
|
93
|
+
readonly supportedTypes: string[];
|
|
94
|
+
/** Check if backend is available/configured */
|
|
95
|
+
isAvailable(): Promise<boolean>;
|
|
96
|
+
/** Spawn and run an agent */
|
|
97
|
+
spawn(config: AgentSpawnConfig): Promise<AgentSession>;
|
|
98
|
+
/** Get status of a running session */
|
|
99
|
+
getSession(sessionId: string): Promise<AgentSession | undefined>;
|
|
100
|
+
/** Send a message to a running session */
|
|
101
|
+
sendMessage?(sessionId: string, message: string): Promise<void>;
|
|
102
|
+
/** Pause a running session */
|
|
103
|
+
pause?(sessionId: string): Promise<void>;
|
|
104
|
+
/** Resume a paused session */
|
|
105
|
+
resume?(sessionId: string): Promise<void>;
|
|
106
|
+
/** Terminate a session */
|
|
107
|
+
terminate?(sessionId: string): Promise<void>;
|
|
108
|
+
/** List active sessions */
|
|
109
|
+
listSessions?(): Promise<AgentSession[]>;
|
|
110
|
+
}
|
|
111
|
+
/**
|
|
112
|
+
* Knowledge injector - formats knowledge for injection into agents
|
|
113
|
+
*/
|
|
114
|
+
export interface KnowledgeInjector {
|
|
115
|
+
/** Format knowledge for a specific agent type */
|
|
116
|
+
format(knowledge: MemoryQueryResultV2, agentType: string, options?: {
|
|
117
|
+
maxTokens?: number;
|
|
118
|
+
format?: 'markdown' | 'json' | 'xml';
|
|
119
|
+
}): string;
|
|
120
|
+
}
|
|
121
|
+
/**
|
|
122
|
+
* Trajectory extractor - extracts trajectories from agent sessions
|
|
123
|
+
*/
|
|
124
|
+
export interface TrajectoryExtractor {
|
|
125
|
+
/** Extract a trajectory from an agent session */
|
|
126
|
+
extract(session: AgentSession, task: Task): Trajectory;
|
|
127
|
+
}
|
|
128
|
+
/**
|
|
129
|
+
* Agent observer callbacks
|
|
130
|
+
*/
|
|
131
|
+
export interface AgentObserverCallbacks {
|
|
132
|
+
onSessionStart?: (session: AgentSession) => void;
|
|
133
|
+
onMessage?: (session: AgentSession, message: AgentMessage) => void;
|
|
134
|
+
onToolCall?: (session: AgentSession, toolCall: ToolCall) => void;
|
|
135
|
+
onSessionEnd?: (session: AgentSession, trajectory: Trajectory) => void;
|
|
136
|
+
onError?: (session: AgentSession, error: Error) => void;
|
|
137
|
+
}
|
|
138
|
+
//# sourceMappingURL=types.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/runtime/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAC1D,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,qBAAqB,CAAC;AAE/D;;GAEG;AACH,MAAM,WAAW,YAAY;IAC3B,IAAI,EAAE,MAAM,GAAG,WAAW,GAAG,QAAQ,GAAG,UAAU,GAAG,aAAa,CAAC;IACnE,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,SAAS,EAAE,IAAI,CAAC;CACjB;AAED;;GAEG;AACH,MAAM,WAAW,QAAQ;IACvB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,OAAO,CAAC;IACf,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,IAAI,CAAC;IAChB,OAAO,CAAC,EAAE,IAAI,CAAC;CAChB;AAED;;GAEG;AACH,MAAM,MAAM,UAAU,GAAG,MAAM,GAAG,SAAS,GAAG,QAAQ,GAAG,WAAW,GAAG,QAAQ,CAAC;AAEhF;;GAEG;AACH,MAAM,WAAW,YAAY;IAC3B,EAAE,EAAE,MAAM,CAAC;IACX,SAAS,EAAE,MAAM,CAAC;IAClB,IAAI,EAAE,IAAI,CAAC;IACX,KAAK,EAAE,UAAU,CAAC;IAClB,QAAQ,EAAE,YAAY,EAAE,CAAC;IACzB,SAAS,EAAE,QAAQ,EAAE,CAAC;IACtB,SAAS,EAAE,IAAI,CAAC;IAChB,OAAO,CAAC,EAAE,IAAI,CAAC;IACf,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACnC;AAED;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAC/B,mEAAmE;IACnE,SAAS,EAAE,MAAM,CAAC;IAClB,oCAAoC;IACpC,IAAI,EAAE,IAAI,CAAC;IACX,4DAA4D;IAC5D,iBAAiB,CAAC,EAAE,mBAAmB,CAAC;IACxC,qCAAqC;IACrC,qBAAqB,CAAC,EAAE,MAAM,CAAC;IAC/B,0CAA0C;IAC1C,GAAG,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC7B,wBAAwB;IACxB,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,8BAA8B;IAC9B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,6CAA6C;IAC7C,gBAAgB,CAAC,EAAE,OAAO,CAAC;IAC3B,yCAAyC;IACzC,SAAS,CAAC,EAAE,CAAC,OAAO,EAAE,YAAY,KAAK,IAAI,CAAC;IAC5C,UAAU,CAAC,EAAE,CAAC,QAAQ,EAAE,QAAQ,KAAK,IAAI,CAAC;IAC1C,0CAA0C;IAC1C,cAAc,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAC1C;AAED;;GAEG;AACH,MAAM,WAAW,WAAW;IAC1B,OAAO,EAAE,YAAY,CAAC;IACtB,UAAU,EAAE,UAAU,CAAC;IACvB,OAAO,EAAE,OAAO,CAAC;IACjB,OAAO,EAAE;QACP,SAAS,EAAE,MAAM,CAAC;QAClB,aAAa,EAAE,MAAM,CAAC;QACtB,YAAY,EAAE,MAAM,CAAC;QACrB,aAAa,CAAC,EAAE,MAAM,CAAC;KACxB,CAAC;CACH;AAED;;GAEG;AACH,MAAM,WAAW,YAAY;IAC3B,mCAAmC;IACnC,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IAEtB,4BAA4B;IAC5B,QAAQ,CAAC,cAAc,EAAE,MAAM,EAAE,CAAC;IAElC,+CAA+C;IAC/C,WAAW,IAAI,OAAO,CAAC,OAAO,CAAC,CAAC;IAEhC,6BAA6B;IAC7B,KAAK,CAAC,MAAM,EAAE,gBAAgB,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC;IAEvD,sCAAsC;IACtC,UAAU,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,GAAG,SAAS,CAAC,CAAC;IAEjE,0CAA0C;IAC1C,WAAW,CAAC,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAEhE,8BAA8B;IAC9B,KAAK,CAAC,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAEzC,8BAA8B;IAC9B,MAAM,CAAC,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAE1C,0BAA0B;IAC1B,SAAS,CAAC,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAE7C,2BAA2B;IAC3B,YAAY,CAAC,IAAI,OAAO,CAAC,YAAY,EAAE,CAAC,CAAC;CAC1C;AAED;;GAEG;AACH,MAAM,WAAW,iBAAiB;IAChC,iDAAiD;IACjD,MAAM,CACJ,SAAS,EAAE,mBAAmB,EAC9B,SAAS,EAAE,MAAM,EACjB,OAAO,CAAC,EAAE;QAAE,SAAS,CAAC,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,UAAU,GAAG,MAAM,GAAG,KAAK,CAAA;KAAE,GACrE,MAAM,CAAC;CACX;AAED;;GAEG;AACH,MAAM,WAAW,mBAAmB;IAClC,iDAAiD;IACjD,OAAO,CAAC,OAAO,EAAE,YAAY,EAAE,IAAI,EAAE,IAAI,GAAG,UAAU,CAAC;CACxD;AAED;;GAEG;AACH,MAAM,WAAW,sBAAsB;IACrC,cAAc,CAAC,EAAE,CAAC,OAAO,EAAE,YAAY,KAAK,IAAI,CAAC;IACjD,SAAS,CAAC,EAAE,CAAC,OAAO,EAAE,YAAY,EAAE,OAAO,EAAE,YAAY,KAAK,IAAI,CAAC;IACnE,UAAU,CAAC,EAAE,CAAC,OAAO,EAAE,YAAY,EAAE,QAAQ,EAAE,QAAQ,KAAK,IAAI,CAAC;IACjE,YAAY,CAAC,EAAE,CAAC,OAAO,EAAE,YAAY,EAAE,UAAU,EAAE,UAAU,KAAK,IAAI,CAAC;IACvE,OAAO,CAAC,EAAE,CAAC,OAAO,EAAE,YAAY,EAAE,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;CACzD"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/runtime/types.ts"],"names":[],"mappings":""}
|