Python 股票 API：通过 QVeris 获取报价

"""Inspect and execute one stock quote route. Running this may use QVeris credits."""
import json
import os
import sys
from pathlib import Path
import requests

BASE = "https://qveris.ai/api/v1"
TOOL = "finnhub_io_api.stock.quote"

def post(path, body, key, query=None):
    response = requests.post(
        BASE + path,
        headers={"Authorization": "Bearer " + key},
        params=query,
        json=body,
        timeout=5,
    )
    response.raise_for_status()
    payload = response.json()
    if not isinstance(payload, dict) or payload.get("success") is False:
        raise ValueError("QVeris returned an unsuccessful or unexpected response")
    return payload

def main():
    key = os.environ["QVERIS_API_KEY"]
    symbol = sys.argv[1] if len(sys.argv) > 1 else "AAPL"
    inspected = post("/tools/by-ids", {"tool_ids": [TOOL]}, key)
    match = next((t for t in inspected.get("results", []) if t.get("tool_id") == TOOL), None)
    if not match:
        raise ValueError("Selected route is unavailable; inspect another route")
    fields = match.get("params") or []
    if not any(p.get("name") == "symbol" and p.get("type") == "string" for p in fields):
        raise ValueError("The symbol contract changed; review the tool before calling")
    if any(p.get("required") and p.get("name") != "symbol" for p in fields):
        raise ValueError("New required parameters need review")
    result = post("/tools/execute", {
        "parameters": {"symbol": symbol}, "max_response_size": 20480
    }, key, {"tool_id": TOOL})
    if result.get("success") is not True or result.get("result") is None:
        raise ValueError("Execution did not return a result")
    Path("quote-response.json").write_text(json.dumps(result, indent=2), encoding="utf-8")
    print("Saved quote-response.json. Validate provider data and timestamps before display.")

if __name__ == "__main__":
    try:
        main()
    except (KeyError, ValueError, OSError, requests.RequestException) as error:
        # Avoid printing authorization headers or complete provider responses.
        raise SystemExit("Request stopped: " + type(error).__name__) from None