from typing import Dict, Any

from . import Provider
from .. import config


class KiloProvider(Provider):
    name = "kilo"
    base_url = "https://api.kilo.ai/api/gateway"

    def is_available(self) -> bool:
        return True

    def is_free_model(self, model: Dict[str, Any]) -> bool:
        if model.get("isFree", False):
            return True
        model_id = model.get("id", "")
        if ":free" in model_id:
            return True
        if model.get("isFree") is False:
            return False
        pricing = model.get("pricing") or {}
        prompt_price = str(pricing.get("prompt", "0"))
        completion_price = str(pricing.get("completion", "0"))
        request_price = str(pricing.get("request", "0"))
        return (prompt_price == "0"
                and completion_price == "0"
                and request_price == "0")

    def get_chat_url(self) -> str:
        return f"{self.base_url}/chat/completions"

    def get_headers(self, model_id: str) -> Dict[str, str]:
        headers = {"Content-Type": "application/json"}
        api_key = config.get_kilo_api_key()
        if api_key:
            headers["Authorization"] = f"Bearer {api_key}"
        return headers