dhisana 0.0.1.dev271__tar.gz → 0.0.1.dev273__tar.gz

{dhisana-0.0.1.dev271 → dhisana-0.0.1.dev273}/PKG-INFO

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

{dhisana-0.0.1.dev271 → dhisana-0.0.1.dev273}/setup.py

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

{dhisana-0.0.1.dev271 → dhisana-0.0.1.dev273}/src/dhisana/utils/generate_email.py

@@ -15,7 +15,25 @@ from pydantic import BaseModel, ConfigDict
 # ---------------------------------------------------------------------------------------
 # CONSTANTS
 # ---------------------------------------------------------------------------------------
-DEFAULT_OPENAI_MODEL = "gpt-4.1"
+DEFAULT_OPENAI_MODEL = "gpt-4.1"  # Larger context length - used for generating instructions
+DEFAULT_EMAIL_GEN_MODEL = "gpt-5.1-chat"  # Better content generation - used for generating email
+
+# -----------------------------------------------------------------------------
+# Email Generation Instructions schema (intermediate step)
+# -----------------------------------------------------------------------------
+class EmailGenerationInstructions(BaseModel):
+    """Instructions generated by the first step to guide email generation."""
+    recipient_context: str  # Key info about the lead/recipient
+    sender_context: str  # Key info about the sender
+    campaign_context: str  # Product, value prop, pain points, etc.
+    conversation_context: str  # Summary of any existing conversation
+    tone_and_style: str  # How the email should sound
+    key_points_to_include: str  # Main points to cover in the email
+    call_to_action: str  # What action to request
+    formatting_requirements: str  # HTML vs plain text, signature format, etc.
+    
+    model_config = ConfigDict(extra="forbid")
+
 
 # -----------------------------------------------------------------------------
 # Email Copy schema
@@ -68,28 +86,18 @@ FRAMEWORK_VARIATIONS = [
 ]
 
 # -----------------------------------------------------------------------------
-# Core function to generate an email copy
+# Step 1: Generate email instructions using gpt-4.1 (larger context)
 # -----------------------------------------------------------------------------
-async def generate_personalized_email_copy(
+async def generate_email_instructions(
     email_context: ContentGenerationContext,
     message_instructions: MessageGenerationInstructions,
     variation_text: str,
     tool_config: Optional[List[Dict]] = None,
-) -> dict:
+) -> EmailGenerationInstructions:
     """
-    Generate a personalized email using the provided context and instructions.
-
-    Steps:
-      1. Build a prompt referencing 6 main info:
-          (a) Lead Info
-          (b) Sender Info
-          (c) Campaign Info
-          (d) Messaging Instructions
-          (e) Additional Data (vector store) if any
-          (f) Current Conversation Context
-      2. Generate an initial draft with or without vector store usage.
-      3. Optionally refine if a vector store was used and user instructions were not provided.
-      4. Return the final subject & body.
+    Step 1 of 2-step email generation process.
+    Uses gpt-4.1 (larger context length) to process all context and generate
+    concise instructions for email generation.
     """
     cleaned_context = cleanup_email_context(email_context)
 
