From 53e1129c95186adace1498b0af0c961dc817d952 Mon Sep 17 00:00:00 2001
From: Yomuko <leafmoes@gmail.com>
Date: Wed, 6 Nov 2024 12:44:38 +0800
Subject: [PATCH] add env: port

---
 .env.example | 2 ++
 api/index.js | 5 +++--
 readme.md    | 2 ++
 3 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/.env.example b/.env.example
index 07b0b32..37028e8 100644
--- a/.env.example
+++ b/.env.example
@@ -1,3 +1,5 @@
+# API 服务使用的端口
+PORT = 8787
 # API 调用的前缀地址
 API_PREFIX = '/'
 # 作为调用 API 验证的 API Key
diff --git a/api/index.js b/api/index.js
index 80eb7eb..5cd408a 100644
--- a/api/index.js
+++ b/api/index.js
@@ -7,6 +7,7 @@ dotenv.config()
 
 class Config {
   constructor() {
+    this.PORT = process.env.PORT || '/'
     this.API_PREFIX = process.env.API_PREFIX || '/'
     this.API_KEY = process.env.API_KEY || ''
     this.MAX_RETRY_COUNT = process.env.MAX_RETRY_COUNT || 3
@@ -319,9 +320,9 @@ function newChatCompletionWithModel(text, model) {
   if (typeof addEventListener === 'function') return
   // For Nodejs
   const ittyServer = createServerAdapter(router.fetch)
-  console.log(`Listening on http://localhost:${process.env.PORT || 8787}`)
+  console.log(`Listening on http://localhost:${config.PORT || 8787}`)
   const httpServer = createServer(ittyServer)
-  httpServer.listen(8787)
+  httpServer.listen(config.PORT || 8787)
 })()
 
 // export default router
diff --git a/readme.md b/readme.md
index 3ed1202..cea260e 100644
--- a/readme.md
+++ b/readme.md
@@ -177,6 +177,8 @@ docker stop ddg-chat # 停止服务
 ## 环境变量
 
 ```bash
+# API 服务使用的端口
+PORT = 8787
 # API 调用的前缀地址
 API_PREFIX = '/'
 # 作为调用 API 验证的 API Key