definite-sdk 0.1.11__py3-none-any.whl → 0.1.14__py3-none-any.whl

definite_sdk/__init__.py

@@ -11,7 +11,7 @@ from definite_sdk.secret import DefiniteSecretStore
 from definite_sdk.sql import DefiniteSqlClient
 from definite_sdk.store import DefiniteKVStore
 
-__version__ = "0.1.9"
+__version__ = "0.1.14"
 __all__ = [
     "DefiniteClient",
     "DefiniteIntegrationStore",
@@ -19,4 +19,4 @@ __all__ = [
     "DefiniteSecretStore",
     "DefiniteSqlClient",
     "DefiniteKVStore",
-]
+]

definite_sdk/client.py

@@ -69,36 +69,45 @@ class DefiniteClient:
 
     def attach_ducklake(self, alias: str = "lake") -> str:
         """Generates SQL statements to attach DuckLake to a DuckDB connection.
-        
+
         This method fetches the team's DuckLake integration credentials and generates
         the necessary SQL statements to create a GCS secret and attach DuckLake.
-        
+
         Args:
             alias: The alias name for the attached DuckLake database (default: "lake")
-            
+
         Returns:
             str: SQL statements to execute for attaching DuckLake
-            
+
         Example:
             >>> client = DefiniteClient(os.environ["DEFINITE_API_KEY"])
             >>> sql = client.attach_ducklake()
             >>> conn.execute(sql)
         """
         # Fetch DuckLake integration details
-        response = requests.get(
-            f"{self.api_url}/v1/api/integration/DuckLake",
-            headers={"Authorization": f"Bearer {self.api_key}"},
-        )
-        response.raise_for_status()
-        dl_integration = response.json()
-        
+        try:
+            response = requests.get(
+                f"{self.api_url}/v1/api/integration/Ducklake",
+                headers={"Authorization": f"Bearer {self.api_key}"},
+            )
+            response.raise_for_status()
+            dl_integration = response.json()
+        except requests.exceptions.HTTPError as e:
+            if response.status_code == 404:
+                raise ValueError(
+                    "DuckLake integration not found. Please ensure you have a DuckLake integration "
+                    "configured in your Definite account at https://ui.definite.app"
+                ) from e
+            else:
+                raise
+
         # Generate SQL statements
         create_secret_sql = f"""CREATE SECRET (
     TYPE gcs,
     KEY_ID '{dl_integration['gcs_access_key_id']}',
     SECRET '{dl_integration['gcs_secret_access_key']}'
 );"""
-        
+
         # Build PostgreSQL connection string
         pg_conn_str = (
             f"postgresql://{dl_integration['pg_user']}:"
@@ -107,13 +116,13 @@ class DefiniteClient:
             f"{dl_integration['pg_port']}/"
             f"{dl_integration['pg_database']}"
         )
-        
+
         attach_sql = (
             f"ATTACH 'ducklake:postgres:{pg_conn_str}' AS {alias} "
             f"(DATA_PATH 'gs://{dl_integration['gcs_bucket_path']}', "
             f"METADATA_SCHEMA '{dl_integration['pg_schema']}');"
         )
-        
+
         return f"{create_secret_sql}\n\n{attach_sql}"
 
     # Alias methods for consistency

definite_sdk/integration.py

@@ -62,7 +62,7 @@ class DefiniteIntegrationStore:
     def lookup_duckdb_integration(self) -> Optional[Tuple[str, str]]:
         """
         Look up the team's DuckDB integration.
-        
+
         Note: Currently, the API only returns extractor (source) integrations.
         Destination integrations like DuckDB are not yet exposed through this endpoint.
         This method is provided for future compatibility when the API is updated.

definite_sdk/message.py

@@ -21,7 +21,7 @@ class DefiniteMessageClient:
     ...     to="C0920MVPWFN",  # channel_id
     ...     content="Hello from Definite SDK!"
     ... )
-    
+
     >>> # Send a Slack message with blocks and thread
     >>> result = message_client.send_message(
     ...     channel="slack",
@@ -53,7 +53,7 @@ class DefiniteMessageClient:
         subject: Optional[str] = None,
         blocks: Optional[List[Dict[str, Any]]] = None,
         thread_ts: Optional[str] = None,
-        **kwargs: Any
+        **kwargs: Any,
     ) -> Dict[str, Any]:
         """
         Sends a message through the specified channel.
@@ -83,7 +83,7 @@ class DefiniteMessageClient:
             ...     to="C0920MVPWFN",
             ...     content="Hello team!"
             ... )
