definite-sdk 0.1.12__tar.gz → 0.1.15__tar.gz

{definite_sdk-0.1.12 → definite_sdk-0.1.15}/PKG-INFO

@@ -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.12 → definite_sdk-0.1.15}/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-0.1.12 → definite_sdk-0.1.15}/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-0.1.12 → definite_sdk-0.1.15}/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-0.1.12 → definite_sdk-0.1.15}/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 → definite_sdk-0.1.15}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "definite-sdk"
-version = "0.1.12"
+version = "0.1.15"
 description = "Definite SDK for Python"
 authors = ["Definite <hello@definite.app>"]
 license = "MIT"