duckrun 0.2.10.dev0__py3-none-any.whl → 0.2.10.dev1__py3-none-any.whl

duckrun/core.py

@@ -94,6 +94,7 @@ class Duckrun:
             pass  # Not in Colab, use default transport
         
         self._attach_lakehouse()
+        self._register_lookup_functions()
 
     @classmethod
     def connect(cls, connection_string: str, sql_folder: Optional[str] = None,
@@ -456,9 +457,6 @@ class Duckrun:
         except Exception as e:
             print(f"❌ Error attaching lakehouse: {e}")
             print("Continuing without pre-attached tables.")
-        
-        # Register lookup functions as DuckDB UDFs
-        self._register_lookup_functions()
 
     def _register_lookup_functions(self):
         """
@@ -488,7 +486,7 @@ class Duckrun:
                 from .auth import get_fabric_api_token
                 token = get_fabric_api_token()
                 if not token:
-                    return f"[Error: No auth token]"
+                    return None
                 
                 url = "https://api.fabric.microsoft.com/v1/workspaces"
                 headers = {"Authorization": f"Bearer {token}", "Content-Type": "application/json"}
@@ -503,9 +501,9 @@ class Duckrun:
                         self._name_cache['workspace_name_to_id'][name] = workspace_id
                         return name
                 
-                return f"[Workspace {workspace_id} not found]"
+                return None
             except Exception as e:
-                return f"[Error: {str(e)}]"
+                return None
         
         def get_lakehouse_name(workspace_id: str, lakehouse_id: str) -> str:
             """Get lakehouse display name from workspace ID and lakehouse ID (GUIDs)"""
@@ -517,7 +515,7 @@ class Duckrun:
                 from .auth import get_fabric_api_token
                 token = get_fabric_api_token()
                 if not token:
-                    return f"[Error: No auth token]"
+                    return None
                 
                 url = f"https://api.fabric.microsoft.com/v1/workspaces/{workspace_id}/lakehouses"
                 headers = {"Authorization": f"Bearer {token}", "Content-Type": "application/json"}
@@ -533,9 +531,9 @@ class Duckrun:
                         self._name_cache['lakehouse_name_to_id'][lh_cache_key] = lakehouse_id
                         return name
                 
-                return f"[Lakehouse {lakehouse_id} not found]"
+                return None
             except Exception as e:
-                return f"[Error: {str(e)}]"
+                return None
         
         def get_workspace_id_from_name(workspace_name: str) -> str:
             """Get workspace ID (GUID) from workspace display name"""
@@ -546,7 +544,7 @@ class Duckrun:
                 from .auth import get_fabric_api_token
                 token = get_fabric_api_token()
                 if not token:
-                    return f"[Error: No auth token]"
+                    return None
                 
                 url = "https://api.fabric.microsoft.com/v1/workspaces"
                 headers = {"Authorization": f"Bearer {token}", "Content-Type": "application/json"}
@@ -561,9 +559,9 @@ class Duckrun:
                         self._name_cache['workspace_id_to_name'][workspace_id] = workspace_name
                         return workspace_id
                 
-                return f"[Workspace '{workspace_name}' not found]"
+                return None
             except Exception as e:
-                return f"[Error: {str(e)}]"
+                return None
         
         def get_lakehouse_id_from_name(workspace_id: str, lakehouse_name: str) -> str:
             """Get lakehouse ID (GUID) from workspace ID and lakehouse display name"""
@@ -575,7 +573,7 @@ class Duckrun:
                 from .auth import get_fabric_api_token
                 token = get_fabric_api_token()
                 if not token:
-                    return f"[Error: No auth token]"
+                    return None
                 
                 url = f"https://api.fabric.microsoft.com/v1/workspaces/{workspace_id}/lakehouses"
                 headers = {"Authorization": f"Bearer {token}", "Content-Type": "application/json"}
@@ -591,9 +589,9 @@ class Duckrun:
                         self._name_cache['lakehouse_id_to_name'][id_cache_key] = lakehouse_name
                         return lakehouse_id
                 
-                return f"[Lakehouse '{lakehouse_name}' not found]"
+                return None
             except Exception as e:
-                return f"[Error: {str(e)}]"
+                return None
         
         # Register functions in DuckDB
         try:

{duckrun-0.2.10.dev0.dist-info → duckrun-0.2.10.dev1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: duckrun
-Version: 0.2.10.dev0
+Version: 0.2.10.dev1
 Summary: Lakehouse task runner powered by DuckDB for Microsoft Fabric
 Author: mim
 License: MIT

{duckrun-0.2.10.dev0.dist-info → duckrun-0.2.10.dev1.dist-info}/RECORD

@@ -1,14 +1,14 @@
 duckrun/__init__.py,sha256=cTj6KQ6hKmgu1z7k9nhDcO5lct049luxjx1V0QnymCo,235
 duckrun/auth.py,sha256=qPaLQ7InlV9leA9r6E6VEeYavFFoBi0zSN8m_l1aoQs,9545
-duckrun/core.py,sha256=imA1Xt3_Qz1PZFzrcv6jbvsfiyxJiYax2iZ14czgrVo,54637
+duckrun/core.py,sha256=MlaHOOz9bg3-EDXR3C4pEcp75QsnEcbTOmvsMjomLKc,54279
 duckrun/files.py,sha256=Fvdjg3DyHJzIVzKo8M_j-eGz4zU61lOB38Y_onbQJkI,10137
 duckrun/lakehouse.py,sha256=j--Z3zo8AOWt1GF9VzRosmmTAy6ey2D0LVubti58twU,14109
 duckrun/runner.py,sha256=yrDxfy1RVkb8iK9GKGmIFZHzCvcO_0GVQlbng7Vw_iM,14171
 duckrun/semantic_model.py,sha256=obzlN2-dbEW3JmDop-vrZGGGLi9u3ThhTbgtDjou7uY,29509
 duckrun/stats.py,sha256=oKIjZ7u5cFVT63FuOl5UqoDsOG3098woSCn-uI6i_sQ,11084
 duckrun/writer.py,sha256=svUuPCYOhrz299NgnpTKhARKjfej0PxnoND2iPDSypk,8098
-duckrun-0.2.10.dev0.dist-info/licenses/LICENSE,sha256=-DeQQwdbCbkB4507ZF3QbocysB-EIjDtaLexvqRkGZc,1083
-duckrun-0.2.10.dev0.dist-info/METADATA,sha256=O6fyWc9A94TlphbmbMEpyrow4iA6ZUrBbs0KbFgmfNE,20629
-duckrun-0.2.10.dev0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-duckrun-0.2.10.dev0.dist-info/top_level.txt,sha256=BknMEwebbUHrVAp3SC92ps8MPhK7XSYsaogTvi_DmEU,8
-duckrun-0.2.10.dev0.dist-info/RECORD,,
+duckrun-0.2.10.dev1.dist-info/licenses/LICENSE,sha256=-DeQQwdbCbkB4507ZF3QbocysB-EIjDtaLexvqRkGZc,1083
+duckrun-0.2.10.dev1.dist-info/METADATA,sha256=ctUMwNP22HX8g1PhY3nERJTbJvJGMrY2qNH2kutdX4c,20629
+duckrun-0.2.10.dev1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+duckrun-0.2.10.dev1.dist-info/top_level.txt,sha256=BknMEwebbUHrVAp3SC92ps8MPhK7XSYsaogTvi_DmEU,8
+duckrun-0.2.10.dev1.dist-info/RECORD,,