nova-api/api/db/logs.py

import os
import time

from dotenv import load_dotenv
from motor.motor_asyncio import AsyncIOMotorClient

from helpers import network

load_dotenv()

def _get_mongo(collection_name: str):
    return AsyncIOMotorClient(os.getenv('MONGO_URI'))['nova-core'][collection_name]

async def log_api_request(user: dict, incoming_request, target_url: str):
    payload = {}

    try:
        payload = await incoming_request.json()
    except Exception as exc:
        if 'JSONDecodeError' in str(exc):
            pass

    last_prompt = None
    if 'messages' in payload:
        last_prompt = payload['messages'][-1]['content']

    model = payload.get('model')
    ip_address = await network.get_ip(incoming_request)

    new_log_item = {
        'timestamp': time.time(),
        'method': incoming_request.method,
        'path': incoming_request.url.path,
        'user_id': user['_id'],
        'security': {
            'ip': ip_address,
            'useragent': incoming_request.headers.get('User-Agent')
        },
        'details': {
            'model': model,
            'last_prompt': last_prompt,
            'target_url': target_url
        }
    }

    inserted = await _get_mongo('logs').insert_one(new_log_item)
    log_item = await _get_mongo('logs').find_one({'_id': inserted.inserted_id})
    return log_item

async def by_id(log_id: str):
    return await _get_mongo('logs').find_one({'_id': log_id})

async def by_user_id(user_id: str):
    return await _get_mongo('logs').find({'user_id': user_id})

async def delete_by_id(log_id: str):
    return await _get_mongo('logs').delete_one({'_id': log_id})

async def delete_by_user_id(user_id: str):
    return await _get_mongo('logs').delete_many({'user_id': user_id})

if __name__ == '__main__':
    pass