@@ -110,110 +118,212 @@ async def generate_personalized_email_copy(
     conversation_data = cleaned_context.current_conversation_context or ConversationContext()
 
     html_note = (
-        f"\n        Provide the HTML body using this guidance/template when possible:\n        {message_instructions.html_template}"
+        f"\n        HTML Template to follow: {message_instructions.html_template}"
         if getattr(message_instructions, "html_template", None)
         else ""
     )
-    important_requirements = """
-        IMPORTANT REQUIREMENTS:
+    
+    allow_html = getattr(message_instructions, "allow_html", False)
+    format_requirement = "HTML email with body_html field" if allow_html else "Plain text only (no HTML tags)"
+
+    # Construct the prompt for generating instructions
+    instructions_prompt = f"""
+Generate email writing instructions based on the following business context.
+
+LEAD INFORMATION:
+{lead_data.dict()}
+
+SENDER INFORMATION:
+Full Name: {sender_data.sender_full_name or ''}
+First Name: {sender_data.sender_first_name or ''}
+Last Name: {sender_data.sender_last_name or ''}
+Bio: {sender_data.sender_bio or ''}
+Appointment Booking URL: {sender_data.sender_appointment_booking_url or ''}
+
+CAMPAIGN INFORMATION:
+Product Name: {campaign_data.product_name or ''}
+Value Proposition: {campaign_data.value_prop or ''}
+Call To Action: {campaign_data.call_to_action or ''}
+Pain Points: {campaign_data.pain_points or []}
+Proof Points: {campaign_data.proof_points or []}
+Email Triage Guidelines: {campaign_data.email_triage_guidelines or ''}
+LinkedIn Triage Guidelines: {campaign_data.linkedin_triage_guidelines or ''}
+
+MESSAGING FRAMEWORK:
+{selected_instructions}{html_note}
+
+CONVERSATION HISTORY:
+Email Thread: {conversation_data.current_email_thread or ''}
+LinkedIn Thread: {conversation_data.current_linkedin_thread or ''}
+
+OUTPUT FIELDS:
+- recipient_context: Key details about the lead (name, company, role, relevant background)
+- sender_context: Sender name and relevant info for signature
+- campaign_context: Product or service being promoted, value proposition, key pain points
+- conversation_context: Summary of any prior conversation if exists
+- tone_and_style: Professional, friendly, etc.
+- key_points_to_include: Specific points to mention in the email
+- call_to_action: What action the recipient should take
+- formatting_requirements: Format is {format_requirement}. Salutation format, signature requirements.
+    """
+
+    # Check if a vector store is available
+    vector_store_id = (email_context.external_known_data.external_openai_vector_store_id
+                       if email_context.external_known_data else None)
+
+    instructions_response = None
+    instructions_status = ""
+
+    # Generate instructions using gpt-4.1 (larger context)
+    if vector_store_id:
+        instructions_response, instructions_status = await get_structured_output_with_assistant_and_vector_store(
+            prompt=instructions_prompt,
+            response_format=EmailGenerationInstructions,
+            vector_store_id=vector_store_id,
+            model=DEFAULT_OPENAI_MODEL,
+            tool_config=tool_config,
+            use_cache=email_context.message_instructions.use_cache if email_context.message_instructions else True
+        )
+    else:
+        instructions_response, instructions_status = await get_structured_output_internal(
+            prompt=instructions_prompt,
+            response_format=EmailGenerationInstructions,
+            model=DEFAULT_OPENAI_MODEL,
+            tool_config=tool_config,
+            use_cache=email_context.message_instructions.use_cache if email_context.message_instructions else True
+        )
+
+    if instructions_status != "SUCCESS":
+        raise Exception("Error: Could not generate email instructions in step 1.")
+    
+    return instructions_response
+
+
+# -----------------------------------------------------------------------------
+# Step 2: Generate email using gpt-5.1-chat (better content generation)
+# -----------------------------------------------------------------------------
+async def generate_email_from_instructions(
+    instructions: EmailGenerationInstructions,
+    message_instructions: MessageGenerationInstructions,
+    tool_config: Optional[List[Dict]] = None,
+    use_cache: bool = True,
+) -> EmailCopy:
+    """
+    Step 2 of 2-step email generation process.
+    Uses gpt-5.1-chat (better content generation) to generate the email
+    from the condensed instructions.
+    """
+    allow_html = getattr(message_instructions, "allow_html", False)
+    
+    if allow_html:
+        output_requirements = """
+        OUTPUT REQUIREMENTS:
         - Output must be JSON with "subject", "body", and "body_html" fields.
         - "body_html" should be clean HTML suitable for email (no external assets), inline styles welcome.
         - "body" must be the plain-text equivalent of "body_html".
-        - Keep it concise and relevant. No placeholders or extra instructions.
-        - Do not include PII or internal references, guids or content identifiers in the email.
-        - Use conversational names for company/person placeholders when provided.
-        - Email has salutation Hi <First Name>, unless otherwise specified.
-        - Make sure the signature in body has the sender_first_name correct and in the format the user specified.
-        - Do Not Make up information. use the information provided in the context and instructions only.
-        - Do Not use em dash in the generated output.
-    """
-    if not getattr(message_instructions, "allow_html", False):
-        important_requirements = """
-        IMPORTANT REQUIREMENTS:
+        """
+    else:
+        output_requirements = """
+        OUTPUT REQUIREMENTS:
         - Output must be JSON with "subject" and "body" fields only.
         - In the subject or body DO NOT include any HTML tags like <a>, <b>, <i>, etc.
         - The body and subject should be in plain text.
-        - If there is a link provided in email use it as is. dont wrap it in any HTML tags.
-        - Keep it concise and relevant. No placeholders or extra instructions.
-        - Do not include PII or internal references, guids or content identifiers in the email.
-        - User conversational name for company name if used.
-        - Email has saluation Hi <First Name>, unless otherwise specified.
-        - <First Name> is the first name of the lead. Its conversational name. It does not have any special characters or spaces.
-        - Make sure the signature in body has the sender_first_name is correct and in the format user has specified.
-        - Do Not Make up information. use the information provided in the context and instructions only.
-        - Make sure the body text is well-formatted and that newline and carriage-return characters are correctly present and preserved in the message body.
-        - Do Not use em dash in the generated output.
-    """
+        - If there is a link provided, use it as is without wrapping in any HTML tags.
+        """
+
+    email_prompt = f"""
+Generate a personalized business email based on these specifications:
 
-    # Construct the consolidated prompt
-    initial_prompt = f"""
-        Hi AI Assistant,
+RECIPIENT CONTEXT:
+{instructions.recipient_context}
 
-        Below is the context in 6 main sections. Use it to craft a concise, professional email:
+SENDER CONTEXT:
+{instructions.sender_context}
 
-        1) Lead Information:
-        {lead_data.dict()}
+CAMPAIGN CONTEXT:
+{instructions.campaign_context}
 
