refactor(frontend/tools): 重命名生成示例 JSON 数据结构的函数
- 将 generate_example_json 函数重命名为 generate_json_example - 更新相关文件中的函数调用和引用 - 此更改旨在使函数名称更具描述性和一致性
This commit is contained in:
parent
e7cf51d662
commit
c28e4819d9
@ -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)
|
||||||
|
|
||||||
|
|
||||||
|
@ -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
|
@ -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))
|
||||||
|
@ -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)
|
||||||
)
|
)
|
||||||
|
|
||||||
# # 单次调用示例
|
# # 单次调用示例
|
||||||
|
Loading…
x
Reference in New Issue
Block a user