Compare commits

...

2 Commits

Author SHA1 Message Date
carry
5c7ced30df fix(db): 修复 prompt_store 初始化逻辑
- 在插入默认模板之前检查数据库是否为空,如果数据库已有数据,则跳过插入默认模板
2025-04-09 10:26:14 +08:00
carry
9741ce6b92 refactor(db): 优化了代码,调整了import顺序,删除了无用变量 2025-04-09 10:19:57 +08:00
3 changed files with 8 additions and 10 deletions

View File

@ -1,5 +1,4 @@
import os
import json
import sys
from typing import Optional
from pathlib import Path
@ -42,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__":

View File

@ -1,9 +1,7 @@
import gradio as gr
from typing import List
from sqlmodel import Session, select
from db import get_sqlite_engine
from schema import APIProvider
import os
from global_var import sql_engine
def setting_page():

View File

@ -1,5 +1,4 @@
from db import get_sqlite_engine
from db import get_prompt_tinydb
from db import get_sqlite_engine,get_prompt_tinydb
prompt_store = get_prompt_tinydb("workdir")
sql_engine = get_sqlite_engine("workdir")