dhisana 0.0.1.dev240__tar.gz → 0.0.1.dev241__tar.gz

{dhisana-0.0.1.dev240 → dhisana-0.0.1.dev241}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dhisana
-Version: 0.0.1.dev240
+Version: 0.0.1.dev241
 Summary: A Python SDK for Dhisana AI Platform
 Home-page: https://github.com/dhisana-ai/dhisana-python-sdk
 Author: Admin

{dhisana-0.0.1.dev240 → dhisana-0.0.1.dev241}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name='dhisana',  
-    version='0.0.1-dev240',
+    version='0.0.1-dev241',
     description='A Python SDK for Dhisana AI Platform',
     author='Admin',
     author_email='contact@dhisana.ai',

{dhisana-0.0.1.dev240 → dhisana-0.0.1.dev241}/src/dhisana/schemas/common.py

@@ -378,6 +378,7 @@ class SendEmailContext(BaseModel):
     sender_email: str
     labels: Optional[List[str]]
     body_format: BodyFormat = BodyFormat.AUTO
+    headers: Optional[Dict[str, str]] = None
     
 class QueryEmailContext(BaseModel):
     start_time: str

{dhisana-0.0.1.dev240 → dhisana-0.0.1.dev241}/src/dhisana/utils/google_oauth_tools.py

@@ -145,6 +145,12 @@ async def send_email_using_google_oauth_async(
     message["from"] = f"{send_email_context.sender_name} <{send_email_context.sender_email}>"
     message["subject"] = send_email_context.subject
 
+    extra_headers = getattr(send_email_context, "headers", None) or {}
+    for header, value in extra_headers.items():
+        if not header or value is None:
+            continue
+        message[header] = str(value)
+
     raw_message = base64.urlsafe_b64encode(message.as_bytes()).decode()
 
     payload: Dict[str, Any] = {"raw": raw_message}

{dhisana-0.0.1.dev240 → dhisana-0.0.1.dev241}/src/dhisana/utils/google_workspace_tools.py

@@ -179,6 +179,12 @@ async def send_email_using_service_account_async(
     message['from'] = f"{send_email_context.sender_name} <{send_email_context.sender_email}>"
     message['subject'] = send_email_context.subject
 
+    extra_headers = getattr(send_email_context, "headers", None) or {}
+    for header, value in extra_headers.items():
+        if not header or value is None:
+            continue
+        message[header] = str(value)
+
     # Base64-encode the message
     raw_message = base64.urlsafe_b64encode(message.as_bytes()).decode()
 
@@ -531,6 +537,7 @@ class SendEmailContext(BaseModel):
     sender_email: str
     labels: Optional[List[str]]
     body_format: BodyFormat = BodyFormat.AUTO
+    headers: Optional[Dict[str, str]] = None
     
 @assistant_tool
 async def send_email_using_service_account_async(

{dhisana-0.0.1.dev240 → dhisana-0.0.1.dev241}/src/dhisana/utils/mailgun_tools.py

@@ -126,6 +126,12 @@ async def send_email_using_mailgun_async(
         "html": html_body,
     }
 
+    extra_headers = getattr(send_email_context, "headers", None) or {}
+    for header, value in extra_headers.items():
+        if not header or value is None:
+            continue
+        data[f"h:{header}"] = str(value)
+
     async with aiohttp.ClientSession() as session:
         async with session.post(
             f"https://api.mailgun.net/v3/{domain}/messages",

{dhisana-0.0.1.dev240 → dhisana-0.0.1.dev241}/src/dhisana/utils/microsoft365_tools.py

@@ -168,6 +168,14 @@ async def send_email_using_microsoft_graph_async(
         ],
     }
 
+    extra_headers = getattr(send_email_context, "headers", None) or {}
+    if extra_headers:
+        message_payload["internetMessageHeaders"] = [
+            {"name": header, "value": str(value)}
+            for header, value in extra_headers.items()
+            if header and value is not None
+        ]
+
     headers = {
         "Authorization": f"Bearer {token}",
         "Content-Type": "application/json",

{dhisana-0.0.1.dev240 → dhisana-0.0.1.dev241}/src/dhisana/utils/sendgrid_tools.py

@@ -59,6 +59,7 @@ async def send_email_with_sendgrid(
     message: str,
     tool_config: Optional[List[Dict]] = None,
     body_format: Optional[str] = None,
+    custom_headers: Optional[Dict[str, str]] = None,
 ):
     """
     Send an email using SendGrid's v3 Mail Send API.
@@ -69,6 +70,7 @@ async def send_email_with_sendgrid(
     - subject: Subject string.
     - message: HTML body content.
     - tool_config: Optional integration configuration list.
+    - custom_headers: Optional mapping of header names to values.
     """
     api_key = get_sendgrid_api_key(tool_config)
 
@@ -98,6 +100,13 @@ async def send_email_with_sendgrid(
         "content": content,
     }
 
+    if custom_headers:
+        payload["headers"] = {
+            header: str(value)
+            for header, value in custom_headers.items()
+            if header and value is not None
+        }
+
     headers = {
         "Authorization": f"Bearer {api_key}",
         "Content-Type": "application/json",
@@ -143,6 +152,7 @@ async def send_email_using_sendgrid_async(
         message=ctx.body or "",
         body_format=getattr(ctx, "body_format", None),
         tool_config=tool_config,
+        custom_headers=getattr(ctx, "headers", None),
     )
     # Normalise output to a string id-like value
     if isinstance(result, dict) and result.get("status") == 202:

{dhisana-0.0.1.dev240 → dhisana-0.0.1.dev241}/src/dhisana/utils/smtp_email_tools.py

@@ -174,6 +174,12 @@ async def send_email_via_smtp_async(
     generated_id = f"<{uuid.uuid4()}@{domain_part}>"
     msg["Message-ID"] = generated_id
 
+    extra_headers = getattr(ctx, "headers", None) or {}
+    for header, value in extra_headers.items():
+        if not header or value is None:
+            continue
+        msg[header] = str(value)
+
     smtp_kwargs = dict(
         hostname=smtp_server,
         port=smtp_port,

{dhisana-0.0.1.dev240 → dhisana-0.0.1.dev241}/src/dhisana.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dhisana
-Version: 0.0.1.dev240
+Version: 0.0.1.dev241
 Summary: A Python SDK for Dhisana AI Platform
 Home-page: https://github.com/dhisana-ai/dhisana-python-sdk
 Author: Admin