relationalai 0.9.18__tar.gz → 0.11.1__tar.gz

{relationalai-0.9.18 → relationalai-0.11.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: relationalai
-Version: 0.9.18
+Version: 0.11.1
 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.18 → relationalai-0.11.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = 'relationalai'
-version = '0.9.18'
+version = '0.11.1'
 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.18 → relationalai-0.11.1}/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}")