@wingman-ai/gateway 0.2.2 → 0.2.3
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/.wingman/agents/coding/agent.md +174 -169
- package/.wingman/agents/coding/implementor.md +25 -1
- package/.wingman/agents/main/agent.md +4 -0
- package/README.md +1 -0
- package/dist/agent/config/agentConfig.cjs +1 -1
- package/dist/agent/config/agentConfig.js +1 -1
- package/dist/agent/config/modelFactory.cjs +22 -2
- package/dist/agent/config/modelFactory.d.ts +2 -0
- package/dist/agent/config/modelFactory.js +22 -2
- package/dist/agent/tests/modelFactory.test.cjs +12 -5
- package/dist/agent/tests/modelFactory.test.js +12 -5
- package/dist/cli/commands/init.cjs +7 -6
- package/dist/cli/commands/init.js +7 -6
- package/dist/cli/commands/provider.cjs +17 -3
- package/dist/cli/commands/provider.js +17 -3
- package/dist/cli/config/loader.cjs +27 -0
- package/dist/cli/config/loader.js +27 -0
- package/dist/cli/config/schema.cjs +80 -2
- package/dist/cli/config/schema.d.ts +88 -0
- package/dist/cli/config/schema.js +67 -1
- package/dist/cli/core/agentInvoker.cjs +191 -13
- package/dist/cli/core/agentInvoker.d.ts +42 -3
- package/dist/cli/core/agentInvoker.js +163 -9
- package/dist/cli/core/sessionManager.cjs +32 -5
- package/dist/cli/core/sessionManager.js +32 -5
- package/dist/cli/index.cjs +6 -5
- package/dist/cli/index.js +6 -5
- package/dist/cli/types.d.ts +32 -0
- package/dist/gateway/http/sessions.cjs +7 -7
- package/dist/gateway/http/sessions.js +7 -7
- package/dist/gateway/server.cjs +191 -41
- package/dist/gateway/server.d.ts +8 -1
- package/dist/gateway/server.js +191 -41
- package/dist/gateway/types.d.ts +1 -0
- package/dist/providers/codex.cjs +167 -0
- package/dist/providers/codex.d.ts +15 -0
- package/dist/providers/codex.js +127 -0
- package/dist/providers/credentials.cjs +8 -0
- package/dist/providers/credentials.js +8 -0
- package/dist/providers/registry.cjs +11 -0
- package/dist/providers/registry.d.ts +1 -1
- package/dist/providers/registry.js +11 -0
- package/dist/tests/agentInvokerSummarization.test.cjs +296 -0
- package/dist/tests/agentInvokerSummarization.test.d.ts +1 -0
- package/dist/tests/agentInvokerSummarization.test.js +290 -0
- package/dist/tests/cli-config-loader.test.cjs +88 -0
- package/dist/tests/cli-config-loader.test.js +88 -0
- package/dist/tests/codex-credentials-precedence.test.cjs +94 -0
- package/dist/tests/codex-credentials-precedence.test.d.ts +1 -0
- package/dist/tests/codex-credentials-precedence.test.js +88 -0
- package/dist/tests/codex-provider.test.cjs +186 -0
- package/dist/tests/codex-provider.test.d.ts +1 -0
- package/dist/tests/codex-provider.test.js +180 -0
- package/dist/tests/gateway.test.cjs +108 -1
- package/dist/tests/gateway.test.js +108 -1
- package/dist/tests/provider-command-codex.test.cjs +57 -0
- package/dist/tests/provider-command-codex.test.d.ts +1 -0
- package/dist/tests/provider-command-codex.test.js +51 -0
- package/dist/tests/sessionStateMessages.test.cjs +38 -0
- package/dist/tests/sessionStateMessages.test.js +38 -0
- package/dist/webui/assets/{index-DDsMIOTX.css → index-BVMavpud.css} +1 -1
- package/dist/webui/assets/index-DCB2aVVf.js +182 -0
- package/dist/webui/index.html +2 -2
- package/package.json +1 -1
- package/dist/webui/assets/index-CPhfGPHc.js +0 -182
|
@@ -1,6 +1,7 @@
|
|
|
1
1
|
import { existsSync, mkdirSync, readFileSync, writeFileSync } from "node:fs";
|
|
2
2
|
import { homedir } from "node:os";
|
|
3
3
|
import { join } from "node:path";
|
|
4
|
+
import { resolveCodexAuthFromFile } from "./codex.js";
|
|
4
5
|
import { getProviderSpec } from "./registry.js";
|
|
5
6
|
const CREDENTIALS_VERSION = 1;
|
|
6
7
|
const CREDENTIALS_DIR = join(homedir(), ".wingman");
|
|
@@ -86,6 +87,13 @@ function resolveProviderToken(providerName) {
|
|
|
86
87
|
envVar
|
|
87
88
|
};
|
|
88
89
|
}
|
|
90
|
+
if ("codex" === provider.name) {
|
|
91
|
+
const codexAuth = resolveCodexAuthFromFile();
|
|
92
|
+
if (codexAuth.accessToken) return {
|
|
93
|
+
token: codexAuth.accessToken,
|
|
94
|
+
source: "credentials"
|
|
95
|
+
};
|
|
96
|
+
}
|
|
89
97
|
const credentials = getProviderCredentials(provider.name);
|
|
90
98
|
const token = credentials?.accessToken ?? credentials?.apiKey ?? credentials?.refreshToken;
|
|
91
99
|
if (token && token.trim()) return {
|
|
@@ -47,6 +47,17 @@ const PROVIDERS = {
|
|
|
47
47
|
],
|
|
48
48
|
category: "model"
|
|
49
49
|
},
|
|
50
|
+
codex: {
|
|
51
|
+
name: "codex",
|
|
52
|
+
label: "OpenAI Codex",
|
|
53
|
+
type: "api-key",
|
|
54
|
+
envVars: [
|
|
55
|
+
"CODEX_ACCESS_TOKEN",
|
|
56
|
+
"CHATGPT_ACCESS_TOKEN"
|
|
57
|
+
],
|
|
58
|
+
baseURL: "https://chatgpt.com/backend-api/codex",
|
|
59
|
+
category: "model"
|
|
60
|
+
},
|
|
50
61
|
openrouter: {
|
|
51
62
|
name: "openrouter",
|
|
52
63
|
label: "OpenRouter",
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
export type ProviderAuthType = "api-key" | "oauth";
|
|
2
2
|
export type ProviderCategory = "model" | "voice";
|
|
3
|
-
export type ProviderName = "anthropic" | "openai" | "openrouter" | "copilot" | "xai" | "lmstudio" | "ollama" | "elevenlabs";
|
|
3
|
+
export type ProviderName = "anthropic" | "openai" | "codex" | "openrouter" | "copilot" | "xai" | "lmstudio" | "ollama" | "elevenlabs";
|
|
4
4
|
export interface ProviderOAuthConfig {
|
|
5
5
|
authorizationUrl: string;
|
|
6
6
|
tokenUrl: string;
|
|
@@ -17,6 +17,17 @@ const PROVIDERS = {
|
|
|
17
17
|
],
|
|
18
18
|
category: "model"
|
|
19
19
|
},
|
|
20
|
+
codex: {
|
|
21
|
+
name: "codex",
|
|
22
|
+
label: "OpenAI Codex",
|
|
23
|
+
type: "api-key",
|
|
24
|
+
envVars: [
|
|
25
|
+
"CODEX_ACCESS_TOKEN",
|
|
26
|
+
"CHATGPT_ACCESS_TOKEN"
|
|
27
|
+
],
|
|
28
|
+
baseURL: "https://chatgpt.com/backend-api/codex",
|
|
29
|
+
category: "model"
|
|
30
|
+
},
|
|
20
31
|
openrouter: {
|
|
21
32
|
name: "openrouter",
|
|
22
33
|
label: "OpenRouter",
|
|
@@ -0,0 +1,296 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __webpack_exports__ = {};
|
|
3
|
+
const external_vitest_namespaceObject = require("vitest");
|
|
4
|
+
const schema_cjs_namespaceObject = require("../cli/config/schema.cjs");
|
|
5
|
+
const agentInvoker_cjs_namespaceObject = require("../cli/core/agentInvoker.cjs");
|
|
6
|
+
const parseConfig = (input)=>{
|
|
7
|
+
const result = (0, schema_cjs_namespaceObject.validateConfig)(input);
|
|
8
|
+
if (!result.success || !result.data) throw new Error(result.error || "Expected config validation to succeed");
|
|
9
|
+
return result.data;
|
|
10
|
+
};
|
|
11
|
+
(0, external_vitest_namespaceObject.describe)("resolveSummarizationMiddlewareSettings", ()=>{
|
|
12
|
+
(0, external_vitest_namespaceObject.it)("returns default summarization settings from config defaults", ()=>{
|
|
13
|
+
const config = parseConfig({});
|
|
14
|
+
(0, external_vitest_namespaceObject.expect)((0, agentInvoker_cjs_namespaceObject.resolveSummarizationMiddlewareSettings)(config)).toEqual({
|
|
15
|
+
maxTokensBeforeSummary: 12000,
|
|
16
|
+
messagesToKeep: 8
|
|
17
|
+
});
|
|
18
|
+
});
|
|
19
|
+
(0, external_vitest_namespaceObject.it)("returns null when summarization is disabled", ()=>{
|
|
20
|
+
const config = parseConfig({
|
|
21
|
+
summarization: {
|
|
22
|
+
enabled: false
|
|
23
|
+
}
|
|
24
|
+
});
|
|
25
|
+
(0, external_vitest_namespaceObject.expect)((0, agentInvoker_cjs_namespaceObject.resolveSummarizationMiddlewareSettings)(config)).toBeNull();
|
|
26
|
+
});
|
|
27
|
+
(0, external_vitest_namespaceObject.it)("returns custom summarization settings when configured", ()=>{
|
|
28
|
+
const config = parseConfig({
|
|
29
|
+
summarization: {
|
|
30
|
+
enabled: true,
|
|
31
|
+
maxTokensBeforeSummary: 20000,
|
|
32
|
+
messagesToKeep: 10
|
|
33
|
+
}
|
|
34
|
+
});
|
|
35
|
+
(0, external_vitest_namespaceObject.expect)((0, agentInvoker_cjs_namespaceObject.resolveSummarizationMiddlewareSettings)(config)).toEqual({
|
|
36
|
+
maxTokensBeforeSummary: 20000,
|
|
37
|
+
messagesToKeep: 10
|
|
38
|
+
});
|
|
39
|
+
});
|
|
40
|
+
});
|
|
41
|
+
(0, external_vitest_namespaceObject.describe)("configureDeepAgentSummarizationMiddleware", ()=>{
|
|
42
|
+
(0, external_vitest_namespaceObject.it)("replaces built-in summarization middleware with configured settings", ()=>{
|
|
43
|
+
const agent = {
|
|
44
|
+
options: {
|
|
45
|
+
middleware: [
|
|
46
|
+
{
|
|
47
|
+
name: "todoListMiddleware"
|
|
48
|
+
},
|
|
49
|
+
{
|
|
50
|
+
name: "SummarizationMiddleware",
|
|
51
|
+
marker: "old"
|
|
52
|
+
}
|
|
53
|
+
]
|
|
54
|
+
}
|
|
55
|
+
};
|
|
56
|
+
(0, agentInvoker_cjs_namespaceObject.configureDeepAgentSummarizationMiddleware)(agent, {
|
|
57
|
+
maxTokensBeforeSummary: 9000,
|
|
58
|
+
messagesToKeep: 5
|
|
59
|
+
}, "openai:gpt-4o-mini");
|
|
60
|
+
(0, external_vitest_namespaceObject.expect)(agent.options.middleware).toHaveLength(2);
|
|
61
|
+
(0, external_vitest_namespaceObject.expect)(agent.options.middleware[1]?.name).toBe("SummarizationMiddleware");
|
|
62
|
+
(0, external_vitest_namespaceObject.expect)(agent.options.middleware[1]?.marker).toBeUndefined();
|
|
63
|
+
});
|
|
64
|
+
(0, external_vitest_namespaceObject.it)("removes built-in summarization middleware when disabled", ()=>{
|
|
65
|
+
const agent = {
|
|
66
|
+
options: {
|
|
67
|
+
middleware: [
|
|
68
|
+
{
|
|
69
|
+
name: "todoListMiddleware"
|
|
70
|
+
},
|
|
71
|
+
{
|
|
72
|
+
name: "SummarizationMiddleware"
|
|
73
|
+
},
|
|
74
|
+
{
|
|
75
|
+
name: "patchToolCallsMiddleware"
|
|
76
|
+
}
|
|
77
|
+
]
|
|
78
|
+
}
|
|
79
|
+
};
|
|
80
|
+
(0, agentInvoker_cjs_namespaceObject.configureDeepAgentSummarizationMiddleware)(agent, null);
|
|
81
|
+
(0, external_vitest_namespaceObject.expect)(agent.options.middleware.map((m)=>m.name)).toEqual([
|
|
82
|
+
"todoListMiddleware",
|
|
83
|
+
"patchToolCallsMiddleware"
|
|
84
|
+
]);
|
|
85
|
+
});
|
|
86
|
+
});
|
|
87
|
+
(0, external_vitest_namespaceObject.describe)("resolveModelRetryMiddlewareSettings", ()=>{
|
|
88
|
+
(0, external_vitest_namespaceObject.it)("returns default retry settings when config defaults are used", ()=>{
|
|
89
|
+
const config = parseConfig({});
|
|
90
|
+
(0, external_vitest_namespaceObject.expect)((0, agentInvoker_cjs_namespaceObject.resolveModelRetryMiddlewareSettings)(config)).toEqual({
|
|
91
|
+
maxRetries: 2,
|
|
92
|
+
backoffFactor: 2,
|
|
93
|
+
initialDelayMs: 1000,
|
|
94
|
+
maxDelayMs: 60000,
|
|
95
|
+
jitter: true,
|
|
96
|
+
onFailure: "continue"
|
|
97
|
+
});
|
|
98
|
+
});
|
|
99
|
+
(0, external_vitest_namespaceObject.it)("returns configured retry settings when enabled", ()=>{
|
|
100
|
+
const config = parseConfig({
|
|
101
|
+
modelRetry: {
|
|
102
|
+
enabled: true,
|
|
103
|
+
maxRetries: 3,
|
|
104
|
+
backoffFactor: 1.5,
|
|
105
|
+
initialDelayMs: 250,
|
|
106
|
+
maxDelayMs: 5000,
|
|
107
|
+
jitter: false,
|
|
108
|
+
onFailure: "error"
|
|
109
|
+
}
|
|
110
|
+
});
|
|
111
|
+
(0, external_vitest_namespaceObject.expect)((0, agentInvoker_cjs_namespaceObject.resolveModelRetryMiddlewareSettings)(config)).toEqual({
|
|
112
|
+
maxRetries: 3,
|
|
113
|
+
backoffFactor: 1.5,
|
|
114
|
+
initialDelayMs: 250,
|
|
115
|
+
maxDelayMs: 5000,
|
|
116
|
+
jitter: false,
|
|
117
|
+
onFailure: "error"
|
|
118
|
+
});
|
|
119
|
+
});
|
|
120
|
+
});
|
|
121
|
+
(0, external_vitest_namespaceObject.describe)("resolveToolRetryMiddlewareSettings", ()=>{
|
|
122
|
+
(0, external_vitest_namespaceObject.it)("returns null by default", ()=>{
|
|
123
|
+
const config = parseConfig({});
|
|
124
|
+
(0, external_vitest_namespaceObject.expect)((0, agentInvoker_cjs_namespaceObject.resolveToolRetryMiddlewareSettings)(config)).toBeNull();
|
|
125
|
+
});
|
|
126
|
+
(0, external_vitest_namespaceObject.it)("returns configured retry settings and tool filter when enabled", ()=>{
|
|
127
|
+
const config = parseConfig({
|
|
128
|
+
toolRetry: {
|
|
129
|
+
enabled: true,
|
|
130
|
+
maxRetries: 4,
|
|
131
|
+
backoffFactor: 2,
|
|
132
|
+
initialDelayMs: 500,
|
|
133
|
+
maxDelayMs: 10000,
|
|
134
|
+
jitter: true,
|
|
135
|
+
onFailure: "continue",
|
|
136
|
+
tools: [
|
|
137
|
+
"internet_search",
|
|
138
|
+
"web_crawler"
|
|
139
|
+
]
|
|
140
|
+
}
|
|
141
|
+
});
|
|
142
|
+
(0, external_vitest_namespaceObject.expect)((0, agentInvoker_cjs_namespaceObject.resolveToolRetryMiddlewareSettings)(config)).toEqual({
|
|
143
|
+
maxRetries: 4,
|
|
144
|
+
backoffFactor: 2,
|
|
145
|
+
initialDelayMs: 500,
|
|
146
|
+
maxDelayMs: 10000,
|
|
147
|
+
jitter: true,
|
|
148
|
+
onFailure: "continue",
|
|
149
|
+
tools: [
|
|
150
|
+
"internet_search",
|
|
151
|
+
"web_crawler"
|
|
152
|
+
]
|
|
153
|
+
});
|
|
154
|
+
});
|
|
155
|
+
});
|
|
156
|
+
(0, external_vitest_namespaceObject.describe)("resolveHumanInTheLoopSettings", ()=>{
|
|
157
|
+
(0, external_vitest_namespaceObject.it)("returns null by default", ()=>{
|
|
158
|
+
const config = parseConfig({});
|
|
159
|
+
(0, external_vitest_namespaceObject.expect)((0, agentInvoker_cjs_namespaceObject.resolveHumanInTheLoopSettings)(config)).toBeNull();
|
|
160
|
+
});
|
|
161
|
+
(0, external_vitest_namespaceObject.it)("returns null when enabled but no tool policy is defined", ()=>{
|
|
162
|
+
const config = parseConfig({
|
|
163
|
+
humanInTheLoop: {
|
|
164
|
+
enabled: true,
|
|
165
|
+
interruptOn: {}
|
|
166
|
+
}
|
|
167
|
+
});
|
|
168
|
+
(0, external_vitest_namespaceObject.expect)((0, agentInvoker_cjs_namespaceObject.resolveHumanInTheLoopSettings)(config)).toBeNull();
|
|
169
|
+
});
|
|
170
|
+
(0, external_vitest_namespaceObject.it)("returns interrupt map when enabled with tool policies", ()=>{
|
|
171
|
+
const config = parseConfig({
|
|
172
|
+
humanInTheLoop: {
|
|
173
|
+
enabled: true,
|
|
174
|
+
interruptOn: {
|
|
175
|
+
command_execute: {
|
|
176
|
+
allowedDecisions: [
|
|
177
|
+
"approve",
|
|
178
|
+
"reject"
|
|
179
|
+
],
|
|
180
|
+
description: "Command execution requires approval"
|
|
181
|
+
},
|
|
182
|
+
internet_search: false
|
|
183
|
+
}
|
|
184
|
+
}
|
|
185
|
+
});
|
|
186
|
+
(0, external_vitest_namespaceObject.expect)((0, agentInvoker_cjs_namespaceObject.resolveHumanInTheLoopSettings)(config)).toEqual({
|
|
187
|
+
interruptOn: {
|
|
188
|
+
command_execute: {
|
|
189
|
+
allowedDecisions: [
|
|
190
|
+
"approve",
|
|
191
|
+
"reject"
|
|
192
|
+
],
|
|
193
|
+
description: "Command execution requires approval"
|
|
194
|
+
},
|
|
195
|
+
internet_search: false
|
|
196
|
+
}
|
|
197
|
+
});
|
|
198
|
+
});
|
|
199
|
+
});
|
|
200
|
+
(0, external_vitest_namespaceObject.describe)("detectToolEventContext", ()=>{
|
|
201
|
+
(0, external_vitest_namespaceObject.it)("extracts tool context for tool start events", ()=>{
|
|
202
|
+
const detected = (0, agentInvoker_cjs_namespaceObject.detectToolEventContext)({
|
|
203
|
+
event: "on_tool_start",
|
|
204
|
+
name: "glob"
|
|
205
|
+
});
|
|
206
|
+
(0, external_vitest_namespaceObject.expect)(detected).toEqual({
|
|
207
|
+
event: "on_tool_start",
|
|
208
|
+
toolName: "glob"
|
|
209
|
+
});
|
|
210
|
+
});
|
|
211
|
+
(0, external_vitest_namespaceObject.it)("returns null for non-tool events", ()=>{
|
|
212
|
+
const detected = (0, agentInvoker_cjs_namespaceObject.detectToolEventContext)({
|
|
213
|
+
event: "on_chat_model_stream",
|
|
214
|
+
data: {
|
|
215
|
+
chunk: {
|
|
216
|
+
text: "hello"
|
|
217
|
+
}
|
|
218
|
+
}
|
|
219
|
+
});
|
|
220
|
+
(0, external_vitest_namespaceObject.expect)(detected).toBeNull();
|
|
221
|
+
});
|
|
222
|
+
});
|
|
223
|
+
(0, external_vitest_namespaceObject.describe)("chunkHasAssistantText", ()=>{
|
|
224
|
+
(0, external_vitest_namespaceObject.it)("detects text in on_chat_model_stream chunks", ()=>{
|
|
225
|
+
(0, external_vitest_namespaceObject.expect)((0, agentInvoker_cjs_namespaceObject.chunkHasAssistantText)({
|
|
226
|
+
event: "on_chat_model_stream",
|
|
227
|
+
data: {
|
|
228
|
+
chunk: {
|
|
229
|
+
content: [
|
|
230
|
+
{
|
|
231
|
+
type: "text",
|
|
232
|
+
text: "hello"
|
|
233
|
+
}
|
|
234
|
+
]
|
|
235
|
+
}
|
|
236
|
+
}
|
|
237
|
+
})).toBe(true);
|
|
238
|
+
});
|
|
239
|
+
(0, external_vitest_namespaceObject.it)("detects text in on_llm_stream chunks", ()=>{
|
|
240
|
+
(0, external_vitest_namespaceObject.expect)((0, agentInvoker_cjs_namespaceObject.chunkHasAssistantText)({
|
|
241
|
+
event: "on_llm_stream",
|
|
242
|
+
data: {
|
|
243
|
+
chunk: {
|
|
244
|
+
text: "delta"
|
|
245
|
+
}
|
|
246
|
+
}
|
|
247
|
+
})).toBe(true);
|
|
248
|
+
});
|
|
249
|
+
(0, external_vitest_namespaceObject.it)("returns false when chunk has no assistant text", ()=>{
|
|
250
|
+
(0, external_vitest_namespaceObject.expect)((0, agentInvoker_cjs_namespaceObject.chunkHasAssistantText)({
|
|
251
|
+
event: "on_tool_start",
|
|
252
|
+
name: "glob"
|
|
253
|
+
})).toBe(false);
|
|
254
|
+
});
|
|
255
|
+
});
|
|
256
|
+
(0, external_vitest_namespaceObject.describe)("selectStreamingFallbackText", ()=>{
|
|
257
|
+
(0, external_vitest_namespaceObject.it)("returns the most recent assistant message in the same invocation window", ()=>{
|
|
258
|
+
const selected = (0, agentInvoker_cjs_namespaceObject.selectStreamingFallbackText)([
|
|
259
|
+
{
|
|
260
|
+
role: "assistant",
|
|
261
|
+
createdAt: 1000,
|
|
262
|
+
content: "stale message"
|
|
263
|
+
},
|
|
264
|
+
{
|
|
265
|
+
role: "assistant",
|
|
266
|
+
createdAt: 4500,
|
|
267
|
+
content: "fresh fallback"
|
|
268
|
+
}
|
|
269
|
+
], 4000);
|
|
270
|
+
(0, external_vitest_namespaceObject.expect)(selected).toBe("fresh fallback");
|
|
271
|
+
});
|
|
272
|
+
(0, external_vitest_namespaceObject.it)("ignores stale, user-role, and empty-content messages", ()=>{
|
|
273
|
+
const selected = (0, agentInvoker_cjs_namespaceObject.selectStreamingFallbackText)([
|
|
274
|
+
{
|
|
275
|
+
role: "assistant",
|
|
276
|
+
createdAt: 1000,
|
|
277
|
+
content: "too old"
|
|
278
|
+
},
|
|
279
|
+
{
|
|
280
|
+
role: "user",
|
|
281
|
+
createdAt: 4900,
|
|
282
|
+
content: "not assistant"
|
|
283
|
+
},
|
|
284
|
+
{
|
|
285
|
+
role: "assistant",
|
|
286
|
+
createdAt: 5000,
|
|
287
|
+
content: " "
|
|
288
|
+
}
|
|
289
|
+
], 5000);
|
|
290
|
+
(0, external_vitest_namespaceObject.expect)(selected).toBeUndefined();
|
|
291
|
+
});
|
|
292
|
+
});
|
|
293
|
+
for(var __rspack_i in __webpack_exports__)exports[__rspack_i] = __webpack_exports__[__rspack_i];
|
|
294
|
+
Object.defineProperty(exports, '__esModule', {
|
|
295
|
+
value: true
|
|
296
|
+
});
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
export {};
|
|
@@ -0,0 +1,290 @@
|
|
|
1
|
+
import { describe, expect, it } from "vitest";
|
|
2
|
+
import { validateConfig } from "../cli/config/schema.js";
|
|
3
|
+
import { chunkHasAssistantText, configureDeepAgentSummarizationMiddleware, detectToolEventContext, resolveHumanInTheLoopSettings, resolveModelRetryMiddlewareSettings, resolveSummarizationMiddlewareSettings, resolveToolRetryMiddlewareSettings, selectStreamingFallbackText } from "../cli/core/agentInvoker.js";
|
|
4
|
+
const parseConfig = (input)=>{
|
|
5
|
+
const result = validateConfig(input);
|
|
6
|
+
if (!result.success || !result.data) throw new Error(result.error || "Expected config validation to succeed");
|
|
7
|
+
return result.data;
|
|
8
|
+
};
|
|
9
|
+
describe("resolveSummarizationMiddlewareSettings", ()=>{
|
|
10
|
+
it("returns default summarization settings from config defaults", ()=>{
|
|
11
|
+
const config = parseConfig({});
|
|
12
|
+
expect(resolveSummarizationMiddlewareSettings(config)).toEqual({
|
|
13
|
+
maxTokensBeforeSummary: 12000,
|
|
14
|
+
messagesToKeep: 8
|
|
15
|
+
});
|
|
16
|
+
});
|
|
17
|
+
it("returns null when summarization is disabled", ()=>{
|
|
18
|
+
const config = parseConfig({
|
|
19
|
+
summarization: {
|
|
20
|
+
enabled: false
|
|
21
|
+
}
|
|
22
|
+
});
|
|
23
|
+
expect(resolveSummarizationMiddlewareSettings(config)).toBeNull();
|
|
24
|
+
});
|
|
25
|
+
it("returns custom summarization settings when configured", ()=>{
|
|
26
|
+
const config = parseConfig({
|
|
27
|
+
summarization: {
|
|
28
|
+
enabled: true,
|
|
29
|
+
maxTokensBeforeSummary: 20000,
|
|
30
|
+
messagesToKeep: 10
|
|
31
|
+
}
|
|
32
|
+
});
|
|
33
|
+
expect(resolveSummarizationMiddlewareSettings(config)).toEqual({
|
|
34
|
+
maxTokensBeforeSummary: 20000,
|
|
35
|
+
messagesToKeep: 10
|
|
36
|
+
});
|
|
37
|
+
});
|
|
38
|
+
});
|
|
39
|
+
describe("configureDeepAgentSummarizationMiddleware", ()=>{
|
|
40
|
+
it("replaces built-in summarization middleware with configured settings", ()=>{
|
|
41
|
+
const agent = {
|
|
42
|
+
options: {
|
|
43
|
+
middleware: [
|
|
44
|
+
{
|
|
45
|
+
name: "todoListMiddleware"
|
|
46
|
+
},
|
|
47
|
+
{
|
|
48
|
+
name: "SummarizationMiddleware",
|
|
49
|
+
marker: "old"
|
|
50
|
+
}
|
|
51
|
+
]
|
|
52
|
+
}
|
|
53
|
+
};
|
|
54
|
+
configureDeepAgentSummarizationMiddleware(agent, {
|
|
55
|
+
maxTokensBeforeSummary: 9000,
|
|
56
|
+
messagesToKeep: 5
|
|
57
|
+
}, "openai:gpt-4o-mini");
|
|
58
|
+
expect(agent.options.middleware).toHaveLength(2);
|
|
59
|
+
expect(agent.options.middleware[1]?.name).toBe("SummarizationMiddleware");
|
|
60
|
+
expect(agent.options.middleware[1]?.marker).toBeUndefined();
|
|
61
|
+
});
|
|
62
|
+
it("removes built-in summarization middleware when disabled", ()=>{
|
|
63
|
+
const agent = {
|
|
64
|
+
options: {
|
|
65
|
+
middleware: [
|
|
66
|
+
{
|
|
67
|
+
name: "todoListMiddleware"
|
|
68
|
+
},
|
|
69
|
+
{
|
|
70
|
+
name: "SummarizationMiddleware"
|
|
71
|
+
},
|
|
72
|
+
{
|
|
73
|
+
name: "patchToolCallsMiddleware"
|
|
74
|
+
}
|
|
75
|
+
]
|
|
76
|
+
}
|
|
77
|
+
};
|
|
78
|
+
configureDeepAgentSummarizationMiddleware(agent, null);
|
|
79
|
+
expect(agent.options.middleware.map((m)=>m.name)).toEqual([
|
|
80
|
+
"todoListMiddleware",
|
|
81
|
+
"patchToolCallsMiddleware"
|
|
82
|
+
]);
|
|
83
|
+
});
|
|
84
|
+
});
|
|
85
|
+
describe("resolveModelRetryMiddlewareSettings", ()=>{
|
|
86
|
+
it("returns default retry settings when config defaults are used", ()=>{
|
|
87
|
+
const config = parseConfig({});
|
|
88
|
+
expect(resolveModelRetryMiddlewareSettings(config)).toEqual({
|
|
89
|
+
maxRetries: 2,
|
|
90
|
+
backoffFactor: 2,
|
|
91
|
+
initialDelayMs: 1000,
|
|
92
|
+
maxDelayMs: 60000,
|
|
93
|
+
jitter: true,
|
|
94
|
+
onFailure: "continue"
|
|
95
|
+
});
|
|
96
|
+
});
|
|
97
|
+
it("returns configured retry settings when enabled", ()=>{
|
|
98
|
+
const config = parseConfig({
|
|
99
|
+
modelRetry: {
|
|
100
|
+
enabled: true,
|
|
101
|
+
maxRetries: 3,
|
|
102
|
+
backoffFactor: 1.5,
|
|
103
|
+
initialDelayMs: 250,
|
|
104
|
+
maxDelayMs: 5000,
|
|
105
|
+
jitter: false,
|
|
106
|
+
onFailure: "error"
|
|
107
|
+
}
|
|
108
|
+
});
|
|
109
|
+
expect(resolveModelRetryMiddlewareSettings(config)).toEqual({
|
|
110
|
+
maxRetries: 3,
|
|
111
|
+
backoffFactor: 1.5,
|
|
112
|
+
initialDelayMs: 250,
|
|
113
|
+
maxDelayMs: 5000,
|
|
114
|
+
jitter: false,
|
|
115
|
+
onFailure: "error"
|
|
116
|
+
});
|
|
117
|
+
});
|
|
118
|
+
});
|
|
119
|
+
describe("resolveToolRetryMiddlewareSettings", ()=>{
|
|
120
|
+
it("returns null by default", ()=>{
|
|
121
|
+
const config = parseConfig({});
|
|
122
|
+
expect(resolveToolRetryMiddlewareSettings(config)).toBeNull();
|
|
123
|
+
});
|
|
124
|
+
it("returns configured retry settings and tool filter when enabled", ()=>{
|
|
125
|
+
const config = parseConfig({
|
|
126
|
+
toolRetry: {
|
|
127
|
+
enabled: true,
|
|
128
|
+
maxRetries: 4,
|
|
129
|
+
backoffFactor: 2,
|
|
130
|
+
initialDelayMs: 500,
|
|
131
|
+
maxDelayMs: 10000,
|
|
132
|
+
jitter: true,
|
|
133
|
+
onFailure: "continue",
|
|
134
|
+
tools: [
|
|
135
|
+
"internet_search",
|
|
136
|
+
"web_crawler"
|
|
137
|
+
]
|
|
138
|
+
}
|
|
139
|
+
});
|
|
140
|
+
expect(resolveToolRetryMiddlewareSettings(config)).toEqual({
|
|
141
|
+
maxRetries: 4,
|
|
142
|
+
backoffFactor: 2,
|
|
143
|
+
initialDelayMs: 500,
|
|
144
|
+
maxDelayMs: 10000,
|
|
145
|
+
jitter: true,
|
|
146
|
+
onFailure: "continue",
|
|
147
|
+
tools: [
|
|
148
|
+
"internet_search",
|
|
149
|
+
"web_crawler"
|
|
150
|
+
]
|
|
151
|
+
});
|
|
152
|
+
});
|
|
153
|
+
});
|
|
154
|
+
describe("resolveHumanInTheLoopSettings", ()=>{
|
|
155
|
+
it("returns null by default", ()=>{
|
|
156
|
+
const config = parseConfig({});
|
|
157
|
+
expect(resolveHumanInTheLoopSettings(config)).toBeNull();
|
|
158
|
+
});
|
|
159
|
+
it("returns null when enabled but no tool policy is defined", ()=>{
|
|
160
|
+
const config = parseConfig({
|
|
161
|
+
humanInTheLoop: {
|
|
162
|
+
enabled: true,
|
|
163
|
+
interruptOn: {}
|
|
164
|
+
}
|
|
165
|
+
});
|
|
166
|
+
expect(resolveHumanInTheLoopSettings(config)).toBeNull();
|
|
167
|
+
});
|
|
168
|
+
it("returns interrupt map when enabled with tool policies", ()=>{
|
|
169
|
+
const config = parseConfig({
|
|
170
|
+
humanInTheLoop: {
|
|
171
|
+
enabled: true,
|
|
172
|
+
interruptOn: {
|
|
173
|
+
command_execute: {
|
|
174
|
+
allowedDecisions: [
|
|
175
|
+
"approve",
|
|
176
|
+
"reject"
|
|
177
|
+
],
|
|
178
|
+
description: "Command execution requires approval"
|
|
179
|
+
},
|
|
180
|
+
internet_search: false
|
|
181
|
+
}
|
|
182
|
+
}
|
|
183
|
+
});
|
|
184
|
+
expect(resolveHumanInTheLoopSettings(config)).toEqual({
|
|
185
|
+
interruptOn: {
|
|
186
|
+
command_execute: {
|
|
187
|
+
allowedDecisions: [
|
|
188
|
+
"approve",
|
|
189
|
+
"reject"
|
|
190
|
+
],
|
|
191
|
+
description: "Command execution requires approval"
|
|
192
|
+
},
|
|
193
|
+
internet_search: false
|
|
194
|
+
}
|
|
195
|
+
});
|
|
196
|
+
});
|
|
197
|
+
});
|
|
198
|
+
describe("detectToolEventContext", ()=>{
|
|
199
|
+
it("extracts tool context for tool start events", ()=>{
|
|
200
|
+
const detected = detectToolEventContext({
|
|
201
|
+
event: "on_tool_start",
|
|
202
|
+
name: "glob"
|
|
203
|
+
});
|
|
204
|
+
expect(detected).toEqual({
|
|
205
|
+
event: "on_tool_start",
|
|
206
|
+
toolName: "glob"
|
|
207
|
+
});
|
|
208
|
+
});
|
|
209
|
+
it("returns null for non-tool events", ()=>{
|
|
210
|
+
const detected = detectToolEventContext({
|
|
211
|
+
event: "on_chat_model_stream",
|
|
212
|
+
data: {
|
|
213
|
+
chunk: {
|
|
214
|
+
text: "hello"
|
|
215
|
+
}
|
|
216
|
+
}
|
|
217
|
+
});
|
|
218
|
+
expect(detected).toBeNull();
|
|
219
|
+
});
|
|
220
|
+
});
|
|
221
|
+
describe("chunkHasAssistantText", ()=>{
|
|
222
|
+
it("detects text in on_chat_model_stream chunks", ()=>{
|
|
223
|
+
expect(chunkHasAssistantText({
|
|
224
|
+
event: "on_chat_model_stream",
|
|
225
|
+
data: {
|
|
226
|
+
chunk: {
|
|
227
|
+
content: [
|
|
228
|
+
{
|
|
229
|
+
type: "text",
|
|
230
|
+
text: "hello"
|
|
231
|
+
}
|
|
232
|
+
]
|
|
233
|
+
}
|
|
234
|
+
}
|
|
235
|
+
})).toBe(true);
|
|
236
|
+
});
|
|
237
|
+
it("detects text in on_llm_stream chunks", ()=>{
|
|
238
|
+
expect(chunkHasAssistantText({
|
|
239
|
+
event: "on_llm_stream",
|
|
240
|
+
data: {
|
|
241
|
+
chunk: {
|
|
242
|
+
text: "delta"
|
|
243
|
+
}
|
|
244
|
+
}
|
|
245
|
+
})).toBe(true);
|
|
246
|
+
});
|
|
247
|
+
it("returns false when chunk has no assistant text", ()=>{
|
|
248
|
+
expect(chunkHasAssistantText({
|
|
249
|
+
event: "on_tool_start",
|
|
250
|
+
name: "glob"
|
|
251
|
+
})).toBe(false);
|
|
252
|
+
});
|
|
253
|
+
});
|
|
254
|
+
describe("selectStreamingFallbackText", ()=>{
|
|
255
|
+
it("returns the most recent assistant message in the same invocation window", ()=>{
|
|
256
|
+
const selected = selectStreamingFallbackText([
|
|
257
|
+
{
|
|
258
|
+
role: "assistant",
|
|
259
|
+
createdAt: 1000,
|
|
260
|
+
content: "stale message"
|
|
261
|
+
},
|
|
262
|
+
{
|
|
263
|
+
role: "assistant",
|
|
264
|
+
createdAt: 4500,
|
|
265
|
+
content: "fresh fallback"
|
|
266
|
+
}
|
|
267
|
+
], 4000);
|
|
268
|
+
expect(selected).toBe("fresh fallback");
|
|
269
|
+
});
|
|
270
|
+
it("ignores stale, user-role, and empty-content messages", ()=>{
|
|
271
|
+
const selected = selectStreamingFallbackText([
|
|
272
|
+
{
|
|
273
|
+
role: "assistant",
|
|
274
|
+
createdAt: 1000,
|
|
275
|
+
content: "too old"
|
|
276
|
+
},
|
|
277
|
+
{
|
|
278
|
+
role: "user",
|
|
279
|
+
createdAt: 4900,
|
|
280
|
+
content: "not assistant"
|
|
281
|
+
},
|
|
282
|
+
{
|
|
283
|
+
role: "assistant",
|
|
284
|
+
createdAt: 5000,
|
|
285
|
+
content: " "
|
|
286
|
+
}
|
|
287
|
+
], 5000);
|
|
288
|
+
expect(selected).toBeUndefined();
|
|
289
|
+
});
|
|
290
|
+
});
|