generated file: dependencies/utils.py

Author: coslynx

Diff for: dependencies/utils.py

@@ -0,0 +1,66 @@
+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)