Spaces:
Runtime error
Runtime error
from json import JSONDecodeError | |
import requests | |
class TradeAssistantAPI: | |
def __init__(self, token: str) -> None: | |
self.token = token | |
self.url = "https://natexcvi-trade-assistant.hf.space" | |
self.base_headers = {"Authorization": f"Bearer {self.token}"} | |
def get_recommendations(self): | |
url = f"{self.url}/recommend.json" | |
response = requests.get(url, headers=self.base_headers) | |
try: | |
body = response.json() | |
if "message" in body: | |
raise Exception(body["message"]) | |
return body | |
except JSONDecodeError: | |
raise Exception(response.text) | |
def train_model(self): | |
url = f"{self.url}/train" | |
response = requests.post(url, headers=self.base_headers) | |
try: | |
body = response.json() | |
if "message" in body: | |
raise Exception(body["message"]) | |
return body | |
except JSONDecodeError: | |
raise Exception(response.text) | |
def get_training_job_status(self, job_id: str): | |
url = f"{self.url}/train/{job_id}" | |
response = requests.get(url, headers=self.base_headers) | |
try: | |
body = response.json() | |
if "message" in body: | |
raise Exception(body["message"]) | |
return body | |
except JSONDecodeError: | |
raise Exception(response.text) | |
def list_training_jobs(self): | |
url = f"{self.url}/train" | |
response = requests.get(url, headers=self.base_headers) | |
try: | |
body = response.json() | |
if "message" in body: | |
raise Exception(body["message"]) | |
return body | |
except JSONDecodeError: | |
raise Exception(response.text) | |
def get_portfolio(self): | |
url = f"{self.url}/portfolio" | |
response = requests.get(url, headers=self.base_headers) | |
try: | |
body = response.json() | |
if "message" in body: | |
raise Exception(body["message"]) | |
return body | |
except JSONDecodeError: | |
raise Exception(response.text) | |