dhisana 0.0.1.dev252__tar.gz → 0.0.1.dev254__tar.gz

{dhisana-0.0.1.dev252 → dhisana-0.0.1.dev254}/PKG-INFO

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

{dhisana-0.0.1.dev252 → dhisana-0.0.1.dev254}/setup.py

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

{dhisana-0.0.1.dev252 → dhisana-0.0.1.dev254}/src/dhisana/schemas/sales.py

@@ -728,7 +728,10 @@ class LeadsQueryFilters(BaseModel):
     # CHANGED: Renamed to be more descriptive
     industries: Optional[List[str]] = Field(
         default=None,
-        description="List of organization industries. Maps to organization_industries in Apollo."
+        description=(
+            "List of organization industries (case sensitive). "
+            "Maps to organization_industries in Apollo."
+        )
     )
 
     # Potential existing fields
@@ -800,7 +803,10 @@ class LeadsQueryFilters(BaseModel):
     # CHANGED: Renamed for consistency
     company_industry_tag_ids: Optional[List[str]] = Field(
         default=None,
-        description="List of industry tag IDs, e.g. ['5567cd4773696439b10b0000']. Maps to organization_industry_tag_ids."
+        description=(
+            "List of industry tag IDs, e.g. ['5567cd4773696439b10b0000']. "
+            "Maps to organization_industry_tag_ids."
+        )
     )
     
     q_organization_keyword_tags: Optional[List[str]] = Field(

{dhisana-0.0.1.dev252 → dhisana-0.0.1.dev254}/src/dhisana/utils/apollo_tools.py

@@ -547,6 +547,7 @@ async def search_leads_with_apollo(
             "revenueRange[max]": "revenue_range_max",
             "revenueRange[min]": "revenue_range_min",
             "currentlyUsingAnyOfTechnologyUids": "currently_using_any_of_technology_uids",
+            "organizationIndustries": "organization_industries",
             "organizationIndustryTagIds": "organization_industry_tag_ids",
             "notOrganizationIds": "not_organization_ids",
             "qOrganizationDomainsList": "q_organization_domains_list",
@@ -614,6 +615,8 @@ async def search_leads_with_apollo(
                     "person_not_titles",  # <--- added so single item is forced into list
                     "q_organization_job_titles",
                     "organization_latest_funding_stage_cd",
+                    "organization_industries",
+                    "organization_industry_tag_ids",
                 ):
                     if isinstance(final_value, str):
                         final_value = [final_value]
@@ -664,6 +667,10 @@ async def search_leads_with_apollo(
             dynamic_payload["sort_ascending"] = query.sort_ascending
         if query.person_seniorities:
             dynamic_payload["person_seniorities"] = query.person_seniorities
+        if query.industries:
+            dynamic_payload["organization_industries"] = query.industries
+        if query.company_industry_tag_ids:
+            dynamic_payload["organization_industry_tag_ids"] = query.company_industry_tag_ids
         # Add company domains to include in search
         if query.company_domains:
             dynamic_payload["q_organization_domains_list"] = query.company_domains
@@ -782,6 +789,7 @@ async def search_leads_with_apollo_page(
             "revenueRange[max]": "revenue_range_max",
             "revenueRange[min]": "revenue_range_min",
             "currentlyUsingAnyOfTechnologyUids": "currently_using_any_of_technology_uids",
+            "organizationIndustries": "organization_industries",
             "organizationIndustryTagIds": "organization_industry_tag_ids",
             "notOrganizationIds": "not_organization_ids",
             "qOrganizationDomainsList": "q_organization_domains_list",
@@ -834,6 +842,8 @@ async def search_leads_with_apollo_page(
                     "person_not_titles",
                     "q_organization_job_titles",
                     "organization_latest_funding_stage_cd",
+                    "organization_industries",
+                    "organization_industry_tag_ids",
                 ):
                     if isinstance(final_value, str):
                         final_value = [final_value]
@@ -880,6 +890,10 @@ async def search_leads_with_apollo_page(
             
         if query.q_not_organization_keyword_tags:
             dynamic_payload["q_not_organization_keyword_tags"] = query.q_not_organization_keyword_tags
+        if query.industries:
+            dynamic_payload["organization_industries"] = query.industries
+        if query.company_industry_tag_ids:
+            dynamic_payload["organization_industry_tag_ids"] = query.company_industry_tag_ids
 
         # Add company domains to include in search (Apollo API: q_organization_domains_list[])
         if query.company_domains:

{dhisana-0.0.1.dev252 → dhisana-0.0.1.dev254}/src/dhisana/utils/test_connect.py

@@ -1714,6 +1714,82 @@ async def test_theorg(api_key: str) -> Dict[str, Any]:
         return {"success": False, "status_code": 0, "error_message": str(exc)}
 
 
+###############################################################################
+#                            CORESIGNAL CONNECTIVITY
+###############################################################################
+
+async def test_coresignal(api_key: str) -> Dict[str, Any]:
+    """
+    Connectivity test for Coresignal using the Multi-source Jobs API search preview.
+
+    Uses a lightweight search query to verify API key validity.
+    Reference: https://docs.coresignal.com/jobs-api/multi-source-jobs-api/search-preview
+    """
+    url = "https://api.coresignal.com/cdapi/v2/job_multi_source/search/es_dsl/preview"
+    headers = {
+        "apikey": api_key,
+        "Accept": "application/json",
+        "Content-Type": "application/json",
+    }
+    # Minimal query to check connectivity
+    payload = {
+        "query": {
+            "bool": {
+                "must": [
+                    {
+                        "match": {
+                            "title": "software engineer"
+                        }
+                    }
+                ]
+            }
+        }
+    }
+
+    try:
+        timeout = aiohttp.ClientTimeout(total=15)
+        async with aiohttp.ClientSession(timeout=timeout) as session:
+            async with session.post(url, headers=headers, json=payload) as resp:
+                status = resp.status
+                data = await safe_json(resp)
+
+                if status == 200:
+                    # Check if we got a valid response (list of job results or empty list)
+                    if isinstance(data, list):
+                        return {"success": True, "status_code": status, "error_message": None}
+                    # Some endpoints may return a dict with results
+                    if isinstance(data, dict):
+                        if "error" in data or "errors" in data:
+                            err = data.get("error") or data.get("errors")
+                            if isinstance(err, dict):
+                                err = err.get("message") or str(err)
+                            return {"success": False, "status_code": status, "error_message": str(err)}
+                        return {"success": True, "status_code": status, "error_message": None}
+                    return {"success": True, "status_code": status, "error_message": None}
+
+                if status in (401, 403):
+                    msg = None
+                    if isinstance(data, dict):
+                        msg = data.get("message") or data.get("error") or data.get("detail")
+                    return {
+                        "success": False,
+                        "status_code": status,
+                        "error_message": msg or "Unauthorized – check Coresignal API key",
+                    }
+
+                msg = None
+                if isinstance(data, dict):
+                    msg = data.get("message") or data.get("error") or data.get("detail")
+                return {
+                    "success": False,
+                    "status_code": status,
+                    "error_message": msg or f"Coresignal responded with {status}",
+                }
+    except Exception as e:
+        logger.error(f"Coresignal connectivity test failed: {e}")
+        return {"success": False, "status_code": 0, "error_message": str(e)}
+
+
 ###############################################################################
 #                              DATAGMA CONNECTIVITY
 ###############################################################################
@@ -1808,6 +1884,7 @@ async def test_connectivity(tool_config: List[Dict[str, Any]]) -> Dict[str, Dict
         "commonRoom":       test_commonroom,
         "scarf":            test_scarf,
         "theorg":           test_theorg,
+        "coresignal":       test_coresignal,
         "salesforce":       test_salesforce,
         "clay":             test_clay,
         "posthog":          test_posthog,

{dhisana-0.0.1.dev252 → dhisana-0.0.1.dev254}/src/dhisana.egg-info/PKG-INFO

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

{dhisana-0.0.1.dev252 → dhisana-0.0.1.dev254}/tests/test_apollo_lead_search.py

@@ -58,7 +58,9 @@ async def test_search_leads_with_apollo_page_parses_url_filters(apollo_people_re
         "https://app.apollo.io/#/people?page=1&personTitles[]=VP%20Sales&"
         "organizationLatestFundingStageCd[]=3&organizationLatestFundingStageCd[]=4&"
         "organizationLatestFundingStageCd[]=5&qOrganizationJobTitles[]=SDR&"
-        "qOrganizationJobTitles[]=account%20executive"
+        "qOrganizationJobTitles[]=account%20executive&"
+        "organizationIndustries[]=information%20technology%20%26%20services&"
+        "organizationIndustryTagIds[]=5567cd4773696439b10b0000"
     )
 
     with patch(
@@ -66,7 +68,7 @@ async def test_search_leads_with_apollo_page_parses_url_filters(apollo_people_re
         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",
+        return_value=("test_api_key", False),
     ):
         result = await search_leads_with_apollo_page(
             query=LeadsQueryFilters(),
@@ -83,6 +85,8 @@ async def test_search_leads_with_apollo_page_parses_url_filters(apollo_people_re
     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"]
+    assert payload["organization_industries"] == ["information technology & services"]
+    assert payload["organization_industry_tag_ids"] == ["5567cd4773696439b10b0000"]
 
 
 @pytest.mark.asyncio
@@ -92,6 +96,8 @@ async def test_search_leads_with_apollo_page_uses_query_filters(apollo_people_re
         job_openings_with_titles=["SDR", "Account Executive"],
         latest_funding_stages=["3", "4"],
         organization_num_employees_ranges=["201,500"],
+        industries=["information technology & services"],
+        company_industry_tag_ids=["5567cd4773696439b10b0000"],
     )
 
     with patch(
@@ -99,7 +105,7 @@ async def test_search_leads_with_apollo_page_uses_query_filters(apollo_people_re
         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",
+        return_value=("test_api_key", False),
     ):
         result = await search_leads_with_apollo_page(
             query=query,
@@ -117,3 +123,5 @@ async def test_search_leads_with_apollo_page_uses_query_filters(apollo_people_re
     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"]
+    assert payload["organization_industries"] == ["information technology & services"]
+    assert payload["organization_industry_tag_ids"] == ["5567cd4773696439b10b0000"]