feat: session 返回接口方式做显示判断

Author: Chanzhaoyu

service/src/index.ts

@@ -1,6 +1,6 @@
 import express from 'express'
 import type { ChatContext, ChatMessage } from './chatgpt'
-import { chatConfig, chatReplyProcess } from './chatgpt'
+import { chatConfig, chatReplyProcess, currentModel } from './chatgpt'
 import { auth } from './middleware/auth'
 import { isNotEmptyString } from './utils/is'
 
@@ -50,7 +50,7 @@ router.post('/session', async (req, res) => {
   try {
     const AUTH_SECRET_KEY = process.env.AUTH_SECRET_KEY
     const hasAuth = isNotEmptyString(AUTH_SECRET_KEY)
-    res.send({ status: 'Success', message: '', data: { auth: hasAuth } })
+    res.send({ status: 'Success', message: '', data: { auth: hasAuth, model: currentModel() } })
   }
   catch (error) {
     res.send({ status: 'Fail', message: error.message, data: null })

src/components/common/Setting/About.vue

@@ -1,8 +1,9 @@
 <script setup lang='ts'>
-import { onMounted, ref } from 'vue'
+import { computed, onMounted, ref } from 'vue'
 import { NSpin } from 'naive-ui'
 import { fetchChatConfig } from '@/api'
 import pkg from '@/../package.json'
+import { useAuthStore } from '@/store'
 
 interface ConfigState {
   timeoutMs?: number
@@ -13,10 +14,14 @@ interface ConfigState {
   balance?: string
 }
 
+const authStore = useAuthStore()
+
 const loading = ref(false)
 
 const config = ref<ConfigState>()
 
+const isChatGPTAPI = computed<boolean>(() => !!authStore.isChatGPTAPI)
+
 async function fetchConfig() {
   try {
     loading.value = true
@@ -56,8 +61,12 @@ onMounted(() => {
         </p>
       </div>
       <p>{{ $t("setting.api") }}：{{ config?.apiModel ?? '-' }}</p>
-      <p>{{ $t("setting.balance") }}：{{ config?.balance ?? '-' }}</p>
-      <p>{{ $t("setting.reverseProxy") }}：{{ config?.reverseProxy ?? '-' }}</p>
+      <p v-if="isChatGPTAPI">
+        {{ $t("setting.balance") }}：{{ config?.balance ?? '-' }}
+      </p>
+      <p v-if="!isChatGPTAPI">
+        {{ $t("setting.reverseProxy") }}：{{ config?.reverseProxy ?? '-' }}
+      </p>
       <p>{{ $t("setting.timeout") }}：{{ config?.timeoutMs ?? '-' }}</p>
       <p>{{ $t("setting.socks") }}：{{ config?.socksProxy ?? '-' }}</p>
       <p>{{ $t("setting.httpsProxy") }}：{{ config?.httpsProxy ?? '-' }}</p>

src/store/modules/auth/index.ts

@@ -3,9 +3,14 @@ import { getToken, removeToken, setToken } from './helper'
 import { store } from '@/store'
 import { fetchSession } from '@/api'
 
+interface SessionResponse {
+  auth: boolean
+  model: 'ChatGPTAPI' | 'ChatGPTUnofficialProxyAPI'
+}
+
 export interface AuthState {
   token: string | undefined
-  session: { auth: boolean } | null
+  session: SessionResponse | null
 }
 
 export const useAuthStore = defineStore('auth-store', {
@@ -14,10 +19,16 @@ export const useAuthStore = defineStore('auth-store', {
     session: null,
   }),
 
+  getters: {
+    isChatGPTAPI(state): boolean {
+      return state.session?.model === 'ChatGPTAPI'
+    },
+  },
+
   actions: {
     async getSession() {
       try {
-        const { data } = await fetchSession<{ auth: boolean }>()
+        const { data } = await fetchSession<SessionResponse>()
         this.session = { ...data }
         return Promise.resolve(data)
       }