Northern_Lights/cherry-studio
1
0
Fork 0
mirror of https://github.com/CherryHQ/cherry-studio.git synced 2025-12-27 04:31:27 +08:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

refactor(types): convert AgentSessionMessageEntity interface to zod schema

Browse Source 
Replace interface with zod schema for better type safety and validation
This commit is contained in:
icarus 2025-09-18 13:15:58 +08:00
parent 7a4952f773
commit 5c578c191b
1 changed files with 13 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
src/renderer/src/types/agent.ts
View File

@ -2,7 +2,7 @@
* Database entity types for Agent, Session, and SessionMessage
* Shared between main and renderer processes
*/
import { ModelMessage, TextStreamPart, UIMessageChunk } from 'ai'
import { ModelMessage, modelMessageSchema, TextStreamPart, UIMessageChunk } from 'ai'
import { z } from 'zod'
export const PermissionModeSchema = z.enum(['default', 'acceptEdits', 'bypassPermissions', 'plan'])
@ -91,15 +91,18 @@ export const AgentSessionEntitySchema = AgentBaseSchema.extend({
export type AgentSessionEntity = z.infer<typeof AgentSessionEntitySchema>
// AgentSessionMessageEntity representing a message within a session
export interface AgentSessionMessageEntity {
id: number // Auto-increment primary key
session_id: string // Reference to session
role: ModelMessage['role'] // 'assistant' | 'user' | 'system' | 'tool'
content: ModelMessage
metadata?: Record<string, any> // Additional metadata (optional)
created_at: string // ISO timestamp
updated_at: string // ISO timestamp
}
export const AgentSessionMessageEntitySchema = z.object({
id: z.number(), // Auto-increment primary key
session_id: z.string(), // Reference to session
// manual defined. may not synced with ai sdk definition
role: z.enum(['assistant', 'user', 'system', 'tool']), // 'assistant' | 'user' | 'system' | 'tool'
content: modelMessageSchema,
metadata: z.record(z.string(), z.any()).optional(), // Additional metadata (optional)
created_at: z.iso.datetime(), // ISO timestamp
updated_at: z.iso.datetime() // ISO timestamp
})
export type AgentSessionMessageEntity = z.infer<typeof AgentSessionMessageEntitySchema>
// Structured content for session messages that preserves both AI SDK and raw data
export interface SessionMessageContent {