api/db/users.py

@@ -15,7 +15,7 @@ with open('config/credits.yml', encoding='utf8') as f:
 async def _get_mongo(collection_name: str):
     return AsyncIOMotorClient(os.getenv('MONGO_URI'))['nova-core'][collection_name]
 
-async def create(discord_id: str='') -> dict:
+async def create(discord_id: int=0) -> dict:
     """Adds a new user to the MongoDB collection."""
 
     chars = string.ascii_letters + string.digits
@@ -36,7 +36,7 @@ async def create(discord_id: str='') -> dict:
             'ban_reason': '',
         },
         'auth': {
-            'discord': str(discord_id),
+            'discord': discord_id,
             'github': None
         }
     }
@@ -52,7 +52,7 @@ async def by_id(user_id: str):
 
 async def by_discord_id(discord_id: str):
     db = await _get_mongo('users')
-    return await db.find_one({'auth.discord': str(int(discord_id))})
+    return await db.find_one({'auth.discord': discord_id})
 
 async def by_api_key(key: str):
     db = await _get_mongo('users')

api/moderation.py

@@ -47,8 +47,7 @@ async def is_policy_violated(inp) -> bool:
                     return False
 
             except Exception as exc:
-                if '401' in str(exc):
-                    await provider_auth.invalidate_key(req.get('provider_auth'))
+                # await provider_auth.invalidate_key(req.get('provider_auth'))
                 print('[!] moderation error:', type(exc), exc)
                 continue
 

api/streaming.py

@@ -10,7 +10,6 @@ from dotenv import load_dotenv
 from python_socks._errors import ProxyError
 
 import proxies
-import provider_auth
 import load_balancing
 
 from db import logs, users, stats
@@ -37,6 +36,8 @@ async def stream(
     input_tokens: int=0,
     incoming_request: starlette.requests.Request=None,
 ):
+
+    payload = payload or DEMO_PAYLOAD
     is_chat = False
     is_stream = payload.get('stream', False)
 
