db/prompt_store.py

@@ -1,4 +1,5 @@
 import os
+import json
 import sys
 from typing import Optional
 from pathlib import Path
@@ -41,15 +42,13 @@ def initialize_prompt_store(db: TinyDB) -> None:
     Args:
         db (TinyDB): TinyDB数据库实例
     """
-    # 检查数据库是否为空
-    if not db.all():  # 如果数据库中没有数据
-        db.insert(promptTempleta(name="default",
-                                 description="默认提示词模板",
-                                 content="""项目名为:{ project_name }
+    db.insert(promptTempleta(name="default",
+                             description="默认提示词模板",
+                             content="""项目名为:{ project_name }
 请依据以下该项目官方文档的部分内容，创造合适的对话数据集用于微调一个了解该项目的小模型的语料，要求兼顾文档中间尽可能多的信息点，使用中文
 文档节选：{ content }
-    按照如下json格式返回：{ json }""").model_dump())
-    # 如果数据库中已有数据，则跳过插入
+	按照如下json格式返回：{ json }""").model_dump())
+    # TinyDB不需要显式创建表结构，首次使用时自动创建
 
 
 if __name__ == "__main__":

frontend/setting_page.py

@@ -1,7 +1,9 @@
 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():

global_var.py

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