60 lines
2.2 KiB
Python
60 lines
2.2 KiB
Python
|
from typing import Any
|
||
|
|
|
||
|
|
import requests
|
||
|
|
|
||
|
|
from config import Config
|
||
|
|
from task_service import TaskError
|
||
|
|
|
||
|
|
|
||
|
|
def _headers() -> dict[str, str]:
|
||
|
|
if not Config.SEARCH_QA_API_KEY:
|
||
|
|
raise TaskError(
|
||
|
|
"SEARCH_QA_NOT_CONFIGURED",
|
||
|
|
"搜索问一问开放 API 尚未配置",
|
||
|
|
503,
|
||
|
|
)
|
||
|
|
return {"X-API-Key": Config.SEARCH_QA_API_KEY}
|
||
|
|
|
||
|
|
|
||
|
|
def _request(method: str, path: str, **kwargs) -> requests.Response:
|
||
|
|
try:
|
||
|
|
session = requests.Session()
|
||
|
|
# 搜索问一问开放 API 必须直连；本机或服务环境的全局代理可能
|
||
|
|
# 指向不可用端口，且不应影响这一条受控的服务端对服务端链路。
|
||
|
|
session.trust_env = False
|
||
|
|
response = session.request(
|
||
|
|
method,
|
||
|
|
f"{Config.SEARCH_QA_API_BASE_URL}{path}",
|
||
|
|
headers={**_headers(), **kwargs.pop("headers", {})},
|
||
|
|
timeout=Config.SEARCH_QA_TIMEOUT_SECONDS,
|
||
|
|
**kwargs,
|
||
|
|
)
|
||
|
|
except requests.RequestException as exc:
|
||
|
|
raise TaskError("SEARCH_QA_UNAVAILABLE", "搜索问一问服务暂时不可用", 502) from exc
|
||
|
|
if response.status_code == 429:
|
||
|
|
raise TaskError("SEARCH_QA_RATE_LIMITED", "搜索问一问接口请求过于频繁，请稍后重试", 429)
|
||
|
|
if response.status_code >= 400:
|
||
|
|
message = "搜索问一问服务请求失败"
|
||
|
|
try:
|
||
|
|
payload = response.json()
|
||
|
|
message = str(payload.get("message") or payload.get("error") or message)
|
||
|
|
except ValueError:
|
||
|
|
pass
|
||
|
|
raise TaskError("SEARCH_QA_UPSTREAM_ERROR", message, 502)
|
||
|
|
return response
|
||
|
|
|
||
|
|
|
||
|
|
def list_runs(task_id: str = "", xhs_account_id: str = "", limit: int = 20) -> dict[str, Any]:
|
||
|
|
limit = max(1, min(int(limit), 100))
|
||
|
|
params: dict[str, Any] = {"limit": limit}
|
||
|
|
if task_id:
|
||
|
|
params = {"task_id": task_id}
|
||
|
|
elif xhs_account_id:
|
||
|
|
params["xhs_account_id"] = xhs_account_id
|
||
|
|
payload = _request("GET", "/api/open/v1/search-qa/runs", params=params).json()
|
||
|
|
if not payload.get("ok"):
|
||
|
|
raise TaskError("SEARCH_QA_UPSTREAM_ERROR", "搜索问一问服务返回失败", 502)
|
||
|
|
runs = payload.get("runs") or ([payload["run"]] if payload.get("run") else [])
|
||
|
|
return {"items": runs, "total": len(runs), "query": params}
|
||
|
|
|