relationalai 0.9.17__tar.gz → 0.11.0__tar.gz

{relationalai-0.9.17 → relationalai-0.11.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: relationalai
-Version: 0.9.17
+Version: 0.11.0
 Summary: RelationalAI Library and CLI
 Author-email: RelationalAI <support@relational.ai>
 License-File: LICENSE
@@ -11,7 +11,7 @@ Requires-Dist: colorama
 Requires-Dist: cryptography
 Requires-Dist: gravis
 Requires-Dist: inquirerpy
-Requires-Dist: lqp==0.1.15
+Requires-Dist: lqp==0.1.16
 Requires-Dist: nicegui
 Requires-Dist: numpy<2
 Requires-Dist: opentelemetry-api

{relationalai-0.9.17 → relationalai-0.11.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = 'relationalai'
-version = '0.9.17'
+version = '0.11.0'
 description = 'RelationalAI Library and CLI'
 readme="docs/pypi/README.md"
 authors = [
@@ -8,7 +8,7 @@ authors = [
 ]
 requires-python = ">= 3.9"
 dependencies = [
-    "lqp==0.1.15",
+    "lqp==0.1.16",
     "snowflake-connector-python[secure-local-storage]",
     "snowflake-snowpark-python>=1.38.0",
     "rai-sdk>=0.7.5",

{relationalai-0.9.17 → relationalai-0.11.0}/src/relationalai/__init__.py

@@ -12,9 +12,10 @@ from . import rel
 from .loaders import csv
 from . import analysis
 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, SnowbookEnvironment
+from .environments import runtime_env, SessionEnvironment
 from relationalai.tools.constants import USE_DIRECT_ACCESS, Generation
 
 import __main__
@@ -135,24 +136,7 @@ def Model(
                 format=format,
             )
 
-            # allow to enable the OpenTelemetry handler only for internal snowflake accounts
-            # Skip OTEL handling entirely for warehouse environments, 
-            # skipping is possible because it is not set up in the warehouse environment
-            if isinstance(runtime_env, SnowbookEnvironment) and runtime_env.runner == "warehouse":
-                pass
-            elif (
-                enable_otel_handler
-                and platform == "snowflake"
-                and config.is_internal_account()
-            ):
-                from .util.otel_handler import enable_otel_export
-                enable_otel_export(
-                    cast(snowflake.Resources, model._client.resources),
-                    config.get('rai_app_name', 'RELATIONALAI')
-                )
-            else:
-                from .util.otel_handler import disable_otel_handling
-                disable_otel_handling()
+            configure_otel(enable_otel_handler, config, model._client.resources)
 
         else:
             raise Exception(f"Unknown platform: {platform}")

{relationalai-0.9.17 → relationalai-0.11.0}/src/relationalai/clients/snowflake.py

@@ -702,22 +702,14 @@ Otherwise, remove it from your '{profile}' configuration profile.
         self._create_engine(name, size, auto_suspend_mins, True)
 
     def delete_engine(self, name:str, force:bool = False, headers: Dict | None = None):
-        request_headers = {}
-        prop_headers = debugging.gen_current_propagation_headers()
-        if headers:
-            request_headers.update(headers)
-        request_headers.update(prop_headers)
+        request_headers = debugging.add_current_propagation_headers(headers)
         self._exec(f"call {APP_NAME}.api.delete_engine('{name}', {force},{request_headers});")
 
     def suspend_engine(self, name:str):
         self._exec(f"call {APP_NAME}.api.suspend_engine('{name}');")
 
     def resume_engine(self, name:str, headers: Dict | None = None) -> Dict:
-        request_headers = {}
-        prop_headers = debugging.gen_current_propagation_headers()
-        if headers:
-            request_headers.update(headers)
-        request_headers.update(prop_headers)
+        request_headers = debugging.add_current_propagation_headers(headers)
         self._exec(f"call {APP_NAME}.api.resume_engine('{name}',{request_headers});")
         # returning empty dict to match the expected return type
         return {}
@@ -1559,12 +1551,7 @@ Otherwise, remove it from your '{profile}' configuration profile.
     ):
         if inputs is None:
             inputs = {}
-        prop_headers = debugging.generate_propagation_headers(debugging.get_current_span())
-        request_headers = {}
-        if headers:
-            request_headers.update(headers)
-        request_headers.update(prop_headers)
-
+        request_headers = debugging.add_current_propagation_headers(headers)
         query_attrs_dict = json.loads(request_headers.get("X-Query-Attributes", "{}"))
 
         with debugging.span("transaction", **query_attrs_dict) as txn_span:
@@ -2980,7 +2967,6 @@ class DirectAccessClient:
         return url, ep.method
 
     def _prepare_headers(self, headers: Dict[str, str] | None) -> Dict[str, str]:
-
         request_headers = {}
         if headers:
             request_headers.update(headers)
@@ -2994,7 +2980,8 @@ class DirectAccessClient:
 
         request_headers["user-agent"] = get_pyrel_version(self.generation)
         request_headers["pyrel_program_id"] = debugging.get_program_span_id() or ""
-        return request_headers
+
+        return debugging.add_current_propagation_headers(request_headers)
 
 class DirectAccessResources(Resources):
     """