1 files changed, 19 insertions, 0 deletions
diff --git a/.venv/lib/python3.12/site-packages/litellm/llms/infinity/rerank/common_utils.py b/.venv/lib/python3.12/site-packages/litellm/llms/infinity/rerank/common_utils.py
new file mode 100644
index 00000000..99477d1a
--- /dev/null
+++ b/.venv/lib/python3.12/site-packages/litellm/llms/infinity/rerank/common_utils.py
@@ -0,0 +1,19 @@
+import httpx
+
+from litellm.llms.base_llm.chat.transformation import BaseLLMException
+
+
+class InfinityError(BaseLLMException):
+    def __init__(self, status_code, message):
+        self.status_code = status_code
+        self.message = message
+        self.request = httpx.Request(
+            method="POST", url="https://github.com/michaelfeil/infinity"
+        )
+        self.response = httpx.Response(status_code=status_code, request=self.request)
+        super().__init__(
+            status_code=status_code,
+            message=message,
+            request=self.request,
+            response=self.response,
+        )  # Call the base class constructor with the parameters it needs