dhisana 0.0.1.dev263__tar.gz → 0.0.1.dev264__tar.gz

{dhisana-0.0.1.dev263 → dhisana-0.0.1.dev264}/PKG-INFO

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

{dhisana-0.0.1.dev263 → dhisana-0.0.1.dev264}/setup.py

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

{dhisana-0.0.1.dev263 → dhisana-0.0.1.dev264}/src/dhisana/utils/enrich_lead_information.py

@@ -165,7 +165,7 @@ async def get_clean_lead_info_with_llm(lead_info_str: str, tool_config: Optional
         model="gpt-5.1-chat",
         tool_config=tool_config
     )
-    if status == "ERROR":
+    if status != "SUCCESS":
         return {}
     return lead_info.model_dump()
 

{dhisana-0.0.1.dev263 → dhisana-0.0.1.dev264}/src/dhisana/utils/generate_email_response.py

@@ -16,6 +16,11 @@ from dhisana.utils.generate_structured_output_internal import (
     get_structured_output_internal
 )
 
+# ---------------------------------------------------------------------------------------
+# CONSTANTS
+# ---------------------------------------------------------------------------------------
+DEFAULT_TRIAGE_MODEL = "gpt-4.1"
+
 # ---------------------------------------------------------------------------------------
 # MODEL
 # ---------------------------------------------------------------------------------------
@@ -194,7 +199,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-5.1-chat",
+            model=DEFAULT_TRIAGE_MODEL,
             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,13 +208,28 @@ async def get_inbound_email_triage_action(
         triage_only, status = await get_structured_output_internal(
             prompt=triage_prompt,
             response_format=InboundEmailTriageResponse,
-            model="gpt-5.1-chat",
+            model=DEFAULT_TRIAGE_MODEL,
             tool_config=tool_config,
             use_cache=cleaned_context.message_instructions.use_cache if cleaned_context.message_instructions else True
         )
 
     if status != "SUCCESS":
-        raise Exception("Error in generating triage action.")
+        campaign_id = context.campaign_context.campaign_id if context.campaign_context else 'N/A'
+        if status == "CONTEXT_LENGTH_EXCEEDED":
+            raise Exception(
+                f"Email thread too long for model. Campaign ID: {campaign_id}. "
+                f"Consider truncating thread or switching models."
+            )
+        elif status in ("API_ERROR", "ERROR"):
+            raise Exception(
+                f"Error in generating triage action. Status: {status}. "
+                f"Campaign ID: {campaign_id}. Details: {triage_only}"
+            )
+        else:
+            raise Exception(
+                f"Error in generating triage action. Status: {status}. "
+                f"Campaign ID: {campaign_id}"
+            )
     return triage_only
 
 
@@ -393,7 +413,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-5.1-chat",
+            model=DEFAULT_TRIAGE_MODEL,
             vector_store_id=cleaned_context.external_known_data.external_openai_vector_store_id,
             tool_config=tool_config
         )
