mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-06 00:57:33 +08:00
58 lines
1.8 KiB
Python
58 lines
1.8 KiB
Python
#!/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
# @author DDDivano
|
||
# encoding=utf-8 vi:ts=4:sw=4:expandtab:ft=python
|
||
|
||
import requests
|
||
from core import TEMPLATES, base_logger
|
||
|
||
|
||
def build_request_payload(template_name: str, case_data: dict) -> dict:
|
||
"""
|
||
基于模板构造请求 payload，按优先级依次合并：
|
||
template < payload 参数 < case_data，后者会覆盖前者的同名字段。
|
||
|
||
:param template_name: 模板变量名，例如 "TOKEN_LOGPROB"
|
||
:return: 构造后的完整请求 payload dict
|
||
"""
|
||
template = TEMPLATES[template_name]
|
||
print(template)
|
||
final_payload = template.copy()
|
||
final_payload.update(case_data)
|
||
|
||
return final_payload
|
||
|
||
|
||
def send_request(url, payload, timeout=600, stream=False):
|
||
"""
|
||
向指定URL发送POST请求，并返回响应结果。
|
||
|
||
Args:
|
||
url (str): 请求的目标URL。
|
||
payload (dict): 请求的负载数据，应该是一个字典类型。
|
||
timeout (int, optional): 请求的超时时间，默认为600秒。
|
||
stream (bool, optional): 是否以流的方式下载响应内容，默认为False。
|
||
|
||
Returns:
|
||
response: 请求的响应结果，如果请求失败则返回None。
|
||
|
||
Raises:
|
||
None
|
||
|
||
"""
|
||
headers = {
|
||
"Content-Type": "application/json",
|
||
}
|
||
base_logger.info("🔄 正在请求模型接口...")
|
||
|
||
try:
|
||
res = requests.post(url, headers=headers, json=payload, stream=stream, timeout=timeout)
|
||
base_logger.info("🟢 接收响应中...\n")
|
||
return res
|
||
except requests.exceptions.Timeout:
|
||
base_logger.error(f"❌ 请求超时（超过 {timeout} 秒）")
|
||
return None
|
||
except requests.exceptions.RequestException as e:
|
||
base_logger.error(f"❌ 请求失败：{e}")
|
||
return None
|