refactor(schema): 重构数据集和文档类的命名

- 将 dataset、dataset_item 和 doc 类的首字母大写,以符合 Python 类命名惯例
- 更新相关模块中的导入和引用,以适应新的类名
- 此更改不影响功能,仅提高了代码的一致性和可读性
This commit is contained in:
carry
2025-04-20 01:46:15 +08:00
parent 9236f49b36
commit 4c9caff668
6 changed files with 16 additions and 16 deletions

View File

@@ -1,19 +1,19 @@
from typing import List
from schema.dataset import dataset, dataset_item, Q_A
from schema.dataset import Dataset, DatasetItem, Q_A
import json
def convert_json_to_dataset(json_data: List[dict]) -> dataset:
def convert_json_to_dataset(json_data: List[dict]) -> Dataset:
# 将JSON数据转换为dataset格式
dataset_items = []
item_id = 1 # 自增ID计数器
for item in json_data:
qa = Q_A(question=item["question"], answer=item["answer"])
dataset_item_obj = dataset_item(id=item_id, message=[qa])
dataset_item_obj = DatasetItem(id=item_id, message=[qa])
dataset_items.append(dataset_item_obj)
item_id += 1 # ID自增
# 创建dataset对象
result_dataset = dataset(
result_dataset = Dataset(
name="Converted Dataset",
model_id=None,
description="Dataset converted from JSON",

View File

@@ -4,7 +4,7 @@ from pathlib import Path
# 添加项目根目录到sys.path
sys.path.append(str(Path(__file__).resolve().parent.parent))
from schema import doc
from schema import Doc
def scan_docs_directory(workdir: str):
docs_dir = os.path.join(workdir, "docs")
@@ -21,7 +21,7 @@ def scan_docs_directory(workdir: str):
for file in files:
if file.endswith(".md"):
markdown_files.append(os.path.join(root, file))
to_return.append(doc(name=doc_name, path=doc_path, markdown_files=markdown_files))
to_return.append(Doc(name=doc_name, path=doc_path, markdown_files=markdown_files))
return to_return

View File

@@ -61,7 +61,7 @@ if __name__ == "__main__":
from pathlib import Path
# 添加项目根目录到sys.path
sys.path.append(str(Path(__file__).resolve().parent.parent))
from schema import dataset
from schema import Dataset
print("示例 JSON:")
print(generate_example_json(dataset))
print(generate_example_json(Dataset))

View File

@@ -96,7 +96,7 @@ if __name__ == "__main__":
from json_example import generate_example_json
from sqlmodel import Session, select
from global_var import get_sql_engine, init_global_var
from schema import dataset_item
from schema import DatasetItem
init_global_var("workdir")
api_state = "1 deepseek-chat"
@@ -105,7 +105,7 @@ if __name__ == "__main__":
llm_request = LLMRequest(
prompt="测试,随便说点什么",
api_provider=api_provider,
format=generate_example_json(dataset_item)
format=generate_example_json(DatasetItem)
)
# # 单次调用示例