refactor(frontend/tools): 重命名生成示例 JSON 数据结构的函数

- 将 generate_example_json 函数重命名为 generate_json_example
- 更新相关文件中的函数调用和引用
- 此更改旨在使函数名称更具描述性和一致性
This commit is contained in:
carry 2025-04-20 16:11:36 +08:00
parent e7cf51d662
commit c28e4819d9
4 changed files with 9 additions and 9 deletions

View File

@ -6,7 +6,7 @@ from sqlmodel import Session, select
sys.path.append(str(Path(__file__).resolve().parent.parent)) sys.path.append(str(Path(__file__).resolve().parent.parent))
from schema import APIProvider, LLMRequest, LLMResponse, DatasetItem 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 from global_var import get_docs, get_prompt_store, get_sql_engine
def dataset_generate_page(): def dataset_generate_page():
@ -128,7 +128,7 @@ def dataset_generate_page():
for document_slice in document_slice_list: for document_slice in document_slice_list:
request = LLMRequest(api_provider=api_provider, request = LLMRequest(api_provider=api_provider,
prompt=prompt.format(document_slice=document_slice), prompt=prompt.format(document_slice=document_slice),
format=generate_example_json(DatasetItem)) format=generate_json_example(DatasetItem))
call_openai_api(request, rounds) call_openai_api(request, rounds)

View File

@ -1,5 +1,5 @@
from .parse_markdown import * from .parse_markdown import *
from .document import * from .document import *
from .json_example import generate_example_json from .json_example import generate_json_example
from .port import * from .port import *
from .reasoning import call_openai_api from .reasoning import call_openai_api

View File

@ -3,7 +3,7 @@ from typing import Any, Dict, List, Optional, Union, get_args, get_origin
import json import json
from datetime import datetime, date 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 数据结构 根据 Pydantic V2 模型生成示例 JSON 数据结构
""" """
@ -37,14 +37,14 @@ def generate_example_json(model: type[BaseModel], include_optional: bool = False
elif field_type is date: elif field_type is date:
return date.today().isoformat() return date.today().isoformat()
elif isinstance(field_type, type) and issubclass(field_type, BaseModel): 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: else:
# 处理直接类型注解(非泛型) # 处理直接类型注解(非泛型)
if field_type is type(None): if field_type is type(None):
return None return None
try: try:
if issubclass(field_type, BaseModel): 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: except TypeError:
pass pass
return "unknown" return "unknown"
@ -64,4 +64,4 @@ if __name__ == "__main__":
from schema import Dataset from schema import Dataset
print("示例 JSON:") print("示例 JSON:")
print(generate_example_json(Dataset)) print(generate_json_example(Dataset))

View File

@ -93,7 +93,7 @@ async def call_openai_api(llm_request: LLMRequest, rounds: int = 1, llm_paramete
return llm_request return llm_request
if __name__ == "__main__": if __name__ == "__main__":
from json_example import generate_example_json from json_example import generate_json_example
from sqlmodel import Session, select from sqlmodel import Session, select
from global_var import get_sql_engine, init_global_var from global_var import get_sql_engine, init_global_var
from schema import DatasetItem from schema import DatasetItem
@ -105,7 +105,7 @@ if __name__ == "__main__":
llm_request = LLMRequest( llm_request = LLMRequest(
prompt="测试,随便说点什么", prompt="测试,随便说点什么",
api_provider=api_provider, api_provider=api_provider,
format=generate_example_json(DatasetItem) format=generate_json_example(DatasetItem)
) )
# # 单次调用示例 # # 单次调用示例