relationalai 1.0.0a2__py3-none-any.whl → 1.0.0a3__py3-none-any.whl

relationalai/shims/executor.py

@@ -10,7 +10,10 @@ from v0.relationalai.semantics.rel.executor import RelExecutor
 from v0.relationalai.semantics.metamodel import ir as v0, factory as v0_factory
 from v0.relationalai.semantics.metamodel.visitor import collect_by_type
 from v0.relationalai.semantics.snowflake import Table as v0Table
-from v0.relationalai.clients.snowflake import Provider as v0Provider
+try:
+    from v0.relationalai.clients.snowflake import Provider as v0Provider #type: ignore
+except ImportError:
+    from v0.relationalai.clients.resources.snowflake import Provider as v0Provider
 from v0.relationalai.clients.config import Config
 
 # from ..config import Config

relationalai/shims/mm2v0.py

@@ -456,6 +456,17 @@ class Translator():
     def translate_table(self, t: mm.Table, parent, ctx):
         return self.translate_scalartype(t, parent, ctx)
 
+    def rewrite_cdc_args(self, l: mm.Lookup, args, parent, ctx):
+        # If this is a lookup for an external table column,
+        # we have to prepend the column symbol to the args
+        root_type = l.relation.fields[0].type
+        if isinstance(root_type, mm.Table):
+            self.used_tables.add(root_type)
+            is_table = l.relation == root_type
+            if is_table or l.relation in root_type.columns:
+                sym = "METADATA$KEY" if is_table else l.relation.name
+                args = (v0.Literal(type=v0_types.Symbol, value=sym), *args)
+        return args
 
     # -----------------------------------------------------------------------------
     # Values
@@ -851,19 +862,11 @@ class Translator():
         if relation is None:
             return None
 
-        # External Table Column lookups
-        # we have to prepend the column symbol to the args
-        root_type = l.relation.fields[0].type
-        if isinstance(root_type, mm.Table):
-            self.used_tables.add(root_type)
-            is_table = l.relation == root_type
-            if is_table or l.relation in root_type.columns:
-                sym = "METADATA$KEY" if is_table else l.relation.name
-                args = (v0.Literal(type=v0_types.Symbol, value=sym), *args)
 
         # Specific rewrites
         rewrite = self.rewrite_lookup(l, parent, ctx)
         if rewrite is None:
