From dba35536e95029c7ac089d34c27a80cb38b46b28 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Fri, 24 May 2024 17:54:38 -0700
Subject: [PATCH] feat - resize logo

---
 litellm/integrations/slack_alerting.py | 16 +++++++++-------
 1 file changed, 9 insertions(+), 7 deletions(-)

diff --git a/litellm/integrations/slack_alerting.py b/litellm/integrations/slack_alerting.py
index 469b319a72..4f63a4ab05 100644
--- a/litellm/integrations/slack_alerting.py
+++ b/litellm/integrations/slack_alerting.py
@@ -18,6 +18,10 @@ from litellm.proxy._types import WebhookEvent
 import random
 
 
+# we use this for the email header, please send a test email if you change this. verify it looks good on email
+LITELLM_LOGO_URL = "https://litellm-listing.s3.amazonaws.com/litellm_logo.png"
+
+
 class LiteLLMBase(BaseModel):
     """
     Implements default functions, all pydantic objects should have.
@@ -785,7 +789,7 @@ Model Info:
 
         # make sure this is a premium user
         from litellm.proxy.proxy_server import premium_user
-        from litellm.proxy.proxy_server import CommonProxyErrors
+        from litellm.proxy.proxy_server import CommonProxyErrors, prisma_client
 
         if premium_user != True:
             raise Exception(
@@ -800,9 +804,6 @@ Model Info:
             and recipient_user_id is not None
             and prisma_client is not None
         ):
-            # try looking up this info in DB
-            from litellm.proxy.proxy_server import prisma_client
-
             user_row = await prisma_client.db.litellm_usertable.find_unique(
                 where={"user_id": recipient_user_id}
             )
@@ -820,7 +821,7 @@ Model Info:
                 "Trying to send email alert to no recipient", extra=webhook_event.dict()
             )
         email_html_content = f"""
-            <h1>LiteLLM</h1>
+            <img src="{LITELLM_LOGO_URL}" alt="LiteLLM Logo" width="150" height="50" />
 
             <p> Hi {recipient_email}, <br/>
  
@@ -830,6 +831,8 @@ Model Info:
 
             <h2>Usage Example</h2>
 
+            Detailed Documentation on <a href="https://docs.litellm.ai/docs/proxy/user_keys">Usage with OpenAI Python SDK, Langchain, LlamaIndex, Curl</a>
+
             <pre>
 
             import openai
@@ -850,7 +853,6 @@ Model Info:
 
             </pre>
 
-            Detailed Documentation on <a href="https://docs.litellm.ai/docs/proxy/user_keys">Usage with OpenAI Python SDK, Langchain, LlamaIndex, Curl</a>
 
             If you have any questions, please send an email to support@berri.ai <br /> <br />
 
@@ -895,7 +897,7 @@ Model Info:
 
         if webhook_event.event == "budget_crossed":
             email_html_content = f"""
-            <h1>LiteLLM</h1>
+            <img src="{LITELLM_LOGO_URL}" alt="LiteLLM Logo" width="150" height="50" />
 
             <p> Hi {user_name}, <br/>