dhisana 0.0.1.dev250__tar.gz → 0.0.1.dev251__tar.gz

{dhisana-0.0.1.dev250 → dhisana-0.0.1.dev251}/PKG-INFO

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

{dhisana-0.0.1.dev250 → dhisana-0.0.1.dev251}/setup.py

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

{dhisana-0.0.1.dev250 → dhisana-0.0.1.dev251}/src/dhisana/utils/cache_output_tools.py

@@ -2,6 +2,7 @@ import os
 import hashlib
 import json
 import logging
+from datetime import datetime, timezone
 
 from azure.storage.blob import BlobServiceClient
 from azure.core.exceptions import ResourceNotFoundError, AzureError
@@ -49,10 +50,11 @@ def cache_output(tool_name: str, key: str, value, ttl: int = None) -> bool:
     # Construct the blob name using a virtual folder for the tool name
     blob_name = f"{tool_name}/{key_hash}.json"
 
-    # Prepare the cache data
+    # Prepare the cache data with timestamp for TTL expiration checking
     cache_data = {
         "value": value,
-        "ttl": ttl
+        "ttl": ttl,
+        "cached_at": datetime.now(timezone.utc).isoformat()
     }
     data = json.dumps(cache_data)
 
@@ -88,6 +90,24 @@ def retrieve_output(tool_name: str, key: str):
         download_stream = blob_client.download_blob()
         content = download_stream.readall()  # content is in bytes
         cache_data = json.loads(content.decode("utf-8"))
+        
+        # Check if TTL has expired
+        ttl = cache_data.get("ttl")
+        cached_at = cache_data.get("cached_at")
+        
+        if ttl is not None and cached_at is not None:
+            try:
+                cached_time = datetime.fromisoformat(cached_at)
+                now = datetime.now(timezone.utc)
+                elapsed_seconds = (now - cached_time).total_seconds()
+                if elapsed_seconds > ttl:
+                    logger.info(f"Cache expired for blob '{blob_name}' (elapsed: {elapsed_seconds}s, ttl: {ttl}s)")
+                    return None
+            except (ValueError, TypeError) as e:
+                logger.warning(f"Error parsing cached_at timestamp: {e}")
+                # If we can't parse the timestamp, treat as expired for safety
+                return None
+        
         return cache_data.get("value")
     except ResourceNotFoundError:
         # Blob does not exist

{dhisana-0.0.1.dev250 → dhisana-0.0.1.dev251}/src/dhisana/utils/proxy_curl_tools.py

@@ -16,6 +16,9 @@ from urllib.parse import urlparse, urlunparse
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+# Cache TTL for Proxycurl responses: 14 days in seconds
+PROXYCURL_CACHE_TTL = 14 * 24 * 60 * 60  # 1,209,600 seconds
+
 
 def get_proxycurl_access_token(tool_config: Optional[List[Dict]] = None) -> str:
     """
@@ -117,7 +120,7 @@ async def enrich_person_info_from_proxycurl(
                 if response.status == 200:
                     result = await response.json()
                     if linkedin_url:
-                        cache_output("enrich_person_info_from_proxycurl", linkedin_url, result)
+                        cache_output("enrich_person_info_from_proxycurl", linkedin_url, result, ttl=PROXYCURL_CACHE_TTL)
                     logger.info("Successfully retrieved person info from Proxycurl.")
                     return result
                 elif response.status == 404:
@@ -195,14 +198,14 @@ async def lookup_person_in_proxy_curl_by_name(
                 logger.debug(f"Received response status: {response.status}")
                 if response.status == 200:
                     result = await response.json()
-                    cache_output("lookup_person_in_proxycurl_by_name", key, result)
+                    cache_output("lookup_person_in_proxycurl_by_name", key, result, ttl=PROXYCURL_CACHE_TTL)
                     logger.info("Successfully retrieved person search info from Proxycurl.")
                     return result
                 elif response.status == 404:
                     msg = "Person not found"
                     logger.warning(msg)
                     if key:
-                        cache_output("lookup_person_in_proxycurl_by_name", key, {'error': msg})
+                        cache_output("lookup_person_in_proxycurl_by_name", key, {'error': msg}, ttl=PROXYCURL_CACHE_TTL)
                     return {'error': msg}
                 elif response.status == 429:
                     msg = "Rate limit exceeded"
@@ -415,7 +418,7 @@ async def enrich_organization_info_from_proxycurl(
                     if response.status == 200:
                         result = await response.json()
                         transformed_result = transform_company_data(result)
-                        cache_output("enrich_organization_info_from_proxycurl", cache_key, transformed_result)
+                        cache_output("enrich_organization_info_from_proxycurl", cache_key, transformed_result, ttl=PROXYCURL_CACHE_TTL)
                         logger.info("Successfully retrieved and transformed organization info from Proxycurl by LinkedIn URL.")
                         return transformed_result
                     elif response.status == 429:
@@ -429,7 +432,7 @@ async def enrich_organization_info_from_proxycurl(
                             f"Proxycurl organization profile not found for LinkedIn URL {standardized_url}: {error_text}"
                         )
                         cache_output(
-                            "enrich_organization_info_from_proxycurl", cache_key, {}
+                            "enrich_organization_info_from_proxycurl", cache_key, {}, ttl=PROXYCURL_CACHE_TTL
                         )
                         return {}
                     else:
@@ -483,7 +486,7 @@ async def enrich_organization_info_from_proxycurl(
                                     if profile_response.status == 200:
                                         result = await profile_response.json()
                                         transformed_result = transform_company_data(result)
-                                        cache_output("enrich_organization_info_from_proxycurl", domain_cache_key, transformed_result)
+                                        cache_output("enrich_organization_info_from_proxycurl", domain_cache_key, transformed_result, ttl=PROXYCURL_CACHE_TTL)
                                         logger.info("Successfully retrieved and transformed organization info from Proxycurl by domain.")
                                         return transformed_result
                                     elif profile_response.status == 429:
@@ -509,7 +512,7 @@ async def enrich_organization_info_from_proxycurl(
                     elif response.status == 404:
                         msg = "Item not found"
                         logger.warning(msg)
-                        cache_output("enrich_organization_info_from_proxycurl", domain_cache_key, {})
+                        cache_output("enrich_organization_info_from_proxycurl", domain_cache_key, {}, ttl=PROXYCURL_CACHE_TTL)
                         return {}
                     else:
                         error_text = await response.text()
@@ -572,7 +575,7 @@ async def enrich_job_info_from_proxycurl(
                 logger.debug(f"Received response status: {response.status}")
                 if response.status == 200:
                     result = await response.json()
-                    cache_output("enrich_job_info_from_proxycurl", job_url, result)
+                    cache_output("enrich_job_info_from_proxycurl", job_url, result, ttl=PROXYCURL_CACHE_TTL)
                     logger.info("Successfully retrieved job info from Proxycurl.")
                     return result
                 elif response.status == 429:
@@ -583,7 +586,7 @@ async def enrich_job_info_from_proxycurl(
                 elif response.status == 404:
                     msg = "Job not found"
                     logger.warning(msg)
-                    cache_output("enrich_job_info_from_proxycurl", job_url, {'error': msg})
+                    cache_output("enrich_job_info_from_proxycurl", job_url, {'error': msg}, ttl=PROXYCURL_CACHE_TTL)
                     return {'error': msg}
                 else:
                     error_text = await response.text()

{dhisana-0.0.1.dev250 → dhisana-0.0.1.dev251}/src/dhisana.egg-info/PKG-INFO

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