import requests
import env
BASE_URL = env.base_url

class APIClient:
    def __init__(self, base_url=BASE_URL):
        self.base_url = base_url
        self.session = requests.Session()

    def prompt(self, prompt,json=True, model='google/gemma-3-12b-it',use_cache=True,reduce_tokens=None):
        """No documentation available."""
        url = f"{self.base_url}/ai/prompt"
        response = self.session.post(url, json={ "prompt": prompt, "model": model,"json":json,'use_cache':use_cache,reduce_tokens:reduce_tokens })
        response.raise_for_status()
        return response.json()