@@ -401,12 +421,30 @@ async def generate_inbound_email_response_copy(
         initial_response, status = await get_structured_output_internal(
             prompt=prompt,
             response_format=InboundEmailTriageResponse,
-            model="gpt-5.1-chat",
+            model=DEFAULT_TRIAGE_MODEL,
             tool_config=tool_config
         )
 
     if status != "SUCCESS":
-        raise Exception("Error in generating the inbound email triage response.")
+        campaign_id = (
+            campaign_context.campaign_context.campaign_id 
+            if campaign_context.campaign_context else 'N/A'
+        )
+        if status == "CONTEXT_LENGTH_EXCEEDED":
+            raise Exception(
+                f"Email thread too long for model. Campaign ID: {campaign_id}. "
+                f"Consider truncating thread or switching models."
+            )
+        elif status in ("API_ERROR", "ERROR"):
+            raise Exception(
+                f"Error in generating inbound email triage response. Status: {status}. "
+                f"Campaign ID: {campaign_id}. Details: {initial_response}"
+            )
+        else:
+            raise Exception(
+                f"Error in generating inbound email triage response. Status: {status}. "
+                f"Campaign ID: {campaign_id}"
+            )
 
     response_item = MessageItem(
         message_id="",  # or generate one if appropriate

{dhisana-0.0.1.dev263 → dhisana-0.0.1.dev264}/src/dhisana/utils/generate_leads_salesnav.py

@@ -215,6 +215,8 @@ async def generate_leads_code(
     )
 
     # Return dict + status
+    if status != "SUCCESS":
+        return {"workflow_python_code": ""}, status
     return response.model_dump(), status
 
 

{dhisana-0.0.1.dev263 → dhisana-0.0.1.dev264}/src/dhisana/utils/generate_structured_output_internal.py

@@ -3,6 +3,7 @@ import hashlib
 import json
 import logging
 import random
+from typing import Any, Dict, List, Optional, Tuple, Union
 
 from fastapi import HTTPException
 from pydantic import BaseModel
@@ -17,11 +18,16 @@ from dhisana.utils.fetch_openai_config import (
     _extract_config,
     create_async_openai_client,
 )
-from typing import Any, Dict, List, Optional, Tuple, Union
 
-from openai import OpenAIError, RateLimitError
-from pydantic import BaseModel
 
+# ──────────────────────────────────────────────────────────────────────────────
+# 1.  Helper functions
+# ──────────────────────────────────────────────────────────────────────────────
+
+def is_context_length_error(error: Exception) -> bool:
+    """Check if an error is due to context length being exceeded."""
+    error_str = str(error).lower()
+    return "context_length_exceeded" in error_str or "context window" in error_str
 
 
 # ──────────────────────────────────────────────────────────────────────────────
@@ -144,11 +150,17 @@ async def get_structured_output_internal(
                 completion = await _make_request()
                 break  # success → exit loop
             except (RateLimitError, OpenAIError) as e:
+                # Check for context length exceeded error
+                if is_context_length_error(e):
+                    logging.error(f"Context length exceeded: {e}")
+                    return f"Context length exceeded: {str(e)}", "CONTEXT_LENGTH_EXCEEDED"
+                
                 # Detect 429 / rate-limit
+                error_str = str(e).lower()
                 is_rl = (
                     isinstance(e, RateLimitError)
                     or getattr(e, "status_code", None) == 429
-                    or "rate_limit" in str(e).lower()
+                    or "rate_limit" in error_str
                 )
                 if is_rl and attempt < max_retries:
                     attempt += 1
@@ -161,7 +173,7 @@ async def get_structured_output_internal(
                     await asyncio.sleep(wait_time)
                     continue  # retry once
                 logging.error(f"OpenAI API error: {e}")
-                raise HTTPException(status_code=502, detail="Error communicating with the OpenAI API.")
+                return f"OpenAI API error: {str(e)}", "API_ERROR"
 
         # ─── handle model output (unchanged) ────────────────────────────────
         if completion and completion.output and len(completion.output) > 0:
@@ -203,12 +215,15 @@ async def get_structured_output_internal(
         else:
             return "No output returned", "FAIL"
 
+    # Safety fallback: catch any OpenAI errors not caught by inner retry loop
     except OpenAIError as e:
         logging.error(f"OpenAI API error: {e}")
-        raise HTTPException(status_code=502, detail="Error communicating with the OpenAI API.")
+        if is_context_length_error(e):
+            return f"Context length exceeded: {str(e)}", "CONTEXT_LENGTH_EXCEEDED"
+        return f"OpenAI API error: {str(e)}", "API_ERROR"
     except Exception as e:
         logging.error(f"Unexpected error: {e}")
-        raise HTTPException(status_code=500, detail="Unexpected server error.")
+        return f"Unexpected error: {str(e)}", "ERROR"
 
 
 
@@ -293,11 +308,17 @@ async def get_structured_output_with_mcp(
         return await client_async.responses.create(**kwargs)
 
     # ─── Retry once for 429s ──────────────────────────────────────────────────
+    completion = None
     for attempt in range(2):
         try:
             completion = await _make_request()
             break
         except (RateLimitError, OpenAIError) as exc:
+            # Check for context length exceeded error
+            if is_context_length_error(exc):
+                logging.error(f"Context length exceeded: {exc}")
+                return f"Context length exceeded: {str(exc)}", "CONTEXT_LENGTH_EXCEEDED"
+            
             if attempt == 0 and (
                 isinstance(exc, RateLimitError)
                 or getattr(exc, "status_code", None) == 429
@@ -308,9 +329,10 @@ async def get_structured_output_with_mcp(
                 await asyncio.sleep(sleep_for)
                 continue
             logging.error("OpenAI API error: %s", exc)
-            raise HTTPException(502, detail="Error communicating with the OpenAI API.") from exc
-    else:  # pragma: no cover
-        raise HTTPException(502, detail="OpenAI request retry loop failed.")
+            return f"OpenAI API error: {str(exc)}", "API_ERROR"
+    
+    if not completion:
+        return "OpenAI request retry loop failed", "API_ERROR"
 
     # ─── Parse the model’s structured output ──────────────────────────────────
     if not (completion and completion.output):
@@ -438,7 +460,7 @@ async def get_structured_output_with_assistant_and_vector_store(
                     break
 
             if not raw_text or not raw_text.strip():
-                raise HTTPException(status_code=502, detail="No response from the model.")
+                return "No response from the model", "FAIL"
 
             try:
                 parsed_obj = response_format.parse_raw(raw_text)
@@ -454,9 +476,12 @@ async def get_structured_output_with_assistant_and_vector_store(
         else:
             return "No output returned", "FAIL"
 
+    # Safety fallback: catch any errors not caught during API call
     except OpenAIError as e:
         logging.error(f"OpenAI API error: {e}")
-        raise HTTPException(status_code=502, detail="Error communicating with the OpenAI API.")
+        if is_context_length_error(e):
+            return f"Context length exceeded: {str(e)}", "CONTEXT_LENGTH_EXCEEDED"
+        return f"OpenAI API error: {str(e)}", "API_ERROR"
     except Exception as e:
         logging.error(f"Unexpected error: {e}")
-        raise HTTPException(status_code=500, detail="Unexpected server error.")
+        return f"Unexpected error: {str(e)}", "ERROR"

{dhisana-0.0.1.dev263 → dhisana-0.0.1.dev264}/src/dhisana.egg-info/PKG-INFO

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