init enterprise pip

.circleci/config.yml

@@ -16,6 +16,14 @@ commands:
             echo "nameserver 127.0.0.11" | sudo tee /etc/resolv.conf
             echo "nameserver 8.8.8.8" | sudo tee -a /etc/resolv.conf
             echo "nameserver 8.8.4.4" | sudo tee -a /etc/resolv.conf
+  setup_litellm_enterprise_pip:
+    steps:
+      - run:
+          name: "Install local version of litellm-enterprise"
+          command: |
+            cd enterprise
+            python -m pip install -e .
+            cd ..
 
 jobs:
   # Add Windows testing job
@@ -445,6 +453,7 @@ jobs:
             pip install "pytest-retry==1.6.3"
             pip install "pytest-asyncio==0.21.1"
       # Run pytest and generate JUnit XML report
+      - setup_litellm_enterprise_pip
       - run:
           name: Run tests
           command: |

litellm/proxy/hooks/user_management_event_hooks.py

@@ -7,10 +7,12 @@ import uuid
 from datetime import datetime, timezone
 from typing import Optional
 
+from litellm_enterprise.enterprise_callbacks.send_emails.base_email import (
+    BaseEmailLogger,
+)
 from pydantic import BaseModel
 
 import litellm
-from enterprise.enterprise_callbacks.send_emails.base_email import BaseEmailLogger
 from litellm._logging import verbose_proxy_logger
 from litellm.proxy._types import (
     AUDIT_ACTIONS,

requirements.txt

@@ -52,4 +52,8 @@ tenacity==8.2.3  # for retrying requests, when litellm.num_retries set
 pydantic==2.10.2 # proxy + openai req.
 jsonschema==4.22.0 # validating json schema
 websockets==13.1.0 # for realtime API
-####
+
+########################
+# LITELLM ENTERPRISE DEPENDENCIES
+########################
+litellm-enterprise==0.1.1