From 98d7588ecfedd580eb54b6dd6af8f5cd2ac6f524 Mon Sep 17 00:00:00 2001
From: nsde <neostyxde@gmail.com>
Date: Mon, 4 Sep 2023 13:25:12 +0200
Subject: [PATCH] No false notifications

---
 admintools/__main__.py |  2 +
 admintools/pruner.py   | 89 ++++++++++++++++++++++++++++++++++++++++++
 api/streaming.py       |  7 ++--
 3 files changed, 95 insertions(+), 3 deletions(-)
 create mode 100644 admintools/__main__.py
 create mode 100644 admintools/pruner.py

diff --git a/admintools/__main__.py b/admintools/__main__.py
new file mode 100644
index 0000000..973793a
--- /dev/null
+++ b/admintools/__main__.py
@@ -0,0 +1,2 @@
+# import pruner
+# pruner.prune()
\ No newline at end of file
diff --git a/admintools/pruner.py b/admintools/pruner.py
new file mode 100644
index 0000000..f71858d
--- /dev/null
+++ b/admintools/pruner.py
@@ -0,0 +1,89 @@
+# import os
+# import sys
+
+# project_root = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
+# sys.path.append(project_root)
+
+# from api.db.users import UserManager
+
+# manager = UserManager()
+
+# async def update_credits(settings=None):
+#     users = await manager.get_all_users()
+
+#     if not settings:
+#         await users.update_many({}, {'$inc': {'credits': 2500}})
+
+#     else:
+#         for key, value in settings.items():
+#             await users.update_many(
+#                 {'level': key}, {'$inc': {'credits': int(value)}})
+
+# get_all_users = manager.get_all_users
+
+
+# ###
+
+# import os
+# import time
+# import aiohttp
+# import pymongo
+# import asyncio
+# import autocredits
+
+# from settings import roles
+# from dotenv import load_dotenv
+
+# load_dotenv()
+
+# async def main():
+#     await update_roles()
+#     await autocredits.update_credits(roles)
+
+# async def update_roles():
+#     async with aiohttp.ClientSession() as session:
+#         try:
+#             async with session.get('http://0.0.0.0:3224/user_ids') as response:
+#                 discord_users = await response.json()
+#         except aiohttp.ClientError as e:
+#             print(f'Error: {e}')
+#             return
+
+#     level_role_names = [f'lvl{lvl}' for lvl in range(10, 110, 10)]
+#     users_doc = await autocredits.get_all_users()
+#     users = users_doc.find({})
+#     users = await users.to_list(length=None)
+
+
+#     for user in users:
+#         if not 'auth' in user:
+#             continue
+
+#         discord = str(user['auth']['discord'])
+
+#         for user_id, role_names in discord_users.items():
+#             if user_id == discord:
+#                 for role in level_role_names:
+#                     if role in role_names:
+#                         users_doc.update_one(
+#                             {'auth.discord': discord},
+#                             {'$set': {'level': role}}
+#                         )
+
+#                         print(f'Updated {discord} to {role}')
+
+#     return users
+
+# def launch():
+#     asyncio.run(main())
+
+#     with open('rewards/last_update.txt', 'w', encoding='utf8') as f:
+#         f.write(str(time.time()))
+
+# # ====================================================================================
+
+# def prune():
+#     # gets all users from 
+
+# if __name__ == '__main__':
+#     launch()
diff --git a/api/streaming.py b/api/streaming.py
index aa1d5ec..e57daf2 100644
--- a/api/streaming.py
+++ b/api/streaming.py
@@ -91,9 +91,10 @@ async def stream(
                     'cookies': incoming_request.cookies
                 })
         except ValueError as exc:
-            webhook = dhooks.Webhook(os.environ['DISCORD_WEBHOOK__API_ISSUE'])
-            webhook.send(content=f'API Issue: **`{exc}`**\nhttps://i.imgflip.com/7uv122.jpg')
-            yield await errors.yield_error(500, 'Sorry, the API has no working keys anymore.', 'The admins have been messaged automatically.')
+            if model in ['gpt-3.5-turbo', 'gpt-4', 'gpt-4-32k']:
+                webhook = dhooks.Webhook(os.environ['DISCORD_WEBHOOK__API_ISSUE'])
+                webhook.send(content=f'API Issue: **`{exc}`**\nhttps://i.imgflip.com/7uv122.jpg')
+                yield await errors.yield_error(500, 'Sorry, the API has no working keys anymore.', 'The admins have been messaged automatically.')
             return
 
         target_request['headers'].update(target_request.get('headers', {}))