definite-sdk 0.1.12__py3-none-any.whl → 0.1.15__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

@@ -16,6 +16,7 @@ class DefiniteIntegrationStore:
     Accessing values:
     >>> integration_store.list_integrations()
     >>> integration_store.get_integration("name")
+    >>> integration_store.get_integration_by_id("integration_id")
     """
 
     def __init__(self, api_key: str, api_url: str):
@@ -59,10 +60,27 @@ class DefiniteIntegrationStore:
         response.raise_for_status()
         return dict(response.json())
 
+    def get_integration_by_id(self, integration_id: str) -> dict:
+        """
+        Retrieves an integration by ID.
+
+        Args:
+            integration_id (str): The ID of the integration.
+
+        Returns:
+            dict: The integration details.
+        """
+        response = requests.get(
+            self._integration_store_url + f"/id/{integration_id}",
+            headers={"Authorization": "Bearer " + self._api_key},
+        )
+        response.raise_for_status()
+        return dict(response.json())
+
     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-0.1.12.dist-info → definite_sdk-0.1.15.dist-info}/METADATA

@@ -1,8 +1,9 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.4
 Name: definite-sdk
-Version: 0.1.12
+Version: 0.1.15
 Summary: Definite SDK for Python
 License: MIT
+License-File: LICENSE
 Author: Definite
 Author-email: hello@definite.app
 Requires-Python: >=3.9,<4.0
@@ -13,6 +14,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
 Provides-Extra: dlt
 Requires-Dist: dlt (>=1.0,<2.0) ; extra == "dlt"
 Requires-Dist: duckdb (>=1.0,<2.0) ; extra == "dlt"

definite_sdk-0.1.15.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=fkLBQyK4BMzwkkuR-Ih8AMbUf7mR7u2psknl5NmNy6E,3970
+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.15.dist-info/METADATA,sha256=igYD7a9epwBJpNK6Tn2HftMRzHuJJNm8TbVxc2m6wjg,9209
+definite_sdk-0.1.15.dist-info/WHEEL,sha256=M5asmiAlL6HEcOq52Yi5mmk9KmTVjY2RDPtO4p9DMrc,88
+definite_sdk-0.1.15.dist-info/licenses/LICENSE,sha256=jMd7PtwNWiMoGIDgFutC1v4taO69W9SRZLKe9o470Vk,1064
+definite_sdk-0.1.15.dist-info/RECORD,,

{definite_sdk-0.1.12.dist-info → definite_sdk-0.1.15.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 2.1.3
+Generator: poetry-core 2.2.0
 Root-Is-Purelib: true
 Tag: py3-none-any

definite_sdk-0.1.12.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=o2ume-S7__wp8hMBy2aC6dmGYV4s-IXMlMAyEuyAP84,4705
-definite_sdk/store.py,sha256=7VYvROIrlWyM_ZX3LpqWWAfVLNDm0XMccMLv6qi9id8,5575
-definite_sdk-0.1.12.dist-info/LICENSE,sha256=jMd7PtwNWiMoGIDgFutC1v4taO69W9SRZLKe9o470Vk,1064
-definite_sdk-0.1.12.dist-info/METADATA,sha256=mR32ZrkWtX_q7BBMF6dtE2f2lHicy2SBo1lkMxOTFto,9136
-definite_sdk-0.1.12.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-definite_sdk-0.1.12.dist-info/RECORD,,