fix(integration): Missing secret in SMTP action

TracecatHQ · Nov 26, 2024 · 01c189c · 01c189c
1 parent 82788b3
commit 01c189c
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/registry/tracecat_registry/base/core/email.py b/registry/tracecat_registry/base/core/email.py
@@ -68,6 +68,7 @@ def _build_email_message(
     namespace="core",
     description="Perform a send email action using SMTP",
     default_title="Send Email (SMTP)",
+    secrets=[smtp_secret],
 )
 def send_email_smtp(
     sender: Annotated[str, Field(..., description="Email address of the sender")],
@@ -106,7 +107,10 @@ def send_email_smtp(
 
     timeout = timeout or socket.getdefaulttimeout() or 10.0
     smtp_host = secrets.get("SMTP_HOST")
-    smtp_port = int(secrets.get("SMTP_PORT"))
+    try:
+        smtp_port = int(secrets.get("SMTP_PORT"))
+    except ValueError as e:
+        raise ValueError(f"Expected SMTP port to be an integer. Got: {e}") from e
     smtp_user = secrets.get("SMTP_USER")
     smtp_pass = secrets.get("SMTP_PASS")