This commit is contained in:
Zhang Yulong
2025-11-28 18:29:16 +08:00
committed by GitHub
parent a535050b11
commit 5b49142988
4 changed files with 561 additions and 29 deletions

View File

@@ -52,6 +52,7 @@ class RequestFuncInput:
language: Optional[str] = None
debug: bool = False
response_format: Optional[dict] = None
random_flag: bool = False
@dataclass
@@ -103,6 +104,13 @@ async def async_request_eb_openai_chat_completions(
# 超参由yaml传入
payload.update(request_func_input.hyper_parameters)
# 随机输入开关
if request_func_input.random_flag:
payload["max_tokens"] = request_func_input.output_len
metadata = payload.get("metadata", {})
metadata["min_tokens"] = request_func_input.output_len
payload["metadata"] = metadata
if request_func_input.ignore_eos:
payload["ignore_eos"] = request_func_input.ignore_eos