Merge pull request #179 from Yidadaa/prompt

fix: middleware match error
This commit is contained in:
Yifei Zhang 2023-03-30 02:11:11 +08:00 committed by GitHub
commit 9c6f3ebb54
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -3,7 +3,7 @@ import { ACCESS_CODES } from "./app/api/access";
import md5 from "spark-md5"; import md5 from "spark-md5";
export const config = { export const config = {
matcher: ["/api/chat", "/api/chat-stream"], matcher: ["/api/openai", "/api/chat-stream"],
}; };
export function middleware(req: NextRequest) { export function middleware(req: NextRequest) {
@ -32,6 +32,7 @@ export function middleware(req: NextRequest) {
if (!token) { if (!token) {
const apiKey = process.env.OPENAI_API_KEY; const apiKey = process.env.OPENAI_API_KEY;
if (apiKey) { if (apiKey) {
console.log("[Auth] set system token");
req.headers.set("token", apiKey); req.headers.set("token", apiKey);
} else { } else {
return NextResponse.json( return NextResponse.json(
@ -44,6 +45,8 @@ export function middleware(req: NextRequest) {
}, },
); );
} }
} else {
console.log("[Auth] set user token");
} }
return NextResponse.next({ return NextResponse.next({