@ariaflowagents/core 0.7.1 → 0.9.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 +90 -1
- package/dist/agents/Agent.d.ts +188 -9
- package/dist/agents/Agent.d.ts.map +1 -1
- package/dist/agents/Agent.js +246 -24
- package/dist/agents/Agent.js.map +1 -1
- package/dist/agents/CompositeAgent.d.ts +4 -3
- package/dist/agents/CompositeAgent.d.ts.map +1 -1
- package/dist/agents/CompositeAgent.js +19 -9
- package/dist/agents/CompositeAgent.js.map +1 -1
- package/dist/agents/FlowAgent.d.ts +3 -2
- package/dist/agents/FlowAgent.d.ts.map +1 -1
- package/dist/agents/FlowAgent.js +16 -6
- package/dist/agents/FlowAgent.js.map +1 -1
- package/dist/agents/TriageAgent.d.ts +8 -2
- package/dist/agents/TriageAgent.d.ts.map +1 -1
- package/dist/agents/TriageAgent.js +39 -6
- package/dist/agents/TriageAgent.js.map +1 -1
- package/dist/agents/index.d.ts +1 -1
- package/dist/agents/index.d.ts.map +1 -1
- package/dist/agents/index.js +0 -1
- package/dist/agents/index.js.map +1 -1
- package/dist/capabilities/AutoRetrieveCapability.d.ts +30 -0
- package/dist/capabilities/AutoRetrieveCapability.d.ts.map +1 -0
- package/dist/capabilities/AutoRetrieveCapability.js +36 -0
- package/dist/capabilities/AutoRetrieveCapability.js.map +1 -0
- package/dist/capabilities/ExtractionCapability.d.ts +25 -0
- package/dist/capabilities/ExtractionCapability.d.ts.map +1 -0
- package/dist/capabilities/ExtractionCapability.js +74 -0
- package/dist/capabilities/ExtractionCapability.js.map +1 -0
- package/dist/capabilities/FlowCapability.d.ts +81 -0
- package/dist/capabilities/FlowCapability.d.ts.map +1 -0
- package/dist/capabilities/FlowCapability.js +482 -0
- package/dist/capabilities/FlowCapability.js.map +1 -0
- package/dist/capabilities/GuardrailCapability.d.ts +30 -0
- package/dist/capabilities/GuardrailCapability.d.ts.map +1 -0
- package/dist/capabilities/GuardrailCapability.js +38 -0
- package/dist/capabilities/GuardrailCapability.js.map +1 -0
- package/dist/capabilities/HandoffCapability.d.ts +19 -0
- package/dist/capabilities/HandoffCapability.d.ts.map +1 -0
- package/dist/capabilities/HandoffCapability.js +58 -0
- package/dist/capabilities/HandoffCapability.js.map +1 -0
- package/dist/capabilities/LivePromptAssembler.d.ts +108 -0
- package/dist/capabilities/LivePromptAssembler.d.ts.map +1 -0
- package/dist/capabilities/LivePromptAssembler.js +157 -0
- package/dist/capabilities/LivePromptAssembler.js.map +1 -0
- package/dist/capabilities/TriageCapability.d.ts +16 -0
- package/dist/capabilities/TriageCapability.d.ts.map +1 -0
- package/dist/capabilities/TriageCapability.js +61 -0
- package/dist/capabilities/TriageCapability.js.map +1 -0
- package/dist/capabilities/adapters/ai-sdk.d.ts +14 -0
- package/dist/capabilities/adapters/ai-sdk.d.ts.map +1 -0
- package/dist/capabilities/adapters/ai-sdk.js +29 -0
- package/dist/capabilities/adapters/ai-sdk.js.map +1 -0
- package/dist/capabilities/adapters/gemini.d.ts +15 -0
- package/dist/capabilities/adapters/gemini.d.ts.map +1 -0
- package/dist/capabilities/adapters/gemini.js +40 -0
- package/dist/capabilities/adapters/gemini.js.map +1 -0
- package/dist/capabilities/index.d.ts +154 -0
- package/dist/capabilities/index.d.ts.map +1 -0
- package/dist/capabilities/index.js +128 -0
- package/dist/capabilities/index.js.map +1 -0
- package/dist/eval/EvalRunner.d.ts +12 -0
- package/dist/eval/EvalRunner.d.ts.map +1 -0
- package/dist/eval/EvalRunner.js +64 -0
- package/dist/eval/EvalRunner.js.map +1 -0
- package/dist/eval/scoring.d.ts +15 -0
- package/dist/eval/scoring.d.ts.map +1 -0
- package/dist/eval/scoring.js +152 -0
- package/dist/eval/scoring.js.map +1 -0
- package/dist/eval/types.d.ts +59 -0
- package/dist/eval/types.d.ts.map +1 -0
- package/dist/eval/types.js +2 -0
- package/dist/eval/types.js.map +1 -0
- package/dist/flows/FlowGraph.d.ts +3 -1
- package/dist/flows/FlowGraph.d.ts.map +1 -1
- package/dist/flows/FlowGraph.js +5 -0
- package/dist/flows/FlowGraph.js.map +1 -1
- package/dist/flows/FlowManager.d.ts +68 -1
- package/dist/flows/FlowManager.d.ts.map +1 -1
- package/dist/flows/FlowManager.js +499 -36
- package/dist/flows/FlowManager.js.map +1 -1
- package/dist/flows/extraction.d.ts +16 -1
- package/dist/flows/extraction.d.ts.map +1 -1
- package/dist/flows/extraction.js +34 -0
- package/dist/flows/extraction.js.map +1 -1
- package/dist/flows/index.d.ts +2 -0
- package/dist/flows/index.d.ts.map +1 -1
- package/dist/flows/index.js +1 -0
- package/dist/flows/index.js.map +1 -1
- package/dist/flows/validation.d.ts +1 -1
- package/dist/flows/validation.d.ts.map +1 -1
- package/dist/flows/validation.js +13 -1
- package/dist/flows/validation.js.map +1 -1
- package/dist/foundation/AgentDefinition.d.ts +18 -0
- package/dist/foundation/AgentDefinition.d.ts.map +1 -0
- package/dist/foundation/AgentDefinition.js +2 -0
- package/dist/foundation/AgentDefinition.js.map +1 -0
- package/dist/foundation/AgentStateController.d.ts +26 -0
- package/dist/foundation/AgentStateController.d.ts.map +1 -0
- package/dist/foundation/AgentStateController.js +2 -0
- package/dist/foundation/AgentStateController.js.map +1 -0
- package/dist/foundation/ConversationEventLog.d.ts +72 -0
- package/dist/foundation/ConversationEventLog.d.ts.map +1 -0
- package/dist/foundation/ConversationEventLog.js +2 -0
- package/dist/foundation/ConversationEventLog.js.map +1 -0
- package/dist/foundation/ConversationState.d.ts +31 -0
- package/dist/foundation/ConversationState.d.ts.map +1 -0
- package/dist/foundation/ConversationState.js +2 -0
- package/dist/foundation/ConversationState.js.map +1 -0
- package/dist/foundation/DefaultAgentStateController.d.ts +24 -0
- package/dist/foundation/DefaultAgentStateController.d.ts.map +1 -0
- package/dist/foundation/DefaultAgentStateController.js +49 -0
- package/dist/foundation/DefaultAgentStateController.js.map +1 -0
- package/dist/foundation/DefaultConversationEventLog.d.ts +28 -0
- package/dist/foundation/DefaultConversationEventLog.d.ts.map +1 -0
- package/dist/foundation/DefaultConversationEventLog.js +195 -0
- package/dist/foundation/DefaultConversationEventLog.js.map +1 -0
- package/dist/foundation/DefaultConversationState.d.ts +34 -0
- package/dist/foundation/DefaultConversationState.d.ts.map +1 -0
- package/dist/foundation/DefaultConversationState.js +100 -0
- package/dist/foundation/DefaultConversationState.js.map +1 -0
- package/dist/foundation/DefaultToolExecutor.d.ts +58 -0
- package/dist/foundation/DefaultToolExecutor.d.ts.map +1 -0
- package/dist/foundation/DefaultToolExecutor.js +128 -0
- package/dist/foundation/DefaultToolExecutor.js.map +1 -0
- package/dist/foundation/ToolExecutor.d.ts +44 -0
- package/dist/foundation/ToolExecutor.d.ts.map +1 -0
- package/dist/foundation/ToolExecutor.js +2 -0
- package/dist/foundation/ToolExecutor.js.map +1 -0
- package/dist/foundation/createFoundation.d.ts +33 -0
- package/dist/foundation/createFoundation.d.ts.map +1 -0
- package/dist/foundation/createFoundation.js +34 -0
- package/dist/foundation/createFoundation.js.map +1 -0
- package/dist/foundation/index.d.ts +15 -0
- package/dist/foundation/index.d.ts.map +1 -0
- package/dist/foundation/index.js +8 -0
- package/dist/foundation/index.js.map +1 -0
- package/dist/hooks/HookRunner.d.ts +5 -1
- package/dist/hooks/HookRunner.d.ts.map +1 -1
- package/dist/hooks/HookRunner.js +7 -0
- package/dist/hooks/HookRunner.js.map +1 -1
- package/dist/hooks/builtin/metrics.d.ts.map +1 -1
- package/dist/hooks/builtin/metrics.js +12 -0
- package/dist/hooks/builtin/metrics.js.map +1 -1
- package/dist/hooks/builtin/observability.d.ts +21 -0
- package/dist/hooks/builtin/observability.d.ts.map +1 -0
- package/dist/hooks/builtin/observability.js +535 -0
- package/dist/hooks/builtin/observability.js.map +1 -0
- package/dist/index.d.ts +24 -3
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +16 -2
- package/dist/index.js.map +1 -1
- package/dist/memory/MemoryService.d.ts +40 -0
- package/dist/memory/MemoryService.d.ts.map +1 -0
- package/dist/memory/MemoryService.js +2 -0
- package/dist/memory/MemoryService.js.map +1 -0
- package/dist/memory/index.d.ts +5 -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/preloadMemory.d.ts +17 -0
- package/dist/memory/preloadMemory.d.ts.map +1 -0
- package/dist/memory/preloadMemory.js +62 -0
- package/dist/memory/preloadMemory.js.map +1 -0
- package/dist/memory/stores/InMemoryMemoryService.d.ts +20 -0
- package/dist/memory/stores/InMemoryMemoryService.d.ts.map +1 -0
- package/dist/memory/stores/InMemoryMemoryService.js +92 -0
- package/dist/memory/stores/InMemoryMemoryService.js.map +1 -0
- package/dist/memory/types.d.ts +49 -0
- package/dist/memory/types.d.ts.map +1 -0
- package/dist/memory/types.js +8 -0
- package/dist/memory/types.js.map +1 -0
- package/dist/orchestration/DefaultOrchestrationAuthority.d.ts +91 -0
- package/dist/orchestration/DefaultOrchestrationAuthority.d.ts.map +1 -0
- package/dist/orchestration/DefaultOrchestrationAuthority.js +786 -0
- package/dist/orchestration/DefaultOrchestrationAuthority.js.map +1 -0
- package/dist/orchestration/OrchestrationAuthority.d.ts +119 -0
- package/dist/orchestration/OrchestrationAuthority.d.ts.map +1 -0
- package/dist/orchestration/OrchestrationAuthority.js +2 -0
- package/dist/orchestration/OrchestrationAuthority.js.map +1 -0
- package/dist/orchestration/RealtimeExtractionRunner.d.ts +25 -0
- package/dist/orchestration/RealtimeExtractionRunner.d.ts.map +1 -0
- package/dist/orchestration/RealtimeExtractionRunner.js +62 -0
- package/dist/orchestration/RealtimeExtractionRunner.js.map +1 -0
- package/dist/orchestration/index.d.ts +5 -0
- package/dist/orchestration/index.d.ts.map +1 -0
- package/dist/orchestration/index.js +4 -0
- package/dist/orchestration/index.js.map +1 -0
- package/dist/orchestration/types.d.ts +134 -0
- package/dist/orchestration/types.d.ts.map +1 -0
- package/dist/orchestration/types.js +2 -0
- package/dist/orchestration/types.js.map +1 -0
- package/dist/prompts/AgentPrompt.d.ts +110 -0
- package/dist/prompts/AgentPrompt.d.ts.map +1 -0
- package/dist/prompts/AgentPrompt.js +373 -0
- package/dist/prompts/AgentPrompt.js.map +1 -0
- package/dist/prompts/PromptAssembly.d.ts +119 -0
- package/dist/prompts/PromptAssembly.d.ts.map +1 -0
- package/dist/prompts/PromptAssembly.js +150 -0
- package/dist/prompts/PromptAssembly.js.map +1 -0
- package/dist/prompts/PromptBuilder.d.ts +22 -3
- package/dist/prompts/PromptBuilder.d.ts.map +1 -1
- package/dist/prompts/PromptBuilder.js +242 -13
- package/dist/prompts/PromptBuilder.js.map +1 -1
- package/dist/prompts/PromptRenderer.d.ts +43 -0
- package/dist/prompts/PromptRenderer.d.ts.map +1 -0
- package/dist/prompts/PromptRenderer.js +114 -0
- package/dist/prompts/PromptRenderer.js.map +1 -0
- package/dist/prompts/brandVoice.d.ts +10 -0
- package/dist/prompts/brandVoice.d.ts.map +1 -0
- package/dist/prompts/brandVoice.js +87 -0
- package/dist/prompts/brandVoice.js.map +1 -0
- package/dist/prompts/index.d.ts +11 -4
- package/dist/prompts/index.d.ts.map +1 -1
- package/dist/prompts/index.js +7 -2
- package/dist/prompts/index.js.map +1 -1
- package/dist/prompts/security.d.ts +5 -0
- package/dist/prompts/security.d.ts.map +1 -0
- package/dist/prompts/security.js +52 -0
- package/dist/prompts/security.js.map +1 -0
- package/dist/prompts/types.d.ts +65 -1
- package/dist/prompts/types.d.ts.map +1 -1
- package/dist/prompts/types.js +26 -0
- package/dist/prompts/types.js.map +1 -1
- package/dist/realtime/RealtimeAudioClient.d.ts +105 -0
- package/dist/realtime/RealtimeAudioClient.d.ts.map +1 -0
- package/dist/realtime/RealtimeAudioClient.js +15 -0
- package/dist/realtime/RealtimeAudioClient.js.map +1 -0
- package/dist/realtime/RealtimeRuntime.d.ts +136 -0
- package/dist/realtime/RealtimeRuntime.d.ts.map +1 -0
- package/dist/realtime/RealtimeRuntime.js +270 -0
- package/dist/realtime/RealtimeRuntime.js.map +1 -0
- package/dist/realtime/index.d.ts +4 -0
- package/dist/realtime/index.d.ts.map +1 -0
- package/dist/realtime/index.js +2 -0
- package/dist/realtime/index.js.map +1 -0
- package/dist/runtime/ContextBudget.d.ts +57 -0
- package/dist/runtime/ContextBudget.d.ts.map +1 -0
- package/dist/runtime/ContextBudget.js +103 -0
- package/dist/runtime/ContextBudget.js.map +1 -0
- package/dist/runtime/ContextManager.d.ts +8 -5
- package/dist/runtime/ContextManager.d.ts.map +1 -1
- package/dist/runtime/ContextManager.js +47 -14
- package/dist/runtime/ContextManager.js.map +1 -1
- package/dist/runtime/ExtractionEngine.d.ts +2 -1
- package/dist/runtime/ExtractionEngine.d.ts.map +1 -1
- package/dist/runtime/ExtractionEngine.js +11 -0
- package/dist/runtime/ExtractionEngine.js.map +1 -1
- package/dist/runtime/FlowExecutor.d.ts +22 -15
- package/dist/runtime/FlowExecutor.d.ts.map +1 -1
- package/dist/runtime/FlowExecutor.js +102 -149
- package/dist/runtime/FlowExecutor.js.map +1 -1
- package/dist/runtime/Runtime.d.ts +53 -78
- package/dist/runtime/Runtime.d.ts.map +1 -1
- package/dist/runtime/Runtime.js +272 -1406
- package/dist/runtime/Runtime.js.map +1 -1
- package/dist/runtime/SessionCache.d.ts +16 -0
- package/dist/runtime/SessionCache.d.ts.map +1 -0
- package/dist/runtime/SessionCache.js +49 -0
- package/dist/runtime/SessionCache.js.map +1 -0
- package/dist/runtime/SessionMutex.d.ts +37 -0
- package/dist/runtime/SessionMutex.d.ts.map +1 -0
- package/dist/runtime/SessionMutex.js +59 -0
- package/dist/runtime/SessionMutex.js.map +1 -0
- package/dist/runtime/StreamEmitter.d.ts +34 -0
- package/dist/runtime/StreamEmitter.d.ts.map +1 -0
- package/dist/runtime/StreamEmitter.js +91 -0
- package/dist/runtime/StreamEmitter.js.map +1 -0
- package/dist/runtime/handoffFilters.d.ts +60 -0
- package/dist/runtime/handoffFilters.d.ts.map +1 -0
- package/dist/runtime/handoffFilters.js +95 -0
- package/dist/runtime/handoffFilters.js.map +1 -0
- package/dist/runtime/pipeline/AgentExecuteStage.d.ts +22 -0
- package/dist/runtime/pipeline/AgentExecuteStage.d.ts.map +1 -0
- package/dist/runtime/pipeline/AgentExecuteStage.js +958 -0
- package/dist/runtime/pipeline/AgentExecuteStage.js.map +1 -0
- package/dist/runtime/pipeline/ContextAssembleStage.d.ts +26 -0
- package/dist/runtime/pipeline/ContextAssembleStage.d.ts.map +1 -0
- package/dist/runtime/pipeline/ContextAssembleStage.js +253 -0
- package/dist/runtime/pipeline/ContextAssembleStage.js.map +1 -0
- package/dist/runtime/pipeline/ContextGatherStage.d.ts +21 -0
- package/dist/runtime/pipeline/ContextGatherStage.d.ts.map +1 -0
- package/dist/runtime/pipeline/ContextGatherStage.js +161 -0
- package/dist/runtime/pipeline/ContextGatherStage.js.map +1 -0
- package/dist/runtime/pipeline/IntakeStage.d.ts +25 -0
- package/dist/runtime/pipeline/IntakeStage.d.ts.map +1 -0
- package/dist/runtime/pipeline/IntakeStage.js +126 -0
- package/dist/runtime/pipeline/IntakeStage.js.map +1 -0
- package/dist/runtime/pipeline/PostStreamStage.d.ts +26 -0
- package/dist/runtime/pipeline/PostStreamStage.d.ts.map +1 -0
- package/dist/runtime/pipeline/PostStreamStage.js +129 -0
- package/dist/runtime/pipeline/PostStreamStage.js.map +1 -0
- package/dist/runtime/pipeline/TurnPipeline.d.ts +54 -0
- package/dist/runtime/pipeline/TurnPipeline.d.ts.map +1 -0
- package/dist/runtime/pipeline/TurnPipeline.js +15 -0
- package/dist/runtime/pipeline/TurnPipeline.js.map +1 -0
- package/dist/runtime/pipeline/TurnServices.d.ts +48 -0
- package/dist/runtime/pipeline/TurnServices.d.ts.map +1 -0
- package/dist/runtime/pipeline/TurnServices.js +2 -0
- package/dist/runtime/pipeline/TurnServices.js.map +1 -0
- package/dist/runtime/pipeline/agentTypeGuards.d.ts +4 -0
- package/dist/runtime/pipeline/agentTypeGuards.d.ts.map +1 -0
- package/dist/runtime/pipeline/agentTypeGuards.js +7 -0
- package/dist/runtime/pipeline/agentTypeGuards.js.map +1 -0
- package/dist/runtime/pipeline/index.d.ts +11 -0
- package/dist/runtime/pipeline/index.d.ts.map +1 -0
- package/dist/runtime/pipeline/index.js +13 -0
- package/dist/runtime/pipeline/index.js.map +1 -0
- package/dist/runtime/pipeline/outputProcessing.d.ts +23 -0
- package/dist/runtime/pipeline/outputProcessing.d.ts.map +1 -0
- package/dist/runtime/pipeline/outputProcessing.js +63 -0
- package/dist/runtime/pipeline/outputProcessing.js.map +1 -0
- package/dist/runtime/pipeline/sessionUtils.d.ts +12 -0
- package/dist/runtime/pipeline/sessionUtils.d.ts.map +1 -0
- package/dist/runtime/pipeline/sessionUtils.js +73 -0
- package/dist/runtime/pipeline/sessionUtils.js.map +1 -0
- package/dist/tools/Tool.d.ts +7 -0
- package/dist/tools/Tool.d.ts.map +1 -1
- package/dist/tools/Tool.js +12 -3
- package/dist/tools/Tool.js.map +1 -1
- package/dist/tools/memory.d.ts +26 -0
- package/dist/tools/memory.d.ts.map +1 -0
- package/dist/tools/memory.js +51 -0
- package/dist/tools/memory.js.map +1 -0
- package/dist/types/index.d.ts +238 -9
- package/dist/types/index.d.ts.map +1 -1
- package/dist/types/index.js +4 -0
- package/dist/types/index.js.map +1 -1
- package/dist/types/telemetry.d.ts +107 -0
- package/dist/types/telemetry.d.ts.map +1 -1
- package/guides/AGENTS.md +173 -0
- package/guides/README.md +12 -0
- package/guides/TOOLS.md +93 -27
- package/package.json +25 -4
- package/dist/agents/LLMAgent.d.ts +0 -11
- package/dist/agents/LLMAgent.d.ts.map +0 -1
- package/dist/agents/LLMAgent.js +0 -31
- package/dist/agents/LLMAgent.js.map +0 -1
|
@@ -0,0 +1,49 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Default agent state controller extracted from Runtime.
|
|
3
|
+
*
|
|
4
|
+
* Handles:
|
|
5
|
+
* - Active agent resolution with fallback
|
|
6
|
+
* - Handoff recording in session.handoffHistory + session.metadata.handoffHistory
|
|
7
|
+
* - Per-agent state updates in session.agentStates
|
|
8
|
+
* - Circular handoff detection
|
|
9
|
+
*/
|
|
10
|
+
export class DefaultAgentStateController {
|
|
11
|
+
getActiveAgent(session, fallbackAgentId) {
|
|
12
|
+
return session.activeAgentId ?? session.currentAgent ?? fallbackAgentId;
|
|
13
|
+
}
|
|
14
|
+
setActiveAgent(session, agentId) {
|
|
15
|
+
session.activeAgentId = agentId;
|
|
16
|
+
session.currentAgent = agentId;
|
|
17
|
+
}
|
|
18
|
+
recordHandoff(args) {
|
|
19
|
+
const record = {
|
|
20
|
+
from: args.fromAgentId,
|
|
21
|
+
to: args.toAgentId,
|
|
22
|
+
reason: args.reason,
|
|
23
|
+
timestamp: new Date(),
|
|
24
|
+
};
|
|
25
|
+
args.session.handoffHistory.push(record);
|
|
26
|
+
if (args.session.metadata) {
|
|
27
|
+
args.session.metadata.handoffHistory.push(record);
|
|
28
|
+
}
|
|
29
|
+
}
|
|
30
|
+
updateAgentState(session, agentId, state) {
|
|
31
|
+
const existing = session.agentStates[agentId];
|
|
32
|
+
if (existing) {
|
|
33
|
+
existing.state = { ...existing.state, ...state };
|
|
34
|
+
existing.lastActive = new Date();
|
|
35
|
+
}
|
|
36
|
+
else {
|
|
37
|
+
session.agentStates[agentId] = {
|
|
38
|
+
agentId,
|
|
39
|
+
state,
|
|
40
|
+
lastActive: new Date(),
|
|
41
|
+
};
|
|
42
|
+
}
|
|
43
|
+
}
|
|
44
|
+
isCircularHandoff(handoffStack, agentId, maxVisits = 2) {
|
|
45
|
+
const priorVisits = handoffStack.filter(id => id === agentId).length;
|
|
46
|
+
return priorVisits >= maxVisits;
|
|
47
|
+
}
|
|
48
|
+
}
|
|
49
|
+
//# sourceMappingURL=DefaultAgentStateController.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"DefaultAgentStateController.js","sourceRoot":"","sources":["../../src/foundation/DefaultAgentStateController.ts"],"names":[],"mappings":"AAGA;;;;;;;;GAQG;AACH,MAAM,OAAO,2BAA2B;IACtC,cAAc,CAAC,OAAgB,EAAE,eAAuB;QACtD,OAAO,OAAO,CAAC,aAAa,IAAI,OAAO,CAAC,YAAY,IAAI,eAAe,CAAC;IAC1E,CAAC;IAED,cAAc,CAAC,OAAgB,EAAE,OAAe;QAC9C,OAAO,CAAC,aAAa,GAAG,OAAO,CAAC;QAChC,OAAO,CAAC,YAAY,GAAG,OAAO,CAAC;IACjC,CAAC;IAED,aAAa,CAAC,IAKb;QACC,MAAM,MAAM,GAAkB;YAC5B,IAAI,EAAE,IAAI,CAAC,WAAW;YACtB,EAAE,EAAE,IAAI,CAAC,SAAS;YAClB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,SAAS,EAAE,IAAI,IAAI,EAAE;SACtB,CAAC;QAEF,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAEzC,IAAI,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC;YAC1B,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACpD,CAAC;IACH,CAAC;IAED,gBAAgB,CAAC,OAAgB,EAAE,OAAe,EAAE,KAA8B;QAChF,MAAM,QAAQ,GAAG,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;QAC9C,IAAI,QAAQ,EAAE,CAAC;YACb,QAAQ,CAAC,KAAK,GAAG,EAAE,GAAG,QAAQ,CAAC,KAAK,EAAE,GAAG,KAAK,EAAE,CAAC;YACjD,QAAQ,CAAC,UAAU,GAAG,IAAI,IAAI,EAAE,CAAC;QACnC,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,GAAG;gBAC7B,OAAO;gBACP,KAAK;gBACL,UAAU,EAAE,IAAI,IAAI,EAAE;aACvB,CAAC;QACJ,CAAC;IACH,CAAC;IAED,iBAAiB,CAAC,YAAsB,EAAE,OAAe,EAAE,YAAoB,CAAC;QAC9E,MAAM,WAAW,GAAG,YAAY,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,KAAK,OAAO,CAAC,CAAC,MAAM,CAAC;QACrE,OAAO,WAAW,IAAI,SAAS,CAAC;IAClC,CAAC;CACF"}
|
|
@@ -0,0 +1,28 @@
|
|
|
1
|
+
import type { HarnessStreamPart, RunContext, Session } from '../types/index.js';
|
|
2
|
+
import type { SessionStore } from '../session/SessionStore.js';
|
|
3
|
+
import type { ConversationEventLog } from './ConversationEventLog.js';
|
|
4
|
+
export interface DefaultConversationEventLogConfig {
|
|
5
|
+
sessionStore: SessionStore;
|
|
6
|
+
}
|
|
7
|
+
/**
|
|
8
|
+
* Default conversation event log extracted from SessionEventManager + Runtime checkpoint logic.
|
|
9
|
+
*
|
|
10
|
+
* Handles:
|
|
11
|
+
* - Recording stream parts into session working memory
|
|
12
|
+
* - Text-delta accumulation and flush on terminal events
|
|
13
|
+
* - Value truncation at depth 5 for safe serialization
|
|
14
|
+
* - Checkpoint determination and persistence
|
|
15
|
+
*/
|
|
16
|
+
export declare class DefaultConversationEventLog implements ConversationEventLog {
|
|
17
|
+
private sessionStore;
|
|
18
|
+
constructor(config: DefaultConversationEventLogConfig);
|
|
19
|
+
record(context: RunContext, part: HarnessStreamPart): void;
|
|
20
|
+
checkpoint(session: Session): Promise<void>;
|
|
21
|
+
shouldCheckpoint(part: HarnessStreamPart): boolean;
|
|
22
|
+
cleanup(session: Session): void;
|
|
23
|
+
private getSessionTurn;
|
|
24
|
+
private appendEvent;
|
|
25
|
+
private flushAssistantText;
|
|
26
|
+
private toEventLogValue;
|
|
27
|
+
}
|
|
28
|
+
//# sourceMappingURL=DefaultConversationEventLog.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"DefaultConversationEventLog.d.ts","sourceRoot":"","sources":["../../src/foundation/DefaultConversationEventLog.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,iBAAiB,EAAE,UAAU,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAC;AAChF,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAC;AAC/D,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,2BAA2B,CAAC;AAmBtE,MAAM,WAAW,iCAAiC;IAChD,YAAY,EAAE,YAAY,CAAC;CAC5B;AAED;;;;;;;;GAQG;AACH,qBAAa,2BAA4B,YAAW,oBAAoB;IACtE,OAAO,CAAC,YAAY,CAAe;gBAEvB,MAAM,EAAE,iCAAiC;IAIrD,MAAM,CAAC,OAAO,EAAE,UAAU,EAAE,IAAI,EAAE,iBAAiB,GAAG,IAAI;IAiFpD,UAAU,CAAC,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC;IASjD,gBAAgB,CAAC,IAAI,EAAE,iBAAiB,GAAG,OAAO;IAIlD,OAAO,CAAC,OAAO,EAAE,OAAO,GAAG,IAAI;IAM/B,OAAO,CAAC,cAAc;IAKtB,OAAO,CAAC,WAAW;IAqBnB,OAAO,CAAC,kBAAkB;IAkB1B,OAAO,CAAC,eAAe;CA4BxB"}
|
|
@@ -0,0 +1,195 @@
|
|
|
1
|
+
import crypto from 'node:crypto';
|
|
2
|
+
import { isRecord } from '../utils/isRecord.js';
|
|
3
|
+
/** Working memory key for the runtime event log array. */
|
|
4
|
+
const EVENT_LOG_KEY = 'runtimeEventLog';
|
|
5
|
+
/** Working memory key for accumulated assistant text between flushes. */
|
|
6
|
+
const ASSISTANT_TEXT_KEY = '__ariaAssistantText';
|
|
7
|
+
/** Working memory key for turn tracking. */
|
|
8
|
+
const SESSION_TURN_KEY = '__ariaSessionTurn';
|
|
9
|
+
/** Max event entries before oldest are evicted. */
|
|
10
|
+
const MAX_ENTRIES = 2000;
|
|
11
|
+
/** Stream part types that trigger a checkpoint save. */
|
|
12
|
+
const CHECKPOINT_TYPES = new Set([
|
|
13
|
+
'tool-result',
|
|
14
|
+
'tool-error',
|
|
15
|
+
'flow-transition',
|
|
16
|
+
]);
|
|
17
|
+
/**
|
|
18
|
+
* Default conversation event log extracted from SessionEventManager + Runtime checkpoint logic.
|
|
19
|
+
*
|
|
20
|
+
* Handles:
|
|
21
|
+
* - Recording stream parts into session working memory
|
|
22
|
+
* - Text-delta accumulation and flush on terminal events
|
|
23
|
+
* - Value truncation at depth 5 for safe serialization
|
|
24
|
+
* - Checkpoint determination and persistence
|
|
25
|
+
*/
|
|
26
|
+
export class DefaultConversationEventLog {
|
|
27
|
+
sessionStore;
|
|
28
|
+
constructor(config) {
|
|
29
|
+
this.sessionStore = config.sessionStore;
|
|
30
|
+
}
|
|
31
|
+
record(context, part) {
|
|
32
|
+
// Fast path: text-delta only accumulates text — no UUID, no base object
|
|
33
|
+
if (part.type === 'text-delta') {
|
|
34
|
+
const prevRaw = context.session.workingMemory[ASSISTANT_TEXT_KEY];
|
|
35
|
+
const prev = typeof prevRaw === 'string' ? prevRaw : '';
|
|
36
|
+
context.session.workingMemory[ASSISTANT_TEXT_KEY] = prev + part.text;
|
|
37
|
+
return;
|
|
38
|
+
}
|
|
39
|
+
const base = {
|
|
40
|
+
id: crypto.randomUUID(),
|
|
41
|
+
sessionId: context.session.id,
|
|
42
|
+
agentId: context.agentId,
|
|
43
|
+
turn: this.getSessionTurn(context.session),
|
|
44
|
+
timestamp: new Date().toISOString(),
|
|
45
|
+
};
|
|
46
|
+
switch (part.type) {
|
|
47
|
+
case 'input':
|
|
48
|
+
this.appendEvent(context, { ...base, type: 'user', text: part.text, userId: part.userId });
|
|
49
|
+
return;
|
|
50
|
+
case 'tool-call':
|
|
51
|
+
this.appendEvent(context, {
|
|
52
|
+
...base,
|
|
53
|
+
type: 'tool_call',
|
|
54
|
+
toolCallId: part.toolCallId,
|
|
55
|
+
toolName: part.toolName,
|
|
56
|
+
args: this.toEventLogValue(part.args),
|
|
57
|
+
});
|
|
58
|
+
return;
|
|
59
|
+
case 'tool-result':
|
|
60
|
+
this.appendEvent(context, {
|
|
61
|
+
...base,
|
|
62
|
+
type: 'tool_result',
|
|
63
|
+
toolCallId: part.toolCallId,
|
|
64
|
+
toolName: part.toolName,
|
|
65
|
+
result: this.toEventLogValue(part.result),
|
|
66
|
+
});
|
|
67
|
+
return;
|
|
68
|
+
case 'tool-error':
|
|
69
|
+
this.appendEvent(context, {
|
|
70
|
+
...base,
|
|
71
|
+
type: 'tool_error',
|
|
72
|
+
toolCallId: part.toolCallId,
|
|
73
|
+
toolName: part.toolName,
|
|
74
|
+
error: part.error,
|
|
75
|
+
});
|
|
76
|
+
return;
|
|
77
|
+
case 'flow-transition':
|
|
78
|
+
this.appendEvent(context, {
|
|
79
|
+
...base,
|
|
80
|
+
type: 'transition',
|
|
81
|
+
kind: 'flow',
|
|
82
|
+
from: part.from,
|
|
83
|
+
to: part.to,
|
|
84
|
+
});
|
|
85
|
+
return;
|
|
86
|
+
case 'handoff':
|
|
87
|
+
this.appendEvent(context, {
|
|
88
|
+
...base,
|
|
89
|
+
type: 'transition',
|
|
90
|
+
kind: 'handoff',
|
|
91
|
+
from: part.from,
|
|
92
|
+
to: part.to,
|
|
93
|
+
reason: part.reason,
|
|
94
|
+
});
|
|
95
|
+
return;
|
|
96
|
+
case 'turn-end':
|
|
97
|
+
this.flushAssistantText(context, 'turn-end');
|
|
98
|
+
return;
|
|
99
|
+
case 'done':
|
|
100
|
+
this.flushAssistantText(context, 'done');
|
|
101
|
+
return;
|
|
102
|
+
case 'error':
|
|
103
|
+
this.flushAssistantText(context, 'error');
|
|
104
|
+
return;
|
|
105
|
+
default:
|
|
106
|
+
return;
|
|
107
|
+
}
|
|
108
|
+
}
|
|
109
|
+
async checkpoint(session) {
|
|
110
|
+
const now = new Date();
|
|
111
|
+
session.updatedAt = now;
|
|
112
|
+
if (session.metadata) {
|
|
113
|
+
session.metadata.lastActiveAt = now;
|
|
114
|
+
}
|
|
115
|
+
await this.sessionStore.save(session);
|
|
116
|
+
}
|
|
117
|
+
shouldCheckpoint(part) {
|
|
118
|
+
return CHECKPOINT_TYPES.has(part.type);
|
|
119
|
+
}
|
|
120
|
+
cleanup(session) {
|
|
121
|
+
delete session.workingMemory[ASSISTANT_TEXT_KEY];
|
|
122
|
+
}
|
|
123
|
+
// --- Private helpers (extracted from SessionEventManager) ---
|
|
124
|
+
getSessionTurn(session) {
|
|
125
|
+
const value = session.workingMemory[SESSION_TURN_KEY];
|
|
126
|
+
return typeof value === 'number' && Number.isFinite(value) ? value : 0;
|
|
127
|
+
}
|
|
128
|
+
appendEvent(context, entry) {
|
|
129
|
+
const current = context.session.workingMemory[EVENT_LOG_KEY];
|
|
130
|
+
let events;
|
|
131
|
+
if (Array.isArray(current)) {
|
|
132
|
+
events = current;
|
|
133
|
+
}
|
|
134
|
+
else {
|
|
135
|
+
events = [];
|
|
136
|
+
context.session.workingMemory[EVENT_LOG_KEY] = events;
|
|
137
|
+
}
|
|
138
|
+
events.push(entry);
|
|
139
|
+
if (events.length > MAX_ENTRIES) {
|
|
140
|
+
events.splice(0, events.length - MAX_ENTRIES);
|
|
141
|
+
}
|
|
142
|
+
// Touch session
|
|
143
|
+
const now = new Date();
|
|
144
|
+
context.session.updatedAt = now;
|
|
145
|
+
if (context.session.metadata) {
|
|
146
|
+
context.session.metadata.lastActiveAt = now;
|
|
147
|
+
}
|
|
148
|
+
}
|
|
149
|
+
flushAssistantText(context, trigger) {
|
|
150
|
+
const textRaw = context.session.workingMemory[ASSISTANT_TEXT_KEY];
|
|
151
|
+
const text = (typeof textRaw === 'string' ? textRaw : '').trim();
|
|
152
|
+
if (text.length > 0) {
|
|
153
|
+
this.appendEvent(context, {
|
|
154
|
+
id: crypto.randomUUID(),
|
|
155
|
+
sessionId: context.session.id,
|
|
156
|
+
agentId: context.agentId,
|
|
157
|
+
turn: this.getSessionTurn(context.session),
|
|
158
|
+
type: 'assistant_final',
|
|
159
|
+
trigger,
|
|
160
|
+
text,
|
|
161
|
+
timestamp: new Date().toISOString(),
|
|
162
|
+
});
|
|
163
|
+
}
|
|
164
|
+
delete context.session.workingMemory[ASSISTANT_TEXT_KEY];
|
|
165
|
+
}
|
|
166
|
+
toEventLogValue(value, depth = 0) {
|
|
167
|
+
if (value === null || value === undefined) {
|
|
168
|
+
return value;
|
|
169
|
+
}
|
|
170
|
+
if (depth >= 5) {
|
|
171
|
+
return '[truncated]';
|
|
172
|
+
}
|
|
173
|
+
if (value instanceof Date) {
|
|
174
|
+
return value.toISOString();
|
|
175
|
+
}
|
|
176
|
+
if (value instanceof Error) {
|
|
177
|
+
return { name: value.name, message: value.message };
|
|
178
|
+
}
|
|
179
|
+
if (Array.isArray(value)) {
|
|
180
|
+
return value.slice(0, 50).map(item => this.toEventLogValue(item, depth + 1));
|
|
181
|
+
}
|
|
182
|
+
if (typeof value === 'function') {
|
|
183
|
+
return '[function]';
|
|
184
|
+
}
|
|
185
|
+
if (isRecord(value)) {
|
|
186
|
+
const out = {};
|
|
187
|
+
for (const [k, v] of Object.entries(value).slice(0, 50)) {
|
|
188
|
+
out[k] = this.toEventLogValue(v, depth + 1);
|
|
189
|
+
}
|
|
190
|
+
return out;
|
|
191
|
+
}
|
|
192
|
+
return value;
|
|
193
|
+
}
|
|
194
|
+
}
|
|
195
|
+
//# sourceMappingURL=DefaultConversationEventLog.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"DefaultConversationEventLog.js","sourceRoot":"","sources":["../../src/foundation/DefaultConversationEventLog.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,aAAa,CAAC;AAIjC,OAAO,EAAE,QAAQ,EAAE,MAAM,sBAAsB,CAAC;AAEhD,0DAA0D;AAC1D,MAAM,aAAa,GAAG,iBAAiB,CAAC;AACxC,yEAAyE;AACzE,MAAM,kBAAkB,GAAG,qBAAqB,CAAC;AACjD,4CAA4C;AAC5C,MAAM,gBAAgB,GAAG,mBAAmB,CAAC;AAC7C,mDAAmD;AACnD,MAAM,WAAW,GAAG,IAAI,CAAC;AAEzB,wDAAwD;AACxD,MAAM,gBAAgB,GAAG,IAAI,GAAG,CAA4B;IAC1D,aAAa;IACb,YAAY;IACZ,iBAAiB;CAClB,CAAC,CAAC;AAMH;;;;;;;;GAQG;AACH,MAAM,OAAO,2BAA2B;IAC9B,YAAY,CAAe;IAEnC,YAAY,MAAyC;QACnD,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC,YAAY,CAAC;IAC1C,CAAC;IAED,MAAM,CAAC,OAAmB,EAAE,IAAuB;QACjD,wEAAwE;QACxE,IAAI,IAAI,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;YAC/B,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,aAAa,CAAC,kBAAkB,CAAC,CAAC;YAClE,MAAM,IAAI,GAAG,OAAO,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;YACxD,OAAO,CAAC,OAAO,CAAC,aAAa,CAAC,kBAAkB,CAAC,GAAG,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC;YACrE,OAAO;QACT,CAAC;QAED,MAAM,IAAI,GAAG;YACX,EAAE,EAAE,MAAM,CAAC,UAAU,EAAE;YACvB,SAAS,EAAE,OAAO,CAAC,OAAO,CAAC,EAAE;YAC7B,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,IAAI,EAAE,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,OAAO,CAAC;YAC1C,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACpC,CAAC;QAEF,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;YAClB,KAAK,OAAO;gBACV,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,EAAE,GAAG,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;gBAC3F,OAAO;YACT,KAAK,WAAW;gBACd,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE;oBACxB,GAAG,IAAI;oBACP,IAAI,EAAE,WAAW;oBACjB,UAAU,EAAE,IAAI,CAAC,UAAU;oBAC3B,QAAQ,EAAE,IAAI,CAAC,QAAQ;oBACvB,IAAI,EAAE,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC;iBACtC,CAAC,CAAC;gBACH,OAAO;YACT,KAAK,aAAa;gBAChB,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE;oBACxB,GAAG,IAAI;oBACP,IAAI,EAAE,aAAa;oBACnB,UAAU,EAAE,IAAI,CAAC,UAAU;oBAC3B,QAAQ,EAAE,IAAI,CAAC,QAAQ;oBACvB,MAAM,EAAE,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,MAAM,CAAC;iBAC1C,CAAC,CAAC;gBACH,OAAO;YACT,KAAK,YAAY;gBACf,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE;oBACxB,GAAG,IAAI;oBACP,IAAI,EAAE,YAAY;oBAClB,UAAU,EAAE,IAAI,CAAC,UAAU;oBAC3B,QAAQ,EAAE,IAAI,CAAC,QAAQ;oBACvB,KAAK,EAAE,IAAI,CAAC,KAAK;iBAClB,CAAC,CAAC;gBACH,OAAO;YACT,KAAK,iBAAiB;gBACpB,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE;oBACxB,GAAG,IAAI;oBACP,IAAI,EAAE,YAAY;oBAClB,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,EAAE,EAAE,IAAI,CAAC,EAAE;iBACZ,CAAC,CAAC;gBACH,OAAO;YACT,KAAK,SAAS;gBACZ,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE;oBACxB,GAAG,IAAI;oBACP,IAAI,EAAE,YAAY;oBAClB,IAAI,EAAE,SAAS;oBACf,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,EAAE,EAAE,IAAI,CAAC,EAAE;oBACX,MAAM,EAAE,IAAI,CAAC,MAAM;iBACpB,CAAC,CAAC;gBACH,OAAO;YACT,KAAK,UAAU;gBACb,IAAI,CAAC,kBAAkB,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;gBAC7C,OAAO;YACT,KAAK,MAAM;gBACT,IAAI,CAAC,kBAAkB,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;gBACzC,OAAO;YACT,KAAK,OAAO;gBACV,IAAI,CAAC,kBAAkB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;gBAC1C,OAAO;YACT;gBACE,OAAO;QACX,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,OAAgB;QAC/B,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,OAAO,CAAC,SAAS,GAAG,GAAG,CAAC;QACxB,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;YACrB,OAAO,CAAC,QAAQ,CAAC,YAAY,GAAG,GAAG,CAAC;QACtC,CAAC;QACD,MAAM,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACxC,CAAC;IAED,gBAAgB,CAAC,IAAuB;QACtC,OAAO,gBAAgB,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACzC,CAAC;IAED,OAAO,CAAC,OAAgB;QACtB,OAAO,OAAO,CAAC,aAAa,CAAC,kBAAkB,CAAC,CAAC;IACnD,CAAC;IAED,+DAA+D;IAEvD,cAAc,CAAC,OAAgB;QACrC,MAAM,KAAK,GAAG,OAAO,CAAC,aAAa,CAAC,gBAAgB,CAAC,CAAC;QACtD,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;IACzE,CAAC;IAEO,WAAW,CAAC,OAAmB,EAAE,KAA8B;QACrE,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,aAAa,CAAC,aAAa,CAAC,CAAC;QAC7D,IAAI,MAAiB,CAAC;QACtB,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;YAC3B,MAAM,GAAG,OAAO,CAAC;QACnB,CAAC;aAAM,CAAC;YACN,MAAM,GAAG,EAAE,CAAC;YACZ,OAAO,CAAC,OAAO,CAAC,aAAa,CAAC,aAAa,CAAC,GAAG,MAAM,CAAC;QACxD,CAAC;QACD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnB,IAAI,MAAM,CAAC,MAAM,GAAG,WAAW,EAAE,CAAC;YAChC,MAAM,CAAC,MAAM,CAAC,CAAC,EAAE,MAAM,CAAC,MAAM,GAAG,WAAW,CAAC,CAAC;QAChD,CAAC;QACD,gBAAgB;QAChB,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,OAAO,CAAC,OAAO,CAAC,SAAS,GAAG,GAAG,CAAC;QAChC,IAAI,OAAO,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC;YAC7B,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,YAAY,GAAG,GAAG,CAAC;QAC9C,CAAC;IACH,CAAC;IAEO,kBAAkB,CAAC,OAAmB,EAAE,OAAsC;QACpF,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,aAAa,CAAC,kBAAkB,CAAC,CAAC;QAClE,MAAM,IAAI,GAAG,CAAC,OAAO,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QACjE,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACpB,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE;gBACxB,EAAE,EAAE,MAAM,CAAC,UAAU,EAAE;gBACvB,SAAS,EAAE,OAAO,CAAC,OAAO,CAAC,EAAE;gBAC7B,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,IAAI,EAAE,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,OAAO,CAAC;gBAC1C,IAAI,EAAE,iBAAiB;gBACvB,OAAO;gBACP,IAAI;gBACJ,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;aACpC,CAAC,CAAC;QACL,CAAC;QACD,OAAO,OAAO,CAAC,OAAO,CAAC,aAAa,CAAC,kBAAkB,CAAC,CAAC;IAC3D,CAAC;IAEO,eAAe,CAAC,KAAc,EAAE,QAAgB,CAAC;QACvD,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YAC1C,OAAO,KAAK,CAAC;QACf,CAAC;QACD,IAAI,KAAK,IAAI,CAAC,EAAE,CAAC;YACf,OAAO,aAAa,CAAC;QACvB,CAAC;QACD,IAAI,KAAK,YAAY,IAAI,EAAE,CAAC;YAC1B,OAAO,KAAK,CAAC,WAAW,EAAE,CAAC;QAC7B,CAAC;QACD,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;YAC3B,OAAO,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC;QACtD,CAAC;QACD,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YACzB,OAAO,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,IAAI,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC;QAC/E,CAAC;QACD,IAAI,OAAO,KAAK,KAAK,UAAU,EAAE,CAAC;YAChC,OAAO,YAAY,CAAC;QACtB,CAAC;QACD,IAAI,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YACpB,MAAM,GAAG,GAA4B,EAAE,CAAC;YACxC,KAAK,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC;gBACxD,GAAG,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,eAAe,CAAC,CAAC,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC;YAC9C,CAAC;YACD,OAAO,GAAG,CAAC;QACb,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;CACF"}
|
|
@@ -0,0 +1,34 @@
|
|
|
1
|
+
import type { ModelMessage } from 'ai';
|
|
2
|
+
import type { Session, WorkingMemory } from '../types/index.js';
|
|
3
|
+
import type { SessionStore } from '../session/SessionStore.js';
|
|
4
|
+
import type { ConversationState } from './ConversationState.js';
|
|
5
|
+
export interface DefaultConversationStateConfig {
|
|
6
|
+
sessionStore: SessionStore;
|
|
7
|
+
defaultAgentId: string;
|
|
8
|
+
}
|
|
9
|
+
/**
|
|
10
|
+
* Default conversation state implementation extracted from Runtime.
|
|
11
|
+
*
|
|
12
|
+
* Handles:
|
|
13
|
+
* - Session load/create/save/delete via SessionStore
|
|
14
|
+
* - Message appending with normalization
|
|
15
|
+
* - Turn counting
|
|
16
|
+
* - Working memory access
|
|
17
|
+
*/
|
|
18
|
+
export declare class DefaultConversationState implements ConversationState {
|
|
19
|
+
private sessionStore;
|
|
20
|
+
private defaultAgentId;
|
|
21
|
+
constructor(config: DefaultConversationStateConfig);
|
|
22
|
+
load(sessionId: string, userId?: string): Promise<Session>;
|
|
23
|
+
save(session: Session): Promise<void>;
|
|
24
|
+
delete(sessionId: string): Promise<void>;
|
|
25
|
+
workingMemory(session: Session): WorkingMemory;
|
|
26
|
+
appendUserMessage(session: Session, text: string): void;
|
|
27
|
+
appendAssistantMessage(session: Session, text: string): void;
|
|
28
|
+
appendMessage(session: Session, message: ModelMessage): void;
|
|
29
|
+
getSessionTurn(session: Session): number;
|
|
30
|
+
bumpSessionTurn(session: Session): number;
|
|
31
|
+
touchSession(session: Session): void;
|
|
32
|
+
createSession(id: string, defaultAgentId: string, userId?: string): Session;
|
|
33
|
+
}
|
|
34
|
+
//# sourceMappingURL=DefaultConversationState.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"DefaultConversationState.d.ts","sourceRoot":"","sources":["../../src/foundation/DefaultConversationState.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,IAAI,CAAC;AACvC,OAAO,KAAK,EAAE,OAAO,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAChE,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAC;AAC/D,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAOhE,MAAM,WAAW,8BAA8B;IAC7C,YAAY,EAAE,YAAY,CAAC;IAC3B,cAAc,EAAE,MAAM,CAAC;CACxB;AAED;;;;;;;;GAQG;AACH,qBAAa,wBAAyB,YAAW,iBAAiB;IAChE,OAAO,CAAC,YAAY,CAAe;IACnC,OAAO,CAAC,cAAc,CAAS;gBAEnB,MAAM,EAAE,8BAA8B;IAK5C,IAAI,CAAC,SAAS,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAQ1D,IAAI,CAAC,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC;IAKrC,MAAM,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAI9C,aAAa,CAAC,OAAO,EAAE,OAAO,GAAG,aAAa;IAI9C,iBAAiB,CAAC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,GAAG,IAAI;IAQvD,sBAAsB,CAAC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,GAAG,IAAI;IAQ5D,aAAa,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,YAAY,GAAG,IAAI;IAQ5D,cAAc,CAAC,OAAO,EAAE,OAAO,GAAG,MAAM;IAKxC,eAAe,CAAC,OAAO,EAAE,OAAO,GAAG,MAAM;IAOzC,YAAY,CAAC,OAAO,EAAE,OAAO,GAAG,IAAI;IAQpC,aAAa,CAAC,EAAE,EAAE,MAAM,EAAE,cAAc,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO;CAuB5E"}
|
|
@@ -0,0 +1,100 @@
|
|
|
1
|
+
import { SessionWorkingMemory } from '../runtime/WorkingMemory.js';
|
|
2
|
+
import { normalizeModelMessage } from '../utils/messageNormalization.js';
|
|
3
|
+
/** Internal working memory key for turn tracking. */
|
|
4
|
+
const SESSION_TURN_KEY = '__ariaSessionTurn';
|
|
5
|
+
/**
|
|
6
|
+
* Default conversation state implementation extracted from Runtime.
|
|
7
|
+
*
|
|
8
|
+
* Handles:
|
|
9
|
+
* - Session load/create/save/delete via SessionStore
|
|
10
|
+
* - Message appending with normalization
|
|
11
|
+
* - Turn counting
|
|
12
|
+
* - Working memory access
|
|
13
|
+
*/
|
|
14
|
+
export class DefaultConversationState {
|
|
15
|
+
sessionStore;
|
|
16
|
+
defaultAgentId;
|
|
17
|
+
constructor(config) {
|
|
18
|
+
this.sessionStore = config.sessionStore;
|
|
19
|
+
this.defaultAgentId = config.defaultAgentId;
|
|
20
|
+
}
|
|
21
|
+
async load(sessionId, userId) {
|
|
22
|
+
const existing = await this.sessionStore.get(sessionId);
|
|
23
|
+
if (existing) {
|
|
24
|
+
return existing;
|
|
25
|
+
}
|
|
26
|
+
return this.createSession(sessionId, this.defaultAgentId, userId);
|
|
27
|
+
}
|
|
28
|
+
async save(session) {
|
|
29
|
+
this.touchSession(session);
|
|
30
|
+
await this.sessionStore.save(session);
|
|
31
|
+
}
|
|
32
|
+
async delete(sessionId) {
|
|
33
|
+
await this.sessionStore.delete(sessionId);
|
|
34
|
+
}
|
|
35
|
+
workingMemory(session) {
|
|
36
|
+
return new SessionWorkingMemory(session);
|
|
37
|
+
}
|
|
38
|
+
appendUserMessage(session, text) {
|
|
39
|
+
const message = {
|
|
40
|
+
role: 'user',
|
|
41
|
+
content: [{ type: 'text', text }],
|
|
42
|
+
};
|
|
43
|
+
this.appendMessage(session, message);
|
|
44
|
+
}
|
|
45
|
+
appendAssistantMessage(session, text) {
|
|
46
|
+
const message = {
|
|
47
|
+
role: 'assistant',
|
|
48
|
+
content: [{ type: 'text', text }],
|
|
49
|
+
};
|
|
50
|
+
this.appendMessage(session, message);
|
|
51
|
+
}
|
|
52
|
+
appendMessage(session, message) {
|
|
53
|
+
const normalized = normalizeModelMessage(message);
|
|
54
|
+
if (normalized) {
|
|
55
|
+
session.messages.push(normalized);
|
|
56
|
+
this.touchSession(session);
|
|
57
|
+
}
|
|
58
|
+
}
|
|
59
|
+
getSessionTurn(session) {
|
|
60
|
+
const value = session.workingMemory[SESSION_TURN_KEY];
|
|
61
|
+
return typeof value === 'number' && Number.isFinite(value) ? value : 0;
|
|
62
|
+
}
|
|
63
|
+
bumpSessionTurn(session) {
|
|
64
|
+
const next = this.getSessionTurn(session) + 1;
|
|
65
|
+
session.workingMemory[SESSION_TURN_KEY] = next;
|
|
66
|
+
this.touchSession(session);
|
|
67
|
+
return next;
|
|
68
|
+
}
|
|
69
|
+
touchSession(session) {
|
|
70
|
+
const now = new Date();
|
|
71
|
+
session.updatedAt = now;
|
|
72
|
+
if (session.metadata) {
|
|
73
|
+
session.metadata.lastActiveAt = now;
|
|
74
|
+
}
|
|
75
|
+
}
|
|
76
|
+
createSession(id, defaultAgentId, userId) {
|
|
77
|
+
const now = new Date();
|
|
78
|
+
return {
|
|
79
|
+
id,
|
|
80
|
+
userId,
|
|
81
|
+
messages: [],
|
|
82
|
+
createdAt: now,
|
|
83
|
+
updatedAt: now,
|
|
84
|
+
workingMemory: {},
|
|
85
|
+
currentAgent: defaultAgentId,
|
|
86
|
+
activeAgentId: defaultAgentId,
|
|
87
|
+
state: {},
|
|
88
|
+
metadata: {
|
|
89
|
+
createdAt: now,
|
|
90
|
+
lastActiveAt: now,
|
|
91
|
+
totalTokens: 0,
|
|
92
|
+
totalSteps: 0,
|
|
93
|
+
handoffHistory: [],
|
|
94
|
+
},
|
|
95
|
+
agentStates: {},
|
|
96
|
+
handoffHistory: [],
|
|
97
|
+
};
|
|
98
|
+
}
|
|
99
|
+
}
|
|
100
|
+
//# sourceMappingURL=DefaultConversationState.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"DefaultConversationState.js","sourceRoot":"","sources":["../../src/foundation/DefaultConversationState.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,oBAAoB,EAAE,MAAM,6BAA6B,CAAC;AACnE,OAAO,EAAE,qBAAqB,EAAE,MAAM,kCAAkC,CAAC;AAEzE,qDAAqD;AACrD,MAAM,gBAAgB,GAAG,mBAAmB,CAAC;AAO7C;;;;;;;;GAQG;AACH,MAAM,OAAO,wBAAwB;IAC3B,YAAY,CAAe;IAC3B,cAAc,CAAS;IAE/B,YAAY,MAAsC;QAChD,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC,YAAY,CAAC;QACxC,IAAI,CAAC,cAAc,GAAG,MAAM,CAAC,cAAc,CAAC;IAC9C,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,SAAiB,EAAE,MAAe;QAC3C,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACxD,IAAI,QAAQ,EAAE,CAAC;YACb,OAAO,QAAQ,CAAC;QAClB,CAAC;QACD,OAAO,IAAI,CAAC,aAAa,CAAC,SAAS,EAAE,IAAI,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC;IACpE,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,OAAgB;QACzB,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAC3B,MAAM,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACxC,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,SAAiB;QAC5B,MAAM,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;IAC5C,CAAC;IAED,aAAa,CAAC,OAAgB;QAC5B,OAAO,IAAI,oBAAoB,CAAC,OAAO,CAAC,CAAC;IAC3C,CAAC;IAED,iBAAiB,CAAC,OAAgB,EAAE,IAAY;QAC9C,MAAM,OAAO,GAAiB;YAC5B,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;SAClC,CAAC;QACF,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IACvC,CAAC;IAED,sBAAsB,CAAC,OAAgB,EAAE,IAAY;QACnD,MAAM,OAAO,GAAiB;YAC5B,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;SAClC,CAAC;QACF,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IACvC,CAAC;IAED,aAAa,CAAC,OAAgB,EAAE,OAAqB;QACnD,MAAM,UAAU,GAAG,qBAAqB,CAAC,OAAO,CAAC,CAAC;QAClD,IAAI,UAAU,EAAE,CAAC;YACf,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAClC,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAC7B,CAAC;IACH,CAAC;IAED,cAAc,CAAC,OAAgB;QAC7B,MAAM,KAAK,GAAG,OAAO,CAAC,aAAa,CAAC,gBAAgB,CAAC,CAAC;QACtD,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;IACzE,CAAC;IAED,eAAe,CAAC,OAAgB;QAC9B,MAAM,IAAI,GAAG,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAC9C,OAAO,CAAC,aAAa,CAAC,gBAAgB,CAAC,GAAG,IAAI,CAAC;QAC/C,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAC3B,OAAO,IAAI,CAAC;IACd,CAAC;IAED,YAAY,CAAC,OAAgB;QAC3B,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,OAAO,CAAC,SAAS,GAAG,GAAG,CAAC;QACxB,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;YACrB,OAAO,CAAC,QAAQ,CAAC,YAAY,GAAG,GAAG,CAAC;QACtC,CAAC;IACH,CAAC;IAED,aAAa,CAAC,EAAU,EAAE,cAAsB,EAAE,MAAe;QAC/D,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,OAAO;YACL,EAAE;YACF,MAAM;YACN,QAAQ,EAAE,EAAE;YACZ,SAAS,EAAE,GAAG;YACd,SAAS,EAAE,GAAG;YACd,aAAa,EAAE,EAAE;YACjB,YAAY,EAAE,cAAc;YAC5B,aAAa,EAAE,cAAc;YAC7B,KAAK,EAAE,EAAE;YACT,QAAQ,EAAE;gBACR,SAAS,EAAE,GAAG;gBACd,YAAY,EAAE,GAAG;gBACjB,WAAW,EAAE,CAAC;gBACd,UAAU,EAAE,CAAC;gBACb,cAAc,EAAE,EAAE;aACnB;YACD,WAAW,EAAE,EAAE;YACf,cAAc,EAAE,EAAE;SACnB,CAAC;IACJ,CAAC;CACF"}
|
|
@@ -0,0 +1,58 @@
|
|
|
1
|
+
import type { Session, ToolCallRecord } from '../types/index.js';
|
|
2
|
+
import type { ToolExecutor, ExecutableTool } from './ToolExecutor.js';
|
|
3
|
+
import type { ToolEnforcer } from '../guards/ToolEnforcer.js';
|
|
4
|
+
import type { HookRunner } from '../hooks/HookRunner.js';
|
|
5
|
+
import type { MemoryService } from '../memory/MemoryService.js';
|
|
6
|
+
export interface DefaultToolExecutorConfig {
|
|
7
|
+
enforcer: ToolEnforcer;
|
|
8
|
+
hookRunner: HookRunner;
|
|
9
|
+
memoryService?: MemoryService;
|
|
10
|
+
/** Default timeout in milliseconds for tool execution. Defaults to 30000 (30s). */
|
|
11
|
+
defaultToolTimeoutMs?: number;
|
|
12
|
+
}
|
|
13
|
+
/**
|
|
14
|
+
* Error thrown when a tool execution exceeds the configured timeout.
|
|
15
|
+
*/
|
|
16
|
+
export declare class ToolTimeoutError extends Error {
|
|
17
|
+
readonly toolName: string;
|
|
18
|
+
readonly timeoutMs: number;
|
|
19
|
+
constructor(toolName: string, timeoutMs: number);
|
|
20
|
+
}
|
|
21
|
+
/**
|
|
22
|
+
* Default tool executor extracted from Runtime.wrapToolsWithEnforcement().
|
|
23
|
+
*
|
|
24
|
+
* Handles:
|
|
25
|
+
* - Enforcement checks via ToolEnforcer
|
|
26
|
+
* - Idempotency key generation
|
|
27
|
+
* - Context enrichment (experimental_context)
|
|
28
|
+
* - Error propagation via HookRunner
|
|
29
|
+
*/
|
|
30
|
+
export declare class DefaultToolExecutor implements ToolExecutor {
|
|
31
|
+
private enforcer;
|
|
32
|
+
private hookRunner;
|
|
33
|
+
private memoryService?;
|
|
34
|
+
private defaultToolTimeoutMs?;
|
|
35
|
+
constructor(config: DefaultToolExecutorConfig);
|
|
36
|
+
execute(args: {
|
|
37
|
+
session: Session;
|
|
38
|
+
userId?: string;
|
|
39
|
+
agentId: string;
|
|
40
|
+
toolName: string;
|
|
41
|
+
tool: ExecutableTool;
|
|
42
|
+
input: unknown;
|
|
43
|
+
toolCallId?: string;
|
|
44
|
+
abortSignal?: AbortSignal;
|
|
45
|
+
step?: number;
|
|
46
|
+
turn?: number;
|
|
47
|
+
toolCallHistory?: ToolCallRecord[];
|
|
48
|
+
}): Promise<unknown>;
|
|
49
|
+
buildIdempotencyKey(args: {
|
|
50
|
+
sessionId: string;
|
|
51
|
+
agentId: string;
|
|
52
|
+
step: number;
|
|
53
|
+
toolName: string;
|
|
54
|
+
toolCallId: string;
|
|
55
|
+
}): string;
|
|
56
|
+
private withToolExecutionMetadata;
|
|
57
|
+
}
|
|
58
|
+
//# sourceMappingURL=DefaultToolExecutor.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"DefaultToolExecutor.d.ts","sourceRoot":"","sources":["../../src/foundation/DefaultToolExecutor.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,OAAO,EAAE,cAAc,EAAsB,MAAM,mBAAmB,CAAC;AACrF,OAAO,KAAK,EAAE,YAAY,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AACtE,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,2BAA2B,CAAC;AAC9D,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,4BAA4B,CAAC;AAGhE,MAAM,WAAW,yBAAyB;IACxC,QAAQ,EAAE,YAAY,CAAC;IACvB,UAAU,EAAE,UAAU,CAAC;IACvB,aAAa,CAAC,EAAE,aAAa,CAAC;IAC9B,mFAAmF;IACnF,oBAAoB,CAAC,EAAE,MAAM,CAAC;CAC/B;AAED;;GAEG;AACH,qBAAa,gBAAiB,SAAQ,KAAK;IACzC,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC;IAC1B,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAC;gBAEf,QAAQ,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM;CAMhD;AAED;;;;;;;;GAQG;AACH,qBAAa,mBAAoB,YAAW,YAAY;IACtD,OAAO,CAAC,QAAQ,CAAe;IAC/B,OAAO,CAAC,UAAU,CAAa;IAC/B,OAAO,CAAC,aAAa,CAAC,CAAgB;IACtC,OAAO,CAAC,oBAAoB,CAAC,CAAS;gBAE1B,MAAM,EAAE,yBAAyB;IAOvC,OAAO,CAAC,IAAI,EAAE;QAClB,OAAO,EAAE,OAAO,CAAC;QACjB,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,OAAO,EAAE,MAAM,CAAC;QAChB,QAAQ,EAAE,MAAM,CAAC;QACjB,IAAI,EAAE,cAAc,CAAC;QACrB,KAAK,EAAE,OAAO,CAAC;QACf,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,WAAW,CAAC,EAAE,WAAW,CAAC;QAC1B,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,eAAe,CAAC,EAAE,cAAc,EAAE,CAAC;KACpC,GAAG,OAAO,CAAC,OAAO,CAAC;IAuFpB,mBAAmB,CAAC,IAAI,EAAE;QACxB,SAAS,EAAE,MAAM,CAAC;QAClB,OAAO,EAAE,MAAM,CAAC;QAChB,IAAI,EAAE,MAAM,CAAC;QACb,QAAQ,EAAE,MAAM,CAAC;QACjB,UAAU,EAAE,MAAM,CAAC;KACpB,GAAG,MAAM;IAIV,OAAO,CAAC,yBAAyB;CAmClC"}
|
|
@@ -0,0 +1,128 @@
|
|
|
1
|
+
import crypto from 'node:crypto';
|
|
2
|
+
import { isRecord } from '../utils/isRecord.js';
|
|
3
|
+
/**
|
|
4
|
+
* Error thrown when a tool execution exceeds the configured timeout.
|
|
5
|
+
*/
|
|
6
|
+
export class ToolTimeoutError extends Error {
|
|
7
|
+
toolName;
|
|
8
|
+
timeoutMs;
|
|
9
|
+
constructor(toolName, timeoutMs) {
|
|
10
|
+
super(`Tool "${toolName}" timeout after ${timeoutMs}ms`);
|
|
11
|
+
this.name = 'ToolTimeoutError';
|
|
12
|
+
this.toolName = toolName;
|
|
13
|
+
this.timeoutMs = timeoutMs;
|
|
14
|
+
}
|
|
15
|
+
}
|
|
16
|
+
/**
|
|
17
|
+
* Default tool executor extracted from Runtime.wrapToolsWithEnforcement().
|
|
18
|
+
*
|
|
19
|
+
* Handles:
|
|
20
|
+
* - Enforcement checks via ToolEnforcer
|
|
21
|
+
* - Idempotency key generation
|
|
22
|
+
* - Context enrichment (experimental_context)
|
|
23
|
+
* - Error propagation via HookRunner
|
|
24
|
+
*/
|
|
25
|
+
export class DefaultToolExecutor {
|
|
26
|
+
enforcer;
|
|
27
|
+
hookRunner;
|
|
28
|
+
memoryService;
|
|
29
|
+
defaultToolTimeoutMs;
|
|
30
|
+
constructor(config) {
|
|
31
|
+
this.enforcer = config.enforcer;
|
|
32
|
+
this.hookRunner = config.hookRunner;
|
|
33
|
+
this.memoryService = config.memoryService;
|
|
34
|
+
this.defaultToolTimeoutMs = config.defaultToolTimeoutMs;
|
|
35
|
+
}
|
|
36
|
+
async execute(args) {
|
|
37
|
+
const { session, agentId, toolName, tool, input, step = 0, turn = 0, toolCallHistory = [], } = args;
|
|
38
|
+
const toolCallId = args.toolCallId ?? crypto.randomUUID();
|
|
39
|
+
const idempotencyKey = this.buildIdempotencyKey({
|
|
40
|
+
sessionId: session.id,
|
|
41
|
+
agentId,
|
|
42
|
+
step,
|
|
43
|
+
toolName,
|
|
44
|
+
toolCallId,
|
|
45
|
+
});
|
|
46
|
+
const callRecord = {
|
|
47
|
+
toolCallId,
|
|
48
|
+
toolName,
|
|
49
|
+
args: input,
|
|
50
|
+
idempotencyKey,
|
|
51
|
+
success: true,
|
|
52
|
+
timestamp: Date.now(),
|
|
53
|
+
};
|
|
54
|
+
// Enforcement check
|
|
55
|
+
const enforcement = await this.enforcer.check(callRecord, {
|
|
56
|
+
previousCalls: toolCallHistory,
|
|
57
|
+
currentStep: step,
|
|
58
|
+
sessionState: session.state ?? {},
|
|
59
|
+
});
|
|
60
|
+
if (!enforcement.allowed) {
|
|
61
|
+
const reason = enforcement.reason ?? 'Tool call blocked by enforcement';
|
|
62
|
+
callRecord.success = false;
|
|
63
|
+
callRecord.error = new Error(reason);
|
|
64
|
+
toolCallHistory.push(callRecord);
|
|
65
|
+
const runContext = {
|
|
66
|
+
session,
|
|
67
|
+
agentId,
|
|
68
|
+
stepCount: step,
|
|
69
|
+
totalTokens: 0,
|
|
70
|
+
handoffStack: [],
|
|
71
|
+
startTime: Date.now(),
|
|
72
|
+
consecutiveErrors: 0,
|
|
73
|
+
toolCallHistory,
|
|
74
|
+
};
|
|
75
|
+
await this.hookRunner.onToolError(runContext, callRecord, callRecord.error);
|
|
76
|
+
throw callRecord.error;
|
|
77
|
+
}
|
|
78
|
+
// Build enriched options for the tool's execute function
|
|
79
|
+
if (!('execute' in tool) || typeof tool.execute !== 'function') {
|
|
80
|
+
throw new Error(`Tool "${toolName}" does not have an execute function`);
|
|
81
|
+
}
|
|
82
|
+
const enrichedOptions = this.withToolExecutionMetadata(undefined, { session, agentId, step, turn, toolName, toolCallId, idempotencyKey });
|
|
83
|
+
try {
|
|
84
|
+
const timeoutMs = tool.timeout ?? this.defaultToolTimeoutMs ?? 30_000;
|
|
85
|
+
const exec = tool.execute.bind(tool);
|
|
86
|
+
const result = await Promise.race([
|
|
87
|
+
exec(input, enrichedOptions),
|
|
88
|
+
new Promise((_, reject) => {
|
|
89
|
+
const timer = setTimeout(() => reject(new ToolTimeoutError(toolName, timeoutMs)), timeoutMs);
|
|
90
|
+
if (typeof timer === 'object' && 'unref' in timer)
|
|
91
|
+
timer.unref();
|
|
92
|
+
}),
|
|
93
|
+
]);
|
|
94
|
+
return result;
|
|
95
|
+
}
|
|
96
|
+
catch (error) {
|
|
97
|
+
console.error(`[ToolExecutor] Tool execution failed for ${toolName}:`, error);
|
|
98
|
+
throw error;
|
|
99
|
+
}
|
|
100
|
+
}
|
|
101
|
+
buildIdempotencyKey(args) {
|
|
102
|
+
return `${args.sessionId}:${args.agentId}:${args.step}:${args.toolName}:${args.toolCallId}`;
|
|
103
|
+
}
|
|
104
|
+
withToolExecutionMetadata(options, ctx) {
|
|
105
|
+
const baseOptions = options ?? {};
|
|
106
|
+
const baseAsAny = baseOptions;
|
|
107
|
+
const existingContext = isRecord(baseAsAny.experimental_context)
|
|
108
|
+
? baseAsAny.experimental_context
|
|
109
|
+
: {};
|
|
110
|
+
return {
|
|
111
|
+
...baseOptions,
|
|
112
|
+
toolCallId: ctx.toolCallId,
|
|
113
|
+
experimental_context: {
|
|
114
|
+
...existingContext,
|
|
115
|
+
session: ctx.session,
|
|
116
|
+
sessionId: ctx.session.id,
|
|
117
|
+
agentId: ctx.agentId,
|
|
118
|
+
step: ctx.step,
|
|
119
|
+
turn: ctx.turn,
|
|
120
|
+
toolName: ctx.toolName,
|
|
121
|
+
toolCallId: ctx.toolCallId,
|
|
122
|
+
idempotencyKey: ctx.idempotencyKey,
|
|
123
|
+
memoryService: this.memoryService,
|
|
124
|
+
},
|
|
125
|
+
};
|
|
126
|
+
}
|
|
127
|
+
}
|
|
128
|
+
//# sourceMappingURL=DefaultToolExecutor.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"DefaultToolExecutor.js","sourceRoot":"","sources":["../../src/foundation/DefaultToolExecutor.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,aAAa,CAAC;AAOjC,OAAO,EAAE,QAAQ,EAAE,MAAM,sBAAsB,CAAC;AAUhD;;GAEG;AACH,MAAM,OAAO,gBAAiB,SAAQ,KAAK;IAChC,QAAQ,CAAS;IACjB,SAAS,CAAS;IAE3B,YAAY,QAAgB,EAAE,SAAiB;QAC7C,KAAK,CAAC,SAAS,QAAQ,mBAAmB,SAAS,IAAI,CAAC,CAAC;QACzD,IAAI,CAAC,IAAI,GAAG,kBAAkB,CAAC;QAC/B,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC7B,CAAC;CACF;AAED;;;;;;;;GAQG;AACH,MAAM,OAAO,mBAAmB;IACtB,QAAQ,CAAe;IACvB,UAAU,CAAa;IACvB,aAAa,CAAiB;IAC9B,oBAAoB,CAAU;IAEtC,YAAY,MAAiC;QAC3C,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;QAChC,IAAI,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,CAAC;QACpC,IAAI,CAAC,aAAa,GAAG,MAAM,CAAC,aAAa,CAAC;QAC1C,IAAI,CAAC,oBAAoB,GAAG,MAAM,CAAC,oBAAoB,CAAC;IAC1D,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,IAYb;QACC,MAAM,EACJ,OAAO,EACP,OAAO,EACP,QAAQ,EACR,IAAI,EACJ,KAAK,EACL,IAAI,GAAG,CAAC,EACR,IAAI,GAAG,CAAC,EACR,eAAe,GAAG,EAAE,GACrB,GAAG,IAAI,CAAC;QAET,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;QAC1D,MAAM,cAAc,GAAG,IAAI,CAAC,mBAAmB,CAAC;YAC9C,SAAS,EAAE,OAAO,CAAC,EAAE;YACrB,OAAO;YACP,IAAI;YACJ,QAAQ;YACR,UAAU;SACX,CAAC,CAAC;QAEH,MAAM,UAAU,GAAmB;YACjC,UAAU;YACV,QAAQ;YACR,IAAI,EAAE,KAAK;YACX,cAAc;YACd,OAAO,EAAE,IAAI;YACb,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC;QAEF,oBAAoB;QACpB,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,UAAU,EAAE;YACxD,aAAa,EAAE,eAAe;YAC9B,WAAW,EAAE,IAAI;YACjB,YAAY,EAAE,OAAO,CAAC,KAAK,IAAI,EAAE;SAClC,CAAC,CAAC;QAEH,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;YACzB,MAAM,MAAM,GAAG,WAAW,CAAC,MAAM,IAAI,kCAAkC,CAAC;YACxE,UAAU,CAAC,OAAO,GAAG,KAAK,CAAC;YAC3B,UAAU,CAAC,KAAK,GAAG,IAAI,KAAK,CAAC,MAAM,CAAC,CAAC;YACrC,eAAe,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAEjC,MAAM,UAAU,GAAG;gBACjB,OAAO;gBACP,OAAO;gBACP,SAAS,EAAE,IAAI;gBACf,WAAW,EAAE,CAAC;gBACd,YAAY,EAAE,EAAE;gBAChB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;gBACrB,iBAAiB,EAAE,CAAC;gBACpB,eAAe;aAChB,CAAC;YACF,MAAM,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,UAAU,EAAE,UAAU,EAAE,UAAU,CAAC,KAAK,CAAC,CAAC;YAC5E,MAAM,UAAU,CAAC,KAAK,CAAC;QACzB,CAAC;QAED,yDAAyD;QACzD,IAAI,CAAC,CAAC,SAAS,IAAI,IAAI,CAAC,IAAI,OAAO,IAAI,CAAC,OAAO,KAAK,UAAU,EAAE,CAAC;YAC/D,MAAM,IAAI,KAAK,CAAC,SAAS,QAAQ,qCAAqC,CAAC,CAAC;QAC1E,CAAC;QAED,MAAM,eAAe,GAAG,IAAI,CAAC,yBAAyB,CACpD,SAAS,EACT,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,cAAc,EAAE,CACvE,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,SAAS,GAAI,IAAY,CAAC,OAAO,IAAI,IAAI,CAAC,oBAAoB,IAAI,MAAM,CAAC;YAC/E,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACrC,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC;gBAChC,IAAI,CAAC,KAAK,EAAE,eAAe,CAAC;gBAC5B,IAAI,OAAO,CAAQ,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE;oBAC/B,MAAM,KAAK,GAAG,UAAU,CACtB,GAAG,EAAE,CAAC,MAAM,CAAC,IAAI,gBAAgB,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC,EACvD,SAAS,CACV,CAAC;oBACF,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,OAAO,IAAI,KAAK;wBAAG,KAAa,CAAC,KAAK,EAAE,CAAC;gBAC5E,CAAC,CAAC;aACH,CAAC,CAAC;YACH,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,4CAA4C,QAAQ,GAAG,EAAE,KAAK,CAAC,CAAC;YAC9E,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED,mBAAmB,CAAC,IAMnB;QACC,OAAO,GAAG,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;IAC9F,CAAC;IAEO,yBAAyB,CAC/B,OAA2C,EAC3C,GAQC;QAED,MAAM,WAAW,GAAG,OAAO,IAAI,EAA4B,CAAC;QAC5D,MAAM,SAAS,GAAG,WAAiD,CAAC;QACpE,MAAM,eAAe,GAAG,QAAQ,CAAC,SAAS,CAAC,oBAAoB,CAAC;YAC9D,CAAC,CAAC,SAAS,CAAC,oBAA+C;YAC3D,CAAC,CAAC,EAAE,CAAC;QAEP,OAAO;YACL,GAAG,WAAW;YACd,UAAU,EAAE,GAAG,CAAC,UAAU;YAC1B,oBAAoB,EAAE;gBACpB,GAAG,eAAe;gBAClB,OAAO,EAAE,GAAG,CAAC,OAAO;gBACpB,SAAS,EAAE,GAAG,CAAC,OAAO,CAAC,EAAE;gBACzB,OAAO,EAAE,GAAG,CAAC,OAAO;gBACpB,IAAI,EAAE,GAAG,CAAC,IAAI;gBACd,IAAI,EAAE,GAAG,CAAC,IAAI;gBACd,QAAQ,EAAE,GAAG,CAAC,QAAQ;gBACtB,UAAU,EAAE,GAAG,CAAC,UAAU;gBAC1B,cAAc,EAAE,GAAG,CAAC,cAAc;gBAClC,aAAa,EAAE,IAAI,CAAC,aAAa;aAClC;SACwB,CAAC;IAC9B,CAAC;CACF"}
|