34 lines
606 B
TypeScript
34 lines
606 B
TypeScript
import {
|
|
Entity,
|
|
Column,
|
|
PrimaryColumn,
|
|
CreateDateColumn,
|
|
} from 'typeorm';
|
|
|
|
@Entity('messages')
|
|
export class MessageORM {
|
|
@PrimaryColumn('uuid')
|
|
id: string;
|
|
|
|
@Column({ name: 'conversation_id' })
|
|
conversationId: string;
|
|
|
|
@Column({ length: 20 })
|
|
role: string;
|
|
|
|
@Column('text')
|
|
content: string;
|
|
|
|
@Column({ name: 'input_tokens', default: 0 })
|
|
inputTokens: number;
|
|
|
|
@Column({ name: 'output_tokens', default: 0 })
|
|
outputTokens: number;
|
|
|
|
@Column('jsonb', { nullable: true })
|
|
metadata: Record<string, unknown>;
|
|
|
|
@CreateDateColumn({ name: 'created_at' })
|
|
createdAt: Date;
|
|
}
|