feat: user openai account

client/src/service/ai/openai.ts

@@ -1,3 +1,4 @@
+import { UserModelSchema } from '@/types/mongoSchema';
 import { Configuration, OpenAIApi } from 'openai';
 
 export const openaiBaseUrl = 'https://api.openai.com/v1';
@@ -5,22 +6,22 @@ export const baseUrl = process.env.ONEAPI_URL || process.env.OPENAI_BASE_URL ||
 
 export const systemAIChatKey = process.env.ONEAPI_KEY || process.env.OPENAIKEY || '';
 
-export const getAIChatApi = (props?: { base?: string; apikey?: string }) => {
+export const getAIChatApi = (props?: UserModelSchema['openaiAccount']) => {
   return new OpenAIApi(
     new Configuration({
-      basePath: props?.base || baseUrl,
-      apiKey: props?.apikey || systemAIChatKey
+      basePath: props?.baseUrl || baseUrl,
+      apiKey: props?.key || systemAIChatKey
     })
   );
 };
 
 /* openai axios config */
-export const axiosConfig = (props?: { base?: string; apikey?: string }) => {
+export const axiosConfig = (props?: UserModelSchema['openaiAccount']) => {
   return {
-    baseURL: props?.base || baseUrl, // 此处仅对非 npm 模块有效
+    baseURL: props?.baseUrl || baseUrl, // 此处仅对非 npm 模块有效
     httpsAgent: global.httpsAgent,
     headers: {
-      Authorization: `Bearer ${props?.apikey || systemAIChatKey}`,
+      Authorization: `Bearer ${props?.key || systemAIChatKey}`,
       auth: process.env.OPENAI_BASE_URL_AUTH || ''
     }
   };

client/src/service/events/pushBill.ts

@@ -19,7 +19,7 @@ export const pushTaskBill = async ({
   shareId?: string;
   response: ChatHistoryItemResType[];
 }) => {
-  const total = response.reduce((sum, item) => sum + item.price, 0);
+  const total = response.reduce((sum, item) => sum + item.price, 0) || 1;
 
   await Promise.allSettled([
     Bill.create({

client/src/service/moduleDispatch/agent/classifyQuestion.ts

@@ -5,12 +5,15 @@ import { ChatModuleEnum, ChatRoleEnum, TaskResponseKeyEnum } from '@/constants/c
 import { getAIChatApi, axiosConfig } from '@/service/ai/openai';
 import type { ClassifyQuestionAgentItemType } from '@/types/app';
 import { countModelPrice } from '@/service/events/pushBill';
+import { UserModelSchema } from '@/types/mongoSchema';
+import { getModel } from '@/service/utils/data';
 
 export type CQProps = {
   systemPrompt?: string;
   history?: ChatItemType[];
   userChatInput: string;
   agents: ClassifyQuestionAgentItemType[];
+  userOpenaiAccount: UserModelSchema['openaiAccount'];
 };
 export type CQResponse = {
   [TaskResponseKeyEnum.responseData]: ChatHistoryItemResType;
@@ -23,7 +26,7 @@ const maxTokens = 2000;
 
 /* request openai chat */
 export const dispatchClassifyQuestion = async (props: Record<string, any>): Promise<CQResponse> => {
-  const { agents, systemPrompt, history = [], userChatInput } = props as CQProps;
+  const { agents, systemPrompt, history = [], userChatInput, userOpenaiAccount } = props as CQProps;
 
   const messages: ChatItemType[] = [
     ...(systemPrompt
@@ -63,7 +66,7 @@ export const dispatchClassifyQuestion = async (props: Record<string, any>): Prom
       required: ['type']
     }
   };
-  const chatAPI = getAIChatApi();
+  const chatAPI = getAIChatApi(userOpenaiAccount);
 
   const response = await chatAPI.createChatCompletion(
     {
@@ -74,7 +77,7 @@ export const dispatchClassifyQuestion = async (props: Record<string, any>): Prom
       functions: [agentFunction]
     },
     {
-      ...axiosConfig()
+      ...axiosConfig(userOpenaiAccount)
     }
   );
 
@@ -88,8 +91,8 @@ export const dispatchClassifyQuestion = async (props: Record<string, any>): Prom
     [result.key]: 1,
     [TaskResponseKeyEnum.responseData]: {
       moduleName: ChatModuleEnum.CQ,
-      price: countModelPrice({ model: agentModel, tokens }),
-      model: agentModel,
+      price: userOpenaiAccount?.key ? 0 : countModelPrice({ model: agentModel, tokens }),
+      model: getModel(agentModel)?.name || agentModel,
       tokens,
       cqList: agents,
       cqResult: result.value

client/src/service/moduleDispatch/chat/oneapi.ts

@@ -14,6 +14,7 @@ import { TaskResponseKeyEnum } from '@/constants/chat';
 import { getChatModel } from '@/service/utils/data';
 import { countModelPrice } from '@/service/events/pushBill';
 import { ChatModelItemType } from '@/types/model';
+import { UserModelSchema } from '@/types/mongoSchema';
 
 export type ChatProps = {
   res: NextApiResponse;
@@ -26,6 +27,7 @@ export type ChatProps = {
   quoteQA?: QuoteItemType[];
   systemPrompt?: string;
   limitPrompt?: string;
+  userOpenaiAccount: UserModelSchema['openaiAccount'];
 };
 export type ChatResponse = {
   [TaskResponseKeyEnum.answerText]: string;
@@ -45,7 +47,8 @@ export const dispatchChatCompletion = async (props: Record<string, any>): Promis
     quoteQA = [],
     userChatInput,
     systemPrompt = '',
-    limitPrompt = ''
+    limitPrompt = '',
+    userOpenaiAccount
   } = props as ChatProps;
 
   // temperature adapt
@@ -77,7 +80,7 @@ export const dispatchChatCompletion = async (props: Record<string, any>): Promis
   // FastGpt temperature range: 1~10
   temperature = +(modelConstantsData.maxTemperature * (temperature / 10)).toFixed(2);
   temperature = Math.max(temperature, 0.01);
-  const chatAPI = getAIChatApi();
+  const chatAPI = getAIChatApi(userOpenaiAccount);
 
   const response = await chatAPI.createChatCompletion(
     {
@@ -92,7 +95,7 @@ export const dispatchChatCompletion = async (props: Record<string, any>): Promis
     {
       timeout: stream ? 60000 : 480000,
       responseType: stream ? 'stream' : 'json',
-      ...axiosConfig()
+      ...axiosConfig(userOpenaiAccount)
     }
   );
 
@@ -136,7 +139,7 @@ export const dispatchChatCompletion = async (props: Record<string, any>): Promis
     [TaskResponseKeyEnum.answerText]: answerText,
     [TaskResponseKeyEnum.responseData]: {
       moduleName: ChatModuleEnum.AIChat,
-      price: countModelPrice({ model, tokens: totalTokens }),
+      price: userOpenaiAccount?.key ? 0 : countModelPrice({ model, tokens: totalTokens }),
       model: modelConstantsData.name,
       tokens: totalTokens,
       question: userChatInput,

client/src/service/utils/auth.ts

@@ -2,7 +2,7 @@ import type { NextApiRequest } from 'next';
 import jwt from 'jsonwebtoken';
 import Cookie from 'cookie';
 import { App, OpenApi, User, OutLink, KB } from '../mongo';
-import type { AppSchema } from '@/types/mongoSchema';
+import type { AppSchema, UserModelSchema } from '@/types/mongoSchema';
 import { formatPrice } from '@/utils/user';
 import { ERROR_ENUM } from '../errorCode';
 
@@ -37,7 +37,7 @@ export const authBalanceByUid = async (uid: string) => {
     return Promise.reject(ERROR_ENUM.unAuthorization);
   }
 
-  if (!user.openaiKey && formatPrice(user.balance) <= 0) {
+  if (user.balance <= 0) {
     return Promise.reject(ERROR_ENUM.insufficientQuota);
   }
   return user;
@@ -151,14 +151,17 @@ export const authUser = async ({
   }
 
   // balance check
-  if (authBalance) {
-    await authBalanceByUid(uid);
-  }
+  const user = await (() => {
+    if (authBalance) {
+      return authBalanceByUid(uid);
+    }
+  })();
 
   return {
     userId: uid,
     appId,
-    authType
+    authType,
+    user
   };
 };
 
@@ -217,7 +220,13 @@ export const authShareChat = async ({ shareId }: { shareId: string }) => {
     return Promise.reject('分享链接已失效');
   }
 
+  const uid = String(shareChat.userId);
+
+  // authBalance
+  const user = await authBalanceByUid(uid);
+
   return {
+    user,
     userId: String(shareChat.userId),
     appId: String(shareChat.appId),
     authType: 'token' as AuthType