relationalai 0.11.4__py3-none-any.whl → 0.12.0__py3-none-any.whl

relationalai/semantics/rel/compiler.py

@@ -6,13 +6,13 @@ from decimal import Decimal as PyDecimal
 from relationalai.semantics.metamodel import ir, compiler as c, builtins as bt, types, visitor, helpers, factory as f
 from relationalai.semantics.metamodel.typer import Checker, InferTypes
 from relationalai.semantics.metamodel.typer.typer import to_base_primitive, to_type, _NON_PARAMETRIC_PRIMITIVES
-from relationalai.semantics.metamodel.rewrite import (Flatten, ExtractKeys, FDConstraints, Splinter,
-                                                      ExtractNestedLogicals, DNFUnionSplitter, DischargeConstraints)
 from relationalai.semantics.metamodel.visitor import ReadWriteVisitor
 from relationalai.semantics.metamodel.util import OrderedSet, group_by, NameCache, ordered_set
 
 from relationalai.semantics.rel import rel, rel_utils as u, builtins as rel_bt
-from relationalai.semantics.rel.rewrite import CDC, QuantifyVars, ExtractCommon
+
+from ..metamodel.rewrite import (Flatten, ExtractNestedLogicals, DNFUnionSplitter, DischargeConstraints)
+from ..lqp.rewrite import CDC, ExtractCommon, ExtractKeys, FDConstraints, QuantifyVars, Splinter
 
 import math
 

relationalai/semantics/sql/compiler.py

@@ -11,7 +11,7 @@ from decimal import Decimal as PyDecimal
 
 import math
 
-from relationalai.semantics.metamodel.rewrite import (Flatten, ExtractNestedLogicals, FDConstraints, DNFUnionSplitter,
+from relationalai.semantics.metamodel.rewrite import (Flatten, ExtractNestedLogicals, DNFUnionSplitter,
                                                       DischargeConstraints)
 from relationalai.semantics.metamodel.visitor import ReadWriteVisitor
 from relationalai.util.graph import topological_sort
@@ -28,7 +28,6 @@ from relationalai.semantics.sql import sql, rewrite
 class Compiler(c.Compiler):
     def __init__(self, skip_denormalization:bool=False):
         rewrites = [
-            FDConstraints(),
             DischargeConstraints(),
             Checker(),
             ExtractNestedLogicals(),  # before InferTypes to avoid extracting casts
@@ -2495,4 +2494,4 @@ class DerivedRelationsVisitor(v.Visitor):
 
     def visit_relation(self, node: ir.Relation, parent: Optional[ir.Node]):
         if self._is_derived and from_cdc_annotation in node.annotations:
-            self._is_derived = False
+            self._is_derived = False

relationalai/semantics/sql/executor/duck_db.py

@@ -8,7 +8,7 @@ from scipy.special import erfinv as special_erfinv
 
 from relationalai.semantics.sql import Compiler
 from relationalai.semantics.sql.executor.result_helpers import format_duckdb_columns
-from relationalai.semantics.metamodel import ir, executor as e
+from relationalai.semantics.metamodel import ir, executor as e, factory as f
 
 class DuckDBExecutor(e.Executor):
 
@@ -29,8 +29,12 @@ class DuckDBExecutor(e.Executor):
         connection.create_function("erfinv", self.erfinv)
 
         try:
-            sql, _ = self.compiler.compile(model, {"is_duck_db": True})
-            arrow_table = connection.query(sql).fetch_arrow_table()
+            model_sql, _ = self.compiler.compile(model, {"is_duck_db": True})
+            query_options = {"is_duck_db": True, "query_compilation": True}
+            query_sql, _ = self.compiler.compile(f.compute_model(f.logical([task])), query_options)
+
+            full_sql = model_sql + "\n" + query_sql
+            arrow_table = connection.query(full_sql).fetch_arrow_table()
             return format_duckdb_columns(arrow_table.to_pandas(), arrow_table.schema)
         finally:
             connection.close()
@@ -45,4 +49,4 @@ class DuckDBExecutor(e.Executor):
 
     @staticmethod
     def acot(x: float) -> float:
-        return math.atan(1 / x) if x != 0 else math.copysign(math.pi / 2, x)
+        return math.atan(1 / x) if x != 0 else math.copysign(math.pi / 2, x)

relationalai/tools/cli.py

@@ -16,8 +16,10 @@ from InquirerPy.base.control import Choice
 from relationalai.clients.util import IdentityParser
 from .cli_controls import divider, Spinner
 from . import cli_controls as controls
-from relationalai.clients import azure
-from typing import Sequence, cast, Any, List
+from typing import Sequence, cast, Any, List, TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from relationalai.clients import azure
 from relationalai.errors import RAIException
 from relationalai.loaders.types import LoadType, UnsupportedTypeError
 from relationalai.loaders.csv import CSVLoader
@@ -1150,9 +1152,14 @@ def import_source_flow(provider: ResourcesBase) -> Sequence[ImportSource]:
 
     if isinstance(provider, snowflake.Resources):
         return snowflake_import_source_flow(provider)
-    elif isinstance(provider, azure.Resources):
-        return azure_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):
+                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]:

