dhisana 0.0.1.dev227__py3-none-any.whl → 0.0.1.dev229__py3-none-any.whl

dhisana/schemas/common.py

@@ -364,6 +364,12 @@ class Integration(IntegrationBase):
 Integration.model_rebuild()
 IntegrationUpdate.model_rebuild()
 
+class BodyFormat(str, Enum):
+    AUTO = "auto"
+    HTML = "html"
+    TEXT = "text"
+
+
 class SendEmailContext(BaseModel):
     recipient: str
     subject: str
@@ -371,6 +377,7 @@ class SendEmailContext(BaseModel):
     sender_name: str
     sender_email: str
     labels: Optional[List[str]]
+    body_format: BodyFormat = BodyFormat.AUTO
     
 class QueryEmailContext(BaseModel):
     start_time: str
@@ -386,4 +393,5 @@ class ReplyEmailContext(BaseModel):
     sender_email: str
     sender_name: str
     mark_as_read: str = "True"
-    add_labels: Optional[List[str]] = None
+    add_labels: Optional[List[str]] = None
+    reply_body_format: BodyFormat = BodyFormat.AUTO

dhisana/utils/compose_search_query.py

@@ -472,7 +472,7 @@ Output must be valid JSON, e.g.:
         prompt=prompt,
         response_format=TechnologyUsedCheck,
         effort="high",
-        model="gpt-4.1-mini",
+        model="gpt-5.1-chat",
         tool_config=tool_config
     )
 
@@ -534,7 +534,7 @@ Output must be valid JSON, e.g.:
         prompt=prompt,
         response_format=TechnologyAndRoleCheck,
         effort="high",
-        model="gpt-4.1-mini",
+        model="gpt-5.1-chat",
         tool_config=tool_config
     )
 

dhisana/utils/dataframe_tools.py

@@ -33,13 +33,13 @@ class PandasQuery(BaseModel):
     
 
 @assistant_tool   
-async def get_structured_output(message: str, response_type, model: str = "gpt-4.1-mini"):
+async def get_structured_output(message: str, response_type, model: str = "gpt-5.1-chat"):
     """
     Asynchronously retrieves structured output from the OpenAI API based on the input message.
 
     :param message: The input message to be processed by the OpenAI API.
     :param response_type: The expected format of the response (e.g., JSON).
-    :param model: The model to be used for processing the input message. Defaults to "gpt-4.1-mini".
+    :param model: The model to be used for processing the input message. Defaults to "gpt-5.1-chat".
     :return: A tuple containing the parsed response and a status string ('SUCCESS' or 'FAIL').
     """
     try:

dhisana/utils/email_body_utils.py

@@ -0,0 +1,72 @@
+"""Small helpers for handling e-mail bodies across providers."""
+
+from typing import Optional, Tuple
+import html as html_lib
+import re
+
+
+def looks_like_html(text: str) -> bool:
+    """Heuristically determine whether the body contains HTML markup."""
+    return bool(text and re.search(r"<[a-zA-Z][^>]*>", text))
+
+
+def _normalize_format_hint(format_hint: Optional[str]) -> str:
+    """
+    Normalize a user-supplied format hint into html/text/auto.
+
+    Accepts variations like "plain" or "plaintext" as text.
+    """
+    if not format_hint:
+        return "auto"
+    fmt_raw = getattr(format_hint, "value", format_hint)
+    fmt = str(fmt_raw).strip().lower()
+    if fmt in ("html",):
+        return "html"
+    if fmt in ("text", "plain", "plain_text", "plaintext"):
+        return "text"
+    return "auto"
+
+
+def html_to_plain_text(html: str) -> str:
+    """
+    Produce a very lightweight plain-text version of an HTML fragment.
+    This keeps newlines on block boundaries and strips tags.
+    """
+    if not html:
+        return ""
+    text = re.sub(r"(?is)<(script|style).*?>.*?</\1>", " ", html)
+    text = re.sub(r"(?i)<br\s*/?>", "\n", text)
+    text = re.sub(r"(?i)</(p|div|li|h[1-6])\s*>", "\n", text)
+    text = re.sub(r"(?is)<.*?>", "", text)
+    text = html_lib.unescape(text)
+    text = re.sub(r"\s+\n", "\n", text)
+    text = re.sub(r"\n{3,}", "\n\n", text)
+    return text.strip()
+
+
+def plain_text_to_html(text: str) -> str:
+    """Wrap plain text in a minimal HTML container that preserves newlines."""
+    if text is None:
+        return ""
+    escaped = html_lib.escape(text)
+    return f'<div style="white-space: pre-wrap">{escaped}</div>'
+
+
+def body_variants(body: Optional[str], format_hint: Optional[str]) -> Tuple[str, str, str]:
+    """
+    Return (plain, html, resolved_format) honoring an optional format hint.
+
+    resolved_format is "html" or "text" after applying auto-detection.
+    """
+    content = body or ""
+    fmt = _normalize_format_hint(format_hint)
+
+    if fmt == "html":
+        return html_to_plain_text(content), content, "html"
+    if fmt == "text":
+        return content, plain_text_to_html(content), "text"
+
+    if looks_like_html(content):
+        return html_to_plain_text(content), content, "html"
+
+    return content, plain_text_to_html(content), "text"

dhisana/utils/enrich_lead_information.py

