Update index.js

index.js

@@ -36,7 +36,9 @@ const CONFIG = {
         "grok-3": "grok-3",
         "grok-3-deepsearch": "grok-3",
         "grok-3-reasoning": "grok-3",
-    }
+    },
+    IS_IMG_GEN: false,
+    IS_THINKING: false
 };
 
 // HTTP 请求头配置
@@ -296,8 +298,8 @@ class ResponseHandler {
 
         const reader = response.body;
         let buffer = '';
-        let isImgGen = false;
-        let isThinking = false;
+        CONFIG.IS_IMG_GEN = false;
+        CONFIG.IS_THINKING = false;
 
         try {
             for await (const chunk of reader) {
@@ -306,7 +308,7 @@ class ResponseHandler {
 
                 for (const line of lines) {
                     if (!line.trim()) continue;
-                    await this.processStreamLine(JSON.parse(line), model, res, isImgGen, isThinking);
+                    await this.processStreamLine(JSON.parse(line), model, res);
                 }
             }
 
@@ -318,19 +320,19 @@ class ResponseHandler {
         }
     }
 
-    static async processStreamLine(jsonData, model, res, isImgGen, isThinking) {
+    static async processStreamLine(jsonData, model, res) {
         if (jsonData.result?.doImgGen) {
-            isImgGen = true;
+            CONFIG.IS_IMG_GEN = true;
             return;
         }
 
-        if (isImgGen && jsonData.result?.event?.imageAttachmentUpdate?.progress === 100) {
+        if (CONFIG.IS_IMG_GEN && jsonData.result?.event?.imageAttachmentUpdate?.progress === 100) {
             await this.handleImageGeneration(jsonData, model, res);
             return;
         }
 
-        if (!isImgGen && jsonData.result?.message) {
-            await this.handleTextMessage(jsonData, model, res, isThinking);
+        if (!CONFIG.IS_IMG_GEN && jsonData.result?.message) {
+            await this.handleTextMessage(jsonData, model, res);
         }
     }
 
@@ -354,17 +356,17 @@ class ResponseHandler {
         res.write(`data: ${JSON.stringify(responseData)}\n\n`);
     }
 
-    static async handleTextMessage(jsonData, model, res, isThinking) {
+    static async handleTextMessage(jsonData, model, res) {
         let message = jsonData.result.message;
 
         switch (model) {
             case "grok-3-reasoning":
-                if (!isThinking && jsonData.result?.isThinking) {
+                if (!CONFIG.IS_THINKING && jsonData.result?.isThinking) {
                     message = "<think>" + message;
-                    isThinking = true;
-                } else if (isThinking && !jsonData.result?.isThinking) {
+                    CONFIG.IS_THINKING = true;
+                } else if (CONFIG.IS_THINKING && !jsonData.result?.isThinking) {
                     message = "</think>" + message;
-                    isThinking = false;
+                    CONFIG.IS_THINKING = false;
                 }
                 break;
             case "grok-3-deepsearch":
@@ -381,7 +383,6 @@ class ResponseHandler {
         let buffer = '';
         let fullResponse = '';
         let imageUrl = null;
-        let isThinking = false;
 
         try {
             for await (const chunk of reader) {
@@ -390,10 +391,10 @@ class ResponseHandler {
 
                 for (const line of lines) {
                     if (!line.trim()) continue;
-                    const result = await this.processNormalLine(JSON.parse(line), model, isThinking);
+                    const result = await this.processNormalLine(JSON.parse(line), model, CONFIG.IS_THINKING);
                     fullResponse += result.text || '';
                     imageUrl = result.imageUrl || imageUrl;
-                    isThinking = result.isThinking;
+                    CONFIG.IS_THINKING = result.isThinking;
                 }
             }
 
@@ -480,7 +481,7 @@ app.use(express.json({ limit: CONFIG.SERVER.BODY_LIMIT }));
 app.use(express.urlencoded({ extended: true, limit: CONFIG.SERVER.BODY_LIMIT }));
 
 // API 路由
-app.get('/hf/v1/models', (req, res) => {
+app.get('/v1/models', (req, res) => {
     res.json({
         object: "list",
         data: Object.keys(CONFIG.MODELS).map(model => ({
@@ -492,7 +493,7 @@ app.get('/hf/v1/models', (req, res) => {
     });
 });
 
-app.post('/hf/v1/chat/completions', async (req, res) => {
+app.post('/v1/chat/completions', async (req, res) => {
     try {
         const authToken = req.headers.authorization?.replace('Bearer ', '');
         if (authToken !== CONFIG.API.API_KEY) {