
- 新增 dataset_store.py 文件,实现数据集的存储和读取功能 - 添加 get_all_dataset 函数,用于获取所有数据集 - 使用 tinydb 和 json 进行数据持久化 - 在项目根目录下创建 workdir/dataset 目录用于存储数据集文件
11 lines
318 B
Python
11 lines
318 B
Python
from .init_db import get_sqlite_engine, initialize_sqlite_db
|
|
from .prompt_store import get_prompt_tinydb, initialize_prompt_store
|
|
from .dataset_store import get_all_dataset
|
|
|
|
__all__ = [
|
|
"get_sqlite_engine",
|
|
"initialize_sqlite_db",
|
|
"get_prompt_tinydb",
|
|
"initialize_prompt_store",
|
|
"get_all_dataset"
|
|
] |