From 32cc9b155985f5c88a96f48c620bf59cb562498c Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Wed, 17 Jan 2024 12:56:26 -0800
Subject: [PATCH] (fix) prisma - non expiring keys

---
 litellm/proxy/utils.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/litellm/proxy/utils.py b/litellm/proxy/utils.py
index c4c76846a0..ae4aebad52 100644
--- a/litellm/proxy/utils.py
+++ b/litellm/proxy/utils.py
@@ -12,6 +12,7 @@ from fastapi import HTTPException, status
 import smtplib
 from email.mime.text import MIMEText
 from email.mime.multipart import MIMEMultipart
+from datetime import datetime
 
 
 def print_verbose(print_statement):
@@ -376,7 +377,8 @@ class PrismaClient:
                 print_verbose(f"PrismaClient: response={response}")
                 if response is not None:
                     # for prisma we need to cast the expires time to str
-                    response.expires = response.expires.isoformat()
+                    if isinstance(response.expires, datetime):
+                        response.expires = response.expires.isoformat()
                     return response
                 else:
                     # Token does not exist.