Add database entity for chat message feedback

This commit is contained in:
Ilango
2024-02-06 17:07:22 +05:30
parent 7faaf13ccc
commit 7d00f6fbe9
3 changed files with 45 additions and 0 deletions
+15
View File
@@ -6,6 +6,11 @@ export enum chatType {
INTERNAL = 'INTERNAL',
EXTERNAL = 'EXTERNAL'
}
export enum ChatMessageRatingType {
THUMBS_UP = 'THUMBS_UP',
THUMBS_DOWN = 'THUMBS_DOWN'
}
/**
* Databases
*/
@@ -38,6 +43,16 @@ export interface IChatMessage {
createdDate: Date
}
export interface IChatMessageFeedback {
id: string
content?: string
chatflowid: string
chatId: string
messageId: string
rating: ChatMessageRatingType
createdDate: Date
}
export interface ITool {
id: string
name: string
@@ -0,0 +1,28 @@
/* eslint-disable */
import { Entity, Column, CreateDateColumn, PrimaryGeneratedColumn, Index } from 'typeorm'
import { ChatMessageRatingType, IChatMessageFeedback } from '../../Interface'
@Entity()
export class ChatMessageFeedback implements IChatMessageFeedback {
@PrimaryGeneratedColumn('uuid')
id: string
@Index()
@Column()
chatflowid: string
@Column({ type: 'text' })
content?: string
@Column()
chatId: string
@Column()
messageId: string
@Column()
rating: ChatMessageRatingType
@CreateDateColumn()
createdDate: Date
}
@@ -1,5 +1,6 @@
import { ChatFlow } from './ChatFlow'
import { ChatMessage } from './ChatMessage'
import { ChatMessageFeedback } from './ChatMessageFeedback'
import { Credential } from './Credential'
import { Tool } from './Tool'
import { Assistant } from './Assistant'
@@ -8,6 +9,7 @@ import { Variable } from './Variable'
export const entities = {
ChatFlow,
ChatMessage,
ChatMessageFeedback,
Credential,
Tool,
Assistant,