+            args = self.rewrite_cdc_args(l, args, parent, ctx)
             # General translation
             rewrite = v0.Lookup(
                 engine=self.translate_reasoner(l.reasoner, l, Context.MODEL),
@@ -1100,10 +1103,12 @@ class Translator():
             else:
                 replaced_args.append(arg)
         # translate the lookup
+        args = tuple(self.translate_value(arg, l, ctx) for arg in replaced_args)
+        args = self.rewrite_cdc_args(l, args, l, ctx)
         lookup = v0.Lookup(
             engine=self.translate_reasoner(l.reasoner, l, Context.MODEL),
             relation=self.translate_node(l.relation), # type: ignore
-            args=tuple(self.translate_value(arg, l, ctx) for arg in replaced_args),
+            args=args,
             annotations=self.translate_frozen(l.annotations, l, ctx) # type: ignore
         )
         # subtract 1 from the index to convert from 1-based to 0-based

{relationalai-1.0.0a2.dist-info → relationalai-1.0.0a3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: relationalai
-Version: 1.0.0a2
+Version: 1.0.0a3
 Summary: RelationalAI Library and CLI
 Author-email: RelationalAI <support@relational.ai>
 Requires-Python: >=3.10

{relationalai-1.0.0a2.dist-info → relationalai-1.0.0a3.dist-info}/RECORD

@@ -42,10 +42,10 @@ relationalai/semantics/std/numbers.py,sha256=AcmESbBXIQDkDL6YGZXYKmpo7uyYD1or7su
 relationalai/semantics/std/re.py,sha256=xMuQY5sG3fEL1efHDNFnVpnXVkDJMmDVUEbHiPr1oqA,4375
 relationalai/semantics/std/strings.py,sha256=w0k06gDfyMS6GODPjhG0gl2SkJ0aH5aYezS8jG9jsU8,5617
 relationalai/shims/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-relationalai/shims/executor.py,sha256=KKPno0Jfeb8KK9jSdLl-T5EeBkcxWDC7HDUcMxTQkw4,7405
+relationalai/shims/executor.py,sha256=LtpBx74mhbJ1bCUVvetDlk5iDAEi_x-wNZ6iq5WPW2s,7531
 relationalai/shims/helpers.py,sha256=1v6D-wDq3nCwDz4egdGUeleRC0fK6wfo9fIQIeQCfZA,4508
 relationalai/shims/hoister.py,sha256=YXXi9pWpKQ7Sd5DzRVl_78z4OYCoKzFfIMMp6OIN3EM,8797
-relationalai/shims/mm2v0.py,sha256=xt08HGv06ABxR1u1EZsdDc_2qVuNw5XNLQ8HKuK_YxU,63715
+relationalai/shims/mm2v0.py,sha256=_fqEfd6b3sVwmkuozfggoopRdNeqbsrtJgj8b9Teij8,63957
 relationalai/tools/debugger.py,sha256=B8QIE1-WFdfkiPoFFn1yztr9t6yf5PI32Y9Y9XzQOJk,10798
 relationalai/tools/typer_debugger.py,sha256=__oLODEBwgt6pYIn1jgIPlenFmiK7iqXfTY_mozHxXA,3182
 relationalai/tools/cli/__init__.py,sha256=jkIlp7wftUuSIlczwZD4_siT5cs6mzpsNGrQ_kMkLZ4,74
@@ -66,7 +66,7 @@ relationalai/util/schema.py,sha256=8pzpI4eMg1W4rsyBZtrU_AzCuBxNx4m_cX_XR1BFjCI,6
 relationalai/util/source.py,sha256=k49NhkCDST4Wa_1xJnEsfG2JZIdADlqzqvHt4FCYfq4,6645
 relationalai/util/structures.py,sha256=3QxHkgJSy9oPom6eJX2z1O3pJugINqD5DUjXwoCWwRg,4990
 relationalai/util/tracing.py,sha256=peOVe55R3B9gVdXAsVPGZ7_jw0KvYlkito-_5zERYtg,9632
-v0/relationalai/__init__.py,sha256=iub3QOhxIMGilcne0MUyhvs65k4bU58v-H4Ku-N4cso,8219
+v0/relationalai/__init__.py,sha256=IcozhVGyoDvBaLAbCgGxJ7JBDbgsMhNkEgVzJ2rZFVY,10464
 v0/relationalai/compiler.py,sha256=5LZY8AyqadkTLSe_xoJ3J9iXOmetz5YQn9EjJKk8XeQ,6411
 v0/relationalai/debugging.py,sha256=AuzbwLT1s71tctuIbYNhfWY2vR_opB7ItbmlqFVHgrY,12024
 v0/relationalai/dependencies.py,sha256=tL113efcISkJUiDXYHmRdU_usdD7gmee-VRHA7N4EFA,16574
@@ -86,20 +86,29 @@ v0/relationalai/auth/jwt_generator.py,sha256=bQY2-EhvKcidamWB4tiRO8VVl_5bH0GC9Y6
 v0/relationalai/auth/oauth_callback_server.py,sha256=vbcpz77n_WKMDZ4sac6IYyrpxScR0DHlf5VFHMwjHGg,3908
 v0/relationalai/auth/token_handler.py,sha256=d5aueGEiK6BmzkxSab3reCdkhtjdphb0CeecyvvNaQU,19443
 v0/relationalai/auth/util.py,sha256=oXOUwW5gaBhEV5v5A-q2ME1VnfjfRWswvRlXW4oBYpk,1116
-v0/relationalai/clients/__init__.py,sha256=-LHT6vQmleYqAaNerV0OdHy56-s4nxFSmZFsQXfgnrI,318
-v0/relationalai/clients/azure.py,sha256=nmOFdw64e7z7eWMS-TZ7X4_qCAdTOEe5V7uTzqS9MFY,20640
-v0/relationalai/clients/cache_store.py,sha256=A-qd11wcwN3TkIqvlN0_iFUU3aEjJal3T2pqFBwkkzQ,3966
-v0/relationalai/clients/client.py,sha256=46_fu8xKR5m6HnTeI9UhIEZtojPrsc-3hMhTutObzxY,35798
+v0/relationalai/clients/__init__.py,sha256=LQ_yHsutRMpoW2mOTmOPGF8mrbP0OiV5E68t8uVwDyQ,833
+v0/relationalai/clients/client.py,sha256=gk_V9KS7_MM2dLL2OCO7EPLHD9dsRwR6R-30SW8lDwU,35759
 v0/relationalai/clients/config.py,sha256=PriDlD0_VZFV0fEQng6U-F5xjGwz5gC2F5_lM159AjM,24481
 v0/relationalai/clients/direct_access_client.py,sha256=p3hqjwdiXQQytRke1wIeNRM-HlZLAxH1U6-yPhsPOdQ,6454
 v0/relationalai/clients/hash_util.py,sha256=NXVtUFgRokZz-45CwV-r5Z5KvmEXD2dh4ypNOrEWAwY,1501
-v0/relationalai/clients/local.py,sha256=cq3fRoayPxjqh6iW3hqfdmDgDbcuM73IcndMeEb_Nhk,23589
+v0/relationalai/clients/local.py,sha256=vo5ikSWg38l3xQAh9yL--4sMAj_T5Tn7YEZiw7TCH08,23504
 v0/relationalai/clients/profile_polling.py,sha256=c1ixxAOKvy-dDfVqkKENwfcsvBB8jhLA1oT9RIFIrY8,2571
 v0/relationalai/clients/result_helpers.py,sha256=dgj4mVg7ZsFsVppI98VQjNMdDEBXuBF7BnyNozidmeY,17817
-v0/relationalai/clients/snowflake.py,sha256=xAyCYWfDE0n7MGdwFBfhVg248QaDzxaKArAObJe0Gho,168282
 v0/relationalai/clients/types.py,sha256=VU2LRtlnFsBRYdokxUccrsUkEYLE9N62EUiWtA_qdYc,2893
-v0/relationalai/clients/use_index_poller.py,sha256=34KQyNr_TL85hWsd803R0D_Ms4P6grQhOr5KSXDyszs,46830
 v0/relationalai/clients/util.py,sha256=Vrw_kr-Oqp_DKTkJ_rTbzpDm4_Z4h8w10T8GUyGLxbA,12323
+v0/relationalai/clients/resources/__init__.py,sha256=pymn8gB86Q3C2bVoFei0KAL8pX_U04uDY9TE4TKzTBs,260
+v0/relationalai/clients/resources/azure/azure.py,sha256=TDapfM5rLoHrPrXg5cUe827m3AO0gSqQjNid1VUlUFo,20631
+v0/relationalai/clients/resources/snowflake/__init__.py,sha256=9VR-hSIw4ZSEWisKcWhNEcRVBmBfueXNCTOOfLt-8rs,871
+v0/relationalai/clients/resources/snowflake/cache_store.py,sha256=A-qd11wcwN3TkIqvlN0_iFUU3aEjJal3T2pqFBwkkzQ,3966
+v0/relationalai/clients/resources/snowflake/cli_resources.py,sha256=xTIcCzvgbkxuNAEvzZoRpj0n-js0hZCK30q7IZXztbI,3252
+v0/relationalai/clients/resources/snowflake/direct_access_resources.py,sha256=Xvh1e6TxUW2dTSS-9HadrfWVKrxNQ5GikqM4yjohJkM,29849
+v0/relationalai/clients/resources/snowflake/engine_state_handlers.py,sha256=SQBu4GfbyABU6xrEV-koivC-ubsVrfCBTF0FEQgJM5g,12054
+v0/relationalai/clients/resources/snowflake/error_handlers.py,sha256=581G2xOihUoiPlucC_Z2FOzhKu_swdIc3uORd0yJQuA,8805
+v0/relationalai/clients/resources/snowflake/resources_factory.py,sha256=4LGd4IQ6z8hGeGlO1TIjSFJEeUNHutaB7j9q1a9rYfQ,3385
+v0/relationalai/clients/resources/snowflake/snowflake.py,sha256=UiSE2jbJVGbe2_UfLI9Gud9aA5SJysuVu3F0McOz95k,133046
+v0/relationalai/clients/resources/snowflake/use_index_poller.py,sha256=4lPMgaeuxUdWo-ds_78OJpc7pyogWhSVgGzuTzj13wE,48460
+v0/relationalai/clients/resources/snowflake/use_index_resources.py,sha256=69PNWHI_uf-Aw_evfwC6j8HLVdjhp84vs8hLkjnhwbg,6462
+v0/relationalai/clients/resources/snowflake/util.py,sha256=BEnm1B1-nqqHdm41RNxblbb-zqXbtqEGGZmTdAYeN_M,13841
 v0/relationalai/early_access/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 v0/relationalai/early_access/builder/__init__.py,sha256=h2zZSEF6xcuk0Ysy8CRzF172usliAs9VUW5t6P3rOBw,996
 v0/relationalai/early_access/builder/builder/__init__.py,sha256=0nUNYRT0QdAlBU0S7aovypHQBdBrBHumr3Ps_gG3R_E,1468
@@ -176,7 +185,7 @@ v0/relationalai/early_access/dsl/core/types/constrained/nominal.py,sha256=jfpqGs
 v0/relationalai/early_access/dsl/core/types/constrained/subtype.py,sha256=qODz36SaWywRHALQLy1d2V8gPqHfb6F357ShRrbacJA,4311
 v0/relationalai/early_access/dsl/ir/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 v0/relationalai/early_access/dsl/ir/compiler.py,sha256=W_hLa6FoqfF140G01aq6qpYtfpxVZRsxS37Ss7N2mUY,15313
-v0/relationalai/early_access/dsl/ir/executor.py,sha256=U10L5OnV9HEL_3-o9FMuJrEei_Ua6nv9FmIVJWqWLjA,12200
+v0/relationalai/early_access/dsl/ir/executor.py,sha256=RSuFpgvkV7ILdE3krUNOHg0ZplpHKlQlbV6GWCAR1d4,12241
 v0/relationalai/early_access/dsl/ontologies/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 v0/relationalai/early_access/dsl/ontologies/constraints.py,sha256=do7IR0sN9S5mBEf79adtmZq-RkOc3HNg9Q1ZbXjlAsY,2429
 v0/relationalai/early_access/dsl/ontologies/export.py,sha256=wyDUjwQ9TwCFKwN8m9S-7G4kKaI5dk4nvecbCy-W5Hc,999
@@ -219,7 +228,7 @@ v0/relationalai/early_access/dsl/serialize/binding_model.py,sha256=RvfUUnTjjmtWw
 v0/relationalai/early_access/dsl/serialize/exporter.py,sha256=oXBxSjHvms7bzIHYb_dU1ItOchbjpDiSjkhXRD-oF2c,8024
 v0/relationalai/early_access/dsl/serialize/model.py,sha256=bIzNuw5hV4iPyUZIPTWdG6BevAtl9RnhNW_cymSk7zU,2378
 v0/relationalai/early_access/dsl/snow/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-v0/relationalai/early_access/dsl/snow/api.py,sha256=UYveBVEKVP1xYzYsax1e2ujc_mMzwz7-v7zP1VlSgAA,4332
+v0/relationalai/early_access/dsl/snow/api.py,sha256=NyOPLanXZaqtINOo9e7M-tdu2Jlo9KFw2G7mYrAKMes,4375
 v0/relationalai/early_access/dsl/snow/common.py,sha256=i8L-TXw1RLUVB9NK8LpFkQAkUquIkbdh1nU0ePsGkCA,2172
 v0/relationalai/early_access/dsl/state_mgmt/__init__.py,sha256=W0hzq5Oa2yy9YoQfC5Ru5UqggyzIhsryljmVt_NCkKQ,4339
 v0/relationalai/early_access/dsl/state_mgmt/state_charts.py,sha256=zBQS9vdEYx5iRSLbp31k4xz8HFDp2EMKTgVQfzaPQlM,4054
@@ -279,7 +288,7 @@ v0/relationalai/experimental/SF.py,sha256=ER6NFB_Vo6hjxdwgDFshFMx_huCwhqOYfEqzFn
 v0/relationalai/experimental/__init__.py,sha256=alaOJulJ9YHRl4WGyQ3Kzxhrpc2gP542oNheWXzs7NQ,74
 v0/relationalai/experimental/graphs.py,sha256=vJQ_8ZF_Q2pL-w1xI4O22VGqbnTDimkkqTNqqIkGL7c,3521
 v0/relationalai/experimental/inspect.py,sha256=CkZVzq1MTXf5aOsqHYZaDXYNegW8Er-7Z4NAJqT9QWM,1650
-v0/relationalai/experimental/solvers.py,sha256=FY9l9NCbEmJxyK8QJgUq6KwRkSw-4R8GQYzrNWyQYI4,43328
+v0/relationalai/experimental/solvers.py,sha256=boZc8DNKTZa-p3AAAtMrYX23nzNSnhw03TlXPm0OO8I,43446
 v0/relationalai/experimental/pathfinder/__init__.py,sha256=GSGTur8UMhhv9u3VmZAWp6OxtDymZRaHDqgV9V2IFuc,5519
 v0/relationalai/experimental/pathfinder/api.py,sha256=iRf9QKq5xIVqiPPQEQGx9BHxbT-dME0kmn5ox6IBCDM,4241
 v0/relationalai/experimental/pathfinder/automaton.py,sha256=y5i3ry_gIPoAp0ZTsCNXTXSutst6VcGCg2VZ5Z_N9Yw,21910
@@ -352,18 +361,18 @@ v0/relationalai/loaders/loader.py,sha256=SmdV5u8NPvc72eb03sQWDD2nqCNDEsueh8LJ3kL
 v0/relationalai/loaders/types.py,sha256=1Fnr-TQc21Vh_Vwrzpmqm7ILr9-lvogOBE0OXGLG7mg,622
 v0/relationalai/semantics/__init__.py,sha256=SnJCjUGj-y9e5-75jxLsiZRM_xLmhOVaZwmgxtscinI,935
 v0/relationalai/semantics/devtools/__init__.py,sha256=hckMw2EtrVBiE71ABXX8bWPdI-6DV9uHqkhYG_nl8S4,85
-v0/relationalai/semantics/devtools/benchmark_lqp.py,sha256=kULTss_dHTsCdMlj_2FfaMGT5hPAXEs1BdMwK3rCMgQ,21260
+v0/relationalai/semantics/devtools/benchmark_lqp.py,sha256=02bMXOYAVYLzi1pYF9qGa9cBV7IH_-VTRNBL76Iq158,21254
 v0/relationalai/semantics/devtools/compilation_manager.py,sha256=DDHY5DNwRK0hD9P8ffBUfrGIG24QI7HDj8XKZulgpog,11181
-v0/relationalai/semantics/devtools/extract_lqp.py,sha256=Ccvhw3W-8VHYi2Kui64lCUmCrliygGqFc6C0LQlvgWM,3702
+v0/relationalai/semantics/devtools/extract_lqp.py,sha256=R4F718cH2uYHz0lx66JHy-2Qj6QbWfqeR-ApAxyVqvU,3712
 v0/relationalai/semantics/internal/__init__.py,sha256=BS0WE1GbT-HpQ5p_zMr9Fcvph562WMvMQNI9czfKNt0,798
 v0/relationalai/semantics/internal/annotations.py,sha256=PkrRN-gHO2ksh1hDKB1VVIB39dONvLdTd8_Y0rCR3fE,367
 v0/relationalai/semantics/internal/internal.py,sha256=-juUQ8AL1kAoh6PgpBQJ5Jp3qjwSgjx64s3IKeKXVYk,150821
-v0/relationalai/semantics/internal/snowflake.py,sha256=Zw-jC2XmsWwZlOn_5zwfu_gCxs-dH0XlSOcAELAc70k,13588
+v0/relationalai/semantics/internal/snowflake.py,sha256=rQpxtR8BImj_ulTbmN0PIpMz0CvJrcC0-TPrTxCSa3k,13598
 v0/relationalai/semantics/lqp/__init__.py,sha256=XgcQZxK-zz_LqPDVtwREhsIvjTuUIt4BZhIedCeMY-s,48
 v0/relationalai/semantics/lqp/builtins.py,sha256=Xst9-UxkCAwDqL-hFOby8B3GKh705oidFUjEozxn71Y,585
 v0/relationalai/semantics/lqp/compiler.py,sha256=xpGIQAqTMwLi3Vrbe79Yal7O_mfNxpBLioZjqeV2hzc,955
 v0/relationalai/semantics/lqp/constructors.py,sha256=x7G85vbJzLIXzUCMKVdJVsr4PrcUsAOzCCfvimHLMVU,2366
-v0/relationalai/semantics/lqp/executor.py,sha256=DeZw71K3TAXI8mZDWa4c9N3V3HQfOVn1YNIPhuKcFcU,21875
+v0/relationalai/semantics/lqp/executor.py,sha256=3tJVxFsUMdevC-HGYR2BjexC4nBoblUBSf87xlIZ_g0,21469
 v0/relationalai/semantics/lqp/intrinsics.py,sha256=8kgH4ndYKBCNnoNOn7iWOU0wnMTSiov0kYaKocQ45RY,880
 v0/relationalai/semantics/lqp/ir.py,sha256=6W9mUH0W7u5eIfF1S3o33uSOfQuM3UcqEkxrxpr1X_8,1867
 v0/relationalai/semantics/lqp/model2lqp.py,sha256=J_QofcYgq_Yb5dhOOP4lXvxsxFMEGHVnaNXoCMK_ak8,38639
@@ -393,7 +402,7 @@ v0/relationalai/semantics/metamodel/factory.py,sha256=ECOg7Fg6DJrJDKIzv3TZqhuN4R
 v0/relationalai/semantics/metamodel/helpers.py,sha256=G6p6XgqeM9b95D4yUaiq71l5WY40Y1neLgMjqk-ofeU,15409
 v0/relationalai/semantics/metamodel/ir.py,sha256=4Xl3oc92Q7_s33axtrZUXr-GL8VGJsKc7yaSNcO6jXY,33578
 v0/relationalai/semantics/metamodel/types.py,sha256=9ErIkHgsTcPgsoEet-meiftLs4W1MJb-G1H_1EzFfLY,11557
-v0/relationalai/semantics/metamodel/util.py,sha256=cmSmeww34JVMqcFudwVAY820IPM2ETSELJylwJ3GRJk,16654
+v0/relationalai/semantics/metamodel/util.py,sha256=-_9UMyLvz6bzAI0paXWLWlbqOcPwI9dKopPwEK3ZZhk,16755
 v0/relationalai/semantics/metamodel/visitor.py,sha256=DFY0DACLhxlZ0e4p0vWqbK6ZJr_GWEvH66CU_HVuoTk,35527
 v0/relationalai/semantics/metamodel/rewrite/__init__.py,sha256=9ONWFSdMPHkWpObDMSljt8DywhpFf4Ehsq1aT3fTPt8,344
 v0/relationalai/semantics/metamodel/rewrite/discharge_constraints.py,sha256=jilU6ZNaqhoFRwNdxArSfYb3Rbeahig9kDjdYoJR9Q4,1955
@@ -415,7 +424,7 @@ v0/relationalai/semantics/reasoners/optimization/solvers_pb.py,sha256=1RobVkhUDP
 v0/relationalai/semantics/rel/__init__.py,sha256=pMlVTC_TbQ45mP1LpzwFBBgPxpKc0H3uJDvvDXEWzvs,55
 v0/relationalai/semantics/rel/builtins.py,sha256=0M9r5GQb9CnkkbdhO8Psw9TgiZk5CC3z5RMncUa6uuM,1571
 v0/relationalai/semantics/rel/compiler.py,sha256=t7gitonfRewTk8IQ53DTqPegytwax1jGu43VgQOqBWA,43062
-v0/relationalai/semantics/rel/executor.py,sha256=47TeoS7qlKCp5uwddSujg9B2sdjdy796TnJF-3owrTU,17348
+v0/relationalai/semantics/rel/executor.py,sha256=fJnOb6Lgv32Qrr8V72i4M6ebOuEiqqyOcSuffUok3AI,17660
 v0/relationalai/semantics/rel/rel.py,sha256=mQTykcHz63P1w8nhvA8SwYRtIi0LEud19vGrdh0ZrG8,15747
 v0/relationalai/semantics/rel/rel_utils.py,sha256=B9qR7hUouybwluuo9zUYemv6tWPnDPaV4da-emsTrRQ,9569
 v0/relationalai/semantics/snowflake/__init__.py,sha256=QRcuGvnK-TeNOWvHpBu5Z4hJHcMRXb5IPOokvLihy1U,169
@@ -425,7 +434,7 @@ v0/relationalai/semantics/sql/sql.py,sha256=a3EBJKPp92EoVpl1qct2jdBwXQflFpG1nJ2O
 v0/relationalai/semantics/sql/executor/__init__.py,sha256=F3HqQPJVP9wgV3rkwI5jy1_QBCD_3qj2IGxbdT_pX9k,120
 v0/relationalai/semantics/sql/executor/duck_db.py,sha256=FriWhN_IvpQ1YH0YcQ9vlnPRkM1ARoqKDZd0koSk0SM,1918
 v0/relationalai/semantics/sql/executor/result_helpers.py,sha256=WFeijyYXUSl2Ut25SNFYTGZ-XktTlz8CZQt5xm9QW9o,3224
-v0/relationalai/semantics/sql/executor/snowflake.py,sha256=DVRxApKTzsVjoutOGzYrp8pM1ksAiPBgQ7awA0Z-9iQ,5939
+v0/relationalai/semantics/sql/executor/snowflake.py,sha256=ItM-dcC-OynlAQNtXTdmVdSyJtKuY24_9_vIXDFP9-A,6024
 v0/relationalai/semantics/sql/rewrite/__init__.py,sha256=AT1WR0rqQHQ7E06NLoVym0zrZpBVPqK85uRFNJUTDp4,254
 v0/relationalai/semantics/sql/rewrite/denormalize.py,sha256=zCwjt8FvcOjs88GUOzLzoqYcxPVlARrYtbVc78yQeBI,9046
 v0/relationalai/semantics/sql/rewrite/double_negation.py,sha256=Rm2-g_gbPPYzqMfsYWX0Z9rm_d-kGVuNRBFNrUaMBbo,1713
@@ -444,7 +453,7 @@ v0/relationalai/semantics/std/std.py,sha256=T9cPRC19I3WlolDyHk8nwVzGRyyYCpbagN1v
 v0/relationalai/semantics/std/strings.py,sha256=3fw1VAggsuFosH_3-5BIB4KbKDmqVque9FlV3yJusrE,2705
 v0/relationalai/semantics/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 v0/relationalai/semantics/tests/logging.py,sha256=oJTrUS_Bq6WxqqO8QLtrjdkUB02Uu5erZ7FTl__-lNY,1432
-v0/relationalai/semantics/tests/test_snapshot_abstract.py,sha256=FMzia4QeiNP-JZ5E0vt6YlL0SMSESB3flg8-ybWb4jU,6326
+v0/relationalai/semantics/tests/test_snapshot_abstract.py,sha256=Q2rIUhzwVAJMJVoCsBtrCVBYafUR7nTQUxO2bL3ECa8,6336
 v0/relationalai/semantics/tests/test_snapshot_base.py,sha256=Q2zThxDOp2FrXK4uhV4nStBc_Q2-ML6MJ-dfE-e6R7U,469
 v0/relationalai/semantics/tests/utils.py,sha256=qcZ5IgU6_vRdKI9yLuoUkBE0Cq0iU6yzPh5HQaH3ttA,1834
 v0/relationalai/std/__init__.py,sha256=r-o7hYZip7xe1stQP4R10jFNA_Km1dcFwmr2j7LBF3s,2184
@@ -457,9 +466,9 @@ v0/relationalai/std/re.py,sha256=7B0dPaYyEdIlEgZfDzs7HJ_MTqccYTfIZieqst6MHbk,445
 v0/relationalai/std/strings.py,sha256=vHvex_W5GHhhsVws6Dfyl4w1EHdbDE3nPjT09DnpvSE,4260
 v0/relationalai/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 v0/relationalai/tools/cleanup_snapshots.py,sha256=s1q1ophi1RmiTZ9QQolQ6q1K9ZwseCaOuDtI1Rm99MQ,3434
-v0/relationalai/tools/cli.py,sha256=AbfliJiUJUDIbqrazLQIv-sHTJ12qI3jcEmIRE-O0R0,77053
+v0/relationalai/tools/cli.py,sha256=7Ads7TGIqqUviVCW1YoVK14chhLu2v2Vr4Zo9GrInNg,76883
 v0/relationalai/tools/cli_controls.py,sha256=PgXdmJ7JJxk_yUyhljPBV7A4qbAzvSMvjhybni7bJwo,70474
-v0/relationalai/tools/cli_helpers.py,sha256=0oF3XeVcq8Ti4nqegkb_tJh2zXLje_AzC3pgdRrah4k,14053
+v0/relationalai/tools/cli_helpers.py,sha256=z6RGKDNwOK8DjWtd-rxiSNdEtnwdF8Kh5d0xV9xtjkI,14449
 v0/relationalai/tools/constants.py,sha256=A8Qz0CPs2cRUs5xaZ-Y7BCyvqmI6RLvipzwLDYZ3_PU,4505
 v0/relationalai/tools/debugger.py,sha256=CiPy3czIR80TfnjGTTpiDSJC2ZlPneAtlddPEMri7Us,6937
 v0/relationalai/tools/debugger_client.py,sha256=lh2ejDZkAmOBu2qOAdVqH0XnqMKM7azGmXXDpTSoq30,3741
@@ -474,16 +483,16 @@ v0/relationalai/util/constants.py,sha256=rq6c1Q5tLIewiG_M_lUtRAbmrKFUUL1jfl_nxU0
 v0/relationalai/util/format.py,sha256=fLRovumUa2cu0_2gy3O5vaEbpND4p9_IIgr-vDaIGDc,3754
 v0/relationalai/util/graph.py,sha256=eT8s0yCiJIu6D1T1fjZsLSPCcuQb2Mzl6qnljtQ5TuA,1504
 v0/relationalai/util/list_databases.py,sha256=dHu9xQkL1MxORrHMqR5E38GybzIjsfayCnFKFKEzXL4,155
-v0/relationalai/util/otel_configuration.py,sha256=38WfHkDMsgTX0H8AWc67XWgGA36kMoXxmk1u8IkAeAM,1075
-v0/relationalai/util/otel_handler.py,sha256=a-It_P-vxBLJjyf9apOVerEcj4aU6IT0JMlD4io-UCo,18121
+v0/relationalai/util/otel_configuration.py,sha256=rt6jQsWBevQNMkTBT3RDHR7r-py2O05vRgx4y2XAdZc,1127
+v0/relationalai/util/otel_handler.py,sha256=lG4fcCb4C8PvQ38c53uhW1-9--son6qztT0QmzbenWU,18131
 v0/relationalai/util/snowflake_handler.py,sha256=mm5hm4FjjaTkDwZGocOkRK4DmLQqFWLMV05pv7mmvUg,2945
 v0/relationalai/util/span_format_test.py,sha256=djFxvHx7IDdh0QwU4DBThTd8OQBBZTBDN4dGcpOwf1Y,1348
 v0/relationalai/util/span_tracker.py,sha256=7vyvEBdI9uCj1dxVZCQT_W5vPMxIzxewrlRRkvD5FGA,7900
 v0/relationalai/util/spans_file_handler.py,sha256=w34WVw19ubtrjPBkKmg0Hb-9MldICmowF1OW85KCJN0,3209
 v0/relationalai/util/timeout.py,sha256=2o6BVNFnFc-B2j-i1pEkZcQbMRto9ps2emci0XwiA4I,783
 v0/relationalai/util/tracing_handler.py,sha256=wJQN52PRw8R2XB1_qd-POgfGZStrj9OFL7wa5Xr6SEM,1732
-relationalai-1.0.0a2.dist-info/METADATA,sha256=j3ofGijkfDIsTfki1G_l7ycajPA4m9hLrUWYIuNI9dg,1368
-relationalai-1.0.0a2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-relationalai-1.0.0a2.dist-info/entry_points.txt,sha256=u_anMN5_VCOQNA5E2mTLT2LWc9r1i-F9yQmy3ZNAAck,91
-relationalai-1.0.0a2.dist-info/top_level.txt,sha256=Y9cfzWf-p2omqqmVy_98m287xf0OJbj6OB5RRdkGql4,16
-relationalai-1.0.0a2.dist-info/RECORD,,
+relationalai-1.0.0a3.dist-info/METADATA,sha256=4hBJy9pOGv9uDUiDfIbGTGOX0M_N94_EptAIzzR7csc,1368
+relationalai-1.0.0a3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+relationalai-1.0.0a3.dist-info/entry_points.txt,sha256=u_anMN5_VCOQNA5E2mTLT2LWc9r1i-F9yQmy3ZNAAck,91
+relationalai-1.0.0a3.dist-info/top_level.txt,sha256=Y9cfzWf-p2omqqmVy_98m287xf0OJbj6OB5RRdkGql4,16
+relationalai-1.0.0a3.dist-info/RECORD,,

v0/relationalai/__init__.py

@@ -7,8 +7,8 @@ import importlib.metadata
 
 from typing import cast
 
-from .clients import config as cfg
 from .clients.config import Config, save_config
+from . import clients
 from . import dsl
 from . import debugging
 from . import metamodel
@@ -19,11 +19,14 @@ from . import tools
 from .util.otel_configuration import configure_otel
 from snowflake.snowpark import Session
 from .errors import RAIException, handle_missing_integration
-from .environments import runtime_env, SessionEnvironment
-from v0.relationalai.tools.constants import USE_DIRECT_ACCESS, Generation
+from .environments import runtime_env, SessionEnvironment, SnowbookEnvironment
+from v0.relationalai.tools.constants import Generation
 
 import __main__
 
+# Define cfg after all imports since it depends on clients being imported
+cfg = clients.config
+
 # __version__ = importlib.metadata.version(__package__ or __name__)
 
 __version__ = '1.0.0a'
@@ -114,9 +117,9 @@ def Model(
 
     try:
         if platform == "azure":
-            import v0.relationalai.clients.azure as azure
+            from v0.relationalai.clients.resources.azure.azure import Graph
             from .util.otel_handler import disable_otel_handling, is_otel_initialized
-            model = azure.Graph(
+            model = Graph(
                 name,
                 profile=profile,
                 config=config,
@@ -128,8 +131,18 @@ def Model(
             if is_otel_initialized:
                 disable_otel_handling()
         elif platform == "snowflake":
-            from v0.relationalai.clients import snowflake
-            model = snowflake.Graph(
+            try:
+                from v0.relationalai.clients.resources.snowflake import Graph
+            except ImportError as e:
+                # Provide a helpful error message for Snowflake notebook environments
+                if isinstance(runtime_env, SnowbookEnvironment):
+                    raise ImportError(
+                        "Failed to import relationalai.clients.resources.snowflake. "
+                        "This may indicate that the relationalai.zip package structure is incomplete. "
+                        "Please ensure the zip file includes the full clients/resources/snowflake directory structure."
+                    ) from e
+                raise
+            model = Graph(
                 name,
                 profile=profile,
                 config=config,
@@ -164,18 +177,41 @@ def Resources(
     config = config or Config(profile)
     platform = config.get("platform", "snowflake")
     if platform == "azure":
-        from v0.relationalai.clients.azure import Resources
+        from v0.relationalai.clients.resources.azure.azure import Resources
         return Resources(config=config)
     elif platform == "snowflake":
-        from v0.relationalai.clients.snowflake import Resources, DirectAccessResources
-        use_direct_access = config.get("use_direct_access", USE_DIRECT_ACCESS)
-        if use_direct_access:
-            return DirectAccessResources(config=config, connection=connection, reset_session=reset_session, generation=generation)
-        else:
-            return Resources(config=config, connection=connection, reset_session=reset_session, generation=generation)
+        try:
+            from v0.relationalai.clients.resources.snowflake.resources_factory import create_resources_instance
+        except ImportError as e:
+            # Provide a helpful error message for Snowflake notebook environments
+            from v0.relationalai.environments import runtime_env, SnowbookEnvironment
+            if isinstance(runtime_env, SnowbookEnvironment):
+                raise ImportError(
+                    "Failed to import relationalai.clients.resources.snowflake. "
+                    "This may indicate that the relationalai.zip package structure is incomplete. "
+                    "Please ensure the zip file includes the full clients/resources/snowflake directory structure."
+                ) from e
+            raise
+        return create_resources_instance(
+            config=config,
+            profile=profile,
+            connection=connection,
+            reset_session=reset_session,
+            generation=generation or Generation.V0,
+            dry_run=False,
+            language="rel",
+        )
     elif platform == "local":
-        from v0.relationalai.clients.local import LocalResources
-        return LocalResources(config=config, connection=connection, reset_session=reset_session, generation=generation)
+        from v0.relationalai.clients.resources.snowflake.resources_factory import create_resources_instance
+        return create_resources_instance(
+            config=config,
+            profile=profile,
+            connection=connection,
+            reset_session=reset_session,
+            generation=generation or Generation.V0,
+            dry_run=False,
+            language="rel",
+        )
     else:
         raise Exception(f"Unknown platform: {platform}")
 
@@ -188,15 +224,26 @@ def Provider(
     resources = Resources(profile, config, connection, generation=generation)
     platform = resources.config.get("platform", "snowflake")
     if platform == "azure":
-        import v0.relationalai.clients.azure
-        resources = cast(v0.relationalai.clients.azure.Resources, resources)
-        return v0.relationalai.clients.azure.Provider(
+        from v0.relationalai.clients.resources.azure.azure import Resources as AzureResources, Provider as AzureProvider
+        resources = cast(AzureResources, resources)
+        return AzureProvider(
             resources=resources
         )
     elif platform == "snowflake":
-        import v0.relationalai.clients.snowflake
-        resources = cast(v0.relationalai.clients.snowflake.Resources, resources)
-        return v0.relationalai.clients.snowflake.Provider(
+        try:
+            from v0.relationalai.clients.resources.snowflake import Resources as SnowflakeResources, Provider as SnowflakeProvider
+        except ImportError as e:
+            # Provide a helpful error message for Snowflake notebook environments
+            from v0.relationalai.environments import runtime_env, SnowbookEnvironment
+            if isinstance(runtime_env, SnowbookEnvironment):
+                raise ImportError(
+                    "Failed to import relationalai.clients.resources.snowflake. "
+                    "This may indicate that the relationalai.zip package structure is incomplete. "
+                    "Please ensure the zip file includes the full clients/resources/snowflake directory structure."
+                ) from e
+            raise
+        resources = cast(SnowflakeResources, resources)
+        return SnowflakeProvider(
             resources=resources,
             generation=generation
         )

v0/relationalai/clients/__init__.py

@@ -1,5 +1,18 @@
 # Azure import is dropped because we need to to do runtime import of azure module to support our package running on Snowflake Notebook
-from . import config, snowflake, client, local
+from . import config
+
+# Lazy imports for client and local to avoid circular import issues
+# These modules import from relationalai which imports from clients, creating a cycle
+def __getattr__(name: str):
+    if name == 'client':
+        from . import client
+        return client
+    elif name == 'local':
+        from . import local
+        return local
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
 
 # note: user must do `import relationalai.clients.azure` to get `azure` submodule
-__all__ = ['snowflake', 'config', 'client', 'local']
+# note: snowflake module is now at relationalai.clients.resources.snowflake
+# note: 'client' and 'local' are lazy-loaded via __getattr__, so they're not in __all__
+__all__ = ['config']

v0/relationalai/clients/client.py

@@ -10,12 +10,12 @@ from dataclasses import dataclass
 from pandas import DataFrame
 import time
 
-from v0.relationalai.clients.hash_util import database_name_from_sproc_name
-from v0.relationalai.tools.constants import USE_GRAPH_INDEX, USE_PACKAGE_MANAGER
+from .hash_util import database_name_from_sproc_name
+from ..tools.constants import USE_GRAPH_INDEX, USE_PACKAGE_MANAGER
 
 
 from .types import AvailableModel, EngineState, Import, ImportSource, ImportsStatus, SourceMapEntry
-from ..clients.config import Config
+from .config import Config
 from ..compiler import Compiler
 from ..environments import runtime_env, NotebookRuntimeEnvironment
 from .. import dsl, debugging, metamodel as m
@@ -210,7 +210,7 @@ class ResourcesBase(ABC):
         return engine
 
     @abstractmethod
-    def auto_create_engine_async(self, name: str | None = None):
+    def auto_create_engine_async(self, name: str | None = None) -> str:
         pass
 
     _active_engine: EngineState|None = None

v0/relationalai/clients/local.py

@@ -13,10 +13,10 @@ from typing import Any, Dict, List, Iterable, Literal, Optional, Tuple, Union
 from requests.adapters import HTTPAdapter
 from urllib3.util.retry import Retry
 
-from v0.relationalai.clients.client import ResourcesBase, ProviderBase
-from v0.relationalai.clients.config import Config
-from v0.relationalai.clients.types import TransactionAsyncResponse
-from v0.relationalai.clients.util import get_pyrel_version
+from .client import ResourcesBase, ProviderBase
+from .config import Config
+from .types import TransactionAsyncResponse
+from .util import get_pyrel_version
 from ..errors import ResponseStatusException
 from .. import debugging
 
@@ -202,7 +202,7 @@ class LocalResources(ResourcesBase):
     def alter_engine_pool(self, size: str | None = None, mins: int | None = None, maxs: int | None = None):
         raise NotImplementedError("alter_engine_pool not supported in local mode")
 
-    def auto_create_engine_async(self, name: str | None = None):
+    def auto_create_engine_async(self, name: str | None = None) -> str:
         raise NotImplementedError("auto_create_engine_async not supported in local mode")
 
     #--------------------------------------------------

v0/relationalai/clients/resources/__init__.py

@@ -0,0 +1,8 @@
+"""
+Resources module for RelationalAI clients.
+"""
+
+# This package contains resource implementations for different platforms:
+# - snowflake: Snowflake-specific resources
+# - azure: Azure-specific resources (optional, may not be available in all environments)
+

v0/relationalai/clients/{azure.py → resources/azure/azure.py}

@@ -9,18 +9,18 @@ from urllib.error import HTTPError
 
 from pandas import DataFrame
 
-from v0.relationalai import debugging
-from v0.relationalai.clients.util import poll_with_specified_overhead
-
-from ..errors import EngineNotFoundException, RAIException, AzureUnsupportedQueryTimeoutException
-from ..rel_utils import assert_no_problems
-from ..loaders.loader import emit_delete_import, import_file, list_available_resources
-from .config import Config
-from .types import EngineState, ImportSource, ImportSourceFile, TransactionAsyncResponse
-from .client import Client, ExportParams, ProviderBase, ResourcesBase
-from .. import dsl, rel, metamodel as m
+from .... import debugging
+from ...util import poll_with_specified_overhead
+
+from ....errors import EngineNotFoundException, RAIException, AzureUnsupportedQueryTimeoutException
+from ....rel_utils import assert_no_problems
+from ....loaders.loader import emit_delete_import, import_file, list_available_resources
+from ...config import Config
+from ...types import EngineState, ImportSource, ImportSourceFile, TransactionAsyncResponse
+from ...client import Client, ExportParams, ProviderBase, ResourcesBase
+from .... import dsl, rel, metamodel as m
 from railib import api
-from . import result_helpers
+from ... import result_helpers
 
 #--------------------------------------------------
 # Constants
@@ -105,7 +105,7 @@ class Resources(ResourcesBase):
     def resume_engine_async(self, name:str):
         return api.resume_engine(self._api_ctx(), name)
 
-    def auto_create_engine_async(self, name: str | None = None):
+    def auto_create_engine_async(self, name: str | None = None) -> str:
         raise Exception("Azure doesn't support auto_create_engine_async")
 
     def alter_engine_pool(self, size: str | None = None, mins: int | None = None, maxs: int | None = None):

v0/relationalai/clients/resources/snowflake/__init__.py

@@ -0,0 +1,20 @@
+"""
+Snowflake resources module.
+"""
+# Import order matters - Resources must be imported first since other classes depend on it
+from .snowflake import Resources, Provider, Graph, SnowflakeClient, APP_NAME, PYREL_ROOT_DB, ExecContext, INTERNAL_ENGINE_SIZES, ENGINE_SIZES_AWS, ENGINE_SIZES_AZURE, PrimaryKey
+
+# These imports depend on Resources, so they come after
+from .cli_resources import CLIResources
+from .use_index_resources import UseIndexResources
+from .direct_access_resources import DirectAccessResources
+from .resources_factory import create_resources_instance
+
+__all__ = [
+    'Resources', 'DirectAccessResources', 'Provider', 'Graph', 'SnowflakeClient',
+    'APP_NAME', 'PYREL_ROOT_DB', 'CLIResources', 'UseIndexResources', 'ExecContext',
+    'INTERNAL_ENGINE_SIZES', 'ENGINE_SIZES_AWS', 'ENGINE_SIZES_AZURE', 'PrimaryKey',
+    'create_resources_instance',
+]
+
+

v0/relationalai/clients/resources/snowflake/cli_resources.py

@@ -0,0 +1,87 @@
+"""
+CLI Resources - Resources class for CLI operations.
+Re-raises exceptions without transformation and provides CLI utility methods.
+"""
+from __future__ import annotations
+import json
+from typing import Any
+
+from ....tools.constants import RAI_APP_NAME
+
+# Import Resources from snowflake - this creates a dependency but no circular import
+# since snowflake.py doesn't import from this file
+from .snowflake import Resources, ExecContext
+
+
+class CLIResources(Resources):
+    """
+    Resources class for CLI operations.
+    Re-raises exceptions without transformation and provides CLI utility methods.
+    """
+
+    def _handle_standard_exec_errors(self, e: Exception, ctx: ExecContext) -> Any | None:
+        """For CLI resources, re-raise exceptions without transformation."""
+        raise e
+
+    def list_warehouses(self):
+        """List all warehouses in the Snowflake account."""
+        results = self._exec("SHOW WAREHOUSES")
+        if not results:
+            return []
+        return [{"name": name}
+                for (name, *rest) in results]
+
+    def list_compute_pools(self):
+        """List all compute pools in the Snowflake account."""
+        results = self._exec("SHOW COMPUTE POOLS")
+        if not results:
+            return []
+        return [{"name": name, "status": status, "min_nodes": min_nodes, "max_nodes": max_nodes, "instance_family": instance_family}
+                for (name, status, min_nodes, max_nodes, instance_family, *rest) in results]
+
+    def list_roles(self):
+        """List all available roles in the Snowflake account."""
+        results = self._exec("SELECT CURRENT_AVAILABLE_ROLES()")
+        if not results:
+            return []
+        # the response is a single row with a single column containing
+        # a stringified JSON array of role names:
+        row = results[0]
+        if not row:
+            return []
+        return [{"name": name} for name in json.loads(row[0])]
+
+    def list_apps(self):
+        """List all applications in the Snowflake account."""
+        all_apps = self._exec(f"SHOW APPLICATIONS LIKE '{RAI_APP_NAME}'")
+        if not all_apps:
+            all_apps = self._exec("SHOW APPLICATIONS")
+            if not all_apps:
+                return []
+        return [{"name": name}
+                for (time, name, *rest) in all_apps]
+
+    def list_databases(self):
+        """List all databases in the Snowflake account."""
+        results = self._exec("SHOW DATABASES")
+        if not results:
+            return []
+        return [{"name": name}
+                for (time, name, *rest) in results]
+
+    def list_sf_schemas(self, database: str):
+        """List all schemas in a given database."""
+        results = self._exec(f"SHOW SCHEMAS IN {database}")
+        if not results:
+            return []
+        return [{"name": name}
+                for (time, name, *rest) in results]
+
+    def list_tables(self, database: str, schema: str):
+        """List all tables and views in a given schema."""
+        results = self._exec(f"SHOW OBJECTS IN {database}.{schema}")
+        items = []
+        if results:
+            for (time, name, db_name, schema_name, kind, *rest) in results:
+                items.append({"name": name, "kind": kind.lower()})
+        return items