Skip to content

Commit 158b110

Browse files
[AIC-1636] refactor: extract SDK info into dedicated sdk_info module
Co-Authored-By: jbailey@launchdarkly.com <accounts@sidewaysgravity.com>
1 parent 583883f commit 158b110

3 files changed

Lines changed: 15 additions & 8 deletions

File tree

packages/sdk/server-ai/src/ldai/client.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,6 @@
44
from ldclient import Context
55
from ldclient.client import LDClient
66

7-
from ldai import __version__ as ai_sdk_version
87
from ldai import log
98
from ldai.agent_graph import AgentGraphDefinition
109
from ldai.chat import Chat
@@ -16,6 +15,7 @@
1615
JudgeConfiguration, LDMessage, ModelConfig,
1716
ProviderConfig)
1817
from ldai.providers.ai_provider_factory import AIProviderFactory
18+
from ldai.sdk_info import AI_SDK_LANGUAGE, AI_SDK_NAME, AI_SDK_VERSION
1919
from ldai.tracker import LDAIConfigTracker
2020

2121
_TRACK_SDK_INFO = '$ld:ai:sdk-info'
@@ -38,9 +38,9 @@ def __init__(self, client: LDClient):
3838
_TRACK_SDK_INFO,
3939
_INIT_TRACK_CONTEXT,
4040
{
41-
'aiSdkName': 'launchdarkly-server-sdk-ai',
42-
'aiSdkVersion': ai_sdk_version,
43-
'aiSdkLanguage': 'python',
41+
'aiSdkName': AI_SDK_NAME,
42+
'aiSdkVersion': AI_SDK_VERSION,
43+
'aiSdkLanguage': AI_SDK_LANGUAGE,
4444
},
4545
1,
4646
)
Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,7 @@
1+
from importlib.metadata import metadata
2+
3+
_meta = metadata('launchdarkly-server-sdk-ai')
4+
5+
AI_SDK_NAME: str = _meta['Name']
6+
AI_SDK_VERSION: str = _meta['Version']
7+
AI_SDK_LANGUAGE: str = 'python'

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

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -334,8 +334,8 @@ def test_config_method_tracking(ldai_client: LDAIClient):
334334
def test_sdk_info_tracked_on_init():
335335
from unittest.mock import Mock
336336

337-
from ldai import __version__ as ai_sdk_version
338337
from ldai.client import _INIT_TRACK_CONTEXT
338+
from ldai.sdk_info import AI_SDK_LANGUAGE, AI_SDK_NAME, AI_SDK_VERSION
339339

340340
mock_client = Mock()
341341

@@ -345,9 +345,9 @@ def test_sdk_info_tracked_on_init():
345345
'$ld:ai:sdk-info',
346346
_INIT_TRACK_CONTEXT,
347347
{
348-
'aiSdkName': 'launchdarkly-server-sdk-ai',
349-
'aiSdkVersion': ai_sdk_version,
350-
'aiSdkLanguage': 'python',
348+
'aiSdkName': AI_SDK_NAME,
349+
'aiSdkVersion': AI_SDK_VERSION,
350+
'aiSdkLanguage': AI_SDK_LANGUAGE,
351351
},
352352
1,
353353
)

0 commit comments

Comments
 (0)