Spaces:
Running
Running
File size: 3,168 Bytes
27127dd |
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 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 |
import { pgTable, text, serial, integer, boolean, timestamp, json } from "drizzle-orm/pg-core";
import { createInsertSchema } from "drizzle-zod";
import { z } from "zod";
export const users = pgTable("users", {
id: serial("id").primaryKey(),
username: text("username").notNull().unique(),
password: text("password").notNull(),
fullName: text("full_name"),
location: text("location"),
interests: text("interests").array(),
profession: text("profession"),
pets: text("pets"),
additionalInfo: text("additional_info"),
systemContext: text("system_context"),
});
export const insertUserSchema = createInsertSchema(users).pick({
username: true,
password: true,
});
export const updateUserProfileSchema = createInsertSchema(users)
.pick({
fullName: true,
location: true,
interests: true,
profession: true,
pets: true,
additionalInfo: true,
systemContext: true,
})
.partial();
export type InsertUser = z.infer<typeof insertUserSchema>;
export type User = typeof users.$inferSelect;
// Message model for chat
export const messages = pgTable("messages", {
id: serial("id").primaryKey(),
content: text("content").notNull(),
role: text("role").notNull(), // 'user' or 'assistant'
conversationId: text("conversation_id").notNull(),
createdAt: timestamp("created_at").defaultNow().notNull(),
});
export const insertMessageSchema = createInsertSchema(messages).pick({
content: true,
role: true,
conversationId: true,
});
export type InsertMessage = z.infer<typeof insertMessageSchema>;
export type Message = typeof messages.$inferSelect;
// Define the personality types
export const personalityTypeSchema = z.enum([
"default",
"professional",
"friendly",
"expert",
"poetic",
"concise"
]);
export type PersonalityType = z.infer<typeof personalityTypeSchema>;
// Conversation model to group messages
export const conversations = pgTable("conversations", {
id: text("id").primaryKey(),
title: text("title").notNull(),
createdAt: timestamp("created_at").defaultNow().notNull(),
personality: text("personality").default("default").notNull(),
userId: integer("user_id").references(() => users.id),
});
export const insertConversationSchema = createInsertSchema(conversations).pick({
id: true,
title: true,
personality: true,
userId: true,
});
export type InsertConversation = z.infer<typeof insertConversationSchema>;
export type Conversation = typeof conversations.$inferSelect;
// Message role schema for validation
export const messageRoleSchema = z.enum(["user", "assistant", "system"]);
export type MessageRole = z.infer<typeof messageRoleSchema>;
// Message schema for API
export const messageSchema = z.object({
content: z.string(),
role: messageRoleSchema,
});
export type MessageType = z.infer<typeof messageSchema>;
// Conversation message format for OpenAI
export const conversationSchema = z.object({
messages: z.array(messageSchema),
personality: personalityTypeSchema.optional().default("default"),
conversationId: z.string().optional(),
userId: z.number().optional()
});
export type ConversationType = z.infer<typeof conversationSchema>; |