This repository has been archived by the owner on Sep 15, 2024. It is now read-only.
forked from ChatGPTNextWeb/NextChat
-
Notifications
You must be signed in to change notification settings - Fork 8
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request ChatGPTNextWeb#3609 from fredliang44/main
[Feature] support Gemini Pro API
- Loading branch information
1 parent
36a31af
commit 9e25a3e
Showing
23 changed files
with
858 additions
and
114 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,114 @@ | ||
| import { NextRequest, NextResponse } from "next/server"; | ||
| import { auth } from "../../auth"; | ||
| import { getServerSideConfig } from "@/app/config/server"; | ||
| import { GEMINI_BASE_URL, Google } from "@/app/constant"; | ||
|
|
||
| async function handle( | ||
| req: NextRequest, | ||
| { params }: { params: { path: string[] } }, | ||
| ) { | ||
| console.log("[Google Route] params ", params); | ||
|
|
||
| if (req.method === "OPTIONS") { | ||
| return NextResponse.json({ body: "OK" }, { status: 200 }); | ||
| } | ||
|
|
||
| const controller = new AbortController(); | ||
|
|
||
| const serverConfig = getServerSideConfig(); | ||
|
|
||
| let baseUrl = serverConfig.googleUrl || GEMINI_BASE_URL; | ||
|
|
||
| if (!baseUrl.startsWith("http")) { | ||
| baseUrl = `https://${baseUrl}`; | ||
| } | ||
|
|
||
| if (baseUrl.endsWith("/")) { | ||
| baseUrl = baseUrl.slice(0, -1); | ||
| } | ||
|
|
||
| let path = `${req.nextUrl.pathname}`.replaceAll("/api/google/", ""); | ||
|
|
||
| console.log("[Proxy] ", path); | ||
| console.log("[Base Url]", baseUrl); | ||
|
|
||
| const timeoutId = setTimeout( | ||
| () => { | ||
| controller.abort(); | ||
| }, | ||
| 10 * 60 * 1000, | ||
| ); | ||
|
|
||
| const bearToken = req.headers.get("Authorization") ?? ""; | ||
| const token = bearToken.trim().replaceAll("Bearer ", "").trim(); | ||
|
|
||
| const key = token ? token : serverConfig.googleApiKey; | ||
| if (!key) { | ||
| return NextResponse.json( | ||
| { | ||
| error: true, | ||
| message: `missing GOOGLE_API_KEY in server env vars`, | ||
| }, | ||
| { | ||
| status: 401, | ||
| }, | ||
| ); | ||
| } | ||
|
|
||
| const fetchUrl = `${baseUrl}/${path}?key=${key}`; | ||
|
|
||
| const fetchOptions: RequestInit = { | ||
| headers: { | ||
| "Content-Type": "application/json", | ||
| "Cache-Control": "no-store", | ||
| }, | ||
| method: req.method, | ||
| body: req.body, | ||
| // to fix #2485: https://stackoverflow.com/questions/55920957/cloudflare-worker-typeerror-one-time-use-body | ||
| redirect: "manual", | ||
| // @ts-ignore | ||
| duplex: "half", | ||
| signal: controller.signal, | ||
| }; | ||
|
|
||
| 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); | ||
| } | ||
| } | ||
|
|
||
| export const GET = handle; | ||
| export const POST = handle; | ||
|
|
||
| export const runtime = "edge"; | ||
| export const preferredRegion = [ | ||
| "arn1", | ||
| "bom1", | ||
| "cdg1", | ||
| "cle1", | ||
| "cpt1", | ||
| "dub1", | ||
| "fra1", | ||
| "gru1", | ||
| "hnd1", | ||
| "iad1", | ||
| "icn1", | ||
| "kix1", | ||
| "lhr1", | ||
| "pdx1", | ||
| "sfo1", | ||
| "sin1", | ||
| "syd1", | ||
| ]; |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.