AXON-1596: Agent mode #1532

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft

jwang19-atlassian wants to merge 36 commits into main from axon-1596-agent-mode

package-lock.json

Large diffs are not rendered by default.

package.json

-Original file line number
+Diff line change
@@ Expand Up / @@ -1702,6 +1702,7 @@ @@
             "@atlaskit/pragmatic-drag-and-drop-hitbox": "^1.1.0",
             "@atlaskit/pragmatic-drag-and-drop-live-region": "^1.3.1",
             "@atlaskit/pragmatic-drag-and-drop-react-drop-indicator": "^3.2.5",
+            "@atlaskit/primitives": "^17.1.1",
             "@atlaskit/radio": "^8.3.0",
             "@atlaskit/renderer": "^124.1.4",
             "@atlaskit/section-message": "^8.5.1",
@@ Expand Down @@

src/rovo-dev/client/rovoDevApiClient.test.ts

-Original file line number
+Diff line change
@@ Expand Up / @@ -1618,4 +1618,254 @@ describe('RovoDevApiClient', () => { @@
                 );
             });
         });
+        describe('getAgentMode method', () => {
+            it('should return current agent mode successfully', async () => {
+                const mockAgentModeResponse = {
+                    mode: 'ask' as const,
+                    message: 'Agent mode is set to ask',
+                };
+                const mockResponse = {
+                    status: 200,
+                    json: jest.fn().mockResolvedValue(mockAgentModeResponse),
+                    headers: mockStandardResponseHeaders(),
+                } as unknown as Response;
+                mockFetch.mockResolvedValue(mockResponse);
+                const result = await client.getAgentMode();
+                expect(mockFetch).toHaveBeenCalledWith('http://localhost:8080/v3/agent-mode', {
+                    method: 'GET',
+                    headers: {
+                        accept: 'text/event-stream',
+                        'Content-Type': 'application/json',
+                        Authorization: 'Bearer sessionToken',
+                    },
+                    body: undefined,
+                });
+                expect(result).toEqual(mockAgentModeResponse);
+                expect(result.mode).toBe('ask');
+            });
+            it('should return default mode', async () => {
+                const mockAgentModeResponse = {
+                    mode: 'default' as const,
+                    message: 'Agent mode is set to default',
+                };
+                const mockResponse = {
+                    status: 200,
+                    json: jest.fn().mockResolvedValue(mockAgentModeResponse),
+                    headers: mockStandardResponseHeaders(),
+                } as unknown as Response;
+                mockFetch.mockResolvedValue(mockResponse);
+                const result = await client.getAgentMode();
+                expect(result.mode).toBe('default');
+                expect(result.message).toBe('Agent mode is set to default');
+            });
+            it('should return plan mode', async () => {
+                const mockAgentModeResponse = {
+                    mode: 'plan' as const,
+                    message: 'Agent mode is set to plan',
+                };
+                const mockResponse = {
+                    status: 200,
+                    json: jest.fn().mockResolvedValue(mockAgentModeResponse),
+                    headers: mockStandardResponseHeaders(),
+                } as unknown as Response;
+                mockFetch.mockResolvedValue(mockResponse);
+                const result = await client.getAgentMode();
+                expect(result.mode).toBe('plan');
+            });
+            it('should throw error when API call fails', async () => {
+                const mockResponse = {
+                    status: 500,
+                    statusText: 'Internal Server Error',
+                    headers: mockStandardResponseHeaders(),
+                } as Response;
+                mockFetch.mockResolvedValue(mockResponse);
+                await expect(client.getAgentMode()).rejects.toThrow("Failed to fetch '/v3/agent-mode API: HTTP 500");
+            });
+            it('should throw error when API returns 404', async () => {
+                const mockResponse = {
+                    status: 404,
+                    statusText: 'Not Found',
+                    headers: mockStandardResponseHeaders(),
+                } as Response;
+                mockFetch.mockResolvedValue(mockResponse);
+                await expect(client.getAgentMode()).rejects.toThrow("Failed to fetch '/v3/agent-mode API: HTTP 404");
+            });
+        });
+        describe('setAgentMode method', () => {
+            it('should set agent mode to ask successfully', async () => {
+                const mockSetAgentModeResponse = {
+                    mode: 'ask' as const,
+                    message: 'Agent mode set to ask',
+                };
+                const mockResponse = {
+                    status: 200,
+                    json: jest.fn().mockResolvedValue(mockSetAgentModeResponse),
+                    headers: mockStandardResponseHeaders(),
+                } as unknown as Response;
+                mockFetch.mockResolvedValue(mockResponse);
+                const result = await client.setAgentMode('ask');
+                expect(mockFetch).toHaveBeenCalledWith('http://localhost:8080/v3/agent-mode', {
+                    method: 'PUT',
+                    headers: {
+                        accept: 'text/event-stream',
+                        'Content-Type': 'application/json',
+                        Authorization: 'Bearer sessionToken',
+                    },
+                    body: JSON.stringify({ mode: 'ask' }),
+                });
+                expect(result).toEqual(mockSetAgentModeResponse);
+                expect(result.mode).toBe('ask');
+            });
+            it('should set agent mode to default successfully', async () => {
+                const mockSetAgentModeResponse = {
+                    mode: 'default' as const,
+                    message: 'Agent mode set to default',
+                };
+                const mockResponse = {
+                    status: 200,
+                    json: jest.fn().mockResolvedValue(mockSetAgentModeResponse),
+                    headers: mockStandardResponseHeaders(),
+                } as unknown as Response;
+                mockFetch.mockResolvedValue(mockResponse);
+                const result = await client.setAgentMode('default');
+                expect(mockFetch).toHaveBeenCalledWith('http://localhost:8080/v3/agent-mode', {
+                    method: 'PUT',
+                    headers: {
+                        accept: 'text/event-stream',
+                        'Content-Type': 'application/json',
+                        Authorization: 'Bearer sessionToken',
+                    },
+                    body: JSON.stringify({ mode: 'default' }),
+                });
+                expect(result.mode).toBe('default');
+            });
+            it('should set agent mode to plan successfully', async () => {
+                const mockSetAgentModeResponse = {
+                    mode: 'plan' as const,
+                    message: 'Agent mode set to plan',
+                };
+                const mockResponse = {
+                    status: 200,
+                    json: jest.fn().mockResolvedValue(mockSetAgentModeResponse),
+                    headers: mockStandardResponseHeaders(),
+                } as unknown as Response;
+                mockFetch.mockResolvedValue(mockResponse);
+                const result = await client.setAgentMode('plan');
+                expect(mockFetch).toHaveBeenCalledWith('http://localhost:8080/v3/agent-mode', {
+                    method: 'PUT',
+                    headers: {
+                        accept: 'text/event-stream',
+                        'Content-Type': 'application/json',
+                        Authorization: 'Bearer sessionToken',
+                    },
+                    body: JSON.stringify({ mode: 'plan' }),
+                });
+                expect(result.mode).toBe('plan');
+            });
+            it('should throw error when API call fails', async () => {
+                const mockResponse = {
+                    status: 500,
+                    statusText: 'Internal Server Error',
+                    headers: mockStandardResponseHeaders(),
+                } as Response;
+                mockFetch.mockResolvedValue(mockResponse);
+                await expect(client.setAgentMode('ask')).rejects.toThrow("Failed to fetch '/v3/agent-mode API: HTTP 500");
+            });
+            it('should throw error when API returns 400 (invalid mode)', async () => {
+                const mockResponse = {
+                    status: 400,
+                    statusText: 'Bad Request',
+                    headers: mockStandardResponseHeaders(),
+                } as Response;
+                mockFetch.mockResolvedValue(mockResponse);
+                await expect(client.setAgentMode('invalid' as any)).rejects.toThrow(
+                    "Failed to fetch '/v3/agent-mode API: HTTP 400",
+                );
+            });
+        });
+        describe('getAvailableModes method', () => {
+            it('should return list of available modes successfully', async () => {
+                const mockAvailableModesResponse = {
+                    modes: [
+                        { mode: 'ask', description: 'Ask questions without editing code' },
+                        { mode: 'default', description: 'Default agent mode' },
+                        { mode: 'plan', description: 'Generate plans before executing' },
+                    ],
+                };
+                const mockResponse = {
+                    status: 200,
+                    json: jest.fn().mockResolvedValue(mockAvailableModesResponse),
+                    headers: mockStandardResponseHeaders(),
+                } as unknown as Response;
+                mockFetch.mockResolvedValue(mockResponse);
+                const result = await client.getAvailableModes();
+                expect(mockFetch).toHaveBeenCalledWith('http://localhost:8080/v3/available-modes', {
+                    method: 'GET',
+                    headers: {
+                        accept: 'text/event-stream',
+                        'Content-Type': 'application/json',
+                        Authorization: 'Bearer sessionToken',
+                    },
+                    body: undefined,
+                });
+                expect(result).toEqual(mockAvailableModesResponse);
+                expect(result.modes).toHaveLength(3);
+                expect(result.modes[0].mode).toBe('ask');
+                expect(result.modes[0].description).toBe('Ask questions without editing code');
+            });
+            it('should throw error when API call fails', async () => {
+                const mockResponse = {
+                    status: 500,
+                    statusText: 'Internal Server Error',
+                    headers: mockStandardResponseHeaders(),
+                } as Response;
+                mockFetch.mockResolvedValue(mockResponse);
+                await expect(client.getAvailableModes()).rejects.toThrow(
+                    "Failed to fetch '/v3/available-modes API: HTTP 500",
+                );
+            });
+        });
     });

