File size: 896 Bytes
b5e5eac |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 |
import {
Entity,
PrimaryGeneratedColumn,
Column,
CreateDateColumn,
ManyToOne,
Index,
} from 'typeorm';
import { Agent } from './Agent';
import { User } from './User';
export enum MessageRole {
USER = 'user',
ASSISTANT = 'assistant',
SYSTEM = 'system',
}
@Entity('chat_messages')
@Index(['agentId', 'userId', 'createdAt'])
export class ChatMessage {
@PrimaryGeneratedColumn('uuid')
id: string;
@ManyToOne(() => Agent, (agent) => agent.messages)
agent: Agent;
@Column()
agentId: string;
@ManyToOne(() => User, { nullable: true })
user: User;
@Column({ nullable: true })
userId?: string;
@Column({
type: 'enum',
enum: MessageRole,
})
role: MessageRole;
@Column('text')
content: string;
@Column('jsonb', { nullable: true })
metadata?: Record<string, any>; // Additional message metadata
@CreateDateColumn()
createdAt: Date;
}
|