85 lines
2.7 KiB
TypeScript
85 lines
2.7 KiB
TypeScript
import { index, integer, sqliteTable, text, uniqueIndex } from "drizzle-orm/sqlite-core";
|
|
|
|
export const projects = sqliteTable("projects", {
|
|
archivedAt: text("archived_at"),
|
|
createdAt: text("created_at").notNull(),
|
|
description: text("description").notNull().default(""),
|
|
id: text("id").primaryKey(),
|
|
name: text("name").notNull().unique(),
|
|
status: text("status", { enum: ["active", "archived"] })
|
|
.notNull()
|
|
.default("active"),
|
|
updatedAt: text("updated_at").notNull(),
|
|
});
|
|
|
|
export const providers = sqliteTable("providers", {
|
|
apiKey: text("api_key").notNull(),
|
|
baseUrl: text("base_url").notNull(),
|
|
createdAt: text("created_at").notNull(),
|
|
id: text("id").primaryKey(),
|
|
name: text("name").notNull().unique(),
|
|
type: text("type", { enum: ["anthropic", "openai", "openai-compatible"] })
|
|
.notNull()
|
|
.default("openai-compatible"),
|
|
updatedAt: text("updated_at").notNull(),
|
|
});
|
|
|
|
export const models = sqliteTable(
|
|
"models",
|
|
{
|
|
capabilities: text("capabilities").notNull(),
|
|
contextLength: integer("context_length"),
|
|
createdAt: text("created_at").notNull(),
|
|
id: text("id").primaryKey(),
|
|
maxOutputTokens: integer("max_output_tokens"),
|
|
modelId: text("model_id").notNull(),
|
|
name: text("name").notNull(),
|
|
providerId: text("provider_id")
|
|
.notNull()
|
|
.references(() => providers.id),
|
|
updatedAt: text("updated_at").notNull(),
|
|
},
|
|
(table) => [
|
|
uniqueIndex("models_provider_id_model_id_unique").on(table.providerId, table.modelId),
|
|
index("models_provider_id_idx").on(table.providerId),
|
|
],
|
|
);
|
|
|
|
export const conversations = sqliteTable(
|
|
"conversations",
|
|
{
|
|
createdAt: text("created_at").notNull(),
|
|
id: text("id").primaryKey(),
|
|
modelId: text("model_id")
|
|
.notNull()
|
|
.references(() => models.id),
|
|
projectId: text("project_id")
|
|
.notNull()
|
|
.references(() => projects.id),
|
|
title: text("title").notNull().default("新会话"),
|
|
updatedAt: text("updated_at").notNull(),
|
|
},
|
|
(table) => [index("conversations_project_id_idx").on(table.projectId)],
|
|
);
|
|
|
|
export const messages = sqliteTable(
|
|
"messages",
|
|
{
|
|
content: text("content").notNull().default(""),
|
|
conversationId: text("conversation_id")
|
|
.notNull()
|
|
.references(() => conversations.id, { onDelete: "cascade" }),
|
|
createdAt: text("created_at").notNull(),
|
|
id: text("id").primaryKey(),
|
|
parts: text("parts"),
|
|
role: text("role", { enum: ["assistant", "system", "user"] }).notNull(),
|
|
},
|
|
(table) => [index("messages_conversation_id_idx").on(table.conversationId)],
|
|
);
|
|
|
|
export const schemaMigrations = sqliteTable("schema_migrations", {
|
|
appliedAt: text("applied_at").notNull(),
|
|
checksum: text("checksum").notNull(),
|
|
id: text("id").primaryKey(),
|
|
});
|