Python 调用

python -m pip install requests

import os
import uuid

import requests


API_URL = "https://api.aibase.cn/v1/openapi/tasks"


def discover_questions(keyword: str) -> dict:
    api_key = os.environ.get("AIBASE_API_KEY")
    if not api_key:
        raise RuntimeError("请先设置 AIBASE_API_KEY 环境变量")

    try:
        response = requests.post(
            API_URL,
            headers={
                "AIBase-API-Key": api_key,
                "AIBase-Request-Id": str(uuid.uuid4()),
                "Content-Type": "application/json",
            },
            json={
                "apiCode": "geo.questions_corr_recommend",
                "data": {"keyword": keyword},
            },
            timeout=30,
        )
        response.raise_for_status()
    except requests.RequestException as exc:
        raise RuntimeError(f"AIBase HTTP 请求失败: {exc}") from exc

    try:
        payload = response.json()
    except requests.JSONDecodeError as exc:
        raise RuntimeError("AIBase 返回了非 JSON 响应") from exc

    if payload.get("code") != 200:
        raise RuntimeError(
            f"AIBase 业务错误 {payload.get('code')}: {payload.get('msg')}"
        )

    data = payload.get("data") or {}
    request_id = data.get("requestId")
    print("requestId:", request_id)
    return data.get("result") or {}


if __name__ == "__main__":
    result = discover_questions("多智能体系统")
    print("AI 相关问题:", result.get("aiQuestions", []))
    print("百度相关问题:", result.get("baiduQuestions", []))

python aibase_example.py