57 lines
1.9 KiB
Python
57 lines
1.9 KiB
Python
"""BaoDan Workflow 调用封装。"""
|
||
import requests
|
||
from flask import current_app
|
||
|
||
|
||
class WorkflowHelper:
|
||
"""Workflow API 调用封装。"""
|
||
|
||
def run_workflow(self, inputs: dict, user_id: str) -> dict:
|
||
"""
|
||
执行 Workflow 获取推荐方案。
|
||
|
||
Args:
|
||
inputs: 客户信息和保险需求
|
||
user_id: 用户 ID
|
||
|
||
Returns:
|
||
Workflow 执行结果
|
||
"""
|
||
base_url = current_app.config.get("BAODAN_API_URL", "http://localhost:5001")
|
||
api_key = current_app.config.get("BAODAN_WORKFLOW_API_KEY", "")
|
||
|
||
if not api_key:
|
||
return {"code": 5001, "message": "Workflow API Key 未配置"}
|
||
|
||
try:
|
||
resp = requests.post(
|
||
f"{base_url}/v1/workflows/run",
|
||
json={
|
||
"inputs": inputs,
|
||
"response_mode": "blocking",
|
||
"user": f"recommend_{user_id}",
|
||
},
|
||
headers={"Authorization": f"Bearer {api_key}"},
|
||
timeout=120,
|
||
)
|
||
|
||
if resp.status_code != 200:
|
||
return {"code": 5001, "message": f"Workflow 调用失败: HTTP {resp.status_code}"}
|
||
|
||
result = resp.json()
|
||
|
||
# 解析 Workflow 输出
|
||
if result.get("status") == "succeeded":
|
||
outputs = result.get("data", {}).get("outputs", {})
|
||
return {"code": 0, "data": outputs}
|
||
else:
|
||
error = result.get("error", "未知错误")
|
||
return {"code": 5001, "message": f"Workflow 执行失败: {error}"}
|
||
|
||
except requests.Timeout:
|
||
return {"code": 5001, "message": "Workflow 调用超时（120秒）"}
|
||
except requests.RequestException as e:
|
||
return {"code": 5001, "message": f"Workflow 调用异常: {str(e)}"}
|
||
except Exception as e:
|
||
return {"code": 9999, "message": f"未知错误: {str(e)}"}
|