{relationalai-0.11.4.dist-info → relationalai-0.12.0.dist-info}/METADATA

@@ -1,12 +1,12 @@
 Metadata-Version: 2.4
 Name: relationalai
-Version: 0.11.4
+Version: 0.12.0
 Summary: RelationalAI Library and CLI
 Author-email: RelationalAI <support@relational.ai>
 License-File: LICENSE
-Requires-Python: >=3.9
+Requires-Python: >=3.10
 Requires-Dist: aiohttp
-Requires-Dist: click<=8.1.8
+Requires-Dist: click==8.2.1
 Requires-Dist: colorama
 Requires-Dist: cryptography
 Requires-Dist: gravis
@@ -21,7 +21,6 @@ Requires-Dist: packaging
 Requires-Dist: pandas
 Requires-Dist: pyarrow
 Requires-Dist: pyjwt
-Requires-Dist: rai-sdk>=0.7.5
 Requires-Dist: requests
 Requires-Dist: rich
 Requires-Dist: snowflake-connector-python[secure-local-storage]
@@ -64,6 +63,8 @@ Requires-Dist: parameterized; extra == 'ea'
 Requires-Dist: pysmt; extra == 'ea'
 Requires-Dist: simple-ddl-parser; extra == 'ea'
 Requires-Dist: xmltodict; extra == 'ea'
+Provides-Extra: legacy
+Requires-Dist: rai-sdk>=0.7.5; extra == 'legacy'
 Description-Content-Type: text/markdown
 
 # The RelationalAI Python Library

{relationalai-0.11.4.dist-info → relationalai-0.12.0.dist-info}/RECORD

@@ -4,7 +4,7 @@ relationalai/debugging.py,sha256=dfT0gEDpL56GflVmtUrYVbYcVcZmQL26Vf_F8cI1QKY,118
 relationalai/dependencies.py,sha256=tL113efcISkJUiDXYHmRdU_usdD7gmee-VRHA7N4EFA,16574
 relationalai/docutils.py,sha256=1gVv9mk0ytdMB2W7_NvslJefmSQtTOg8LHTCDcGCjyE,1554
 relationalai/dsl.py,sha256=pvwH12zLvyumn8jRpkb57KkBPLrLkp5vnDdN5kBO388,66099
-relationalai/errors.py,sha256=4v-J91SfsgowPaU7xssJC8t1VFs1TlJGuIN378itoLE,94019
+relationalai/errors.py,sha256=ERyVg5SJd3O689QSnp-rtbBCLUB3Bbcp5VgkRX1HZzg,94948
 relationalai/metagen.py,sha256=o10PNvR_myr_61DC8g6lkB093bFo9qXGUkZKgKyfXiE,26821
 relationalai/metamodel.py,sha256=P1hliwHd1nYxbXON4LZeaYZD6T6pZm97HgmFBFrWyCk,32886
 relationalai/rel.py,sha256=ePmAXx4NxOdsPcHNHyGH3Jkp_cB3QzfKu5p_EQSHPh0,38293
@@ -22,12 +22,13 @@ relationalai/clients/__init__.py,sha256=0c_oqvpVN7pccc6q8fw11zSOVK-utlRTmd1ECjgM
 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/config.py,sha256=aerlAhg6DWICa4AUwetgsYKXUWiyEyymnHTIWAN2sAE,24087
+relationalai/clients/config.py,sha256=hERaKjc3l4kd-kf0l-NUOHrWunCn8gmFWpuE0j3ScJg,24457
+relationalai/clients/direct_access_client.py,sha256=uDML4nFTctvvya4GFWoef2oirlWHiRugn5qBk9fxH9k,5961
 relationalai/clients/export_procedure.py.jinja,sha256=nhvVcs5hQyWExFDuROQbi1VyYzOCa_ZIRPR2KzZwDtI,10582
 relationalai/clients/hash_util.py,sha256=pZVR1FX3q4G_19p_r6wpIR2tIM8_WUlfAR7AVZJjIYM,1495
 relationalai/clients/profile_polling.py,sha256=pUH7WKH4nYDD0SlQtg3wsWdj0K7qt6nZqUw8jTthCBs,2565
 relationalai/clients/result_helpers.py,sha256=wDSD02Ngx6W-YQqBIGKnpXD4Ju3pA1e9Nz6ORRI6SRI,17808
