@just-every/ensemble 0.2.87 → 0.2.88
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/cjs/config/tool_execution.cjs +47 -0
- package/dist/cjs/config/tool_execution.d.ts +12 -0
- package/dist/cjs/config/tool_execution.d.ts.map +1 -0
- package/dist/cjs/config/tool_execution.js.map +1 -0
- package/dist/cjs/core/ensemble_embed.cjs +35 -0
- package/dist/cjs/core/ensemble_embed.d.ts +3 -0
- package/dist/cjs/core/ensemble_embed.d.ts.map +1 -0
- package/dist/cjs/core/ensemble_embed.js.map +1 -0
- package/dist/cjs/core/ensemble_image.cjs +13 -0
- package/dist/cjs/core/ensemble_image.d.ts +3 -0
- package/dist/cjs/core/ensemble_image.d.ts.map +1 -0
- package/dist/cjs/core/ensemble_image.js.map +1 -0
- package/dist/cjs/core/ensemble_listen.cjs +162 -0
- package/dist/cjs/core/ensemble_listen.d.ts +5 -0
- package/dist/cjs/core/ensemble_listen.d.ts.map +1 -0
- package/dist/cjs/core/ensemble_listen.js.map +1 -0
- package/dist/cjs/core/ensemble_live.cjs +387 -0
- package/dist/cjs/core/ensemble_live.d.ts +14 -0
- package/dist/cjs/core/ensemble_live.d.ts.map +1 -0
- package/dist/cjs/core/ensemble_live.js.map +1 -0
- package/dist/cjs/core/ensemble_request.cjs +409 -0
- package/dist/cjs/core/ensemble_request.d.ts +4 -0
- package/dist/cjs/core/ensemble_request.d.ts.map +1 -0
- package/dist/cjs/core/ensemble_request.js.map +1 -0
- package/dist/cjs/core/ensemble_voice.cjs +284 -0
- package/dist/cjs/core/ensemble_voice.d.ts +4 -0
- package/dist/cjs/core/ensemble_voice.d.ts.map +1 -0
- package/dist/cjs/core/ensemble_voice.js.map +1 -0
- package/dist/cjs/data/model_data.cjs +1460 -0
- package/dist/cjs/data/model_data.d.ts +71 -0
- package/dist/cjs/data/model_data.d.ts.map +1 -0
- package/dist/cjs/data/model_data.js.map +1 -0
- package/dist/cjs/index.cjs +139 -0
- package/dist/cjs/index.d.ts +37 -0
- package/dist/cjs/index.d.ts.map +1 -0
- package/dist/cjs/index.js.map +1 -0
- package/dist/cjs/model_providers/base_provider.cjs +46 -0
- package/dist/cjs/model_providers/base_provider.d.ts +13 -0
- package/dist/cjs/model_providers/base_provider.d.ts.map +1 -0
- package/dist/cjs/model_providers/base_provider.js.map +1 -0
- package/dist/cjs/model_providers/claude.cjs +782 -0
- package/dist/cjs/model_providers/claude.d.ts +12 -0
- package/dist/cjs/model_providers/claude.d.ts.map +1 -0
- package/dist/cjs/model_providers/claude.js.map +1 -0
- package/dist/cjs/model_providers/deepseek.cjs +129 -0
- package/dist/cjs/model_providers/deepseek.d.ts +8 -0
- package/dist/cjs/model_providers/deepseek.d.ts.map +1 -0
- package/dist/cjs/model_providers/deepseek.js.map +1 -0
- package/dist/cjs/model_providers/elevenlabs.cjs +140 -0
- package/dist/cjs/model_providers/elevenlabs.d.ts +37 -0
- package/dist/cjs/model_providers/elevenlabs.d.ts.map +1 -0
- package/dist/cjs/model_providers/elevenlabs.js.map +1 -0
- package/dist/cjs/model_providers/gemini.cjs +1654 -0
- package/dist/cjs/model_providers/gemini.d.ts +22 -0
- package/dist/cjs/model_providers/gemini.d.ts.map +1 -0
- package/dist/cjs/model_providers/gemini.js.map +1 -0
- package/dist/cjs/model_providers/grok.cjs +25 -0
- package/dist/cjs/model_providers/grok.d.ts +8 -0
- package/dist/cjs/model_providers/grok.d.ts.map +1 -0
- package/dist/cjs/model_providers/grok.js.map +1 -0
- package/dist/cjs/model_providers/model_provider.cjs +296 -0
- package/dist/cjs/model_providers/model_provider.d.ts +10 -0
- package/dist/cjs/model_providers/model_provider.d.ts.map +1 -0
- package/dist/cjs/model_providers/model_provider.js.map +1 -0
- package/dist/cjs/model_providers/openai.cjs +1117 -0
- package/dist/cjs/model_providers/openai.d.ts +19 -0
- package/dist/cjs/model_providers/openai.d.ts.map +1 -0
- package/dist/cjs/model_providers/openai.js.map +1 -0
- package/dist/cjs/model_providers/openai_chat.cjs +787 -0
- package/dist/cjs/model_providers/openai_chat.d.ts +20 -0
- package/dist/cjs/model_providers/openai_chat.d.ts.map +1 -0
- package/dist/cjs/model_providers/openai_chat.js.map +1 -0
- package/dist/cjs/model_providers/openrouter.cjs +22 -0
- package/dist/cjs/model_providers/openrouter.d.ts +6 -0
- package/dist/cjs/model_providers/openrouter.d.ts.map +1 -0
- package/dist/cjs/model_providers/openrouter.js.map +1 -0
- package/dist/cjs/model_providers/test_provider.cjs +236 -0
- package/dist/cjs/model_providers/test_provider.d.ts +29 -0
- package/dist/cjs/model_providers/test_provider.d.ts.map +1 -0
- package/dist/cjs/model_providers/test_provider.js.map +1 -0
- package/dist/cjs/tsconfig.cjs.tsbuildinfo +1 -0
- package/dist/cjs/types/api_types.cjs +3 -0
- package/dist/cjs/types/api_types.d.ts +249 -0
- package/dist/cjs/types/api_types.d.ts.map +1 -0
- package/dist/cjs/types/api_types.js.map +1 -0
- package/dist/cjs/types/errors.cjs +76 -0
- package/dist/cjs/types/errors.d.ts +34 -0
- package/dist/cjs/types/errors.d.ts.map +1 -0
- package/dist/cjs/types/errors.js.map +1 -0
- package/dist/cjs/types/types.cjs +3 -0
- package/dist/cjs/types/types.d.ts +638 -0
- package/dist/cjs/types/types.d.ts.map +1 -0
- package/dist/cjs/types/types.js.map +1 -0
- package/dist/cjs/utils/agent.cjs +384 -0
- package/dist/cjs/utils/agent.d.ts +48 -0
- package/dist/cjs/utils/agent.d.ts.map +1 -0
- package/dist/cjs/utils/agent.js.map +1 -0
- package/dist/cjs/utils/audio_stream_player.cjs +342 -0
- package/dist/cjs/utils/audio_stream_player.d.ts +37 -0
- package/dist/cjs/utils/audio_stream_player.d.ts.map +1 -0
- package/dist/cjs/utils/audio_stream_player.js.map +1 -0
- package/dist/cjs/utils/citation_tracker.cjs +25 -0
- package/dist/cjs/utils/citation_tracker.d.ts +12 -0
- package/dist/cjs/utils/citation_tracker.d.ts.map +1 -0
- package/dist/cjs/utils/citation_tracker.js.map +1 -0
- package/dist/cjs/utils/config_manager.cjs +105 -0
- package/dist/cjs/utils/config_manager.d.ts +31 -0
- package/dist/cjs/utils/config_manager.d.ts.map +1 -0
- package/dist/cjs/utils/config_manager.js.map +1 -0
- package/dist/cjs/utils/cost_tracker.cjs +226 -0
- package/dist/cjs/utils/cost_tracker.d.ts +33 -0
- package/dist/cjs/utils/cost_tracker.d.ts.map +1 -0
- package/dist/cjs/utils/cost_tracker.js.map +1 -0
- package/dist/cjs/utils/create_tool_function.cjs +182 -0
- package/dist/cjs/utils/create_tool_function.d.ts +3 -0
- package/dist/cjs/utils/create_tool_function.d.ts.map +1 -0
- package/dist/cjs/utils/create_tool_function.js.map +1 -0
- package/dist/cjs/utils/delta_buffer.cjs +65 -0
- package/dist/cjs/utils/delta_buffer.d.ts +14 -0
- package/dist/cjs/utils/delta_buffer.d.ts.map +1 -0
- package/dist/cjs/utils/delta_buffer.js.map +1 -0
- package/dist/cjs/utils/ensemble_result.cjs +167 -0
- package/dist/cjs/utils/ensemble_result.d.ts +33 -0
- package/dist/cjs/utils/ensemble_result.d.ts.map +1 -0
- package/dist/cjs/utils/ensemble_result.js.map +1 -0
- package/dist/cjs/utils/event_controller.cjs +59 -0
- package/dist/cjs/utils/event_controller.d.ts +13 -0
- package/dist/cjs/utils/event_controller.d.ts.map +1 -0
- package/dist/cjs/utils/event_controller.js.map +1 -0
- package/dist/cjs/utils/external_models.cjs +42 -0
- package/dist/cjs/utils/external_models.d.ts +9 -0
- package/dist/cjs/utils/external_models.d.ts.map +1 -0
- package/dist/cjs/utils/external_models.js.map +1 -0
- package/dist/cjs/utils/image_to_text.cjs +58 -0
- package/dist/cjs/utils/image_to_text.d.ts +3 -0
- package/dist/cjs/utils/image_to_text.d.ts.map +1 -0
- package/dist/cjs/utils/image_to_text.js.map +1 -0
- package/dist/cjs/utils/image_utils.cjs +168 -0
- package/dist/cjs/utils/image_utils.d.ts +18 -0
- package/dist/cjs/utils/image_utils.d.ts.map +1 -0
- package/dist/cjs/utils/image_utils.js.map +1 -0
- package/dist/cjs/utils/image_validation.cjs +31 -0
- package/dist/cjs/utils/image_validation.d.ts +3 -0
- package/dist/cjs/utils/image_validation.d.ts.map +1 -0
- package/dist/cjs/utils/image_validation.js.map +1 -0
- package/dist/cjs/utils/llm_logger.cjs +31 -0
- package/dist/cjs/utils/llm_logger.d.ts +8 -0
- package/dist/cjs/utils/llm_logger.d.ts.map +1 -0
- package/dist/cjs/utils/llm_logger.js.map +1 -0
- package/dist/cjs/utils/message_history.cjs +560 -0
- package/dist/cjs/utils/message_history.d.ts +65 -0
- package/dist/cjs/utils/message_history.d.ts.map +1 -0
- package/dist/cjs/utils/message_history.js.map +1 -0
- package/dist/cjs/utils/model_class_config.cjs +105 -0
- package/dist/cjs/utils/model_class_config.d.ts +12 -0
- package/dist/cjs/utils/model_class_config.d.ts.map +1 -0
- package/dist/cjs/utils/model_class_config.js.map +1 -0
- package/dist/cjs/utils/pause_controller.cjs +90 -0
- package/dist/cjs/utils/pause_controller.d.ts +14 -0
- package/dist/cjs/utils/pause_controller.d.ts.map +1 -0
- package/dist/cjs/utils/pause_controller.js.map +1 -0
- package/dist/cjs/utils/quota_tracker.cjs +311 -0
- package/dist/cjs/utils/quota_tracker.d.ts +22 -0
- package/dist/cjs/utils/quota_tracker.d.ts.map +1 -0
- package/dist/cjs/utils/quota_tracker.js.map +1 -0
- package/dist/cjs/utils/retry_handler.cjs +131 -0
- package/dist/cjs/utils/retry_handler.d.ts +15 -0
- package/dist/cjs/utils/retry_handler.d.ts.map +1 -0
- package/dist/cjs/utils/retry_handler.js.map +1 -0
- package/dist/cjs/utils/running_tool_tracker.cjs +133 -0
- package/dist/cjs/utils/running_tool_tracker.d.ts +42 -0
- package/dist/cjs/utils/running_tool_tracker.d.ts.map +1 -0
- package/dist/cjs/utils/running_tool_tracker.js.map +1 -0
- package/dist/cjs/utils/sequential_queue.cjs +73 -0
- package/dist/cjs/utils/sequential_queue.d.ts +13 -0
- package/dist/cjs/utils/sequential_queue.d.ts.map +1 -0
- package/dist/cjs/utils/sequential_queue.js.map +1 -0
- package/dist/cjs/utils/stream_handler.cjs +73 -0
- package/dist/cjs/utils/stream_handler.d.ts +16 -0
- package/dist/cjs/utils/stream_handler.d.ts.map +1 -0
- package/dist/cjs/utils/stream_handler.js.map +1 -0
- package/dist/cjs/utils/summary_utils.cjs +211 -0
- package/dist/cjs/utils/summary_utils.d.ts +7 -0
- package/dist/cjs/utils/summary_utils.d.ts.map +1 -0
- package/dist/cjs/utils/summary_utils.js.map +1 -0
- package/dist/cjs/utils/test_utils.cjs +212 -0
- package/dist/cjs/utils/test_utils.d.ts +58 -0
- package/dist/cjs/utils/test_utils.d.ts.map +1 -0
- package/dist/cjs/utils/test_utils.js.map +1 -0
- package/dist/cjs/utils/tool_execution_manager.cjs +139 -0
- package/dist/cjs/utils/tool_execution_manager.d.ts +7 -0
- package/dist/cjs/utils/tool_execution_manager.d.ts.map +1 -0
- package/dist/cjs/utils/tool_execution_manager.js.map +1 -0
- package/dist/cjs/utils/tool_parameter_utils.cjs +168 -0
- package/dist/cjs/utils/tool_parameter_utils.d.ts +5 -0
- package/dist/cjs/utils/tool_parameter_utils.d.ts.map +1 -0
- package/dist/cjs/utils/tool_parameter_utils.js.map +1 -0
- package/dist/cjs/utils/tool_result_processor.cjs +363 -0
- package/dist/cjs/utils/tool_result_processor.d.ts +11 -0
- package/dist/cjs/utils/tool_result_processor.d.ts.map +1 -0
- package/dist/cjs/utils/tool_result_processor.js.map +1 -0
- package/dist/cjs/utils/verification.cjs +59 -0
- package/dist/cjs/utils/verification.d.ts +7 -0
- package/dist/cjs/utils/verification.d.ts.map +1 -0
- package/dist/cjs/utils/verification.js.map +1 -0
- package/dist/tsconfig.tsbuildinfo +1 -1
- package/dist/utils/audio_stream_player.d.ts +2 -0
- package/dist/utils/audio_stream_player.d.ts.map +1 -1
- package/dist/utils/audio_stream_player.js +83 -1
- package/dist/utils/audio_stream_player.js.map +1 -1
- package/dist/utils/cost_tracker.d.ts +7 -0
- package/dist/utils/cost_tracker.d.ts.map +1 -1
- package/dist/utils/cost_tracker.js +29 -0
- package/dist/utils/cost_tracker.js.map +1 -1
- package/package.json +18 -3
|
@@ -0,0 +1,212 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.StreamAssertions = exports.EnhancedRequestMock = void 0;
|
|
4
|
+
exports.createMockContext = createMockContext;
|
|
5
|
+
const vitest_1 = require("vitest");
|
|
6
|
+
class EnhancedRequestMock {
|
|
7
|
+
responses;
|
|
8
|
+
callIndex = 0;
|
|
9
|
+
constructor(responses) {
|
|
10
|
+
this.responses = Array.isArray(responses) ? responses : [responses];
|
|
11
|
+
}
|
|
12
|
+
getMock() {
|
|
13
|
+
return (model, messages, options) => {
|
|
14
|
+
return this.createAsyncGenerator(options);
|
|
15
|
+
};
|
|
16
|
+
}
|
|
17
|
+
async *createAsyncGenerator(options) {
|
|
18
|
+
for (const response of this.responses) {
|
|
19
|
+
if (response.delay) {
|
|
20
|
+
await new Promise(resolve => setTimeout(resolve, response.delay));
|
|
21
|
+
}
|
|
22
|
+
if (response.error) {
|
|
23
|
+
const error = typeof response.error === 'string' ? new Error(response.error) : response.error;
|
|
24
|
+
yield {
|
|
25
|
+
type: 'error',
|
|
26
|
+
error: error.message,
|
|
27
|
+
timestamp: new Date().toISOString(),
|
|
28
|
+
};
|
|
29
|
+
return;
|
|
30
|
+
}
|
|
31
|
+
if (response.thinking && options?.includeThinking) {
|
|
32
|
+
yield {
|
|
33
|
+
type: 'thinking_start',
|
|
34
|
+
timestamp: new Date().toISOString(),
|
|
35
|
+
};
|
|
36
|
+
const chunks = response.thinking.match(/.{1,10}/g) || [];
|
|
37
|
+
for (const chunk of chunks) {
|
|
38
|
+
yield {
|
|
39
|
+
type: 'thinking_delta',
|
|
40
|
+
delta: chunk,
|
|
41
|
+
timestamp: new Date().toISOString(),
|
|
42
|
+
};
|
|
43
|
+
}
|
|
44
|
+
yield {
|
|
45
|
+
type: 'thinking_complete',
|
|
46
|
+
content: response.thinking,
|
|
47
|
+
timestamp: new Date().toISOString(),
|
|
48
|
+
};
|
|
49
|
+
}
|
|
50
|
+
if (response.message) {
|
|
51
|
+
yield {
|
|
52
|
+
type: 'message_start',
|
|
53
|
+
timestamp: new Date().toISOString(),
|
|
54
|
+
};
|
|
55
|
+
const chunks = response.message.match(/.{1,5}/g) || [];
|
|
56
|
+
for (const chunk of chunks) {
|
|
57
|
+
yield {
|
|
58
|
+
type: 'text_delta',
|
|
59
|
+
delta: chunk,
|
|
60
|
+
timestamp: new Date().toISOString(),
|
|
61
|
+
};
|
|
62
|
+
}
|
|
63
|
+
yield {
|
|
64
|
+
type: 'message_complete',
|
|
65
|
+
content: response.message,
|
|
66
|
+
timestamp: new Date().toISOString(),
|
|
67
|
+
};
|
|
68
|
+
}
|
|
69
|
+
if (response.toolCalls && response.toolCalls.length > 0) {
|
|
70
|
+
const toolCallEvents = response.toolCalls.map((call, index) => ({
|
|
71
|
+
id: `call_${Date.now()}_${index}`,
|
|
72
|
+
type: 'function',
|
|
73
|
+
function: {
|
|
74
|
+
name: call.name,
|
|
75
|
+
arguments: JSON.stringify(call.arguments),
|
|
76
|
+
},
|
|
77
|
+
}));
|
|
78
|
+
yield {
|
|
79
|
+
type: 'tool_start',
|
|
80
|
+
tool_call: toolCallEvents,
|
|
81
|
+
timestamp: new Date().toISOString(),
|
|
82
|
+
};
|
|
83
|
+
if (options?.onToolCall) {
|
|
84
|
+
for (const call of toolCallEvents) {
|
|
85
|
+
options.onToolCall(call);
|
|
86
|
+
}
|
|
87
|
+
}
|
|
88
|
+
}
|
|
89
|
+
}
|
|
90
|
+
yield {
|
|
91
|
+
type: 'stream_end',
|
|
92
|
+
timestamp: new Date().toISOString(),
|
|
93
|
+
};
|
|
94
|
+
}
|
|
95
|
+
static success(message = 'Success', result = 'Task completed') {
|
|
96
|
+
return new EnhancedRequestMock({
|
|
97
|
+
message,
|
|
98
|
+
toolCalls: [{ name: 'task_complete', arguments: { result } }],
|
|
99
|
+
});
|
|
100
|
+
}
|
|
101
|
+
static error(message = 'Error occurred', error = 'Task failed') {
|
|
102
|
+
return new EnhancedRequestMock({
|
|
103
|
+
message,
|
|
104
|
+
toolCalls: [{ name: 'task_fatal_error', arguments: { error } }],
|
|
105
|
+
});
|
|
106
|
+
}
|
|
107
|
+
static throws(error) {
|
|
108
|
+
return new EnhancedRequestMock({
|
|
109
|
+
error: typeof error === 'string' ? new Error(error) : error,
|
|
110
|
+
});
|
|
111
|
+
}
|
|
112
|
+
static thinking(thinking, message) {
|
|
113
|
+
return new EnhancedRequestMock({
|
|
114
|
+
thinking,
|
|
115
|
+
message,
|
|
116
|
+
});
|
|
117
|
+
}
|
|
118
|
+
static toolCalls(...calls) {
|
|
119
|
+
return new EnhancedRequestMock({
|
|
120
|
+
message: '',
|
|
121
|
+
toolCalls: calls,
|
|
122
|
+
});
|
|
123
|
+
}
|
|
124
|
+
static sequence(...responses) {
|
|
125
|
+
return new EnhancedRequestMock(responses);
|
|
126
|
+
}
|
|
127
|
+
}
|
|
128
|
+
exports.EnhancedRequestMock = EnhancedRequestMock;
|
|
129
|
+
function createMockContext(overrides = {}) {
|
|
130
|
+
return {
|
|
131
|
+
shouldContinue: true,
|
|
132
|
+
metadata: {},
|
|
133
|
+
toolCallCount: 0,
|
|
134
|
+
turnCount: 0,
|
|
135
|
+
startTime: Date.now(),
|
|
136
|
+
messages: [],
|
|
137
|
+
isPaused: false,
|
|
138
|
+
isHalted: false,
|
|
139
|
+
halt: vitest_1.vi.fn(function () {
|
|
140
|
+
this.shouldContinue = false;
|
|
141
|
+
this.isHalted = true;
|
|
142
|
+
}),
|
|
143
|
+
pause: vitest_1.vi.fn(function () {
|
|
144
|
+
this.isPaused = true;
|
|
145
|
+
}),
|
|
146
|
+
resume: vitest_1.vi.fn(function () {
|
|
147
|
+
this.isPaused = false;
|
|
148
|
+
}),
|
|
149
|
+
setMetadata: vitest_1.vi.fn(function (key, value) {
|
|
150
|
+
this.metadata[key] = value;
|
|
151
|
+
}),
|
|
152
|
+
getMetadata: vitest_1.vi.fn(function (key) {
|
|
153
|
+
return this.metadata[key];
|
|
154
|
+
}),
|
|
155
|
+
addMessage: vitest_1.vi.fn(function (message) {
|
|
156
|
+
this.messages.push(message);
|
|
157
|
+
}),
|
|
158
|
+
getHistory: vitest_1.vi.fn(function () {
|
|
159
|
+
return this.messages;
|
|
160
|
+
}),
|
|
161
|
+
...overrides,
|
|
162
|
+
};
|
|
163
|
+
}
|
|
164
|
+
class StreamAssertions {
|
|
165
|
+
events = [];
|
|
166
|
+
constructor(eventGenerator) {
|
|
167
|
+
(async () => {
|
|
168
|
+
for await (const event of eventGenerator) {
|
|
169
|
+
this.events.push(event);
|
|
170
|
+
}
|
|
171
|
+
})();
|
|
172
|
+
}
|
|
173
|
+
async waitForCompletion() {
|
|
174
|
+
await new Promise(resolve => setTimeout(resolve, 100));
|
|
175
|
+
}
|
|
176
|
+
hasEvent(type) {
|
|
177
|
+
return this.events.some(e => e.type === type);
|
|
178
|
+
}
|
|
179
|
+
getEvents(type) {
|
|
180
|
+
return this.events.filter(e => e.type === type);
|
|
181
|
+
}
|
|
182
|
+
hasToolCall(name) {
|
|
183
|
+
const toolEvents = this.getEvents('tool_start');
|
|
184
|
+
return toolEvents.some(event => {
|
|
185
|
+
if ('tool_call' in event && event.tool_call) {
|
|
186
|
+
return event.tool_call.function.name === name;
|
|
187
|
+
}
|
|
188
|
+
return false;
|
|
189
|
+
});
|
|
190
|
+
}
|
|
191
|
+
getFinalMessage() {
|
|
192
|
+
const messageEvents = this.getEvents('message_complete');
|
|
193
|
+
if (messageEvents.length > 0) {
|
|
194
|
+
const lastEvent = messageEvents[messageEvents.length - 1];
|
|
195
|
+
return 'content' in lastEvent ? lastEvent.content : undefined;
|
|
196
|
+
}
|
|
197
|
+
return undefined;
|
|
198
|
+
}
|
|
199
|
+
hasError() {
|
|
200
|
+
return this.hasEvent('error');
|
|
201
|
+
}
|
|
202
|
+
getError() {
|
|
203
|
+
const errorEvents = this.getEvents('error');
|
|
204
|
+
if (errorEvents.length > 0) {
|
|
205
|
+
const errorEvent = errorEvents[0];
|
|
206
|
+
return 'error' in errorEvent ? errorEvent.error : undefined;
|
|
207
|
+
}
|
|
208
|
+
return undefined;
|
|
209
|
+
}
|
|
210
|
+
}
|
|
211
|
+
exports.StreamAssertions = StreamAssertions;
|
|
212
|
+
//# sourceMappingURL=test_utils.js.map
|
|
@@ -0,0 +1,58 @@
|
|
|
1
|
+
import { ProviderStreamEvent, ToolCall } from "../types/types.cjs";
|
|
2
|
+
export interface MockToolCall {
|
|
3
|
+
name: string;
|
|
4
|
+
arguments: Record<string, any>;
|
|
5
|
+
}
|
|
6
|
+
export interface MockResponse {
|
|
7
|
+
message?: string;
|
|
8
|
+
toolCalls?: MockToolCall[];
|
|
9
|
+
error?: Error | string;
|
|
10
|
+
delay?: number;
|
|
11
|
+
thinking?: string;
|
|
12
|
+
}
|
|
13
|
+
export interface MockStreamOptions {
|
|
14
|
+
onToolCall?: (call: ToolCall) => void;
|
|
15
|
+
includeThinking?: boolean;
|
|
16
|
+
}
|
|
17
|
+
export declare class EnhancedRequestMock {
|
|
18
|
+
private responses;
|
|
19
|
+
private callIndex;
|
|
20
|
+
constructor(responses: MockResponse | MockResponse[]);
|
|
21
|
+
getMock(): (model: string, messages: any, options: any) => AsyncGenerator<ProviderStreamEvent, any, any>;
|
|
22
|
+
private createAsyncGenerator;
|
|
23
|
+
static success(message?: string, result?: string): EnhancedRequestMock;
|
|
24
|
+
static error(message?: string, error?: string): EnhancedRequestMock;
|
|
25
|
+
static throws(error: Error | string): EnhancedRequestMock;
|
|
26
|
+
static thinking(thinking: string, message: string): EnhancedRequestMock;
|
|
27
|
+
static toolCalls(...calls: MockToolCall[]): EnhancedRequestMock;
|
|
28
|
+
static sequence(...responses: MockResponse[]): EnhancedRequestMock;
|
|
29
|
+
}
|
|
30
|
+
export declare function createMockContext(overrides?: Partial<any>): {
|
|
31
|
+
shouldContinue: boolean;
|
|
32
|
+
metadata: {};
|
|
33
|
+
toolCallCount: number;
|
|
34
|
+
turnCount: number;
|
|
35
|
+
startTime: number;
|
|
36
|
+
messages: any[];
|
|
37
|
+
isPaused: boolean;
|
|
38
|
+
isHalted: boolean;
|
|
39
|
+
halt: import("vitest").Mock<() => void>;
|
|
40
|
+
pause: import("vitest").Mock<() => void>;
|
|
41
|
+
resume: import("vitest").Mock<() => void>;
|
|
42
|
+
setMetadata: import("vitest").Mock<(key: string, value: any) => void>;
|
|
43
|
+
getMetadata: import("vitest").Mock<(key: string) => any>;
|
|
44
|
+
addMessage: import("vitest").Mock<(message: any) => void>;
|
|
45
|
+
getHistory: import("vitest").Mock<() => any>;
|
|
46
|
+
};
|
|
47
|
+
export declare class StreamAssertions {
|
|
48
|
+
private events;
|
|
49
|
+
constructor(eventGenerator: AsyncGenerator<ProviderStreamEvent>);
|
|
50
|
+
waitForCompletion(): Promise<void>;
|
|
51
|
+
hasEvent(type: string): boolean;
|
|
52
|
+
getEvents(type: string): ProviderStreamEvent[];
|
|
53
|
+
hasToolCall(name: string): boolean;
|
|
54
|
+
getFinalMessage(): string | undefined;
|
|
55
|
+
hasError(): boolean;
|
|
56
|
+
getError(): string | undefined;
|
|
57
|
+
}
|
|
58
|
+
//# sourceMappingURL=test_utils.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"test_utils.d.ts","sourceRoot":"","sources":["../../../utils/test_utils.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,mBAAmB,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAGlE,MAAM,WAAW,YAAY;IACzB,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAClC;AAED,MAAM,WAAW,YAAY;IACzB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,SAAS,CAAC,EAAE,YAAY,EAAE,CAAC;IAC3B,KAAK,CAAC,EAAE,KAAK,GAAG,MAAM,CAAC;IACvB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,iBAAiB;IAC9B,UAAU,CAAC,EAAE,CAAC,IAAI,EAAE,QAAQ,KAAK,IAAI,CAAC;IACtC,eAAe,CAAC,EAAE,OAAO,CAAC;CAC7B;AAKD,qBAAa,mBAAmB;IAC5B,OAAO,CAAC,SAAS,CAAiB;IAClC,OAAO,CAAC,SAAS,CAAK;gBAEV,SAAS,EAAE,YAAY,GAAG,YAAY,EAAE;IAOpD,OAAO,KACK,OAAO,MAAM,EAAE,UAAU,GAAG,EAAE,SAAS,GAAG;YAQvC,oBAAoB;IAoGnC,MAAM,CAAC,OAAO,CAAC,OAAO,SAAY,EAAE,MAAM,SAAmB;IAO7D,MAAM,CAAC,KAAK,CAAC,OAAO,SAAmB,EAAE,KAAK,SAAgB;IAO9D,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,KAAK,GAAG,MAAM;IAMnC,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM;IAOjD,MAAM,CAAC,SAAS,CAAC,GAAG,KAAK,EAAE,YAAY,EAAE;IAOzC,MAAM,CAAC,QAAQ,CAAC,GAAG,SAAS,EAAE,YAAY,EAAE;CAG/C;AAKD,wBAAgB,iBAAiB,CAAC,SAAS,GAAE,OAAO,CAAC,GAAG,CAAM;;;;;;;;;;;;6CAwBpB,MAAM,SAAS,GAAG;6CAIlB,MAAM;gDAIH,GAAG;;EAU/C;AAKD,qBAAa,gBAAgB;IACzB,OAAO,CAAC,MAAM,CAA6B;gBAE/B,cAAc,EAAE,cAAc,CAAC,mBAAmB,CAAC;IAYzD,iBAAiB,IAAI,OAAO,CAAC,IAAI,CAAC;IAQxC,QAAQ,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO;IAO/B,SAAS,CAAC,IAAI,EAAE,MAAM,GAAG,mBAAmB,EAAE;IAO9C,WAAW,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO;IAalC,eAAe,IAAI,MAAM,GAAG,SAAS;IAYrC,QAAQ,IAAI,OAAO;IAOnB,QAAQ,IAAI,MAAM,GAAG,SAAS;CAQjC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"test_utils.js","sourceRoot":"","sources":["../../../utils/test_utils.ts"],"names":[],"mappings":";;;AA8LA,8CA0CC;AAnOD,mCAA4B;AAuB5B,MAAa,mBAAmB;IACpB,SAAS,CAAiB;IAC1B,SAAS,GAAG,CAAC,CAAC;IAEtB,YAAY,SAAwC;QAChD,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;IACxE,CAAC;IAKD,OAAO;QACH,OAAO,CAAC,KAAa,EAAE,QAAa,EAAE,OAAY,EAAE,EAAE;YAClD,OAAO,IAAI,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC;QAC9C,CAAC,CAAC;IACN,CAAC;IAKO,KAAK,CAAC,CAAC,oBAAoB,CAAC,OAA2B;QAC3D,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YAEpC,IAAI,QAAQ,CAAC,KAAK,EAAE,CAAC;gBACjB,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC;YACtE,CAAC;YAGD,IAAI,QAAQ,CAAC,KAAK,EAAE,CAAC;gBACjB,MAAM,KAAK,GAAG,OAAO,QAAQ,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC;gBAC9F,MAAM;oBACF,IAAI,EAAE,OAAO;oBACb,KAAK,EAAE,KAAK,CAAC,OAAO;oBACpB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;iBACf,CAAC;gBACzB,OAAO;YACX,CAAC;YAGD,IAAI,QAAQ,CAAC,QAAQ,IAAI,OAAO,EAAE,eAAe,EAAE,CAAC;gBAChD,MAAM;oBACF,IAAI,EAAE,gBAAgB;oBACtB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;iBAC/B,CAAC;gBAGT,MAAM,MAAM,GAAG,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC;gBACzD,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;oBACzB,MAAM;wBACF,IAAI,EAAE,gBAAgB;wBACtB,KAAK,EAAE,KAAK;wBACZ,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;qBAC/B,CAAC;gBACb,CAAC;gBAED,MAAM;oBACF,IAAI,EAAE,mBAAmB;oBACzB,OAAO,EAAE,QAAQ,CAAC,QAAQ;oBAC1B,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;iBAC/B,CAAC;YACb,CAAC;YAGD,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;gBACnB,MAAM;oBACF,IAAI,EAAE,eAAe;oBACrB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;iBACf,CAAC;gBAGzB,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;gBACvD,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;oBACzB,MAAM;wBACF,IAAI,EAAE,YAAY;wBAClB,KAAK,EAAE,KAAK;wBACZ,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;qBAC/B,CAAC;gBACb,CAAC;gBAED,MAAM;oBACF,IAAI,EAAE,kBAAkB;oBACxB,OAAO,EAAE,QAAQ,CAAC,OAAO;oBACzB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;iBACf,CAAC;YAC7B,CAAC;YAGD,IAAI,QAAQ,CAAC,SAAS,IAAI,QAAQ,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACtD,MAAM,cAAc,GAAe,QAAQ,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;oBACxE,EAAE,EAAE,QAAQ,IAAI,CAAC,GAAG,EAAE,IAAI,KAAK,EAAE;oBACjC,IAAI,EAAE,UAAmB;oBACzB,QAAQ,EAAE;wBACN,IAAI,EAAE,IAAI,CAAC,IAAI;wBACf,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC;qBAC5C;iBACJ,CAAC,CAAC,CAAC;gBAEJ,MAAM;oBACF,IAAI,EAAE,YAAY;oBAClB,SAAS,EAAE,cAAc;oBACzB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;iBACf,CAAC;gBAGzB,IAAI,OAAO,EAAE,UAAU,EAAE,CAAC;oBACtB,KAAK,MAAM,IAAI,IAAI,cAAc,EAAE,CAAC;wBAChC,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;oBAC7B,CAAC;gBACL,CAAC;YACL,CAAC;QACL,CAAC;QAGD,MAAM;YACF,IAAI,EAAE,YAAY;YAClB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACf,CAAC;IAC7B,CAAC;IAGD,MAAM,CAAC,OAAO,CAAC,OAAO,GAAG,SAAS,EAAE,MAAM,GAAG,gBAAgB;QACzD,OAAO,IAAI,mBAAmB,CAAC;YAC3B,OAAO;YACP,SAAS,EAAE,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,SAAS,EAAE,EAAE,MAAM,EAAE,EAAE,CAAC;SAChE,CAAC,CAAC;IACP,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,OAAO,GAAG,gBAAgB,EAAE,KAAK,GAAG,aAAa;QAC1D,OAAO,IAAI,mBAAmB,CAAC;YAC3B,OAAO;YACP,SAAS,EAAE,CAAC,EAAE,IAAI,EAAE,kBAAkB,EAAE,SAAS,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC;SAClE,CAAC,CAAC;IACP,CAAC;IAED,MAAM,CAAC,MAAM,CAAC,KAAqB;QAC/B,OAAO,IAAI,mBAAmB,CAAC;YAC3B,KAAK,EAAE,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK;SAC9D,CAAC,CAAC;IACP,CAAC;IAED,MAAM,CAAC,QAAQ,CAAC,QAAgB,EAAE,OAAe;QAC7C,OAAO,IAAI,mBAAmB,CAAC;YAC3B,QAAQ;YACR,OAAO;SACV,CAAC,CAAC;IACP,CAAC;IAED,MAAM,CAAC,SAAS,CAAC,GAAG,KAAqB;QACrC,OAAO,IAAI,mBAAmB,CAAC;YAC3B,OAAO,EAAE,EAAE;YACX,SAAS,EAAE,KAAK;SACnB,CAAC,CAAC;IACP,CAAC;IAED,MAAM,CAAC,QAAQ,CAAC,GAAG,SAAyB;QACxC,OAAO,IAAI,mBAAmB,CAAC,SAAS,CAAC,CAAC;IAC9C,CAAC;CACJ;AA7JD,kDA6JC;AAKD,SAAgB,iBAAiB,CAAC,YAA0B,EAAE;IAC1D,OAAO;QACH,cAAc,EAAE,IAAI;QACpB,QAAQ,EAAE,EAAE;QACZ,aAAa,EAAE,CAAC;QAChB,SAAS,EAAE,CAAC;QACZ,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;QACrB,QAAQ,EAAE,EAAE;QACZ,QAAQ,EAAE,KAAK;QACf,QAAQ,EAAE,KAAK;QAEf,IAAI,EAAE,WAAE,CAAC,EAAE,CAAC;YACR,IAAI,CAAC,cAAc,GAAG,KAAK,CAAC;YAC5B,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QACzB,CAAC,CAAC;QAEF,KAAK,EAAE,WAAE,CAAC,EAAE,CAAC;YACT,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QACzB,CAAC,CAAC;QAEF,MAAM,EAAE,WAAE,CAAC,EAAE,CAAC;YACV,IAAI,CAAC,QAAQ,GAAG,KAAK,CAAC;QAC1B,CAAC,CAAC;QAEF,WAAW,EAAE,WAAE,CAAC,EAAE,CAAC,UAAU,GAAW,EAAE,KAAU;YAChD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;QAC/B,CAAC,CAAC;QAEF,WAAW,EAAE,WAAE,CAAC,EAAE,CAAC,UAAU,GAAW;YACpC,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;QAC9B,CAAC,CAAC;QAEF,UAAU,EAAE,WAAE,CAAC,EAAE,CAAC,UAAU,OAAY;YACpC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAChC,CAAC,CAAC;QAEF,UAAU,EAAE,WAAE,CAAC,EAAE,CAAC;YACd,OAAO,IAAI,CAAC,QAAQ,CAAC;QACzB,CAAC,CAAC;QAEF,GAAG,SAAS;KACf,CAAC;AACN,CAAC;AAKD,MAAa,gBAAgB;IACjB,MAAM,GAA0B,EAAE,CAAC;IAE3C,YAAY,cAAmD;QAE3D,CAAC,KAAK,IAAI,EAAE;YACR,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,cAAc,EAAE,CAAC;gBACvC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC5B,CAAC;QACL,CAAC,CAAC,EAAE,CAAC;IACT,CAAC;IAKD,KAAK,CAAC,iBAAiB;QAEnB,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;IAC3D,CAAC;IAKD,QAAQ,CAAC,IAAY;QACjB,OAAO,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;IAClD,CAAC;IAKD,SAAS,CAAC,IAAY;QAClB,OAAO,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;IACpD,CAAC;IAKD,WAAW,CAAC,IAAY;QACpB,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;QAChD,OAAO,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE;YAC3B,IAAI,WAAW,IAAI,KAAK,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;gBAC1C,OAAO,KAAK,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,KAAK,IAAI,CAAC;YAClD,CAAC;YACD,OAAO,KAAK,CAAC;QACjB,CAAC,CAAC,CAAC;IACP,CAAC;IAKD,eAAe;QACX,MAAM,aAAa,GAAG,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC;QACzD,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC3B,MAAM,SAAS,GAAG,aAAa,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAC1D,OAAO,SAAS,IAAI,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC;QAClE,CAAC;QACD,OAAO,SAAS,CAAC;IACrB,CAAC;IAKD,QAAQ;QACJ,OAAO,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;IAClC,CAAC;IAKD,QAAQ;QACJ,MAAM,WAAW,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAC5C,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzB,MAAM,UAAU,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC;YAClC,OAAO,OAAO,IAAI,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;QAChE,CAAC;QACD,OAAO,SAAS,CAAC;IACrB,CAAC;CACJ;AA7ED,4CA6EC"}
|
|
@@ -0,0 +1,139 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.timeoutPromise = timeoutPromise;
|
|
4
|
+
exports.agentHasStatusTracking = agentHasStatusTracking;
|
|
5
|
+
exports.executeToolWithLifecycle = executeToolWithLifecycle;
|
|
6
|
+
exports.handleToolCall = handleToolCall;
|
|
7
|
+
exports.prepareToolArguments = prepareToolArguments;
|
|
8
|
+
const uuid_1 = require("uuid");
|
|
9
|
+
const running_tool_tracker_js_1 = require("./running_tool_tracker.cjs");
|
|
10
|
+
const sequential_queue_js_1 = require("./sequential_queue.cjs");
|
|
11
|
+
const tool_parameter_utils_js_1 = require("./tool_parameter_utils.cjs");
|
|
12
|
+
const tool_execution_js_1 = require("../config/tool_execution.cjs");
|
|
13
|
+
function timeoutPromise(ms) {
|
|
14
|
+
return new Promise(resolve => {
|
|
15
|
+
setTimeout(() => resolve('TIMEOUT'), ms);
|
|
16
|
+
});
|
|
17
|
+
}
|
|
18
|
+
function agentHasStatusTracking(agent) {
|
|
19
|
+
if (!agent.tools)
|
|
20
|
+
return false;
|
|
21
|
+
return agent.tools.some(tool => tool_execution_js_1.STATUS_TRACKING_TOOLS.has(tool.definition.function.name));
|
|
22
|
+
}
|
|
23
|
+
async function executeToolWithLifecycle(toolCall, tool, agent, signal) {
|
|
24
|
+
const fnId = toolCall.id || (0, uuid_1.v4)();
|
|
25
|
+
const toolName = toolCall.function.name;
|
|
26
|
+
const argsString = toolCall.function.arguments || '{}';
|
|
27
|
+
let args;
|
|
28
|
+
try {
|
|
29
|
+
args = prepareToolArguments(argsString, tool);
|
|
30
|
+
}
|
|
31
|
+
catch (error) {
|
|
32
|
+
throw new Error(`Invalid JSON in tool arguments: ${error}`);
|
|
33
|
+
}
|
|
34
|
+
const runningTool = running_tool_tracker_js_1.runningToolTracker.addRunningTool(fnId, toolName, agent.agent_id || 'unknown', argsString);
|
|
35
|
+
const abortSignal = signal || runningTool.abortController?.signal;
|
|
36
|
+
try {
|
|
37
|
+
if (tool.injectAgentId) {
|
|
38
|
+
args.unshift(agent.agent_id || 'ensemble');
|
|
39
|
+
}
|
|
40
|
+
if (tool.injectAbortSignal && abortSignal) {
|
|
41
|
+
args.push(abortSignal);
|
|
42
|
+
}
|
|
43
|
+
const result = await tool.function(...args);
|
|
44
|
+
await running_tool_tracker_js_1.runningToolTracker.completeRunningTool(fnId, String(result), agent);
|
|
45
|
+
return String(result);
|
|
46
|
+
}
|
|
47
|
+
catch (error) {
|
|
48
|
+
await running_tool_tracker_js_1.runningToolTracker.failRunningTool(fnId, String(error), agent);
|
|
49
|
+
throw error;
|
|
50
|
+
}
|
|
51
|
+
}
|
|
52
|
+
async function handleToolCall(toolCall, tool, agent) {
|
|
53
|
+
const fnId = toolCall.id || (0, uuid_1.v4)();
|
|
54
|
+
const toolName = toolCall.function.name;
|
|
55
|
+
const executeFunction = async () => {
|
|
56
|
+
if (toolName === 'wait_for_running_tool') {
|
|
57
|
+
return executeToolWithLifecycle(toolCall, tool, agent);
|
|
58
|
+
}
|
|
59
|
+
return Promise.race([
|
|
60
|
+
executeToolWithLifecycle(toolCall, tool, agent),
|
|
61
|
+
new Promise((_, reject) => {
|
|
62
|
+
const runningTool = running_tool_tracker_js_1.runningToolTracker.getRunningTool(fnId);
|
|
63
|
+
if (runningTool?.abortController?.signal) {
|
|
64
|
+
runningTool.abortController.signal.addEventListener('abort', () => reject(new Error('Operation was aborted')), { once: true });
|
|
65
|
+
}
|
|
66
|
+
}),
|
|
67
|
+
]);
|
|
68
|
+
};
|
|
69
|
+
const sequential = !!agent.modelSettings?.sequential_tools;
|
|
70
|
+
const hasStatusTools = agentHasStatusTracking(agent);
|
|
71
|
+
const excludedFromTimeout = tool_execution_js_1.EXCLUDED_FROM_TIMEOUT_FUNCTIONS.has(toolName);
|
|
72
|
+
const shouldTimeout = !excludedFromTimeout && hasStatusTools && !sequential;
|
|
73
|
+
const execute = sequential ? () => (0, sequential_queue_js_1.runSequential)(agent.agent_id || 'unknown', executeFunction) : executeFunction;
|
|
74
|
+
if (!shouldTimeout) {
|
|
75
|
+
return execute();
|
|
76
|
+
}
|
|
77
|
+
const result = await Promise.race([
|
|
78
|
+
execute().catch(error => {
|
|
79
|
+
throw error;
|
|
80
|
+
}),
|
|
81
|
+
timeoutPromise(tool_execution_js_1.FUNCTION_TIMEOUT_MS),
|
|
82
|
+
]);
|
|
83
|
+
if (result === 'TIMEOUT') {
|
|
84
|
+
running_tool_tracker_js_1.runningToolTracker.markTimedOut(fnId);
|
|
85
|
+
return `Tool ${toolName} is running in the background (RunningTool: ${fnId}).`;
|
|
86
|
+
}
|
|
87
|
+
return result;
|
|
88
|
+
}
|
|
89
|
+
function prepareToolArguments(argsString, tool) {
|
|
90
|
+
let args;
|
|
91
|
+
try {
|
|
92
|
+
if (!argsString || argsString.trim() === '') {
|
|
93
|
+
args = {};
|
|
94
|
+
}
|
|
95
|
+
else {
|
|
96
|
+
args = JSON.parse(argsString);
|
|
97
|
+
}
|
|
98
|
+
}
|
|
99
|
+
catch (error) {
|
|
100
|
+
throw new Error(`Invalid JSON in tool arguments: ${error}`);
|
|
101
|
+
}
|
|
102
|
+
if (typeof args === 'object' && args !== null && !Array.isArray(args)) {
|
|
103
|
+
const paramNames = Object.keys(tool.definition.function.parameters.properties);
|
|
104
|
+
Object.keys(args).forEach(key => {
|
|
105
|
+
if (!paramNames.includes(key)) {
|
|
106
|
+
console.warn(`Removing unknown parameter "${key}" for tool "${tool.definition.function.name}"`);
|
|
107
|
+
delete args[key];
|
|
108
|
+
}
|
|
109
|
+
});
|
|
110
|
+
if (paramNames.length > 0) {
|
|
111
|
+
return paramNames.map(param => {
|
|
112
|
+
const value = args[param];
|
|
113
|
+
const paramSpec = tool.definition.function.parameters.properties[param];
|
|
114
|
+
if ((value === undefined || value === '') &&
|
|
115
|
+
!tool.definition.function.parameters.required?.includes(param)) {
|
|
116
|
+
return undefined;
|
|
117
|
+
}
|
|
118
|
+
const [coercedValue, error] = (0, tool_parameter_utils_js_1.coerceValue)(value, paramSpec, param);
|
|
119
|
+
if (error && tool.definition.function.parameters.required?.includes(param)) {
|
|
120
|
+
throw new Error(JSON.stringify({
|
|
121
|
+
error: {
|
|
122
|
+
param,
|
|
123
|
+
expected: paramSpec.type + (paramSpec.items?.type ? `<${paramSpec.items.type}>` : ''),
|
|
124
|
+
received: String(value),
|
|
125
|
+
message: error,
|
|
126
|
+
},
|
|
127
|
+
}));
|
|
128
|
+
}
|
|
129
|
+
else if (error) {
|
|
130
|
+
console.warn(`Parameter coercion warning for ${param}: ${error}`);
|
|
131
|
+
}
|
|
132
|
+
return coercedValue;
|
|
133
|
+
});
|
|
134
|
+
}
|
|
135
|
+
return Object.values(args);
|
|
136
|
+
}
|
|
137
|
+
return [args];
|
|
138
|
+
}
|
|
139
|
+
//# sourceMappingURL=tool_execution_manager.js.map
|
|
@@ -0,0 +1,7 @@
|
|
|
1
|
+
import { ToolCall, ToolFunction, AgentDefinition } from "../types/types.cjs";
|
|
2
|
+
export declare function timeoutPromise(ms: number): Promise<'TIMEOUT'>;
|
|
3
|
+
export declare function agentHasStatusTracking(agent: AgentDefinition): boolean;
|
|
4
|
+
export declare function executeToolWithLifecycle(toolCall: ToolCall, tool: ToolFunction, agent: AgentDefinition, signal?: AbortSignal): Promise<string>;
|
|
5
|
+
export declare function handleToolCall(toolCall: ToolCall, tool: ToolFunction, agent: AgentDefinition): Promise<string>;
|
|
6
|
+
export declare function prepareToolArguments(argsString: string, tool: ToolFunction): any[];
|
|
7
|
+
//# sourceMappingURL=tool_execution_manager.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tool_execution_manager.d.ts","sourceRoot":"","sources":["../../../utils/tool_execution_manager.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,QAAQ,EAAE,YAAY,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAa5E,wBAAgB,cAAc,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,CAAC,CAI7D;AAKD,wBAAgB,sBAAsB,CAAC,KAAK,EAAE,eAAe,GAAG,OAAO,CAItE;AAKD,wBAAsB,wBAAwB,CAC1C,QAAQ,EAAE,QAAQ,EAClB,IAAI,EAAE,YAAY,EAClB,KAAK,EAAE,eAAe,EACtB,MAAM,CAAC,EAAE,WAAW,GACrB,OAAO,CAAC,MAAM,CAAC,CAwCjB;AAKD,wBAAsB,cAAc,CAAC,QAAQ,EAAE,QAAQ,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,eAAe,GAAG,OAAO,CAAC,MAAM,CAAC,CAuDpH;AAKD,wBAAgB,oBAAoB,CAAC,UAAU,EAAE,MAAM,EAAE,IAAI,EAAE,YAAY,GAAG,GAAG,EAAE,CAmElF"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tool_execution_manager.js","sourceRoot":"","sources":["../../../utils/tool_execution_manager.ts"],"names":[],"mappings":";;AAkBA,wCAIC;AAKD,wDAIC;AAKD,4DA6CC;AAKD,wCAuDC;AAKD,oDAmEC;AAjND,+BAAoC;AAEpC,uEAA+D;AAC/D,+DAAsD;AACtD,uEAAwD;AACxD,mEAIqC;AAKrC,SAAgB,cAAc,CAAC,EAAU;IACrC,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE;QACzB,UAAU,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,EAAE,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;AACP,CAAC;AAKD,SAAgB,sBAAsB,CAAC,KAAsB;IACzD,IAAI,CAAC,KAAK,CAAC,KAAK;QAAE,OAAO,KAAK,CAAC;IAE/B,OAAO,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,yCAAqB,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;AAC9F,CAAC;AAKM,KAAK,UAAU,wBAAwB,CAC1C,QAAkB,EAClB,IAAkB,EAClB,KAAsB,EACtB,MAAoB;IAEpB,MAAM,IAAI,GAAG,QAAQ,CAAC,EAAE,IAAI,IAAA,SAAM,GAAE,CAAC;IACrC,MAAM,QAAQ,GAAG,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC;IACxC,MAAM,UAAU,GAAG,QAAQ,CAAC,QAAQ,CAAC,SAAS,IAAI,IAAI,CAAC;IAGvD,IAAI,IAAW,CAAC;IAChB,IAAI,CAAC;QACD,IAAI,GAAG,oBAAoB,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC;IAClD,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACb,MAAM,IAAI,KAAK,CAAC,mCAAmC,KAAK,EAAE,CAAC,CAAC;IAChE,CAAC;IAGD,MAAM,WAAW,GAAG,4CAAkB,CAAC,cAAc,CAAC,IAAI,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,IAAI,SAAS,EAAE,UAAU,CAAC,CAAC;IAG/G,MAAM,WAAW,GAAG,MAAM,IAAI,WAAW,CAAC,eAAe,EAAE,MAAM,CAAC;IAElE,IAAI,CAAC;QAED,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YACrB,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,QAAQ,IAAI,UAAU,CAAC,CAAC;QAC/C,CAAC;QACD,IAAI,IAAI,CAAC,iBAAiB,IAAI,WAAW,EAAE,CAAC;YACxC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC3B,CAAC;QAGD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,GAAG,IAAI,CAAC,CAAC;QAG5C,MAAM,4CAAkB,CAAC,mBAAmB,CAAC,IAAI,EAAE,MAAM,CAAC,MAAM,CAAC,EAAE,KAAK,CAAC,CAAC;QAE1E,OAAO,MAAM,CAAC,MAAM,CAAC,CAAC;IAC1B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAEb,MAAM,4CAAkB,CAAC,eAAe,CAAC,IAAI,EAAE,MAAM,CAAC,KAAK,CAAC,EAAE,KAAK,CAAC,CAAC;QACrE,MAAM,KAAK,CAAC;IAChB,CAAC;AACL,CAAC;AAKM,KAAK,UAAU,cAAc,CAAC,QAAkB,EAAE,IAAkB,EAAE,KAAsB;IAC/F,MAAM,IAAI,GAAG,QAAQ,CAAC,EAAE,IAAI,IAAA,SAAM,GAAE,CAAC;IACrC,MAAM,QAAQ,GAAG,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC;IAGxC,MAAM,eAAe,GAAG,KAAK,IAAqB,EAAE;QAEhD,IAAI,QAAQ,KAAK,uBAAuB,EAAE,CAAC;YACvC,OAAO,wBAAwB,CAAC,QAAQ,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;QAC3D,CAAC;QAED,OAAO,OAAO,CAAC,IAAI,CAAC;YAChB,wBAAwB,CAAC,QAAQ,EAAE,IAAI,EAAE,KAAK,CAAC;YAC/C,IAAI,OAAO,CAAS,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE;gBAC9B,MAAM,WAAW,GAAG,4CAAkB,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;gBAC5D,IAAI,WAAW,EAAE,eAAe,EAAE,MAAM,EAAE,CAAC;oBACvC,WAAW,CAAC,eAAe,CAAC,MAAM,CAAC,gBAAgB,CAC/C,OAAO,EACP,GAAG,EAAE,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC,EAChD,EAAE,IAAI,EAAE,IAAI,EAAE,CACjB,CAAC;gBACN,CAAC;YACL,CAAC,CAAC;SACL,CAAC,CAAC;IACP,CAAC,CAAC;IAGF,MAAM,UAAU,GAAG,CAAC,CAAC,KAAK,CAAC,aAAa,EAAE,gBAAgB,CAAC;IAG3D,MAAM,cAAc,GAAG,sBAAsB,CAAC,KAAK,CAAC,CAAC;IACrD,MAAM,mBAAmB,GAAG,mDAA+B,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IAC1E,MAAM,aAAa,GAAG,CAAC,mBAAmB,IAAI,cAAc,IAAI,CAAC,UAAU,CAAC;IAG5E,MAAM,OAAO,GAAG,UAAU,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,IAAA,mCAAa,EAAC,KAAK,CAAC,QAAQ,IAAI,SAAS,EAAE,eAAe,CAAC,CAAC,CAAC,CAAC,eAAe,CAAC;IAEjH,IAAI,CAAC,aAAa,EAAE,CAAC;QACjB,OAAO,OAAO,EAAE,CAAC;IACrB,CAAC;IAGD,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC;QAC9B,OAAO,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE;YACpB,MAAM,KAAK,CAAC;QAChB,CAAC,CAAC;QACF,cAAc,CAAC,uCAAmB,CAAC;KACtC,CAAC,CAAC;IAEH,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;QACvB,4CAAkB,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;QACtC,OAAO,QAAQ,QAAQ,+CAA+C,IAAI,IAAI,CAAC;IACnF,CAAC;IAED,OAAO,MAAM,CAAC;AAClB,CAAC;AAKD,SAAgB,oBAAoB,CAAC,UAAkB,EAAE,IAAkB;IAEvE,IAAI,IAAS,CAAC;IACd,IAAI,CAAC;QACD,IAAI,CAAC,UAAU,IAAI,UAAU,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;YAC1C,IAAI,GAAG,EAAE,CAAC;QACd,CAAC;aAAM,CAAC;YACJ,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;QAClC,CAAC;IACL,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACb,MAAM,IAAI,KAAK,CAAC,mCAAmC,KAAK,EAAE,CAAC,CAAC;IAChE,CAAC;IAED,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;QAEpE,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;QAG/E,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE;YAC5B,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC5B,OAAO,CAAC,IAAI,CAAC,+BAA+B,GAAG,eAAe,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,IAAI,GAAG,CAAC,CAAC;gBAChG,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC;YACrB,CAAC;QACL,CAAC,CAAC,CAAC;QAGH,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxB,OAAO,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE;gBAC1B,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC;gBAC1B,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;gBAGxE,IACI,CAAC,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,EAAE,CAAC;oBACrC,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,UAAU,CAAC,QAAQ,EAAE,QAAQ,CAAC,KAAK,CAAC,EAChE,CAAC;oBACC,OAAO,SAAS,CAAC;gBACrB,CAAC;gBAGD,MAAM,CAAC,YAAY,EAAE,KAAK,CAAC,GAAG,IAAA,qCAAW,EAAC,KAAK,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC;gBAEnE,IAAI,KAAK,IAAI,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,UAAU,CAAC,QAAQ,EAAE,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;oBACzE,MAAM,IAAI,KAAK,CACX,IAAI,CAAC,SAAS,CAAC;wBACX,KAAK,EAAE;4BACH,KAAK;4BACL,QAAQ,EAAE,SAAS,CAAC,IAAI,GAAG,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,SAAS,CAAC,KAAK,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;4BACrF,QAAQ,EAAE,MAAM,CAAC,KAAK,CAAC;4BACvB,OAAO,EAAE,KAAK;yBACjB;qBACJ,CAAC,CACL,CAAC;gBACN,CAAC;qBAAM,IAAI,KAAK,EAAE,CAAC;oBACf,OAAO,CAAC,IAAI,CAAC,kCAAkC,KAAK,KAAK,KAAK,EAAE,CAAC,CAAC;gBACtE,CAAC;gBAED,OAAO,YAAY,CAAC;YACxB,CAAC,CAAC,CAAC;QACP,CAAC;QAGD,OAAO,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IAC/B,CAAC;IAGD,OAAO,CAAC,IAAI,CAAC,CAAC;AAClB,CAAC"}
|
|
@@ -0,0 +1,168 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.coerceValue = coerceValue;
|
|
4
|
+
exports.extractParameterNames = extractParameterNames;
|
|
5
|
+
exports.mapNamedToPositionalArgs = mapNamedToPositionalArgs;
|
|
6
|
+
function coerceValue(value, paramSpec, paramName) {
|
|
7
|
+
const expectedType = paramSpec.type || 'string';
|
|
8
|
+
if (value === undefined || value === null) {
|
|
9
|
+
if (paramSpec.default) {
|
|
10
|
+
return [paramSpec.default, null];
|
|
11
|
+
}
|
|
12
|
+
if (paramSpec.optional) {
|
|
13
|
+
return [undefined, null];
|
|
14
|
+
}
|
|
15
|
+
return [null, `Required parameter "${paramName}" is missing`];
|
|
16
|
+
}
|
|
17
|
+
if (value === '' && paramSpec.optional) {
|
|
18
|
+
return [undefined, null];
|
|
19
|
+
}
|
|
20
|
+
const actualType = Array.isArray(value) ? 'array' : typeof value;
|
|
21
|
+
switch (expectedType) {
|
|
22
|
+
case 'string':
|
|
23
|
+
if (actualType === 'string') {
|
|
24
|
+
return [value, null];
|
|
25
|
+
}
|
|
26
|
+
return [String(value), null];
|
|
27
|
+
case 'number':
|
|
28
|
+
if (actualType === 'number') {
|
|
29
|
+
return [value, null];
|
|
30
|
+
}
|
|
31
|
+
if (actualType === 'string') {
|
|
32
|
+
const num = Number(value);
|
|
33
|
+
if (!isNaN(num)) {
|
|
34
|
+
return [num, null];
|
|
35
|
+
}
|
|
36
|
+
return [null, `Cannot convert "${value}" to number`];
|
|
37
|
+
}
|
|
38
|
+
return [null, `Expected number but got ${actualType}`];
|
|
39
|
+
case 'boolean':
|
|
40
|
+
if (actualType === 'boolean') {
|
|
41
|
+
return [value, null];
|
|
42
|
+
}
|
|
43
|
+
if (actualType === 'string') {
|
|
44
|
+
const lower = value.toLowerCase();
|
|
45
|
+
if (lower === 'true')
|
|
46
|
+
return [true, null];
|
|
47
|
+
if (lower === 'false')
|
|
48
|
+
return [false, null];
|
|
49
|
+
return [null, `Cannot convert "${value}" to boolean`];
|
|
50
|
+
}
|
|
51
|
+
if (actualType === 'number') {
|
|
52
|
+
return [value !== 0, null];
|
|
53
|
+
}
|
|
54
|
+
return [null, `Expected boolean but got ${actualType}`];
|
|
55
|
+
case 'array':
|
|
56
|
+
if (actualType === 'array') {
|
|
57
|
+
return [value, null];
|
|
58
|
+
}
|
|
59
|
+
if (actualType === 'string') {
|
|
60
|
+
try {
|
|
61
|
+
const parsed = JSON.parse(value);
|
|
62
|
+
if (Array.isArray(parsed)) {
|
|
63
|
+
return [parsed, null];
|
|
64
|
+
}
|
|
65
|
+
}
|
|
66
|
+
catch {
|
|
67
|
+
if (value.includes(',')) {
|
|
68
|
+
return [value.split(',').map(s => s.trim()), null];
|
|
69
|
+
}
|
|
70
|
+
}
|
|
71
|
+
return [[value], null];
|
|
72
|
+
}
|
|
73
|
+
return [[value], null];
|
|
74
|
+
case 'object':
|
|
75
|
+
if (actualType === 'object' && !Array.isArray(value)) {
|
|
76
|
+
return [value, null];
|
|
77
|
+
}
|
|
78
|
+
if (actualType === 'string') {
|
|
79
|
+
try {
|
|
80
|
+
const parsed = JSON.parse(value);
|
|
81
|
+
if (typeof parsed === 'object' && !Array.isArray(parsed)) {
|
|
82
|
+
return [parsed, null];
|
|
83
|
+
}
|
|
84
|
+
}
|
|
85
|
+
catch {
|
|
86
|
+
return [null, `Cannot parse "${value}" as object`];
|
|
87
|
+
}
|
|
88
|
+
}
|
|
89
|
+
return [null, `Expected object but got ${actualType}`];
|
|
90
|
+
case 'null':
|
|
91
|
+
return [null, null];
|
|
92
|
+
default:
|
|
93
|
+
return [value, null];
|
|
94
|
+
}
|
|
95
|
+
}
|
|
96
|
+
function extractParameterNames(func) {
|
|
97
|
+
const funcStr = func.toString();
|
|
98
|
+
const cleanFuncStr = funcStr.replaceAll(/\n\s*/g, ' ');
|
|
99
|
+
const paramMatch = cleanFuncStr.match(/\(([^)]*)\)/);
|
|
100
|
+
if (!paramMatch || !paramMatch[1]) {
|
|
101
|
+
return [];
|
|
102
|
+
}
|
|
103
|
+
const params = paramMatch[1]
|
|
104
|
+
.split(',')
|
|
105
|
+
.map(p => p.trim())
|
|
106
|
+
.filter(Boolean);
|
|
107
|
+
const paramNames = [];
|
|
108
|
+
for (const param of params) {
|
|
109
|
+
let paramName = param;
|
|
110
|
+
if (paramName.includes('=')) {
|
|
111
|
+
paramName = paramName.split('=')[0].trim();
|
|
112
|
+
}
|
|
113
|
+
if (paramName.includes(':')) {
|
|
114
|
+
paramName = paramName.split(':')[0].trim();
|
|
115
|
+
}
|
|
116
|
+
if (paramName.startsWith('...')) {
|
|
117
|
+
paramName = paramName.substring(3);
|
|
118
|
+
}
|
|
119
|
+
if (paramName === 'inject_agent_id' || paramName === 'abort_signal') {
|
|
120
|
+
continue;
|
|
121
|
+
}
|
|
122
|
+
paramNames.push(paramName);
|
|
123
|
+
}
|
|
124
|
+
return paramNames;
|
|
125
|
+
}
|
|
126
|
+
function mapNamedToPositionalArgs(namedArgs, tool, injectAgentId, abortSignal) {
|
|
127
|
+
const paramNames = extractParameterNames(tool.function);
|
|
128
|
+
const paramSpecs = tool.definition.function.parameters.properties || {};
|
|
129
|
+
const requiredParams = tool.definition.function.parameters.required || [];
|
|
130
|
+
Object.keys(namedArgs).forEach(key => {
|
|
131
|
+
if (!paramNames.includes(key)) {
|
|
132
|
+
console.warn(`Removing unknown parameter "${key}" for tool "${tool.definition.function.name}"`);
|
|
133
|
+
delete namedArgs[key];
|
|
134
|
+
}
|
|
135
|
+
});
|
|
136
|
+
const positionalArgs = [];
|
|
137
|
+
for (const paramName of paramNames) {
|
|
138
|
+
const value = namedArgs[paramName];
|
|
139
|
+
const paramSpec = paramSpecs[paramName] || { type: 'string' };
|
|
140
|
+
if ((value === undefined || value === '') && !requiredParams.includes(paramName)) {
|
|
141
|
+
positionalArgs.push(undefined);
|
|
142
|
+
continue;
|
|
143
|
+
}
|
|
144
|
+
const [coercedValue, error] = coerceValue(value, paramSpec, paramName);
|
|
145
|
+
if (error && requiredParams.includes(paramName)) {
|
|
146
|
+
throw new Error(JSON.stringify({
|
|
147
|
+
error: {
|
|
148
|
+
param: paramName,
|
|
149
|
+
expected: paramSpec.type + (paramSpec.items?.type ? `<${paramSpec.items.type}>` : ''),
|
|
150
|
+
received: String(value),
|
|
151
|
+
message: error,
|
|
152
|
+
},
|
|
153
|
+
}));
|
|
154
|
+
}
|
|
155
|
+
else if (error) {
|
|
156
|
+
console.warn(`Parameter coercion warning for ${paramName}: ${error}`);
|
|
157
|
+
}
|
|
158
|
+
positionalArgs.push(coercedValue);
|
|
159
|
+
}
|
|
160
|
+
if (tool.injectAgentId && injectAgentId) {
|
|
161
|
+
positionalArgs.unshift(injectAgentId);
|
|
162
|
+
}
|
|
163
|
+
if (tool.injectAbortSignal && abortSignal) {
|
|
164
|
+
positionalArgs.push(abortSignal);
|
|
165
|
+
}
|
|
166
|
+
return positionalArgs;
|
|
167
|
+
}
|
|
168
|
+
//# sourceMappingURL=tool_parameter_utils.js.map
|
|
@@ -0,0 +1,5 @@
|
|
|
1
|
+
import { ToolParameter } from "../types/types.cjs";
|
|
2
|
+
export declare function coerceValue(value: any, paramSpec: ToolParameter, paramName: string): [any, string | null];
|
|
3
|
+
export declare function extractParameterNames(func: (...args: any[]) => any): string[];
|
|
4
|
+
export declare function mapNamedToPositionalArgs(namedArgs: Record<string, any>, tool: any, injectAgentId?: string, abortSignal?: AbortSignal): any[];
|
|
5
|
+
//# sourceMappingURL=tool_parameter_utils.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tool_parameter_utils.d.ts","sourceRoot":"","sources":["../../../utils/tool_parameter_utils.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAMlD,wBAAgB,WAAW,CAAC,KAAK,EAAE,GAAG,EAAE,SAAS,EAAE,aAAa,EAAE,SAAS,EAAE,MAAM,GAAG,CAAC,GAAG,EAAE,MAAM,GAAG,IAAI,CAAC,CAsGzG;AAKD,wBAAgB,qBAAqB,CAAC,IAAI,EAAE,CAAC,GAAG,IAAI,EAAE,GAAG,EAAE,KAAK,GAAG,GAAG,MAAM,EAAE,CA6C7E;AAKD,wBAAgB,wBAAwB,CACpC,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAC9B,IAAI,EAAE,GAAG,EACT,aAAa,CAAC,EAAE,MAAM,EACtB,WAAW,CAAC,EAAE,WAAW,GAC1B,GAAG,EAAE,CA0DP"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tool_parameter_utils.js","sourceRoot":"","sources":["../../../utils/tool_parameter_utils.ts"],"names":[],"mappings":";;AAUA,kCAsGC;AAKD,sDA6CC;AAKD,4DA+DC;AA5ND,SAAgB,WAAW,CAAC,KAAU,EAAE,SAAwB,EAAE,SAAiB;IAC/E,MAAM,YAAY,GAAG,SAAS,CAAC,IAAI,IAAI,QAAQ,CAAC;IAGhD,IAAI,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;QACxC,IAAI,SAAS,CAAC,OAAO,EAAE,CAAC;YACpB,OAAO,CAAC,SAAS,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QACrC,CAAC;QACD,IAAI,SAAS,CAAC,QAAQ,EAAE,CAAC;YACrB,OAAO,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;QAC7B,CAAC;QACD,OAAO,CAAC,IAAI,EAAE,uBAAuB,SAAS,cAAc,CAAC,CAAC;IAClE,CAAC;IAGD,IAAI,KAAK,KAAK,EAAE,IAAI,SAAS,CAAC,QAAQ,EAAE,CAAC;QACrC,OAAO,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;IAC7B,CAAC;IAED,MAAM,UAAU,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,KAAK,CAAC;IAGjE,QAAQ,YAAY,EAAE,CAAC;QACnB,KAAK,QAAQ;YACT,IAAI,UAAU,KAAK,QAAQ,EAAE,CAAC;gBAC1B,OAAO,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;YACzB,CAAC;YAED,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,IAAI,CAAC,CAAC;QAEjC,KAAK,QAAQ;YACT,IAAI,UAAU,KAAK,QAAQ,EAAE,CAAC;gBAC1B,OAAO,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;YACzB,CAAC;YACD,IAAI,UAAU,KAAK,QAAQ,EAAE,CAAC;gBAC1B,MAAM,GAAG,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;gBAC1B,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC;oBACd,OAAO,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;gBACvB,CAAC;gBACD,OAAO,CAAC,IAAI,EAAE,mBAAmB,KAAK,aAAa,CAAC,CAAC;YACzD,CAAC;YACD,OAAO,CAAC,IAAI,EAAE,2BAA2B,UAAU,EAAE,CAAC,CAAC;QAE3D,KAAK,SAAS;YACV,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;gBAC3B,OAAO,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;YACzB,CAAC;YACD,IAAI,UAAU,KAAK,QAAQ,EAAE,CAAC;gBAC1B,MAAM,KAAK,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;gBAClC,IAAI,KAAK,KAAK,MAAM;oBAAE,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;gBAC1C,IAAI,KAAK,KAAK,OAAO;oBAAE,OAAO,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;gBAC5C,OAAO,CAAC,IAAI,EAAE,mBAAmB,KAAK,cAAc,CAAC,CAAC;YAC1D,CAAC;YACD,IAAI,UAAU,KAAK,QAAQ,EAAE,CAAC;gBAC1B,OAAO,CAAC,KAAK,KAAK,CAAC,EAAE,IAAI,CAAC,CAAC;YAC/B,CAAC;YACD,OAAO,CAAC,IAAI,EAAE,4BAA4B,UAAU,EAAE,CAAC,CAAC;QAE5D,KAAK,OAAO;YACR,IAAI,UAAU,KAAK,OAAO,EAAE,CAAC;gBACzB,OAAO,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;YACzB,CAAC;YACD,IAAI,UAAU,KAAK,QAAQ,EAAE,CAAC;gBAC1B,IAAI,CAAC;oBACD,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;oBACjC,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;wBACxB,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;oBAC1B,CAAC;gBACL,CAAC;gBAAC,MAAM,CAAC;oBAEL,IAAI,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;wBACtB,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,EAAE,IAAI,CAAC,CAAC;oBACvD,CAAC;gBACL,CAAC;gBAED,OAAO,CAAC,CAAC,KAAK,CAAC,EAAE,IAAI,CAAC,CAAC;YAC3B,CAAC;YAED,OAAO,CAAC,CAAC,KAAK,CAAC,EAAE,IAAI,CAAC,CAAC;QAE3B,KAAK,QAAQ;YACT,IAAI,UAAU,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;gBACnD,OAAO,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;YACzB,CAAC;YACD,IAAI,UAAU,KAAK,QAAQ,EAAE,CAAC;gBAC1B,IAAI,CAAC;oBACD,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;oBACjC,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;wBACvD,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;oBAC1B,CAAC;gBACL,CAAC;gBAAC,MAAM,CAAC;oBACL,OAAO,CAAC,IAAI,EAAE,iBAAiB,KAAK,aAAa,CAAC,CAAC;gBACvD,CAAC;YACL,CAAC;YACD,OAAO,CAAC,IAAI,EAAE,2BAA2B,UAAU,EAAE,CAAC,CAAC;QAE3D,KAAK,MAAM;YACP,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;QAExB;YACI,OAAO,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IAC7B,CAAC;AACL,CAAC;AAKD,SAAgB,qBAAqB,CAAC,IAA6B;IAC/D,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;IAGhC,MAAM,YAAY,GAAG,OAAO,CAAC,UAAU,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;IACvD,MAAM,UAAU,GAAG,YAAY,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;IAErD,IAAI,CAAC,UAAU,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC;QAChC,OAAO,EAAE,CAAC;IACd,CAAC;IAED,MAAM,MAAM,GAAG,UAAU,CAAC,CAAC,CAAC;SACvB,KAAK,CAAC,GAAG,CAAC;SACV,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;SAClB,MAAM,CAAC,OAAO,CAAC,CAAC;IAErB,MAAM,UAAU,GAAa,EAAE,CAAC;IAEhC,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QACzB,IAAI,SAAS,GAAG,KAAK,CAAC;QAGtB,IAAI,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;YAC1B,SAAS,GAAG,SAAS,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAC/C,CAAC;QAGD,IAAI,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;YAC1B,SAAS,GAAG,SAAS,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAC/C,CAAC;QAGD,IAAI,SAAS,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;YAC9B,SAAS,GAAG,SAAS,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;QACvC,CAAC;QAGD,IAAI,SAAS,KAAK,iBAAiB,IAAI,SAAS,KAAK,cAAc,EAAE,CAAC;YAClE,SAAS;QACb,CAAC;QAED,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAC/B,CAAC;IAED,OAAO,UAAU,CAAC;AACtB,CAAC;AAKD,SAAgB,wBAAwB,CACpC,SAA8B,EAC9B,IAAS,EACT,aAAsB,EACtB,WAAyB;IAGzB,MAAM,UAAU,GAAG,qBAAqB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACxD,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,UAAU,CAAC,UAAU,IAAI,EAAE,CAAC;IACxE,MAAM,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,UAAU,CAAC,QAAQ,IAAI,EAAE,CAAC;IAG1E,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE;QACjC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;YAC5B,OAAO,CAAC,IAAI,CAAC,+BAA+B,GAAG,eAAe,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,IAAI,GAAG,CAAC,CAAC;YAChG,OAAO,SAAS,CAAC,GAAG,CAAC,CAAC;QAC1B,CAAC;IACL,CAAC,CAAC,CAAC;IAGH,MAAM,cAAc,GAAU,EAAE,CAAC;IAEjC,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;QACjC,MAAM,KAAK,GAAG,SAAS,CAAC,SAAS,CAAC,CAAC;QACnC,MAAM,SAAS,GAAG,UAAU,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;QAG9D,IAAI,CAAC,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;YAC/E,cAAc,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC/B,SAAS;QACb,CAAC;QAGD,MAAM,CAAC,YAAY,EAAE,KAAK,CAAC,GAAG,WAAW,CAAC,KAAK,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC;QAEvE,IAAI,KAAK,IAAI,cAAc,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;YAC9C,MAAM,IAAI,KAAK,CACX,IAAI,CAAC,SAAS,CAAC;gBACX,KAAK,EAAE;oBACH,KAAK,EAAE,SAAS;oBAChB,QAAQ,EAAE,SAAS,CAAC,IAAI,GAAG,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,SAAS,CAAC,KAAK,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;oBACrF,QAAQ,EAAE,MAAM,CAAC,KAAK,CAAC;oBACvB,OAAO,EAAE,KAAK;iBACjB;aACJ,CAAC,CACL,CAAC;QACN,CAAC;aAAM,IAAI,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,IAAI,CAAC,kCAAkC,SAAS,KAAK,KAAK,EAAE,CAAC,CAAC;QAC1E,CAAC;QAED,cAAc,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IACtC,CAAC;IAGD,IAAI,IAAI,CAAC,aAAa,IAAI,aAAa,EAAE,CAAC;QACtC,cAAc,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;IAC1C,CAAC;IAED,IAAI,IAAI,CAAC,iBAAiB,IAAI,WAAW,EAAE,CAAC;QACxC,cAAc,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IACrC,CAAC;IAED,OAAO,cAAc,CAAC;AAC1B,CAAC"}
|