Add support for Sonic model

Author: mrubens

packages/types/src/provider-settings.ts

@@ -47,6 +47,7 @@ export const providerNames = [
 	"zai",
 	"fireworks",
 	"io-intelligence",
+	"roo",
 ] as const
 
 export const providerNamesSchema = z.enum(providerNames)
@@ -288,6 +289,10 @@ const ioIntelligenceSchema = apiModelIdProviderModelSchema.extend({
 	ioIntelligenceApiKey: z.string().optional(),
 })
 
+const rooSchema = apiModelIdProviderModelSchema.extend({
+	// No additional fields needed - uses cloud authentication
+})
+
 const defaultSchema = z.object({
 	apiProvider: z.undefined(),
 })
@@ -324,6 +329,7 @@ export const providerSettingsSchemaDiscriminated = z.discriminatedUnion("apiProv
 	zaiSchema.merge(z.object({ apiProvider: z.literal("zai") })),
 	fireworksSchema.merge(z.object({ apiProvider: z.literal("fireworks") })),
 	ioIntelligenceSchema.merge(z.object({ apiProvider: z.literal("io-intelligence") })),
+	rooSchema.merge(z.object({ apiProvider: z.literal("roo") })),
 	defaultSchema,
 ])
 
@@ -360,6 +366,7 @@ export const providerSettingsSchema = z.object({
 	...zaiSchema.shape,
 	...fireworksSchema.shape,
 	...ioIntelligenceSchema.shape,
+	...rooSchema.shape,
 	...codebaseIndexProviderSchema.shape,
 })
 

packages/types/src/providers/index.ts

@@ -25,3 +25,4 @@ export * from "./xai.js"
 export * from "./doubao.js"
 export * from "./zai.js"
 export * from "./fireworks.js"
+export * from "./roo.js"

packages/types/src/providers/roo.ts

@@ -0,0 +1,19 @@
+import type { ModelInfo } from "../model.js"
+
+// Roo provider with single model
+export type RooModelId = "roo/sonic"
+
+export const rooDefaultModelId: RooModelId = "roo/sonic"
+
+export const rooModels = {
+	"roo/sonic": {
+		maxTokens: 8192,
+		contextWindow: 262_144,
+		supportsImages: false,
+		supportsPromptCache: false,
+		inputPrice: 0,
+		outputPrice: 0,
+		description:
+			"Stealth coding model with 262K context window, accessible for free through Roo Code Cloud for a limited time. (Note: prompts and completions are logged by the model creator and used to improve the model.)",
+	},
+} as const satisfies Record<string, ModelInfo>

src/api/index.ts

@@ -35,6 +35,7 @@ import {
 	DoubaoHandler,
 	ZAiHandler,
 	FireworksHandler,
+	RooHandler,
 } from "./providers"
 import { NativeOllamaHandler } from "./providers/native-ollama"
 
@@ -140,6 +141,8 @@ export function buildApiHandler(configuration: ProviderSettings): ApiHandler {
 			return new FireworksHandler(options)
 		case "io-intelligence":
 			return new IOIntelligenceHandler(options)
+		case "roo":
+			return new RooHandler(options)
 		default:
 			apiProvider satisfies "gemini-cli" | undefined
 			return new AnthropicHandler(options)

src/api/providers/index.ts

@@ -29,3 +29,4 @@ export { VsCodeLmHandler } from "./vscode-lm"
 export { XAIHandler } from "./xai"
 export { ZAiHandler } from "./zai"
 export { FireworksHandler } from "./fireworks"
+export { RooHandler } from "./roo"

src/api/providers/roo.ts

@@ -0,0 +1,53 @@
+import { rooDefaultModelId, rooModels, type RooModelId } from "@roo-code/types"
+import { CloudService } from "@roo-code/cloud"
+
+import type { ApiHandlerOptions } from "../../shared/api"
+import { BaseOpenAiCompatibleProvider } from "./base-openai-compatible-provider"
+import { t } from "../../i18n"
+
+export class RooHandler extends BaseOpenAiCompatibleProvider<RooModelId> {
+	constructor(options: ApiHandlerOptions) {
+		// Check if CloudService is available and get the session token
+		if (!CloudService.hasInstance()) {
+			throw new Error(t("common:errors.roo.authenticationRequired"))
+		}
+
+		const sessionToken = CloudService.instance.authService?.getSessionToken()
+
+		if (!sessionToken) {
+			throw new Error(t("common:errors.roo.authenticationRequired"))
+		}
+
+		super({
+			...options,
+			providerName: "Roo Code Cloud",
+			baseURL: "https://api.roocode.com/v1",
+			apiKey: sessionToken,
+			defaultProviderModelId: rooDefaultModelId,
+			providerModels: rooModels,
+			defaultTemperature: 0.7,
+		})
+	}
+
+	override getModel() {
+		const modelId = this.options.apiModelId || rooDefaultModelId
+		const modelInfo = this.providerModels[modelId as RooModelId] ?? this.providerModels[rooDefaultModelId]
+
+		if (modelInfo) {
+			return { id: modelId as RooModelId, info: modelInfo }
+		}
+
+		// Return the requested model ID even if not found, with fallback info
+		return {
+			id: modelId as RooModelId,
+			info: {
+				maxTokens: 8192,
+				contextWindow: 262_144,
+				supportsImages: false,
+				supportsPromptCache: false,
+				inputPrice: 0,
+				outputPrice: 0,
+			},
+		}
+	}
+}

src/i18n/locales/ca/common.json

@@ -100,6 +100,9 @@
 			"genericError": "Cerebras API Error ({{status}}): {{message}}",
 			"noResponseBody": "Cerebras API Error: No response body",
 			"completionError": "Cerebras completion error: {{error}}"
+		},
+		"roo": {
+			"authenticationRequired": "Roo provider requires cloud authentication. Please sign in to Roo Code Cloud."
 		}
 	},
 	"warnings": {