From 371c086d4564cb7bffb3eb039930e43f88244af8 Mon Sep 17 00:00:00 2001
From: Teo <teocns@gmail.com>
Date: Mon, 30 Dec 2024 20:47:39 +0100
Subject: [PATCH] =?UTF-8?q?Revert=20"Replace=20usage=20of=20functools.cach?=
 =?UTF-8?q?ed=5Fproperty=20with=20DIY=20class=20for=20backwar=E2=80=A6=20(?=
 =?UTF-8?q?#613)"?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

This reverts commit 82f9bb989e726c88e84265fc2399093cdcdfecf5.
---
 agentops/client.py    |  2 +-
 agentops/helpers.py   | 19 ------------
 tests/test_helpers.py | 67 -------------------------------------------
 3 files changed, 1 insertion(+), 87 deletions(-)
 delete mode 100644 tests/test_helpers.py

diff --git a/agentops/client.py b/agentops/client.py
index 8e0b6de52..fb3e17937 100644
--- a/agentops/client.py
+++ b/agentops/client.py
@@ -14,6 +14,7 @@
 import threading
 import traceback
 from decimal import Decimal
+from functools import cached_property
 from typing import List, Optional, Tuple, Union
 from uuid import UUID, uuid4
 
@@ -27,7 +28,6 @@
 from .meta_client import MetaClient
 from .session import Session, active_sessions
 from .singleton import conditional_singleton
-from .helpers import cached_property
 
 
 @conditional_singleton
diff --git a/agentops/helpers.py b/agentops/helpers.py
index 3e8df36e1..ca0c4f0e3 100644
--- a/agentops/helpers.py
+++ b/agentops/helpers.py
@@ -174,22 +174,3 @@ def wrapper(self, *args, **kwargs):
         return func(self, *args, **kwargs)
 
     return wrapper
-
-
-class cached_property:
-    """
-    Decorator that converts a method with a single self argument into a
-    property cached on the instance.
-    See: https://github.com/AgentOps-AI/agentops/issues/612
-    """
-
-    def __init__(self, func):
-        self.func = func
-        self.__doc__ = func.__doc__
-
-    def __get__(self, instance, cls=None):
-        if instance is None:
-            return self
-        value = self.func(instance)
-        setattr(instance, self.func.__name__, value)
-        return value
diff --git a/tests/test_helpers.py b/tests/test_helpers.py
deleted file mode 100644
index 401bb5665..000000000
--- a/tests/test_helpers.py
+++ /dev/null
@@ -1,67 +0,0 @@
-import pytest
-from agentops.helpers import cached_property
-
-
-def test_cached_property():
-    class TestClass:
-        def __init__(self):
-            self.compute_count = 0
-
-        @cached_property
-        def expensive_computation(self):
-            self.compute_count += 1
-            return 42
-
-    # Create instance
-    obj = TestClass()
-
-    # First access should compute the value
-    assert obj.expensive_computation == 42
-    assert obj.compute_count == 1
-
-    # Second access should use cached value
-    assert obj.expensive_computation == 42
-    assert obj.compute_count == 1  # Count shouldn't increase
-
-    # Third access should still use cached value
-    assert obj.expensive_computation == 42
-    assert obj.compute_count == 1  # Count shouldn't increase
-
-
-def test_cached_property_different_instances():
-    class TestClass:
-        def __init__(self):
-            self.compute_count = 0
-
-        @cached_property
-        def expensive_computation(self):
-            self.compute_count += 1
-            return id(self)  # Return unique id for each instance
-
-    # Create two different instances
-    obj1 = TestClass()
-    obj2 = TestClass()
-
-    # Each instance should compute its own value
-    val1 = obj1.expensive_computation
-    val2 = obj2.expensive_computation
-
-    assert val1 != val2  # Values should be different
-    assert obj1.compute_count == 1
-    assert obj2.compute_count == 1
-
-    # Accessing again should use cached values
-    assert obj1.expensive_computation == val1
-    assert obj2.expensive_computation == val2
-    assert obj1.compute_count == 1  # Counts shouldn't increase
-    assert obj2.compute_count == 1
-
-
-def test_cached_property_class_access():
-    class TestClass:
-        @cached_property
-        def expensive_computation(self):
-            return 42
-
-    # Accessing via class should return the descriptor
-    assert isinstance(TestClass.expensive_computation, cached_property)