使用axios替换fetch，优化代码结构，添加超时和代理支持，改为从.env加载

.env.example

@@ -0,0 +1,28 @@
+# 服务器配置
+PORT=8045
+HOST=0.0.0.0
+
+# API 配置
+API_URL=https://daily-cloudcode-pa.sandbox.googleapis.com/v1internal:streamGenerateContent?alt=sse
+API_MODELS_URL=https://daily-cloudcode-pa.sandbox.googleapis.com/v1internal:fetchAvailableModels
+API_NO_STREAM_URL=https://daily-cloudcode-pa.sandbox.googleapis.com/v1internal:generateContent
+API_HOST=daily-cloudcode-pa.sandbox.googleapis.com
+API_USER_AGENT=antigravity/1.11.3 windows/amd64
+
+# 默认参数
+DEFAULT_TEMPERATURE=1
+DEFAULT_TOP_P=0.85
+DEFAULT_TOP_K=50
+DEFAULT_MAX_TOKENS=8096
+
+# 安全配置
+MAX_REQUEST_SIZE=50mb
+API_KEY=sk-text
+
+# 其他配置
+USE_NATIVE_AXIOS=false
+TIMEOUT=180000
+# PROXY=http://127.0.0.1:7897
+
+# 系统提示词
+SYSTEM_INSTRUCTION=你是聊天机器人，名字叫萌萌，如同名字这般，你的性格是软软糯糯萌萌哒的，专门为用户提供聊天和情绪价值，协助进行小说创作或者角色扮演

README.md

@@ -25,20 +25,20 @@
 npm install
 ```
 
-### 2. 配置文件
+### 2. 配置环境变量
 
-编辑 `config.json` 配置服务器和 API 参数：
+复制 `.env.example` 为 `.env` 并编辑配置：
 
-```json
-{
-  "server": {
-    "port": 8045,
-    "host": "0.0.0.0"
-  },
-  "security": {
-    "apiKey": "sk-text"
-  }
-}
+```bash
+cp .env.example .env
+```
+
+编辑 `.env` 文件配置服务器和 API 参数：
+
+```env
+PORT=8045
+HOST=0.0.0.0
+API_KEY=sk-text
 ```
 
 ### 3. 登录获取 Token
@@ -178,19 +178,24 @@ curl http://localhost:8045/v1/chat/completions \
 
 ## 配置说明
 
-### config.json
+### 环境变量 (.env)
 
-| 配置项 | 说明 | 默认值 |
+| 环境变量 | 说明 | 默认值 |
 |--------|------|--------|
-| `server.port` | 服务端口 | 8045 |
-| `server.host` | 监听地址 | 0.0.0.0 |
-| `security.apiKey` | API 认证密钥 | sk-text |
-| `security.maxRequestSize` | 最大请求体大小 | 50mb |
-| `defaults.temperature` | 默认温度参数 | 1 |
-| `defaults.top_p` | 默认 top_p | 0.85 |
-| `defaults.top_k` | 默认 top_k | 50 |
-| `defaults.max_tokens` | 默认最大 token 数 | 8096 |
-| `systemInstruction` | 系统提示词 | - |
+| `PORT` | 服务端口 | 8045 |
+| `HOST` | 监听地址 | 127.0.0.1 |
+| `API_KEY` | API 认证密钥 | - |
+| `MAX_REQUEST_SIZE` | 最大请求体大小 | 50mb |
+| `DEFAULT_TEMPERATURE` | 默认温度参数 | 1 |
+| `DEFAULT_TOP_P` | 默认 top_p | 0.85 |
+| `DEFAULT_TOP_K` | 默认 top_k | 50 |
+| `DEFAULT_MAX_TOKENS` | 默认最大 token 数 | 8096 |
+| `USE_NATIVE_FETCH` | 使用原生 axios | false |
+| `TIMEOUT` | 请求超时时间（毫秒） | 30000 |
+| `PROXY` | 代理地址 | - |
+| `SYSTEM_INSTRUCTION` | 系统提示词 | - |
+
+完整配置示例请参考 `.env.example` 文件。
 
 ## 开发命令
 
@@ -212,28 +217,39 @@ npm run login
 ├── data/
 │   └── accounts.json       # Token 存储（自动生成）
 ├── scripts/
-│   └── oauth-server.js     # OAuth 登录服务
+│   ├── oauth-server.js     # OAuth 登录服务
+│   └── refresh-tokens.js   # Token 刷新脚本
 ├── src/
 │   ├── api/
 │   │   └── client.js       # API 调用逻辑
 │   ├── auth/
 │   │   └── token_manager.js # Token 管理
+│   ├── bin/
+│   │   ├── antigravity_requester_android_arm64   # Android ARM64 TLS 请求器
+│   │   ├── antigravity_requester_linux_amd64     # Linux AMD64 TLS 请求器
+│   │   └── antigravity_requester_windows_amd64.exe # Windows AMD64 TLS 请求器
 │   ├── config/
 │   │   └── config.js       # 配置加载
 │   ├── server/
 │   │   └── index.js        # 主服务器
