utils.py

from typing import Optional, Dict, Any, List
import logging
import os
from datetime import datetime, timedelta

from fastapi import HTTPException, status

from .config import settings
from .database import get_db
from .models import User
from .schemas.openai import OpenAIRequest, OpenAIResponse, OpenAIModel

logger = logging.getLogger(__name__)

# --- Constants ---

# ... (Constants specific to this file if any)

# --- Functions ---

def format_datetime(dt: datetime) -> str:
    """Formats a datetime object into a human-readable string."""
    return dt.strftime("%Y-%m-%d %H:%M:%S")

def get_api_key(user: User) -> Optional[str]:
    """Retrieves the OpenAI API key for the given user."""
    if user.api_key:
        return user.api_key
    else:
        return None

def generate_api_key():
    """Generates a random API key."""
    # ... (Implementation for generating a random API key)

def log_api_usage(user: User, endpoint: str, response_time: float) -> None:
    """Logs API usage statistics."""
    db = get_db()
    try:
        # ... (Implementation for logging API usage)
    except Exception as e:
        logger.error(f"Error logging API usage: {e}")

def format_openai_response(response: Dict[str, Any]) -> OpenAIResponse:
    """Formats the OpenAI API response into a structured OpenAIResponse object."""
    # ... (Implementation for formatting the OpenAI API response)

async def get_openai_model(model_id: str, user: User) -> OpenAIModel:
    """Retrieves information about a specific OpenAI model."""
    api_key = get_api_key(user)
    if not api_key:
        raise HTTPException(status_code=status.HTTP_401_UNAUTHORIZED, detail="API key not found.")
    openai.api_key = api_key
    try:
        model = openai.Model.retrieve(model_id)
        return OpenAIModel(**model)
    except openai.error.APIError as e:
        raise HTTPException(status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, detail=f"Error retrieving model: {e}")

# --- Classes ---

# ... (Classes specific to this file if any)

# --- Main ---

# ... (Main function or script execution logic if any)