@@ -44,6 +45,7 @@ async def stream(
         is_chat = True
         model = payload['model']
 
+
     if is_chat and is_stream:
         chat_id = await chat.create_chat_id()
 
@@ -86,7 +88,7 @@ async def stream(
             webhook = dhooks.Webhook(os.getenv('DISCORD_WEBHOOK__API_ISSUE'))
 
             webhook.send(content=f'API Issue: **`{exc}`**\nhttps://i.imgflip.com/7uv122.jpg')
-            error = await errors.yield_error(
+            error = errors.yield_error(
                 500,
                 'Sorry, the API has no working keys anymore.',
                 'The admins have been messaged automatically.'
@@ -95,10 +97,7 @@ async def stream(
             return
 
         for k, v in target_request.get('headers', {}).items():
-            target_request['headers'][k] = v
-
-        if target_request['method'] == 'GET' and not payload:
-            target_request['payload'] = None
+            headers[k] = v
 
         async with aiohttp.ClientSession(connector=proxies.default_proxy.connector) as session:
             try:
@@ -109,30 +108,24 @@ async def stream(
                     data=target_request.get('data'),
                     json=target_request.get('payload'),
 
-                    headers=target_request.get('headers', {}),
+                    headers=headers,
                     cookies=target_request.get('cookies'),
 
                     ssl=False,
 
                     timeout=aiohttp.ClientTimeout(total=float(os.getenv('TRANSFER_TIMEOUT', '120'))),
                 ) as response:
-                    if response.content_type == 'application/json':
-                        data = await response.json()
 
-                        if data.get('code') == 'invalid_api_key':
-                            await provider_auth.invalidate_key(target_request.get('provider_auth'))
+                    if not is_stream:
+                        json_response = await response.json()
+
+                    try:
+                        response.raise_for_status()
+                    except Exception as exc:
+                        if 'Too Many Requests' in str(exc):
                             continue
 
-                        if response.ok:
-                            json_response = data
-
                     if is_stream:
-                        try:
-                            response.raise_for_status()
-                        except Exception as exc:
-                            if 'Too Many Requests' in str(exc):
-                                continue
-
                         try:
                             async for chunk in response.content.iter_any():
                                 send = False
@@ -166,7 +159,7 @@ async def stream(
 
                         except Exception as exc:
                             if 'Connection closed' in str(exc):
-                                error = await errors.yield_error(
+                                error = errors.yield_error(
                                     500,
                                     'Sorry, there was an issue with the connection.',
                                     'Please first check if the issue on your end. If this error repeats, please don\'t heistate to contact the staff!.'
@@ -187,9 +180,10 @@ async def stream(
             content=chat.CompletionStop
         )
         yield chunk
+
         yield 'data: [DONE]\n\n'
 
-    if not is_stream and json_response:
+    if not is_stream:
         yield json.dumps(json_response)
 
     # DONE =========================================================

api/transfer.py

@@ -14,8 +14,6 @@ from helpers import tokens, errors
 
 load_dotenv()
 
-models_list = json.load(open('models.json'))
-
 with open('config/credits.yml', encoding='utf8') as f:
     credits_config = yaml.safe_load(f)
 
@@ -67,9 +65,6 @@ async def handle(incoming_request):
         error = await errors.error(418, 'Your NovaAI account is not active (paused).', 'Simply re-activate your account using a Discord command or the web panel.')
         return error
 
-    if '/models' in path:
-        return fastapi.responses.JSONResponse(content=models_list)
-
     # COST
     costs = credits_config['costs']
     cost = costs['other']
@@ -106,6 +101,8 @@ async def handle(incoming_request):
 
     # READY
 
+    # payload['user'] = str(user['_id'])
+
     if 'chat/completions' in path and not payload.get('stream') is True:
         payload['stream'] = False
 

rewards/last_update.txt

@@ -1 +1 @@
-1691546405.042006
+1691460004.7354248

rewards/main.py

@@ -35,7 +35,7 @@ async def update_roles(mongo):
                 for role in level_role_names:
                     if role in role_names:
                         users.update_one(
-                            {'auth.discord': discord},
+                            {'auth.discord': int(discord)},
                             {'$set': {'level': role}}
                         )
                         print(f'Updated {discord} to {role}')

tests/__main__.py

@@ -40,6 +40,11 @@ def test_server():
 def test_api(model: str=MODEL, messages: List[dict]=None) -> dict:
     """Tests an API api_endpoint."""
 
+    headers = {
+        'Content-Type': 'application/json',
+        'Authorization': 'Bearer ' + closedai.api_key
+    }
+
     json_data = {
         'model': model,
         'messages': messages or MESSAGES,
@@ -48,7 +53,7 @@ def test_api(model: str=MODEL, messages: List[dict]=None) -> dict:
 
     response = httpx.post(
         url=f'{api_endpoint}/chat/completions',
-        headers=HEADERS,
+        headers=headers,
         json=json_data,
         timeout=20
     )
@@ -69,30 +74,25 @@ def test_library():
 def test_library_moderation():
     return closedai.Moderation.create("I wanna kill myself, I wanna kill myself; It's all I hear right now, it's all I hear right now")
 
-def test_models():
-    response = httpx.get(
-        url=f'{api_endpoint}/models',
-        headers=HEADERS,
-        timeout=5
-    )
-    response.raise_for_status()
-    return response.json()
-
 def test_all():
     """Runs all tests."""
 
     # print(test_server())
     # print(test_api())
-    # print(test_library())
+    print(test_library())
     # print(test_library_moderation())
-    print(test_models())
+
 
 def test_api_moderation(model: str=MODEL, messages: List[dict]=None) -> dict:
     """Tests an API api_endpoint."""
 
+    headers = {
+        'Authorization': 'Bearer ' + closedai.api_key
+    }
+
     response = httpx.get(
         url=f'{api_endpoint}/moderations',
-        headers=HEADERS,
+        headers=headers,
         timeout=20
     )
     response.raise_for_status()
@@ -104,9 +104,4 @@ if __name__ == '__main__':
     closedai.api_base = api_endpoint
     closedai.api_key = os.getenv('TEST_NOVA_KEY')
 
-    HEADERS = {
-        'Content-Type': 'application/json',
-        'Authorization': 'Bearer ' + closedai.api_key
-    }
-
     test_all()