@wingman-ai/gateway 0.2.2 → 0.2.4

package/dist/tests/agentInvokerSummarization.test.cjs

@@ -0,0 +1,455 @@
+"use strict";
+var __webpack_exports__ = {};
+const external_vitest_namespaceObject = require("vitest");
+const schema_cjs_namespaceObject = require("../cli/config/schema.cjs");
+const agentInvoker_cjs_namespaceObject = require("../cli/core/agentInvoker.cjs");
+const parseConfig = (input)=>{
+    const result = (0, schema_cjs_namespaceObject.validateConfig)(input);
+    if (!result.success || !result.data) throw new Error(result.error || "Expected config validation to succeed");
+    return result.data;
+};
+(0, external_vitest_namespaceObject.describe)("resolveSummarizationMiddlewareSettings", ()=>{
+    (0, external_vitest_namespaceObject.it)("returns default summarization settings from config defaults", ()=>{
+        const config = parseConfig({});
+        (0, external_vitest_namespaceObject.expect)((0, agentInvoker_cjs_namespaceObject.resolveSummarizationMiddlewareSettings)(config)).toEqual({
+            maxTokensBeforeSummary: 12000,
+            messagesToKeep: 8
+        });
+    });
+    (0, external_vitest_namespaceObject.it)("returns null when summarization is disabled", ()=>{
+        const config = parseConfig({
+            summarization: {
+                enabled: false
+            }
+        });
+        (0, external_vitest_namespaceObject.expect)((0, agentInvoker_cjs_namespaceObject.resolveSummarizationMiddlewareSettings)(config)).toBeNull();
+    });
+    (0, external_vitest_namespaceObject.it)("returns custom summarization settings when configured", ()=>{
+        const config = parseConfig({
+            summarization: {
+                enabled: true,
+                maxTokensBeforeSummary: 20000,
+                messagesToKeep: 10
+            }
+        });
+        (0, external_vitest_namespaceObject.expect)((0, agentInvoker_cjs_namespaceObject.resolveSummarizationMiddlewareSettings)(config)).toEqual({
+            maxTokensBeforeSummary: 20000,
+            messagesToKeep: 10
+        });
+    });
+});
+(0, external_vitest_namespaceObject.describe)("configureDeepAgentSummarizationMiddleware", ()=>{
+    (0, external_vitest_namespaceObject.it)("replaces built-in summarization middleware with configured settings", ()=>{
+        const agent = {
+            options: {
+                middleware: [
+                    {
+                        name: "todoListMiddleware"
+                    },
+                    {
+                        name: "SummarizationMiddleware",
+                        marker: "old"
+                    }
+                ]
+            }
+        };
+        (0, agentInvoker_cjs_namespaceObject.configureDeepAgentSummarizationMiddleware)(agent, {
+            maxTokensBeforeSummary: 9000,
+            messagesToKeep: 5
+        }, "openai:gpt-4o-mini");
+        (0, external_vitest_namespaceObject.expect)(agent.options.middleware).toHaveLength(2);
+        (0, external_vitest_namespaceObject.expect)(agent.options.middleware[1]?.name).toBe("SummarizationMiddleware");
+        (0, external_vitest_namespaceObject.expect)(agent.options.middleware[1]?.marker).toBeUndefined();
+    });
+    (0, external_vitest_namespaceObject.it)("removes built-in summarization middleware when disabled", ()=>{
+        const agent = {
+            options: {
+                middleware: [
+                    {
+                        name: "todoListMiddleware"
+                    },
+                    {
+                        name: "SummarizationMiddleware"
+                    },
+                    {
+                        name: "patchToolCallsMiddleware"
+                    }
+                ]
+            }
+        };
+        (0, agentInvoker_cjs_namespaceObject.configureDeepAgentSummarizationMiddleware)(agent, null);
+        (0, external_vitest_namespaceObject.expect)(agent.options.middleware.map((m)=>m.name)).toEqual([
+            "todoListMiddleware",
+            "patchToolCallsMiddleware"
+        ]);
+    });
+});
+(0, external_vitest_namespaceObject.describe)("resolveModelRetryMiddlewareSettings", ()=>{
+    (0, external_vitest_namespaceObject.it)("returns default retry settings when config defaults are used", ()=>{
+        const config = parseConfig({});
+        (0, external_vitest_namespaceObject.expect)((0, agentInvoker_cjs_namespaceObject.resolveModelRetryMiddlewareSettings)(config)).toEqual({
+            maxRetries: 2,
+            backoffFactor: 2,
+            initialDelayMs: 1000,
+            maxDelayMs: 60000,
+            jitter: true,
+            onFailure: "continue"
+        });
+    });
+    (0, external_vitest_namespaceObject.it)("returns configured retry settings when enabled", ()=>{
+        const config = parseConfig({
+            modelRetry: {
+                enabled: true,
+                maxRetries: 3,
+                backoffFactor: 1.5,
+                initialDelayMs: 250,
+                maxDelayMs: 5000,
+                jitter: false,
+                onFailure: "error"
+            }
+        });
+        (0, external_vitest_namespaceObject.expect)((0, agentInvoker_cjs_namespaceObject.resolveModelRetryMiddlewareSettings)(config)).toEqual({
+            maxRetries: 3,
+            backoffFactor: 1.5,
+            initialDelayMs: 250,
+            maxDelayMs: 5000,
+            jitter: false,
+            onFailure: "error"
+        });
+    });
+});
+(0, external_vitest_namespaceObject.describe)("resolveToolRetryMiddlewareSettings", ()=>{
+    (0, external_vitest_namespaceObject.it)("returns null by default", ()=>{
+        const config = parseConfig({});
+        (0, external_vitest_namespaceObject.expect)((0, agentInvoker_cjs_namespaceObject.resolveToolRetryMiddlewareSettings)(config)).toBeNull();
+    });
+    (0, external_vitest_namespaceObject.it)("returns configured retry settings and tool filter when enabled", ()=>{
+        const config = parseConfig({
+            toolRetry: {
+                enabled: true,
+                maxRetries: 4,
+                backoffFactor: 2,
+                initialDelayMs: 500,
+                maxDelayMs: 10000,
+                jitter: true,
+                onFailure: "continue",
+                tools: [
+                    "internet_search",
+                    "web_crawler"
+                ]
+            }
+        });
+        (0, external_vitest_namespaceObject.expect)((0, agentInvoker_cjs_namespaceObject.resolveToolRetryMiddlewareSettings)(config)).toEqual({
+            maxRetries: 4,
+            backoffFactor: 2,
+            initialDelayMs: 500,
+            maxDelayMs: 10000,
+            jitter: true,
+            onFailure: "continue",
+            tools: [
+                "internet_search",
+                "web_crawler"
+            ]
+        });
+    });
+});
+(0, external_vitest_namespaceObject.describe)("resolveHumanInTheLoopSettings", ()=>{
+    (0, external_vitest_namespaceObject.it)("returns null by default", ()=>{
+        const config = parseConfig({});
+        (0, external_vitest_namespaceObject.expect)((0, agentInvoker_cjs_namespaceObject.resolveHumanInTheLoopSettings)(config)).toBeNull();
+    });
+    (0, external_vitest_namespaceObject.it)("returns null when enabled but no tool policy is defined", ()=>{
+        const config = parseConfig({
+            humanInTheLoop: {
+                enabled: true,
+                interruptOn: {}
+            }
+        });
+        (0, external_vitest_namespaceObject.expect)((0, agentInvoker_cjs_namespaceObject.resolveHumanInTheLoopSettings)(config)).toBeNull();
+    });
+    (0, external_vitest_namespaceObject.it)("returns interrupt map when enabled with tool policies", ()=>{
+        const config = parseConfig({
+            humanInTheLoop: {
+                enabled: true,
+                interruptOn: {
+                    command_execute: {
+                        allowedDecisions: [
+                            "approve",
+                            "reject"
+                        ],
+                        description: "Command execution requires approval"
+                    },
+                    internet_search: false
+                }
+            }
+        });
+        (0, external_vitest_namespaceObject.expect)((0, agentInvoker_cjs_namespaceObject.resolveHumanInTheLoopSettings)(config)).toEqual({
+            interruptOn: {
+                command_execute: {
+                    allowedDecisions: [
+                        "approve",
+                        "reject"
+                    ],
+                    description: "Command execution requires approval"
+                },
+                internet_search: false
+            }
+        });
+    });
+});
+(0, external_vitest_namespaceObject.describe)("detectToolEventContext", ()=>{
+    (0, external_vitest_namespaceObject.it)("extracts tool context for tool start events", ()=>{
+        const detected = (0, agentInvoker_cjs_namespaceObject.detectToolEventContext)({
+            event: "on_tool_start",
+            name: "glob"
+        });
+        (0, external_vitest_namespaceObject.expect)(detected).toEqual({
+            event: "on_tool_start",
+            toolName: "glob"
+        });
+    });
+    (0, external_vitest_namespaceObject.it)("extracts tool context for tool error events", ()=>{
+        const detected = (0, agentInvoker_cjs_namespaceObject.detectToolEventContext)({
+            event: "on_tool_error",
+            name: "grep"
+        });
+        (0, external_vitest_namespaceObject.expect)(detected).toEqual({
+            event: "on_tool_error",
+            toolName: "grep"
+        });
+    });
+    (0, external_vitest_namespaceObject.it)("returns null for non-tool events", ()=>{
+        const detected = (0, agentInvoker_cjs_namespaceObject.detectToolEventContext)({
+            event: "on_chat_model_stream",
+            data: {
+                chunk: {
+                    text: "hello"
+                }
+            }
+        });
+        (0, external_vitest_namespaceObject.expect)(detected).toBeNull();
+    });
+});
+(0, external_vitest_namespaceObject.describe)("chunkHasAssistantText", ()=>{
+    (0, external_vitest_namespaceObject.it)("detects text in on_chat_model_stream chunks", ()=>{
+        (0, external_vitest_namespaceObject.expect)((0, agentInvoker_cjs_namespaceObject.chunkHasAssistantText)({
+            event: "on_chat_model_stream",
+            data: {
+                chunk: {
+                    content: [
+                        {
+                            type: "text",
+                            text: "hello"
+                        }
+                    ]
+                }
+            }
+        })).toBe(true);
+    });
+    (0, external_vitest_namespaceObject.it)("detects text in on_llm_stream chunks", ()=>{
+        (0, external_vitest_namespaceObject.expect)((0, agentInvoker_cjs_namespaceObject.chunkHasAssistantText)({
+            event: "on_llm_stream",
+            data: {
+                chunk: {
+                    text: "delta"
+                }
+            }
+        })).toBe(true);
+    });
+    (0, external_vitest_namespaceObject.it)("returns false when chunk has no assistant text", ()=>{
+        (0, external_vitest_namespaceObject.expect)((0, agentInvoker_cjs_namespaceObject.chunkHasAssistantText)({
+            event: "on_tool_start",
+            name: "glob"
+        })).toBe(false);
+    });
+});
+(0, external_vitest_namespaceObject.describe)("selectStreamingFallbackText", ()=>{
+    (0, external_vitest_namespaceObject.it)("returns assistant text introduced during the current invocation", ()=>{
+        const selected = (0, agentInvoker_cjs_namespaceObject.selectStreamingFallbackText)([
+            {
+                role: "assistant",
+                content: "stale message"
+            }
+        ], [
+            {
+                role: "assistant",
+                content: "stale message"
+            },
+            {
+                role: "assistant",
+                content: "fresh fallback"
+            }
+        ]);
+        (0, external_vitest_namespaceObject.expect)(selected).toBe("fresh fallback");
+    });
+    (0, external_vitest_namespaceObject.it)("returns undefined when no new assistant text is introduced", ()=>{
+        const selected = (0, agentInvoker_cjs_namespaceObject.selectStreamingFallbackText)([
+            {
+                role: "assistant",
+                content: "existing response"
+            }
+        ], [
+            {
+                role: "user",
+                content: "not assistant"
+            },
+            {
+                role: "assistant",
+                content: "existing response"
+            }
+        ]);
+        (0, external_vitest_namespaceObject.expect)(selected).toBeUndefined();
+    });
+    (0, external_vitest_namespaceObject.it)("handles repeated assistant messages by count", ()=>{
+        const selected = (0, agentInvoker_cjs_namespaceObject.selectStreamingFallbackText)([
+            {
+                role: "assistant",
+                content: "repeat"
+            }
+        ], [
+            {
+                role: "assistant",
+                content: "repeat"
+            },
+            {
+                role: "assistant",
+                content: "repeat"
+            }
+        ]);
+        (0, external_vitest_namespaceObject.expect)(selected).toBe("repeat");
+    });
+});
+(0, external_vitest_namespaceObject.describe)("detectStreamErrorMessage", ()=>{
+    (0, external_vitest_namespaceObject.it)("extracts non-tool stream error messages", ()=>{
+        const detected = (0, agentInvoker_cjs_namespaceObject.detectStreamErrorMessage)({
+            event: "on_chat_model_error",
+            data: {
+                error: {
+                    message: "Model call failed after retries"
+                }
+            }
+        });
+        (0, external_vitest_namespaceObject.expect)(detected).toBe("Model call failed after retries");
+    });
+    (0, external_vitest_namespaceObject.it)("ignores tool errors and non-error events", ()=>{
+        (0, external_vitest_namespaceObject.expect)((0, agentInvoker_cjs_namespaceObject.detectStreamErrorMessage)({
+            event: "on_tool_error",
+            data: {
+                error: "tool failed"
+            }
+        })).toBeUndefined();
+        (0, external_vitest_namespaceObject.expect)((0, agentInvoker_cjs_namespaceObject.detectStreamErrorMessage)({
+            event: "on_chat_model_stream",
+            data: {
+                chunk: {
+                    text: "hello"
+                }
+            }
+        })).toBeUndefined();
+    });
+});
+(0, external_vitest_namespaceObject.describe)("evaluateStreamingCompletion", ()=>{
+    (0, external_vitest_namespaceObject.it)("blocks with stream_error when no assistant text and stream error exists", ()=>{
+        const result = (0, agentInvoker_cjs_namespaceObject.evaluateStreamingCompletion)({
+            sawAssistantText: false,
+            fallbackText: void 0,
+            streamErrorMessage: "provider timeout"
+        });
+        (0, external_vitest_namespaceObject.expect)(result).toEqual({
+            status: "blocked",
+            reason: "stream_error",
+            message: "Model call failed: provider timeout"
+        });
+    });
+    (0, external_vitest_namespaceObject.it)("blocks with empty_stream_response when no text or fallback is present", ()=>{
+        const result = (0, agentInvoker_cjs_namespaceObject.evaluateStreamingCompletion)({
+            sawAssistantText: false,
+            fallbackText: void 0,
+            streamErrorMessage: void 0
+        });
+        (0, external_vitest_namespaceObject.expect)(result).toEqual({
+            status: "blocked",
+            reason: "empty_stream_response",
+            message: "Model completed without a response. Check provider logs for request errors."
+        });
+    });
+    (0, external_vitest_namespaceObject.it)("returns ok when assistant text exists", ()=>{
+        const result = (0, agentInvoker_cjs_namespaceObject.evaluateStreamingCompletion)({
+            sawAssistantText: true,
+            fallbackText: void 0,
+            streamErrorMessage: void 0
+        });
+        (0, external_vitest_namespaceObject.expect)(result).toEqual({
+            status: "ok"
+        });
+    });
+});
+(0, external_vitest_namespaceObject.describe)("LangGraph lifecycle termination", ()=>{
+    (0, external_vitest_namespaceObject.it)("tracks root LangGraph run id from parentless on_chain_start", ()=>{
+        const rootRunId = (0, agentInvoker_cjs_namespaceObject.trackRootLangGraphRunId)(void 0, {
+            event: "on_chain_start",
+            name: "LangGraph",
+            run_id: "root-run",
+            parent_ids: []
+        });
+        (0, external_vitest_namespaceObject.expect)(rootRunId).toBe("root-run");
+    });
+    (0, external_vitest_namespaceObject.it)("ignores non-root or non-LangGraph start events", ()=>{
+        const nestedRunId = (0, agentInvoker_cjs_namespaceObject.trackRootLangGraphRunId)(void 0, {
+            event: "on_chain_start",
+            name: "LangGraph",
+            run_id: "child-run",
+            parent_ids: [
+                "root-run"
+            ]
+        });
+        (0, external_vitest_namespaceObject.expect)(nestedRunId).toBeUndefined();
+        const otherChainRunId = (0, agentInvoker_cjs_namespaceObject.trackRootLangGraphRunId)(void 0, {
+            event: "on_chain_start",
+            name: "CustomChain",
+            run_id: "other-run",
+            parent_ids: []
+        });
+        (0, external_vitest_namespaceObject.expect)(otherChainRunId).toBeUndefined();
+    });
+    (0, external_vitest_namespaceObject.it)("detects terminal root LangGraph on_chain_end by run id", ()=>{
+        (0, external_vitest_namespaceObject.expect)((0, agentInvoker_cjs_namespaceObject.isRootLangGraphTerminalEvent)({
+            event: "on_chain_end",
+            name: "LangGraph",
+            run_id: "root-run",
+            parent_ids: []
+        }, "root-run")).toBe(true);
+    });
+    (0, external_vitest_namespaceObject.it)("does not treat nested, mismatched, or non-LangGraph chain end as terminal", ()=>{
+        (0, external_vitest_namespaceObject.expect)((0, agentInvoker_cjs_namespaceObject.isRootLangGraphTerminalEvent)({
+            event: "on_chain_end",
+            name: "LangGraph",
+            run_id: "child-run",
+            parent_ids: [
+                "root-run"
+            ]
+        }, "root-run")).toBe(false);
+        (0, external_vitest_namespaceObject.expect)((0, agentInvoker_cjs_namespaceObject.isRootLangGraphTerminalEvent)({
+            event: "on_chain_end",
+            name: "LangGraph",
+            run_id: "other-run",
+            parent_ids: []
+        }, "root-run")).toBe(false);
+        (0, external_vitest_namespaceObject.expect)((0, agentInvoker_cjs_namespaceObject.isRootLangGraphTerminalEvent)({
+            event: "on_chain_end",
+            name: "CustomChain",
+            run_id: "root-run",
+            parent_ids: []
+        }, "root-run")).toBe(false);
+        (0, external_vitest_namespaceObject.expect)((0, agentInvoker_cjs_namespaceObject.isRootLangGraphTerminalEvent)({
+            event: "on_chain_end",
+            name: "LangGraph",
+            run_id: "root-run",
+            parent_ids: []
+        }, void 0)).toBe(false);
+    });
+});
+for(var __rspack_i in __webpack_exports__)exports[__rspack_i] = __webpack_exports__[__rspack_i];
+Object.defineProperty(exports, '__esModule', {
+    value: true
+});

package/dist/tests/agentInvokerSummarization.test.d.ts

@@ -0,0 +1 @@
+export {};