dhisana 0.0.1.dev220__tar.gz → 0.0.1.dev222__tar.gz

{dhisana-0.0.1.dev220 → dhisana-0.0.1.dev222}/PKG-INFO

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

{dhisana-0.0.1.dev220 → dhisana-0.0.1.dev222}/setup.py

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

{dhisana-0.0.1.dev220 → dhisana-0.0.1.dev222}/src/dhisana/schemas/sales.py

@@ -49,6 +49,7 @@ class Lead(BaseModel):
     engaged: bool = False
     last_contact: Optional[int] = None
     research_summary: Optional[str] = None
+    task_ids: Optional[List[str]] = None
     email_validation_status: Optional[str] = None
     linkedin_validation_status: Optional[str] = None
     research_status: Optional[str] = None

{dhisana-0.0.1.dev220 → dhisana-0.0.1.dev222}/src/dhisana/utils/apollo_tools.py

@@ -573,7 +573,7 @@ async def search_leads_with_apollo(
             # Important: handle personNotTitles as well
             "personNotTitles": "person_not_titles",
 
-            "qOrganizationJobTitles": "q_keywords",
+            "qOrganizationJobTitles": "q_organization_job_titles",
             "sortAscending": "sort_ascending",
             "sortByField": "sort_by_field",
             "contactEmailStatusV2": "contact_email_status",
@@ -646,6 +646,8 @@ async def search_leads_with_apollo(
                     "organization_ids",
                     "organization_num_employees_ranges",
                     "person_not_titles",  # <--- added so single item is forced into list
+                    "q_organization_job_titles",
+                    "organization_latest_funding_stage_cd",
                 ):
                     if isinstance(final_value, str):
                         final_value = [final_value]
@@ -675,6 +677,10 @@ async def search_leads_with_apollo(
             "page": 1,
             "per_page": min(max_items, 100),
         }
+        if query.job_openings_with_titles:
+            dynamic_payload["q_organization_job_titles"] = query.job_openings_with_titles
+        if query.latest_funding_stages:
+            dynamic_payload["organization_latest_funding_stage_cd"] = query.latest_funding_stages
         if query.sort_by_field is not None:
             dynamic_payload["sort_by_field"] = query.sort_by_field
         if query.sort_ascending is not None:
@@ -785,7 +791,7 @@ async def search_leads_with_apollo_page(
             "organizationNumEmployeesRanges": "organization_num_employees_ranges",
             "personTitles": "person_titles",
             "personNotTitles": "person_not_titles",
-            "qOrganizationJobTitles": "q_keywords",
+            "qOrganizationJobTitles": "q_organization_job_titles",
             "sortAscending": "sort_ascending",
             "sortByField": "sort_by_field",
             "contactEmailStatusV2": "contact_email_status",
@@ -843,6 +849,8 @@ async def search_leads_with_apollo_page(
                     "organization_ids",
                     "organization_num_employees_ranges",
                     "person_not_titles",
+                    "q_organization_job_titles",
+                    "organization_latest_funding_stage_cd",
                 ):
                     if isinstance(final_value, str):
                         final_value = [final_value]
@@ -866,6 +874,10 @@ async def search_leads_with_apollo_page(
                 or [f"{query.min_employees_in_organization or 1},{query.max_employees_in_organization or 1000}"]
             ),
         }
+        if query.job_openings_with_titles:
+            dynamic_payload["q_organization_job_titles"] = query.job_openings_with_titles
+        if query.latest_funding_stages:
+            dynamic_payload["organization_latest_funding_stage_cd"] = query.latest_funding_stages
         if query.sort_by_field is not None:
             dynamic_payload["sort_by_field"] = query.sort_by_field
         if query.sort_ascending is not None:

{dhisana-0.0.1.dev220 → dhisana-0.0.1.dev222}/src/dhisana/utils/test_connect.py

@@ -296,6 +296,68 @@ async def test_proxycurl(api_key: str) -> Dict[str, Any]:
         return {"success": False, "status_code": 0, "error_message": str(e)}
 
 
+async def test_exa(api_key: str) -> Dict[str, Any]:
+    """Verify Exa connectivity by issuing a minimal search request."""
+
+    url = "https://api.exa.ai/search"
+    headers = {
+        "x-api-key": api_key,
+        "Content-Type": "application/json",
+        "Accept": "application/json",
+    }
+    payload = {
+        "query": "Dhisana connectivity check",
+        "numResults": 1,
+    }
+
+    try:
+        timeout = aiohttp.ClientTimeout(total=10)
+        async with aiohttp.ClientSession(timeout=timeout) as session:
+            async with session.post(url, headers=headers, json=payload) as response:
+                status = response.status
+                data = await safe_json(response)
+
+                if status != 200:
+                    err_message = None
+                    if isinstance(data, dict):
+                        err_message = (
+                            data.get("message")
+                            or data.get("error")
+                            or data.get("detail")
+                        )
+                        if isinstance(err_message, dict):
+                            err_message = err_message.get("message") or str(err_message)
+                    return {
+                        "success": False,
+                        "status_code": status,
+                        "error_message": err_message or f"Non-200 from Exa: {status}",
+                    }
+
+                if isinstance(data, dict):
+                    if "error" in data:
+                        error_value = data["error"]
+                        if isinstance(error_value, dict):
+                            error_value = error_value.get("message") or str(error_value)
+                        return {
+                            "success": False,
+                            "status_code": status,
+                            "error_message": str(error_value),
+                        }
+
+                    results = data.get("results")
+                    if isinstance(results, list):
+                        return {"success": True, "status_code": status, "error_message": None}
+
+                return {
+                    "success": False,
+                    "status_code": status,
+                    "error_message": "Unexpected response from Exa API.",
+                }
+    except Exception as exc:
+        logger.error(f"Exa test failed: {exc}")
+        return {"success": False, "status_code": 0, "error_message": str(exc)}
+
+
 async def test_apollo(api_key: str) -> Dict[str, Any]:
     organization_domain = 'microsoft.com'
     url = f'https://api.apollo.io/api/v1/organizations/enrich?domain={organization_domain}'
@@ -1179,6 +1241,7 @@ async def test_connectivity(tool_config: List[Dict[str, Any]]) -> Dict[str, Dict
         "serpapi":          test_serpapi,
         "serperdev":        test_serperdev,
         "proxycurl":        test_proxycurl,
+        "exa":             test_exa,
         "apollo":           test_apollo,
         "hubspot":          test_hubspot,
         "github":           test_github,

{dhisana-0.0.1.dev220 → dhisana-0.0.1.dev222}/src/dhisana.egg-info/PKG-INFO

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

{dhisana-0.0.1.dev220 → dhisana-0.0.1.dev222}/src/dhisana.egg-info/SOURCES.txt

@@ -103,6 +103,7 @@ src/dhisana/workflow/task.py
 src/dhisana/workflow/test.py
 tests/test_agent_tools.py
 tests/test_apollo_company_search.py
+tests/test_apollo_lead_search.py
 tests/test_connectivity.py
 tests/test_google_document.py
 tests/test_hubspot_call_logs.py

dhisana-0.0.1.dev222/tests/test_apollo_lead_search.py

@@ -0,0 +1,119 @@
+import pytest
+from unittest.mock import AsyncMock, patch
+
+from src.dhisana.schemas.sales import LeadsQueryFilters
+from src.dhisana.utils.apollo_tools import search_leads_with_apollo_page
+
+
+@pytest.fixture
+def apollo_people_response():
+    return {
+        "pagination": {
+            "page": 1,
+            "per_page": 25,
+            "total_entries": 1,
+            "total_pages": 1,
+        },
+        "people": [
+            {
+                "id": "person_1",
+                "name": "Alex Example",
+                "first_name": "Alex",
+                "last_name": "Example",
+                "title": "VP Sales",
+                "headline": "Revenue leader",
+                "email": "alex@example.com",
+                "linkedin_url": "https://linkedin.com/in/alex-example",
+                "city": "San Francisco",
+                "state": "California",
+                "organization": {
+                    "name": "Example Inc",
+                    "primary_domain": "example.com",
+                    "linkedin_url": "https://linkedin.com/company/example",
+                    "website_url": "https://example.com",
+                    "keywords": ["SaaS", "Sales"],
+                },
+                "contact": {"sanitized_phone": "+1-555-0100"},
+            }
+        ],
+        "contacts": [],
+    }
+
+
+@pytest.fixture
+def tool_config():
+    return [
+        {
+            "name": "apollo",
+            "configuration": [
+                {"name": "apiKey", "value": "test_api_key"},
+            ],
+        }
+    ]
+
+
+@pytest.mark.asyncio
+async def test_search_leads_with_apollo_page_parses_url_filters(apollo_people_response, tool_config):
+    example_url = (
+        "https://app.apollo.io/#/people?page=1&personTitles[]=VP%20Sales&"
+        "organizationLatestFundingStageCd[]=3&organizationLatestFundingStageCd[]=4&"
+        "organizationLatestFundingStageCd[]=5&qOrganizationJobTitles[]=SDR&"
+        "qOrganizationJobTitles[]=account%20executive"
+    )
+
+    with patch(
+        "src.dhisana.utils.apollo_tools.fetch_apollo_data",
+        new=AsyncMock(return_value=apollo_people_response),
+    ) as mock_fetch_data, patch(
+        "src.dhisana.utils.apollo_tools.get_apollo_access_token",
+        return_value="test_api_key",
+    ):
+        result = await search_leads_with_apollo_page(
+            query=LeadsQueryFilters(),
+            page=1,
+            per_page=25,
+            example_url=example_url,
+            tool_config=tool_config,
+        )
+
+    assert result["current_page"] == 1
+    assert result["total_entries"] == 1
+    assert len(result["results"]) == 1
+
+    payload = mock_fetch_data.call_args[0][3]
+    assert payload["q_organization_job_titles"] == ["SDR", "account executive"]
+    assert payload["organization_latest_funding_stage_cd"] == ["3", "4", "5"]
+
+
+@pytest.mark.asyncio
+async def test_search_leads_with_apollo_page_uses_query_filters(apollo_people_response, tool_config):
+    query = LeadsQueryFilters(
+        person_current_titles=["Head of Sales"],
+        job_openings_with_titles=["SDR", "Account Executive"],
+        latest_funding_stages=["3", "4"],
+        organization_num_employees_ranges=["201,500"],
+    )
+
+    with patch(
+        "src.dhisana.utils.apollo_tools.fetch_apollo_data",
+        new=AsyncMock(return_value=apollo_people_response),
+    ) as mock_fetch_data, patch(
+        "src.dhisana.utils.apollo_tools.get_apollo_access_token",
+        return_value="test_api_key",
+    ):
+        result = await search_leads_with_apollo_page(
+            query=query,
+            page=2,
+            per_page=50,
+            tool_config=tool_config,
+        )
+
+    assert result["current_page"] == 1
+    assert len(result["results"]) == 1
+
+    payload = mock_fetch_data.call_args[0][3]
+    assert payload["page"] == 2
+    assert payload["per_page"] == 50
+    assert payload["person_titles"] == ["Head of Sales"]
+    assert payload["q_organization_job_titles"] == ["SDR", "Account Executive"]
+    assert payload["organization_latest_funding_stage_cd"] == ["3", "4"]