Spaces:
Sleeping
Sleeping
File size: 1,067 Bytes
5e4ca56 2c5812c 5e4ca56 2c5812c 5e4ca56 2c5812c 5e4ca56 2c5812c 5e4ca56 2c5812c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
from modules.presets import COMPLETION_URL, BALANCE_API_URL, USAGE_API_URL, API_HOST
import os
class State:
interrupted = False
completion_url = COMPLETION_URL
balance_api_url = BALANCE_API_URL
usage_api_url = USAGE_API_URL
def interrupt(self):
self.interrupted = True
def recover(self):
self.interrupted = False
def set_api_host(self, api_host):
self.completion_url = f"https://{api_host}/v1/chat/completions"
self.balance_api_url = f"https://{api_host}/dashboard/billing/credit_grants"
self.usage_api_url = f"https://{api_host}/dashboard/billing/usage"
os.environ["OPENAI_API_BASE"] = f"https://{api_host}/v1"
def reset_api_host(self):
self.completion_url = COMPLETION_URL
self.balance_api_url = BALANCE_API_URL
self.usage_api_url = USAGE_API_URL
os.environ["OPENAI_API_BASE"] = f"https://{API_HOST}/v1"
return API_HOST
def reset_all(self):
self.interrupted = False
self.completion_url = COMPLETION_URL
state = State()
|