fix: the specified model is invalid

Author: Kerwin

service/src/chatgpt/index.ts

@@ -326,9 +326,9 @@ async function randomKeyConfig(keys: KeyConfig[]): Promise < KeyConfig | null >
   return thisKey
 }
 
-async function getRandomApiKey(user: UserInfo): Promise<KeyConfig | undefined> {
+async function getRandomApiKey(user: UserInfo, chatModel: CHATMODEL): Promise<KeyConfig | undefined> {
   const keys = (await getCacheApiKeys()).filter(d => hasAnyRole(d.userRoles, user.roles))
-  return randomKeyConfig(keys)
+  return randomKeyConfig(keys.filter(d => d.chatModels.includes(chatModel)))
 }
 
 async function releaseApiKey(key: KeyConfig) {

service/src/index.ts

@@ -430,7 +430,7 @@ router.post('/chat-process', [auth, limiter], async (req, res) => {
       temperature,
       top_p,
       chatModel: user.config.chatModel,
-      key: await getRandomApiKey(user),
+      key: await getRandomApiKey(user, user.config.chatModel),
     })
     // return the whole response including usage
     res.write(`\n${JSON.stringify(result.data)}`)