Skip to content

Commit ca20f09

Browse files
authored
fix: Remove deprecated track generation event (#301)
1 parent 9a84146 commit ca20f09

File tree

3 files changed

+7
-36
lines changed

3 files changed

+7
-36
lines changed

ldai/README.md

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -39,12 +39,12 @@ aiClient, err := ldai.NewClient(sdkClient)
3939

4040
Fetch a model configuration for a specific LaunchDarkly context:
4141
```go
42-
// The default value 'ldai.Disabled()' be returned if LaunchDarkly is unavailable or the config
42+
// The default value 'ldai.Disabled()' be returned if LaunchDarkly is unavailable or the config
4343
// cannot be fetched. To customize the default value, use ldai.NewConfig().
4444
config, tracker := aiClient.Config("your-model-key", ldcontext.New("user-key"), ldai.Disabled(), nil)
4545

4646
// Access the methods on config, and optionally use the returned tracker to generate analytic events
47-
// related to usage of the model config.
47+
// related to usage of the model config.
4848
```
4949
Learn more
5050
-----------
@@ -56,7 +56,7 @@ You can also head straight to the [complete reference guide for this SDK](https:
5656
Contributing
5757
------------
5858

59-
We encourage pull requests and other contributions from the community. Check out our [contributing guidelines](CONTRIBUTING.md) for instructions on how to contribute to this library.
59+
We encourage pull requests and other contributions from the community. Check out our [contributing guidelines](../CONTRIBUTING.md) for instructions on how to contribute to this library.
6060

6161
About LaunchDarkly
6262
-----------

ldai/tracker.go

Lines changed: 2 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,6 @@ const (
1414
duration = "$ld:ai:duration:total"
1515
feedbackPositive = "$ld:ai:feedback:user:positive"
1616
feedbackNegative = "$ld:ai:feedback:user:negative"
17-
generation = "$ld:ai:generation"
1817
generationSuccess = "$ld:ai:generation:success"
1918
generationError = "$ld:ai:generation:error"
2019
//nolint:gosec
@@ -219,24 +218,14 @@ func (t *Tracker) TrackFeedback(feedback Feedback) error {
219218
func (t *Tracker) TrackSuccess() error {
220219
t.success = ldcommon.Some(true)
221220

222-
err := t.events.TrackMetric(generation, t.context, 1, t.trackData)
223-
if err := t.events.TrackMetric(generationSuccess, t.context, 1, t.trackData); err != nil {
224-
return err
225-
}
226-
227-
return err
221+
return t.events.TrackMetric(generationSuccess, t.context, 1, t.trackData)
228222
}
229223

230224
// TrackError tracks an unsuccessful model evaluation.
231225
func (t *Tracker) TrackError() error {
232226
t.success = ldcommon.Some(false)
233227

234-
err := t.events.TrackMetric(generation, t.context, 1, t.trackData)
235-
if err := t.events.TrackMetric(generationError, t.context, 1, t.trackData); err != nil {
236-
return err
237-
}
238-
239-
return err
228+
return t.events.TrackMetric(generationError, t.context, 1, t.trackData)
240229
}
241230

242231
// TrackTimeToFirstToken tracks the time to the first token of the streamed response.

ldai/tracker_test.go

Lines changed: 2 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -64,12 +64,6 @@ func TestTracker_TrackSuccess(t *testing.T) {
6464
assert.NoError(t, tracker.TrackSuccess())
6565

6666
expectedEvents := []trackEvent{
67-
{
68-
name: "$ld:ai:generation",
69-
context: ldcontext.New("key"),
70-
metricValue: 1.0,
71-
data: makeTrackData("key", "variationKey", 1, config),
72-
},
7367
{
7468
name: "$ld:ai:generation:success",
7569
context: ldcontext.New("key"),
@@ -88,12 +82,6 @@ func TestTracker_TrackError(t *testing.T) {
8882
assert.NoError(t, tracker.TrackError())
8983

9084
expectedEvents := []trackEvent{
91-
{
92-
name: "$ld:ai:generation",
93-
context: ldcontext.New("key"),
94-
metricValue: 1.0,
95-
data: makeTrackData("key", "variationKey", 2, config),
96-
},
9785
{
9886
name: "$ld:ai:generation:error",
9987
context: ldcontext.New("key"),
@@ -128,12 +116,6 @@ func TestTracker_TrackRequest(t *testing.T) {
128116
assert.Equal(t, expectedResponse, r)
129117

130118
expectedEvents := []trackEvent{
131-
{
132-
name: "$ld:ai:generation",
133-
context: ldcontext.New("key"),
134-
metricValue: 1,
135-
data: makeTrackData("key", "variationKey", 3, config),
136-
},
137119
{
138120
name: "$ld:ai:generation:success",
139121
context: ldcontext.New("key"),
@@ -214,8 +196,8 @@ func TestTracker_LatencyMeasuredIfNotProvided(t *testing.T) {
214196
assert.NoError(t, err)
215197
assert.Equal(t, expectedResponse, r)
216198

217-
require.Equal(t, 4, len(events.events))
218-
gotEvent := events.events[2]
199+
require.Equal(t, 3, len(events.events))
200+
gotEvent := events.events[1]
219201
assert.Equal(t, "$ld:ai:duration:total", gotEvent.name)
220202
assert.Equal(t, 42.0, gotEvent.metricValue)
221203
}

0 commit comments

Comments
 (0)