From d3fc5d4a1793d060dba37c608cac89a9cb1ec07c Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Mon, 8 Jul 2024 12:42:27 -0700
Subject: [PATCH] use types for ProxyErrorTypes

---
 litellm/proxy/_types.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/litellm/proxy/_types.py b/litellm/proxy/_types.py
index 966889ee24..42f81df7c4 100644
--- a/litellm/proxy/_types.py
+++ b/litellm/proxy/_types.py
@@ -1641,3 +1641,8 @@ class SpendCalculateRequest(LiteLLMBase):
     model: Optional[str] = None
     messages: Optional[List] = None
     completion_response: Optional[dict] = None
+
+
+class ProxyErrorTypes(str, enum.Enum):
+    budget_exceeded = "budget_exceeded"
+    auth_error = "auth_error"