diff --git a/.circleci/config.yml b/.circleci/config.yml index 9fb1aeaf4f5..0a5b081b80a 100644 --- a/.circleci/config.yml +++ b/.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: | diff --git a/litellm/proxy/hooks/user_management_event_hooks.py b/litellm/proxy/hooks/user_management_event_hooks.py index b56d3143daf..eb0b003d23b 100644 --- a/litellm/proxy/hooks/user_management_event_hooks.py +++ b/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, diff --git a/requirements.txt b/requirements.txt index b803516d3e3..08ed4c75348 100644 --- a/requirements.txt +++ b/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