23 lines
719 B
Python
23 lines
719 B
Python
import aiohttp
|
|
from bot.db import db, DBTables, decrypt
|
|
|
|
|
|
async def get_models() -> list | None:
|
|
endpoint = decrypt(db[DBTables.config].get('endpoint'))
|
|
async with aiohttp.ClientSession() as session:
|
|
r = await session.get(endpoint + "/sdapi/v1/sd-models")
|
|
if r.status != 200:
|
|
return None
|
|
return [x["title"] for x in await r.json()]
|
|
|
|
|
|
async def set_model(model_name: str):
|
|
endpoint = decrypt(db[DBTables.config].get('endpoint'))
|
|
async with aiohttp.ClientSession() as session:
|
|
r = await session.post(endpoint + "/sdapi/v1/options", json={
|
|
"sd_model_checkpoint": model_name
|
|
})
|
|
if r.status != 200:
|
|
return False
|
|
return True
|