c5e4684b44
### What problem does this PR solve? Feat: Let system variables appear in operator prompts #3221 ### Type of change - [x] New Feature (non-breaking change which adds functionality)
57 lines
1.7 KiB
TypeScript
57 lines
1.7 KiB
TypeScript
import { MessageType } from '@/constants/chat';
|
|
import { IConversation, IReference } from '@/interfaces/database/chat';
|
|
import { isEmpty } from 'lodash';
|
|
import { EmptyConversationId } from './constants';
|
|
import { IMessage } from './interface';
|
|
|
|
export const isConversationIdExist = (conversationId: string) => {
|
|
return conversationId !== EmptyConversationId && conversationId !== '';
|
|
};
|
|
|
|
export const getDocumentIdsFromConversionReference = (data: IConversation) => {
|
|
const documentIds = data.reference.reduce(
|
|
(pre: Array<string>, cur: IReference) => {
|
|
cur.doc_aggs
|
|
?.map((x) => x.doc_id)
|
|
.forEach((x) => {
|
|
if (pre.every((y) => y !== x)) {
|
|
pre.push(x);
|
|
}
|
|
});
|
|
return pre;
|
|
},
|
|
[],
|
|
);
|
|
return documentIds.join(',');
|
|
};
|
|
|
|
export const buildMessageItemReference = (
|
|
conversation: { message: IMessage[]; reference: IReference[] },
|
|
message: IMessage,
|
|
) => {
|
|
const assistantMessages = conversation.message?.filter(
|
|
(x) => x.role === MessageType.Assistant,
|
|
);
|
|
const referenceIndex = assistantMessages.findIndex(
|
|
(x) => x.id === message.id,
|
|
);
|
|
const reference = !isEmpty(message?.reference)
|
|
? message?.reference
|
|
: (conversation?.reference ?? [])[referenceIndex];
|
|
|
|
return reference ?? { doc_aggs: [], chunks: [], total: 0 };
|
|
};
|
|
|
|
const oldReg = /(#{2}\d+\${2})/g;
|
|
export const currentReg = /\[ID:(\d+)\]/g;
|
|
|
|
// To be compatible with the old index matching mode
|
|
export const replaceTextByOldReg = (text: string) => {
|
|
return (
|
|
text
|
|
// ?.replace(currentReg, transformReg)
|
|
.replace(oldReg, (substring: string) => {
|
|
return `[ID:${substring.slice(2, -2)}]`;
|
|
})
|
|
);
|
|
};
|