From 464248651babf66656ec7aae82669e86a5e2e11a Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Wed, 25 Oct 2023 15:01:28 -0700
Subject: [PATCH] (fix) proxy server openapi.json and main.py

---
 litellm_server/main.py      | 2 +-
 litellm_server/openapi.json | 3 +++
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/litellm_server/main.py b/litellm_server/main.py
index 1782e28c1..4f2586b7a 100644
--- a/litellm_server/main.py
+++ b/litellm_server/main.py
@@ -9,7 +9,7 @@ from typing import Optional
 try:
     from utils import set_callbacks, load_router_config
 except ImportError:
-    from openai_proxy.utils import set_callbacks, load_router_config
+    from litellm_server.utils import set_callbacks, load_router_config
 import dotenv
 dotenv.load_dotenv() # load env variables
 
diff --git a/litellm_server/openapi.json b/litellm_server/openapi.json
index c72d55212..2879b331c 100644
--- a/litellm_server/openapi.json
+++ b/litellm_server/openapi.json
@@ -171,6 +171,9 @@
             },
             "user": {
               "type": "string"
+            },
+            "caching": {
+              "type": "boolean"
             }
           },
           "required": ["messages", "model"]