dhisana 0.0.1.dev253__py3-none-any.whl → 0.0.1.dev254__py3-none-any.whl

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.dev253.dist-info → dhisana-0.0.1.dev254.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dhisana
-Version: 0.0.1.dev253
+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.dev253.dist-info → dhisana-0.0.1.dev254.dist-info}/RECORD

@@ -81,7 +81,7 @@ dhisana/utils/serperdev_google_jobs.py,sha256=m5_2f_5y79FOFZz1A_go6m0hIUfbbAoZ0Y
 dhisana/utils/serperdev_local_business.py,sha256=JoZfTg58Hojv61cyuwA2lcnPdLT1lawnWaBNrUYWnuQ,6447
 dhisana/utils/serperdev_search.py,sha256=_iBKIfHMq4gFv5StYz58eArriygoi1zW6VnLlux8vto,9363
 dhisana/utils/smtp_email_tools.py,sha256=J9uDHCEu2BTyFzDBru0e1lC8bsAMt9c_mYXTzJgk9Kc,22054
-dhisana/utils/test_connect.py,sha256=pYR1Ki6WKx-vAhLJxQ6At627xfOnlQeVRjnW7FBdGKM,86702
+dhisana/utils/test_connect.py,sha256=UbVNVPS8OXRCrntwRCZVieX2m3OJ3mRYZG-gj2oajHA,90040
 dhisana/utils/trasform_json.py,sha256=7V72XNDpuxUX0GHN5D83z4anj_gIf5zabaHeQm7b1_E,6979
 dhisana/utils/web_download_parse_tools.py,sha256=ouXwH7CmjcRjoBfP5BWat86MvcGO-8rLCmWQe_eZKjc,7810
 dhisana/utils/workflow_code_model.py,sha256=YPWse5vBb3O6Km2PvKh1Q3AB8qBkzLt1CrR5xOL9Mro,99
@@ -95,8 +95,8 @@ dhisana/workflow/agent.py,sha256=esv7_i_XuMkV2j1nz_UlsHov_m6X5WZZiZm_tG4OBHU,565
 dhisana/workflow/flow.py,sha256=xWE3qQbM7j2B3FH8XnY3zOL_QXX4LbTW4ArndnEYJE0,1638
 dhisana/workflow/task.py,sha256=HlWz9mtrwLYByoSnePOemBUBrMEcj7KbgNjEE1oF5wo,1830
 dhisana/workflow/test.py,sha256=E7lRnXK0PguTNzyasHytLzTJdkqIPxG5_4qk4hMEeKc,3399
-dhisana-0.0.1.dev253.dist-info/METADATA,sha256=zs8sGuJ9Lv2en8iIMns92NleCIMmMQ4ub0bqzzimMsY,1190
-dhisana-0.0.1.dev253.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-dhisana-0.0.1.dev253.dist-info/entry_points.txt,sha256=jujxteZmNI9EkEaK-pOCoWuBujU8TCevdkfl9ZcKHek,49
-dhisana-0.0.1.dev253.dist-info/top_level.txt,sha256=NETTHt6YifG_P7XtRHbQiXZlgSFk9Qh9aR-ng1XTf4s,8
-dhisana-0.0.1.dev253.dist-info/RECORD,,
+dhisana-0.0.1.dev254.dist-info/METADATA,sha256=Y4lCiPFjHmSRLN2QfJsTlKVxdu_m3XRXHl9c8DmIm6k,1190
+dhisana-0.0.1.dev254.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+dhisana-0.0.1.dev254.dist-info/entry_points.txt,sha256=jujxteZmNI9EkEaK-pOCoWuBujU8TCevdkfl9ZcKHek,49
+dhisana-0.0.1.dev254.dist-info/top_level.txt,sha256=NETTHt6YifG_P7XtRHbQiXZlgSFk9Qh9aR-ng1XTf4s,8
+dhisana-0.0.1.dev254.dist-info/RECORD,,