rahyana_client.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
from __future__ import annotations

import os
import json
from typing import Any, Dict, List, Optional

import requests


class RahyanaClient:
    def __init__(self, api_key: Optional[str] = None, base_url: Optional[str] = None, timeout: int = 30) -> None:
        self.api_key = api_key or os.getenv("API_KEY_OVERRIDE") or "YOUR_API_KEY_HERE"
        self.base_url = base_url or os.getenv("BASE_URL_OVERRIDE") or "https://rahyana.ir/api/v1"
        self.timeout = timeout

    def _headers(self) -> Dict[str, str]:
        return {
            "Content-Type": "application/json",
            "Authorization": f"Bearer {self.api_key}",
            "Accept": "application/json",
        }

    def post_json(self, path: str, payload: Dict[str, Any]) -> Dict[str, Any]:
        url = f"{self.base_url.rstrip('/')}/{path.lstrip('/')}"
        try:
            resp = requests.post(url, headers=self._headers(), data=json.dumps(payload), timeout=self.timeout)
        except Exception as e:
            return {"error": {"message": str(e), "type": "network_error"}}
        try:
            return resp.json()
        except Exception:
            return {"status": resp.status_code, "text": resp.text}


def chat_completions(
    client: RahyanaClient,
    *,
    model: str,
    messages: List[Dict[str, Any]],
    max_tokens: Optional[int] = None,
    temperature: Optional[float] = None,
) -> Dict[str, Any]:
    payload: Dict[str, Any] = {"model": model, "messages": messages}
    if max_tokens is not None:
        payload["max_tokens"] = max_tokens
    if temperature is not None:
        payload["temperature"] = temperature
    return client.post_json("chat/completions", payload)