carry/gzhu-biyesheji
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(db): 为 promptStore 添加空数据库初始化逻辑

Browse Source 
- 在 initialize_prompt_store 函数中增加空数据库检查和初始化逻辑
- 为默认模板添加 id 字段，设置为 0
This commit is contained in:
carry 2025-04-09 10:28:31 +08:00
parent 5c7ced30df
commit daddcd34da
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
db/prompt_store.py
View File

@ -43,9 +43,11 @@ def initialize_prompt_store(db: TinyDB) -> None:
""" """
# 检查数据库是否为空 # 检查数据库是否为空
if not db.all(): # 如果数据库中没有数据 if not db.all(): # 如果数据库中没有数据
db.insert(promptTempleta(name="default", db.insert(promptTempleta(
description="默认提示词模板", id=0,
content="""项目名为:{ project_name } name="default",
description="默认提示词模板",
content="""项目名为:{ project_name }
请依据以下该项目官方文档的部分内容，创造合适的对话数据集用于微调一个了解该项目的小模型的语料，要求兼顾文档中间尽可能多的信息点，使用中文 请依据以下该项目官方文档的部分内容，创造合适的对话数据集用于微调一个了解该项目的小模型的语料，要求兼顾文档中间尽可能多的信息点，使用中文
文档节选：{ content } 文档节选：{ content }
按照如下json格式返回：{ json }""").model_dump()) 按照如下json格式返回：{ json }""").model_dump())