-relationalai/clients/snowflake.py,sha256=lfMwypFnsw18ZG4ixOCuEqZwf0lvvthk4pPtj97pZNE,155717
+relationalai/clients/snowflake.py,sha256=zO9mVSCrLbjWwu0PIp3cZimDF9nxvU4mgp5AImeSA0c,151653
 relationalai/clients/types.py,sha256=z33RlnXFml-aVAtZ1pJsbixrGnlPfhkAjZXM_ySZGuY,2776
 relationalai/clients/use_index_poller.py,sha256=rQaHmlNLYDQ4NJh3i6CVAS1XxUjazwzTv-OiacrldGI,44662
 relationalai/clients/util.py,sha256=3iLY5DgUkQcRAPXrvP4xdIYZtc3trrryy5jfX2QAFEA,12014
@@ -178,7 +179,7 @@ relationalai/early_access/metamodel/dependency/__init__.py,sha256=bkrnbq8aRMwCPD
 relationalai/early_access/metamodel/factory/__init__.py,sha256=_boK2qJ4v_JVFn4OelJx3GTA3X48Yc5yCxq_DxG6fUs,946
 relationalai/early_access/metamodel/helpers/__init__.py,sha256=bkEbkPFhvcVhnwRAAJxEUMWlI5r3iYuyTIDINfY1frs,320
 relationalai/early_access/metamodel/ir/__init__.py,sha256=1oJzbPkQrRf5Cng51u1rZUa_sxIzSaSE4m6pvnHTUfE,571
-relationalai/early_access/metamodel/rewrite/__init__.py,sha256=57_5cY6vlCdwtGGWS9K-JrhOkwEsXfwYAVJrxsHc1Fc,371
+relationalai/early_access/metamodel/rewrite/__init__.py,sha256=TH1FykUuhb34Gd06klxhGvAqfpIB4ZhuHBSXOMeOhsI,342
 relationalai/early_access/metamodel/typer/__init__.py,sha256=va6-xIBBh3B28O5JEUeXsu38FcZPTDTQuym4BDrshmw,123
 relationalai/early_access/metamodel/typer/typer/__init__.py,sha256=gUfw_Zkk_wq1iycTp7yrOxpZJs5GctvsLmatlmKlcFM,432
 relationalai/early_access/metamodel/types/__init__.py,sha256=M7KDNQyO7uA_p_U_IRT3jYaOzhVVDm5MBjgQdxZqWsY,741
@@ -202,7 +203,7 @@ relationalai/early_access/paths/utilities/iterators/__init__.py,sha256=HRO3elGRr
 relationalai/early_access/rel/__init__.py,sha256=1Uf_eHjZcUZvaFoQ1nmMIcth4icRLxM3WsvCf99_Fj0,290
 relationalai/early_access/rel/executor/__init__.py,sha256=viY7VTCVe3GVOpHlnVhrlXYDIAeIA01BcxtLtiaRaqI,285
 relationalai/early_access/rel/rel_utils/__init__.py,sha256=FS5qahNNxiqev6DG0f66MaCnPuneWEysFAOZTWHwvL8,298
-relationalai/early_access/rel/rewrite/__init__.py,sha256=pe8wuwhqACG427DWJm7ch10QChpkgZEIMemF3d6fqvA,147
+relationalai/early_access/rel/rewrite/__init__.py,sha256=gA0aPFs1BYV8Pc_FJejjjAEIB3lA8JpCQGTWDz1mhd4,147
 relationalai/early_access/solvers/__init__.py,sha256=Ix9cRdW_glqtGkk3KyH5sJIR3X4yauVDSH_JlPrMYBQ,447
 relationalai/early_access/sql/__init__.py,sha256=B2L23yTW-kYMk52zPvCo_D2XD22vZFdJm1iBnuliEec,267
 relationalai/early_access/sql/executor/__init__.py,sha256=yyt8dxigsUvfRVOPV2rW2QrbHiUK750Dil1IIY3Y3B8,132
