dhisana 0.0.1.dev238__tar.gz → 0.0.1.dev239__tar.gz

{dhisana-0.0.1.dev238 → dhisana-0.0.1.dev239}/PKG-INFO

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

{dhisana-0.0.1.dev238 → dhisana-0.0.1.dev239}/setup.py

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

{dhisana-0.0.1.dev238 → dhisana-0.0.1.dev239}/src/dhisana/utils/test_connect.py

@@ -1646,6 +1646,49 @@ async def test_scarf(api_key: str) -> Dict[str, Any]:
         return {"success": False, "status_code": 0, "error_message": str(exc)}
 
 
+async def test_theorg(api_key: str) -> Dict[str, Any]:
+    """Validate The Org API key by calling the Usage endpoint."""
+    url = "https://api.theorg.com/v1.1/usage"
+    headers = {
+        "X-Api-Key": api_key,
+        "Accept": "application/json",
+    }
+
+    try:
+        timeout = aiohttp.ClientTimeout(total=10)
+        async with aiohttp.ClientSession(timeout=timeout) as session:
+            async with session.get(url, headers=headers) as response:
+                status = response.status
+                data = await safe_json(response)
+
+                if status == 200:
+                    payload = data if isinstance(data, dict) else None
+                    usage = payload.get("data") if payload else None
+                    if isinstance(usage, dict):
+                        return {"success": True, "status_code": status, "error_message": None}
+                    return {
+                        "success": False,
+                        "status_code": status,
+                        "error_message": "The Org usage endpoint returned an unexpected payload.",
+                    }
+
+                message = None
+                if isinstance(data, dict):
+                    message = (
+                        data.get("message")
+                        or data.get("error")
+                        or data.get("detail")
+                    )
+                return {
+                    "success": False,
+                    "status_code": status,
+                    "error_message": message or f"The Org responded with {status}",
+                }
+    except Exception as exc:
+        logger.error(f"The Org connectivity test failed: {exc}")
+        return {"success": False, "status_code": 0, "error_message": str(exc)}
+
+
 ###############################################################################
 #                              DATAGMA CONNECTIVITY
 ###############################################################################
@@ -1739,6 +1782,7 @@ async def test_connectivity(tool_config: List[Dict[str, Any]]) -> Dict[str, Dict
         "nooks":            test_nooks,
         "commonRoom":       test_commonroom,
         "scarf":            test_scarf,
+        "theorg":           test_theorg,
         "salesforce":       test_salesforce,
         "clay":             test_clay,
         "posthog":          test_posthog,

{dhisana-0.0.1.dev238 → dhisana-0.0.1.dev239}/src/dhisana.egg-info/PKG-INFO

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

{dhisana-0.0.1.dev238 → dhisana-0.0.1.dev239}/tests/test_connectivity.py

@@ -54,6 +54,22 @@ class TestConnectivity(unittest.TestCase):
                 self.assertTrue(result['firefliesai']['success'])
         asyncio.run(runner())
 
+    def test_theorg_dispatch(self):
+        async def runner():
+            with patch('src.dhisana.utils.test_connect.test_theorg', new=AsyncMock(return_value={'success': True, 'status_code': 200, 'error_message': None})):
+                tool_config = [
+                    {
+                        'name': 'theorg',
+                        'configuration': [
+                            {'name': 'apiKey', 'value': 'dummy'}
+                        ]
+                    }
+                ]
+                result = await run_test_connectivity(tool_config)
+                self.assertIn('theorg', result)
+                self.assertTrue(result['theorg']['success'])
+        asyncio.run(runner())
+
     def test_salesforce_dispatch(self):
         async def runner():
             with patch(