mirror of
https://github.com/labring/FastGPT.git
synced 2025-07-23 13:03:50 +00:00

* 4.7-alpha3 (#62) * doc * Optimize possible null Pointers and parts of Ux * fix: mulity index training error * feat: doc and rename question guide * fix ios speech input (#59) * fix: prompt editor variables nowrap (#61) * change openapi import in http module with curl import (#60) * chore(ui): dataset import modal ui (#58) * chore(ui): dataset import modal ui * use component * fix height * 4.7 (#63) * fix: claude3 image type verification failed (#1038) (#1040) * perf: curl import modal * doc img * perf: adapt cohere rerank * perf: code * perf: input style * doc --------- Co-authored-by: xiaotian <dimsky@163.com> * fix: ts * docker deploy * perf: prompt call * doc * ts * finish ui * perf: outlink detail ux * perf: user schema * fix: plugin update * feat: get current time plugin * fix: ts * perf: fetch anamation * perf: mark ux * doc * perf: select app ux * fix: split text custom string conflict * peref: inform readed * doc * memo flow component * perf: version * faq * feat: flow max runtimes * feat: similarity tip * feat: auto detect file encoding * Supports asymmetric vector model * fix: ts * perf: max w * move code * perf: hide whisper * fix: ts * feat: system msg modal * perf: catch error * perf: inform tip * fix: inform --------- Co-authored-by: heheer <71265218+newfish-cmyk@users.noreply.github.com> Co-authored-by: xiaotian <dimsky@163.com>
110 lines
2.7 KiB
TypeScript
110 lines
2.7 KiB
TypeScript
import { TeamItemType, TeamMemberWithTeamSchema } from '@fastgpt/global/support/user/team/type';
|
|
import { ClientSession, Types } from '../../../common/mongo';
|
|
import {
|
|
TeamMemberRoleEnum,
|
|
TeamMemberStatusEnum,
|
|
notLeaveStatus
|
|
} from '@fastgpt/global/support/user/team/constant';
|
|
import { MongoTeamMember } from './teamMemberSchema';
|
|
import { MongoTeam } from './teamSchema';
|
|
|
|
async function getTeamMember(match: Record<string, any>): Promise<TeamItemType> {
|
|
const tmb = (await MongoTeamMember.findOne(match).populate('teamId')) as TeamMemberWithTeamSchema;
|
|
|
|
if (!tmb) {
|
|
return Promise.reject('member not exist');
|
|
}
|
|
|
|
return {
|
|
userId: String(tmb.userId),
|
|
teamId: String(tmb.teamId._id),
|
|
teamName: tmb.teamId.name,
|
|
memberName: tmb.name,
|
|
avatar: tmb.teamId.avatar,
|
|
balance: tmb.teamId.balance,
|
|
tmbId: String(tmb._id),
|
|
teamDomain: tmb.teamId?.teamDomain,
|
|
role: tmb.role,
|
|
status: tmb.status,
|
|
defaultTeam: tmb.defaultTeam,
|
|
canWrite: tmb.role !== TeamMemberRoleEnum.visitor
|
|
};
|
|
}
|
|
|
|
export async function getTmbInfoByTmbId({ tmbId }: { tmbId: string }) {
|
|
if (!tmbId) {
|
|
return Promise.reject('tmbId or userId is required');
|
|
}
|
|
return getTeamMember({
|
|
_id: new Types.ObjectId(String(tmbId)),
|
|
status: notLeaveStatus
|
|
});
|
|
}
|
|
|
|
export async function getUserDefaultTeam({ userId }: { userId: string }) {
|
|
if (!userId) {
|
|
return Promise.reject('tmbId or userId is required');
|
|
}
|
|
return getTeamMember({
|
|
userId: new Types.ObjectId(userId),
|
|
defaultTeam: true
|
|
});
|
|
}
|
|
export async function createDefaultTeam({
|
|
userId,
|
|
teamName = 'My Team',
|
|
avatar = '/icon/logo.svg',
|
|
balance,
|
|
session
|
|
}: {
|
|
userId: string;
|
|
teamName?: string;
|
|
avatar?: string;
|
|
balance?: number;
|
|
session: ClientSession;
|
|
}) {
|
|
// auth default team
|
|
const tmb = await MongoTeamMember.findOne({
|
|
userId: new Types.ObjectId(userId),
|
|
defaultTeam: true
|
|
});
|
|
|
|
if (!tmb) {
|
|
// create
|
|
const [{ _id: insertedId }] = await MongoTeam.create(
|
|
[
|
|
{
|
|
ownerId: userId,
|
|
name: teamName,
|
|
avatar,
|
|
balance,
|
|
createTime: new Date()
|
|
}
|
|
],
|
|
{ session }
|
|
);
|
|
await MongoTeamMember.create(
|
|
[
|
|
{
|
|
teamId: insertedId,
|
|
userId,
|
|
name: 'Owner',
|
|
role: TeamMemberRoleEnum.owner,
|
|
status: TeamMemberStatusEnum.active,
|
|
createTime: new Date(),
|
|
defaultTeam: true
|
|
}
|
|
],
|
|
{ session }
|
|
);
|
|
console.log('create default team', userId);
|
|
} else {
|
|
console.log('default team exist', userId);
|
|
await MongoTeam.findByIdAndUpdate(tmb.teamId, {
|
|
$set: {
|
|
...(balance !== undefined && { balance })
|
|
}
|
|
});
|
|
}
|
|
}
|