src/rovo-dev/client/rovoDevApiClient.ts

-Original file line number
+Diff line change
@@ -1,8 +1,13 @@
     import { RovoDevLogger } from '../util/rovoDevLogger';
     import {
+        AgentMode,
+        RovoDevAvailableModesResponse,
         RovoDevCancelResponse,
         RovoDevChatRequest,
+        RovoDevGetAgentModeResponse,
         RovoDevHealthcheckResponse,
+        RovoDevSetAgentModeRequest,
+        RovoDevSetAgentModeResponse,
         RovoDevStatusAPIResponse,
         ToolPermissionChoice,
     } from './rovoDevApiClientInterfaces';
@@ Expand Down Expand Up / @@ -68,7 +73,12 @@ export class RovoDevApiClient { @@
         ): Promise<Response>;
         private async fetchApi(
             restApi: string,
-            method: 'GET' | 'DELETE' | 'POST' | 'PATCH',
+            method: 'POST' | 'PATCH' | 'PUT',
+            body?: BodyInit | null,
+        ): Promise<Response>;
+        private async fetchApi(
+            restApi: string,
+            method: 'GET' | 'DELETE' | 'POST' | 'PATCH' | 'PUT',
             body?: BodyInit | null,
             abortSignal?: AbortSignal | null,
         ): Promise<Response> {
@@ Expand Down Expand Up / @@ -289,4 +299,30 @@ export class RovoDevApiClient { @@
             await this.fetchApi('/accept-mcp-terms', 'POST', JSON.stringify(message));
         }
+        /** Invokes the GET `/v3/agent-mode` API.
+         * @returns {Promise<RovoDevGetAgentModeResponse>} An object representing the current agent mode.
+         */
+        public async getAgentMode(): Promise<RovoDevGetAgentModeResponse> {
+            const response = await this.fetchApi('/v3/agent-mode', 'GET');
+            return await response.json();
+        }
+        /** Invokes the PUT `/v3/agent-mode` API.
+         * @param {AgentMode} mode The agent mode to set ('ask', 'default', or 'plan').
+         * @returns {Promise<RovoDevSetAgentModeResponse>} An object representing the API response.
+         */
+        public async setAgentMode(mode: AgentMode): Promise<RovoDevSetAgentModeResponse> {
+            const request: RovoDevSetAgentModeRequest = { mode };
+            const response = await this.fetchApi('/v3/agent-mode', 'PUT', JSON.stringify(request));
+            return await response.json();
+        }
+        /** Invokes the GET `/v3/available-modes` API.
+         * @returns {Promise<RovoDevAvailableModesResponse>} An object representing all available agent modes.
+         */
+        public async getAvailableModes(): Promise<RovoDevAvailableModesResponse> {
+            const response = await this.fetchApi('/v3/available-modes', 'GET');
+            return await response.json();
+        }
     }

