mirror of
https://github.com/CherryHQ/cherry-studio.git
synced 2026-01-02 10:29:02 +08:00
9c47937714
- Added `setAsActive` property to `CreateMessageDto` for controlling active node status in topics. - Updated `messageTable` and `topicTable` schemas to include foreign key constraints and improved handling of active node references. - Refactored message and topic service methods to utilize `.returning()` for better data retrieval after inserts and updates. - Implemented hard delete functionality for messages and topics, replacing soft delete logic to ensure data integrity.
66 lines
2.4 KiB
TypeScript
66 lines
2.4 KiB
TypeScript
import type { MessageData, MessageStats } from '@shared/data/types/message'
|
|
import type { AssistantMeta, ModelMeta } from '@shared/data/types/meta'
|
|
import { sql } from 'drizzle-orm'
|
|
import { check, foreignKey, index, integer, sqliteTable, text } from 'drizzle-orm/sqlite-core'
|
|
|
|
import { createUpdateDeleteTimestamps, uuidPrimaryKeyOrdered } from './columnHelpers'
|
|
import { topicTable } from './topic'
|
|
|
|
/**
|
|
* Message table - stores chat messages with tree structure
|
|
*
|
|
* Uses adjacency list pattern (parentId) for tree navigation.
|
|
* Block content is stored as JSON in the data field.
|
|
* searchableText is a generated column for FTS5 indexing.
|
|
*/
|
|
export const messageTable = sqliteTable(
|
|
'message',
|
|
{
|
|
id: uuidPrimaryKeyOrdered(),
|
|
// Adjacency list parent reference for tree structure
|
|
parentId: text(),
|
|
// FK to topic - CASCADE: delete messages when topic is deleted
|
|
topicId: text()
|
|
.notNull()
|
|
.references(() => topicTable.id, { onDelete: 'cascade' }),
|
|
// Message role: user, assistant, system
|
|
role: text().notNull(),
|
|
// Main content - contains blocks[], mentions, etc.
|
|
data: text({ mode: 'json' }).$type<MessageData>().notNull(),
|
|
// Searchable text extracted from data.blocks (populated by trigger, used for FTS5)
|
|
searchableText: text(),
|
|
|
|
// Final status: SUCCESS, ERROR, PAUSED
|
|
status: text().notNull(),
|
|
|
|
// Group ID for siblings (0 = normal branch)
|
|
siblingsGroupId: integer().default(0),
|
|
// FK to assistant
|
|
assistantId: text(),
|
|
// Preserved assistant info for display
|
|
assistantMeta: text({ mode: 'json' }).$type<AssistantMeta>(),
|
|
// Model identifier
|
|
modelId: text(),
|
|
// Preserved model info (provider, name)
|
|
modelMeta: text({ mode: 'json' }).$type<ModelMeta>(),
|
|
// Trace ID for tracking
|
|
|
|
traceId: text(),
|
|
// Statistics: token usage, performance metrics, etc.
|
|
stats: text({ mode: 'json' }).$type<MessageStats>(),
|
|
|
|
...createUpdateDeleteTimestamps
|
|
},
|
|
(t) => [
|
|
// Foreign keys
|
|
foreignKey({ columns: [t.parentId], foreignColumns: [t.id] }).onDelete('set null'),
|
|
// Indexes
|
|
index('message_parent_id_idx').on(t.parentId),
|
|
index('message_topic_created_idx').on(t.topicId, t.createdAt),
|
|
index('message_trace_id_idx').on(t.traceId),
|
|
// Check constraints for enum fields
|
|
check('message_role_check', sql`${t.role} IN ('user', 'assistant', 'system')`),
|
|
check('message_status_check', sql`${t.status} IN ('success', 'error', 'paused')`)
|
|
]
|
|
)
|