@@ -254,26 +255,33 @@ relationalai/semantics/devtools/benchmark_lqp.py,sha256=btySKVtUUNBuPY-l6DrkbRNl
 relationalai/semantics/devtools/compilation_manager.py,sha256=XBqG_nYWtK3s_J6MeCmkLqy3qm7l778zDAhH7DEO91g,11169
 relationalai/semantics/devtools/extract_lqp.py,sha256=gxI3EvPUTPAkwgnkCKAkEm2vA6QkLfoM8AXXiVz0c34,3696
 relationalai/semantics/internal/__init__.py,sha256=JXrpFaL-fdZrvKpWTEn1UoLXITOoTGnAYwmgeiglhSk,774
-relationalai/semantics/internal/annotations.py,sha256=NuOcaca6H8osWqJAwe9nxezu10X2sPV7ti7iROAxiBM,267
+relationalai/semantics/internal/annotations.py,sha256=P_nRpBm4wLmE_8L0VO3TDORL1p2flXaPOVDC0GG7KsQ,306
 relationalai/semantics/internal/internal.py,sha256=5Wf0XitUqWo0j30aRTtkQ6KKsBDmD3mklLitpPSzB90,148333
 relationalai/semantics/internal/snowflake.py,sha256=DR6rgbPawen2ZOFegRuPkXt8aQUfciiCYzDI74u1Iwc,13036
 relationalai/semantics/lqp/__init__.py,sha256=XgcQZxK-zz_LqPDVtwREhsIvjTuUIt4BZhIedCeMY-s,48
-relationalai/semantics/lqp/builtins.py,sha256=N1hJwyvwgFvBbt2X7-4U-Pq1LVOYV-zAUeSqS39OZcQ,515
+relationalai/semantics/lqp/builtins.py,sha256=bRmQ6fdceWU-4xf4l-W-YiuyDxJTPey1s6O4xlyW6iM,540
 relationalai/semantics/lqp/compiler.py,sha256=Nury1gPw_-Oi_mqT1-rhr13L4UmyIP2BGuotbuklQKA,949
 relationalai/semantics/lqp/constructors.py,sha256=8U4eUL8-m1wYRQnws_YWC1coGquTugVH5YC0Zek6VT8,2309
 relationalai/semantics/lqp/executor.py,sha256=At0IdBlNlOM0E533V0urRgeIkfbv-pPWuQ_sfRkKqPQ,18926
 relationalai/semantics/lqp/intrinsics.py,sha256=Pb1mLIme7Q-5Y-CVacUOEvapfhKs076bgtRNi3f0ayY,833
 relationalai/semantics/lqp/ir.py,sha256=DUw0ltul0AS9CRjntNlmllWTwXpxMyYg4iJ9t7NFYMA,1791
 relationalai/semantics/lqp/model2lqp.py,sha256=L63A2ECLiLTbUoZB7wTB2dP0ahUAA-8WhxiddVO2znQ,31647
-relationalai/semantics/lqp/passes.py,sha256=TBsYmWYh4qZtKzlSiZcJJDt3-Bwl_wRv0c9L8P2H1fE,27683
+relationalai/semantics/lqp/passes.py,sha256=nLppoHvIQkGP6VuG56OAZ1oOrYhEqpR_0w91gfJ7t_s,27540
 relationalai/semantics/lqp/pragmas.py,sha256=FzzldrJEAZ1AIcEw6D-FfaVg3CoahRYgPCFo7xHfg1g,375
 relationalai/semantics/lqp/primitives.py,sha256=Gbh6cohoAArhqEJTN_TgIRc7wmtdxXt231NRW0beEj0,10898
 relationalai/semantics/lqp/result_helpers.py,sha256=oYpLoTBnzsiyOVIWA2rLMHlgs7P7BoEkqthQ2aMosnk,10123
 relationalai/semantics/lqp/types.py,sha256=3TZ61ybwNV8lDyUMujZIWNFz3Fgn4uifsJb8ExfoMDg,4508
 relationalai/semantics/lqp/utils.py,sha256=x8dcfVoyqzCznNFtI4qSNSL-vWcdxzC-QrnFJlAARXs,6121
 relationalai/semantics/lqp/validators.py,sha256=YO_ciSgEVNILWUbkxIagKpIxI4oqV0fRSTO2Ok0rPJk,1526
