补充欢迎页和错误请求引导，对小白更友好

LLM-Red-Team · Apr 3, 2024 · 74ef23e · 74ef23e
1 parent 6f22635
commit 74ef23e
Show file tree

Hide file tree

Showing 5 changed files with 30 additions and 3 deletions.
diff --git a/public/welcome.html b/public/welcome.html
@@ -0,0 +1,10 @@
+<!DOCTYPE html>
+<html>
+    <head>
+        <meta charset="utf-8"/>
+        <title>🚀 服务已启动</title>
+    </head>
+    <body>
+        <p>step-free-api已启动！<br>请通过LobeChat / NextChat / Dify等客户端或OpenAI SDK接入！</p>
+    </body>
+</html>
diff --git a/src/api/routes/chat.ts b/src/api/routes/chat.ts
@@ -3,7 +3,6 @@ import _ from 'lodash';
 import Request from '@/lib/request/Request.ts';
 import Response from '@/lib/response/Response.ts';
 import chat from '@/api/controllers/chat.ts';
-import logger from '@/lib/logger.ts';
 
 export default {
 

diff --git a/src/api/routes/index.ts b/src/api/routes/index.ts
@@ -1,7 +1,23 @@
+import fs from 'fs-extra';
+
+import Response from '@/lib/response/Response.ts';
 import chat from "./chat.ts";
 import ping from "./ping.ts";
 
 export default [
+    {
+        get: {
+            '/': async () => {
+                const content = await fs.readFile('public/welcome.html');
+                return new Response(content, {
+                    type: 'html',
+                    headers: {
+                        Expires: '-1'
+                    }
+                });
+            }
+        }
+    },
     chat,
     ping
 ];
diff --git a/src/lib/response/FailureBody.ts b/src/lib/response/FailureBody.ts
@@ -15,7 +15,7 @@ export default class FailureBody extends Body {
         else if(error instanceof APIException || error instanceof Exception)
             ({ errcode, errmsg, data, httpStatusCode } = error);
         else if(_.isError(error))
-            error = new Exception(EX.SYSTEM_ERROR, error.message);
+            ({ errcode, errmsg, data, httpStatusCode } = new Exception(EX.SYSTEM_ERROR, error.message));
         super({
             code: errcode || -1,
             message: errmsg || 'Internal error',

diff --git a/src/lib/server.ts b/src/lib/server.ts
@@ -73,7 +73,9 @@ class Server {
         this.app.use((ctx: any) => {
             const request = new Request(ctx);
             logger.debug(`-> ${ctx.request.method} ${ctx.request.url} request is not supported - ${request.remoteIP || "unknown"}`);
-            const failureBody = new FailureBody(new Exception(EX.SYSTEM_NOT_ROUTE_MATCHING, "Request is not supported"));
+            // const failureBody = new FailureBody(new Exception(EX.SYSTEM_NOT_ROUTE_MATCHING, "Request is not supported"));
+            // const response = new Response(failureBody);
+            const failureBody = new FailureBody(new Error(`[请求有误]: 正确请求为 POST -> /v1/chat/completions ，当前请求为 ${ctx.request.method} -> ${ctx.request.url} 请纠正`));
             const response = new Response(failureBody);
             response.injectTo(ctx);
             if(config.system.requestLog)