Add IO Intelligence Provider (#6875)

Co-authored-by: daniel-lxs <[email protected]>

Author: ertan2002

packages/types/src/global-settings.ts

@@ -194,6 +194,7 @@ export const SECRET_STATE_KEYS = [
 	"huggingFaceApiKey",
 	"sambaNovaApiKey",
 	"fireworksApiKey",
+	"ioIntelligenceApiKey",
 ] as const satisfies readonly (keyof ProviderSettings)[]
 export type SecretState = Pick<ProviderSettings, (typeof SECRET_STATE_KEYS)[number]>
 

packages/types/src/provider-settings.ts

@@ -43,6 +43,7 @@ export const providerNames = [
 	"sambanova",
 	"zai",
 	"fireworks",
+	"io-intelligence",
 ] as const
 
 export const providerNamesSchema = z.enum(providerNames)
@@ -276,6 +277,11 @@ const fireworksSchema = apiModelIdProviderModelSchema.extend({
 	fireworksApiKey: z.string().optional(),
 })
 
+const ioIntelligenceSchema = apiModelIdProviderModelSchema.extend({
+	ioIntelligenceModelId: z.string().optional(),
+	ioIntelligenceApiKey: z.string().optional(),
+})
+
 const defaultSchema = z.object({
 	apiProvider: z.undefined(),
 })
@@ -311,6 +317,7 @@ export const providerSettingsSchemaDiscriminated = z.discriminatedUnion("apiProv
 	sambaNovaSchema.merge(z.object({ apiProvider: z.literal("sambanova") })),
 	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") })),
 	defaultSchema,
 ])
 
@@ -346,6 +353,7 @@ export const providerSettingsSchema = z.object({
 	...sambaNovaSchema.shape,
 	...zaiSchema.shape,
 	...fireworksSchema.shape,
+	...ioIntelligenceSchema.shape,
 	...codebaseIndexProviderSchema.shape,
 })
 
@@ -371,6 +379,7 @@ export const MODEL_ID_KEYS: Partial<keyof ProviderSettings>[] = [
 	"requestyModelId",
 	"litellmModelId",
 	"huggingFaceModelId",
+	"ioIntelligenceModelId",
 ]
 
 export const getModelId = (settings: ProviderSettings): string | undefined => {

packages/types/src/providers/index.ts

@@ -8,6 +8,7 @@ export * from "./gemini.js"
 export * from "./glama.js"
 export * from "./groq.js"
 export * from "./huggingface.js"
+export * from "./io-intelligence.js"
 export * from "./lite-llm.js"
 export * from "./lm-studio.js"
 export * from "./mistral.js"

packages/types/src/providers/io-intelligence.ts

@@ -0,0 +1,44 @@
+import type { ModelInfo } from "../model.js"
+
+export type IOIntelligenceModelId =
+	| "deepseek-ai/DeepSeek-R1-0528"
+	| "meta-llama/Llama-4-Maverick-17B-128E-Instruct-FP8"
+	| "Intel/Qwen3-Coder-480B-A35B-Instruct-int4-mixed-ar"
+	| "openai/gpt-oss-120b"
+
+export const ioIntelligenceDefaultModelId: IOIntelligenceModelId = "meta-llama/Llama-4-Maverick-17B-128E-Instruct-FP8"
+
+export const ioIntelligenceDefaultBaseUrl = "https://api.intelligence.io.solutions/api/v1"
+
+export const IO_INTELLIGENCE_CACHE_DURATION = 1000 * 60 * 60 // 1 hour
+
+export const ioIntelligenceModels = {
+	"deepseek-ai/DeepSeek-R1-0528": {
+		maxTokens: 8192,
+		contextWindow: 128000,
+		supportsImages: false,
+		supportsPromptCache: false,
+		description: "DeepSeek R1 reasoning model",
+	},
+	"meta-llama/Llama-4-Maverick-17B-128E-Instruct-FP8": {
+		maxTokens: 8192,
+		contextWindow: 430000,
+		supportsImages: true,
+		supportsPromptCache: false,
+		description: "Llama 4 Maverick 17B model",
+	},
+	"Intel/Qwen3-Coder-480B-A35B-Instruct-int4-mixed-ar": {
+		maxTokens: 8192,
+		contextWindow: 106000,
+		supportsImages: false,
+		supportsPromptCache: false,
+		description: "Qwen3 Coder 480B specialized for coding",
+	},
+	"openai/gpt-oss-120b": {
+		maxTokens: 8192,
+		contextWindow: 131072,
+		supportsImages: false,
+		supportsPromptCache: false,
+		description: "OpenAI GPT-OSS 120B model",
+	},
+} as const satisfies Record<string, ModelInfo>

src/api/index.ts

@@ -32,6 +32,7 @@ import {
 	LiteLLMHandler,
 	ClaudeCodeHandler,
 	SambaNovaHandler,
+	IOIntelligenceHandler,
 	DoubaoHandler,
 	ZAiHandler,
 	FireworksHandler,
@@ -137,6 +138,8 @@ export function buildApiHandler(configuration: ProviderSettings): ApiHandler {
 			return new ZAiHandler(options)
 		case "fireworks":
 			return new FireworksHandler(options)
+		case "io-intelligence":
+			return new IOIntelligenceHandler(options)
 		default:
 			apiProvider satisfies "gemini-cli" | undefined
 			return new AnthropicHandler(options)