+relationalai/semantics/lqp/rewrite/__init__.py,sha256=NQMkrZMUrLksScGXs0EiZTT-In5VltvswfeB4kCkeZM,337
+relationalai/semantics/lqp/rewrite/cdc.py,sha256=I6DeMOZScx-3UAVoSCMn9cuOgLzwdvJVKNwsgFa6R_k,10390
+relationalai/semantics/lqp/rewrite/extract_common.py,sha256=TY4wYhMy6bbKC9YZqrBeQgaZ41Sf4BCI3Zy26yavgAM,17301
+relationalai/semantics/lqp/rewrite/extract_keys.py,sha256=dSr5SVkYmrhiR0XPY5eRAnWD66dcZYgXdilXcERv634,18682
+relationalai/semantics/lqp/rewrite/fd_constraints.py,sha256=a4jetchlOfm2rb3QD0Nil0bvermpBx1up7p1uNgcJHg,3503
+relationalai/semantics/lqp/rewrite/quantify_vars.py,sha256=wYMEXzCW_D_Y_1rSLvuAAqw9KN1oIOn_vIMxELzRVb4,11568
+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=0NIy94ay2-Yy0tvL750kRLoavyjnwARdmn07hn6LDFs,35310
+relationalai/semantics/metamodel/builtins.py,sha256=EOOi3AjeF5uyYVjie0WhpVTVatEK3zWyBvggStH1aGo,35897
 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=YU-nTm9Hq6ufkwoA5D1iPciGw7WW6Hk4Cyq0EQbkl2s,32845
@@ -284,23 +292,18 @@ relationalai/semantics/metamodel/ir.py,sha256=4Xl3oc92Q7_s33axtrZUXr-GL8VGJsKc7y
 relationalai/semantics/metamodel/types.py,sha256=m3NolF3B_fBSnkQKA9o1feEawaW7RhR-t6W9OSxSFfc,11492
 relationalai/semantics/metamodel/util.py,sha256=cmSmeww34JVMqcFudwVAY820IPM2ETSELJylwJ3GRJk,16654
 relationalai/semantics/metamodel/visitor.py,sha256=DFY0DACLhxlZ0e4p0vWqbK6ZJr_GWEvH66CU_HVuoTk,35527
-relationalai/semantics/metamodel/rewrite/__init__.py,sha256=iKg1YsWjWhY9UpmQt0nHJjSto_kMD9bN1vLdfblyP_c,577
+relationalai/semantics/metamodel/rewrite/__init__.py,sha256=aPgZuRGpULwPVWtENUEHmGBSNAEsxz1i0t257IOCO-E,285
 relationalai/semantics/metamodel/rewrite/discharge_constraints.py,sha256=0v613BqCLlo4sgWuZjcLSxxakp3d34mYWbG4ldhzGno,1949
 relationalai/semantics/metamodel/rewrite/dnf_union_splitter.py,sha256=dlyD868Pg424BLowY4A0gOmSziSy1U-dYGXZEE3SW8E,7956
-relationalai/semantics/metamodel/rewrite/extract_keys.py,sha256=dSr5SVkYmrhiR0XPY5eRAnWD66dcZYgXdilXcERv634,18682
 relationalai/semantics/metamodel/rewrite/extract_nested_logicals.py,sha256=mzZLsu4pYWzRUScAgmWQ58WYVW30Sst7xMJteoSxm-4,3629
-relationalai/semantics/metamodel/rewrite/fd_constraints.py,sha256=a4jetchlOfm2rb3QD0Nil0bvermpBx1up7p1uNgcJHg,3503
 relationalai/semantics/metamodel/rewrite/flatten.py,sha256=B-LH9OApak7Cst7bcVWzKZcgI0fuTDTBprXJ0AL7q6w,28448
-relationalai/semantics/metamodel/rewrite/gc_nodes.py,sha256=5_hXEk-Gg2E2bmE56wATcaZ7nIHfp5BsqkVEyCU-r3I,2090
-relationalai/semantics/metamodel/rewrite/list_types.py,sha256=RBmi0iWMaADjbqptV0W7P9ncmZFqCSDMKmWxKuS9Mh0,4454
-relationalai/semantics/metamodel/rewrite/splinter.py,sha256=oeDjP_F2PVLVexAKFn8w7CLtO9oy-R-tS2IOmzw_Ujk,3199
 relationalai/semantics/metamodel/typer/__init__.py,sha256=E3ydmhWRdm-cAqWsNR24_Qd3NcwiHx8ElO2tzNysAXc,143
 relationalai/semantics/metamodel/typer/checker.py,sha256=frY0gilDO6skbDiYFiIpDUOWyt9s9jAJsRBs848DcG0,19184
 relationalai/semantics/metamodel/typer/typer.py,sha256=jBo0CwY6G0qKBzqoAmQ09dPtcPDHvCiutolnwbV4gus,62294
 relationalai/semantics/reasoners/__init__.py,sha256=Tu4U26rrkBIzAk3a4tXRJaeD5mAtK9Z7JXh2c6VJ-J4,249
 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=pdjfEUlPtX3rbVObqG9RuhUckaFSkUym-cmZmmxrPpc,262065
