feat: share窗口

src/pages/api/chat/chatGpt.ts

@@ -17,6 +17,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
     prompt: ChatItemType;
     chatId: string;
   };
+  const { authorization } = req.headers;
 
   try {
     if (!chatId || !prompt) {
@@ -25,7 +26,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
 
     await connectToDatabase();
 
-    const { chat, userApiKey, systemKey, userId } = await authChat(chatId);
+    const { chat, userApiKey, systemKey, userId } = await authChat(chatId, authorization);
 
     const model: ModelSchema = chat.modelId;
 

src/pages/api/chat/generate.ts

@@ -7,7 +7,10 @@ import type { ModelSchema } from '@/types/mongoSchema';
 /* 获取我的模型 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
   try {
-    const { modelId } = req.query;
+    const { modelId, isShare = 'false' } = req.query as {
+      modelId: string;
+      isShare?: 'true' | 'false';
+    };
     const { authorization } = req.headers;
 
     if (!authorization) {
@@ -40,6 +43,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
       expiredTime: Date.now() + model.security.expiredTime,
       loadAmount: model.security.maxLoadAmount,
       updateTime: Date.now(),
+      isShare: isShare === 'true',
       content: []
     });
 

src/pages/api/chat/gpt3.ts

@@ -12,6 +12,7 @@ import { pushBill } from '@/service/events/bill';
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
   try {
     const { prompt, chatId } = req.body as { prompt: ChatItemType[]; chatId: string };
+    const { authorization } = req.headers;
 
     if (!prompt || !chatId) {
       throw new Error('缺少参数');
@@ -19,7 +20,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
 
     await connectToDatabase();
 
-    const { chat, userApiKey, systemKey, userId } = await authChat(chatId);
+    const { chat, userApiKey, systemKey, userId } = await authChat(chatId, authorization);
 
     const model = chat.modelId;