407 lines
14 KiB
Plaintext
407 lines
14 KiB
Plaintext
|
generator client {
|
|||
|
|
provider = "prisma-client-js"
|
|||
|
|
binaryTargets = ["native", "darwin-arm64", "darwin", "windows", "debian-openssl-3.0.x", "linux-arm64-openssl-3.0.x", "debian-openssl-1.1.x"]
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
datasource db {
|
|||
|
|
provider = "sqlite"
|
|||
|
|
url = env("DATABASE_URL")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
model Projects {
|
|||
|
|
id String @id @default(nanoid(12))
|
|||
|
|
name String
|
|||
|
|
description String
|
|||
|
|
globalPrompt String @default("")
|
|||
|
|
questionPrompt String @default("")
|
|||
|
|
answerPrompt String @default("")
|
|||
|
|
labelPrompt String @default("")
|
|||
|
|
domainTreePrompt String @default("")
|
|||
|
|
cleanPrompt String @default("")
|
|||
|
|
defaultModelConfigId String?
|
|||
|
|
test String @default("")
|
|||
|
|
createAt DateTime @default(now())
|
|||
|
|
updateAt DateTime @updatedAt
|
|||
|
|
Questions Questions[]
|
|||
|
|
Datasets Datasets[]
|
|||
|
|
DatasetConversations DatasetConversations[]
|
|||
|
|
Chunks Chunks[]
|
|||
|
|
ModelConfig ModelConfig[]
|
|||
|
|
UploadFiles UploadFiles[]
|
|||
|
|
Tags Tags[]
|
|||
|
|
Task Task[]
|
|||
|
|
GaPairs GaPairs[]
|
|||
|
|
CustomPrompts CustomPrompts[]
|
|||
|
|
Images Images[]
|
|||
|
|
ImageDatasets ImageDatasets[]
|
|||
|
|
QuestionTemplates QuestionTemplates[]
|
|||
|
|
EvalDatasets EvalDatasets[]
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
model UploadFiles {
|
|||
|
|
id String @id @default(nanoid())
|
|||
|
|
project Projects @relation(fields: [projectId], references: [id], onDelete: Cascade)
|
|||
|
|
projectId String
|
|||
|
|
fileName String
|
|||
|
|
fileExt String
|
|||
|
|
path String
|
|||
|
|
size Int
|
|||
|
|
md5 String
|
|||
|
|
createAt DateTime @default(now())
|
|||
|
|
updateAt DateTime @updatedAt
|
|||
|
|
GaPairs GaPairs[]
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
model Chunks {
|
|||
|
|
id String @id @default(nanoid())
|
|||
|
|
name String
|
|||
|
|
project Projects @relation(fields: [projectId], references: [id], onDelete: Cascade)
|
|||
|
|
projectId String
|
|||
|
|
fileId String
|
|||
|
|
fileName String
|
|||
|
|
content String
|
|||
|
|
summary String
|
|||
|
|
size Int
|
|||
|
|
createAt DateTime @default(now())
|
|||
|
|
updateAt DateTime @updatedAt
|
|||
|
|
Questions Questions[]
|
|||
|
|
EvalDatasets EvalDatasets[]
|
|||
|
|
|
|||
|
|
@@index([projectId])
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
model Tags {
|
|||
|
|
id String @id @default(nanoid())
|
|||
|
|
label String
|
|||
|
|
project Projects @relation(fields: [projectId], references: [id], onDelete: Cascade)
|
|||
|
|
projectId String
|
|||
|
|
parentId String?
|
|||
|
|
parent Tags? @relation("Tags", fields: [parentId], references: [id])
|
|||
|
|
children Tags[] @relation("Tags")
|
|||
|
|
|
|||
|
|
@@index([projectId, label])
|
|||
|
|
@@index([projectId, parentId])
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
model Questions {
|
|||
|
|
id String @id @default(nanoid())
|
|||
|
|
project Projects @relation(fields: [projectId], references: [id], onDelete: Cascade)
|
|||
|
|
projectId String
|
|||
|
|
chunk Chunks @relation(fields: [chunkId], references: [id])
|
|||
|
|
chunkId String
|
|||
|
|
gaPair GaPairs? @relation(fields: [gaPairId], references: [id])
|
|||
|
|
gaPairId String? // Optional: links question to the GA pair that generated it
|
|||
|
|
question String
|
|||
|
|
label String
|
|||
|
|
answered Boolean @default(false)
|
|||
|
|
imageId String? // Optional: for image-based questions
|
|||
|
|
imageName String? // Optional: for image-based questions
|
|||
|
|
templateId String? // Optional: links to ImageQuestionTemplates
|
|||
|
|
createAt DateTime @default(now())
|
|||
|
|
updateAt DateTime @updatedAt
|
|||
|
|
|
|||
|
|
@@index([projectId])
|
|||
|
|
@@index([imageId])
|
|||
|
|
@@index([templateId])
|
|||
|
|
@@index([projectId, label])
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
model Datasets {
|
|||
|
|
id String @id @default(nanoid())
|
|||
|
|
project Projects @relation(fields: [projectId], references: [id], onDelete: Cascade)
|
|||
|
|
projectId String
|
|||
|
|
questionId String
|
|||
|
|
question String
|
|||
|
|
answer String
|
|||
|
|
answerType String? @default("text") // 'text' | 'label' | 'custom_format'
|
|||
|
|
chunkName String
|
|||
|
|
chunkContent String
|
|||
|
|
model String
|
|||
|
|
questionLabel String
|
|||
|
|
cot String
|
|||
|
|
confirmed Boolean @default(false)
|
|||
|
|
score Float @default(0)
|
|||
|
|
aiEvaluation String @default("") // AI评估结论
|
|||
|
|
tags String @default("")
|
|||
|
|
note String @default("")
|
|||
|
|
other String @default("") // 存储其他字段的JSON字符串
|
|||
|
|
createAt DateTime @default(now())
|
|||
|
|
updateAt DateTime @updatedAt
|
|||
|
|
|
|||
|
|
@@index([projectId])
|
|||
|
|
@@index([projectId, confirmed, createAt, id], name: "idx_export_confirmed")
|
|||
|
|
@@index([projectId, createAt], name: "idx_project_createAt")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
model DatasetConversations {
|
|||
|
|
id String @id @default(nanoid())
|
|||
|
|
project Projects @relation(fields: [projectId], references: [id], onDelete: Cascade)
|
|||
|
|
projectId String
|
|||
|
|
questionId String // 第一个问题 Id（初始问题）
|
|||
|
|
question String // 第一个问题（初始问题）
|
|||
|
|
chunkId String // 基于哪个文本块生成
|
|||
|
|
model String
|
|||
|
|
questionLabel String
|
|||
|
|
score Float @default(0)
|
|||
|
|
aiEvaluation String @default("") // AI评估结论
|
|||
|
|
tags String @default("")
|
|||
|
|
note String @default("")
|
|||
|
|
scenario String // 对话场景（教学/咨询/讨论等）
|
|||
|
|
roleA String // 角色A设定
|
|||
|
|
roleB String // 角色B设定
|
|||
|
|
turnCount Int // 实际轮数
|
|||
|
|
maxTurns Int // 设置的最大轮数
|
|||
|
|
rawMessages String // JSON存储完整对话（和 ShareGPT 格式保持完全一致）
|
|||
|
|
confirmed Boolean @default(false)
|
|||
|
|
createAt DateTime @default(now())
|
|||
|
|
updateAt DateTime @updatedAt
|
|||
|
|
|
|||
|
|
@@index([projectId])
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
model LlmProviders {
|
|||
|
|
id String @id
|
|||
|
|
name String
|
|||
|
|
apiUrl String
|
|||
|
|
createAt DateTime @default(now())
|
|||
|
|
updateAt DateTime @updatedAt
|
|||
|
|
LlmModels LlmModels[]
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
model LlmModels {
|
|||
|
|
id String @id @default(nanoid())
|
|||
|
|
modelId String
|
|||
|
|
modelName String
|
|||
|
|
provider LlmProviders @relation(fields: [providerId], references: [id])
|
|||
|
|
providerId String
|
|||
|
|
createAt DateTime @default(now())
|
|||
|
|
updateAt DateTime @updatedAt
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
model ModelConfig {
|
|||
|
|
id String @id @default(nanoid())
|
|||
|
|
project Projects @relation(fields: [projectId], references: [id], onDelete: Cascade)
|
|||
|
|
projectId String
|
|||
|
|
providerId String
|
|||
|
|
providerName String
|
|||
|
|
endpoint String
|
|||
|
|
apiKey String
|
|||
|
|
modelId String
|
|||
|
|
modelName String
|
|||
|
|
type String
|
|||
|
|
temperature Float
|
|||
|
|
maxTokens Int
|
|||
|
|
topP Float
|
|||
|
|
topK Float
|
|||
|
|
status Int
|
|||
|
|
createAt DateTime @default(now())
|
|||
|
|
updateAt DateTime @updatedAt
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
model Task {
|
|||
|
|
id String @id @default(nanoid())
|
|||
|
|
project Projects @relation(fields: [projectId], references: [id], onDelete: Cascade)
|
|||
|
|
projectId String
|
|||
|
|
taskType String // 任务类型: text-processing, question-generation, answer-generation, data-distillation
|
|||
|
|
status Int // 任务状态: 0-处理中, 1-已完成, 2-失败, 3-已中断
|
|||
|
|
startTime DateTime @default(now())
|
|||
|
|
endTime DateTime?
|
|||
|
|
completedCount Int @default(0)
|
|||
|
|
totalCount Int @default(0)
|
|||
|
|
modelInfo String // JSON格式存储，包含使用的模型信息
|
|||
|
|
language String @default("zh-CN")
|
|||
|
|
detail String @default("") // 任务详情
|
|||
|
|
note String @default("") // 任务备注
|
|||
|
|
createAt DateTime @default(now())
|
|||
|
|
updateAt DateTime @updatedAt
|
|||
|
|
|
|||
|
|
@@index([projectId])
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
model CustomPrompts {
|
|||
|
|
id String @id @default(nanoid())
|
|||
|
|
project Projects @relation(fields: [projectId], references: [id], onDelete: Cascade)
|
|||
|
|
projectId String
|
|||
|
|
promptType String // 提示词类型，对应 lib/llm/prompts 下的文件名
|
|||
|
|
promptKey String // 提示词在模块中的键名，如 QUESTION_PROMPT, QUESTION_PROMPT_EN
|
|||
|
|
language String // 语言: zh-CN, en
|
|||
|
|
content String // 自定义的提示词内容
|
|||
|
|
isActive Boolean @default(true) // 是否启用
|
|||
|
|
createAt DateTime @default(now())
|
|||
|
|
updateAt DateTime @updatedAt
|
|||
|
|
|
|||
|
|
@@unique([projectId, promptType, promptKey, language])
|
|||
|
|
@@index([projectId, promptType])
|
|||
|
|
@@index([projectId, language])
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
model GaPairs {
|
|||
|
|
id String @id @default(nanoid())
|
|||
|
|
project Projects @relation(fields: [projectId], references: [id], onDelete: Cascade)
|
|||
|
|
projectId String
|
|||
|
|
uploadFile UploadFiles @relation(fields: [fileId], references: [id], onDelete: Cascade)
|
|||
|
|
fileId String
|
|||
|
|
pairNumber Int // 1-5, representing the 5 generated pairs
|
|||
|
|
genreTitle String // Genre name/title
|
|||
|
|
genreDesc String // Genre description
|
|||
|
|
audienceTitle String // Audience name/title
|
|||
|
|
audienceDesc String // Audience description
|
|||
|
|
isActive Boolean @default(true) // Whether this pair is active for use
|
|||
|
|
questions Questions[] // Questions generated by this GA pair
|
|||
|
|
createAt DateTime @default(now())
|
|||
|
|
updateAt DateTime @updatedAt
|
|||
|
|
|
|||
|
|
@@unique([fileId, pairNumber])
|
|||
|
|
@@index([projectId])
|
|||
|
|
@@index([fileId])
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
model Images {
|
|||
|
|
id String @id @default(nanoid())
|
|||
|
|
project Projects @relation(fields: [projectId], references: [id], onDelete: Cascade)
|
|||
|
|
projectId String
|
|||
|
|
imageName String
|
|||
|
|
path String // 图片存储路径
|
|||
|
|
size Int // 文件大小（字节）
|
|||
|
|
width Int? // 图片宽度
|
|||
|
|
height Int? // 图片高度
|
|||
|
|
createAt DateTime @default(now())
|
|||
|
|
updateAt DateTime @updatedAt
|
|||
|
|
ImageDatasets ImageDatasets[]
|
|||
|
|
|
|||
|
|
@@unique([projectId, imageName])
|
|||
|
|
@@index([projectId])
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
model ImageDatasets {
|
|||
|
|
id String @id @default(nanoid())
|
|||
|
|
project Projects @relation(fields: [projectId], references: [id], onDelete: Cascade)
|
|||
|
|
projectId String
|
|||
|
|
image Images @relation(fields: [imageId], references: [id], onDelete: Cascade)
|
|||
|
|
imageId String
|
|||
|
|
imageName String
|
|||
|
|
questionId String? // Optional: links to Questions table
|
|||
|
|
question String
|
|||
|
|
answer String // Stores all answer types: text, JSON array for labels, or custom format JSON
|
|||
|
|
answerType String @default("text") // 'text' | 'label' | 'custom_format'
|
|||
|
|
model String
|
|||
|
|
confirmed Boolean @default(false)
|
|||
|
|
score Float @default(0)
|
|||
|
|
tags String @default("")
|
|||
|
|
note String @default("")
|
|||
|
|
createAt DateTime @default(now())
|
|||
|
|
updateAt DateTime @updatedAt
|
|||
|
|
|
|||
|
|
@@index([projectId])
|
|||
|
|
@@index([imageId])
|
|||
|
|
@@index([questionId])
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
model QuestionTemplates {
|
|||
|
|
id String @id @default(nanoid())
|
|||
|
|
project Projects @relation(fields: [projectId], references: [id], onDelete: Cascade)
|
|||
|
|
projectId String
|
|||
|
|
question String // Question content
|
|||
|
|
sourceType String // 'image' | 'text' - data source type
|
|||
|
|
answerType String // 'text' | 'label' | 'custom_format'
|
|||
|
|
description String @default("") // Question description
|
|||
|
|
labels String @default("") // JSON array of label options (for answerType='label')
|
|||
|
|
customFormat String @default("") // Custom format definition (for answerType='custom_format')
|
|||
|
|
order Int @default(0) // Display order
|
|||
|
|
createAt DateTime @default(now())
|
|||
|
|
updateAt DateTime @updatedAt
|
|||
|
|
|
|||
|
|
@@index([projectId])
|
|||
|
|
@@index([projectId, sourceType])
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
model LlmUsageLogs {
|
|||
|
|
id String @id @default(nanoid())
|
|||
|
|
projectId String
|
|||
|
|
provider String // 提供商: openai, anthropic, google 等
|
|||
|
|
model String // 模型名称
|
|||
|
|
|
|||
|
|
// 核心指标
|
|||
|
|
inputTokens Int @default(0)
|
|||
|
|
outputTokens Int @default(0)
|
|||
|
|
totalTokens Int @default(0)
|
|||
|
|
latency Int @default(0) // 响应耗时（毫秒）
|
|||
|
|
|
|||
|
|
// 状态与追踪
|
|||
|
|
status String @default("SUCCESS") // 状态: "SUCCESS", "FAILED"
|
|||
|
|
errorMessage String? // 失败原因，status="FAILED" 时填写
|
|||
|
|
|
|||
|
|
// 时间维度
|
|||
|
|
createAt DateTime @default(now())
|
|||
|
|
dateString String // 格式 "YYYY-MM-DD"，用于快速按天聚合
|
|||
|
|
|
|||
|
|
@@index([projectId, dateString])
|
|||
|
|
@@index([dateString])
|
|||
|
|
@@index([provider])
|
|||
|
|
@@index([model])
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
model EvalDatasets {
|
|||
|
|
id String @id @default(nanoid())
|
|||
|
|
project Projects @relation(fields: [projectId], references: [id], onDelete: Cascade)
|
|||
|
|
projectId String
|
|||
|
|
|
|||
|
|
// 题目内容
|
|||
|
|
question String // 题目内容
|
|||
|
|
questionType String // 题型: true_false, single_choice, multiple_choice, short_answer, open_ended
|
|||
|
|
|
|||
|
|
// 上下文信息（关联到文本块）
|
|||
|
|
chunkId String? // 关联到 Chunks 表
|
|||
|
|
chunks Chunks? @relation(fields: [chunkId], references: [id])
|
|||
|
|
|
|||
|
|
// 选项（仅选择题使用）
|
|||
|
|
options String @default("") // JSON数组: ["选项A", "选项B", "选项C", "选项D"]
|
|||
|
|
|
|||
|
|
// 标准答案
|
|||
|
|
correctAnswer String // 标准答案
|
|||
|
|
|
|||
|
|
tags String @default("") // 标签，逗号分隔
|
|||
|
|
note String @default("") // 备注
|
|||
|
|
|
|||
|
|
// 时间戳
|
|||
|
|
createAt DateTime @default(now())
|
|||
|
|
updateAt DateTime @updatedAt
|
|||
|
|
|
|||
|
|
// 关联评估结果
|
|||
|
|
EvalResults EvalResults[]
|
|||
|
|
|
|||
|
|
@@index([projectId])
|
|||
|
|
@@index([projectId, questionType])
|
|||
|
|
@@index([chunkId])
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
model EvalResults {
|
|||
|
|
id String @id @default(nanoid())
|
|||
|
|
projectId String
|
|||
|
|
taskId String // 关联到 Task 表
|
|||
|
|
|
|||
|
|
// 关联评估题目
|
|||
|
|
evalDataset EvalDatasets @relation(fields: [evalDatasetId], references: [id], onDelete: Cascade)
|
|||
|
|
evalDatasetId String
|
|||
|
|
|
|||
|
|
// 评估结果
|
|||
|
|
modelAnswer String // 模型的回答
|
|||
|
|
score Float @default(0) // 得分 (0-1 之间)
|
|||
|
|
isCorrect Boolean @default(false) // 是否正确（用于客观题）
|
|||
|
|
judgeResponse String @default("") // LLM 评分的响应（用于主观题）
|
|||
|
|
|
|||
|
|
// 答题详情
|
|||
|
|
duration Int @default(0) // 答题耗时（毫秒）
|
|||
|
|
status Int @default(0) // 答题状态：0-成功, 1-输出不符合规范, 2-LLM调用报错
|
|||
|
|
errorMessage String @default("") // 答题报错信息
|
|||
|
|
|
|||
|
|
// 时间戳
|
|||
|
|
createAt DateTime @default(now())
|
|||
|
|
updateAt DateTime @updatedAt
|
|||
|
|
|
|||
|
|
@@unique([taskId, evalDatasetId]) // 每个任务对每道题只能有一个结果
|
|||
|
|
@@index([projectId])
|
|||
|
|
@@index([taskId])
|
|||
|
|
@@index([evalDatasetId])
|
|||
|
|
}
|