feat: model share market

src/pages/api/chat/init.ts

@@ -22,7 +22,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
     await connectToDatabase();
 
     // 获取 model 数据
-    const { model } = await authModel(modelId, userId);
+    const { model } = await authModel({ modelId, userId, authUser: false, authOwner: false });
 
     // 历史记录
     let history: ChatItemType[] = [];
@@ -53,6 +53,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
         modelId: modelId,
         name: model.name,
         avatar: model.avatar,
+        intro: model.share.intro,
         modelName: model.service.modelName,
         chatModel: model.service.chatModel,
         history

src/pages/api/chat/saveChat.ts

@@ -27,9 +27,10 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
       value: item.value
     }));
 
+    await authModel({ modelId, userId, authOwner: false });
+
     // 没有 chatId, 创建一个对话
     if (!chatId) {
-      await authModel(modelId, userId);
       const { _id } = await Chat.create({
         userId,
         modelId,

src/pages/api/model/data/getModelData.ts

@@ -4,6 +4,7 @@ import { connectToDatabase } from '@/service/mongo';
 import { authToken } from '@/service/utils/tools';
 import { PgClient } from '@/service/pg';
 import type { PgModelDataItemType } from '@/types/pg';
+import { authModel } from '@/service/utils/auth';
 
 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
   try {
@@ -36,9 +37,14 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
 
     await connectToDatabase();
 
+    const { model } = await authModel({
+      userId,
+      modelId,
+      authOwner: false
+    });
+
     const where: any = [
-      ['user_id', userId],
-      'AND',
+      ...(model.share.isShareDetail ? [] : [['user_id', userId], 'AND']),
       ['model_id', modelId],
       ...(searchText ? ['AND', `(q LIKE '%${searchText}%' OR a LIKE '%${searchText}%')`] : [])
     ];

src/pages/api/model/data/pushModelDataCsv.ts

@@ -1,10 +1,11 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
-import { connectToDatabase, Model } from '@/service/mongo';
+import { connectToDatabase } from '@/service/mongo';
 import { authToken } from '@/service/utils/tools';
 import { generateVector } from '@/service/events/generateVector';
 import { ModelDataStatusEnum } from '@/constants/model';
 import { PgClient } from '@/service/pg';
+import { authModel } from '@/service/utils/auth';
 
 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
   try {
@@ -28,15 +29,11 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
     await connectToDatabase();
 
     // 验证是否是该用户的 model
-    const model = await Model.findOne({
-      _id: modelId,
-      userId
+    await authModel({
+      userId,
+      modelId
     });
 
-    if (!model) {
-      throw new Error('无权操作该模型');
-    }
-
     // 去重
     const searchRes = await Promise.allSettled(
       data.map(async ([q, a]) => {

src/pages/api/model/data/pushModelDataInput.ts

@@ -1,10 +1,11 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
-import { connectToDatabase, Model } from '@/service/mongo';
+import { connectToDatabase } from '@/service/mongo';
 import { authToken } from '@/service/utils/tools';
 import { ModelDataSchema } from '@/types/mongoSchema';
 import { generateVector } from '@/service/events/generateVector';
 import { PgClient } from '@/service/pg';
+import { authModel } from '@/service/utils/auth';
 
 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
   try {
@@ -28,15 +29,11 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
     await connectToDatabase();
 
     // 验证是否是该用户的 model
-    const model = await Model.findOne({
-      _id: modelId,
-      userId
+    await authModel({
+      userId,
+      modelId
     });
 
-    if (!model) {
-      throw new Error('无权操作该模型');
-    }
-
     // 插入记录
     await PgClient.insert('modelData', {
       values: data.map((item) => [

src/pages/api/model/del.ts

@@ -3,6 +3,7 @@ import { jsonRes } from '@/service/response';
 import { Chat, Model, connectToDatabase } from '@/service/mongo';
 import { authToken } from '@/service/utils/tools';
 import { PgClient } from '@/service/pg';
+import { authModel } from '@/service/utils/auth';
 
 /* 获取我的模型 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
@@ -21,18 +22,14 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
     // 凭证校验
     const userId = await authToken(authorization);
 
+    await connectToDatabase();
+
     // 验证是否是该用户的 model
-    const model = await Model.findOne({
-      _id: modelId,
+    await authModel({
+      modelId,
       userId
     });
 
-    if (!model) {
-      throw new Error('无权操作该模型');
-    }
-
-    await connectToDatabase();
-
     // 删除 pg 中所有该模型的数据
     await PgClient.delete('modelData', {
       where: [['user_id', userId], 'AND', ['model_id', modelId]]

src/pages/api/model/detail.tsx

@@ -2,8 +2,7 @@ import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { connectToDatabase } from '@/service/mongo';
 import { authToken } from '@/service/utils/tools';
-import { Model } from '@/service/models/model';
-import type { ModelSchema } from '@/types/mongoSchema';
+import { authModel } from '@/service/utils/auth';
 
 /* 获取我的模型 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
@@ -14,7 +13,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
       throw new Error('无权操作');
     }
 
-    const { modelId } = req.query;
+    const { modelId } = req.query as { modelId: string };
 
     if (!modelId) {
       throw new Error('参数错误');
@@ -25,16 +24,12 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
 
     await connectToDatabase();
 
-    // 根据 userId 获取模型信息
-    const model = await Model.findOne<ModelSchema>({
+    const { model } = await authModel({
+      modelId,
       userId,
-      _id: modelId
+      authOwner: false
     });
 
-    if (!model) {
-      throw new Error('模型不存在');
-    }
-
     jsonRes(res, {
       data: model
     });

src/pages/api/model/list.ts

@@ -4,7 +4,7 @@ import { connectToDatabase } from '@/service/mongo';
 import { authToken } from '@/service/utils/tools';
 import { Model } from '@/service/models/model';
 
-/* 获取我的模型 */
+/* 获取模型列表 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
   try {
     const { authorization } = req.headers;

src/pages/api/model/share/getModels.ts

@@ -0,0 +1,54 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import { connectToDatabase } from '@/service/mongo';
+import { authToken } from '@/service/utils/tools';
+import { Model } from '@/service/models/model';
+import type { PagingData } from '@/types';
+import type { ShareModelItem } from '@/types/model';
+
+/* 获取模型列表 */
+export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
+  try {
+    // 凭证校验
+    await authToken(req.headers.authorization);
+
+    const {
+      searchText = '',
+      pageNum = 1,
+      pageSize = 20
+    } = req.body as { searchText: string; pageNum: number; pageSize: number };
+
+    await connectToDatabase();
+
+    const regex = new RegExp(searchText, 'i');
+
+    const where = {
+      $and: [
+        { 'share.isShare': true },
+        { $or: [{ name: { $regex: regex } }, { 'share.intro': { $regex: regex } }] }
+      ]
+    };
+
+    // 根据分享的模型
+    const models = await Model.find(where, '_id avatar name userId share')
+      .sort({
+        'share.collection': -1
+      })
+      .limit(pageSize)
+      .skip((pageNum - 1) * pageSize);
+
+    jsonRes<PagingData<ShareModelItem>>(res, {
+      data: {
+        pageNum,
+        pageSize,
+        data: models,
+        total: await Model.countDocuments(where)
+      }
+    });
+  } catch (err) {
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}

src/pages/api/model/update.ts

@@ -4,11 +4,12 @@ import { connectToDatabase } from '@/service/mongo';
 import { authToken } from '@/service/utils/tools';
 import { Model } from '@/service/models/model';
 import type { ModelUpdateParams } from '@/types/model';
+import { authModel } from '@/service/utils/auth';
 
 /* 获取我的模型 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
   try {
-    const { name, search, service, security, systemPrompt, temperature } =
+    const { name, search, share, service, security, systemPrompt, temperature } =
       req.body as ModelUpdateParams;
     const { modelId } = req.query as { modelId: string };
     const { authorization } = req.headers;
@@ -26,6 +27,11 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
 
     await connectToDatabase();
 
+    await authModel({
+      modelId,
+      userId
+    });
+
     // 更新模型
     await Model.updateOne(
       {
@@ -36,6 +42,9 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
         name,
         systemPrompt,
         temperature,
+        'share.isShare': share.isShare,
+        'share.isShareDetail': share.isShareDetail,
+        'share.intro': share.intro,
         search,
         security
       }