relationalai 0.12.13__py3-none-any.whl → 0.13.0__py3-none-any.whl

relationalai/tools/cli.py

@@ -27,7 +27,9 @@ from relationalai.loaders.loader import Loader, rel_schema_to_type
 from ..clients.types import ImportSource, ImportSourceFile, ImportSourceTable
 from ..clients.client import ResourcesBase
 from ..tools import debugger as deb, qb_debugger as qb_deb
-from ..clients import config, snowflake
+from ..clients import config
+from relationalai.tools.constants import RAI_APP_NAME
+from relationalai.clients.resources.snowflake.cli_resources import CLIResources
 from relationalai.tools.cli_helpers import (
     EMPTY_STRING_REGEX,
     ENGINE_NAME_ERROR,
@@ -337,7 +339,7 @@ def check_snowflake_connections_flow(cfg:config.Config):
     return True
 
 def role_flow(provider:ResourcesBase, cfg:config.Config):
-    roles = cast(snowflake.Resources, provider).list_roles()
+    roles = cast(CLIResources, provider).list_roles()
     result = controls.fuzzy_with_refetch(
             "Select a role:",
             "roles",
@@ -355,7 +357,7 @@ def warehouse_flow(provider:ResourcesBase, cfg:config.Config):
     result = controls.fuzzy_with_refetch(
             "Select a warehouse:",
             "warehouses",
-            lambda: [w["name"] for w in cast(snowflake.Resources, provider).list_warehouses()],
+            lambda: [w["name"] for w in cast(CLIResources, provider).list_warehouses()],
             default=cfg.get("warehouse", None),
         )
     if not result or isinstance(result, Exception):
@@ -367,12 +369,12 @@ def warehouse_flow(provider:ResourcesBase, cfg:config.Config):
 def rai_app_flow(provider:ResourcesBase, cfg:config.Config):
     auto_select = None
     if provider.config.get("platform") == "snowflake":
-        auto_select=snowflake.RAI_APP_NAME
+        auto_select=RAI_APP_NAME
 
     result = controls.fuzzy_with_refetch(
             "Select RelationalAI app name:",
             "apps",
-            lambda: [w["name"] for w in cast(snowflake.Resources, provider).list_apps()],
+            lambda: [w["name"] for w in cast(CLIResources, provider).list_apps()],
             default=cfg.get("rai_app_name", None),
             auto_select=auto_select
         )
@@ -454,7 +456,8 @@ def name_profile_flow(cfg: config.Config):
         )
         return name_profile_flow(cfg)
     config_store = ConfigStore()
-    if profile in config_store.get_profiles():
+    profiles = config_store.get_profiles() or {}
+    if profile in profiles:
         overwrite = controls.confirm(f"[yellow]Overwrite existing {profile} profile?")
         if overwrite:
             return profile
@@ -464,7 +467,8 @@ def name_profile_flow(cfg: config.Config):
 
 def save_flow(cfg:config.Config):
     config_store = ConfigStore()
-    if cfg.profile != PARTIAL_PROFILE_NAME and cfg.profile in config_store.get_profiles():
+    profiles = config_store.get_profiles() or {}
+    if cfg.profile != PARTIAL_PROFILE_NAME and cfg.profile in profiles:
         if not controls.confirm(f"Overwrite existing {cfg.profile} profile"):
             rich.print()
             profile_name = controls.text("Profile name:")
@@ -570,7 +574,7 @@ def profile_switch(profile:str|None=None):
         rich.print(f"\n[yellow]Error: {e}")
         pass
 
-    profiles = list(config_store.get_profiles().keys())
+    profiles = list((config_store.get_profiles() or {}).keys())
     divider()
     if not profile:
         if len(profiles) == 0:
@@ -672,7 +676,8 @@ def config_explain(profile:str|None=None, all_profiles:bool=False):
             ))
 
     if all_profiles:
-        for profile, props in config_store.get_profiles().items():
+        profiles = config_store.get_profiles() or {}
+        for profile, props in profiles.items():
             if profile == cfg.profile:
                 continue
             if len(props):
@@ -872,7 +877,6 @@ def engines_get(name:str|None=None):
 def create_engine_flow(cfg:config.Config, name=None, size=None, auto_suspend_mins=None):
     provider = get_resource_provider(None, cfg)
     engine = None
-    create_exception = None
     is_engine_present = False
     is_engine_suspended = False
     is_interactive = name is None or size is None
@@ -957,15 +961,7 @@ def create_engine_flow(cfg:config.Config, name=None, size=None, auto_suspend_min
         f"Engine '{name}' created!",
         failed_message="Error:"
     ):
-        try:
-            provider.create_engine(name, size, auto_suspend_mins)
-        except Exception as e:
-            create_exception = e
-            # We do not want to print the success context message if the engine creation fails
-            # Since we are raising here and passing a "fail_message", the spinner will print the actual error message
-            raise e
-    if isinstance(create_exception, Exception):
-        exit_with_error(f"[yellow]{create_exception}")
+        provider.create_engine(name, size, auto_suspend_mins)
     return name
 
 @cli.command(name="engines:create", help="Create a new engine")
@@ -1164,19 +1160,19 @@ def engines_alter_pool(size:str|None=None, min:int|None=None, max:int|None=None)
 def import_source_flow(provider: ResourcesBase) -> Sequence[ImportSource]:
     provider_type = type(provider)
 
-    if isinstance(provider, snowflake.Resources):
+    if isinstance(provider, CLIResources):
         return snowflake_import_source_flow(provider)
     else:
         # Lazy import for azure to avoid optional dependency issues
         try:
-            from relationalai.clients import azure
-            if isinstance(provider, azure.Resources):
+            from relationalai.clients.resources.azure.azure import Resources as AzureResources
+            if isinstance(provider, AzureResources):
                 return azure_import_source_flow(provider)
         except ImportError:
             pass
         raise Exception(f"No import source flow available for {provider_type.__module__}.{provider_type.__name__}")
 
-def snowflake_import_source_flow(provider: snowflake.Resources) -> Sequence[ImportSource]:
+def snowflake_import_source_flow(provider: CLIResources) -> Sequence[ImportSource]:
     with Spinner("Fetching databases", "Databases fetched"):
         try:
             dbs = provider.list_databases()
