|
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()
|
|
|
|
|