Fixes

tdagent/tools/lookup_company_cloud_account_information.py

@@ -75,14 +75,18 @@ def lookup_cloud_account(public_cloud_provider, account_id):
     "error": "Account not found",
     "public_cloud_provider": "{public_cloud_provider}",
     "account_id": "{account_id}",
-    "message": "No cloud account found with these credentials"
+    "message": "No cloud account information found"
 }}"""
 
 
 # Create Gradio Interface
 gr_lookup_company_cloud_account_information = gr.Interface(
     fn=lookup_cloud_account,
-    inputs=["text","text"],
+    inputs=[gr.Dropdown(
+        choices=["AWS", "Azure", "GCP"],
+        label="Cloud Provider",
+        info="Select the cloud provider"
+    ), "text"],
     outputs="text",
     title="Company Cloud Account Lookup System",
     description="""
@@ -94,4 +98,4 @@ gr_lookup_company_cloud_account_information = gr.Interface(
     GCP: project-id-123456, project-id-789012
     """,
     theme="default"
-)
+)

tdagent/tools/send_email.py

@@ -3,18 +3,16 @@ import datetime
 import random
 
 
-def send_email(sender, recipient, subject, message):
+def send_email(recipient, subject, message):
     """
-    Simulates sending an email and prints it to the console in a well-formatted way.
+    Simulates sending an email from [email protected] and prints it to the console in a well-formatted way.
 
     This function takes email details, formats them into a standard email structure
     with headers and body, prints the formatted email to the console, and returns
-    a success message.
+    a success message. The sender is always set to [email protected].
 
     Parameters:
     -----------
-    sender : str
-        The email address of the sender (From field)
     recipient : str
         The email address of the recipient (To field)
     subject : str
@@ -30,24 +28,21 @@ def send_email(sender, recipient, subject, message):
 
     Example:
     --------
-    >>> send_email("john@example.com", "[email protected]", "Hello", "How are you?")
+    >>> send_email("jane@example.com", "Security Alert", "Please update your password.")
     ------ EMAIL SENT ------
-    From: john@example.com
+    From: cert@company.com
     To: [email protected]
-    Subject: Hello
+    Subject: Security Alert
     Date: Wed, 04 Jun 2025 16:40:58 +0000
     Message-ID: <123456.7890123456@mail-server>
 
-    How are you?
+    Please update your password.
     ------------------------
     'Email successfully sent to [email protected] at 16:40:58'
-
-    Notes:
-    ------
-    - This is a simulation function and does not actually send emails
-    - The Message-ID is randomly generated for each call
-    - The timestamp reflects the current system time
     """
+    # Fixed sender email
+    sender = "[email protected]"
+
     # Generate a random message ID
     message_id = f"<{random.randint(100000, 999999)}.{random.randint(1000000000, 9999999999)}@mail-server>"
 
@@ -71,13 +66,13 @@ Message-ID: {message_id}
     print(email_format)
 
     # Return a success message
-    return f"Email successfully sent to {recipient} at {datetime.datetime.now().strftime('%H:%M:%S')}"
+    return f"Email successfully sent from {sender} to {recipient} at {datetime.datetime.now().strftime('%H:%M:%S')}"
 
 
 # Create Gradio Interface
 gr_send_email = gr.Interface(
     fn=send_email,
-    inputs=["text", "text", "text", "text"],
+    inputs=["text", "text", "text"],
     outputs="text",
     title="Email Sender Simulator",
     description="This tool simulates sending an email by formatting and printing it to the console."