diff --git a/frontend/dataset_generate_page.py b/frontend/dataset_generate_page.py index 1765a59..7d21123 100644 --- a/frontend/dataset_generate_page.py +++ b/frontend/dataset_generate_page.py @@ -6,7 +6,7 @@ from sqlmodel import Session, select sys.path.append(str(Path(__file__).resolve().parent.parent)) from schema import APIProvider, LLMRequest, LLMResponse, DatasetItem -from tools import call_openai_api, process_markdown_file, generate_example_json +from tools import call_openai_api, process_markdown_file, generate_json_example from global_var import get_docs, get_prompt_store, get_sql_engine def dataset_generate_page(): @@ -128,7 +128,7 @@ def dataset_generate_page(): for document_slice in document_slice_list: request = LLMRequest(api_provider=api_provider, prompt=prompt.format(document_slice=document_slice), - format=generate_example_json(DatasetItem)) + format=generate_json_example(DatasetItem)) call_openai_api(request, rounds) diff --git a/tools/__init__.py b/tools/__init__.py index b384ee1..6eeeaf3 100644 --- a/tools/__init__.py +++ b/tools/__init__.py @@ -1,5 +1,5 @@ from .parse_markdown import * from .document import * -from .json_example import generate_example_json +from .json_example import generate_json_example from .port import * from .reasoning import call_openai_api \ No newline at end of file diff --git a/tools/json_example.py b/tools/json_example.py index 8ebc68b..273b725 100644 --- a/tools/json_example.py +++ b/tools/json_example.py @@ -3,7 +3,7 @@ from typing import Any, Dict, List, Optional, Union, get_args, get_origin import json from datetime import datetime, date -def generate_example_json(model: type[BaseModel], include_optional: bool = False,list_length = 2) -> str: +def generate_json_example(model: type[BaseModel], include_optional: bool = False,list_length = 2) -> str: """ 根据 Pydantic V2 模型生成示例 JSON 数据结构。 """ @@ -37,14 +37,14 @@ def generate_example_json(model: type[BaseModel], include_optional: bool = False elif field_type is date: return date.today().isoformat() elif isinstance(field_type, type) and issubclass(field_type, BaseModel): - return json.loads(generate_example_json(field_type, include_optional)) + return json.loads(generate_json_example(field_type, include_optional)) else: # 处理直接类型注解(非泛型) if field_type is type(None): return None try: if issubclass(field_type, BaseModel): - return json.loads(generate_example_json(field_type, include_optional)) + return json.loads(generate_json_example(field_type, include_optional)) except TypeError: pass return "unknown" @@ -64,4 +64,4 @@ if __name__ == "__main__": from schema import Dataset print("示例 JSON:") - print(generate_example_json(Dataset)) + print(generate_json_example(Dataset)) diff --git a/tools/reasoning.py b/tools/reasoning.py index a9fde0e..ee632be 100644 --- a/tools/reasoning.py +++ b/tools/reasoning.py @@ -93,7 +93,7 @@ async def call_openai_api(llm_request: LLMRequest, rounds: int = 1, llm_paramete return llm_request if __name__ == "__main__": - from json_example import generate_example_json + from json_example import generate_json_example from sqlmodel import Session, select from global_var import get_sql_engine, init_global_var from schema import DatasetItem @@ -105,7 +105,7 @@ if __name__ == "__main__": llm_request = LLMRequest( prompt="测试,随便说点什么", api_provider=api_provider, - format=generate_example_json(DatasetItem) + format=generate_json_example(DatasetItem) ) # # 单次调用示例