Update api/utils.py

api/utils.py

@@ -1,12 +1,12 @@
 # api/utils.py
 
-import json
 from datetime import datetime
-import uuid
+import json
 from typing import Any, Dict, Optional
+import uuid
+import re
 
 import httpx
-from fastapi import HTTPException
 from api.config import (
     MODEL_MAPPING,
     MODEL_ALIASES,
@@ -15,13 +15,13 @@ from api.config import (
     TRENDING_AGENT_MODE,
     BASE_URL
 )
+from fastapi import HTTPException
 from api.models import ChatRequest
 
 from api.logger import setup_logger
 
 logger = setup_logger(__name__)
 
-
 def create_chat_completion_data(
     content: str, model: str, timestamp: int, finish_reason: Optional[str] = None
 ) -> Dict[str, Any]:
@@ -40,7 +40,6 @@ def create_chat_completion_data(
         "usage": None,
     }
 
-
 def message_to_dict(message):
     if isinstance(message.content, str):
         return {"role": message.role, "content": message.content}
@@ -57,7 +56,6 @@ def message_to_dict(message):
     else:
         return {"role": message.role, "content": message.content}
 
-
 async def process_streaming_response(request: ChatRequest):
     # Map the requested model to the actual model used by the API
     model = MODEL_MAPPING.get(request.model, MODEL_ALIASES.get(request.model, "blackboxai"))