definite-sdk 0.1.15__py3-none-any.whl → 0.1.16__py3-none-any.whl

definite_sdk/client.py

@@ -1,7 +1,6 @@
 import os
 from typing import Optional
 
-import requests
 from definite_sdk.integration import DefiniteIntegrationStore
 from definite_sdk.message import DefiniteMessageClient
 from definite_sdk.secret import DefiniteSecretStore
@@ -73,6 +72,7 @@ class DefiniteClient:
         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")
 
@@ -85,42 +85,38 @@ class DefiniteClient:
             >>> conn.execute(sql)
         """
         # Fetch DuckLake integration details
-        try:
-            response = requests.get(
-                f"{self.api_url}/v1/api/integration/Ducklake",
-                headers={"Authorization": f"Bearer {self.api_key}"},
+        integrations_client = self.get_integration_store()
+        integrations = integrations_client.list_integrations(
+            integration_type="ducklake"
+        )
+        if len(integrations) == 0:
+            raise Exception(
+                "DuckLake integration not found. Please make sure one is"
+                "created for your team at https://ui.definite.app/settings/integrations"
             )
-            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
+
+        integration = integrations.pop()
 
         # 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']}'
-);"""
+            TYPE gcs,
+            KEY_ID '{integration["gcs_access_key_id"]}',
+            SECRET '{integration["gcs_secret_access_key"]}'
+        );"""
 
         # Build PostgreSQL connection string
         pg_conn_str = (
-            f"postgresql://{dl_integration['pg_user']}:"
-            f"{dl_integration['pg_password']}@"
-            f"{dl_integration['pg_host']}:"
-            f"{dl_integration['pg_port']}/"
-            f"{dl_integration['pg_database']}"
+            f"postgresql://{integration['pg_user']}:"
+            f"{integration['pg_password']}@"
+            f"{integration['pg_host']}:"
+            f"{integration['pg_port']}/"
+            f"{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']}');"
+            f"(DATA_PATH 'gs://{integration['gcs_bucket_path']}', "
+            f"METADATA_SCHEMA '{integration['pg_schema']}');"
         )
 
         return f"{create_secret_sql}\n\n{attach_sql}"

definite_sdk/integration.py

@@ -1,8 +1,8 @@
-from typing import Iterator, Optional, Tuple
+from typing import Dict, List, Optional, cast
 
 import requests
 
-INTEGRATION_ENDPOINT = "/v1/api/integration"
+INTEGRATION_ENDPOINT = "/v1/api/integrations"
 
 
 class DefiniteIntegrationStore:
@@ -27,23 +27,42 @@ class DefiniteIntegrationStore:
             api_key (str): The API key for authorization.
         """
         self._api_key = api_key
-        self._integration_store_url = api_url + INTEGRATION_ENDPOINT
-
-    def list_integrations(self) -> Iterator[dict]:
+        self._integrations_url = api_url + INTEGRATION_ENDPOINT
+
+    def list_integrations(
+        self,
+        *,
+        integration_type: Optional[str] = None,
+        category: Optional[str] = None,
+    ) -> List[Dict]:
         """
         Lists all integrations in the store.
 
+        Args:
+            integration_type (str): Optional type filter
+            category (str): Optional category filter
+
         Returns:
             Iterator[str]: An iterator of integrations.
         """
+        params = {}
+        if integration_type:
+            params |= {"type": integration_type}
+        if category:
+            params |= {"category": category}
+
         response = requests.get(
-            self._integration_store_url,
+            self._integrations_url,
+            params=params,
             headers={"Authorization": "Bearer " + self._api_key},
         )
         response.raise_for_status()
-        return iter(response.json())
+        cursor_page = response.json()
+        integrations = cursor_page.get("data", [])
+        details = [i.get("details", {}) for i in integrations]
+        return cast(List[Dict], details)
 
-    def get_integration(self, name: str) -> dict:
+    def get_integration(self, name: str) -> Dict:
         """
         Retrieves an integration by name.
 
@@ -54,13 +73,19 @@ class DefiniteIntegrationStore:
             str: The value of the integration.
         """
         response = requests.get(
-            self._integration_store_url + f"/{name}",
+            self._integrations_url,
+            params={"name": name, "limit": 1},
             headers={"Authorization": "Bearer " + self._api_key},
         )
         response.raise_for_status()
