tgbots/TeleDiffusionBot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Image info, endpoint encryption

Browse Source
This commit is contained in:
BarsTiger
2023-02-24 23:41:29 +02:00
parent 846e65e2ce
commit 8f0cd0ac05
19 changed files with 211 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
bot/modules/api/models.py
View File

@@ -1,10 +1,10 @@
import aiohttp
from bot.db import db, DBTables
from bot.db import db, DBTables, decrypt
from rich import print
async def get_models():
endpoint = db[DBTables.config].get('endpoint')
endpoint = decrypt(db[DBTables.config].get('endpoint'))
try:
async with aiohttp.ClientSession() as session:
r = await session.get(endpoint + "/sdapi/v1/sd-models")
@@ -17,7 +17,7 @@ async def get_models():
async def set_model(model_name: str):
endpoint = db[DBTables.config].get('endpoint')
endpoint = decrypt(db[DBTables.config].get('endpoint'))
try:
async with aiohttp.ClientSession() as session:
r = await session.post(endpoint + "/sdapi/v1/options", json={