diff --git a/src/drizzle/schema.ts b/src/drizzle/schema.ts index f2f347d..585866a 100644 --- a/src/drizzle/schema.ts +++ b/src/drizzle/schema.ts @@ -131,6 +131,10 @@ export const messages = pgTable('messages', { toolResults: jsonb('tool_results').$type(), // 代码执行产生的图片(Base64 编码数组) images: jsonb('images').$type(), + // 用户上传的图片(Base64 编码数组) + uploadedImages: jsonb('uploaded_images').$type(), + // 用户上传的文档 + uploadedDocuments: jsonb('uploaded_documents').$type(), // Token 统计 inputTokens: integer('input_tokens').default(0), outputTokens: integer('output_tokens').default(0), @@ -243,6 +247,14 @@ export interface ToolResult { isError?: boolean; } +// 上传的文档数据(用于持久化存储) +export interface UploadedDocumentData { + name: string; + size: number; + type: string; + content: string; +} + // 导出类型 export type User = typeof users.$inferSelect; export type NewUser = typeof users.$inferInsert;