fix(db): 修复 prompt_store 初始化逻辑
- 在插入默认模板之前检查数据库是否为空,如果数据库已有数据,则跳过插入默认模板
This commit is contained in:
parent
9741ce6b92
commit
5c7ced30df
@ -41,13 +41,15 @@ def initialize_prompt_store(db: TinyDB) -> None:
|
||||
Args:
|
||||
db (TinyDB): TinyDB数据库实例
|
||||
"""
|
||||
db.insert(promptTempleta(name="default",
|
||||
description="默认提示词模板",
|
||||
content="""项目名为:{ project_name }
|
||||
# 检查数据库是否为空
|
||||
if not db.all(): # 如果数据库中没有数据
|
||||
db.insert(promptTempleta(name="default",
|
||||
description="默认提示词模板",
|
||||
content="""项目名为:{ project_name }
|
||||
请依据以下该项目官方文档的部分内容,创造合适的对话数据集用于微调一个了解该项目的小模型的语料,要求兼顾文档中间尽可能多的信息点,使用中文
|
||||
文档节选:{ content }
|
||||
按照如下json格式返回:{ json }""").model_dump())
|
||||
# TinyDB不需要显式创建表结构,首次使用时自动创建
|
||||
按照如下json格式返回:{ json }""").model_dump())
|
||||
# 如果数据库中已有数据,则跳过插入
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
Loading…
x
Reference in New Issue
Block a user