feat: add EdgeOne Pages as AI provider (#456)

* feat: add edgeone provider * feat: add edgeone provider * feat: add edgeone provider * feat: add edgeone provider * feat: add edgeone provider * feat: add edgeone provider * feat: add edgeone provider * feat: add edgeone provider * feat: add edgeone provider * feat: add edgeone provider * feat: add edgeone provider * feat: add edgeone provider * feat: add edgeone provider * feat: add edgeone provider * feat: add edgeone provider * feat: add edgeone provider * feat: add edgeone provider * feat: add edgeone provider * feat: add edgeone provider * feat: edit diagram * feat: edit diagram * feat: edit diagram * feat: edit diagram * feat: edit diagram * feat: edit diagram * feat: edit diagram * feat: add edgeone provider * feat: add edgeone provider * feat: add edgeone provider * fix: build error * fix: build error * fix: build error * fix: build error * fix: build error * fix: build error * fix: build error * fix: add cookie * fix: add cookie * fix: add cookie * fix: add cookie * fix: add cookie * fix: build error * fix: build error * fix: build error * fix: build error * fix: build error * fix: build error * fix: build error * fix: build error * fix: build error * fix: build error * fix: build error * fix: build error * fix: build error * fix: build error * fix: build error * feat: validate * feat: document link --------- Co-authored-by: zoejiezhou <zoejiezhou@tencent.com>
2026-01-02 14:22:28 +08:00 · 2025-12-30 21:13:22 +08:00
parent ad80e9c6f5
commit ca21a5bb27
13 changed files with 456 additions and 9 deletions
--- a/lib/ai-providers.ts
+++ b/lib/ai-providers.ts
@@ -21,6 +21,7 @@ export type ProviderName =
    | "siliconflow"
    | "sglang"
    | "gateway"
+    | "edgeone"
    | "doubao"

 interface ModelConfig {
@@ -40,6 +41,8 @@ export interface ClientOverrides {
    awsSecretAccessKey?: string | null
    awsRegion?: string | null
    awsSessionToken?: string | null
+    // Custom headers (e.g., for EdgeOne cookie auth)
+    headers?: Record<string, string>
 }

 // Providers that can be used with client-provided API keys
@@ -54,6 +57,7 @@ const ALLOWED_CLIENT_PROVIDERS: ProviderName[] = [
    "siliconflow",
    "sglang",
    "gateway",
+    "edgeone",
    "doubao",
 ]

@@ -375,6 +379,7 @@ const PROVIDER_ENV_VARS: Record<ProviderName, string | null> = {
    siliconflow: "SILICONFLOW_API_KEY",
    sglang: "SGLANG_API_KEY",
    gateway: "AI_GATEWAY_API_KEY",
+    edgeone: null, // No credentials needed - uses EdgeOne Edge AI
    doubao: "DOUBAO_API_KEY",
 }

@@ -463,7 +468,12 @@ export function getAIModel(overrides?: ClientOverrides): ModelConfig {
    // SECURITY: Prevent SSRF attacks (GHSA-9qf7-mprq-9qgm)
    // If a custom baseUrl is provided, an API key MUST also be provided.
    // This prevents attackers from redirecting server API keys to malicious endpoints.
-    if (overrides?.baseUrl && !overrides?.apiKey) {
+    // Exception: EdgeOne provider doesn't require API key (uses Edge AI runtime)
+    if (
+        overrides?.baseUrl &&
+        !overrides?.apiKey &&
+        overrides?.provider !== "edgeone"
+    ) {
        throw new Error(
            `API key is required when using a custom base URL. ` +
                `Please provide your own API key in Settings.`,
@@ -840,6 +850,21 @@ export function getAIModel(overrides?: ClientOverrides): ModelConfig {
            break
        }

+        case "edgeone": {
+            // EdgeOne Pages Edge AI - uses OpenAI-compatible API
+            // AI SDK appends /chat/completions to baseURL
+            // /api/edgeai + /chat/completions = /api/edgeai/chat/completions
+            const baseURL = overrides?.baseUrl || "/api/edgeai"
+            const edgeoneProvider = createOpenAI({
+                apiKey: "edgeone", // Dummy key - EdgeOne doesn't require API key
+                baseURL,
+                // Pass cookies for EdgeOne Pages authentication (eo_token, eo_time)
+                ...(overrides?.headers && { headers: overrides.headers }),
+            })
+            model = edgeoneProvider.chat(modelId)
+            break
+        }
+
        case "doubao": {
            const apiKey = overrides?.apiKey || process.env.DOUBAO_API_KEY
            const baseURL =
@@ -856,7 +881,7 @@ export function getAIModel(overrides?: ClientOverrides): ModelConfig {

        default:
            throw new Error(
-                `Unknown AI provider: ${provider}. Supported providers: bedrock, openai, anthropic, google, azure, ollama, openrouter, deepseek, siliconflow, sglang, gateway, doubao`,
+                `Unknown AI provider: ${provider}. Supported providers: bedrock, openai, anthropic, google, azure, ollama, openrouter, deepseek, siliconflow, sglang, gateway, edgeone, doubao`,
            )
    }