mirror of
https://github.com/CherryHQ/cherry-studio.git
synced 2025-12-29 23:12:38 +08:00
0d2dc2c257
- Replace custom migration system with modern Drizzle ORM implementation - Add drizzle-orm and drizzle-kit dependencies for type-safe database operations - Refactor BaseService to use Drizzle client with full type safety - Create schema definitions in /database/schema/ using Drizzle patterns - Remove legacy migration files, queries, and migrator classes - Add comprehensive documentation for new Drizzle-based architecture - Maintain backward compatibility in service layer APIs - Simplify database operations with modern ORM patterns This migration eliminates custom SQL generation in favor of a proven, type-safe ORM solution that provides better developer experience and maintainability.
80 lines
4.5 KiB
TypeScript
80 lines
4.5 KiB
TypeScript
/**
|
|
* Drizzle ORM schema for sessions and session_logs tables
|
|
*/
|
|
|
|
import { foreignKey, index, integer, sqliteTable, text } from 'drizzle-orm/sqlite-core'
|
|
|
|
export const sessionsTable = sqliteTable('sessions', {
|
|
id: text('id').primaryKey(),
|
|
name: text('name'), // Session name
|
|
main_agent_id: text('main_agent_id').notNull(), // Primary agent ID for the session
|
|
sub_agent_ids: text('sub_agent_ids'), // JSON array of sub-agent IDs involved in the session
|
|
user_goal: text('user_goal'), // Initial user goal for the session
|
|
status: text('status').notNull().default('idle'), // 'idle', 'running', 'completed', 'failed', 'stopped'
|
|
external_session_id: text('external_session_id'), // Agent session for external agent management/tracking
|
|
// AgentConfiguration fields that can override agent defaults
|
|
model: text('model'), // Main model ID (inherits from agent if null)
|
|
plan_model: text('plan_model'), // Optional plan/thinking model ID
|
|
small_model: text('small_model'), // Optional small/fast model ID
|
|
built_in_tools: text('built_in_tools'), // JSON array of built-in tool IDs
|
|
mcps: text('mcps'), // JSON array of MCP tool IDs
|
|
knowledges: text('knowledges'), // JSON array of enabled knowledge base IDs
|
|
configuration: text('configuration'), // JSON, extensible settings like temperature, top_p
|
|
accessible_paths: text('accessible_paths'), // JSON array of directory paths the agent can access
|
|
permission_mode: text('permission_mode').default('readOnly'), // 'readOnly', 'acceptEdits', 'bypassPermissions'
|
|
max_steps: integer('max_steps').default(10), // Maximum number of steps the agent can take
|
|
created_at: text('created_at').notNull(),
|
|
updated_at: text('updated_at').notNull()
|
|
})
|
|
|
|
// Indexes for sessions table
|
|
export const sessionsNameIdx = index('idx_sessions_name').on(sessionsTable.name)
|
|
export const sessionsStatusIdx = index('idx_sessions_status').on(sessionsTable.status)
|
|
export const sessionsCreatedAtIdx = index('idx_sessions_created_at').on(sessionsTable.created_at)
|
|
export const sessionsExternalSessionIdIdx = index('idx_sessions_external_session_id').on(
|
|
sessionsTable.external_session_id
|
|
)
|
|
export const sessionsMainAgentIdIdx = index('idx_sessions_main_agent_id').on(sessionsTable.main_agent_id)
|
|
export const sessionsModelIdx = index('idx_sessions_model').on(sessionsTable.model)
|
|
export const sessionsPlanModelIdx = index('idx_sessions_plan_model').on(sessionsTable.plan_model)
|
|
export const sessionsSmallModelIdx = index('idx_sessions_small_model').on(sessionsTable.small_model)
|
|
|
|
export const sessionMessagesTable = sqliteTable('session_messages', {
|
|
id: integer('id').primaryKey({ autoIncrement: true }),
|
|
session_id: text('session_id').notNull(),
|
|
parent_id: integer('parent_id'), // Foreign Key to session_logs.id, nullable for tree structure
|
|
role: text('role').notNull(), // 'user', 'agent', 'system', 'tool'
|
|
type: text('type').notNull(), // 'message', 'thought', 'action', 'observation', etc.
|
|
content: text('content').notNull(), // JSON structured data
|
|
metadata: text('metadata'), // JSON metadata (optional)
|
|
created_at: text('created_at').notNull(),
|
|
updated_at: text('updated_at').notNull()
|
|
})
|
|
|
|
// Indexes for session_messages table
|
|
export const sessionMessagesSessionIdIdx = index('idx_session_messages_session_id').on(sessionMessagesTable.session_id)
|
|
export const sessionMessagesParentIdIdx = index('idx_session_messages_parent_id').on(sessionMessagesTable.parent_id)
|
|
export const sessionMessagesRoleIdx = index('idx_session_messages_role').on(sessionMessagesTable.role)
|
|
export const sessionMessagesTypeIdx = index('idx_session_messages_type').on(sessionMessagesTable.type)
|
|
export const sessionMessagesCreatedAtIdx = index('idx_session_messages_created_at').on(sessionMessagesTable.created_at)
|
|
export const sessionMessagesUpdatedAtIdx = index('idx_session_messages_updated_at').on(sessionMessagesTable.updated_at)
|
|
|
|
// Foreign keys for session_messages table
|
|
export const sessionMessagesFkSession = foreignKey({
|
|
columns: [sessionMessagesTable.session_id],
|
|
foreignColumns: [sessionsTable.id],
|
|
name: 'fk_session_messages_session_id'
|
|
}).onDelete('cascade')
|
|
|
|
export const sessionMessagesFkParent = foreignKey({
|
|
columns: [sessionMessagesTable.parent_id],
|
|
foreignColumns: [sessionMessagesTable.id],
|
|
name: 'fk_session_messages_parent_id'
|
|
})
|
|
|
|
export type SessionRow = typeof sessionsTable.$inferSelect
|
|
export type InsertSessionRow = typeof sessionsTable.$inferInsert
|
|
|
|
export type SessionMessageRow = typeof sessionMessagesTable.$inferSelect
|
|
export type InsertSessionMessageRow = typeof sessionMessagesTable.$inferInsert
|