diff --git a/ldai/testing/test_tracker.py b/ldai/testing/test_tracker.py index 66d8df9..30a20bd 100644 --- a/ldai/testing/test_tracker.py +++ b/ldai/testing/test_tracker.py @@ -176,12 +176,6 @@ def test_tracks_bedrock_metrics(client: LDClient): tracker.track_bedrock_converse_metrics(bedrock_result) calls = [ - call( - "$ld:ai:generation", - context, - {"variationKey": "variation-key", "configKey": "config-key", "version": 3}, - 1, - ), call( "$ld:ai:generation:success", context, @@ -239,12 +233,6 @@ def test_tracks_bedrock_metrics_with_error(client: LDClient): tracker.track_bedrock_converse_metrics(bedrock_result) calls = [ - call( - "$ld:ai:generation", - context, - {"variationKey": "variation-key", "configKey": "config-key", "version": 3}, - 1, - ), call( "$ld:ai:generation:error", context, @@ -303,12 +291,6 @@ def to_dict(self): tracker.track_openai_metrics(lambda: Result()) calls = [ - call( - "$ld:ai:generation", - context, - {"variationKey": "variation-key", "configKey": "config-key", "version": 3}, - 1, - ), call( "$ld:ai:generation:success", context, @@ -354,12 +336,6 @@ def raise_exception(): pass calls = [ - call( - "$ld:ai:generation", - context, - {"variationKey": "variation-key", "configKey": "config-key", "version": 3}, - 1, - ), call( "$ld:ai:generation:error", context, @@ -401,12 +377,6 @@ def test_tracks_success(client: LDClient): tracker.track_success() calls = [ - call( - "$ld:ai:generation", - context, - {"variationKey": "variation-key", "configKey": "config-key", "version": 3}, - 1, - ), call( "$ld:ai:generation:success", context, @@ -426,12 +396,6 @@ def test_tracks_error(client: LDClient): tracker.track_error() calls = [ - call( - "$ld:ai:generation", - context, - {"variationKey": "variation-key", "configKey": "config-key", "version": 3}, - 1, - ), call( "$ld:ai:generation:error", context, @@ -452,24 +416,12 @@ def test_error_overwrites_success(client: LDClient): tracker.track_error() calls = [ - call( - "$ld:ai:generation", - context, - {"variationKey": "variation-key", "configKey": "config-key", "version": 3}, - 1, - ), call( "$ld:ai:generation:success", context, {"variationKey": "variation-key", "configKey": "config-key", "version": 3}, 1, ), - call( - "$ld:ai:generation", - context, - {"variationKey": "variation-key", "configKey": "config-key", "version": 3}, - 1, - ), call( "$ld:ai:generation:error", context, diff --git a/ldai/tracker.py b/ldai/tracker.py index ef10bcb..9099833 100644 --- a/ldai/tracker.py +++ b/ldai/tracker.py @@ -176,9 +176,6 @@ def track_success(self) -> None: Track a successful AI generation. """ self._summary._success = True - self._ld_client.track( - "$ld:ai:generation", self._context, self.__get_track_data(), 1 - ) self._ld_client.track( "$ld:ai:generation:success", self._context, self.__get_track_data(), 1 ) @@ -188,9 +185,6 @@ def track_error(self) -> None: Track an unsuccessful AI generation attempt. """ self._summary._success = False - self._ld_client.track( - "$ld:ai:generation", self._context, self.__get_track_data(), 1 - ) self._ld_client.track( "$ld:ai:generation:error", self._context, self.__get_track_data(), 1 )