Setting generation properties

bot/modules/api/models.py

@@ -1,31 +1,22 @@
 import aiohttp
 from bot.db import db, DBTables, decrypt
-from rich import print
 
 
 async def get_models():
     endpoint = decrypt(db[DBTables.config].get('endpoint'))
-    try:
-        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()]
-    except Exception as e:
-        print(e)
-        return None
+    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'))
-    try:
-        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
-    except Exception as e:
-        print(e)
-        return False
+    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