-        2) Sender Information:
-        Full Name: {sender_data.sender_full_name or ''}
-        First Name: {sender_data.sender_first_name or ''}
-        Last Name: {sender_data.sender_last_name or ''}
-        Bio: {sender_data.sender_bio or ''}
-        Appointment Booking URL: {sender_data.sender_appointment_booking_url or ''}
+CONVERSATION CONTEXT:
+{instructions.conversation_context}
 
-        3) Campaign Information:
-        Product Name: {campaign_data.product_name or ''}
-        Value Proposition: {campaign_data.value_prop or ''}
-        Call To Action: {campaign_data.call_to_action or ''}
-        Pain Points: {campaign_data.pain_points or []}
-        Proof Points: {campaign_data.proof_points or []}
-        Triage Guidelines (Email): {campaign_data.email_triage_guidelines or ''}
-        Triage Guidelines (LinkedIn): {campaign_data.linkedin_triage_guidelines or ''}
+TONE AND STYLE:
+{instructions.tone_and_style}
 
-        4) Messaging Instructions (template/framework):
-        {selected_instructions}{html_note}
+KEY POINTS TO INCLUDE:
+{instructions.key_points_to_include}
 
-        5) External Data / Vector Store:
-        (I will be provided with file_search tool if present.)
+CALL TO ACTION:
+{instructions.call_to_action}
 
-        6) Current Conversation Context:
-        Email Thread: {conversation_data.current_email_thread or ''}
-        LinkedIn Thread: {conversation_data.current_linkedin_thread or ''}
+FORMATTING REQUIREMENTS:
+{instructions.formatting_requirements}
 
