增加token存活检测

src/api/controllers/chat.ts

@@ -350,9 +350,32 @@ function generateXssParams() {
   };
 }
 
+/**
+ * 获取Token存活状态
+ */
+async function getTokenLiveStatus(token: string) {
+  const result = await axios.get("https://ai-role.cn/echo-prod/convs?role=echo", {
+    headers: {
+      Authorization: `Bearer ${token}`,
+      ...FAKE_HEADERS,
+      ...generateXssParams()
+    },
+    timeout: 15000,
+    validateStatus: () => true
+  });
+  try {
+    const data = checkResult(result);
+    return _.isArray(data);
+  }
+  catch(err) {
+    return false;
+  }
+}
+
 export default {
   createConversation,
   createCompletion,
   createCompletionStream,
+  getTokenLiveStatus,
   tokenSplit
 };

src/api/routes/token.ts

@@ -0,0 +1,25 @@
+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 {
+
+    prefix: '/token',
+
+    post: {
+
+        '/check': async (request: Request) => {
+            request
+                .validate('body.token', _.isString)
+            const live = await chat.getTokenLiveStatus(request.body.token);
+            return {
+                live
+            }
+        }
+
+    }
+
+}