@@ -149,7 +149,6 @@ def test_tracks_bedrock_metrics(client: LDClient):
149
149
tracker .track_bedrock_converse_metrics (bedrock_result )
150
150
151
151
calls = [
152
- call ('$ld:ai:generation' , context , {'variationKey' : 'variation-key' , 'configKey' : 'config-key' , 'version' : 3 }, 1 ),
153
152
call ('$ld:ai:generation:success' , context , {'variationKey' : 'variation-key' , 'configKey' : 'config-key' , 'version' : 3 }, 1 ),
154
153
call ('$ld:ai:duration:total' , context , {'variationKey' : 'variation-key' , 'configKey' : 'config-key' , 'version' : 3 }, 50 ),
155
154
call ('$ld:ai:tokens:total' , context , {'variationKey' : 'variation-key' , 'configKey' : 'config-key' , 'version' : 3 }, 330 ),
@@ -182,7 +181,6 @@ def test_tracks_bedrock_metrics_with_error(client: LDClient):
182
181
tracker .track_bedrock_converse_metrics (bedrock_result )
183
182
184
183
calls = [
185
- call ('$ld:ai:generation' , context , {'variationKey' : 'variation-key' , 'configKey' : 'config-key' , 'version' : 3 }, 1 ),
186
184
call ('$ld:ai:generation:error' , context , {'variationKey' : 'variation-key' , 'configKey' : 'config-key' , 'version' : 3 }, 1 ),
187
185
call ('$ld:ai:duration:total' , context , {'variationKey' : 'variation-key' , 'configKey' : 'config-key' , 'version' : 3 }, 50 ),
188
186
call ('$ld:ai:tokens:total' , context , {'variationKey' : 'variation-key' , 'configKey' : 'config-key' , 'version' : 3 }, 330 ),
@@ -216,7 +214,6 @@ def to_dict(self):
216
214
tracker .track_openai_metrics (lambda : Result ())
217
215
218
216
calls = [
219
- call ('$ld:ai:generation' , context , {'variationKey' : 'variation-key' , 'configKey' : 'config-key' , 'version' : 3 }, 1 ),
220
217
call ('$ld:ai:generation:success' , context , {'variationKey' : 'variation-key' , 'configKey' : 'config-key' , 'version' : 3 }, 1 ),
221
218
call ('$ld:ai:tokens:total' , context , {'variationKey' : 'variation-key' , 'configKey' : 'config-key' , 'version' : 3 }, 330 ),
222
219
call ('$ld:ai:tokens:input' , context , {'variationKey' : 'variation-key' , 'configKey' : 'config-key' , 'version' : 3 }, 220 ),
@@ -242,7 +239,6 @@ def raise_exception():
242
239
pass
243
240
244
241
calls = [
245
- call ('$ld:ai:generation' , context , {'variationKey' : 'variation-key' , 'configKey' : 'config-key' , 'version' : 3 }, 1 ),
246
242
call ('$ld:ai:generation:error' , context , {'variationKey' : 'variation-key' , 'configKey' : 'config-key' , 'version' : 3 }, 1 ),
247
243
]
248
244
@@ -279,7 +275,6 @@ def test_tracks_success(client: LDClient):
279
275
tracker .track_success ()
280
276
281
277
calls = [
282
- call ('$ld:ai:generation' , context , {'variationKey' : 'variation-key' , 'configKey' : 'config-key' , 'version' : 3 }, 1 ),
283
278
call ('$ld:ai:generation:success' , context , {'variationKey' : 'variation-key' , 'configKey' : 'config-key' , 'version' : 3 }, 1 ),
284
279
]
285
280
@@ -294,7 +289,6 @@ def test_tracks_error(client: LDClient):
294
289
tracker .track_error ()
295
290
296
291
calls = [
297
- call ('$ld:ai:generation' , context , {'variationKey' : 'variation-key' , 'configKey' : 'config-key' , 'version' : 3 }, 1 ),
298
292
call ('$ld:ai:generation:error' , context , {'variationKey' : 'variation-key' , 'configKey' : 'config-key' , 'version' : 3 }, 1 ),
299
293
]
300
294
@@ -310,9 +304,7 @@ def test_error_overwrites_success(client: LDClient):
310
304
tracker .track_error ()
311
305
312
306
calls = [
313
- call ('$ld:ai:generation' , context , {'variationKey' : 'variation-key' , 'configKey' : 'config-key' , 'version' : 3 }, 1 ),
314
307
call ('$ld:ai:generation:success' , context , {'variationKey' : 'variation-key' , 'configKey' : 'config-key' , 'version' : 3 }, 1 ),
315
- call ('$ld:ai:generation' , context , {'variationKey' : 'variation-key' , 'configKey' : 'config-key' , 'version' : 3 }, 1 ),
316
308
call ('$ld:ai:generation:error' , context , {'variationKey' : 'variation-key' , 'configKey' : 'config-key' , 'version' : 3 }, 1 ),
317
309
]
318
310
0 commit comments