mirror of
https://github.com/CherryHQ/cherry-studio.git
synced 2026-01-01 09:49:03 +08:00
e4b5e70c34
Use dayjs().valueOf() instead of dayjs().unix() to get timestamps in milliseconds for consistency with the updated schema comment
50 lines
2.1 KiB
TypeScript
50 lines
2.1 KiB
TypeScript
import type { OcrProviderCapabilityRecord, OcrProviderConfig } from '@types'
|
|
import { index, sqliteTable, text } from 'drizzle-orm/sqlite-core'
|
|
|
|
import { createUpdateTimestamps } from '../columnHelpers'
|
|
|
|
export const ocrProviderTable = sqliteTable(
|
|
'ocr_provider',
|
|
{
|
|
/**
|
|
* Unique identifier for the provider.
|
|
* For built-in providers, it's 'tesseract', 'system', etc.
|
|
* For custom providers, it can be any unique string (we typically use UUID v4).
|
|
* As the primary key, it ensures the uniqueness of each provider.
|
|
*/
|
|
id: text('id').primaryKey(),
|
|
|
|
/**
|
|
* Display name of the provider, e.g., "Tesseract OCR".
|
|
* For built-in providers, this value is used internally and is not exposed to users; the display name shown in the UI is locale-based by i18n.
|
|
* Cannot be null.
|
|
*/
|
|
name: text('name').notNull(),
|
|
|
|
/**
|
|
* Object describing the provider's capabilities, e.g., { image: true }.
|
|
* Stored as JSON in a text column. Drizzle's `mode: 'json'` handles
|
|
* serialization and deserialization automatically. `$type` provides strong typing.
|
|
* Cannot be null; should store an empty object `{}` even if no specific capabilities.
|
|
*/
|
|
capabilities: text('capabilities', { mode: 'json' }).$type<OcrProviderCapabilityRecord>().notNull(),
|
|
|
|
/**
|
|
* Provider-specific configuration. This is a polymorphic field, its structure varies by provider type.
|
|
* For example, Tesseract's configuration is entirely different from PaddleOCR's.
|
|
* Storing it as JSON is the most flexible approach to accommodate any configuration structure.
|
|
* Since this is a polymorphic field, both frontend and backend must validate
|
|
* that the structure matches the expected schema for the corresponding provider type
|
|
* before saving.
|
|
*/
|
|
config: text('config', { mode: 'json' }).$type<OcrProviderConfig>().notNull(),
|
|
|
|
/** Unix timestamp (milliseconds since epoch) for creation and last update. */
|
|
...createUpdateTimestamps
|
|
},
|
|
(t) => [index('name').on(t.name)]
|
|
)
|
|
|
|
export type OcrProviderInsert = typeof ocrProviderTable.$inferInsert
|
|
export type OcrProviderSelect = typeof ocrProviderTable.$inferSelect
|