+relationalai/semantics/reasoners/graph/core.py,sha256=RjzjbQcRptK0F9WfRO9a4XBpoxa7AT_opJjnpzJb-uU,270731
 relationalai/semantics/reasoners/graph/design/beyond_demand_transform.md,sha256=Givh0W6B6Hlow6TpmK-8adpEYd8b3O_WmdgMOQIyKs0,55749
 relationalai/semantics/reasoners/graph/paths/README.md,sha256=ydm6CzMN_vOOgq7a6_hBCiyGi3D6g5gxAbf4OXbQbDE,3433
 relationalai/semantics/reasoners/graph/paths/__init__.py,sha256=oCoDJVg4moGe0tNMNeRn7XQqa1AIEZGlxAiaD5bBahg,211
@@ -371,21 +374,17 @@ relationalai/semantics/reasoners/optimization/common.py,sha256=TVa9n9n97BwfMyYOT
 relationalai/semantics/reasoners/optimization/solvers_dev.py,sha256=BjFcEfmqmdxM6jnd34pt1AlkrdwHd_HapqpB2oIvNP4,24142
 relationalai/semantics/reasoners/optimization/solvers_pb.py,sha256=ryNARpyph3uvr1dhWb7pNU3uuxuL1FLZZiTqFcr3Xf8,22148
 relationalai/semantics/rel/__init__.py,sha256=pMlVTC_TbQ45mP1LpzwFBBgPxpKc0H3uJDvvDXEWzvs,55
-relationalai/semantics/rel/builtins.py,sha256=bLMX4pJ62J62qn-TX5WQWl3Nhhy6q9Zz827pY3ihaBc,1286
-relationalai/semantics/rel/compiler.py,sha256=9d0-75t08vkdS5n3wvyCN68qlxVywIZ8JD7sph4f5LM,42298
+relationalai/semantics/rel/builtins.py,sha256=qu4yZvLovn4Vn2x44D4XugqGD6Qo5xLxj_RKA34cpF4,1527
+relationalai/semantics/rel/compiler.py,sha256=hiLIaZzhVU5VMtU6rdo3tH2pmKMEtXvh6AHh94CvRAg,42203
 relationalai/semantics/rel/executor.py,sha256=BcQUJht5G2so2Af9GszLBTK90LqFUmv3gshRGFeiG-s,14943
 relationalai/semantics/rel/rel.py,sha256=9I_V6dQ83QRaLzq04Tt-KjBWhmNxNO3tFzeornBK4zc,15738
 relationalai/semantics/rel/rel_utils.py,sha256=F14Ch8mn45J8SmM7HZnIHUNqDnb3WQLnkEGLme04iBk,9386
-relationalai/semantics/rel/rewrite/__init__.py,sha256=Ke85L9DMO_d6qQbjb17k_LoMqar8-vOHKmxHGynIh3s,170
-relationalai/semantics/rel/rewrite/cdc.py,sha256=I6DeMOZScx-3UAVoSCMn9cuOgLzwdvJVKNwsgFa6R_k,10390
-relationalai/semantics/rel/rewrite/extract_common.py,sha256=TY4wYhMy6bbKC9YZqrBeQgaZ41Sf4BCI3Zy26yavgAM,17301
-relationalai/semantics/rel/rewrite/quantify_vars.py,sha256=wYMEXzCW_D_Y_1rSLvuAAqw9KN1oIOn_vIMxELzRVb4,11568
 relationalai/semantics/snowflake/__init__.py,sha256=BW_zvPQBWGTAtY6cluG6tDDG-QmU_jRb-F7PeCpDhIU,134
 relationalai/semantics/sql/__init__.py,sha256=59uuKmQOFUzwlYwc7anSuwhQ45bJTYVmd9nOJXr9mQs,80
-relationalai/semantics/sql/compiler.py,sha256=4NeOp4YnMZz25IfLpPeEWL1iO46y8h2D0HW05d4RY4A,128816
+relationalai/semantics/sql/compiler.py,sha256=BcvEIjcD9SVkTtTvDIw-yjeC3dE5dv01P3Ohd7HCNxo,128773
 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=pS_93nV5vKPnjkK62fc0YHHFdRyBEHlXe_Y4H2zB6lQ,1654