@@ -162,7 +162,7 @@ async def get_clean_lead_info_with_llm(lead_info_str: str, tool_config: Optional
     lead_info, status = await get_structured_output_internal(
         prompt,
         BasicLeadInformation,
-        model="gpt-4.1-mini",
+        model="gpt-5.1-chat",
         tool_config=tool_config
     )
     if status == "ERROR":
@@ -493,7 +493,7 @@ async def get_user_linkedin_url_from_github_profile(
     response, status = await get_structured_output_internal(
         instructions, 
         UserInfoFromGithubProfileId, 
-        model="gpt-4.1-mini", 
+        model="gpt-5.1-chat", 
         use_web_search=True,
         tool_config=tool_config
     )
@@ -903,7 +903,7 @@ async def get_company_domain_from_llm_web_search(
         response, status = await get_structured_output_internal(
             instructions,
             CompanyInfoFromName,
-            model="gpt-4.1-mini",
+            model="gpt-5.1-chat",
             use_web_search=True,
             tool_config=tool_config
         )

dhisana/utils/generate_email.py

@@ -152,7 +152,7 @@ async def generate_personalized_email_copy(
             prompt=initial_prompt,
             response_format=EmailCopy,
             vector_store_id=vector_store_id,
-            model="gpt-4.1",
+            model="gpt-5.1-chat",
             tool_config=tool_config,
             use_cache=email_context.message_instructions.use_cache if email_context.message_instructions else True
         )
@@ -161,7 +161,7 @@ async def generate_personalized_email_copy(
         initial_response, initial_status = await get_structured_output_internal(
             prompt=initial_prompt,
             response_format=EmailCopy,
-            model="gpt-4.1",
+            model="gpt-5.1-chat",
             tool_config=tool_config,
             use_cache=email_context.message_instructions.use_cache if email_context.message_instructions else True
         )

dhisana/utils/generate_email_response.py

@@ -194,7 +194,7 @@ async def get_inbound_email_triage_action(
         triage_only, status = await get_structured_output_with_assistant_and_vector_store(
             prompt=triage_prompt,
             response_format=InboundEmailTriageResponse,
-            model="gpt-4.1",
+            model="gpt-5.1-chat",
             vector_store_id=cleaned_context.external_known_data.external_openai_vector_store_id,
             tool_config=tool_config,
             use_cache=cleaned_context.message_instructions.use_cache if cleaned_context.message_instructions else True
@@ -203,7 +203,7 @@ async def get_inbound_email_triage_action(
         triage_only, status = await get_structured_output_internal(
             prompt=triage_prompt,
             response_format=InboundEmailTriageResponse,
-            model="gpt-4.1",
+            model="gpt-5.1-chat",
             tool_config=tool_config,
             use_cache=cleaned_context.message_instructions.use_cache if cleaned_context.message_instructions else True
         )
@@ -376,7 +376,7 @@ async def generate_inbound_email_response_copy(
         initial_response, status = await get_structured_output_with_assistant_and_vector_store(
             prompt=prompt,
             response_format=InboundEmailTriageResponse,
-            model="gpt-4.1",
+            model="gpt-5.1-chat",
             vector_store_id=cleaned_context.external_known_data.external_openai_vector_store_id,
             tool_config=tool_config
         )
@@ -384,7 +384,7 @@ async def generate_inbound_email_response_copy(
         initial_response, status = await get_structured_output_internal(
             prompt=prompt,
             response_format=InboundEmailTriageResponse,
-            model="gpt-4.1",
+            model="gpt-5.1-chat",
             tool_config=tool_config
         )
 

dhisana/utils/generate_linkedin_connect_message.py

@@ -149,7 +149,7 @@ async def generate_personalized_linkedin_copy(
             prompt=prompt,
             response_format=LinkedInConnectMessage,
             vector_store_id=vector_store_id,
-            model="gpt-4.1",
+            model="gpt-5.1-chat",
             tool_config=tool_config,
             use_cache=linkedin_context.message_instructions.use_cache if linkedin_context.message_instructions else True
         )
@@ -158,7 +158,7 @@ async def generate_personalized_linkedin_copy(
         response_data, status = await get_structured_output_internal(
             prompt=prompt,
             response_format=LinkedInConnectMessage,
-            model="gpt-4.1",
+            model="gpt-5.1-chat",
             tool_config=tool_config,
             use_cache=linkedin_context.message_instructions.use_cache if linkedin_context.message_instructions else True
         )

dhisana/utils/generate_linkedin_response_message.py

@@ -224,7 +224,7 @@ async def generate_linkedin_response_message_copy(
         initial_response, status = await get_structured_output_with_assistant_and_vector_store(
             prompt=prompt,
             response_format=LinkedInTriageResponse,
-            model="gpt-4.1",
+            model="gpt-5.1-chat",
             vector_store_id=cleaned_context.external_known_data.external_openai_vector_store_id,
             tool_config=tool_config,
             use_cache=linkedin_context.message_instructions.use_cache if linkedin_context.message_instructions else True
@@ -233,7 +233,7 @@ async def generate_linkedin_response_message_copy(
         initial_response, status = await get_structured_output_internal(
             prompt,
             LinkedInTriageResponse,
-            model="gpt-4.1",
+            model="gpt-5.1-chat",
             tool_config=tool_config,
             use_cache=linkedin_context.message_instructions.use_cache if linkedin_context.message_instructions else True
         )

dhisana/utils/generate_structured_output_internal.py

@@ -56,7 +56,7 @@ async def get_structured_output_internal(
     response_format: BaseModel,
     effort: str = "low",
     use_web_search: bool = False,
-    model: str = "gpt-4.1-mini",
+    model: str = "gpt-5.1-chat",
     tool_config: Optional[List[Dict]] = None,
     use_cache: bool = True
 ):
@@ -217,7 +217,7 @@ async def get_structured_output_with_mcp(
     response_format: BaseModel,
     effort: str = "low",
     use_web_search: bool = False,
-    model: str = "gpt-4.1-mini",
+    model: str = "gpt-5.1-chat",
     tool_config: Optional[List[Dict[str, Any]]] = None,
 ) -> Tuple[Union[BaseModel, str], str]:
     """
@@ -376,7 +376,7 @@ async def get_structured_output_with_assistant_and_vector_store(
     response_format: BaseModel,
     vector_store_id: str,
     effort: str = "low",
-    model="gpt-4.1-mini",
+    model="gpt-5.1-chat",
     tool_config: Optional[List[Dict]] = None,
     use_cache: bool = True
 ):

dhisana/utils/google_oauth_tools.py

@@ -2,6 +2,7 @@ import base64
 import json
 import logging
 import re
+from email.mime.multipart import MIMEMultipart
 from email.mime.text import MIMEText
 from typing import Any, Dict, List, Optional
 
@@ -22,6 +23,7 @@ from dhisana.utils.email_parse_helpers import (
 )
 from dhisana.utils.assistant_tool_tag import assistant_tool
 from dhisana.utils.cache_output_tools import retrieve_output, cache_output
+from dhisana.utils.email_body_utils import body_variants
 from typing import Optional as _Optional  # avoid name clash in wrappers
 
 def _status_phrase(code: int) -> str:
@@ -127,7 +129,18 @@ async def send_email_using_google_oauth_async(
     """
     token = get_google_access_token(tool_config)
 
-    message = MIMEText(send_email_context.body, _subtype="html")
+    plain_body, html_body, resolved_fmt = body_variants(
+        send_email_context.body,
+        getattr(send_email_context, "body_format", None),
+    )
+    # Use multipart/alternative when we have both; fall back to single part for pure text.
+    if resolved_fmt == "text":
+        message = MIMEText(plain_body, "plain", _charset="utf-8")
+    else:
+        message = MIMEMultipart("alternative")
+        message.attach(MIMEText(plain_body, "plain", _charset="utf-8"))
+        message.attach(MIMEText(html_body, "html", _charset="utf-8"))
+
     message["to"] = send_email_context.recipient
     message["from"] = f"{send_email_context.sender_name} <{send_email_context.sender_email}>"
     message["subject"] = send_email_context.subject
@@ -265,7 +278,17 @@ async def reply_to_email_google_oauth_async(
     message_id_header = headers_map.get("Message-ID", "") or ""
 
     # 2) Build reply MIME
-    msg = MIMEText(reply_email_context.reply_body, _subtype="html")
+    plain_reply, html_reply, resolved_reply_fmt = body_variants(
+        reply_email_context.reply_body,
+        getattr(reply_email_context, "reply_body_format", None),
+    )
+    if resolved_reply_fmt == "text":
+        msg = MIMEText(plain_reply, "plain", _charset="utf-8")
+    else:
+        msg = MIMEMultipart("alternative")
+        msg.attach(MIMEText(plain_reply, "plain", _charset="utf-8"))
+        msg.attach(MIMEText(html_reply, "html", _charset="utf-8"))
+
     msg["To"] = to_addresses
     if cc_addresses:
         msg["Cc"] = cc_addresses

dhisana/utils/google_workspace_tools.py

@@ -24,8 +24,9 @@ from googleapiclient.http import MediaFileUpload, MediaIoBaseDownload
 from dhisana.schemas.sales import MessageItem
 from dhisana.utils.assistant_tool_tag import assistant_tool
 from dhisana.utils.email_parse_helpers import *
+from dhisana.utils.email_body_utils import body_variants
 import asyncio
-from dhisana.schemas.common import (SendEmailContext, QueryEmailContext, ReplyEmailContext)
+from dhisana.schemas.common import (SendEmailContext, QueryEmailContext, ReplyEmailContext, BodyFormat)
 
 
 ################################################################################
@@ -161,15 +162,18 @@ async def send_email_using_service_account_async(
 
     gmail_api_url = 'https://gmail.googleapis.com/gmail/v1/users/me/messages/send'
 
-    body = send_email_context.body or ""
+    plain_body, html_body, resolved_fmt = body_variants(
+        send_email_context.body,
+        getattr(send_email_context, "body_format", None),
+    )
 
-    if _looks_like_html(body):
+    if resolved_fmt == "text":
+        message = MIMEText(plain_body, _subtype="plain", _charset="utf-8")
+    else:
         # Gmail prefers multipart/alternative when HTML is present.
         message = MIMEMultipart("alternative")
-        message.attach(MIMEText(_html_to_plain_text(body), "plain", _charset="utf-8"))
-        message.attach(MIMEText(body, "html", _charset="utf-8"))
-    else:
-        message = MIMEText(body, _subtype="plain", _charset="utf-8")
+        message.attach(MIMEText(plain_body, "plain", _charset="utf-8"))
+        message.attach(MIMEText(html_body, "html", _charset="utf-8"))
 
     message['to'] = send_email_context.recipient
     message['from'] = f"{send_email_context.sender_name} <{send_email_context.sender_email}>"
@@ -509,6 +513,7 @@ class SendEmailContext(BaseModel):
     sender_name: str
     sender_email: str
     labels: Optional[List[str]]
+    body_format: BodyFormat = BodyFormat.AUTO
     
 @assistant_tool
 async def send_email_using_service_account_async(
@@ -537,8 +542,18 @@ async def send_email_using_service_account_async(
 
     gmail_api_url = 'https://gmail.googleapis.com/gmail/v1/users/me/messages/send'
 
+    plain_body, html_body, resolved_fmt = body_variants(
+        send_email_context.body,
+        getattr(send_email_context, "body_format", None),
+    )
+
     # Construct the MIME text message
-    message = MIMEText(send_email_context.body)
+    if resolved_fmt == "text":
+        message = MIMEText(plain_body, _subtype="plain", _charset="utf-8")
+    else:
+        message = MIMEMultipart("alternative")
+        message.attach(MIMEText(plain_body, "plain", _charset="utf-8"))
+        message.attach(MIMEText(html_body, "html", _charset="utf-8"))
     message['to'] = send_email_context.recipient
     message['from'] = f"{send_email_context.sender_name} <{send_email_context.sender_email}>"
     message['subject'] = send_email_context.subject
@@ -893,7 +908,16 @@ async def reply_to_email_async(
     message_id_header = headers_dict.get('Message-ID', '')
 
     # 3. Create the reply email message
-    msg = MIMEText(reply_email_context.reply_body)
+    plain_reply, html_reply, resolved_reply_fmt = body_variants(
+        reply_email_context.reply_body,
+        getattr(reply_email_context, "reply_body_format", None),
+    )
+    if resolved_reply_fmt == "text":
+        msg = MIMEText(plain_reply, _subtype="plain", _charset="utf-8")
+    else:
+        msg = MIMEMultipart("alternative")
+        msg.attach(MIMEText(plain_reply, "plain", _charset="utf-8"))
+        msg.attach(MIMEText(html_reply, "html", _charset="utf-8"))
     msg['To'] = to_addresses
     if cc_addresses:
         msg['Cc'] = cc_addresses

dhisana/utils/mailgun_tools.py

@@ -7,6 +7,7 @@ import aiohttp
 
 from dhisana.utils.assistant_tool_tag import assistant_tool
 from dhisana.schemas.common import SendEmailContext
+from dhisana.utils.email_body_utils import body_variants
 
 
 def get_mailgun_notify_key(tool_config: Optional[List[Dict]] = None) -> str:
@@ -59,6 +60,7 @@ async def send_email_with_mailgun(
     subject: str,
     message: str,
     tool_config: Optional[List[Dict]] = None,
+    body_format: Optional[str] = None,
 ):
     """
     Send an email using the Mailgun API.
@@ -74,13 +76,17 @@ async def send_email_with_mailgun(
         api_key = get_mailgun_notify_key(tool_config)
         domain = get_mailgun_notify_domain(tool_config)
 
+        body = message or ""
         data = {
             "from": sender,
             "to": recipients,
             "subject": subject,
-            "html": message,
         }
 
+        plain_body, html_body, _ = body_variants(body, body_format)
+        data["text"] = plain_body
+        data["html"] = html_body
+
         async with aiohttp.ClientSession() as session:
             async with session.post(
                 f"https://api.mailgun.net/v3/{domain}/messages",
@@ -107,11 +113,17 @@ async def send_email_using_mailgun_async(
     api_key = get_mailgun_notify_key(tool_config)
     domain = get_mailgun_notify_domain(tool_config)
 
+    plain_body, html_body, _ = body_variants(
+        send_email_context.body,
+        getattr(send_email_context, "body_format", None),
+    )
+
     data = {
         "from": f"{send_email_context.sender_name} <{send_email_context.sender_email}>",
         "to": [send_email_context.recipient],
         "subject": send_email_context.subject,
-        "html": send_email_context.body or "",
+        "text": plain_body,
+        "html": html_body,
     }
 
     async with aiohttp.ClientSession() as session:

dhisana/utils/microsoft365_tools.py

@@ -12,6 +12,7 @@ from dhisana.schemas.common import (
     ReplyEmailContext,
 )
 from dhisana.schemas.sales import MessageItem
+from dhisana.utils.email_body_utils import body_variants
 
 
 def get_microsoft365_access_token(tool_config: Optional[List[Dict]] = None) -> str:
@@ -149,11 +150,18 @@ async def send_email_using_microsoft_graph_async(
     base_url = "https://graph.microsoft.com/v1.0"
     base_res = _base_resource(sender_email, tool_config, auth_mode)
 
+    plain_body, html_body, resolved_fmt = body_variants(
+        send_email_context.body,
+        getattr(send_email_context, "body_format", None),
+    )
+    content_type = "Text" if resolved_fmt == "text" else "HTML"
+    content_body = plain_body if resolved_fmt == "text" else html_body
+
     message_payload: Dict[str, Any] = {
         "subject": send_email_context.subject,
         "body": {
-            "contentType": "HTML",
-            "content": send_email_context.body or "",
+            "contentType": content_type,
+            "content": content_body,
         },
         "toRecipients": [
             {"emailAddress": {"address": send_email_context.recipient}}

dhisana/utils/openai_assistant_and_file_utils.py

@@ -169,7 +169,7 @@ async def delete_files(
 async def run_file_search(
     query: str,
     vector_store_id: str,
-    model: str = "gpt-4.1-mini",
+    model: str = "gpt-5.1-chat",
     max_num_results: int = 5,
     store: bool = True,
     tool_config: Optional[List[Dict]] = None,
@@ -213,7 +213,7 @@ async def run_file_search(
 
 async def run_response_text(
     prompt: str,
-    model: str = "gpt-4.1-mini",
+    model: str = "gpt-5.1-chat",
     max_tokens: int = 2048,
     store: bool = True,
     tool_config: Optional[List[Dict]] = None,
@@ -237,7 +237,7 @@ async def run_response_text(
 async def run_response_structured(
     prompt: str,
     response_format: dict,
-    model: str = "gpt-4.1-mini",
+    model: str = "gpt-5.1-chat",
     max_tokens: int = 1024,
     store: bool = True,
     tool_config: Optional[List[Dict]] = None,

dhisana/utils/openai_helpers.py

@@ -330,7 +330,7 @@ async def process_agent_request(row_batch: List[Dict], workflow: Dict, custom_in
             name="AI Assistant",
             instructions=instructions,   
             tools=[],
-            model="gpt-4.1-mini"
+            model="gpt-5.1-chat"
         )
         thread = await client.beta.threads.create()
         parsed_outputs = []
@@ -955,7 +955,7 @@ async def get_function_call_arguments(input_text: str, function_name: str) -> Tu
         
         # Make the API call
         response = await client.beta.chat.completions.parse(
-            model="gpt-4.1-mini",
+            model="gpt-5.1-chat",
             messages=[
                 {"role": "system", "content": "Extract function arguments in JSON format."},
                 {"role": "user", "content": prompt},

dhisana/utils/research_lead.py

@@ -91,7 +91,7 @@ async def research_lead_with_full_info_ai(
     response, status = await get_structured_output_internal(
         instructions, 
         LeadResearchInformation, 
-        model="gpt-4.1-mini", 
+        model="gpt-5.1-chat", 
         tool_config=tool_config
     )
     if status == "SUCCESS":
@@ -165,7 +165,7 @@ async def research_company_with_full_info_ai(
     response, status = await get_structured_output_internal(
         instructions,
         CompanyResearchInformation,
-        model="gpt-4.1-mini",
+        model="gpt-5.1-chat",
         use_web_search=False,
         tool_config=tool_config
     )

dhisana/utils/sendgrid_tools.py

@@ -14,6 +14,7 @@ import aiohttp
 
 from dhisana.utils.assistant_tool_tag import assistant_tool
 from dhisana.schemas.common import SendEmailContext
+from dhisana.utils.email_body_utils import body_variants
 
 # --------------------------------------------------------------------------- #
 # Mailgun (re-exported from dedicated module for backward compatibility)
@@ -57,6 +58,7 @@ async def send_email_with_sendgrid(
     subject: str,
     message: str,
     tool_config: Optional[List[Dict]] = None,
+    body_format: Optional[str] = None,
 ):
     """
     Send an email using SendGrid's v3 Mail Send API.
@@ -79,6 +81,12 @@ async def send_email_with_sendgrid(
     if not to_list:
         return {"error": "No recipients provided"}
 
+    plain_body, html_body, _ = body_variants(message, body_format)
+    content = [
+        {"type": "text/plain", "value": plain_body},
+        {"type": "text/html", "value": html_body},
+    ]
+
     payload = {
         "personalizations": [
             {
@@ -87,9 +95,7 @@ async def send_email_with_sendgrid(
             }
         ],
         "from": from_obj,
-        "content": [
-            {"type": "text/html", "value": message or ""}
-        ],
+        "content": content,
     }
 
     headers = {
@@ -126,11 +132,16 @@ async def send_email_using_sendgrid_async(
     Provider-style wrapper for SendGrid using SendEmailContext.
     Returns an opaque token since SendGrid does not return a message id.
     """
+    plain_body, html_body, _ = body_variants(
+        ctx.body,
+        getattr(ctx, "body_format", None),
+    )
     result = await send_email_with_sendgrid(
         sender=f"{ctx.sender_name} <{ctx.sender_email}>",
         recipients=[ctx.recipient],
         subject=ctx.subject,
         message=ctx.body or "",
+        body_format=getattr(ctx, "body_format", None),
         tool_config=tool_config,
     )
     # Normalise output to a string id-like value

dhisana/utils/serpapi_search_tools.py

@@ -51,7 +51,7 @@ async def get_structured_output(text: str, tool_config: Optional[List[Dict]] = N
         f"Text:\n{text}"
     )
     result, status = await get_structured_output_internal(
-        prompt, LeadSearchResult, model = "gpt-4.1-nano", tool_config=tool_config
+        prompt, LeadSearchResult, model = "gpt-5.1-chat", tool_config=tool_config
     )
     if status != "SUCCESS" or result is None:
         return LeadSearchResult()
@@ -137,7 +137,7 @@ async def pick_best_linkedin_candidate_with_llm(
     result, status = await get_structured_output_internal(
         prompt,
         LinkedinCandidateChoice,
-        model="gpt-4.1-mini",
+        model="gpt-5.1-chat",
         tool_config=tool_config,
     )
 

dhisana/utils/smtp_email_tools.py

@@ -33,6 +33,7 @@ from dhisana.utils.google_workspace_tools import (
     QueryEmailContext,
     SendEmailContext,
 )
+from dhisana.utils.email_body_utils import body_variants
 
 
 # --------------------------------------------------------------------------- #
@@ -151,15 +152,18 @@ async def send_email_via_smtp_async(
     str
         The Message-ID of the sent message (e.g., "<uuid@yourdomain.com>").
     """
-    body = ctx.body or ""
+    plain_body, html_body, resolved_fmt = body_variants(
+        ctx.body,
+        getattr(ctx, "body_format", None),
+    )
 
-    if _looks_like_html(body):
+    if resolved_fmt == "text":
+        msg = MIMEText(plain_body, _subtype="plain", _charset="utf-8")
+    else:
         # Build multipart/alternative so HTML-capable clients see rich content.
         msg = MIMEMultipart("alternative")
-        msg.attach(MIMEText(_html_to_plain_text(body), "plain", _charset="utf-8"))
-        msg.attach(MIMEText(body, "html", _charset="utf-8"))
-    else:
-        msg = MIMEText(body, _subtype="plain", _charset="utf-8")
+        msg.attach(MIMEText(plain_body, "plain", _charset="utf-8"))
+        msg.attach(MIMEText(html_body, "html", _charset="utf-8"))
 
     msg["From"] = f"{ctx.sender_name} <{ctx.sender_email}>"
     msg["To"] = ctx.recipient

dhisana/utils/test_connect.py

@@ -1619,7 +1619,7 @@ async def test_connectivity(tool_config: List[Dict[str, Any]]) -> Dict[str, Dict
 
         # OpenAI needs extra args
         if tool_name == "openai":
-            model_name = next((c["value"] for c in config_entries if c["name"] == "modelName"), "gpt-4o-mini")
+            model_name = next((c["value"] for c in config_entries if c["name"] == "modelName"), "gpt-5.1-chat")
             reasoning_effort = next((c["value"] for c in config_entries if c["name"] == "reasoningEffort"), "medium")
             results[tool_name] = await test_openai(api_key, model_name, reasoning_effort)
 

dhisana/utils/trasform_json.py

@@ -159,7 +159,7 @@ async def create_property_mapping(
         prompt=user_prompt,
         response_format=PropertyMappingList,
         effort="high",
-        model="gpt-4.1-mini",
+        model="gpt-5.1-chat",
         tool_config=tool_config
     )
     if status == "SUCCESS" and response and response.properties:

dhisana/workflow/test.py

@@ -21,7 +21,7 @@ async def call_openai_api(system_content: str, user_content: str, max_tokens: in
     try:
         # Call the OpenAI API using the new client method
         response = client.chat.completions.create(
-            model="gpt-4.1-mini",
+            model="gpt-5.1-chat",
             messages=[
                 {"role": "system", "content": system_content},
                 {"role": "user", "content": user_content}

{dhisana-0.0.1.dev227.dist-info → dhisana-0.0.1.dev229.dist-info}/METADATA

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

{dhisana-0.0.1.dev227.dist-info → dhisana-0.0.1.dev229.dist-info}/RECORD

@@ -5,7 +5,7 @@ dhisana/cli/datasets.py,sha256=OwzoCrVQqmh0pKpUAKAg_w9uGYncbWU7ZrAL_QukxAk,839
 dhisana/cli/models.py,sha256=IzUFZW_X32mL3fpM1_j4q8AF7v5nrxJcxBoqvG-TTgA,706
 dhisana/cli/predictions.py,sha256=VYgoLK1Ksv6MFImoYZqjQJkds7e5Hso65dHwbxTNNzE,646
 dhisana/schemas/__init__.py,sha256=jv2YF__bseklT3OWEzlqJ5qE24c4aWd5F4r0TTjOrWQ,65
-dhisana/schemas/common.py,sha256=WErBbyZAQtkmSOXrMkamJkMptj3uCY1S2CGHJWkB_Ps,9103
+dhisana/schemas/common.py,sha256=SI6nldyfvZhNhfJwy2Qo1VJ7xwuAlg4QHYGLWPoVvZo,9287
 dhisana/schemas/sales.py,sha256=KUicIFC8uy4WLJegD6xIdOCUb63yGoqlDq-lALIrJKU,33395
 dhisana/ui/__init__.py,sha256=jv2YF__bseklT3OWEzlqJ5qE24c4aWd5F4r0TTjOrWQ,65
 dhisana/ui/components.py,sha256=4NXrAyl9tx2wWwoVYyABO-EOGnreGMvql1AkXWajIIo,14316
@@ -24,62 +24,63 @@ dhisana/utils/clay_tools.py,sha256=Pi3UMcxSDz2pd2Xq_tn9G8Zot5DUQM_cYNgWI2478EM,1
 dhisana/utils/clean_properties.py,sha256=D3GHL9grpe6baV7JyS9ZdHboESk1-ddcfLXHOLoFRws,4372
 dhisana/utils/company_utils.py,sha256=6icf3MuCzBp7iSoiq6LkQHg8A9hOqKuFyUwv0og1vBg,2038
 dhisana/utils/compose_salesnav_query.py,sha256=dPf23JEr_9i7E91EU6h4jd2IGHf4Rhma5WgGdlAockY,16486
-dhisana/utils/compose_search_query.py,sha256=QOBM1rUmBqKHsZIoyrewpgtl3k97SpkkWvYFcEWVgHs,29435
+dhisana/utils/compose_search_query.py,sha256=bqx7PyIgubRn_qd3pxgquQrofkqIyuEclzuwFNgo5GE,29435
 dhisana/utils/compose_three_step_workflow.py,sha256=-I-8Mr6k_fjRM5qciziwZ5L_mBYjW81vEsgHeahy-Sk,10512
 dhisana/utils/composite_tools.py,sha256=ZlwHCp7PXjYFUWUEeR_fTF0Z4Wg-4F6eBi1reE3FfUA,6092
-dhisana/utils/dataframe_tools.py,sha256=jxyvyXAMKxccST_W6o6FnBqAsvp7mGNOD6HV5V6xgeA,9242
+dhisana/utils/dataframe_tools.py,sha256=R6eUXjwR5SG6_K87rWjj4T5PT2w6xvVF2EKBajIv-RE,9242
 dhisana/utils/domain_parser.py,sha256=Kw5MPP06wK2azWQzuSiOE-DffOezLqDyF-L9JEBsMSU,1206
+dhisana/utils/email_body_utils.py,sha256=rlCVjdBlqNnEiUberJGXGcrYY1GQOkW0-aB6AEpS3L4,2302
 dhisana/utils/email_parse_helpers.py,sha256=LIdm1B1IyGSW50y8EkxOk6YRjvxO2SJTgTKPLxYls_o,4613
 dhisana/utils/email_provider.py,sha256=spjbNdnaVfCZEUw62EEHKijuXjI7vTVNqsftxJ15Erw,14352
-dhisana/utils/enrich_lead_information.py,sha256=DzhaAO5scOcQ95oSgIyxkYRbz96gbXoASjsgrAalOo8,38730
+dhisana/utils/enrich_lead_information.py,sha256=xJXxhLqQcR2wukM9MDBJ80BAoE-SH0cwTpDWOEHx-gs,38730
 dhisana/utils/extract_email_content_for_llm.py,sha256=SQmMZ3YJtm3ZI44XiWEVAItcAwrsSSy1QzDne7LTu_Q,3713
 dhisana/utils/fetch_openai_config.py,sha256=LjWdFuUeTNeAW106pb7DLXZNElos2PlmXRe6bHZJ2hw,5159
 dhisana/utils/field_validators.py,sha256=BZgNCpBG264aRqNUu_J67c6zfr15zlAaIw2XRy8J7DY,11809
 dhisana/utils/g2_tools.py,sha256=a4vmBYCBvLae5CdpOhMN1oNlvO8v9J1B5Sd8T5PzuU8,3346
 dhisana/utils/generate_content.py,sha256=1ycqHhxuxXnsX_5CNp5M6zW40VsM6toYMCwJU9jf__4,2111
-dhisana/utils/generate_email.py,sha256=N3XlswJV0b8v_eoCKpVW2UY345BsNzao1OKLOPLYp8Q,10732
-dhisana/utils/generate_email_response.py,sha256=YXa9gjvYN2UxOWyfsl1Lop5Q8LEHn1SMTNBmxhh6VH8,21142
+dhisana/utils/generate_email.py,sha256=CNRL5wjFHnsi3EUpqwIhMtbS2g0W-loM-1dBd7fI8AM,10742
+dhisana/utils/generate_email_response.py,sha256=4ikjGMc0NIKETWMxS-bHxCrB7H1AdTvDgXda0VpFwww,21162
 dhisana/utils/generate_flow.py,sha256=QMn6bWo0nH0fBvy2Ebub1XfH5udnVAqsPsbIqCtQPXU,4728
 dhisana/utils/generate_leads_salesnav.py,sha256=AONP1KXDJdg95JQBmKx5PQXUD2BHctc-MZOMuRfuK9U,12156
-dhisana/utils/generate_linkedin_connect_message.py,sha256=eL4RV2B1ByyMnjoGohdTA066rjFRaMKOU2hBfFMLsrE,9821
-dhisana/utils/generate_linkedin_response_message.py,sha256=udAt4V_vNuieyyfhrtTFWA1CiBx63jfac9E35wunS5k,14404
-dhisana/utils/generate_structured_output_internal.py,sha256=83SaThDAa_fANJEZ5CSCMcPpD_MN5zMI9NU1uEtQO2E,20705
+dhisana/utils/generate_linkedin_connect_message.py,sha256=xOe41CgYWJNyZAehk_IuLudKbCxVtYZAexlmAMd2vmI,9831
+dhisana/utils/generate_linkedin_response_message.py,sha256=W_nwMjmUT8Sk2pilR7HmPhrRtooj12Zw4-FbU8vwirw,14414
+dhisana/utils/generate_structured_output_internal.py,sha256=DmZ5QzW-79Jo3JL5nDCZQ-Fjl8Nz7FHK6S0rZxXbKyg,20705
 dhisana/utils/google_custom_search.py,sha256=5rQ4uAF-hjFpd9ooJkd6CjRvSmhZHhqM0jfHItsbpzk,10071
-dhisana/utils/google_oauth_tools.py,sha256=cYynzqRgv79aPMK8DehrXUyAPReO3hyDJiJ8-1008XM,25232
-dhisana/utils/google_workspace_tools.py,sha256=UVxEicpPsm_DfadXygp6ELzXuVfFv9cVSaQrQwFoxqM,45898
+dhisana/utils/google_oauth_tools.py,sha256=pN5YGkM50OieCFpz9RlmEwfrnGzkh342e0h5XschuuE,26211
+dhisana/utils/google_workspace_tools.py,sha256=wyBy5WN3-eUCrKz1HYr_CS0vdsiQgOA-SFb368jSDrY,46957
 dhisana/utils/hubspot_clearbit.py,sha256=keNX1F_RnDl9AOPxYEOTMdukV_A9g8v9j1fZyT4tuP4,3440
 dhisana/utils/hubspot_crm_tools.py,sha256=lbXFCeq690_TDLjDG8Gm5E-2f1P5EuDqNf5j8PYpMm8,99298
 dhisana/utils/instantly_tools.py,sha256=hhqjDPyLE6o0dzzuvryszbK3ipnoGU2eBm6NlsUGJjY,4771
 dhisana/utils/linkedin_crawler.py,sha256=6fMQTY5lTw2kc65SFHgOAM6YfezAS0Yhg-jkiX8LGHo,6533
 dhisana/utils/lusha_tools.py,sha256=MdiWlxBBjSNpSKz8rhNOyLPtbeh-YWHgGiUq54vN_gM,12734
-dhisana/utils/mailgun_tools.py,sha256=AOnU0B_wFRoCtF-QmkLR97eZEAR3IvmO-cesjsjuQ0g,4900
-dhisana/utils/microsoft365_tools.py,sha256=W5kAkLE_94fvU1v8x16XgwckglW05QJvPcBpBsO-QTk,16387
-dhisana/utils/openai_assistant_and_file_utils.py,sha256=qg3MEKN6wfpFZmkyIZUpWPPKXBUh8YUW7gahGFr1LoE,8957
-dhisana/utils/openai_helpers.py,sha256=NkTqbdql31GmwTcRd90KsSHpEoAFGpiimchjVTG5Rdc,40155
+dhisana/utils/mailgun_tools.py,sha256=qUD-jFMZpmkkkKtyihVSe9tgFzYe-UiiBDHQKtsLq0M,5284
+dhisana/utils/microsoft365_tools.py,sha256=AwWSdE-xeHkCx9T_cVgDbzBmsz7Co4KE45rAmt_lnAc,16723
+dhisana/utils/openai_assistant_and_file_utils.py,sha256=-eyPcxFvtS-DDtYQGle1SU6C6CuxjulVIojFy27HeWc,8957
+dhisana/utils/openai_helpers.py,sha256=ZK9S5-jcLCpiiD6XBLkCqYcNz-AGYmO9xh4e2H-FDLo,40155
 dhisana/utils/openapi_spec_to_tools.py,sha256=oBLVq3WeDWvW9O02NCvY8bxQURQdKwHJHGcX8bC_b2I,1926
 dhisana/utils/parse_linkedin_messages_txt.py,sha256=g3N_ac70mAEuDDQ7Ott6mkOaBwI3ZvcsJD3R9RlYwPQ,3320
 dhisana/utils/profile.py,sha256=12IhefaLp3j74zzBzVRe50_KWqtWZ_cdzUKlYNy9T2Y,1192
 dhisana/utils/proxy_curl_tools.py,sha256=V54zXOP3K2EWGSYanvw43n45hP_4KG8kw2n_LBiL0ak,49971
 dhisana/utils/proxycurl_search_leads.py,sha256=6PlraPNYQ4fIDzTYnY-T2g_ip5fPkqHigbGoPD8ZosQ,16131
 dhisana/utils/python_function_to_tools.py,sha256=jypddM6WTlIQmRWnqAugYJXvaPYaXaMgWAZRYeeGlj4,2682
-dhisana/utils/research_lead.py,sha256=i7xk3edNzYKeJ_-JzKXwGL-NeeApZuWpx4vd4Uvguw4,7009
+dhisana/utils/research_lead.py,sha256=L6w2fK5in8z2xmWnXBjbkvTdrwPf8ZfvAXq3gb7-S6s,7009
 dhisana/utils/sales_navigator_crawler.py,sha256=z8yurwUTLXdM71xWPDSAFNuDyA_SlanTpTncayG8YP8,44670
 dhisana/utils/salesforce_crm_tools.py,sha256=r6tROej4PtfcRN2AViPD7tV24oxBNm6QCE7uwhDH5Hc,17169
 dhisana/utils/search_router.py,sha256=p_1MPHbjalBM8gZuU4LADbmqSLNtZ4zll6CbPOc0POU,4610
 dhisana/utils/search_router_jobs.py,sha256=LgCHNGLMSv-ovgzF32muprfaDTdTpIKgrP5F7swAqhk,1721
-dhisana/utils/sendgrid_tools.py,sha256=uBaGuDFX1ef6WBNBBJTs_znw6T0nVYLaVqBOgc0n0I8,4761
+dhisana/utils/sendgrid_tools.py,sha256=Q3Mr7mF0iM_zoF-Vhb1lH5r8AyqkVGgYXaCMR2i7zLQ,5157
 dhisana/utils/serarch_router_local_business.py,sha256=n9yZjeXKOSgBnr0lCSQomP1nN3ucbC9ZTTSmSHQLeVo,2920
 dhisana/utils/serpapi_additional_tools.py,sha256=Xb1tc_oK-IjI9ZrEruYhFg8UJMLHQDaO9B51YiNbeBs,10569
 dhisana/utils/serpapi_google_jobs.py,sha256=HUJFZEW8UvYqsW0sWlEDXgI_IUomh5fTkzRJzEgsDGc,4509
 dhisana/utils/serpapi_google_search.py,sha256=B3sVq2OXdrYmPbH7cjQN4RFoek96qgKzXIayKXn0HLU,7318
 dhisana/utils/serpapi_local_business_search.py,sha256=vinmuXLaQ_0BpEdwnONZ2vLTq5xnRh6ICmPbnpckSN4,5775
-dhisana/utils/serpapi_search_tools.py,sha256=MglMPN9wHkGAHb7YAQXvEDsWK3RGS-zu3wUIvZAYxfo,31738
+dhisana/utils/serpapi_search_tools.py,sha256=_lPnrGyqO6iYGtB714_FBKw6_2XSDQWz2TZnZBOfSCk,31738
 dhisana/utils/serperdev_google_jobs.py,sha256=m5_2f_5y79FOFZz1A_go6m0hIUfbbAoZ0YTjUMO2BSI,4508
 dhisana/utils/serperdev_local_business.py,sha256=JoZfTg58Hojv61cyuwA2lcnPdLT1lawnWaBNrUYWnuQ,6447
 dhisana/utils/serperdev_search.py,sha256=_iBKIfHMq4gFv5StYz58eArriygoi1zW6VnLlux8vto,9363
-dhisana/utils/smtp_email_tools.py,sha256=yWg5BmQgRfnHiID7waHkq2sCNuCFBHe0-uVFgWtlr7c,17035
-dhisana/utils/test_connect.py,sha256=wkswcwEyMSVhZACHtqsZDhP6QpDMW4wq5MTcfRokvOM,69021
-dhisana/utils/trasform_json.py,sha256=s48DoyzVVCI4dTvSUwF5X-exX6VH6nPWrjbUENkYuNE,6979
+dhisana/utils/smtp_email_tools.py,sha256=tF6GoNqkS9pWP52VTTrYSgL7wPdIp3XTklxrHLdzU5o,17186
+dhisana/utils/test_connect.py,sha256=fTJW2UzC4zbzFKk_CvJJk-hRmMXYb4XJmyGa_ja-pbA,69022
+dhisana/utils/trasform_json.py,sha256=7V72XNDpuxUX0GHN5D83z4anj_gIf5zabaHeQm7b1_E,6979
 dhisana/utils/web_download_parse_tools.py,sha256=ouXwH7CmjcRjoBfP5BWat86MvcGO-8rLCmWQe_eZKjc,7810
 dhisana/utils/workflow_code_model.py,sha256=YPWse5vBb3O6Km2PvKh1Q3AB8qBkzLt1CrR5xOL9Mro,99
 dhisana/utils/zoominfo_tools.py,sha256=8RwOuhnfABWvrJAjDHvC32JsG1vgVfgckpRuoU6BvEU,13409
@@ -91,9 +92,9 @@ dhisana/workflow/__init__.py,sha256=jv2YF__bseklT3OWEzlqJ5qE24c4aWd5F4r0TTjOrWQ,
 dhisana/workflow/agent.py,sha256=esv7_i_XuMkV2j1nz_UlsHov_m6X5WZZiZm_tG4OBHU,565
 dhisana/workflow/flow.py,sha256=xWE3qQbM7j2B3FH8XnY3zOL_QXX4LbTW4ArndnEYJE0,1638
 dhisana/workflow/task.py,sha256=HlWz9mtrwLYByoSnePOemBUBrMEcj7KbgNjEE1oF5wo,1830
-dhisana/workflow/test.py,sha256=kwW8jWqSBNcRmoyaxlTuZCMOpGJpTbJQgHI7gSjwdzM,3399
-dhisana-0.0.1.dev227.dist-info/METADATA,sha256=xJCNdyGsigROGAFmkncnMU5ogDb2sScbvmOmCfTh1SQ,1190
-dhisana-0.0.1.dev227.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-dhisana-0.0.1.dev227.dist-info/entry_points.txt,sha256=jujxteZmNI9EkEaK-pOCoWuBujU8TCevdkfl9ZcKHek,49
-dhisana-0.0.1.dev227.dist-info/top_level.txt,sha256=NETTHt6YifG_P7XtRHbQiXZlgSFk9Qh9aR-ng1XTf4s,8
-dhisana-0.0.1.dev227.dist-info/RECORD,,
+dhisana/workflow/test.py,sha256=E7lRnXK0PguTNzyasHytLzTJdkqIPxG5_4qk4hMEeKc,3399
+dhisana-0.0.1.dev229.dist-info/METADATA,sha256=Op2HWN1M1FV1xx_flhumbWWujZg84_paPU2ax6Ft2Z0,1190
+dhisana-0.0.1.dev229.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+dhisana-0.0.1.dev229.dist-info/entry_points.txt,sha256=jujxteZmNI9EkEaK-pOCoWuBujU8TCevdkfl9ZcKHek,49
+dhisana-0.0.1.dev229.dist-info/top_level.txt,sha256=NETTHt6YifG_P7XtRHbQiXZlgSFk9Qh9aR-ng1XTf4s,8
+dhisana-0.0.1.dev229.dist-info/RECORD,,