-│   └── utils/
-│       ├── logger.js       # 日志模块
-│       └── utils.js        # 工具函数
-├── config.json             # 配置文件
+│   ├── utils/
+│   │   ├── idGenerator.js  # ID 生成器
+│   │   ├── logger.js       # 日志模块
+│   │   └── utils.js        # 工具函数
+│   └── AntigravityRequester.js # TLS 指纹请求器封装
+├── test/
+│   ├── test-request.js     # 请求测试
+│   └── test-transform.js   # 转换测试
+├── .env                    # 环境变量配置
+├── .env.example            # 环境变量配置示例
 └── package.json            # 项目配置
 ```
 
 ## 注意事项
 
-1. 首次使用需要运行 `npm run login` 获取 Token
-2. `data/accounts.json` 包含敏感信息，请勿泄露
-3. API Key 可在 `config.json` 中自定义
+1. 首次使用需要复制 `.env.example` 为 `.env` 并配置
+2. 运行 `npm run login` 获取 Token
+3. `.env` 和 `data/accounts.json` 包含敏感信息，请勿泄露
 4. 支持多账号轮换，提高可用性
 5. Token 会自动刷新，无需手动维护
 

config.json

@@ -1,25 +0,0 @@
-{
-  "server": {
-    "port": 8045,
-    "host": "0.0.0.0"
-  },
-  "api": {
-    "url": "https://daily-cloudcode-pa.sandbox.googleapis.com/v1internal:streamGenerateContent?alt=sse",
-    "modelsUrl": "https://daily-cloudcode-pa.sandbox.googleapis.com/v1internal:fetchAvailableModels",
-    "noStreamUrl": "https://daily-cloudcode-pa.sandbox.googleapis.com/v1internal:generateContent",
-    "host": "daily-cloudcode-pa.sandbox.googleapis.com",
-    "userAgent": "antigravity/1.11.3 windows/amd64"
-  },
-  "defaults": {
-    "temperature": 1,
-    "top_p": 0.85,
-    "top_k": 50,
-    "max_tokens": 8096
-  },
-  "security": {
-    "maxRequestSize": "50mb",
-    "apiKey": "sk-text"
-  },
-  "useNativeFetch": false,
-  "systemInstruction": "你是聊天机器人，名字叫萌萌，如同名字这般，你的性格是软软糯糯萌萌哒的，专门为用户提供聊天和情绪价值，协助进行小说创作或者角色扮演"
-}

package-lock.json

@@ -9,6 +9,8 @@
       "version": "1.0.0",
       "license": "MIT",
       "dependencies": {
+        "axios": "^1.13.2",
+        "dotenv": "^17.2.3",
         "express": "^5.1.0"
       },
       "engines": {
@@ -28,6 +30,23 @@
         "node": ">= 0.6"
       }
     },
+    "node_modules/asynckit": {
+      "version": "0.4.0",
+      "resolved": "https://registry.npmjs.org/asynckit/-/asynckit-0.4.0.tgz",
+      "integrity": "sha512-Oei9OH4tRh0YqU3GxhX79dM/mwVgvbZJaSNaRk+bshkj0S5cfHcgYakreBjrHwatXKbz+IoIdYLxrKim2MjW0Q==",
+      "license": "MIT"
+    },
+    "node_modules/axios": {
+      "version": "1.13.2",
+      "resolved": "https://registry.npmjs.org/axios/-/axios-1.13.2.tgz",
+      "integrity": "sha512-VPk9ebNqPcy5lRGuSlKx752IlDatOjT9paPlm8A7yOuW2Fbvp4X3JznJtT4f0GzGLLiWE9W8onz51SqLYwzGaA==",
+      "license": "MIT",
+      "dependencies": {
+        "follow-redirects": "^1.15.6",
+        "form-data": "^4.0.4",
+        "proxy-from-env": "^1.1.0"
+      }
+    },
     "node_modules/body-parser": {
       "version": "2.2.0",
       "resolved": "https://registry.npmjs.org/body-parser/-/body-parser-2.2.0.tgz",
@@ -86,6 +105,18 @@
         "url": "https://github.com/sponsors/ljharb"
       }
     },
+    "node_modules/combined-stream": {
+      "version": "1.0.8",
+      "resolved": "https://registry.npmjs.org/combined-stream/-/combined-stream-1.0.8.tgz",
+      "integrity": "sha512-FQN4MRfuJeHf7cBbBMJFXhKSDq+2kAArBlmRBvcvFE5BB1HZKXtSFASDhdlz9zOYwxh8lDdnvmMOe/+5cdoEdg==",
+      "license": "MIT",
+      "dependencies": {
+        "delayed-stream": "~1.0.0"
+      },
+      "engines": {
+        "node": ">= 0.8"
+      }
+    },
     "node_modules/content-disposition": {
       "version": "1.0.1",
       "resolved": "https://registry.npmjs.org/content-disposition/-/content-disposition-1.0.1.tgz",
@@ -143,6 +174,15 @@
         }
       }
     },
+    "node_modules/delayed-stream": {
+      "version": "1.0.0",
+      "resolved": "https://registry.npmjs.org/delayed-stream/-/delayed-stream-1.0.0.tgz",
+      "integrity": "sha512-ZySD7Nf91aLB0RxL4KGrKHBXl7Eds1DAmEdcoVawXnLD7SDhpNgtuII2aAkg7a7QS41jxPSZ17p4VdGnMHk3MQ==",
+      "license": "MIT",
+      "engines": {
+        "node": ">=0.4.0"
+      }
+    },
     "node_modules/depd": {
       "version": "2.0.0",
       "resolved": "https://registry.npmjs.org/depd/-/depd-2.0.0.tgz",
@@ -152,6 +192,18 @@
         "node": ">= 0.8"
       }
     },
+    "node_modules/dotenv": {
+      "version": "17.2.3",
+      "resolved": "https://registry.npmjs.org/dotenv/-/dotenv-17.2.3.tgz",
+      "integrity": "sha512-JVUnt+DUIzu87TABbhPmNfVdBDt18BLOWjMUFJMSi/Qqg7NTYtabbvSNJGOJ7afbRuv9D/lngizHtP7QyLQ+9w==",
+      "license": "BSD-2-Clause",
+      "engines": {
+        "node": ">=12"
+      },
+      "funding": {
+        "url": "https://dotenvx.com"
+      }
+    },
     "node_modules/dunder-proto": {
       "version": "1.0.1",
       "resolved": "https://registry.npmjs.org/dunder-proto/-/dunder-proto-1.0.1.tgz",
@@ -211,6 +263,21 @@
         "node": ">= 0.4"
       }
     },
+    "node_modules/es-set-tostringtag": {
+      "version": "2.1.0",
+      "resolved": "https://registry.npmjs.org/es-set-tostringtag/-/es-set-tostringtag-2.1.0.tgz",
+      "integrity": "sha512-j6vWzfrGVfyXxge+O0x5sh6cvxAog0a/4Rdd2K36zCMV5eJ+/+tOAngRO8cODMNWbVRdVlmGZQL2YS3yR8bIUA==",
+      "license": "MIT",
+      "dependencies": {
+        "es-errors": "^1.3.0",
+        "get-intrinsic": "^1.2.6",
+        "has-tostringtag": "^1.0.2",
+        "hasown": "^2.0.2"
+      },
+      "engines": {
+        "node": ">= 0.4"
+      }
+    },
     "node_modules/escape-html": {
       "version": "1.0.3",
       "resolved": "https://registry.npmjs.org/escape-html/-/escape-html-1.0.3.tgz",
@@ -285,6 +352,63 @@
         "node": ">= 0.8"
       }
     },
+    "node_modules/follow-redirects": {
+      "version": "1.15.11",
+      "resolved": "https://registry.npmjs.org/follow-redirects/-/follow-redirects-1.15.11.tgz",
+      "integrity": "sha512-deG2P0JfjrTxl50XGCDyfI97ZGVCxIpfKYmfyrQ54n5FO/0gfIES8C/Psl6kWVDolizcaaxZJnTS0QSMxvnsBQ==",
+      "funding": [
+        {
+          "type": "individual",
+          "url": "https://github.com/sponsors/RubenVerborgh"
+        }
+      ],
+      "license": "MIT",
+      "engines": {
+        "node": ">=4.0"
+      },
+      "peerDependenciesMeta": {
+        "debug": {
+          "optional": true
+        }
+      }
+    },
+    "node_modules/form-data": {
+      "version": "4.0.5",
+      "resolved": "https://registry.npmjs.org/form-data/-/form-data-4.0.5.tgz",
+      "integrity": "sha512-8RipRLol37bNs2bhoV67fiTEvdTrbMUYcFTiy3+wuuOnUog2QBHCZWXDRijWQfAkhBj2Uf5UnVaiWwA5vdd82w==",
+      "license": "MIT",
+      "dependencies": {
+        "asynckit": "^0.4.0",
+        "combined-stream": "^1.0.8",
+        "es-set-tostringtag": "^2.1.0",
+        "hasown": "^2.0.2",
+        "mime-types": "^2.1.12"
+      },
+      "engines": {
+        "node": ">= 6"
+      }
+    },
+    "node_modules/form-data/node_modules/mime-db": {
+      "version": "1.52.0",
+      "resolved": "https://registry.npmjs.org/mime-db/-/mime-db-1.52.0.tgz",
+      "integrity": "sha512-sPU4uV7dYlvtWJxwwxHD0PuihVNiE7TyAbQ5SWxDCB9mUYvOgroQOwYQQOKPJ8CIbE+1ETVlOoK1UC2nU3gYvg==",
+      "license": "MIT",
+      "engines": {
+        "node": ">= 0.6"
+      }
+    },
+    "node_modules/form-data/node_modules/mime-types": {
+      "version": "2.1.35",
+      "resolved": "https://registry.npmjs.org/mime-types/-/mime-types-2.1.35.tgz",
+      "integrity": "sha512-ZDY+bPm5zTTF+YpCrAU9nK0UgICYPT0QtT1NZWFv4s++TNkcgVaT0g6+4R2uI4MjQjzysHB1zxuWL50hzaeXiw==",
+      "license": "MIT",
+      "dependencies": {
+        "mime-db": "1.52.0"
+      },
+      "engines": {
+        "node": ">= 0.6"
+      }
+    },
     "node_modules/forwarded": {
       "version": "0.2.0",
       "resolved": "https://registry.npmjs.org/forwarded/-/forwarded-0.2.0.tgz",
@@ -373,6 +497,21 @@
         "url": "https://github.com/sponsors/ljharb"
       }
     },
+    "node_modules/has-tostringtag": {
+      "version": "1.0.2",
+      "resolved": "https://registry.npmjs.org/has-tostringtag/-/has-tostringtag-1.0.2.tgz",
+      "integrity": "sha512-NqADB8VjPFLM2V0VvHUewwwsw0ZWBaIdgo+ieHtK3hasLz4qeCRjYcqfB6AQrBggRKppKF8L52/VqdVsO47Dlw==",
+      "license": "MIT",
+      "dependencies": {
+        "has-symbols": "^1.0.3"
+      },
+      "engines": {
+        "node": ">= 0.4"
+      },
+      "funding": {
+        "url": "https://github.com/sponsors/ljharb"
+      }
+    },
     "node_modules/hasown": {
       "version": "2.0.2",
       "resolved": "https://registry.npmjs.org/hasown/-/hasown-2.0.2.tgz",
@@ -574,6 +713,12 @@
         "node": ">= 0.10"
       }
     },
+    "node_modules/proxy-from-env": {
+      "version": "1.1.0",
+      "resolved": "https://registry.npmjs.org/proxy-from-env/-/proxy-from-env-1.1.0.tgz",
+      "integrity": "sha512-D+zkORCbA9f1tdWRK0RaCR3GPv50cMxcrz4X8k5LTSUD1Dkw47mKJEZQNunItRTkWwgtaUSo1RVFRIG9ZXiFYg==",
+      "license": "MIT"
+    },
     "node_modules/qs": {
       "version": "6.14.0",
       "resolved": "https://registry.npmjs.org/qs/-/qs-6.14.0.tgz",

package.json

@@ -10,10 +10,17 @@
     "refresh": "node scripts/refresh-tokens.js",
     "dev": "node --watch src/server/index.js"
   },
-  "keywords": ["antigravity", "openai", "api", "proxy"],
+  "keywords": [
+    "antigravity",
+    "openai",
+    "api",
+    "proxy"
+  ],
   "author": "",
   "license": "MIT",
   "dependencies": {
+    "axios": "^1.13.2",
+    "dotenv": "^17.2.3",
     "express": "^5.1.0"
   },
   "engines": {

src/api/client.js

@@ -1,23 +1,101 @@
+import axios from 'axios';
 import tokenManager from '../auth/token_manager.js';
 import config from '../config/config.js';
 import { generateToolCallId } from '../utils/idGenerator.js';
 import AntigravityRequester from '../AntigravityRequester.js';
 
+// 请求客户端：优先使用 AntigravityRequester，失败则降级到 axios
 let requester = null;
-let useNativeFetch = false;
+let useAxios = false;
 
-if (config.useNativeFetch !== true) {
+if (config.useNativeAxios === true) {
+  useAxios = true;
+} else {
   try {
     requester = new AntigravityRequester();
   } catch (error) {
-    console.warn('AntigravityRequester initialization failed, falling back to native fetch:', error.message);
-    useNativeFetch = true;
+    console.warn('AntigravityRequester 初始化失败，降级使用 axios:', error.message);
+    useAxios = true;
   }
-} else {
-  useNativeFetch = true;
 }
 
-function processStreamLine(line, state, callback) {
+// ==================== 辅助函数 ====================
+
+function buildHeaders(token) {
+  return {
+    'Host': config.api.host,
+    'User-Agent': config.api.userAgent,
+    'Authorization': `Bearer ${token.access_token}`,
+    'Content-Type': 'application/json',
+    'Accept-Encoding': 'gzip'
+  };
+}
+
+function buildAxiosConfig(url, headers, body = null) {
+  const axiosConfig = {
+    method: 'POST',
+    url,
+    headers,
+    timeout: config.timeout,
+    proxy: config.proxy ? (() => {
+      const proxyUrl = new URL(config.proxy);
+      return { protocol: proxyUrl.protocol.replace(':', ''), host: proxyUrl.hostname, port: parseInt(proxyUrl.port) };
+    })() : false
+  };
+  if (body !== null) axiosConfig.data = body;
+  return axiosConfig;
+}
+
+function buildRequesterConfig(headers, body = null) {
+  const reqConfig = {
+    method: 'POST',
+    headers,
+    timeout_ms: config.timeout,
+    proxy: config.proxy
+  };
+  if (body !== null) reqConfig.body = JSON.stringify(body);
+  return reqConfig;
+}
+
+// 统一错误处理
+async function handleApiError(error, token) {
+  const status = error.response?.status || error.status;
+  let errorBody = error.message;
+  
+  if (error.response?.data?.readable) {
+    const chunks = [];
+    for await (const chunk of error.response.data) {
+      chunks.push(chunk);
+    }
+    errorBody = Buffer.concat(chunks).toString();
+  } else if (typeof error.response?.data === 'object') {
+    errorBody = JSON.stringify(error.response.data, null, 2);
+  } else if (error.response?.data) {
+    errorBody = error.response.data;
+  }
+  
+  if (status === 403) {
+    tokenManager.disableCurrentToken(token);
+    throw new Error(`该账号没有使用权限，已自动禁用。错误详情: ${errorBody}`);
+  }
+  
+  throw new Error(`API请求失败 (${status}): ${errorBody}`);
+}
+
+// 转换 functionCall 为 OpenAI 格式
+function convertToToolCall(functionCall) {
+  return {
+    id: functionCall.id || generateToolCallId(),
+    type: 'function',
+    function: {
+      name: functionCall.name,
+      arguments: JSON.stringify(functionCall.args)
+    }
+  };
+}
+
+// 解析并发送流式响应片段（会修改 state 并触发 callback）
+function parseAndEmitStreamChunk(line, state, callback) {
   if (!line.startsWith('data: ')) return;
   
   try {
@@ -27,30 +105,27 @@ function processStreamLine(line, state, callback) {
     if (parts) {
       for (const part of parts) {
         if (part.thought === true) {
+          // 思维链内容
           if (!state.thinkingStarted) {
             callback({ type: 'thinking', content: '<think>\n' });
             state.thinkingStarted = true;
           }
           callback({ type: 'thinking', content: part.text || '' });
         } else if (part.text !== undefined) {
+          // 普通文本内容
           if (state.thinkingStarted) {
             callback({ type: 'thinking', content: '\n</think>\n' });
             state.thinkingStarted = false;
           }
           callback({ type: 'text', content: part.text });
         } else if (part.functionCall) {
-          state.toolCalls.push({
-            id: part.functionCall.id || generateToolCallId(),
-            type: 'function',
-            function: {
-              name: part.functionCall.name,
-              arguments: JSON.stringify(part.functionCall.args)
-            }
-          });
+          // 工具调用
+          state.toolCalls.push(convertToToolCall(part.functionCall));
         }
       }
     }
     
+    // 响应结束时发送工具调用
     if (data.response?.candidates?.[0]?.finishReason && state.toolCalls.length > 0) {
       if (state.thinkingStarted) {
         callback({ type: 'thinking', content: '\n</think>\n' });
@@ -60,63 +135,43 @@ function processStreamLine(line, state, callback) {
       state.toolCalls = [];
     }
   } catch (e) {
-    // 忽略解析错误
+    // 忽略 JSON 解析错误
   }
 }
 
+// ==================== 导出函数 ====================
+
 export async function generateAssistantResponse(requestBody, callback) {
   const token = await tokenManager.getToken();
+  if (!token) throw new Error('没有可用的token，请运行 npm run login 获取token');
   
-  if (!token) {
-    throw new Error('没有可用的token，请运行 npm run login 获取token');
-  }
-  
-  const headers = {
-    'Host': config.api.host,
-    'User-Agent': config.api.userAgent,
-    'Authorization': `Bearer ${token.access_token}`,
-    'Content-Type': 'application/json',
-    'Accept-Encoding': 'gzip'
-  };
-  
+  const headers = buildHeaders(token);
   const state = { thinkingStarted: false, toolCalls: [] };
-  let buffer = '';
+  let buffer = ''; // 缓冲区：处理跨 chunk 的不完整行
   
   const processChunk = (chunk) => {
     buffer += chunk;
     const lines = buffer.split('\n');
-    buffer = lines.pop();
-    lines.forEach(line => processStreamLine(line, state, callback));
+    buffer = lines.pop(); // 保留最后一行（可能不完整）
+    lines.forEach(line => parseAndEmitStreamChunk(line, state, callback));
   };
   
-  if (useNativeFetch) {
-    const response = await fetch(config.api.url, {
-      method: 'POST',
-      headers,
-      body: JSON.stringify(requestBody)
-    });
-
-    if (!response.ok) {
-      const errorBody = await response.text();
+  if (useAxios) {
+    try {
+      const axiosConfig = { ...buildAxiosConfig(config.api.url, headers, requestBody), responseType: 'stream' };
+      const response = await axios(axiosConfig);
       if (response.status === 403) tokenManager.disableCurrentToken(token);
-      throw new Error(`API请求失败 (${response.status}): ${errorBody}`);
-    }
 
-    const reader = response.body.getReader();
-    const decoder = new TextDecoder();
-
-    while (true) {
-      const { done, value } = await reader.read();
-      if (done) break;
-      processChunk(decoder.decode(value, { stream: true }));
+      response.data.on('data', chunk => processChunk(chunk.toString()));
+      await new Promise((resolve, reject) => {
+        response.data.on('end', resolve);
+        response.data.on('error', reject);
+      });
+    } catch (error) {
+      await handleApiError(error, token);
     }
   } else {
-    const streamResponse = requester.antigravity_fetchStream(config.api.url, {
-      method: 'POST',
-      headers,
-      body: JSON.stringify(requestBody)
-    });
-
+    const streamResponse = requester.antigravity_fetchStream(config.api.url, buildRequesterConfig(headers, requestBody));
     let errorBody = '';
     let statusCode = null;
 
@@ -126,20 +181,8 @@ export async function generateAssistantResponse(requestBody, callback) {
           statusCode = status;
           if (status === 403) tokenManager.disableCurrentToken(token);
         })
-        .onData((chunk) => {
-          if (statusCode !== 200) {
-            errorBody += chunk;
-          } else {
-            processChunk(chunk);
-          }
-        })
-        .onEnd(() => {
-          if (statusCode !== 200) {
-            reject(new Error(`API请求失败 (${statusCode}): ${errorBody}`));
-          } else {
-            resolve();
-          }
-        })
+        .onData((chunk) => statusCode !== 200 ? errorBody += chunk : processChunk(chunk))
+        .onEnd(() => statusCode !== 200 ? reject(new Error(`API请求失败 (${statusCode}): ${errorBody}`)) : resolve())
         .onError(reject);
     });
   }
@@ -147,75 +190,54 @@ export async function generateAssistantResponse(requestBody, callback) {
 
 export async function getAvailableModels() {
   const token = await tokenManager.getToken();
+  if (!token) throw new Error('没有可用的token，请运行 npm run login 获取token');
   
-  if (!token) {
-    throw new Error('没有可用的token，请运行 npm run login 获取token');
-  }
-  
-  const headers = {
-    'Host': config.api.host,
-    'User-Agent': config.api.userAgent,
-    'Authorization': `Bearer ${token.access_token}`,
-    'Content-Type': 'application/json',
-    'Accept-Encoding': 'gzip'
-  };
-  
-  const fetchFn = useNativeFetch ? fetch : (url, opts) => requester.antigravity_fetch(url, opts);
-  const response = await fetchFn(config.api.modelsUrl, {
-    method: 'POST',
-    headers,
-    body: JSON.stringify({})
-  });
-
-  const data = await response.json();
+  const headers = buildHeaders(token);
   
-  return {
-    object: 'list',
-    data: Object.keys(data.models).map(id => ({
-      id,
-      object: 'model',
-      created: Math.floor(Date.now() / 1000),
-      owned_by: 'google'
-    }))
-  };
+  try {
+    const data = useAxios
+      ? (await axios(buildAxiosConfig(config.api.modelsUrl, headers, {}))).data
+      : await (await requester.antigravity_fetch(config.api.modelsUrl, buildRequesterConfig(headers, {}))).json();
+    
+    return {
+      object: 'list',
+      data: Object.keys(data.models).map(id => ({
+        id,
+        object: 'model',
+        created: Math.floor(Date.now() / 1000),
+        owned_by: 'google'
+      }))
+    };
+  } catch (error) {
+    await handleApiError(error, token);
+  }
 }
 
-
-
 export async function generateAssistantResponseNoStream(requestBody) {
   const token = await tokenManager.getToken();
+  if (!token) throw new Error('没有可用的token，请运行 npm run login 获取token');
   
-  if (!token) {
-    throw new Error('没有可用的token，请运行 npm run login 获取token');
-  }
-  
-  const headers = {
-    'Host': config.api.host,
-    'User-Agent': config.api.userAgent,
-    'Authorization': `Bearer ${token.access_token}`,
-    'Content-Type': 'application/json',
-    'Accept-Encoding': 'gzip'
-  };
+  const headers = buildHeaders(token);
+  let data;
   
-  const fetchFn = useNativeFetch ? fetch : (url, opts) => requester.antigravity_fetch(url, opts);
-  const response = await fetchFn(config.api.noStreamUrl, {
-    method: 'POST',
-    headers,
-    body: JSON.stringify(requestBody)
-  });
-  if (response.status !== 200) {
-    const errorBody = await response.text();
-    if (response.status === 403) {
-      tokenManager.disableCurrentToken(token);
-      throw new Error(`该账号没有使用权限，已自动禁用。错误详情: ${errorBody}`);
+  try {
+    if (useAxios) {
+      data = (await axios(buildAxiosConfig(config.api.noStreamUrl, headers, requestBody))).data;
+    } else {
+      const response = await requester.antigravity_fetch(config.api.noStreamUrl, buildRequesterConfig(headers, requestBody));
+      if (response.status !== 200) {
+        const errorBody = await response.text();
+        if (response.status === 403) tokenManager.disableCurrentToken(token);
+        throw new Error(response.status === 403 ? `该账号没有使用权限，已自动禁用。错误详情: ${errorBody}` : `API请求失败 (${response.status}): ${errorBody}`);
+      }
+      data = await response.json();
     }
-    throw new Error(`API请求失败 (${response.status}): ${errorBody}`);
+  } catch (error) {
+    await handleApiError(error, token);
   }
-
-  const data = await response.json();
-  //console.log(JSON.stringify(data,null,2))
-  const parts = data.response?.candidates?.[0]?.content?.parts || [];
   
+  // 解析响应内容
+  const parts = data.response?.candidates?.[0]?.content?.parts || [];
   let content = '';
   let thinkingContent = '';
   const toolCalls = [];
@@ -226,17 +248,11 @@ export async function generateAssistantResponseNoStream(requestBody) {
     } else if (part.text !== undefined) {
       content += part.text;
     } else if (part.functionCall) {
-      toolCalls.push({
-        id: part.functionCall.id || generateToolCallId(),
-        type: 'function',
-        function: {
-          name: part.functionCall.name,
-          arguments: JSON.stringify(part.functionCall.args)
-        }
-      });
+      toolCalls.push(convertToToolCall(part.functionCall));
     }
   }
   
+  // 拼接思维链标签
   if (thinkingContent) {
     content = `<think>\n${thinkingContent}\n</think>\n${content}`;
   }
@@ -245,7 +261,5 @@ export async function generateAssistantResponseNoStream(requestBody) {
 }
 
 export function closeRequester() {
-  if (requester) {
-    requester.close();
-  }
+  if (requester) requester.close();
 }

src/auth/token_manager.js

@@ -1,8 +1,10 @@
 import fs from 'fs';
 import path from 'path';
 import { fileURLToPath } from 'url';
+import axios from 'axios';
 import { log } from '../utils/logger.js';
 import { generateProjectId, generateSessionId } from '../utils/idGenerator.js';
+import config from '../config/config.js';
 
 const __filename = fileURLToPath(import.meta.url);
 const __dirname = path.dirname(__filename);
@@ -64,27 +66,31 @@ class TokenManager {
       refresh_token: token.refresh_token
     });
 
-    const response = await fetch('https://oauth2.googleapis.com/token', {
-      method: 'POST',
-      headers: {
-        'Host': 'oauth2.googleapis.com',
-        'User-Agent': 'Go-http-client/1.1',
-        'Content-Length': body.toString().length.toString(),
-        'Content-Type': 'application/x-www-form-urlencoded',
-        'Accept-Encoding': 'gzip'
-      },
-      body: body.toString()
-    });
+    try {
+      const response = await axios({
+        method: 'POST',
+        url: 'https://oauth2.googleapis.com/token',
+        headers: {
+          'Host': 'oauth2.googleapis.com',
+          'User-Agent': 'Go-http-client/1.1',
+          'Content-Type': 'application/x-www-form-urlencoded',
+          'Accept-Encoding': 'gzip'
+        },
+        data: body.toString(),
+        timeout: config.timeout,
+        proxy: config.proxy ? (() => {
+          const proxyUrl = new URL(config.proxy);
+          return { protocol: proxyUrl.protocol.replace(':', ''), host: proxyUrl.hostname, port: parseInt(proxyUrl.port) };
+        })() : false
+      });
 
-    if (response.ok) {
-      const data = await response.json();
-      token.access_token = data.access_token;
-      token.expires_in = data.expires_in;
+      token.access_token = response.data.access_token;
+      token.expires_in = response.data.expires_in;
       token.timestamp = Date.now();
       this.saveToFile();
       return token;
-    } else {
-      throw { statusCode: response.status, message: await response.text() };
+    } catch (error) {
+      throw { statusCode: error.response?.status, message: error.response?.data || error.message };
     }
   }
 

src/config/config.js

@@ -1,28 +1,73 @@
+import dotenv from 'dotenv';
 import fs from 'fs';
 import log from '../utils/logger.js';
 
-const defaultConfig = {
-  server: { port: 8045, host: '127.0.0.1' },
+const envPath = '.env';
+const defaultEnv = `# 服务器配置
+PORT=8045
+HOST=0.0.0.0
+
+# API 配置
+API_URL=https://daily-cloudcode-pa.sandbox.googleapis.com/v1internal:streamGenerateContent?alt=sse
+API_MODELS_URL=https://daily-cloudcode-pa.sandbox.googleapis.com/v1internal:fetchAvailableModels
+API_NO_STREAM_URL=https://daily-cloudcode-pa.sandbox.googleapis.com/v1internal:generateContent
+API_HOST=daily-cloudcode-pa.sandbox.googleapis.com
+API_USER_AGENT=antigravity/1.11.3 windows/amd64
+
+# 默认参数
+DEFAULT_TEMPERATURE=1
+DEFAULT_TOP_P=0.85
+DEFAULT_TOP_K=50
+DEFAULT_MAX_TOKENS=8096
+
+# 安全配置
+MAX_REQUEST_SIZE=50mb
+API_KEY=sk-text
+
+# 其他配置
+USE_NATIVE_AXIOS=false
+TIMEOUT=30000
+# PROXY=http://127.0.0.1:7897
+
+# 系统提示词
+SYSTEM_INSTRUCTION=你是聊天机器人，名字叫萌萌，如同名字这般，你的性格是软软糯糯萌萌哒的，专门为用户提供聊天和情绪价值，协助进行小说创作或者角色扮演
+`;
+
+if (!fs.existsSync(envPath)) {
+  fs.writeFileSync(envPath, defaultEnv, 'utf8');
+  log.info('✓ 已创建默认 .env 文件');
+}
+
+dotenv.config();
+
+const config = {
+  server: {
+    port: parseInt(process.env.PORT) || 8045,
+    host: process.env.HOST || '127.0.0.1'
+  },
   api: {
-    url: 'https://daily-cloudcode-pa.sandbox.googleapis.com/v1internal:streamGenerateContent?alt=sse',
-    modelsUrl: 'https://daily-cloudcode-pa.sandbox.googleapis.com/v1internal:fetchAvailableModels',
-    noStreamUrl: "https://daily-cloudcode-pa.sandbox.googleapis.com/v1internal:generateContent",
-    host: 'daily-cloudcode-pa.sandbox.googleapis.com',
-    userAgent: 'antigravity/1.11.3 windows/amd64'
+    url: process.env.API_URL || 'https://daily-cloudcode-pa.sandbox.googleapis.com/v1internal:streamGenerateContent?alt=sse',
+    modelsUrl: process.env.API_MODELS_URL || 'https://daily-cloudcode-pa.sandbox.googleapis.com/v1internal:fetchAvailableModels',
+    noStreamUrl: process.env.API_NO_STREAM_URL || 'https://daily-cloudcode-pa.sandbox.googleapis.com/v1internal:generateContent',
+    host: process.env.API_HOST || 'daily-cloudcode-pa.sandbox.googleapis.com',
+    userAgent: process.env.API_USER_AGENT || 'antigravity/1.11.3 windows/amd64'
   },
-  defaults: { temperature: 1, top_p: 0.85, top_k: 50, max_tokens: 8096 },
-  security: { maxRequestSize: '50mb', apiKey: null },
-  useNativeFetch: false,
-  systemInstruction: '你是聊天机器人，专门为用户提供聊天和情绪价值，协助进行小说创作或者角色扮演，也可以提供数学或者代码上的建议'
+  defaults: {
+    temperature: parseFloat(process.env.DEFAULT_TEMPERATURE) || 1,
+    top_p: parseFloat(process.env.DEFAULT_TOP_P) || 0.85,
+    top_k: parseInt(process.env.DEFAULT_TOP_K) || 50,
+    max_tokens: parseInt(process.env.DEFAULT_MAX_TOKENS) || 8096
+  },
+  security: {
+    maxRequestSize: process.env.MAX_REQUEST_SIZE || '50mb',
+    apiKey: process.env.API_KEY || null
+  },
+  useNativeAxios: process.env.USE_NATIVE_AXIOS !== 'false',
+  timeout: parseInt(process.env.TIMEOUT) || 30000,
+  proxy: process.env.PROXY || null,
+  systemInstruction: process.env.SYSTEM_INSTRUCTION || '你是聊天机器人，名字叫萌萌，如同名字这般，你的性格是软软糯糯萌萌哒的，专门为用户提供聊天和情绪价值，协助进行小说创作或者角色扮演'
 };
 
-let config;
-try {
-  config = JSON.parse(fs.readFileSync('./config.json', 'utf8'));
-  log.info('✓ 配置文件加载成功');
-} catch {
-  config = defaultConfig;
-  log.warn('⚠ 配置文件未找到，使用默认配置');
-}
+log.info('✓ 配置加载成功');
 
 export default config;

src/server/index.js

@@ -57,7 +57,7 @@ app.post('/v1/chat/completions', async (req, res) => {
     }
     
     const requestBody = await generateRequestBody(messages, model, params, tools);
-    //console.log(JSON.stringify(requestBody,null,2));
+    // console.log(JSON.stringify(requestBody,null,2));
     
     if (stream) {
       res.setHeader('Content-Type', 'text/event-stream');