support multi search parameters

antergone · Dec 18, 2023 · a6db0ee · a6db0ee
1 parent b8d0c2d
commit a6db0ee
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 3 deletions.
diff --git a/next.config.mjs b/next.config.mjs
@@ -3,7 +3,7 @@ const config = {
   async rewrites() {
     return [
       {
-        source: "/:path*",
+        source: "/:_path*",
         "has": [
           {
             "type": "query",

diff --git a/package.json b/package.json
@@ -9,7 +9,8 @@
   "private": true,
   "scripts": {
     "next": "next",
-    "build": "next build"
+    "build": "next build",
+    "dev": "PORT=4000 next dev"
   },
   "dependencies": {
     "next": "canary",

diff --git a/src/handle-request.ts b/src/handle-request.ts
@@ -33,7 +33,13 @@ export default async function handleRequest(request: NextRequest & { nextUrl?: U
   // -d '{ "prompt": { "text": "Write a story about a magic backpack"} }' \
   // "https://generativelanguage.googleapis.com/v1beta3/models/text-bison-001:generateText?key={YOUR_KEY}"
 
-  const url = new URL(pathname + `?key=${searchParams.get('key')}`, "https://generativelanguage.googleapis.com").href;
+  const url = new URL(pathname, "https://generativelanguage.googleapis.com");
+  searchParams.delete("_path");
+
+  searchParams.forEach((value, key) => {
+    url.searchParams.append(key, value);
+  });
+
   const headers = pickHeaders(request.headers, ["content-type"]);
 
   const response = await fetch(url, {