From ee95fad7e5cbe27d54259d59baf380d4a3553ec5 Mon Sep 17 00:00:00 2001
From: liyuyun-lyy <liyuyun.lyy@alibaba-inc.com>
Date: Tue, 16 Sep 2025 21:24:24 +0800
Subject: [PATCH] feat: add support for iflow cli (#10198)
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: 本x <liyuyun.lyy@alibaba-inc.com>
---
 packages/shared/config/constant.ts    |  3 ++-
 src/main/services/CodeToolsService.ts |  4 ++++
 src/renderer/src/pages/code/index.ts  | 12 ++++++++++--
 3 files changed, 16 insertions(+), 3 deletions(-)

diff --git a/packages/shared/config/constant.ts b/packages/shared/config/constant.ts
index 18246cd1e7..3dc2a45a6a 100644
--- a/packages/shared/config/constant.ts
+++ b/packages/shared/config/constant.ts
@@ -216,5 +216,6 @@ export enum codeTools {
   qwenCode = 'qwen-code',
   claudeCode = 'claude-code',
   geminiCli = 'gemini-cli',
-  openaiCodex = 'openai-codex'
+  openaiCodex = 'openai-codex',
+  iFlowCli = 'iflow-cli'
 }
diff --git a/src/main/services/CodeToolsService.ts b/src/main/services/CodeToolsService.ts
index 6e3bb19022..1372bf1f88 100644
--- a/src/main/services/CodeToolsService.ts
+++ b/src/main/services/CodeToolsService.ts
@@ -51,6 +51,8 @@ class CodeToolsService {
         return '@openai/codex'
       case codeTools.qwenCode:
         return '@qwen-code/qwen-code'
+      case codeTools.iFlowCli:
+        return '@iflow-ai/iflow-cli'
       default:
         throw new Error(`Unsupported CLI tool: ${cliTool}`)
     }
@@ -66,6 +68,8 @@ class CodeToolsService {
         return 'codex'
       case codeTools.qwenCode:
         return 'qwen'
+      case codeTools.iFlowCli:
+        return 'iflow'
       default:
         throw new Error(`Unsupported CLI tool: ${cliTool}`)
     }
diff --git a/src/renderer/src/pages/code/index.ts b/src/renderer/src/pages/code/index.ts
index 2c36da3ec6..f286704d39 100644
--- a/src/renderer/src/pages/code/index.ts
+++ b/src/renderer/src/pages/code/index.ts
@@ -19,7 +19,8 @@ export const CLI_TOOLS = [
   { value: codeTools.claudeCode, label: 'Claude Code' },
   { value: codeTools.qwenCode, label: 'Qwen Code' },
   { value: codeTools.geminiCli, label: 'Gemini CLI' },
-  { value: codeTools.openaiCodex, label: 'OpenAI Codex' }
+  { value: codeTools.openaiCodex, label: 'OpenAI Codex' },
+  { value: codeTools.iFlowCli, label: 'iFlow CLI' }
 ]
 
 export const GEMINI_SUPPORTED_PROVIDERS = ['aihubmix', 'dmxapi', 'new-api']
@@ -35,7 +36,8 @@ export const CLI_TOOL_PROVIDER_MAP: Record<string, (providers: Provider[]) => Pr
     providers.filter((p) => p.type === 'gemini' || GEMINI_SUPPORTED_PROVIDERS.includes(p.id)),
   [codeTools.qwenCode]: (providers) => providers.filter((p) => p.type.includes('openai')),
   [codeTools.openaiCodex]: (providers) =>
-    providers.filter((p) => p.id === 'openai' || OPENAI_CODEX_SUPPORTED_PROVIDERS.includes(p.id))
+    providers.filter((p) => p.id === 'openai' || OPENAI_CODEX_SUPPORTED_PROVIDERS.includes(p.id)),
+  [codeTools.iFlowCli]: (providers) => providers.filter((p) => p.type.includes('openai'))
 }
 
 export const getCodeToolsApiBaseUrl = (model: Model, type: EndpointType) => {
@@ -144,6 +146,12 @@ export const generateToolEnvironment = ({
       env.OPENAI_MODEL = model.id
       env.OPENAI_MODEL_PROVIDER = modelProvider.id
       break
+
+    case codeTools.iFlowCli:
+      env.IFLOW_API_KEY = apiKey
+      env.IFLOW_BASE_URL = baseUrl
+      env.IFLOW_MODEL_NAME = model.id
+      break
   }
 
   return env