+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=GMfeutObRWOUnH5UVflHZgFxn-soUtjTky37daBtBi4,5825
 relationalai/semantics/sql/rewrite/__init__.py,sha256=AT1WR0rqQHQ7E06NLoVym0zrZpBVPqK85uRFNJUTDp4,254
@@ -419,7 +418,7 @@ 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=k7r1VG9YiY3o8ZqgEhWI2naYyMzKyJ2gR-KAL-vfzIs,76195
+relationalai/tools/cli.py,sha256=MophMqK-_sw0LYNuNH0L-wCmEmAN49bvf_Fvb7JZKoI,76424
 relationalai/tools/cli_controls.py,sha256=6k2nu6km9IYGDwG7pSZGBXrEN64Ehw0tamKgcPotZoM,71822
 relationalai/tools/cli_helpers.py,sha256=r-gJnWnb3d6yIn_ccyp3qx_Vpqh-FINCFMyPBVCA-Jg,14041
 relationalai/tools/constants.py,sha256=mDdWelpf-u9tb0tspFgq8V0fCxLSzbS0Hl9WGJMJa_4,4417
@@ -453,8 +452,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.11.4.dist-info/METADATA,sha256=B3VzqLxv4_NTOySacfpH2EgQmwYc5yFOhojNgkKXego,2519
-relationalai-0.11.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-relationalai-0.11.4.dist-info/entry_points.txt,sha256=fo_oLFJih3PUgYuHXsk7RnCjBm9cqRNR--ab6DgI6-0,88
-relationalai-0.11.4.dist-info/licenses/LICENSE,sha256=pPyTVXFYhirkEW9VsnHIgUjT0Vg8_xsE6olrF5SIgpc,11343
-relationalai-0.11.4.dist-info/RECORD,,
+relationalai-0.12.0.dist-info/METADATA,sha256=2VPdNMd0QUh5hrmUIpaU7YijfeRP68L3cjUaUwAftJI,2562
+relationalai-0.12.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+relationalai-0.12.0.dist-info/entry_points.txt,sha256=fo_oLFJih3PUgYuHXsk7RnCjBm9cqRNR--ab6DgI6-0,88
+relationalai-0.12.0.dist-info/licenses/LICENSE,sha256=pPyTVXFYhirkEW9VsnHIgUjT0Vg8_xsE6olrF5SIgpc,11343
+relationalai-0.12.0.dist-info/RECORD,,

relationalai/semantics/metamodel/rewrite/gc_nodes.py

@@ -1,58 +0,0 @@
-"""
-Garbage collection pass that removes unused types and relations from the model.
-"""
-from dataclasses import dataclass, field
-
-from relationalai.semantics.metamodel import ir, visitor, compiler
-from relationalai.semantics.metamodel.util import FrozenOrderedSet
-
-@dataclass
-class GarbageCollectNodes(compiler.Pass):
-    """
-    A pass that removes unused types and relations from the model.
-    """
-
-    # TODO: Since sometimes models don't initially have the correct set of relations,
-    # we sometimes actually end up /adding/ relations to the model here.
-
-    # Flags to determine which nodes to gargbage collect
-    # By default, we only garbage collect types and relations
-    types: bool = field(default=True)
-    relations: bool = field(default=True)
-
-    @staticmethod
-    def used_relations(model, engines) -> FrozenOrderedSet[ir.Relation]:
-        return visitor.collect_by_type(ir.Relation, *engines, model.root, *model.annotations).frozen()
-
-    @staticmethod
-    def used_types(model, engines, relations) -> FrozenOrderedSet[ir.Type]:
-        return visitor.collect_by_type(ir.Type, *engines, *relations, model.root, *model.annotations).frozen()
-
-    def rewrite(self, model: ir.Model, options:dict={}) -> ir.Model:
-        gc_types = self.types
-        gc_relations = self.relations
-
-        # No flags set, nothing to GC
-        if not gc_relations and not gc_types:
-            return model
-
-        engines = model.engines
-        relations = model.relations
-        types = model.types
-
-        # Note that engines and relations are mutually dependent. If we were to also GC engines,
-        # then we would need to iterate until we reach a fixed point.
-        if gc_relations:
-            relations = self.used_relations(model, engines)
-
-        # Types are independent of engines and relations, so we can just collect them last
-        if gc_types:
-            types = self.used_types(model, engines, relations)
-
-        return ir.Model(
-            engines,
-            relations,
-            types,
-            model.root,
-            model.annotations,
-        )

