35 lines
1.2 KiB
TypeScript
35 lines
1.2 KiB
TypeScript
import { pgTable, serial, integer, text, timestamp } from "drizzle-orm/pg-core";
|
|
import { createInsertSchema } from "drizzle-zod";
|
|
import { z } from "zod";
|
|
import { sql } from "drizzle-orm";
|
|
|
|
export const conversations = pgTable("conversations", {
|
|
id: serial("id").primaryKey(),
|
|
title: text("title").notNull(),
|
|
createdAt: timestamp("created_at").default(sql`CURRENT_TIMESTAMP`).notNull(),
|
|
});
|
|
|
|
export const messages = pgTable("messages", {
|
|
id: serial("id").primaryKey(),
|
|
conversationId: integer("conversation_id").notNull().references(() => conversations.id, { onDelete: "cascade" }),
|
|
role: text("role").notNull(),
|
|
content: text("content").notNull(),
|
|
createdAt: timestamp("created_at").default(sql`CURRENT_TIMESTAMP`).notNull(),
|
|
});
|
|
|
|
export const insertConversationSchema = createInsertSchema(conversations).omit({
|
|
id: true,
|
|
createdAt: true,
|
|
});
|
|
|
|
export const insertMessageSchema = createInsertSchema(messages).omit({
|
|
id: true,
|
|
createdAt: true,
|
|
});
|
|
|
|
export type Conversation = typeof conversations.$inferSelect;
|
|
export type InsertConversation = z.infer<typeof insertConversationSchema>;
|
|
export type Message = typeof messages.$inferSelect;
|
|
export type InsertMessage = z.infer<typeof insertMessageSchema>;
|
|
|