-            
+
             >>> # Slack message with blocks
             >>> result = message_client.send_message(
             ...     channel="slack",
@@ -97,7 +97,7 @@ class DefiniteMessageClient:
             ... )
         """
         channel_lower = channel.lower()
-        
+
         if channel_lower == "slack":
             return self._send_slack_message(
                 integration_id=integration_id,
@@ -105,7 +105,7 @@ class DefiniteMessageClient:
                 text=content,
                 blocks=blocks,
                 thread_ts=thread_ts,
-                **kwargs
+                **kwargs,
             )
         else:
             raise ValueError(f"Unsupported channel: {channel}")
@@ -117,7 +117,7 @@ class DefiniteMessageClient:
         text: str,
         blocks: Optional[List[Dict[str, Any]]] = None,
         thread_ts: Optional[str] = None,
-        **kwargs: Any
+        **kwargs: Any,
     ) -> Dict[str, Any]:
         """
         Internal method to send a Slack message.
@@ -134,18 +134,18 @@ class DefiniteMessageClient:
             Dict[str, Any]: The API response.
         """
         url = f"{self._message_url}/slack/message"
-        
+
         payload = {
             "integration_id": integration_id,
             "channel_id": channel_id,
-            "text": text
+            "text": text,
         }
-        
+
         if blocks:
             payload["blocks"] = blocks
         if thread_ts:
             payload["thread_ts"] = thread_ts
-            
+
         # Add any additional kwargs to the payload
         payload.update(kwargs)
 
@@ -164,7 +164,7 @@ class DefiniteMessageClient:
         text: str,
         blocks: Optional[List[Dict[str, Any]]] = None,
         thread_ts: Optional[str] = None,
-        **kwargs: Any
+        **kwargs: Any,
     ) -> Dict[str, Any]:
         """
         Convenience method to send a Slack message directly.
@@ -195,5 +195,5 @@ class DefiniteMessageClient:
             content=text,
             blocks=blocks,
             thread_ts=thread_ts,
-            **kwargs
-        )
+            **kwargs,
+        )

definite_sdk/sql.py

@@ -1,4 +1,4 @@
-from typing import Any, Dict, List, Optional, Union
+from typing import Any, Dict, Optional
 
 import requests
 
@@ -31,7 +31,9 @@ class DefiniteSqlClient:
     ...     "timeDimensions": [{"dimension": "sales.date", "granularity": "month"}],
     ...     "limit": 1000
     ... }
-    >>> result = sql_client.execute_cube_query(cube_query, integration_id="my_cube_integration")
+    >>> result = sql_client.execute_cube_query(
+    ...     cube_query, integration_id="my_cube_integration"
+    ... )
     >>> print(result)
     """
 
@@ -46,11 +48,7 @@ class DefiniteSqlClient:
         self._api_key = api_key
         self._sql_url = api_url + SQL_ENDPOINT
 
-    def execute(
-        self, 
-        sql: str, 
-        integration_id: Optional[str] = None
-    ) -> Dict[str, Any]:
+    def execute(self, sql: str, integration_id: Optional[str] = None) -> Dict[str, Any]:
         """
         Executes a SQL query against a database integration.
 
@@ -69,7 +67,7 @@ class DefiniteSqlClient:
             >>> result = sql_client.execute("SELECT COUNT(*) FROM users")
             >>> print(result)
         """
-        payload = {"sql": sql}
+        payload: Dict[str, Any] = {"sql": sql}
         if integration_id:
             payload["integration_id"] = integration_id
 
@@ -82,11 +80,12 @@ class DefiniteSqlClient:
         return response.json()
 
     def execute_cube_query(
-        self, 
-        cube_query: Dict[str, Any], 
+        self,
+        cube_query: Dict[str, Any],
         integration_id: Optional[str] = None,
         persist: bool = True,
         invalidate: bool = False,
+        raw: bool = False,
     ) -> Dict[str, Any]:
         """
         Executes a Cube query against a Cube integration.
@@ -97,6 +96,7 @@ class DefiniteSqlClient:
                 If not provided, the default integration will be used.
             persist (bool): Whether to persist the query result to the cache.
             invalidate (bool): Whether to invalidate the cached result.
+            raw (bool): Whether to return raw/unformatted cube results.
 
         Returns:
             Dict[str, Any]: The query result as returned by the API.
@@ -110,24 +110,38 @@ class DefiniteSqlClient:
             ...     "measures": ["sales.total_amount"],
             ...     "timeDimensions": [{
             ...         "dimension": "sales.date",
-            ...         "granularity": "month"
+            ...         "granularity": "month",
             ...     }],
             ...     "limit": 1000
             ... }
-            >>> result = sql_client.execute_cube_query(cube_query, "my_cube_integration")
+            >>> result = sql_client.execute_cube_query(
+            ...     cube_query, "my_cube_integration"
+            ... )
             >>> print(result)
