mirror of
https://github.com/labring/FastGPT.git
synced 2025-07-23 05:12:39 +00:00
34602b25df
* perf: redirect request and err log replace perf: dataset openapi feat: session fix: retry input error feat: 468 doc sub page feat: standard sub perf: rerank tip perf: rerank tip perf: api sdk perf: openapi sub plan perf: sub ui fix: ts * perf: init log * fix: variable select * sub page * icon * perf: llm model config * perf: menu ux * perf: system store * perf: publish app name * fix: init data * perf: flow edit ux * fix: value type format and ux * fix prompt editor default value (#13) * fix prompt editor default value * fix prompt editor update when not focus * add key with variable --------- Co-authored-by: Archer <545436317@qq.com> * fix: value type * doc * i18n * import path * home page * perf: mongo session running * fix: ts * perf: use toast * perf: flow edit * perf: sse response * slider ui * fetch error * fix prompt editor rerender when not focus by key defaultvalue (#14) * perf: prompt editor * feat: dataset search concat * perf: doc * fix:ts * perf: doc * fix json editor onblur value (#15) * faq * vector model default config * ipv6 --------- Co-authored-by: heheer <71265218+newfish-cmyk@users.noreply.github.com>
106 lines
2.6 KiB
TypeScript
106 lines
2.6 KiB
TypeScript
import { connectionMongo, type Model } from '../../common/mongo';
|
|
const { Schema, model, models } = connectionMongo;
|
|
import { ChatItemSchema as ChatItemType } from '@fastgpt/global/core/chat/type';
|
|
import { ChatRoleMap } from '@fastgpt/global/core/chat/constants';
|
|
import { getNanoid } from '@fastgpt/global/common/string/tools';
|
|
import {
|
|
TeamCollectionName,
|
|
TeamMemberCollectionName
|
|
} from '@fastgpt/global/support/user/team/constant';
|
|
import { appCollectionName } from '../app/schema';
|
|
import { userCollectionName } from '../../support/user/schema';
|
|
import { ModuleOutputKeyEnum } from '@fastgpt/global/core/module/constants';
|
|
|
|
export const ChatItemCollectionName = 'chatitems';
|
|
|
|
const ChatItemSchema = new Schema({
|
|
teamId: {
|
|
type: Schema.Types.ObjectId,
|
|
ref: TeamCollectionName,
|
|
required: true
|
|
},
|
|
tmbId: {
|
|
type: Schema.Types.ObjectId,
|
|
ref: TeamMemberCollectionName,
|
|
required: true
|
|
},
|
|
userId: {
|
|
type: Schema.Types.ObjectId,
|
|
ref: userCollectionName
|
|
},
|
|
chatId: {
|
|
type: String,
|
|
require: true
|
|
},
|
|
dataId: {
|
|
type: String,
|
|
require: true,
|
|
default: () => getNanoid(22)
|
|
},
|
|
appId: {
|
|
type: Schema.Types.ObjectId,
|
|
ref: appCollectionName,
|
|
required: true
|
|
},
|
|
time: {
|
|
type: Date,
|
|
default: () => new Date()
|
|
},
|
|
obj: {
|
|
// chat role
|
|
type: String,
|
|
required: true,
|
|
enum: Object.keys(ChatRoleMap)
|
|
},
|
|
value: {
|
|
// chat content
|
|
type: String,
|
|
default: ''
|
|
},
|
|
userGoodFeedback: {
|
|
type: String
|
|
},
|
|
userBadFeedback: {
|
|
type: String
|
|
},
|
|
customFeedbacks: {
|
|
type: [String]
|
|
},
|
|
adminFeedback: {
|
|
type: {
|
|
datasetId: String,
|
|
collectionId: String,
|
|
dataId: String,
|
|
q: String,
|
|
a: String
|
|
}
|
|
},
|
|
[ModuleOutputKeyEnum.responseData]: {
|
|
type: Array,
|
|
default: []
|
|
}
|
|
});
|
|
|
|
try {
|
|
ChatItemSchema.index({ dataId: 1 }, { background: true });
|
|
/* delete by app;
|
|
delete by chat id;
|
|
get chat list;
|
|
get chat logs;
|
|
close custom feedback;
|
|
*/
|
|
ChatItemSchema.index({ appId: 1, chatId: 1, dataId: 1 }, { background: true });
|
|
ChatItemSchema.index({ time: -1, obj: 1 }, { background: true });
|
|
ChatItemSchema.index({ userGoodFeedback: 1 }, { background: true });
|
|
ChatItemSchema.index({ userBadFeedback: 1 }, { background: true });
|
|
ChatItemSchema.index({ customFeedbacks: 1 }, { background: true });
|
|
ChatItemSchema.index({ adminFeedback: 1 }, { background: true });
|
|
} catch (error) {
|
|
console.log(error);
|
|
}
|
|
|
|
export const MongoChatItem: Model<ChatItemType> =
|
|
models[ChatItemCollectionName] || model(ChatItemCollectionName, ChatItemSchema);
|
|
|
|
MongoChatItem.syncIndexes();
|