From 5e5466dfaf8dc1b66f3d0351a44a99b2df275640 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Wed, 11 Oct 2023 10:43:42 -0700
Subject: [PATCH] fix(proxy_server.py): use tomli instead of tomllib

---
 litellm/proxy/proxy_server.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py
index f9f342032..5c2cbe8a4 100644
--- a/litellm/proxy/proxy_server.py
+++ b/litellm/proxy/proxy_server.py
@@ -1,5 +1,4 @@
 import sys, os, platform, appdirs
-import tomllib
 import shutil, random, traceback
 sys.path.insert(
     0, os.path.abspath("../..")
@@ -8,11 +7,15 @@ sys.path.insert(
 try:
     import uvicorn
     import fastapi
+    import tomli as tomllib
 except ImportError:
     import subprocess
     import sys
 
-    subprocess.check_call([sys.executable, "-m", "pip", "install", "uvicorn", "fastapi"])
+    subprocess.check_call([sys.executable, "-m", "pip", "install", "uvicorn", "fastapi", "tomli"])
+    import uvicorn
+    import fastapi
+    import tomli as tomllib
 
 
 print()