+
+            >>> # To get raw/unformatted results:
+            >>> raw_result = sql_client.execute_cube_query(
+            ...     cube_query, "my_cube_integration", raw=True
+            ... )
+            >>> print(raw_result)
         """
-        payload = {"cube_query": cube_query}
+        payload: Dict[str, Any] = {"cube_query": cube_query}
         if integration_id:
             payload["integration_id"] = integration_id
         if persist:
             payload["persist"] = persist
         if invalidate:
             payload["invalidate"] = invalidate
+
+        # Build URL with query parameters
+        url = self._sql_url
+        if raw:
+            url += "?raw=true"
+
         response = requests.post(
-            self._sql_url,
+            url,
             json=payload,
             headers={"Authorization": "Bearer " + self._api_key},
         )
         response.raise_for_status()
-        return response.json()
+        return response.json()

{definite_sdk-0.1.11.dist-info → definite_sdk-0.1.14.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: definite-sdk
-Version: 0.1.11
+Version: 0.1.14
 Summary: Definite SDK for Python
 License: MIT
 Author: Definite

definite_sdk-0.1.14.dist-info/RECORD

@@ -0,0 +1,13 @@
+definite_sdk/__init__.py,sha256=pzw1-eWBcD5mvWRxbo59dMUU8Wue4zNsdDYK0Pj1hXQ,606
+definite_sdk/client.py,sha256=5v4mLIsvVf9SS3kE4_vfXjx9AegrGzNsUReIz_n4sxI,5420
+definite_sdk/dlt.py,sha256=JC-S1jd6zqnpGSnvk9kYb8GQ1IPZRd1zVKTVYAbeA4w,7188
+definite_sdk/integration.py,sha256=ATfp8F9yXU6nVMmrLf8fM5BXsb24BG5WBjUE6HPk4WA,3387
+definite_sdk/message.py,sha256=k5TpAoPYSqK_w7ZOphnDz4ntTMmsYVNfG5ROCg6oqxg,6504
+definite_sdk/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+definite_sdk/secret.py,sha256=3wvEnTZ946hOxJBdB29dxvppwJ0x-TJV_LxRcXGqtis,2503
+definite_sdk/sql.py,sha256=o2ume-S7__wp8hMBy2aC6dmGYV4s-IXMlMAyEuyAP84,4705
+definite_sdk/store.py,sha256=7VYvROIrlWyM_ZX3LpqWWAfVLNDm0XMccMLv6qi9id8,5575
+definite_sdk-0.1.14.dist-info/LICENSE,sha256=jMd7PtwNWiMoGIDgFutC1v4taO69W9SRZLKe9o470Vk,1064
+definite_sdk-0.1.14.dist-info/METADATA,sha256=tRUd9oDiV3X2AbbTe3tiiUKe6fGZc1mUYEHSDG14h30,9136
+definite_sdk-0.1.14.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+definite_sdk-0.1.14.dist-info/RECORD,,

definite_sdk-0.1.11.dist-info/RECORD

@@ -1,13 +0,0 @@
-definite_sdk/__init__.py,sha256=Y3GwfDOkpWt6EILoZ7PRT7nXuoVnk0zHrNOPHlkEHS4,604
-definite_sdk/client.py,sha256=mpXFpJVmK3JVYsNrn7f4bzqlxMbIsS3meRu3XMiXWZ4,5075
-definite_sdk/dlt.py,sha256=JC-S1jd6zqnpGSnvk9kYb8GQ1IPZRd1zVKTVYAbeA4w,7188
-definite_sdk/integration.py,sha256=v15QIpWm3eAZr-cmavfA39LXTjpskEEQ2SM_TBV4E1U,3395
-definite_sdk/message.py,sha256=I3qx2v9gIcC9NZyr5zEp_Seft8clRoER4WeUbAaDiFc,6549
-definite_sdk/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-definite_sdk/secret.py,sha256=3wvEnTZ946hOxJBdB29dxvppwJ0x-TJV_LxRcXGqtis,2503
-definite_sdk/sql.py,sha256=nVPUuQy4zLMoeUe9bRgU1eMmIJBwlNCNXZm2tO0M8sw,4223
-definite_sdk/store.py,sha256=7VYvROIrlWyM_ZX3LpqWWAfVLNDm0XMccMLv6qi9id8,5575
-definite_sdk-0.1.11.dist-info/LICENSE,sha256=jMd7PtwNWiMoGIDgFutC1v4taO69W9SRZLKe9o470Vk,1064
-definite_sdk-0.1.11.dist-info/METADATA,sha256=Xvh6IRpK-0vcD1mY814f_otQpY-300jXxRHR-cXtWro,9136
-definite_sdk-0.1.11.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-definite_sdk-0.1.11.dist-info/RECORD,,