add variables

main.py

@@ -18,7 +18,7 @@ MODEL_CATALOG_WEBHOOK_SECRET = os.environ.get("MODEL_CATALOG_WEBHOOK_SECRET")
 SIMSHIP_WEBHOOK = os.environ.get("SIMSHIP_WEBHOOK")
 SIMSHIP_WEBHOOK_SECRET = os.environ.get("SIMSHIP_WEBHOOK_SECRET")
 
-def send_slack_message(message: str, max_retries: int = 3, timeout: int = 10, retry_delay: int = 2):
+def send_slack_message(payload: dict, max_retries: int = 3, timeout: int = 10, retry_delay: int = 2):
     """
     Send a message to Slack using webhook URL with timeout and retry mechanism
     
@@ -29,11 +29,9 @@ def send_slack_message(message: str, max_retries: int = 3, timeout: int = 10, re
         retry_delay: Delay between retries in seconds (default: 2)
     """
     if not SLACK_WEBHOOK_URL:
-        logger.warning(f"No Slack webhook URL configured. Message: {message}")
+        logger.warning(f"No Slack webhook URL configured. Payload: {payload}")
         return
     
-    payload = {"text": message}
-    
     for attempt in range(max_retries + 1):  # +1 for initial attempt
         try:
             logger.info(f"Sending Slack message (attempt {attempt + 1}/{max_retries + 1})")
@@ -43,7 +41,7 @@ def send_slack_message(message: str, max_retries: int = 3, timeout: int = 10, re
                 timeout=timeout
             )
             response.raise_for_status()
-            logger.info(f"Slack message sent successfully: {message}")
+            logger.info(f"Slack message sent successfully: {payload}")
             return  # Success, exit the function
             
         except requests.exceptions.Timeout as e:
@@ -109,7 +107,7 @@ def process_model_catalog_webhook(data: dict):
             f"Model in the catalog renamed 🔄 \n"
             f"{repo.get('name', '')} → https://hf.co/{movedTo.get('name', '')}"
         )
-        send_slack_message(message)
+        send_slack_message({"text": message, "type": "", "link": ""})
 
     # repo deleted
     elif (
@@ -122,7 +120,7 @@ def process_model_catalog_webhook(data: dict):
             f"Model in the catalog deleted 🗑️\n"
             f"https://hf.co/{repo.get('name', '')}"
         )
-        send_slack_message(message)
+        send_slack_message({"text": message, "type": "", "link": ""})
 
     # other events
     else:
@@ -142,12 +140,13 @@ def process_simship_webhook(data: dict):
         event.get("scope") == "repo" and
         event.get("action") == "create"
     ):
+        link = f"https://hf.co/{repo.get('name', '')}"
         message = (
             "🛥️ SimShip Provider Alert 🛥️\n\n"
             f"SimShip Model created 🆕\n"
-            f"https://hf.co/{repo.get('name', '')}"
+            f"{link}"
         )
-        send_slack_message(message)
+        send_slack_message({"text": message, "link": link, "type": "simship"})
 
     # repo visibility update
     elif (
@@ -156,12 +155,13 @@ def process_simship_webhook(data: dict):
         event.get("action") == "update" and
         updatedConfig.get("private") is False
     ):
+        link = f"https://hf.co/{repo.get('name', '')}"
         message = (
             "🛥️ SimShip Provider Alert 🛥️\n\n"
             f"SimShip Model made public 🆕\n"
-            f"https://hf.co/{repo.get('name', '')}"
+            f"{link}"
         )
-        send_slack_message(message)
+        send_slack_message({"text": message, "link": link, "type": "simship"})
 
     # other events
     else: