From 14da2d5adede440a0504941aead98725c54bcca0 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Thu, 20 Jun 2024 15:20:21 -0700
Subject: [PATCH] fix(_logging.py): fix timestamp format for json logs

Outputs timestamp in ISO 8601 format to work with fluentbit
---
 litellm/_logging.py       | 8 +++++++-
 litellm/proxy/_logging.py | 2 ++
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/litellm/_logging.py b/litellm/_logging.py
index b0935066b..e3835458c 100644
--- a/litellm/_logging.py
+++ b/litellm/_logging.py
@@ -2,6 +2,7 @@ import json
 import logging
 import os
 import traceback
+from datetime import datetime
 from logging import Formatter
 
 set_verbose = False
@@ -22,11 +23,16 @@ class JsonFormatter(Formatter):
     def __init__(self):
         super(JsonFormatter, self).__init__()
 
+    def formatTime(self, record, datefmt=None):
+        # Use datetime to format the timestamp in ISO 8601 format
+        dt = datetime.fromtimestamp(record.created)
+        return dt.isoformat()
+
     def format(self, record):
         json_record = {
             "message": record.getMessage(),
             "level": record.levelname,
-            "timestamp": self.formatTime(record, self.datefmt),
+            "timestamp": self.formatTime(record),
         }
 
         return json.dumps(json_record)
diff --git a/litellm/proxy/_logging.py b/litellm/proxy/_logging.py
index a3eb35180..dc6b34fd3 100644
--- a/litellm/proxy/_logging.py
+++ b/litellm/proxy/_logging.py
@@ -1,3 +1,5 @@
+### DEPRECATED ###
+## unused file. initially written for json logging on proxy.
 import json
 import logging
 import os