diff --git a/enterprise/dist/litellm_enterprise-0.1.1-py3-none-any.whl b/enterprise/dist/litellm_enterprise-0.1.1-py3-none-any.whl new file mode 100644 index 00000000000..d9a8ef41e62 Binary files /dev/null and b/enterprise/dist/litellm_enterprise-0.1.1-py3-none-any.whl differ diff --git a/enterprise/dist/litellm_enterprise-0.1.1.tar.gz b/enterprise/dist/litellm_enterprise-0.1.1.tar.gz new file mode 100644 index 00000000000..98cf132b213 Binary files /dev/null and b/enterprise/dist/litellm_enterprise-0.1.1.tar.gz differ diff --git a/enterprise/litellm_enterprise/__init__.py b/enterprise/litellm_enterprise/__init__.py new file mode 100644 index 00000000000..e69de29bb2d diff --git a/enterprise/enterprise_callbacks/example_logging_api.py b/enterprise/litellm_enterprise/enterprise_callbacks/example_logging_api.py similarity index 83% rename from enterprise/enterprise_callbacks/example_logging_api.py rename to enterprise/litellm_enterprise/enterprise_callbacks/example_logging_api.py index 2084ffb548e..14d34f5d1e8 100644 --- a/enterprise/enterprise_callbacks/example_logging_api.py +++ b/enterprise/litellm_enterprise/enterprise_callbacks/example_logging_api.py @@ -7,11 +7,11 @@ app = FastAPI() @app.post("/log-event") async def log_event(request: Request): try: - print("Received /log-event request") # noqa + print("Received /log-event request") # noqa # Assuming the incoming request has JSON data data = await request.json() - print("Received request data:") # noqa - print(data) # noqa + print("Received request data:") # noqa + print(data) # noqa # Your additional logic can go here # For now, just printing the received data diff --git a/enterprise/enterprise_callbacks/generic_api_callback.py b/enterprise/litellm_enterprise/enterprise_callbacks/generic_api_callback.py similarity index 100% rename from enterprise/enterprise_callbacks/generic_api_callback.py rename to enterprise/litellm_enterprise/enterprise_callbacks/generic_api_callback.py diff --git a/enterprise/enterprise_callbacks/send_emails/base_email.py b/enterprise/litellm_enterprise/enterprise_callbacks/send_emails/base_email.py similarity index 97% rename from enterprise/enterprise_callbacks/send_emails/base_email.py rename to enterprise/litellm_enterprise/enterprise_callbacks/send_emails/base_email.py index e476852b2e0..d9f3ce46bac 100644 --- a/enterprise/enterprise_callbacks/send_emails/base_email.py +++ b/enterprise/litellm_enterprise/enterprise_callbacks/send_emails/base_email.py @@ -105,9 +105,9 @@ class BaseEmailLogger(CustomLogger): support_contact = os.getenv("EMAIL_SUPPORT_CONTACT", self.DEFAULT_SUPPORT_EMAIL) base_url = os.getenv("PROXY_BASE_URL", "http://0.0.0.0:4000") - recipient_email: Optional[str] = ( - user_email or await self._lookup_user_email_from_db(user_id=user_id) - ) + recipient_email: Optional[ + str + ] = user_email or await self._lookup_user_email_from_db(user_id=user_id) if recipient_email is None: raise ValueError( f"User email not found for user_id: {user_id}. User email is required to send email." diff --git a/enterprise/enterprise_callbacks/send_emails/endpoints.py b/enterprise/litellm_enterprise/enterprise_callbacks/send_emails/endpoints.py similarity index 100% rename from enterprise/enterprise_callbacks/send_emails/endpoints.py rename to enterprise/litellm_enterprise/enterprise_callbacks/send_emails/endpoints.py diff --git a/enterprise/enterprise_callbacks/send_emails/resend_email.py b/enterprise/litellm_enterprise/enterprise_callbacks/send_emails/resend_email.py similarity index 100% rename from enterprise/enterprise_callbacks/send_emails/resend_email.py rename to enterprise/litellm_enterprise/enterprise_callbacks/send_emails/resend_email.py diff --git a/enterprise/pyproject.toml b/enterprise/pyproject.toml new file mode 100644 index 00000000000..a04accb96cb --- /dev/null +++ b/enterprise/pyproject.toml @@ -0,0 +1,30 @@ +[tool.poetry] +name = "litellm-enterprise" +version = "0.1.1" +description = "Package for LiteLLM Enterprise features" +authors = ["BerriAI"] +readme = "README.md" + + +[tool.poetry.urls] +homepage = "https://litellm.ai" +Homepage = "https://litellm.ai" +repository = "https://github.com/BerriAI/litellm" +Repository = "https://github.com/BerriAI/litellm" +documentation = "https://docs.litellm.ai" +Documentation = "https://docs.litellm.ai" + +[tool.poetry.dependencies] +python = ">=3.8.1,<4.0, !=3.9.7" + +[build-system] +requires = ["poetry-core"] +build-backend = "poetry.core.masonry.api" + +[tool.commitizen] +version = "0.1.18" +version_files = [ + "pyproject.toml:version", + "../requirements.txt:litellm-enterprise==", + "../pyproject.toml:litellm-enterprise = {version = \"" +] \ No newline at end of file