@@ -1227,7 +1223,7 @@ def azure_import_source_flow(provider: azure.Resources) -> Sequence[ImportSource
 
 def import_source_options_flow(provider: ResourcesBase, source: ImportSource, default_options:dict) -> dict:
     if isinstance(source, ImportSourceFile):
-        type: LoadType = default_options.get("type", None)
+        type: LoadType | None = default_options.get("type", None)
         if type is None or type == "auto":
             type = Loader.get_type_for(source)
         if type == "csv":
@@ -1329,7 +1325,7 @@ def parse_source(provider: ResourcesBase, raw: str) -> ImportSource:
 def imports_setup(engine_size:str|None=None, resume:bool=False, suspend:bool=False):
     divider(flush=True)
     ensure_config()
-    provider = get_resource_provider()
+    provider = cast(CLIResources, get_resource_provider())
     data = None
 
     if resume or suspend:
@@ -1394,7 +1390,7 @@ def imports_setup(engine_size:str|None=None, resume:bool=False, suspend:bool=Fal
 def imports_get(id:str|None=None):
     divider(flush=True)
     ensure_config()
-    provider = get_resource_provider()
+    provider = cast(CLIResources, get_resource_provider())
     import_streams = []
     import_response = []
     with Spinner("Fetching imports", "Imports fetched", "Error:"):
@@ -1567,7 +1563,7 @@ def imports_stream(
 ):
     divider(flush=True)
     ensure_config()
-    provider = get_resource_provider()
+    provider = cast(CLIResources, get_resource_provider())
     default_options = ImportOptionsType.reduce(options)
 
     # Resume or suspend import stream
@@ -1713,7 +1709,7 @@ def imports_snapshot(source:str|None, model:str|None, name:str|None, type:str|No
 def imports_delete(object, model, force):
     divider(flush=True)
     ensure_config()
-    provider = cast(snowflake.Resources, get_resource_provider())
+    provider = cast(CLIResources, get_resource_provider())
     if not model:
         with Spinner("Fetching models", "Models fetched"):
             try:
@@ -1768,7 +1764,7 @@ def imports_delete(object, model, force):
 def exports_list(model):
     divider(flush=True)
     ensure_config()
-    provider = cast(snowflake.Resources, get_resource_provider())
+    provider = cast(CLIResources, get_resource_provider())
     coming_soon()
     if not model:
         with Spinner("Fetching models", "Models fetched"):
@@ -1810,7 +1806,7 @@ def exports_list(model):
 def exports_delete(export, model):
     divider(flush=True)
     ensure_config()
-    provider = cast(snowflake.Resources, get_resource_provider())
+    provider = cast(CLIResources, get_resource_provider())
     coming_soon()
     if not model:
         with Spinner("Fetching models", "Models fetched"):

relationalai/tools/cli_helpers.py

@@ -11,7 +11,6 @@ import functools
 import pytz
 
 from relationalai.util.constants import TOP_LEVEL_PROFILE_NAME
-from .. import Resources
 from rich.table import Table
 from typing import Callable, Dict, Any
 from ..clients import config
@@ -39,7 +38,16 @@ def get_config(profile:str|Dict[str, Any]|None = None):
 @functools.cache
 def get_resource_provider(platform:str|None=None, _cfg:config.Config|None = None) -> ResourcesBase:
     cfg = _cfg or get_config()
-    provider = Resources(config=cfg)
+    platform = platform or cfg.get("platform", "snowflake")
+    if platform == "snowflake":
+        from relationalai.clients.resources.snowflake.cli_resources import CLIResources
+        provider = CLIResources(config=cfg)
+    elif platform == "azure":
+        from relationalai.clients.resources.azure.azure import Resources
+        provider = Resources(config=cfg)
+    else:
+        from .. import Resources
+        provider = Resources(config=cfg)
     return provider
 
 def unexpand_user_path(path):

relationalai/util/otel_configuration.py

@@ -1,7 +1,8 @@
 from typing import cast
 
 from relationalai.clients.config import Config
-from relationalai.clients.snowflake import Resources, ResourcesBase
+from relationalai.clients.resources.snowflake import Resources
+from relationalai.clients.client import ResourcesBase
 from relationalai.environments import runtime_env, SnowbookEnvironment
 
 def configure_otel(enable_otel_handler: bool, config: Config, resources: ResourcesBase):

relationalai/util/otel_handler.py

@@ -17,7 +17,7 @@ from relationalai.util.span_tracker import record_is_span, get_span_path_as_str
 from relationalai.util.span_tracker import get_span_allowed_attributes_values, get_span_value_from_key
 from relationalai.debugging import logger, Span, filter_span_attrs, otel_traceid_to_opentracing, get_current_span
 
-from relationalai.clients.snowflake import Resources
+from relationalai.clients.resources.snowflake import Resources
 
 MAX_PAYLOAD_SIZE = 25*1024 # 25KB
 MAX_ATTRIBUTE_LENGTH = 1000

{relationalai-0.12.13.dist-info → relationalai-0.13.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: relationalai
-Version: 0.12.13
+Version: 0.13.0
 Summary: RelationalAI Library and CLI
 Author-email: RelationalAI <support@relational.ai>
 License-File: LICENSE

{relationalai-0.12.13.dist-info → relationalai-0.13.0.dist-info}/RECORD

@@ -1,10 +1,10 @@
-relationalai/__init__.py,sha256=TxQjYI8QhbHBXvQa4ktMlBbFbgMqGd0-KrlYPvB334Q,8146
+relationalai/__init__.py,sha256=GCSkhHArhDIdRvQHvdkJLs_d3PheeWfD1xJ-IRnaHA8,10397
 relationalai/compiler.py,sha256=5LZY8AyqadkTLSe_xoJ3J9iXOmetz5YQn9EjJKk8XeQ,6411
 relationalai/debugging.py,sha256=wqGly2Yji4ErdV9F_S1Bny35SiGqI3C7S-hap7B8yUs,11997
 relationalai/dependencies.py,sha256=tL113efcISkJUiDXYHmRdU_usdD7gmee-VRHA7N4EFA,16574
 relationalai/docutils.py,sha256=1gVv9mk0ytdMB2W7_NvslJefmSQtTOg8LHTCDcGCjyE,1554
 relationalai/dsl.py,sha256=UJr93X8kwnnyUY-kjPzp_jhsp2pYBUnDfu8mhNXPNII,66116
-relationalai/errors.py,sha256=ERyVg5SJd3O689QSnp-rtbBCLUB3Bbcp5VgkRX1HZzg,94948
+relationalai/errors.py,sha256=JeYaycLGFeT0jidSMERGkpN02zswW5TGrIV647robmw,95973
 relationalai/metagen.py,sha256=o10PNvR_myr_61DC8g6lkB093bFo9qXGUkZKgKyfXiE,26821
 relationalai/metamodel.py,sha256=P1hliwHd1nYxbXON4LZeaYZD6T6pZm97HgmFBFrWyCk,32886
 relationalai/rel.py,sha256=ePmAXx4NxOdsPcHNHyGH3Jkp_cB3QzfKu5p_EQSHPh0,38293
@@ -18,21 +18,30 @@ relationalai/auth/jwt_generator.py,sha256=bQY2-EhvKcidamWB4tiRO8VVl_5bH0GC9Y6ySH
 relationalai/auth/oauth_callback_server.py,sha256=vbcpz77n_WKMDZ4sac6IYyrpxScR0DHlf5VFHMwjHGg,3908
 relationalai/auth/token_handler.py,sha256=d5aueGEiK6BmzkxSab3reCdkhtjdphb0CeecyvvNaQU,19443
 relationalai/auth/util.py,sha256=oXOUwW5gaBhEV5v5A-q2ME1VnfjfRWswvRlXW4oBYpk,1116
-relationalai/clients/__init__.py,sha256=-LHT6vQmleYqAaNerV0OdHy56-s4nxFSmZFsQXfgnrI,318
-relationalai/clients/azure.py,sha256=6tYHxKVN2fHMIDV7J_EHZD9WntkYh2IreMRMqlq1Bhg,20634
-relationalai/clients/cache_store.py,sha256=A-qd11wcwN3TkIqvlN0_iFUU3aEjJal3T2pqFBwkkzQ,3966
-relationalai/clients/client.py,sha256=4SSunUwuFEcRFXOPYotpSLDPr0CKuwJ4335W0DesR90,35792
+relationalai/clients/__init__.py,sha256=LQ_yHsutRMpoW2mOTmOPGF8mrbP0OiV5E68t8uVwDyQ,833
+relationalai/clients/client.py,sha256=gk_V9KS7_MM2dLL2OCO7EPLHD9dsRwR6R-30SW8lDwU,35759
 relationalai/clients/config.py,sha256=hERaKjc3l4kd-kf0l-NUOHrWunCn8gmFWpuE0j3ScJg,24457
 relationalai/clients/direct_access_client.py,sha256=VGjQ7wzduxCo04BkxSZjlPAgqK-aBc32zIXcMfAzzSU,6436
-relationalai/clients/export_procedure.py.jinja,sha256=nhvVcs5hQyWExFDuROQbi1VyYzOCa_ZIRPR2KzZwDtI,10582
 relationalai/clients/hash_util.py,sha256=pZVR1FX3q4G_19p_r6wpIR2tIM8_WUlfAR7AVZJjIYM,1495
-relationalai/clients/local.py,sha256=uX1Or2WO0juDuqa6TCCvm3G2ieP6p0PtYp_jfrCMlVc,23577
+relationalai/clients/local.py,sha256=vo5ikSWg38l3xQAh9yL--4sMAj_T5Tn7YEZiw7TCH08,23504
 relationalai/clients/profile_polling.py,sha256=pUH7WKH4nYDD0SlQtg3wsWdj0K7qt6nZqUw8jTthCBs,2565
 relationalai/clients/result_helpers.py,sha256=wDSD02Ngx6W-YQqBIGKnpXD4Ju3pA1e9Nz6ORRI6SRI,17808
-relationalai/clients/snowflake.py,sha256=pB_4t70vTmfGPUJnJpbrrsNH5pojPJyqodsMtUL6b9c,168261
 relationalai/clients/types.py,sha256=eNo6akcMTbnBFbBbHd5IgVeY-zuAgtXlOs8Bo1SWmVU,2890
-relationalai/clients/use_index_poller.py,sha256=rrkg35xiHqY0-2dZlPkgixEGENrIrl7bf_2TboX_qew,46794
 relationalai/clients/util.py,sha256=NJC8fnrWHR01NydwESPSetIHRWf7jQJURYpaWJjmDyE,12311
+relationalai/clients/resources/__init__.py,sha256=pymn8gB86Q3C2bVoFei0KAL8pX_U04uDY9TE4TKzTBs,260
+relationalai/clients/resources/azure/azure.py,sha256=TDapfM5rLoHrPrXg5cUe827m3AO0gSqQjNid1VUlUFo,20631
+relationalai/clients/resources/snowflake/__init__.py,sha256=9VR-hSIw4ZSEWisKcWhNEcRVBmBfueXNCTOOfLt-8rs,871
+relationalai/clients/resources/snowflake/cache_store.py,sha256=A-qd11wcwN3TkIqvlN0_iFUU3aEjJal3T2pqFBwkkzQ,3966
+relationalai/clients/resources/snowflake/cli_resources.py,sha256=xTIcCzvgbkxuNAEvzZoRpj0n-js0hZCK30q7IZXztbI,3252
+relationalai/clients/resources/snowflake/direct_access_resources.py,sha256=Xvh1e6TxUW2dTSS-9HadrfWVKrxNQ5GikqM4yjohJkM,29849
+relationalai/clients/resources/snowflake/engine_state_handlers.py,sha256=SQBu4GfbyABU6xrEV-koivC-ubsVrfCBTF0FEQgJM5g,12054
+relationalai/clients/resources/snowflake/error_handlers.py,sha256=581G2xOihUoiPlucC_Z2FOzhKu_swdIc3uORd0yJQuA,8805
+relationalai/clients/resources/snowflake/export_procedure.py.jinja,sha256=00iLO2qmvJoqAeJUWt3bAsFDDnof7Ab2spggzelbwK4,10566
+relationalai/clients/resources/snowflake/resources_factory.py,sha256=4LGd4IQ6z8hGeGlO1TIjSFJEeUNHutaB7j9q1a9rYfQ,3385
+relationalai/clients/resources/snowflake/snowflake.py,sha256=kWs7W_jsf_iy2N0VO5Uqu9VHXsbH2rQghPn1s5-9GAg,133037
+relationalai/clients/resources/snowflake/use_index_poller.py,sha256=gMcfILcpD1-wkJ0C1rGZQSy1oZBq6Gm8x9N7jdgEkhA,48448
+relationalai/clients/resources/snowflake/use_index_resources.py,sha256=69PNWHI_uf-Aw_evfwC6j8HLVdjhp84vs8hLkjnhwbg,6462
+relationalai/clients/resources/snowflake/util.py,sha256=BEnm1B1-nqqHdm41RNxblbb-zqXbtqEGGZmTdAYeN_M,13841
 relationalai/early_access/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 relationalai/early_access/builder/__init__.py,sha256=mrR-aGLPquUGc_e9-DOkVCCCo8QRE5A56GTTtgILNa4,993
 relationalai/early_access/builder/builder/__init__.py,sha256=QOVGGdEf-5OirBLjY75W4ANiqGKDknwAAl3YUFEzHVI,1462
@@ -109,7 +118,7 @@ relationalai/early_access/dsl/core/types/constrained/nominal.py,sha256=FsDzuvJgb
 relationalai/early_access/dsl/core/types/constrained/subtype.py,sha256=thEUF3sFpcdlvKdDxl-xNfF1SePOXUva3GRw0nGxsAY,4305
 relationalai/early_access/dsl/ir/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 relationalai/early_access/dsl/ir/compiler.py,sha256=ZU30sfkneBKi3qB3tCY-6xY9Nv8nHsV9dAfNKocj4Hs,15259
-relationalai/early_access/dsl/ir/executor.py,sha256=C9c99vsN6NnDIxE2fLILCvY0ooEHSlZd0g78wKvOow0,12164
+relationalai/early_access/dsl/ir/executor.py,sha256=sct6dS7Ewk37C5Up8TAvth1UlIKde6YwMKFI5h2pobw,12205
 relationalai/early_access/dsl/ontologies/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 relationalai/early_access/dsl/ontologies/constraints.py,sha256=HEQowcD6vFt0WApCmIQ_6mW3Iow_qODi1WpDtnvfYkQ,2423
 relationalai/early_access/dsl/ontologies/export.py,sha256=VJB1n7tjs0-fOBgmLFcRdnR2YRrIHESqob11uS6-KGI,987
@@ -152,7 +161,7 @@ relationalai/early_access/dsl/serialize/binding_model.py,sha256=RvfUUnTjjmtWwDxK
 relationalai/early_access/dsl/serialize/exporter.py,sha256=Z7AO8QNw754uilRq_T7y-LbVW6jb1KHz0Xm1Dxr0ZEM,8009
 relationalai/early_access/dsl/serialize/model.py,sha256=bIzNuw5hV4iPyUZIPTWdG6BevAtl9RnhNW_cymSk7zU,2378
 relationalai/early_access/dsl/snow/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-relationalai/early_access/dsl/snow/api.py,sha256=te2A3U8rfr59R5_PKZQ8O_c4-tdxkuwz4XHU7Pbb9-w,4323
+relationalai/early_access/dsl/snow/api.py,sha256=KDOKX0tF6EAj83p290fYDfoYrXK-VYUwcHVNBvieH8I,4363
 relationalai/early_access/dsl/snow/common.py,sha256=bwUt7z4XgTT3jhPeLlsz6SlZ1YCZDypEHbxQ72PiWfA,2166
 relationalai/early_access/dsl/state_mgmt/__init__.py,sha256=v0dbwg7eGYTcLKuBl5QDAUpN_IAFipBfIVituzBsuk8,4333
 relationalai/early_access/dsl/state_mgmt/state_charts.py,sha256=GmUg-sJk741RRwJEH_wY5cwJdod9EzeEpp7s__cGrZA,4048
@@ -212,7 +221,7 @@ relationalai/experimental/SF.py,sha256=cLIPCgTa6yCkmAU4zw3o2OlSM_Sl8ah3JUpHVoHzr
 relationalai/experimental/__init__.py,sha256=alaOJulJ9YHRl4WGyQ3Kzxhrpc2gP542oNheWXzs7NQ,74
 relationalai/experimental/graphs.py,sha256=vJQ_8ZF_Q2pL-w1xI4O22VGqbnTDimkkqTNqqIkGL7c,3521
 relationalai/experimental/inspect.py,sha256=jpk_xl_bWRcX96xVrHsex-YE3sTqa-W2b5IjGXoxeWk,1635
-relationalai/experimental/solvers.py,sha256=qN35ju-qs5zI_CZlcx9nCFSVr4XDKS8IZ7JzRxztzpM,43316
+relationalai/experimental/solvers.py,sha256=8unvFvzZdRoP0DlHhJvrUFBPvyfXoiqpV3vlZHb3a1Y,43440
 relationalai/experimental/pathfinder/__init__.py,sha256=wzF7zzl5Cvp2CkFaPpY2VU-Tg7I03wvxGq7kp8FVvU0,5429
 relationalai/experimental/pathfinder/api.py,sha256=HT6PAt5gszgJhbnqsHACzZrr9ps9kwcHYu9Nf2EYtTA,4235
 relationalai/experimental/pathfinder/automaton.py,sha256=Z_TYjf5j89-48rxjxl0jV9Q0BR4VhV9cKRTFES7c1t0,21898
@@ -298,23 +307,23 @@ relationalai/loaders/types.py,sha256=w4kvPMpuRoz2ckOCakHrSvtXLetVsQHLi4V3up8hzig
 relationalai/semantics/__init__.py,sha256=SnJCjUGj-y9e5-75jxLsiZRM_xLmhOVaZwmgxtscinI,935
 relationalai/semantics/designs/query_builder/identify_by.md,sha256=AsbusFqT6YB44-iEGfuH4GVOxf1ZjxkpFovvWPMJHuI,4877
 relationalai/semantics/devtools/__init__.py,sha256=hckMw2EtrVBiE71ABXX8bWPdI-6DV9uHqkhYG_nl8S4,85
-relationalai/semantics/devtools/benchmark_lqp.py,sha256=btySKVtUUNBuPY-l6DrkbRNlvZcNXrKUqdvo_Hqsc98,21242
+relationalai/semantics/devtools/benchmark_lqp.py,sha256=VjusVXpyp5u6jEEDoAOISi9A2ALA-DoxBNUmI3t8i9o,21239
 relationalai/semantics/devtools/compilation_manager.py,sha256=XBqG_nYWtK3s_J6MeCmkLqy3qm7l778zDAhH7DEO91g,11169
-relationalai/semantics/devtools/extract_lqp.py,sha256=gxI3EvPUTPAkwgnkCKAkEm2vA6QkLfoM8AXXiVz0c34,3696
+relationalai/semantics/devtools/extract_lqp.py,sha256=uy1R27IA3sNikv6YWtRwZM1wFdMZpH_K_Z6vQX3MkeY,3706
 relationalai/semantics/internal/__init__.py,sha256=BS0WE1GbT-HpQ5p_zMr9Fcvph562WMvMQNI9czfKNt0,798
 relationalai/semantics/internal/annotations.py,sha256=PkrRN-gHO2ksh1hDKB1VVIB39dONvLdTd8_Y0rCR3fE,367
-relationalai/semantics/internal/internal.py,sha256=U3FV4jgWRJGFPk7ZwPzOQl_YMHcwBFH6kXrxJNs5OjA,150750
-relationalai/semantics/internal/snowflake.py,sha256=8D6WYDFKtt8R-sc9o1Oxgtl6Xwehs2Txw_lKNBid7UA,13467
+relationalai/semantics/internal/internal.py,sha256=6sQ9U53zrA9vCdADs3OtmuFwOJG-0LLBAdFf6-kK5mY,150782
+relationalai/semantics/internal/snowflake.py,sha256=yziGDkRMZV-IuRf6qDIm6QzBuQQi0IWy8NLiksTtfMg,13571
 relationalai/semantics/lqp/README.md,sha256=9DsEL81q5z49YteGzLPBs_HhpZS1vO--MOBlQpCkI_Q,1116
 relationalai/semantics/lqp/__init__.py,sha256=XgcQZxK-zz_LqPDVtwREhsIvjTuUIt4BZhIedCeMY-s,48
 relationalai/semantics/lqp/builtins.py,sha256=IWRYJ1J-HGEQqBn8QVOyjZvgEiq6W9tZ0nBLdHz5wjA,576
 relationalai/semantics/lqp/compiler.py,sha256=oOGlN03NVvltKN5KSDOqvb0TsAER_igUe9CnOcHAuoY,943
 relationalai/semantics/lqp/constructors.py,sha256=bGFUv3tLCJD8lGOSGRAPntTBOgczio96_Gn49YoHpl0,2363
-relationalai/semantics/lqp/executor.py,sha256=GuPiSJpLaMYqjZOlVEFLOUilyuLcfkQgZ62UoxMpvtM,21597
+relationalai/semantics/lqp/executor.py,sha256=ibxdY62gTZb6SptcgX86mgPengtxksHqkVXQo-qEERE,21496
 relationalai/semantics/lqp/intrinsics.py,sha256=oKPIcW8PYgU-yPTO21iSF00RBsFKPFFP5MICe6izjKk,871
 relationalai/semantics/lqp/ir.py,sha256=6W9mUH0W7u5eIfF1S3o33uSOfQuM3UcqEkxrxpr1X_8,1867
-relationalai/semantics/lqp/model2lqp.py,sha256=PTe2PtHioWAclu6tdkCu5iGaHJxPIe7zIG1Ew72qxDo,36957
-relationalai/semantics/lqp/passes.py,sha256=bpRCdssqcSlVw77tam6EIjI0BypDoFKGWJix8dTywhU,28621
+relationalai/semantics/lqp/model2lqp.py,sha256=_Udn0SFBu35BOlH__Tc4reXBDiz-izMxP1D6qKfQ7-g,38606
+relationalai/semantics/lqp/passes.py,sha256=ApNFT3R4TmeZzRpNmmm8K9Rl2h6R1T92ioXQPjh8NVc,28631
 relationalai/semantics/lqp/pragmas.py,sha256=FzzldrJEAZ1AIcEw6D-FfaVg3CoahRYgPCFo7xHfg1g,375
 relationalai/semantics/lqp/primitives.py,sha256=9Hjow-Yp06jt0xatuUrH1dw0ErnzknIr9K0TB_AwdjU,11029
 relationalai/semantics/lqp/result_helpers.py,sha256=oYpLoTBnzsiyOVIWA2rLMHlgs7P7BoEkqthQ2aMosnk,10123
@@ -323,15 +332,15 @@ relationalai/semantics/lqp/utils.py,sha256=iOoS-f8kyFjrgAnpK4cWDvAA-WmPgDRggSKUX
 relationalai/semantics/lqp/validators.py,sha256=FlKMKclHj0L71QUtl0aqKknqksSWM-di4N9bjGDJvnY,1561
 relationalai/semantics/lqp/rewrite/__init__.py,sha256=V9ERED9qdh4VvY9Ud_M8Zn8lhVANdOGIgW03l55sGj0,492
 relationalai/semantics/lqp/rewrite/annotate_constraints.py,sha256=b_Ly4_80dQpRzWbeLC72JVfxzhwOPBpiCdEqtBiEiwM,2310
-relationalai/semantics/lqp/rewrite/cdc.py,sha256=I6DeMOZScx-3UAVoSCMn9cuOgLzwdvJVKNwsgFa6R_k,10390
+relationalai/semantics/lqp/rewrite/cdc.py,sha256=8By-BrsayAWYxvrOcoqQ9w0-Nj4KQnaRH8haiog0-4o,10400
 relationalai/semantics/lqp/rewrite/extract_common.py,sha256=ZRvmeYHN8JEkU-j3fRx1e0_JK-46n6NqhxtwZe6L10c,14690
-relationalai/semantics/lqp/rewrite/extract_keys.py,sha256=iSbwGQG9p8j-erknEwl2pZkunJEpXTlnL9ohr4KVS8M,19317
+relationalai/semantics/lqp/rewrite/extract_keys.py,sha256=gC-AkDA4DpqpFkXYc3u-LExbK6e79SrSxs5G6-TEb58,21797
 relationalai/semantics/lqp/rewrite/function_annotations.py,sha256=9ZzLASvXh_OgQ04eup0AyoMIh2HxWHkoRETLm1-XtWs,4660
 relationalai/semantics/lqp/rewrite/functional_dependencies.py,sha256=4oQcVQtAGDqY850B1bNszigQopf6y9Y_CaUyWx42PtM,12718
 relationalai/semantics/lqp/rewrite/quantify_vars.py,sha256=bOowgQ45zmP0HOhsTlE92WdVBCTXSkszcCYbPMeIibw,12004
 relationalai/semantics/lqp/rewrite/splinter.py,sha256=oeDjP_F2PVLVexAKFn8w7CLtO9oy-R-tS2IOmzw_Ujk,3199
 relationalai/semantics/metamodel/__init__.py,sha256=I-XqQAGycD0nKkKYvnF3G9d0QK_1LIM4xXICw8g8fBA,805
-relationalai/semantics/metamodel/builtins.py,sha256=O26klqMH4ud5nqb2Os6ISsxaPsCj5X1eVKD5RuGUmtA,38657
+relationalai/semantics/metamodel/builtins.py,sha256=HN-F5lpnwZiMsl2kBUxdXDSa3LPjVDUIuPF0AbbgqNU,39024
 relationalai/semantics/metamodel/compiler.py,sha256=XBsAnbFwgZ_TcRry6yXGWLyw_MaO2WJDp1EnC_ubhps,4525
 relationalai/semantics/metamodel/dataflow.py,sha256=wfj1tARrR4yEAaTwUTrAcxEcz81VkUal4U_AX1esovk,3929
 relationalai/semantics/metamodel/dependency.py,sha256=iJLx-w_zqde7CtbGcXxLxZBdUKZYl7AUykezPI9ccck,33926
@@ -340,13 +349,13 @@ relationalai/semantics/metamodel/factory.py,sha256=Vk3ASwWan08mfGehoOOwMixuU_mEb
 relationalai/semantics/metamodel/helpers.py,sha256=aeXWkS-iKfLqqXtlMjQZyqIIiIsG9dqP4cQA3cUmM08,15403
 relationalai/semantics/metamodel/ir.py,sha256=4Xl3oc92Q7_s33axtrZUXr-GL8VGJsKc7yaSNcO6jXY,33578
 relationalai/semantics/metamodel/types.py,sha256=9ErIkHgsTcPgsoEet-meiftLs4W1MJb-G1H_1EzFfLY,11557
-relationalai/semantics/metamodel/util.py,sha256=cmSmeww34JVMqcFudwVAY820IPM2ETSELJylwJ3GRJk,16654
+relationalai/semantics/metamodel/util.py,sha256=-_9UMyLvz6bzAI0paXWLWlbqOcPwI9dKopPwEK3ZZhk,16755
 relationalai/semantics/metamodel/visitor.py,sha256=DFY0DACLhxlZ0e4p0vWqbK6ZJr_GWEvH66CU_HVuoTk,35527
 relationalai/semantics/metamodel/rewrite/__init__.py,sha256=9ONWFSdMPHkWpObDMSljt8DywhpFf4Ehsq1aT3fTPt8,344
 relationalai/semantics/metamodel/rewrite/discharge_constraints.py,sha256=0v613BqCLlo4sgWuZjcLSxxakp3d34mYWbG4ldhzGno,1949
 relationalai/semantics/metamodel/rewrite/dnf_union_splitter.py,sha256=piV8FEr4GHKSDcCcsu_TLHJHrsX7blTq4TUvGpGrN8Q,7986
 relationalai/semantics/metamodel/rewrite/extract_nested_logicals.py,sha256=vQ0-7t_GORskB1ZG50KuzM4phm6YNPvehfFn3v_LbgI,3354
-relationalai/semantics/metamodel/rewrite/flatten.py,sha256=CMCFrMCPIQJs9Ln8TJBJYbZaw6EpIcyWh6N6HaZlLQA,22513
+relationalai/semantics/metamodel/rewrite/flatten.py,sha256=oyCYyJrq84YpNC6UpLDfSlUHu5RiE_byLHugQuQ0bCw,22831
 relationalai/semantics/metamodel/rewrite/format_outputs.py,sha256=n0IxC3RL3UMly6MWsq342EGfL2yGj3vOgVG_wg7kt-o,6225
 relationalai/semantics/metamodel/typer/__init__.py,sha256=E3ydmhWRdm-cAqWsNR24_Qd3NcwiHx8ElO2tzNysAXc,143
 relationalai/semantics/metamodel/typer/checker.py,sha256=frY0gilDO6skbDiYFiIpDUOWyt9s9jAJsRBs848DcG0,19184
@@ -355,7 +364,7 @@ relationalai/semantics/reasoners/__init__.py,sha256=cLrGNKFX859EdPjk2n6MdYLvueaF
 relationalai/semantics/reasoners/experimental/__init__.py,sha256=ZWXb3Oun7m_G2c3ijKnqxbEsAzTMVa7ciBXjdi4dCNI,144
 relationalai/semantics/reasoners/graph/README.md,sha256=QgKEXTllp5PO-yK8oDfMx1PNTYF2uVoneMRKsWTY5GU,23953
 relationalai/semantics/reasoners/graph/__init__.py,sha256=jSXR6J05SQZdjxQ5Y-ovqFqGTAXCOWeQDcvpfoBYgDA,1282
-relationalai/semantics/reasoners/graph/core.py,sha256=2ZzGTWlS-IUTqVcdVx0w_GrgxIyppCvHsls1uBjFe4Q,399524
+relationalai/semantics/reasoners/graph/core.py,sha256=sYjSbmn8EKckJq0cN2_UptuYOfvtTjB0CHHzZw4nGkA,399489
 relationalai/semantics/reasoners/graph/design/beyond_demand_transform.md,sha256=Givh0W6B6Hlow6TpmK-8adpEYd8b3O_WmdgMOQIyKs0,55749
 relationalai/semantics/reasoners/graph/tests/README.md,sha256=XbauTzt6VA_YEOcrlZwsu4WOW5UoWV8M6F57wmxe5QU,1023
 relationalai/semantics/reasoners/optimization/__init__.py,sha256=lpavly1Qa3VKvLgrbpp-tsxY9hcqHL6buxuekgKPakw,2212
@@ -365,17 +374,17 @@ relationalai/semantics/reasoners/optimization/solvers_pb.py,sha256=ESwraHU9c4NCE
 relationalai/semantics/rel/__init__.py,sha256=pMlVTC_TbQ45mP1LpzwFBBgPxpKc0H3uJDvvDXEWzvs,55
 relationalai/semantics/rel/builtins.py,sha256=kQToiELc4NnvCmXyFtu9CsGZNdTQtSzTB-nuyIfQcsM,1562
 relationalai/semantics/rel/compiler.py,sha256=pFkEbuPKVd8AI4tiklcv06LbNnK8KfoV4FwmY9Lrhqo,43044
-relationalai/semantics/rel/executor.py,sha256=v-yHl9R8AV0AA2xnm5YZDzue83pr8j2Q97Ky1MKkU70,17309
+relationalai/semantics/rel/executor.py,sha256=GJQea-f1jLLXovcKI4rJ0kYbnnRij2LW_h1XrVY9GII,17621
 relationalai/semantics/rel/rel.py,sha256=9I_V6dQ83QRaLzq04Tt-KjBWhmNxNO3tFzeornBK4zc,15738
 relationalai/semantics/rel/rel_utils.py,sha256=EH-NBROA4vIJXajLKniapt4Dxt7cXSqY4NEjD-wD8Mc,9566
 relationalai/semantics/snowflake/__init__.py,sha256=wRvLb8sQ_SvoY6DEPKpMz6sdSJbrbgyQJVcHAPoSlAw,166
 relationalai/semantics/sql/__init__.py,sha256=59uuKmQOFUzwlYwc7anSuwhQ45bJTYVmd9nOJXr9mQs,80
-relationalai/semantics/sql/compiler.py,sha256=FFcaZYg64bY3HzAk5JugmHced8h2H8eqG4MsIqoDqTM,129171
+relationalai/semantics/sql/compiler.py,sha256=bcntEvcr74HHHAqgqHyrJZwa_TokQcEpVdtcisK9_LA,129203
 relationalai/semantics/sql/sql.py,sha256=7nUnm0RbHlOGSGQbnFrgzPYdmnoppifQ5jylR5Wtgcc,17122
 relationalai/semantics/sql/executor/__init__.py,sha256=F3HqQPJVP9wgV3rkwI5jy1_QBCD_3qj2IGxbdT_pX9k,120
 relationalai/semantics/sql/executor/duck_db.py,sha256=laI0jquMNNhj1pcFlaqxYAvvnCmSuvzzkibfjMz7liY,1909
 relationalai/semantics/sql/executor/result_helpers.py,sha256=kVfspHHuzyq4SNklrtvhYte1wqRFzct-dAKin_lOmR4,3215
-relationalai/semantics/sql/executor/snowflake.py,sha256=ctbSssn91f6cV-_sPGizjkRn7vw_Lcqgd4Ai9S-DTFk,5912
+relationalai/semantics/sql/executor/snowflake.py,sha256=cdQhJL7yO0QacJleoSD45vnLMpK9CmbG4FtANvp3Ud0,5994
 relationalai/semantics/sql/rewrite/__init__.py,sha256=AT1WR0rqQHQ7E06NLoVym0zrZpBVPqK85uRFNJUTDp4,254
 relationalai/semantics/sql/rewrite/denormalize.py,sha256=gPF0s1edthoHfuBIdS9-hnEaX5RFYBdtOlK5iAfB5CA,9040
 relationalai/semantics/sql/rewrite/double_negation.py,sha256=QXynhjwfPW52Hpdo78VWPNk2TXe_mMTWYN470in51z8,1710
@@ -394,7 +403,7 @@ relationalai/semantics/std/std.py,sha256=Ql27y2Rs0d1kluktWi-t6_M_uYIxQUfO94GjlVf
 relationalai/semantics/std/strings.py,sha256=Q_7kvx5dud6gppNURHOi4SMvgZNPRuWwEKDIPSEIOJI,2702
 relationalai/semantics/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 relationalai/semantics/tests/logging.py,sha256=oJTrUS_Bq6WxqqO8QLtrjdkUB02Uu5erZ7FTl__-lNY,1432
-relationalai/semantics/tests/test_snapshot_abstract.py,sha256=ob_5p43CQzz9nfB7Q9W1geKqwNmLyUmQx-mpJfTOwhg,6305
+relationalai/semantics/tests/test_snapshot_abstract.py,sha256=LeXcb3WAPcXZZaSB4JZ9JzPH9ZnjBoN8dSRzSeyt8TQ,6315
 relationalai/semantics/tests/test_snapshot_base.py,sha256=vlqqSyQf_IsDb7feDnkiHZKSCsJqDq0Ep4V3CKtKWns,466
 relationalai/semantics/tests/utils.py,sha256=h4GPNCJGmAWvwskGWxc523BtFEd2ayff4zfBNHapoCo,1825
 relationalai/std/__init__.py,sha256=U8-DpdOcxeGXVSf3LqJl9mSXKztNgB9iwropQcXp960,2178
@@ -407,9 +416,9 @@ relationalai/std/re.py,sha256=7B0dPaYyEdIlEgZfDzs7HJ_MTqccYTfIZieqst6MHbk,4457
 relationalai/std/strings.py,sha256=vHvex_W5GHhhsVws6Dfyl4w1EHdbDE3nPjT09DnpvSE,4260
 relationalai/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 relationalai/tools/cleanup_snapshots.py,sha256=s1q1ophi1RmiTZ9QQolQ6q1K9ZwseCaOuDtI1Rm99MQ,3434
-relationalai/tools/cli.py,sha256=EBrPdJQsltUQGClQTYaEDVmEmm4w_UGdb8fT_DKIeOQ,77017
+relationalai/tools/cli.py,sha256=z6FpwRe3pVro78QDlTkfq6D-Tpb7KpgyscGpVKhJGVM,76841
 relationalai/tools/cli_controls.py,sha256=84K0sjfsVwqi9C8ZlpvPi6XZPhjyW-R06Pir9Ex_Xtg,70468
-relationalai/tools/cli_helpers.py,sha256=r-gJnWnb3d6yIn_ccyp3qx_Vpqh-FINCFMyPBVCA-Jg,14041
+relationalai/tools/cli_helpers.py,sha256=dUIx9Z1I1rTZ3VKtn4jlKGCY5TNk1rl1CY8g3-m99A0,14431
 relationalai/tools/constants.py,sha256=A8Qz0CPs2cRUs5xaZ-Y7BCyvqmI6RLvipzwLDYZ3_PU,4505
 relationalai/tools/debugger.py,sha256=iWO1tMJud3dXHIkrjdZ5Temp0MR2AwN5_QGDt-ddmcs,6934
 relationalai/tools/debugger_client.py,sha256=g9vBaeJLKVq6vW63W-KKSaPJ--g5Wuf4LcIuYRA_pk4,3729
@@ -425,8 +434,8 @@ relationalai/util/constants.py,sha256=rq6c1Q5tLIewiG_M_lUtRAbmrKFUUL1jfl_nxU0989
 relationalai/util/format.py,sha256=fLRovumUa2cu0_2gy3O5vaEbpND4p9_IIgr-vDaIGDc,3754
 relationalai/util/graph.py,sha256=eT8s0yCiJIu6D1T1fjZsLSPCcuQb2Mzl6qnljtQ5TuA,1504
 relationalai/util/list_databases.py,sha256=xJZGHzE0VLaDItWo5XvQSx75OwV045h2rjCBBnhNB3o,152
-relationalai/util/otel_configuration.py,sha256=SV1SFqdKm8q-OD1TsmiIB0x5VxtAFsBRyTwD8zNtnQc,1066
-relationalai/util/otel_handler.py,sha256=2h3ROnra798RRRPLTlEBxEcj-jXXk1ri_l10RAFaiWk,18109
+relationalai/util/otel_configuration.py,sha256=EslwZoaML0WZuEIsMgP1PlrEwsN4Qy3VGabP3Y15yzk,1115
+relationalai/util/otel_handler.py,sha256=a7F5wsc4dI5KF5xpUev2cUwXjd3Lurb198z3AoBS3ME,18119
 relationalai/util/snowflake_handler.py,sha256=8HUo6eU5wPxa6m_EocYuEZD_XpVIh9EqBnGwaS06Usw,2942
 relationalai/util/span_format_test.py,sha256=1rU-M3RXhRt8vCX6rtuIJogEVkB164N3qiKcTmb-sl8,1345
 relationalai/util/span_tracker.py,sha256=SKj4UB43clUxWShq3lFChl3dZ4JGRhcnY_xAh-7Xzvg,7888
@@ -434,7 +443,7 @@ relationalai/util/spans_file_handler.py,sha256=a0sDwDPBBvGsM6be2En3mId9sXpuJlXia
 relationalai/util/timeout.py,sha256=2o6BVNFnFc-B2j-i1pEkZcQbMRto9ps2emci0XwiA4I,783
 relationalai/util/tracing_handler.py,sha256=H919ETAxh7Z1tRz9x8m90qP51_264UunHAPw8Sr6x2g,1729
 relationalai_test_util/__init__.py,sha256=Io_9_IQXXnrUlaL7S1Ndv-4YHilNxy36LrL723MI7lw,118
-relationalai_test_util/fixtures.py,sha256=UL3--4nGWgf6bpOYxFloYCgh5lGVsCeTnTTeKMcA7lY,9169
+relationalai_test_util/fixtures.py,sha256=rNOd8HbguWQi0j63QGoh5iFHLFfM1JUBgDyf87dEsKY,9214
 relationalai_test_util/snapshot.py,sha256=FeH2qYBzLxr2-9qs0yElPIgWUjm_SrzawB3Jgn-aSuE,9291
 relationalai_test_util/traceback.py,sha256=lD0qaEmCyO-7xg9CNf6IzwS-Q-sTS8N9YIv8RroAE50,3298
 frontend/debugger/dist/.gitignore,sha256=JAo-DTfS6GthQGP1NH6wLU-ZymwlTea4KHH_jZVTKn0,14
@@ -442,8 +451,8 @@ frontend/debugger/dist/index.html,sha256=0wIQ1Pm7BclVV1wna6Mj8OmgU73B9rSEGPVX-Wo
 frontend/debugger/dist/assets/favicon-Dy0ZgA6N.png,sha256=tPXOEhOrM4tJyZVJQVBO_yFgNAlgooY38ZsjyrFstgg,620
 frontend/debugger/dist/assets/index-Cssla-O7.js,sha256=MxgIGfdKQyBWgufck1xYggQNhW5nj6BPjCF6Wleo-f0,298886
 frontend/debugger/dist/assets/index-DlHsYx1V.css,sha256=21pZtAjKCcHLFjbjfBQTF6y7QmOic-4FYaKNmwdNZVE,60141
-relationalai-0.12.13.dist-info/METADATA,sha256=qiiOC2n_SXtlfEHafFp8hVqY1M2VSzGOl5fqOPt1_ZQ,2563
-relationalai-0.12.13.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-relationalai-0.12.13.dist-info/entry_points.txt,sha256=fo_oLFJih3PUgYuHXsk7RnCjBm9cqRNR--ab6DgI6-0,88
-relationalai-0.12.13.dist-info/licenses/LICENSE,sha256=pPyTVXFYhirkEW9VsnHIgUjT0Vg8_xsE6olrF5SIgpc,11343
-relationalai-0.12.13.dist-info/RECORD,,
+relationalai-0.13.0.dist-info/METADATA,sha256=RsdnclgO5I9Akv8gxCW-DM5-8ohVCYowAwY-ulQmq6A,2562
+relationalai-0.13.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+relationalai-0.13.0.dist-info/entry_points.txt,sha256=fo_oLFJih3PUgYuHXsk7RnCjBm9cqRNR--ab6DgI6-0,88
+relationalai-0.13.0.dist-info/licenses/LICENSE,sha256=pPyTVXFYhirkEW9VsnHIgUjT0Vg8_xsE6olrF5SIgpc,11343
+relationalai-0.13.0.dist-info/RECORD,,

relationalai_test_util/fixtures.py

@@ -219,7 +219,8 @@ def root_span_fixture(get_full_config=False, span_type: str = "test_session", ex
 
 def otel_collector_fixture(generation):
     config = make_config()
-    resources = rai.clients.snowflake.Resources(config=config, generation=generation)
+    from relationalai.clients.resources.snowflake import Resources
+    resources = Resources(config=config, generation=generation)
     enable_otel_export(
         resources,
         config.get('rai_app_name', 'RELATIONALAI')