dhisana 0.0.1.dev101__tar.gz → 0.0.1.dev102__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dhisana
-Version: 0.0.1.dev101
+Version: 0.0.1.dev102
 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.dev102}/setup.py

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

{dhisana-0.0.1.dev101 → dhisana-0.0.1.dev102}/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.dev102}/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.dev102}/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"""
@@ -69,7 +92,7 @@ async def research_lead_with_full_info_ai(
         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-4o",
         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.dev102}/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.dev102
 Summary: A Python SDK for Dhisana AI Platform
 Home-page: https://github.com/dhisana-ai/dhisana-python-sdk
 Author: Admin