It works

Author: brunobergher

packages/cloud/src/CloudAPI.ts

@@ -142,25 +142,13 @@ export class CloudAPI {
 	}
 
 	async getCloudAgents(): Promise<CloudAgent[]> {
-		try {
-			this.log("[CloudAPI] Fetching cloud agents")
+		this.log("[CloudAPI] Fetching cloud agents")
 
-			const response = await this.request<{ agents: CloudAgent[] }>("/api/cloud_agents", {
-				method: "GET",
-			})
+		const response = await this.request<{ success: boolean; data: CloudAgent[] }>("/api/cloud-agents", {
+			method: "GET",
+		})
 
-			this.log("[CloudAPI] Cloud agents response:", response)
-			return response.agents || []
-		} catch (error) {
-			this.log("[CloudAPI] Failed to fetch cloud agents, returning mock data:", error)
-
-			// Return mock data when API fails as requested
-			return [
-				{ id: "1", name: "Code Assistant", type: "code" },
-				{ id: "2", name: "Test Generator", type: "test" },
-				{ id: "3", name: "Code Reviewer", type: "review" },
-				{ id: "4", name: "Documentation Writer", type: "docs" },
-			]
-		}
+		this.log("[CloudAPI] Cloud agents response:", response)
+		return response.data || []
 	}
 }

packages/cloud/src/__tests__/CloudAPI.test.ts

@@ -0,0 +1,81 @@
+import { describe, it, expect, vi, beforeEach } from "vitest"
+import { CloudAPI } from "../CloudAPI.js"
+import { AuthenticationError } from "../errors.js"
+import type { AuthService } from "@roo-code/types"
+
+// Mock fetch globally
+global.fetch = vi.fn()
+
+describe("CloudAPI", () => {
+	let mockAuthService: Partial<AuthService>
+	let cloudAPI: CloudAPI
+
+	beforeEach(() => {
+		// Mock only the methods we need for testing
+		mockAuthService = {
+			getSessionToken: vi.fn().mockReturnValue("test-token"),
+		}
+
+		cloudAPI = new CloudAPI(mockAuthService as AuthService)
+		vi.clearAllMocks()
+	})
+
+	describe("getCloudAgents", () => {
+		it("should return cloud agents on success", async () => {
+			const mockAgents = [
+				{ id: "1", name: "Agent 1", type: "code", icon: "code" },
+				{ id: "2", name: "Agent 2", type: "chat", icon: "chat" },
+			]
+
+			// Mock successful response
+			;(global.fetch as ReturnType<typeof vi.fn>).mockResolvedValueOnce({
+				ok: true,
+				json: async () => ({ success: true, data: mockAgents }),
+			})
+
+			const agents = await cloudAPI.getCloudAgents()
+
+			expect(agents).toEqual(mockAgents)
+			expect(global.fetch).toHaveBeenCalledWith(
+				expect.stringContaining("/api/cloud-agents"),
+				expect.objectContaining({
+					method: "GET",
+					headers: expect.objectContaining({
+						Authorization: "Bearer test-token",
+					}),
+				}),
+			)
+		})
+
+		it("should throw AuthenticationError on 401 response", async () => {
+			// Mock 401 response
+			;(global.fetch as ReturnType<typeof vi.fn>).mockResolvedValueOnce({
+				ok: false,
+				status: 401,
+				statusText: "Unauthorized",
+				json: async () => ({ error: "Authentication required" }),
+			})
+
+			await expect(cloudAPI.getCloudAgents()).rejects.toThrow(AuthenticationError)
+		})
+
+		it("should throw AuthenticationError when no session token", async () => {
+			// Mock no session token
+			mockAuthService.getSessionToken = vi.fn().mockReturnValue(null)
+
+			await expect(cloudAPI.getCloudAgents()).rejects.toThrow(AuthenticationError)
+		})
+
+		it("should return empty array when data is missing", async () => {
+			// Mock response with no data
+			;(global.fetch as ReturnType<typeof vi.fn>).mockResolvedValueOnce({
+				ok: true,
+				json: async () => ({ success: true }),
+			})
+
+			const agents = await cloudAPI.getCloudAgents()
+
+			expect(agents).toEqual([])
+		})
+	})
+})

packages/types/src/cloud.ts

@@ -730,7 +730,8 @@ export type LeaveResponse = {
 export interface CloudAgent {
 	id: string
 	name: string
-	type: string // e.g., "code", "review", "test", "docs"
+	type: string // e.g., "PR Reviewer", "Documentation Writer"
+	icon?: string // e.g., "pr-reviewer", "documentation-writer"
 }
 
 /**

src/core/webview/ClineProvider.ts

@@ -1034,12 +1034,11 @@ export class ClineProvider
 				window.__vite_plugin_react_preamble_installed__ = true
 			</script>
 		`
-
 		const csp = [
 			"default-src 'none'",
 			`font-src ${webview.cspSource} data:`,
 			`style-src ${webview.cspSource} 'unsafe-inline' https://* http://${localServerUrl} http://0.0.0.0:${localPort}`,
-			`img-src ${webview.cspSource} https://storage.googleapis.com https://img.clerk.com data:`,
+			`img-src ${webview.cspSource} https://storage.googleapis.com https://img.clerk.com ${getRooCodeApiUrl()} data:`,
 			`media-src ${webview.cspSource}`,
 			`script-src 'unsafe-eval' ${webview.cspSource} https://* https://*.posthog.com http://${localServerUrl} http://0.0.0.0:${localPort} 'nonce-${nonce}'`,
 			`connect-src ${webview.cspSource} https://* https://*.posthog.com ws://${localServerUrl} ws://0.0.0.0:${localPort} http://${localServerUrl} http://0.0.0.0:${localPort}`,