Skip to content

Commit 583883f

Browse files
[AIC-1636] fix: clean up unused import and use dynamic version in test
Co-Authored-By: jbailey@launchdarkly.com <accounts@sidewaysgravity.com>
1 parent b61875d commit 583883f

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

packages/sdk/server-ai/tests/test_model_config.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -307,7 +307,7 @@ def test_model_initial_config_enabled(ldai_client: LDAIClient):
307307

308308

309309
def test_config_method_tracking(ldai_client: LDAIClient):
310-
from unittest.mock import Mock, call
310+
from unittest.mock import Mock
311311

312312
mock_client = Mock()
313313
mock_client.variation.return_value = {
@@ -332,8 +332,9 @@ def test_config_method_tracking(ldai_client: LDAIClient):
332332

333333

334334
def test_sdk_info_tracked_on_init():
335-
from unittest.mock import Mock, call
335+
from unittest.mock import Mock
336336

337+
from ldai import __version__ as ai_sdk_version
337338
from ldai.client import _INIT_TRACK_CONTEXT
338339

339340
mock_client = Mock()
@@ -345,7 +346,7 @@ def test_sdk_info_tracked_on_init():
345346
_INIT_TRACK_CONTEXT,
346347
{
347348
'aiSdkName': 'launchdarkly-server-sdk-ai',
348-
'aiSdkVersion': '0.14.0',
349+
'aiSdkVersion': ai_sdk_version,
349350
'aiSdkLanguage': 'python',
350351
},
351352
1,

0 commit comments

Comments
 (0)