feat: set openai account

client/src/pages/api/user/account/update.ts

@@ -5,22 +5,44 @@ import { User } from '@/service/models/user';
 import { connectToDatabase } from '@/service/mongo';
 import { authUser } from '@/service/utils/auth';
 import { UserUpdateParams } from '@/types/user';
+import { getAIChatApi, openaiBaseUrl } from '@/service/ai/openai';
 
 /* 更新一些基本信息 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
   try {
-    const { avatar } = req.body as UserUpdateParams;
+    let { avatar, openaiAccount } = req.body as UserUpdateParams;
 
     const { userId } = await authUser({ req, authToken: true });
 
     await connectToDatabase();
+
+    // auth key
+    if (openaiAccount?.key) {
+      console.log('auth user openai key', openaiAccount?.key);
+
+      const chatAPI = getAIChatApi({
+        base: openaiAccount?.baseUrl || openaiBaseUrl,
+        apikey: openaiAccount?.key
+      });
+
+      const response = await chatAPI.createChatCompletion({
+        model: 'gpt-3.5-turbo',
+        max_tokens: 1,
+        messages: [{ role: 'user', content: 'hi' }]
+      });
+      if (!response?.data?.choices?.[0]?.message?.content) {
+        throw new Error(JSON.stringify(response?.data));
+      }
+    }
+
     // 更新对应的记录
     await User.updateOne(
       {
         _id: userId
       },
       {
-        ...(avatar && { avatar })
+        ...(avatar && { avatar }),
+        ...(openaiAccount && { openaiAccount })
       }
     );