src/rovo-dev/client/rovoDevApiClientInterfaces.ts

-Original file line number
+Diff line change
@@ Expand Up / @@ -78,3 +78,28 @@ export interface RovoDevStatusAPIResponse { @@
     }
     export type ToolPermissionChoice = 'allow' | 'deny';
+    export type AgentMode = 'ask' | 'default' | 'plan';
+    export interface RovoDevGetAgentModeResponse {
+        mode: AgentMode;
+        message: string;
+    }
+    export interface RovoDevSetAgentModeRequest {
+        mode: AgentMode;
+    }
+    export interface RovoDevSetAgentModeResponse {
+        mode: AgentMode;
+        message: string;
+    }
+    export interface RovoDevModeInfo {
+        mode: string; // Use string to allow new modes to be added on Rovo Dev side
+        description: string;
+    }
+    export interface RovoDevAvailableModesResponse {
+        modes: RovoDevModeInfo[];
+    }

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

AXON-1596: Agent mode #1532

Uh oh!

Diff view

Diff view

There are no files selected for viewing

Uh oh!

Uh oh!

Uh oh!

AXON-1596: Agent mode #1532

Are you sure you want to change the base?

Uh oh!

AXON-1596: Agent mode #1532

Uh oh!

Uh oh!

Diff view

Diff view

There are no files selected for viewing

Uh oh!

Uh oh!

Uh oh!