fix: Change attributes datasets and experiment on client to lazyloading and mark datasets as deprecated

packages/traceloop-sdk/tests/test_client.py

@@ -2,6 +2,8 @@
 from traceloop.sdk.client import Client
 from traceloop.sdk.client.http import HTTPClient
 from traceloop.sdk.annotation.user_feedback import UserFeedback
+from traceloop.sdk.datasets.datasets import Datasets
+from traceloop.sdk.experiment import Experiment
 
 
 def test_client_initialization():
@@ -46,3 +48,38 @@ def test_user_feedback_initialization():
     assert isinstance(client.user_feedback, UserFeedback)
     assert client.user_feedback._http == client._http
     assert client.user_feedback._app_name == client.app_name
+
+
+def test_client_lazy_loads_datasets():
+    """Test client.datasets is only initialized lazy."""
+    client = Client(api_key="test-key", app_name="test-app")
+    assert client._datasets is None  # Initial state is None
+
+    datasets = client.datasets
+
+    assert isinstance(datasets, Datasets)
+    assert client._datasets is not None  # Then it's loaded
+    assert client.datasets is datasets  # And always returns same instance
+
+
+def test_datasets_deprecation_warnings():
+    """Test client.datasets emits proper deprecation warnings."""
+    client = Client(api_key="test-key", app_name="test-app")
+    with pytest.deprecated_call():
+        client.datasets
+    with pytest.deprecated_call():
+        client.datasets = Datasets(client._http)
+    with pytest.deprecated_call():
+        del client.datasets
+
+def test_client_lazy_loads_experiment():
+    """Test cilent.experiment is only initialized lazy."""
+    client = Client(api_key="test-key", app_name="test-app")
+    assert client._experiment is None  # Initial state is None
+
+    experiment = client.experiment
+
+    assert isinstance(experiment, Experiment)
+    assert client._experiment is not None  # Then it's loaded
+    assert client.experiment is experiment  # And always returns same instance
+

packages/traceloop-sdk/traceloop/sdk/client/client.py

@@ -1,9 +1,8 @@
 import sys
 import os
+from deprecated import deprecated
 
 from traceloop.sdk.annotation.user_feedback import UserFeedback
-from traceloop.sdk.datasets.datasets import Datasets
-from traceloop.sdk.experiment.experiment import Experiment
 from traceloop.sdk.client.http import HTTPClient
 from traceloop.sdk.version import __version__
 from traceloop.sdk.associations.associations import Associations
@@ -25,8 +24,6 @@ class Client:
     api_endpoint: str
     api_key: str
     user_feedback: UserFeedback
-    datasets: Datasets
-    experiment: Experiment
     associations: Associations
     guardrails: Guardrails
     _http: HTTPClient
@@ -65,9 +62,45 @@ def __init__(
             timeout=httpx.Timeout(120.0),
         )
         self.user_feedback = UserFeedback(self._http, self.app_name)
-        self.datasets = Datasets(self._http)
-        experiment_slug = os.getenv("TRACELOOP_EXP_SLUG")
+        self._datasets = None
+        self._experiment = None
         # TODO: Fix type - Experiment constructor should accept Optional[str]
-        self.experiment = Experiment(self._http, self._async_http, experiment_slug)  # type: ignore[arg-type]
         self.associations = Associations()
         self.guardrails = Guardrails(self._async_http)
+
+    @property
+    def experiment(self):
+        # Lazy load only if accessed.
+        if self._experiment is None:
+            from traceloop.sdk.experiment.experiment import Experiment
+            experiment_slug = os.getenv("TRACELOOP_EXP_SLUG")
+            self._experiment = Experiment(self._http, self._async_http, experiment_slug)  # type: ignore[arg-type]
+        return self._experiment
+
+    @property
+    @deprecated(
+        reason="datasets as client attribute is deprecated. Use a dedicated instance "
+               "of Datasets from traceloop.sdk.datasets"
+    )
+    def datasets(self):
+        # Lazy load only if accessed.
+        if self._datasets is None:
+            from traceloop.sdk.datasets.datasets import Datasets
+            self._datasets = Datasets(self._http)
+        return self._datasets
+
+    @datasets.setter
+    @deprecated(
+        reason="datasets as client attribute is deprecated. Use a dedicated instance "
+               "of Datasets from traceloop.sdk.datasets"
+    )
+    def datasets(self, datasets: "Datasets"):
+        self._datasets = datasets
+
+    @datasets.deleter
+    @deprecated(
+        reason="datasets as client attribute is deprecated. Use a dedicated instance "
+               "of Datasets from traceloop.sdk.datasets"
+    )
+    def datasets(self):
+        self._datasets = None