diff --git a/packages/sdk/server-ai/src/ldai/client.py b/packages/sdk/server-ai/src/ldai/client.py index c5fda5a..de9fbde 100644 --- a/packages/sdk/server-ai/src/ldai/client.py +++ b/packages/sdk/server-ai/src/ldai/client.py @@ -18,7 +18,7 @@ from ldai.sdk_info import AI_SDK_LANGUAGE, AI_SDK_NAME, AI_SDK_VERSION from ldai.tracker import LDAIConfigTracker -_TRACK_SDK_INFO = '$ld:ai:sdk-info' +_TRACK_SDK_INFO = '$ld:ai:sdk:info' _TRACK_USAGE_COMPLETION_CONFIG = '$ld:ai:usage:completion-config' _TRACK_USAGE_CREATE_CHAT = '$ld:ai:usage:create-chat' _TRACK_USAGE_JUDGE_CONFIG = '$ld:ai:usage:judge-config' @@ -26,7 +26,7 @@ _TRACK_USAGE_AGENT_CONFIG = '$ld:ai:usage:agent-config' _TRACK_USAGE_AGENT_CONFIGS = '$ld:ai:usage:agent-configs' -_INIT_TRACK_CONTEXT = Context.builder('ld-internal-tracking').anonymous(True).build() +_INIT_TRACK_CONTEXT = Context.builder('ld-internal-tracking').kind('ld_ai').anonymous(True).build() class LDAIClient: diff --git a/packages/sdk/server-ai/tests/test_model_config.py b/packages/sdk/server-ai/tests/test_model_config.py index d0ae24d..e0fcd4e 100644 --- a/packages/sdk/server-ai/tests/test_model_config.py +++ b/packages/sdk/server-ai/tests/test_model_config.py @@ -342,7 +342,7 @@ def test_sdk_info_tracked_on_init(): client = LDAIClient(mock_client) mock_client.track.assert_called_once_with( - '$ld:ai:sdk-info', + '$ld:ai:sdk:info', _INIT_TRACK_CONTEXT, { 'aiSdkName': AI_SDK_NAME,