repo_description: ✨ Light and Fast AI Assistant. Support: Web | iOS | MacOS | Android |  Linux | Windows
repo_topics: calclaude chatgpt claude cross-platform desktop fe gemini gemini-pro gemini-server gemini-ultra gpt-4o groq nextjs ollama react tauri tauri-app vercel webui
code: import { getServerSideConfig } from "@/app/config/server";
  import {
    IFLYTEK_BASE_URL,
    ApiPath,
    ModelProvider,
    ServiceProvider,
  } from "@/app/constant";
  import { prettyObject } from "@/app/utils/format";
  import { NextRequest, NextResponse } from "next/server";
  import { auth } from "@/app/api/auth";
  import { isModelNotavailableInServer } from "@/app/utils/model";
  // iflytek
  
  const serverConfig = getServerSideConfig();
  
  export async function handle(
    req: NextRequest,
    { params }: { params: { path: string[] } },
  ) {
    console.log("[Iflytek Route] params ", params);
  
    if (req.method === "OPTIONS") {
      return NextResponse.json({ body: "OK" }, { status: 200 });
    }
  
    const authResult = auth(req, ModelProvider.Iflytek);
    if (authResult.error) {
      return NextResponse.json(authResult, {
        status: 401,
      });
    }
  
    try {
      const response = await request(req);
      return response;
    } catch (e) {
      console.error("[Iflytek] ", e);
      return NextResponse.json(prettyObject(e));
    }
  }
  
  async function request(req: NextRequest) {
    const controller = new AbortController();
  
    // iflytek use base url or just remove the path
    let path = `${req.nextUrl.pathname}`.replaceAll(ApiPath.Iflytek, "");
  
    let baseUrl = serverConfig.iflytekUrl || IFLYTEK_BASE_URL;
  
    if (!baseUrl.startsWith("http")) {
      baseUrl = `https://${baseUrl}`;
    }
  
    if (baseUrl.endsWith("/")) {
      baseUrl = baseUrl.slice(0, -1);
    }
  
    console.log("[Proxy] ", path);
    console.log("[Base Url]", baseUrl);
  
    const timeoutId = setTimeout(
      () => {
        controller.abort();
      },
      10 * 60 * 1000,
    );
  
    const fetchUrl = `${baseUrl}${path}`;
    const fetchOptions: RequestInit = {
      headers: {
        "Content-Type": "application/json",
        Authorization: req.headers.get("Authorization") ?? "",
      },
      method: req.method,
      body: req.body,
      redirect: "manual",
      // @ts-ignore
      duplex: "half",
      signal: controller.signal,
    };
  
    // try to refuse some request to some models
    if (serverConfig.customModels && req.body) {
      try {
        const clonedBody = await req.text();
        fetchOptions.body = clonedBody;
  
        const jsonBody = JSON.parse(clonedBody) as { model?: string };
  
        // not undefined and is false
        if (
          isModelNotavailableInServer(
            serverConfig.customModels,
            jsonBody?.model as string,
            ServiceProvider.Iflytek as string,
          )
        ) {
          return NextResponse.json(
            {
              error: true,
              message: `you are not allowed to use ${jsonBody?.model} model`,
            },
            {
              status: 403,
            },
          );
        }
      } catch (e) {
        console.error(`[Iflytek] filter`, e);
      }
    }
    try {
      const res = await fetch(fetchUrl, fetchOptions);
  
      // to prevent browser prompt for credentials
      const newHeaders = new Headers(res.headers);
      newHeaders.delete("www-authenticate");
      // to disable nginx buffering
      newHeaders.set("X-Accel-Buffering", "no");
  
      return new Response(res.body, {
        status: res.status,
        statusText: res.statusText,
        headers: newHeaders,
      });
    } finally {
      clearTimeout(timeoutId);
    }
  }
  