
- 新增 save_dataset 函数,用于将 TinyDB 中的数据集保存为单独的 JSON 文件 - 更新 db/__init__.py,添加 get_dataset_tinydb 函数的引用 - 修改 db/dataset_store.py,实现 save_dataset 函数并添加相关逻辑
12 lines
354 B
Python
12 lines
354 B
Python
from .init_db import load_sqlite_engine, initialize_sqlite_db
|
|
from .prompt_store import get_prompt_tinydb, initialize_prompt_store
|
|
from .dataset_store import get_all_dataset, save_dataset
|
|
|
|
__all__ = [
|
|
"load_sqlite_engine",
|
|
"initialize_sqlite_db",
|
|
"get_prompt_tinydb",
|
|
"initialize_prompt_store",
|
|
"get_all_dataset",
|
|
"save_dataset"
|
|
] |