@illuma-ai/agents 1.0.89 → 1.0.93

package/src/tools/__tests__/BrowserTools.test.js

@@ -0,0 +1,306 @@
+import { describe, it, expect, beforeEach } from '@jest/globals';
+import { createBrowserTools, hasBrowserCapability, EBrowserTools, } from '../BrowserTools';
+describe('BrowserTools', () => {
+    describe('EBrowserTools enum', () => {
+        it('should have all expected browser tool names', () => {
+            expect(EBrowserTools.CLICK).toBe('browser_click');
+            expect(EBrowserTools.TYPE).toBe('browser_type');
+            expect(EBrowserTools.NAVIGATE).toBe('browser_navigate');
+            expect(EBrowserTools.SCROLL).toBe('browser_scroll');
+            expect(EBrowserTools.EXTRACT).toBe('browser_extract');
+            expect(EBrowserTools.HOVER).toBe('browser_hover');
+            expect(EBrowserTools.WAIT).toBe('browser_wait');
+            expect(EBrowserTools.BACK).toBe('browser_back');
+            expect(EBrowserTools.SCREENSHOT).toBe('browser_screenshot');
+            expect(EBrowserTools.GET_PAGE_STATE).toBe('browser_get_page_state');
+            expect(EBrowserTools.KEYPRESS).toBe('browser_keypress');
+            expect(EBrowserTools.SWITCH_TAB).toBe('browser_switch_tab');
+        });
+        it('should have exactly 12 browser tools', () => {
+            expect(Object.keys(EBrowserTools).length).toBe(12);
+        });
+    });
+    describe('hasBrowserCapability', () => {
+        it('should return false when no request is provided', () => {
+            expect(hasBrowserCapability()).toBe(false);
+            expect(hasBrowserCapability(undefined)).toBe(false);
+        });
+        it('should return false when request has no headers', () => {
+            expect(hasBrowserCapability({})).toBe(false);
+            expect(hasBrowserCapability({ headers: undefined })).toBe(false);
+        });
+        it('should return true when X-Illuma-Browser-Extension header is true', () => {
+            const req = {
+                headers: {
+                    'x-illuma-browser-extension': 'true',
+                },
+            };
+            expect(hasBrowserCapability(req)).toBe(true);
+        });
+        it('should return true when X-Illuma-Browser-Capable header is true', () => {
+            const req = {
+                headers: {
+                    'x-illuma-browser-capable': 'true',
+                },
+            };
+            expect(hasBrowserCapability(req)).toBe(true);
+        });
+        it('should return false when browser headers are not true', () => {
+            const req = {
+                headers: {
+                    'x-illuma-browser-extension': 'false',
+                    'x-illuma-browser-capable': 'false',
+                },
+            };
+            expect(hasBrowserCapability(req)).toBe(false);
+        });
+        it('should return false when browser headers are missing', () => {
+            const req = {
+                headers: {
+                    'content-type': 'application/json',
+                },
+            };
+            expect(hasBrowserCapability(req)).toBe(false);
+        });
+    });
+    describe('createBrowserTools', () => {
+        let tools;
+        beforeEach(() => {
+            tools = createBrowserTools();
+        });
+        it('should create exactly 12 browser tools', () => {
+            expect(tools.length).toBe(12);
+        });
+        it('should create tools with correct names', () => {
+            const toolNames = tools.map((t) => t.name);
+            expect(toolNames).toContain('browser_click');
+            expect(toolNames).toContain('browser_type');
+            expect(toolNames).toContain('browser_navigate');
+            expect(toolNames).toContain('browser_scroll');
+            expect(toolNames).toContain('browser_extract');
+            expect(toolNames).toContain('browser_hover');
+            expect(toolNames).toContain('browser_wait');
+            expect(toolNames).toContain('browser_back');
+            expect(toolNames).toContain('browser_screenshot');
+            expect(toolNames).toContain('browser_get_page_state');
+        });
+        it('should have descriptions for all tools', () => {
+            for (const tool of tools) {
+                expect(tool.description).toBeDefined();
+                expect(tool.description.length).toBeGreaterThan(0);
+            }
+        });
+        describe('browser_click tool', () => {
+            it('should return requiresBrowserExecution marker', async () => {
+                const clickTool = tools.find((t) => t.name === 'browser_click');
+                expect(clickTool).toBeDefined();
+                const result = await clickTool.invoke({ index: 5 });
+                const parsed = JSON.parse(result);
+                expect(parsed.requiresBrowserExecution).toBe(true);
+                expect(parsed.action).toBe('click');
+                expect(parsed.args).toEqual({ index: 5 });
+            });
+        });
+        describe('browser_type tool', () => {
+            it('should return requiresBrowserExecution marker with all args', async () => {
+                const typeTool = tools.find((t) => t.name === 'browser_type');
+                expect(typeTool).toBeDefined();
+                const result = await typeTool.invoke({
+                    index: 3,
+                    text: 'Hello World',
+                    pressEnter: true,
+                });
+                const parsed = JSON.parse(result);
+                expect(parsed.requiresBrowserExecution).toBe(true);
+                expect(parsed.action).toBe('type');
+                expect(parsed.args.index).toBe(3);
+                expect(parsed.args.text).toBe('Hello World');
+                expect(parsed.args.pressEnter).toBe(true);
+            });
+        });
+        describe('browser_navigate tool', () => {
+            it('should return requiresBrowserExecution marker with URL', async () => {
+                const navigateTool = tools.find((t) => t.name === 'browser_navigate');
+                expect(navigateTool).toBeDefined();
+                const result = await navigateTool.invoke({
+                    url: 'https://example.com',
+                });
+                const parsed = JSON.parse(result);
+                expect(parsed.requiresBrowserExecution).toBe(true);
+                expect(parsed.action).toBe('navigate');
+                expect(parsed.args.url).toBe('https://example.com');
+            });
+        });
+        describe('browser_scroll tool', () => {
+            it('should return requiresBrowserExecution marker with direction', async () => {
+                const scrollTool = tools.find((t) => t.name === 'browser_scroll');
+                expect(scrollTool).toBeDefined();
+                const result = await scrollTool.invoke({
+                    direction: 'down',
+                    amount: 500,
+                });
+                const parsed = JSON.parse(result);
+                expect(parsed.requiresBrowserExecution).toBe(true);
+                expect(parsed.action).toBe('scroll');
+                expect(parsed.args.direction).toBe('down');
+                expect(parsed.args.amount).toBe(500);
+            });
+        });
+        describe('browser_hover tool', () => {
+            it('should return requiresBrowserExecution marker with index', async () => {
+                const hoverTool = tools.find((t) => t.name === 'browser_hover');
+                expect(hoverTool).toBeDefined();
+                const result = await hoverTool.invoke({ index: 2 });
+                const parsed = JSON.parse(result);
+                expect(parsed.requiresBrowserExecution).toBe(true);
+                expect(parsed.action).toBe('hover');
+                expect(parsed.args.index).toBe(2);
+            });
+        });
+        describe('browser_wait tool', () => {
+            it('should return requiresBrowserExecution marker with duration', async () => {
+                const waitTool = tools.find((t) => t.name === 'browser_wait');
+                expect(waitTool).toBeDefined();
+                const result = await waitTool.invoke({ duration: 2000 });
+                const parsed = JSON.parse(result);
+                expect(parsed.requiresBrowserExecution).toBe(true);
+                expect(parsed.action).toBe('wait');
+                expect(parsed.args.duration).toBe(2000);
+            });
+        });
+        describe('browser_back tool', () => {
+            it('should return requiresBrowserExecution marker', async () => {
+                const backTool = tools.find((t) => t.name === 'browser_back');
+                expect(backTool).toBeDefined();
+                const result = await backTool.invoke({});
+                const parsed = JSON.parse(result);
+                expect(parsed.requiresBrowserExecution).toBe(true);
+                expect(parsed.action).toBe('back');
+            });
+        });
+        describe('browser_screenshot tool', () => {
+            it('should return requiresBrowserExecution marker', async () => {
+                const screenshotTool = tools.find((t) => t.name === 'browser_screenshot');
+                expect(screenshotTool).toBeDefined();
+                const result = await screenshotTool.invoke({});
+                const parsed = JSON.parse(result);
+                expect(parsed.requiresBrowserExecution).toBe(true);
+                expect(parsed.action).toBe('screenshot');
+            });
+        });
+        describe('browser_get_page_state tool', () => {
+            it('should return requiresBrowserExecution marker', async () => {
+                const pageStateTool = tools.find((t) => t.name === 'browser_get_page_state');
+                expect(pageStateTool).toBeDefined();
+                const result = await pageStateTool.invoke({});
+                const parsed = JSON.parse(result);
+                expect(parsed.requiresBrowserExecution).toBe(true);
+                expect(parsed.action).toBe('get_page_state');
+            });
+        });
+        describe('browser_extract tool', () => {
+            it('should return requiresBrowserExecution marker', async () => {
+                const extractTool = tools.find((t) => t.name === 'browser_extract');
+                expect(extractTool).toBeDefined();
+                const result = await extractTool.invoke({ query: 'main content' });
+                const parsed = JSON.parse(result);
+                expect(parsed.requiresBrowserExecution).toBe(true);
+                expect(parsed.action).toBe('extract');
+                expect(parsed.args.query).toBe('main content');
+            });
+        });
+        describe('with waitForResult callback (formatResultForLLM)', () => {
+            it('should format successful result with full page state', async () => {
+                const mockCallback = async () => ({
+                    success: true,
+                    url: 'https://example.com/page',
+                    title: 'Example Page',
+                    elementList: '[0] <button> Click me\n[1] <input> fieldLabel="Search"',
+                });
+                const toolsWithCallback = createBrowserTools({
+                    waitForResult: mockCallback,
+                });
+                const extractTool = toolsWithCallback.find((t) => t.name === 'browser_extract');
+                const result = await extractTool.invoke({});
+                expect(result).toContain('**Current URL:** https://example.com/page');
+                expect(result).toContain('**Page Title:** Example Page');
+                expect(result).toContain('**Interactive Elements**');
+                expect(result).toContain('[0] <button> Click me');
+            });
+            it('should include page state even when success is false', async () => {
+                // Critical: extension often sends success=false with valid page state
+                const mockCallback = async () => ({
+                    success: false,
+                    error: 'Element not found',
+                    url: 'https://example.com/form',
+                    title: 'Login Form',
+                    elementList: '[0] <input> fieldLabel="Username"\n[1] <input> fieldLabel="Password"',
+                });
+                const toolsWithCallback = createBrowserTools({
+                    waitForResult: mockCallback,
+                });
+                const clickTool = toolsWithCallback.find((t) => t.name === 'browser_click');
+                const result = await clickTool.invoke({ index: 5 });
+                // Should include the error note
+                expect(result).toContain('**Note:**');
+                expect(result).toContain('Element not found');
+                // But also include page state — this is the critical fix
+                expect(result).toContain('**Current URL:** https://example.com/form');
+                expect(result).toContain('**Page Title:** Login Form');
+                expect(result).toContain('**Interactive Elements**');
+                expect(result).toContain('fieldLabel="Username"');
+            });
+            it('should return failure message when success=false and no page state', async () => {
+                const mockCallback = async () => ({
+                    success: false,
+                    error: 'Timeout waiting for page',
+                });
+                const toolsWithCallback = createBrowserTools({
+                    waitForResult: mockCallback,
+                });
+                const navTool = toolsWithCallback.find((t) => t.name === 'browser_navigate');
+                const result = await navTool.invoke({ url: 'https://example.com' });
+                expect(result).toContain('**Note:**');
+                expect(result).toContain('Timeout waiting for page');
+            });
+            it('should return success message when success=true and no other data', async () => {
+                const mockCallback = async () => ({
+                    success: true,
+                });
+                const toolsWithCallback = createBrowserTools({
+                    waitForResult: mockCallback,
+                });
+                const waitTool = toolsWithCallback.find((t) => t.name === 'browser_wait');
+                const result = await waitTool.invoke({});
+                expect(result).toContain('completed successfully');
+            });
+            it('should handle screenshot result', async () => {
+                const mockCallback = async () => ({
+                    success: true,
+                    url: 'https://example.com',
+                    screenshot: 'data:image/png;base64,iVBOR...',
+                });
+                const toolsWithCallback = createBrowserTools({
+                    waitForResult: mockCallback,
+                });
+                const ssTool = toolsWithCallback.find((t) => t.name === 'browser_screenshot');
+                const result = await ssTool.invoke({});
+                expect(result).toContain('[Screenshot captured and displayed to user]');
+                expect(result).toContain('**Current URL:** https://example.com');
+            });
+            it('should handle callback error gracefully', async () => {
+                const mockCallback = async () => {
+                    throw new Error('Extension disconnected');
+                };
+                const toolsWithCallback = createBrowserTools({
+                    waitForResult: mockCallback,
+                });
+                const clickTool = toolsWithCallback.find((t) => t.name === 'browser_click');
+                const result = await clickTool.invoke({ index: 1 });
+                expect(result).toContain('failed');
+                expect(result).toContain('Extension disconnected');
+            });
+        });
+    });
+});
+//# sourceMappingURL=BrowserTools.test.js.map

