From 9bbb13c373313107a21d62e6e051d65b0afce91b Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Fri, 10 May 2024 13:54:52 -0700
Subject: [PATCH] fix bug upsert_deployment

---
 litellm/router.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/litellm/router.py b/litellm/router.py
index 20566f255..32c2b61d1 100644
--- a/litellm/router.py
+++ b/litellm/router.py
@@ -2606,7 +2606,7 @@ class Router:
         self.model_names.append(deployment.model_name)
         return deployment
 
-    def upsert_deployment(self, deployment: Deployment) -> Deployment | None:
+    def upsert_deployment(self, deployment: Deployment) -> Optional[Deployment]:
         """
         Add or update deployment
         Parameters: