21 lines
539 B
Python
21 lines
539 B
Python
def check_status(status_code: float, body: str):
|
|
'''
|
|
检查状态码和约定的返回值
|
|
'''
|
|
if status_code != 200:
|
|
raise Exception(f"注册任务失败，状态码：{status_code}")
|
|
if "code" not in body or body["code"] != 1:
|
|
raise Exception(f"注册任务失败，返回值：{body}")
|
|
|
|
def main(status_code: float, body: str):
|
|
try:
|
|
check_status(status_code, body)
|
|
except Exception as e:
|
|
raise e
|
|
|
|
data = body["data"]
|
|
|
|
return {
|
|
"RAG_results": data
|
|
}
|