-        return dict(response.json())
-
-    def get_integration_by_id(self, integration_id: str) -> dict:
+        cursor_page = response.json()
+        integrations = cursor_page.get("data", [])
+        if len(integrations) == 0:
+            raise Exception(f"Integration with name {name} not found")
+        integration = integrations[0]
+        return integration.get("details", {})
+
+    def get_integration_by_id(self, integration_id: str) -> Dict:
         """
         Retrieves an integration by ID.
 
@@ -71,47 +96,35 @@ class DefiniteIntegrationStore:
             dict: The integration details.
         """
         response = requests.get(
-            self._integration_store_url + f"/id/{integration_id}",
+            self._integrations_url,
+            params={"id": integration_id, "limit": 1},
             headers={"Authorization": "Bearer " + self._api_key},
         )
         response.raise_for_status()
-        return dict(response.json())
-
-    def lookup_duckdb_integration(self) -> Optional[Tuple[str, str]]:
+        cursor_page = response.json()
+        integrations = cursor_page.get("data", [])
+        if len(integrations) == 0:
+            raise Exception(f"Integration with ID {integration_id} not found")
+        integration = integrations[0]
+        return integration.get("details", {})
+
+    def lookup_duckdb_integration(self) -> Dict:
         """
         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.
-
         Returns:
             Optional[Tuple[str, str]]: Tuple of (integration_id, connection_uri)
                 if found, None if no DuckDB integration exists.
         """
-        try:
-            response = requests.get(
-                self._integration_store_url,
-                headers={"Authorization": "Bearer " + self._api_key},
-            )
-            response.raise_for_status()
-            integrations = response.json()
-        except:
-            return None
-
-        # Check for DuckDB in the integrations
-        # Note: Currently this will not find DuckDB as it's a destination integration
-        for integration in integrations:
-            integration_type = integration.get("integration_type", "").lower()
-            if integration_type == "duckdb" and integration.get("active", True):
-                integration_id = integration.get("id")
-                # Connection URI might be in config or connection_string field
-                connection_uri = (
-                    integration.get("connection_string")
-                    or integration.get("config", {}).get("database_path")
-                    or integration.get("config", {}).get("connection_string")
-                )
-                if integration_id and connection_uri:
-                    return (integration_id, connection_uri)
-
-        return None
+        response = requests.get(
+            self._integrations_url,
+            params={"type": "duckdb", "limit": 1},
+            headers={"Authorization": "Bearer " + self._api_key},
+        )
+        response.raise_for_status()
+        cursor_page = response.json()
+        integrations = cursor_page.get("data", [])
+        if len(integrations) == 0:
+            raise Exception("Integration with type `duckdb` not found")
+        integration = integrations[0]
+        return integration.get("details", {})

{definite_sdk-0.1.15.dist-info → definite_sdk-0.1.16.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: definite-sdk
-Version: 0.1.15
+Version: 0.1.16
 Summary: Definite SDK for Python
 License: MIT
 License-File: LICENSE

{definite_sdk-0.1.15.dist-info → definite_sdk-0.1.16.dist-info}/RECORD

@@ -1,13 +1,13 @@
 definite_sdk/__init__.py,sha256=pzw1-eWBcD5mvWRxbo59dMUU8Wue4zNsdDYK0Pj1hXQ,606
-definite_sdk/client.py,sha256=5v4mLIsvVf9SS3kE4_vfXjx9AegrGzNsUReIz_n4sxI,5420
+definite_sdk/client.py,sha256=iI1CXPMW1jtIFIPcNT2V4Ley7VqHIzEimmGO2biXZkM,5203
 definite_sdk/dlt.py,sha256=JC-S1jd6zqnpGSnvk9kYb8GQ1IPZRd1zVKTVYAbeA4w,7188
-definite_sdk/integration.py,sha256=fkLBQyK4BMzwkkuR-Ih8AMbUf7mR7u2psknl5NmNy6E,3970
+definite_sdk/integration.py,sha256=oSQz9ojgnCch5CFWEC02nTwNmM6Y-w5-LKz0na4xLOQ,4163
 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.16.dist-info/METADATA,sha256=828vrMGBYuQZlClISEXcYLfKfVUa8pdRqwaS_ufSa4c,9209
+definite_sdk-0.1.16.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+definite_sdk-0.1.16.dist-info/licenses/LICENSE,sha256=jMd7PtwNWiMoGIDgFutC1v4taO69W9SRZLKe9o470Vk,1064
+definite_sdk-0.1.16.dist-info/RECORD,,

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

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