fix: 修复部份 bug (#131)

* fix: 主题模式图标不一致的问题[#125]

* fix: 修复样式问题[#123][#126]

* perf: 优化代码和添加类型

service/src/chatgpt.ts

@@ -1,22 +1,18 @@
 import * as dotenv from 'dotenv'
 import 'isomorphic-fetch'
-import type { ChatGPTAPI, ChatMessage, SendMessageOptions } from 'chatgpt'
-import { ChatGPTUnofficialProxyAPI } from 'chatgpt'
+import type { ChatMessage, SendMessageOptions } from 'chatgpt'
+import { ChatGPTAPI, ChatGPTUnofficialProxyAPI } from 'chatgpt'
 import { SocksProxyAgent } from 'socks-proxy-agent'
 import fetch from 'node-fetch'
 import { sendResponse } from './utils'
+import type { ApiModel, ChatContext, ChatGPTAPIOptions, ChatGPTUnofficialProxyAPIOptions, ModelConfig } from './types'
 
 dotenv.config()
 
-let apiModel: 'ChatGPTAPI' | 'ChatGPTUnofficialProxyAPI' | undefined
-
-interface ChatContext {
-  conversationId?: string
-  parentMessageId?: string
-}
-
 const timeoutMs: number = !isNaN(+process.env.TIMEOUT_MS) ? +process.env.TIMEOUT_MS : 30 * 1000
 
+let apiModel: ApiModel
+
 if (!process.env.OPENAI_API_KEY && !process.env.OPENAI_ACCESS_TOKEN)
   throw new Error('Missing OPENAI_API_KEY or OPENAI_ACCESS_TOKEN environment variable')
 
@@ -25,15 +21,20 @@ let api: ChatGPTAPI | ChatGPTUnofficialProxyAPI
 // To use ESM in CommonJS, you can use a dynamic import
 (async () => {
   // More Info: https://github.com/transitive-bullshit/chatgpt-api
-  const { ChatGPTAPI } = await import('chatgpt')
 
   if (process.env.OPENAI_API_KEY) {
-    api = new ChatGPTAPI({ apiKey: process.env.OPENAI_API_KEY })
+    const options: ChatGPTAPIOptions = {
+      apiKey: process.env.OPENAI_API_KEY,
+      debug: false,
+    }
+
+    api = new ChatGPTAPI({ ...options })
     apiModel = 'ChatGPTAPI'
   }
   else {
-    const options = {
-      debug: true,
+    const options: ChatGPTUnofficialProxyAPIOptions = {
+      accessToken: process.env.OPENAI_ACCESS_TOKEN,
+      debug: false,
     }
 
     if (process.env.SOCKS_PROXY_HOST && process.env.SOCKS_PROXY_PORT) {
@@ -41,16 +42,13 @@ let api: ChatGPTAPI | ChatGPTUnofficialProxyAPI
         hostname: process.env.SOCKS_PROXY_HOST,
         port: process.env.SOCKS_PROXY_PORT,
       })
-      globalThis.console.log(`Using socks proxy: ${process.env.SOCKS_PROXY_HOST}:${process.env.SOCKS_PROXY_PORT}`)
       options.fetch = (url, options) => {
         return fetch(url, { agent, ...options })
       }
     }
 
-    if (process.env.API_REVERSE_PROXY) {
+    if (process.env.API_REVERSE_PROXY)
       options.apiReverseProxyUrl = process.env.API_REVERSE_PROXY
-      globalThis.console.log(`Using api reverse proxy: ${process.env.API_REVERSE_PROXY}`)
-    }
 
     api = new ChatGPTUnofficialProxyAPI({
       accessToken: process.env.OPENAI_ACCESS_TOKEN,
@@ -82,7 +80,6 @@ async function chatReply(
   }
 }
 
-/**  实验性质的函数，用于处理聊天过程中的中间结果 */
 async function chatReplyProcess(
   message: string,
   lastContext?: { conversationId?: string; parentMessageId?: string },
@@ -119,7 +116,7 @@ async function chatConfig() {
       reverseProxy: process.env.API_REVERSE_PROXY,
       timeoutMs,
       socksProxy: (process.env.SOCKS_PROXY_HOST && process.env.SOCKS_PROXY_PORT) ? (`${process.env.SOCKS_PROXY_HOST}:${process.env.SOCKS_PROXY_PORT}`) : '-',
-    },
+    } as ModelConfig,
   })
 }
 

service/src/index.ts

@@ -26,7 +26,6 @@ router.post('/chat', async (req, res) => {
   }
 })
 
-/**  实验性质的函数，用于处理聊天过程中的中间结果 */
 router.post('/chat-process', async (req, res) => {
   res.setHeader('Content-type', 'application/octet-stream')
 

service/src/types.ts

@@ -0,0 +1,30 @@
+import type { FetchFn, openai } from 'chatgpt'
+
+export interface ChatContext {
+  conversationId?: string
+  parentMessageId?: string
+}
+
+export interface ChatGPTAPIOptions {
+  apiKey: string
+  debug?: boolean
+  completionParams?: Partial<openai.CompletionParams>
+}
+
+export interface ChatGPTUnofficialProxyAPIOptions {
+  accessToken: string
+  apiReverseProxyUrl?: string
+  model?: string
+  debug?: boolean
+  headers?: Record<string, string>
+  fetch?: FetchFn
+}
+
+export interface ModelConfig {
+  apiModel?: ApiModel
+  reverseProxy?: string
+  timeoutMs?: number
+  socksProxy?: string
+}
+
+export type ApiModel = 'ChatGPTAPI' | 'ChatGPTUnofficialProxyAPI' | undefined