Merge branch 'main' into kfreeman/REL-8497/ai-tracker-model-provider

Author: knfreemLD

ldai/testing/test_tracker.py

@@ -180,12 +180,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, "modelName": "fakeModel", "providerName": "fakeProvider"},
-            1,
-        ),
         call(
             "$ld:ai:generation:success",
             context,
@@ -243,12 +237,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, "modelName": "fakeModel", "providerName": "fakeProvider"},
-            1,
-        ),
         call(
             "$ld:ai:generation:error",
             context,
@@ -307,12 +295,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, "modelName": "fakeModel", "providerName": "fakeProvider"},
-            1,
-        ),
         call(
             "$ld:ai:generation:success",
             context,
@@ -358,12 +340,6 @@ def raise_exception():
         pass
 
     calls = [
-        call(
-            "$ld:ai:generation",
-            context,
-            {"variationKey": "variation-key", "configKey": "config-key", "version": 3, "modelName": "fakeModel", "providerName": "fakeProvider"},
-            1,
-        ),
         call(
             "$ld:ai:generation:error",
             context,
@@ -405,12 +381,6 @@ def test_tracks_success(client: LDClient):
     tracker.track_success()
 
     calls = [
-        call(
-            "$ld:ai:generation",
-            context,
-            {"variationKey": "variation-key", "configKey": "config-key", "version": 3, "modelName": "fakeModel", "providerName": "fakeProvider"},
-            1,
-        ),
         call(
             "$ld:ai:generation:success",
             context,
@@ -430,12 +400,6 @@ def test_tracks_error(client: LDClient):
     tracker.track_error()
 
     calls = [
-        call(
-            "$ld:ai:generation",
-            context,
-            {"variationKey": "variation-key", "configKey": "config-key", "version": 3, "modelName": "fakeModel", "providerName": "fakeProvider"},
-            1,
-        ),
         call(
             "$ld:ai:generation:error",
             context,
@@ -456,22 +420,10 @@ def test_error_overwrites_success(client: LDClient):
     tracker.track_error()
 
     calls = [
-        call(
-            "$ld:ai:generation",
-            context,
-            {"variationKey": "variation-key", "configKey": "config-key", "version": 3, "modelName": "fakeModel", "providerName": "fakeProvider"},
-            1,
-        ),
         call(
             "$ld:ai:generation:success",
             context,
-            {"variationKey": "variation-key", "configKey": "config-key", "version": 3, "modelName": "fakeModel", "providerName": "fakeProvider"},
-            1,
-        ),
-        call(
-            "$ld:ai:generation",
-            context,
-            {"variationKey": "variation-key", "configKey": "config-key", "version": 3, "modelName": "fakeModel", "providerName": "fakeProvider"},
+            {"variationKey": "variation-key", "configKey": "config-key", "version": 3},
             1,
         ),
         call(

ldai/tracker.py

@@ -184,9 +184,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
         )
@@ -196,9 +193,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
         )