dhisana 0.0.1.dev101__tar.gz → 0.0.1.dev103__tar.gz

{dhisana-0.0.1.dev101 → dhisana-0.0.1.dev103}/PKG-INFO

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

{dhisana-0.0.1.dev101 → dhisana-0.0.1.dev103}/setup.py

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

{dhisana-0.0.1.dev101 → dhisana-0.0.1.dev103}/src/dhisana/utils/apollo_tools.py

@@ -192,6 +192,16 @@ async def lookup_person_in_apollo_by_name(
         "per_page": 10
     }
 
+    # Build a cache key that includes full_name and company_name (if provided)
+    # so that results are correctly cached and retrieved.
+    key_item = f"lookup_person_in_apollo_by_name_{full_name}_{company_name or ''}".lower()
+
+    # Attempt to retrieve a cached response first
+    cached_response = retrieve_output("lookup_person_in_apollo_by_name", key_item)
+    if cached_response is not None:
+        logger.info(f"Cache hit for user: {full_name}, company: {company_name or ''}")
+        return cached_response
+
     url = 'https://api.apollo.io/api/v1/mixed_people/search'
     logger.debug(f"Making request to Apollo with payload: {data}")
 
@@ -202,6 +212,7 @@ async def lookup_person_in_apollo_by_name(
                 if response.status == 200:
                     result = await response.json()
                     logger.info("Successfully looked up person by name on Apollo.")
+                    cache_output("lookup_person_in_apollo_by_name", key_item, result)
                     return result
                 elif response.status == 429:
                     msg = "Rate limit exceeded"
@@ -222,7 +233,6 @@ async def lookup_person_in_apollo_by_name(
             logger.exception("Exception occurred while looking up person by name.")
             return {'error': str(e)}
 
-
 @assistant_tool
 @backoff.on_exception(
     backoff.expo,

{dhisana-0.0.1.dev101 → dhisana-0.0.1.dev103}/src/dhisana/utils/check_email_validity_tools.py

@@ -220,6 +220,18 @@ async def check_email_validity_with_hunter(
             "confidence": "low",
             "is_valid": False
         }
+    
+    cache_key = f"{email_id}"
+    cached_response = retrieve_output("hunter_validate", cache_key)
+    if cached_response is not None:
+        logger.info("Cache hit for Hunter validate.")
+        if not cached_response:
+            return {
+                "email": email_id,
+                "confidence": "low",
+                "is_valid": False
+            }
+        return json.loads(cached_response[0])
 
     HUNTER_API_KEY = get_hunter_access_token(tool_config)
     if not HUNTER_API_KEY:
@@ -241,11 +253,14 @@ async def check_email_validity_with_hunter(
                 if response.status != 200:
                     content = await safe_read_json_or_text(response)
                     logger.warning(f"[Hunter] Non-200 status: {response.status} => {content}")
-                    return {
+                    
+                    final_response = {
                         "email": email_id,
                         "confidence": "low",
                         "is_valid": False
                     }
+                    cache_output("hunter_validate", cache_key, [json.dumps(final_response)])
+                    return final_response
 
                 result = await response.json()
     except Exception as ex:
@@ -266,6 +281,7 @@ async def check_email_validity_with_hunter(
         "confidence": confidence,
         "is_valid": is_valid
     }
+    cache_output("hunter_validate", cache_key, [json.dumps(final_response)])
     logger.info("Exiting check_email_validity_with_hunter.")
     return final_response
 

{dhisana-0.0.1.dev101 → dhisana-0.0.1.dev103}/src/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-4o",
+        model="gpt-4.1",
         tool_config=tool_config
     )
     if status == "ERROR":
@@ -411,7 +411,7 @@ async def get_user_linkedin_url_from_github_profile(
     response, status = await get_structured_output_internal(
         instructions, 
         UserInfoFromGithubProfileId, 
-        model="gpt-4o", 
+        model="gpt-4.1", 
         use_web_search=True,
         tool_config=tool_config
     )
@@ -783,7 +783,7 @@ async def get_company_domain_from_llm_web_search(
         response, status = await get_structured_output_internal(
             instructions,
             CompanyInfoFromName,
-            model="gpt-4o",
+            model="gpt-4.1",
             use_web_search=True,
             tool_config=tool_config
         )

{dhisana-0.0.1.dev101 → dhisana-0.0.1.dev103}/src/dhisana/utils/generate_email.py

@@ -133,6 +133,7 @@ async def generate_personalized_email_copy(
         - 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.
     """
 
     # Check if a vector store is available

{dhisana-0.0.1.dev101 → dhisana-0.0.1.dev103}/src/dhisana/utils/generate_email_response.py

@@ -106,6 +106,10 @@ async def generate_inbound_email_response_copy(
     DO NOT share any link to internal or made up doucment. You can attach or send any document.
     If the user is asking for any document point them to organization's website found in sender information if available:
     {campaign_context.sender_info.model_dump()} 
+    
+    Use conversational name for company name. 
+    Use conversational name when using lead first name.
+    Do not use special characters or spaces when use leads first name. 
 
     Your final output must be valid JSON with the structure:
     {{

{dhisana-0.0.1.dev101 → dhisana-0.0.1.dev103}/src/dhisana/utils/generate_linkedin_connect_message.py

@@ -25,7 +25,6 @@ from dhisana.utils.assistant_tool_tag import assistant_tool
 # LinkedIn Connection Message Schema
 # ----------------------------------------------------------------------
 class LinkedInConnectMessage(BaseModel):
-    subject: str
     body: str
 
 # ----------------------------------------------------------------------
@@ -125,11 +124,12 @@ async def generate_personalized_linkedin_copy(
         LinkedIn Thread: {conversation_data.current_linkedin_thread or ''}
 
         IMPORTANT REQUIREMENTS:
-        - Output must be JSON with "subject" and "body" fields only.
-        - The entire message must be under 40 words total (subject + body).
+        - Output must be JSON "body" of the message.
+        - The entire message body must be under 40 words total.
         - No placeholders or extra instructions in the final output.
         - Do not include personal addresses, IDs, or irrelevant internal data.
         - Linked in message always has saluatation Hi <First Name>, unless specified otherwise.
+        - <First Name> is the first name of the lead. Its conversational name. It does not have any special characters or spaces.
         - User conversational name for company name if used.
     """
 
@@ -166,7 +166,7 @@ async def generate_personalized_linkedin_copy(
         receiver_name=lead_data.full_name or "",
         receiver_email=lead_data.email or "",
         iso_datetime=datetime.utcnow().isoformat(),
-        subject=response_data.subject,
+        subject="Hi",
         body=response_data.body
     )
 

{dhisana-0.0.1.dev101 → dhisana-0.0.1.dev103}/src/dhisana/utils/generate_linkedin_response_message.py

@@ -69,9 +69,24 @@ async def generate_linkedin_response_message_copy(
     ]
     
     cleaned_context = cleanup_reply_linkedin_context(linkedin_context)
-    if not cleaned_context.current_conversation_context.current_email_thread:
-        cleaned_context.current_conversation_context.current_email_thread = []
-
+    
+    # Safely handle the current_conversation_context if it exists.
+    if cleaned_context.current_conversation_context:
+        if not cleaned_context.current_conversation_context.current_email_thread:
+            cleaned_context.current_conversation_context.current_email_thread = []
+        
+        if not cleaned_context.current_conversation_context.current_linkedin_thread:
+            cleaned_context.current_conversation_context.current_linkedin_thread = []
+        
+        # Safely extract the conversation thread for prompt formatting.
+        conversation_thread_dump = [
+            thread_item.model_dump()
+            for thread_item in cleaned_context.current_conversation_context.current_linkedin_thread
+        ]
+    else:
+        # If current_conversation_context is None, use an empty thread.
+        conversation_thread_dump = []
+    
     current_date_iso = datetime.datetime.now().isoformat()
     prompt = f"""
     You are a specialized LinkedIn assistant.
@@ -89,7 +104,7 @@ async def generate_linkedin_response_message_copy(
     {variation}
 
     1. Understand the conversation thread:
-       {[thread_item.model_dump() for thread_item in cleaned_context.current_conversation_context.current_linkedin_thread] if cleaned_context.current_conversation_context.current_linkedin_thread else []}
+       {conversation_thread_dump}
 
     2. User & Company (Lead) Info:
        {cleaned_context.model_dump()}

{dhisana-0.0.1.dev101 → dhisana-0.0.1.dev103}/src/dhisana/utils/openai_assistant_and_file_utils.py

@@ -190,7 +190,7 @@ async def delete_files(
 async def run_file_search(
     query: str,
     vector_store_id: str,
-    model: str = "gpt-4o",
+    model: str = "gpt-4.1",
     max_num_results: int = 5,
     store: bool = True,
     tool_config: Optional[List[Dict]] = None
@@ -260,7 +260,7 @@ async def run_file_search(
 
 async def run_response_text(
     prompt: str,
-    model: str = "gpt-4o",
+    model: str = "gpt-4.1",
     max_tokens: int = 2048,
     store: bool = True,
     tool_config: Optional[List[Dict]] = None
@@ -288,7 +288,7 @@ async def run_response_text(
 async def run_response_structured(
     prompt: str,
     response_format: dict,
-    model: str = "gpt-4o",
+    model: str = "gpt-4.1",
     max_tokens: int = 1024,
     store: bool = True,
     tool_config: Optional[List[Dict]] = None

{dhisana-0.0.1.dev101 → dhisana-0.0.1.dev103}/src/dhisana/utils/research_lead.py

@@ -8,6 +8,23 @@ def clean_nul_bytes(s: str) -> str:
     s = s.replace('```markdown', '')
     return s.replace('\x00', '')
 
+def _remove_excluded_fields(data: Dict) -> Dict:
+    """
+    Return a copy of `data` that excludes keys named 'id'
+    or that end in '_by', '_id', '_to', or '_at'.
+    """
+    excluded_keys = {"id"}
+    excluded_endings = ["_by", "_id", "_to", "_at", "_status", "research_summary"]
+
+    cleaned = {}
+    for k, v in data.items():
+        if k in excluded_keys:
+            continue
+        if any(k.endswith(suffix) for suffix in excluded_endings):
+            continue
+        cleaned[k] = v
+    return cleaned
+
 class LeadResearchInformation(BaseModel):
     research_summary: str
 
@@ -20,7 +37,13 @@ async def research_lead_with_full_info_ai(
     """
     Research on lead provided given input. Provide Detailed Summary.
     """
+    # Clean user properties (e.g. remove newlines, sanitize strings, etc.)
     user_properties = cleanup_email_context(user_properties)
+
+    # Remove excluded fields from user_properties
+    user_properties = _remove_excluded_fields(user_properties)
+
+    # Optionally remove any known keys that should not appear (e.g. 'date_extracted')
     user_properties.pop("date_extracted", None)
 
     instructions = f"""
@@ -62,14 +85,14 @@ async def research_lead_with_full_info_ai(
     response, status = await get_structured_output_internal(
         instructions, 
         LeadResearchInformation, 
-        model="gpt-4o", 
+        model="gpt-4.1", 
         tool_config=tool_config
     )
     if status == "SUCCESS":
         response.research_summary = clean_nul_bytes(response.research_summary)
         return response.model_dump()
     else:
-        {"research_summary", ""}
+        return {"research_summary": ""}
 
 # --------------------------------------------
 # COMPANY-RELATED MODELS & FUNCTION (FIXED)
@@ -94,9 +117,12 @@ async def research_company_with_full_info_ai(
     Returns:
     dict: The JSON response containing the detailed research summary of the company.
     """
+    # Clean company properties (e.g. remove newlines, sanitize strings, etc.)
     company_properties = cleanup_email_context(company_properties)
 
-    # Build the LLM instructions for the company research
+    # Remove excluded fields from company_properties
+    company_properties = _remove_excluded_fields(company_properties)
+
     instructions = f"""
         Please read the following company information and instructions, then produce a detailed summary of the company in the specified format.
         ---
@@ -130,11 +156,10 @@ async def research_company_with_full_info_ai(
             "research_summary": "Detailed summary about the company. The summary should be neatly formatted in GitHub-Flavored Markdown, and include all the key information from the listed sections."
         }}
     """
-
     response, status = await get_structured_output_internal(
-        instructions, 
-        CompanyResearchInformation, 
-        model="gpt-4o", 
+        instructions,
+        CompanyResearchInformation,
+        model="gpt-4.1",
         use_web_search=True,
         tool_config=tool_config
     )
@@ -142,4 +167,4 @@ async def research_company_with_full_info_ai(
         response.research_summary = clean_nul_bytes(response.research_summary)
         return response.model_dump()
     else:
-        return {"research_summary", ""}
+        return {"research_summary": ""}

{dhisana-0.0.1.dev101 → dhisana-0.0.1.dev103}/src/dhisana/utils/test_connect.py

@@ -320,7 +320,7 @@ async def test_connectivity(tool_config: List[Dict[str, Any]]) -> Dict[str, Dict
 
         # Special handling for openai
         if tool_name == "openai":
-            model_name = next((c["value"] for c in config_entries if c["name"] == "modelName"), "gpt-4o")
+            model_name = next((c["value"] for c in config_entries if c["name"] == "modelName"), "gpt-4.1")
             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-0.0.1.dev101 → dhisana-0.0.1.dev103}/src/dhisana.egg-info/PKG-INFO

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