Rerank 4 Fast

import os
import requests

resp = requests.post(
    "https://api.naga.ac/v1/rerank",
    headers={"Authorization": f"Bearer {os.environ['NAGA_API_KEY']}"},
    json={
        "model": "cohere/rerank-4-fast",
        "query": "How do I rotate an API key?",
        "documents": [
            "Create a new key, switch your clients to it, then revoke the old one.",
            "The balance is shown in US dollars.",
        ],
    },
)
for result in resp.json()["results"]:
    print(result["index"], result["relevance_score"])