revert format changes per @dsyme, 1rgs#33 (comment)

Author: Boris Dev

server.py

@@ -1,34 +1,32 @@
-import json
+from fastapi import FastAPI, Request, HTTPException
+import uvicorn
 import logging
-import os
-import re
-import sys
-import time
-import uuid
-from datetime import datetime
-from typing import Any, Dict, List, Literal, Optional, Union
-
+import json
+from pydantic import BaseModel, Field, field_validator
+from typing import List, Dict, Any, Optional, Union, Literal
 import httpx
+import os
+from fastapi.responses import JSONResponse, StreamingResponse
 import litellm
-import uvicorn
+import uuid
+import time
 from dotenv import load_dotenv
-from fastapi import FastAPI, HTTPException, Request
-from fastapi.responses import JSONResponse, StreamingResponse
-from pydantic import BaseModel, Field, field_validator
+import re
+from datetime import datetime
+import sys
 
 # Load environment variables from .env file
 load_dotenv()
 
 # Configure logging
 logging.basicConfig(
     level=logging.WARN,  # Change to INFO level to show more details
-    format="%(asctime)s - %(levelname)s - %(message)s",
+    format='%(asctime)s - %(levelname)s - %(message)s',
 )
 logger = logging.getLogger(__name__)
 
 # Configure uvicorn to be quieter
 import uvicorn
-
 # Tell uvicorn's loggers to be quiet
 logging.getLogger("uvicorn").setLevel(logging.WARNING)
 logging.getLogger("uvicorn.access").setLevel(logging.WARNING)
@@ -41,48 +39,42 @@ def filter(self, record):
         # Block messages containing these strings
         blocked_phrases = [
             "LiteLLM completion()",
-            "HTTP Request:",
+            "HTTP Request:", 
             "selected model name for cost calculation",
             "utils.py",
-            "cost_calculator",
+            "cost_calculator"
         ]
-
-        if hasattr(record, "msg") and isinstance(record.msg, str):
+        
+        if hasattr(record, 'msg') and isinstance(record.msg, str):
             for phrase in blocked_phrases:
                 if phrase in record.msg:
                     return False
         return True
 
-
 # Apply the filter to the root logger to catch all messages
 root_logger = logging.getLogger()
 root_logger.addFilter(MessageFilter())
 
-
 # Custom formatter for model mapping logs
 class ColorizedFormatter(logging.Formatter):
     """Custom formatter to highlight model mappings"""
-
     BLUE = "\033[94m"
     GREEN = "\033[92m"
     YELLOW = "\033[93m"
     RED = "\033[91m"
     RESET = "\033[0m"
     BOLD = "\033[1m"
-
+    
     def format(self, record):
         if record.levelno == logging.debug and "MODEL MAPPING" in record.msg:
             # Apply colors and formatting to model mapping logs
             return f"{self.BOLD}{self.GREEN}{record.msg}{self.RESET}"
         return super().format(record)
 
-
 # Apply custom formatter to console handler
 for handler in logger.handlers:
     if isinstance(handler, logging.StreamHandler):
-        handler.setFormatter(
-            ColorizedFormatter("%(asctime)s - %(levelname)s - %(message)s")
-        )
+        handler.setFormatter(ColorizedFormatter('%(asctime)s - %(levelname)s - %(message)s'))
 
 app = FastAPI()
 
@@ -122,12 +114,14 @@ def format(self, record):
 ]
 
 # List of Gemini models
-GEMINI_MODELS = ["gemini-2.5-pro-preview-03-25", "gemini-2.0-flash"]
+GEMINI_MODELS = [
+    "gemini-2.5-pro-preview-03-25",
+    "gemini-2.0-flash"
+]
 
 # List of Azure OpenAI models (deployment names)
 AZURE_MODELS = []  # These are deployment-specific, so we'll validate at runtime
 
-
 # Helper function to clean schema for Gemini
 def clean_gemini_schema(schema: Any) -> Any:
     """Recursively removes unsupported fields from a JSON schema for Gemini."""
@@ -140,22 +134,17 @@ def clean_gemini_schema(schema: Any) -> Any:
         if schema.get("type") == "string" and "format" in schema:
             allowed_formats = {"enum", "date-time"}
             if schema["format"] not in allowed_formats:
-                logger.debug(
-                    f"Removing unsupported format '{schema['format']}' for string type in Gemini schema."
-                )
+                logger.debug(f"Removing unsupported format '{schema['format']}' for string type in Gemini schema.")
                 schema.pop("format")
 
         # Recursively clean nested schemas (properties, items, etc.)
-        for key, value in list(
-            schema.items()
-        ):  # Use list() to allow modification during iteration
+        for key, value in list(schema.items()): # Use list() to allow modification during iteration
             schema[key] = clean_gemini_schema(value)
     elif isinstance(schema, list):
         # Recursively clean items in a list
         return [clean_gemini_schema(item) for item in schema]
     return schema
 
-
 # Models for Anthropic API requests
 class ContentBlockText(BaseModel):
     type: Literal["text"]