-        {important_requirements}
+{output_requirements}
+
+ADDITIONAL REQUIREMENTS:
+- Keep it concise and relevant. No placeholders.
+- Do not include internal references or content identifiers in the email.
+- Use only the information provided.
+- Ensure the body text is well-formatted with proper newlines.
+- Do not use em dash in the output.
     """
 
-    # Check if a vector store is available
-    vector_store_id = (email_context.external_known_data.external_openai_vector_store_id
-                       if email_context.external_known_data else None)
+    # Generate email using gpt-5.1-chat (better content generation)
+    email_response, email_status = await get_structured_output_internal(
+        prompt=email_prompt,
+        response_format=EmailCopy,
+        model=DEFAULT_EMAIL_GEN_MODEL,
+        tool_config=tool_config,
+        use_cache=use_cache
+    )
 
-    initial_response = None
-    initial_status = ""
+    if email_status != "SUCCESS":
+        raise Exception("Error: Could not generate email in step 2.")
+    
+    return email_response
 
-    # Generate initial draft
-    if vector_store_id:
-        initial_response, initial_status = await get_structured_output_with_assistant_and_vector_store(
-            prompt=initial_prompt,
-            response_format=EmailCopy,
-            vector_store_id=vector_store_id,
-            model=DEFAULT_OPENAI_MODEL,
-            tool_config=tool_config,
-            use_cache=email_context.message_instructions.use_cache if email_context.message_instructions else True
-        )
-    else:
-        # Otherwise, generate the initial draft internally
-        initial_response, initial_status = await get_structured_output_internal(
-            prompt=initial_prompt,
-            response_format=EmailCopy,
-            model=DEFAULT_OPENAI_MODEL,
-            tool_config=tool_config,
-            use_cache=email_context.message_instructions.use_cache if email_context.message_instructions else True
-        )
 
-    if initial_status != "SUCCESS":
-        raise Exception("Error: Could not generate initial draft for the personalized email.")
-    plain_body = initial_response.body
-    html_body = getattr(initial_response, "body_html", None)
+# -----------------------------------------------------------------------------
+# Core function to generate an email copy (2-step process)
+# -----------------------------------------------------------------------------
+async def generate_personalized_email_copy(
+    email_context: ContentGenerationContext,
+    message_instructions: MessageGenerationInstructions,
+    variation_text: str,
+    tool_config: Optional[List[Dict]] = None,
+) -> dict:
+    """
+    Generate a personalized email using a 2-step process:
+
+    Step 1: Use gpt-4.1 (larger context length) to process all context and generate
+            concise instructions for email generation.
+    
+    Step 2: Use gpt-5.1-chat (better content generation) to generate the actual
+            email from the condensed instructions.
+
+    This approach leverages gpt-4.1's larger context window to process extensive
+    lead/campaign data, and gpt-5.1-chat's superior content generation for the
+    final email output.
+    """
+    # Step 1: Generate instructions using gpt-4.1
+    instructions = await generate_email_instructions(
+        email_context=email_context,
+        message_instructions=message_instructions,
+        variation_text=variation_text,
+        tool_config=tool_config,
+    )
+
+    # Step 2: Generate email using gpt-5.1-chat
+    use_cache = email_context.message_instructions.use_cache if email_context.message_instructions else True
+    email_response = await generate_email_from_instructions(
+        instructions=instructions,
+        message_instructions=message_instructions,
+        tool_config=tool_config,
+        use_cache=use_cache,
+    )
+
+    plain_body = email_response.body
+    html_body = getattr(email_response, "body_html", None)
     if not plain_body and html_body:
         plain_body = _html_to_plain_text(html_body)
 
@@ -225,7 +335,7 @@ async def generate_personalized_email_copy(
         receiver_name=email_context.lead_info.full_name or "",
         receiver_email=email_context.lead_info.email or "",
         iso_datetime=datetime.utcnow().isoformat(),
-        subject=initial_response.subject,
+        subject=email_response.subject,
         body=plain_body,
         html_body=html_body if getattr(message_instructions, "allow_html", False) else None,
     )