relationalai/semantics/metamodel/rewrite/list_types.py

@@ -1,109 +0,0 @@
-"""
-Rewrite list types to tuple or scalar types depending on how they are used.
-"""
-from dataclasses import dataclass, field
-from typing import List, cast
-
-from relationalai.semantics.metamodel import ir, visitor, compiler, builtins as bt
-from relationalai.semantics.metamodel.util import ordered_set
-
-@dataclass
-class RewriteListTypes(compiler.Pass):
-    """
-    Rewrite list types to tuple or scalar types depending on how they are used.
-    """
-    def rewrite(self, model: ir.Model, options:dict={}) -> ir.Model:
-        v = RewriteListTypesVisitor()
-        result = v.walk(model)
-        return result
-
-@dataclass
-class RewriteListTypesVisitor(visitor.Rewriter):
-    """
-    A pass that fixes the types of nodes that use ListTypes.
-    """
-    new_relations: List[ir.Relation] = field(default_factory=list, init=False)
-    new_types: List[ir.Type] = field(default_factory=list, init=False)
-
-    def handle_model(self, model: ir.Model, parent: None):
-        result = super().handle_model(model, parent)
-        relations_without_list_type = ordered_set(*[
-            r for r in result.relations if not any(isinstance(f.type, ir.ListType) for f in r.fields)
-        ]).frozen()
-        return model.reconstruct(
-            result.engines,
-            relations_without_list_type | self.new_relations,
-            result.types | self.new_types,
-            result.root,
-            result.annotations,
-        )
-
-    def handle_lookup(self, node: ir.Lookup, parent: ir.Node):
-        if any(isinstance(f.type, ir.ListType) for f in node.relation.fields):
-            return self._rewrite_non_aggr_relation(node, node.relation)
-        else:
-            return node
-
-    def handle_aggregate(self, node: ir.Aggregate, parent: ir.Node):
-        new_aggr = self._rewrite_aggr_relation(node, node.aggregation)
-        return ir.Aggregate(
-            node.engine,
-            new_aggr,
-            node.projection,
-            node.group,
-            node.args,
-        )
-
-    def _rewrite_aggr_relation(self, node: ir.Aggregate, relation: ir.Relation) -> ir.Relation:
-        assert len(relation.fields) >= 1
-        changed = False
-
-        if changed:
-            overloads = ordered_set(*[
-                self._rewrite_aggr_relation(node, overload)
-                for overload in relation.overloads
-            ]).frozen()
-
-            new_relation = ir.Relation(
-                relation.name,
-                tuple(relation.fields),
-                relation.requires,
-                relation.annotations,
-                overloads,
-            )
-            self.new_relations.append(new_relation)
-            return new_relation
-
-        return relation
-
-    def _rewrite_non_aggr_relation(self, node: ir.Lookup, relation: ir.Relation):
-        # Handle the simple varargs case: one list field and all the rest are scalar.
-        # Currently this pattern is only used for `rel_primitive_hash_tuple`.
-        # and `rel_primitive_solverlib_fo_appl`.
-        list_field_indexes = [i for i in range(len(node.relation.fields)) if isinstance(node.relation.fields[i].type, ir.ListType)]
-        if len(list_field_indexes) != 1:
-            return node
-
-        # There exactly one list field, rewrite it to a tuple type.
-        i = list_field_indexes[0]
-        assert isinstance(node.relation.fields[i].type, ir.ListType)
-        scalar_field_count = len(node.relation.fields) - 1
-        tuple_len = len(node.args) - scalar_field_count
-        assert tuple_len >= 0, f"List field {i} has {tuple_len} elements, but there are only {scalar_field_count} scalar fields."
-
-        # Flatten the list field into separate scalar fields.
-        field = node.relation.fields[i]
-        ft = cast(ir.ListType, field.type)
-        new_fields = (
-            list(node.relation.fields[0:i]) +
-            [ir.Field(f"{field.name}@{j}", ft.element_type, field.input) for j in range(tuple_len)] +
-            list(node.relation.fields[i+1:])
-        )
-
-        # Create new relation, adding external annotation to prevent renaming.
-        # NOTE(coey): maybe this is a misuse of this annotation and there could be a nicer way to do it with overloads.
-        annos = ordered_set(*node.relation.annotations, bt.external_annotation).frozen()
-        new_relation = ir.Relation(node.relation.name, tuple(new_fields), node.relation.requires, annos)
-        self.new_relations.append(new_relation)
-
-        return ir.Lookup(node.engine, new_relation, node.args)