package/src/tools/__tests__/BrowserTools.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"BrowserTools.test.js","sourceRoot":"","sources":["BrowserTools.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAiB,MAAM,eAAe,CAAC;AAChF,OAAO,EACL,kBAAkB,EAClB,oBAAoB,EACpB,aAAa,GAEd,MAAM,iBAAiB,CAAC;AAEzB,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YACrD,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YAClD,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAChD,MAAM,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;YACxD,MAAM,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YACpD,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;YACtD,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YAClD,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAChD,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAChD,MAAM,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;YAC5D,MAAM,CAAC,aAAa,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;YACpE,MAAM,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;YACxD,MAAM,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;YAC9C,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,MAAM,CAAC,oBAAoB,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC3C,MAAM,CAAC,oBAAoB,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,MAAM,CAAC,oBAAoB,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC7C,MAAM,CAAC,oBAAoB,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mEAAmE,EAAE,GAAG,EAAE;YAC3E,MAAM,GAAG,GAAG;gBACV,OAAO,EAAE;oBACP,4BAA4B,EAAE,MAAM;iBACrC;aACF,CAAC;YACF,MAAM,CAAC,oBAAoB,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;YACzE,MAAM,GAAG,GAAG;gBACV,OAAO,EAAE;oBACP,0BAA0B,EAAE,MAAM;iBACnC;aACF,CAAC;YACF,MAAM,CAAC,oBAAoB,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;YAC/D,MAAM,GAAG,GAAG;gBACV,OAAO,EAAE;oBACP,4BAA4B,EAAE,OAAO;oBACrC,0BAA0B,EAAE,OAAO;iBACpC;aACF,CAAC;YACF,MAAM,CAAC,oBAAoB,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;YAC9D,MAAM,GAAG,GAAG;gBACV,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;iBACnC;aACF,CAAC;YACF,MAAM,CAAC,oBAAoB,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,IAAI,KAA4C,CAAC;QAEjD,UAAU,CAAC,GAAG,EAAE;YACd,KAAK,GAAG,kBAAkB,EAAE,CAAC;QAC/B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;YAChD,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;YAChD,MAAM,SAAS,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YAC3C,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;YAC7C,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;YAC5C,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC;YAChD,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;YAC9C,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC;YAC/C,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;YAC7C,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;YAC5C,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;YAC5C,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC;YAClD,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,wBAAwB,CAAC,CAAC;QACxD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;YAChD,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,WAAW,EAAE,CAAC;gBACvC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YACrD,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;YAClC,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;gBAC7D,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,eAAe,CAAC,CAAC;gBAChE,MAAM,CAAC,SAAS,CAAC,CAAC,WAAW,EAAE,CAAC;gBAEhC,MAAM,MAAM,GAAG,MAAM,SAAU,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;gBACrD,MAAM,MAAM,GAAwB,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;gBAEvD,MAAM,CAAC,MAAM,CAAC,wBAAwB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACnD,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBACpC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;YAC5C,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;YACjC,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;gBAC3E,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,cAAc,CAAC,CAAC;gBAC9D,MAAM,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;gBAE/B,MAAM,MAAM,GAAG,MAAM,QAAS,CAAC,MAAM,CAAC;oBACpC,KAAK,EAAE,CAAC;oBACR,IAAI,EAAE,aAAa;oBACnB,UAAU,EAAE,IAAI;iBACjB,CAAC,CAAC;gBACH,MAAM,MAAM,GAAwB,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;gBAEvD,MAAM,CAAC,MAAM,CAAC,wBAAwB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACnD,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACnC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;gBAClC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;gBAC7C,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC5C,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;YACrC,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;gBACtE,MAAM,YAAY,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,kBAAkB,CAAC,CAAC;gBACtE,MAAM,CAAC,YAAY,CAAC,CAAC,WAAW,EAAE,CAAC;gBAEnC,MAAM,MAAM,GAAG,MAAM,YAAa,CAAC,MAAM,CAAC;oBACxC,GAAG,EAAE,qBAAqB;iBAC3B,CAAC,CAAC;gBACH,MAAM,MAAM,GAAwB,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;gBAEvD,MAAM,CAAC,MAAM,CAAC,wBAAwB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACnD,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;gBACvC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;YACtD,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;YACnC,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;gBAC5E,MAAM,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,gBAAgB,CAAC,CAAC;gBAClE,MAAM,CAAC,UAAU,CAAC,CAAC,WAAW,EAAE,CAAC;gBAEjC,MAAM,MAAM,GAAG,MAAM,UAAW,CAAC,MAAM,CAAC;oBACtC,SAAS,EAAE,MAAM;oBACjB,MAAM,EAAE,GAAG;iBACZ,CAAC,CAAC;gBACH,MAAM,MAAM,GAAwB,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;gBAEvD,MAAM,CAAC,MAAM,CAAC,wBAAwB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACnD,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;gBACrC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBAC3C,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACvC,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;YAClC,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;gBACxE,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,eAAe,CAAC,CAAC;gBAChE,MAAM,CAAC,SAAS,CAAC,CAAC,WAAW,EAAE,CAAC;gBAEhC,MAAM,MAAM,GAAG,MAAM,SAAU,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;gBACrD,MAAM,MAAM,GAAwB,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;gBAEvD,MAAM,CAAC,MAAM,CAAC,wBAAwB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACnD,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBACpC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACpC,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;YACjC,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;gBAC3E,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,cAAc,CAAC,CAAC;gBAC9D,MAAM,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;gBAE/B,MAAM,MAAM,GAAG,MAAM,QAAS,CAAC,MAAM,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;gBAC1D,MAAM,MAAM,GAAwB,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;gBAEvD,MAAM,CAAC,MAAM,CAAC,wBAAwB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACnD,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACnC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC1C,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;YACjC,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;gBAC7D,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,cAAc,CAAC,CAAC;gBAC9D,MAAM,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;gBAE/B,MAAM,MAAM,GAAG,MAAM,QAAS,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;gBAC1C,MAAM,MAAM,GAAwB,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;gBAEvD,MAAM,CAAC,MAAM,CAAC,wBAAwB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACnD,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACrC,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;YACvC,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;gBAC7D,MAAM,cAAc,GAAG,KAAK,CAAC,IAAI,CAC/B,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,oBAAoB,CACvC,CAAC;gBACF,MAAM,CAAC,cAAc,CAAC,CAAC,WAAW,EAAE,CAAC;gBAErC,MAAM,MAAM,GAAG,MAAM,cAAe,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;gBAChD,MAAM,MAAM,GAAwB,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;gBAEvD,MAAM,CAAC,MAAM,CAAC,wBAAwB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACnD,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YAC3C,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,6BAA6B,EAAE,GAAG,EAAE;YAC3C,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;gBAC7D,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAC9B,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,wBAAwB,CAC3C,CAAC;gBACF,MAAM,CAAC,aAAa,CAAC,CAAC,WAAW,EAAE,CAAC;gBAEpC,MAAM,MAAM,GAAG,MAAM,aAAc,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;gBAC/C,MAAM,MAAM,GAAwB,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;gBAEvD,MAAM,CAAC,MAAM,CAAC,wBAAwB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACnD,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YAC/C,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;YACpC,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;gBAC7D,MAAM,WAAW,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,iBAAiB,CAAC,CAAC;gBACpE,MAAM,CAAC,WAAW,CAAC,CAAC,WAAW,EAAE,CAAC;gBAElC,MAAM,MAAM,GAAG,MAAM,WAAY,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,cAAc,EAAE,CAAC,CAAC;gBACpE,MAAM,MAAM,GAAwB,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;gBAEvD,MAAM,CAAC,MAAM,CAAC,wBAAwB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACnD,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBACtC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YACjD,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAChE,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;gBACpE,MAAM,YAAY,GAAG,KAAK,IAAI,EAAE,CAAC,CAAC;oBAChC,OAAO,EAAE,IAAI;oBACb,GAAG,EAAE,0BAA0B;oBAC/B,KAAK,EAAE,cAAc;oBACrB,WAAW,EAAE,wDAAwD;iBACtE,CAAC,CAAC;gBAEH,MAAM,iBAAiB,GAAG,kBAAkB,CAAC;oBAC3C,aAAa,EAAE,YAAY;iBAC5B,CAAC,CAAC;gBACH,MAAM,WAAW,GAAG,iBAAiB,CAAC,IAAI,CACxC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,iBAAiB,CACpC,CAAC;gBACF,MAAM,MAAM,GAAG,MAAM,WAAY,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;gBAE7C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,2CAA2C,CAAC,CAAC;gBACtE,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,8BAA8B,CAAC,CAAC;gBACzD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,0BAA0B,CAAC,CAAC;gBACrD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,uBAAuB,CAAC,CAAC;YACpD,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;gBACpE,sEAAsE;gBACtE,MAAM,YAAY,GAAG,KAAK,IAAI,EAAE,CAAC,CAAC;oBAChC,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,mBAAmB;oBAC1B,GAAG,EAAE,0BAA0B;oBAC/B,KAAK,EAAE,YAAY;oBACnB,WAAW,EAAE,sEAAsE;iBACpF,CAAC,CAAC;gBAEH,MAAM,iBAAiB,GAAG,kBAAkB,CAAC;oBAC3C,aAAa,EAAE,YAAY;iBAC5B,CAAC,CAAC;gBACH,MAAM,SAAS,GAAG,iBAAiB,CAAC,IAAI,CACtC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,eAAe,CAClC,CAAC;gBACF,MAAM,MAAM,GAAG,MAAM,SAAU,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;gBAErD,gCAAgC;gBAChC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;gBACtC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;gBAC9C,yDAAyD;gBACzD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,2CAA2C,CAAC,CAAC;gBACtE,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,4BAA4B,CAAC,CAAC;gBACvD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,0BAA0B,CAAC,CAAC;gBACrD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,uBAAuB,CAAC,CAAC;YACpD,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,oEAAoE,EAAE,KAAK,IAAI,EAAE;gBAClF,MAAM,YAAY,GAAG,KAAK,IAAI,EAAE,CAAC,CAAC;oBAChC,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,0BAA0B;iBAClC,CAAC,CAAC;gBAEH,MAAM,iBAAiB,GAAG,kBAAkB,CAAC;oBAC3C,aAAa,EAAE,YAAY;iBAC5B,CAAC,CAAC;gBACH,MAAM,OAAO,GAAG,iBAAiB,CAAC,IAAI,CACpC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,kBAAkB,CACrC,CAAC;gBACF,MAAM,MAAM,GAAG,MAAM,OAAQ,CAAC,MAAM,CAAC,EAAE,GAAG,EAAE,qBAAqB,EAAE,CAAC,CAAC;gBAErE,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;gBACtC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,0BAA0B,CAAC,CAAC;YACvD,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,mEAAmE,EAAE,KAAK,IAAI,EAAE;gBACjF,MAAM,YAAY,GAAG,KAAK,IAAI,EAAE,CAAC,CAAC;oBAChC,OAAO,EAAE,IAAI;iBACd,CAAC,CAAC;gBAEH,MAAM,iBAAiB,GAAG,kBAAkB,CAAC;oBAC3C,aAAa,EAAE,YAAY;iBAC5B,CAAC,CAAC;gBACH,MAAM,QAAQ,GAAG,iBAAiB,CAAC,IAAI,CACrC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,cAAc,CACjC,CAAC;gBACF,MAAM,MAAM,GAAG,MAAM,QAAS,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;gBAE1C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,wBAAwB,CAAC,CAAC;YACrD,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;gBAC/C,MAAM,YAAY,GAAG,KAAK,IAAI,EAAE,CAAC,CAAC;oBAChC,OAAO,EAAE,IAAI;oBACb,GAAG,EAAE,qBAAqB;oBAC1B,UAAU,EAAE,gCAAgC;iBAC7C,CAAC,CAAC;gBAEH,MAAM,iBAAiB,GAAG,kBAAkB,CAAC;oBAC3C,aAAa,EAAE,YAAY;iBAC5B,CAAC,CAAC;gBACH,MAAM,MAAM,GAAG,iBAAiB,CAAC,IAAI,CACnC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,oBAAoB,CACvC,CAAC;gBACF,MAAM,MAAM,GAAG,MAAM,MAAO,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;gBAExC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,6CAA6C,CAAC,CAAC;gBACxE,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,sCAAsC,CAAC,CAAC;YACnE,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;gBACvD,MAAM,YAAY,GAAG,KAAK,IAAI,EAAE;oBAC9B,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC;gBAC5C,CAAC,CAAC;gBAEF,MAAM,iBAAiB,GAAG,kBAAkB,CAAC;oBAC3C,aAAa,EAAE,YAAY;iBAC5B,CAAC,CAAC;gBACH,MAAM,SAAS,GAAG,iBAAiB,CAAC,IAAI,CACtC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,eAAe,CAClC,CAAC;gBACF,MAAM,MAAM,GAAG,MAAM,SAAU,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;gBAErD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;gBACnC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,wBAAwB,CAAC,CAAC;YACrD,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/src/tools/__tests__/BrowserTools.test.ts

@@ -39,19 +39,19 @@ describe('BrowserTools', () => {
       expect(hasBrowserCapability({ headers: undefined })).toBe(false);
     });
 
-    it('should return true when X-Ranger-Browser-Extension header is true', () => {
+    it('should return true when X-Illuma-Browser-Extension header is true', () => {
       const req = {
         headers: {
-          'x-ranger-browser-extension': 'true',
+          'x-illuma-browser-extension': 'true',
         },
       };
       expect(hasBrowserCapability(req)).toBe(true);
     });
 
-    it('should return true when X-Ranger-Browser-Capable header is true', () => {
+    it('should return true when X-Illuma-Browser-Capable header is true', () => {
       const req = {
         headers: {
-          'x-ranger-browser-capable': 'true',
+          'x-illuma-browser-capable': 'true',
         },
       };
       expect(hasBrowserCapability(req)).toBe(true);
@@ -60,8 +60,8 @@ describe('BrowserTools', () => {
     it('should return false when browser headers are not true', () => {
       const req = {
         headers: {
-          'x-ranger-browser-extension': 'false',
-          'x-ranger-browser-capable': 'false',
+          'x-illuma-browser-extension': 'false',
+          'x-illuma-browser-capable': 'false',
         },
       };
       expect(hasBrowserCapability(req)).toBe(false);
@@ -261,5 +261,130 @@ describe('BrowserTools', () => {
         expect(parsed.args.query).toBe('main content');
       });
     });
+
+    /* eslint-disable @typescript-eslint/explicit-function-return-type */
+    describe('with waitForResult callback (formatResultForLLM)', () => {
+      it('should format successful result with full page state', async () => {
+        const mockCallback = async () => ({
+          success: true,
+          url: 'https://example.com/page',
+          title: 'Example Page',
+          elementList: '[0] <button> Click me\n[1] <input> fieldLabel="Search"',
+        });
+
+        const toolsWithCallback = createBrowserTools({
+          waitForResult: mockCallback,
+        });
+        const extractTool = toolsWithCallback.find(
+          (t) => t.name === 'browser_extract'
+        );
+        const result = await extractTool!.invoke({});
+
+        expect(result).toContain('**Current URL:** https://example.com/page');
+        expect(result).toContain('**Page Title:** Example Page');
+        expect(result).toContain('**Interactive Elements**');
+        expect(result).toContain('[0] <button> Click me');
+      });
+
+      it('should include page state even when success is false', async () => {
+        // Critical: extension often sends success=false with valid page state
+        const mockCallback = async () => ({
+          success: false,
+          error: 'Element not found',
+          url: 'https://example.com/form',
+          title: 'Login Form',
+          elementList:
+            '[0] <input> fieldLabel="Username"\n[1] <input> fieldLabel="Password"',
+        });
+
+        const toolsWithCallback = createBrowserTools({
+          waitForResult: mockCallback,
+        });
+        const clickTool = toolsWithCallback.find(
+          (t) => t.name === 'browser_click'
+        );
+        const result = await clickTool!.invoke({ index: 5 });
+
+        // Should include the error note
+        expect(result).toContain('**Note:**');
+        expect(result).toContain('Element not found');
+        // But also include page state — this is the critical fix
+        expect(result).toContain('**Current URL:** https://example.com/form');
+        expect(result).toContain('**Page Title:** Login Form');
+        expect(result).toContain('**Interactive Elements**');
+        expect(result).toContain('fieldLabel="Username"');
+      });
+
+      it('should return failure message when success=false and no page state', async () => {
+        const mockCallback = async () => ({
+          success: false,
+          error: 'Timeout waiting for page',
+        });
+
+        const toolsWithCallback = createBrowserTools({
+          waitForResult: mockCallback,
+        });
+        const navTool = toolsWithCallback.find(
+          (t) => t.name === 'browser_navigate'
+        );
+        const result = await navTool!.invoke({ url: 'https://example.com' });
+
+        expect(result).toContain('**Note:**');
+        expect(result).toContain('Timeout waiting for page');
+      });
+
+      it('should return success message when success=true and no other data', async () => {
+        const mockCallback = async () => ({
+          success: true,
+        });
+
+        const toolsWithCallback = createBrowserTools({
+          waitForResult: mockCallback,
+        });
+        const waitTool = toolsWithCallback.find(
+          (t) => t.name === 'browser_wait'
+        );
+        const result = await waitTool!.invoke({});
+
+        expect(result).toContain('completed successfully');
+      });
+
+      it('should handle screenshot result', async () => {
+        const mockCallback = async () => ({
+          success: true,
+          url: 'https://example.com',
+          screenshot: 'data:image/png;base64,iVBOR...',
+        });
+
+        const toolsWithCallback = createBrowserTools({
+          waitForResult: mockCallback,
+        });
+        const ssTool = toolsWithCallback.find(
+          (t) => t.name === 'browser_screenshot'
+        );
+        const result = await ssTool!.invoke({});
+
+        expect(result).toContain('[Screenshot captured and displayed to user]');
+        expect(result).toContain('**Current URL:** https://example.com');
+      });
+
+      it('should handle callback error gracefully', async () => {
+        const mockCallback = async () => {
+          throw new Error('Extension disconnected');
+        };
+
+        const toolsWithCallback = createBrowserTools({
+          waitForResult: mockCallback,
+        });
+        const clickTool = toolsWithCallback.find(
+          (t) => t.name === 'browser_click'
+        );
+        const result = await clickTool!.invoke({ index: 1 });
+
+        expect(result).toContain('failed');
+        expect(result).toContain('Extension disconnected');
+      });
+    });
+    /* eslint-enable @typescript-eslint/explicit-function-return-type */
   });
 });