From 111eac51d7fa223340d3258cfc05af44f0872dc4 Mon Sep 17 00:00:00 2001
From: Matthew Keeler <mkeeler@launchdarkly.com>
Date: Tue, 10 Dec 2024 09:58:32 -0500
Subject: [PATCH] refactor!: Change version_key_to variation_key

---
 ldai/client.py                    |  2 +-
 ldai/testing/test_model_config.py | 10 +++++-----
 ldai/tracker.py                   | 10 +++++-----
 3 files changed, 11 insertions(+), 11 deletions(-)

diff --git a/ldai/client.py b/ldai/client.py
index 0cd2d19..9db71f3 100644
--- a/ldai/client.py
+++ b/ldai/client.py
@@ -185,7 +185,7 @@ def config(
 
         tracker = LDAIConfigTracker(
             self.client,
-            variation.get('_ldMeta', {}).get('versionKey', ''),
+            variation.get('_ldMeta', {}).get('variationKey', ''),
             key,
             context,
         )
diff --git a/ldai/testing/test_model_config.py b/ldai/testing/test_model_config.py
index 53baa1d..48fd00c 100644
--- a/ldai/testing/test_model_config.py
+++ b/ldai/testing/test_model_config.py
@@ -15,7 +15,7 @@ def td() -> TestData:
                 'model': {'name': 'fakeModel', 'parameters': {'temperature': 0.5, 'maxTokens': 4096}, 'custom': {'extra-attribute': 'value'}},
                 'provider': {'name': 'fakeProvider'},
                 'messages': [{'role': 'system', 'content': 'Hello, {{name}}!'}],
-                '_ldMeta': {'enabled': True, 'versionKey': 'abcd'},
+                '_ldMeta': {'enabled': True, 'variationKey': 'abcd'},
             },
             "green",
         )
@@ -31,7 +31,7 @@ def td() -> TestData:
                     {'role': 'system', 'content': 'Hello, {{name}}!'},
                     {'role': 'user', 'content': 'The day is, {{day}}!'},
                 ],
-                '_ldMeta': {'enabled': True, 'versionKey': 'abcd'},
+                '_ldMeta': {'enabled': True, 'variationKey': 'abcd'},
             },
             "green",
         )
@@ -44,7 +44,7 @@ def td() -> TestData:
             {
                 'model': {'name': 'fakeModel', 'parameters': {'extra-attribute': 'I can be anything I set my mind/type to'}},
                 'messages': [{'role': 'system', 'content': 'Hello, {{ldctx.name}}! Is your last name {{ldctx.last}}?'}],
-                '_ldMeta': {'enabled': True, 'versionKey': 'abcd'},
+                '_ldMeta': {'enabled': True, 'variationKey': 'abcd'},
             }
         )
         .variation_for_all(0)
@@ -56,7 +56,7 @@ def td() -> TestData:
             {
                 'model': {'name': 'fakeModel', 'parameters': {'extra-attribute': 'I can be anything I set my mind/type to'}},
                 'messages': [{'role': 'system', 'content': 'Hello, {{ldctx.user.name}}! Do you work for {{ldctx.org.shortname}}?'}],
-                '_ldMeta': {'enabled': True, 'versionKey': 'abcd'},
+                '_ldMeta': {'enabled': True, 'variationKey': 'abcd'},
             }
         )
         .variation_for_all(0)
@@ -68,7 +68,7 @@ def td() -> TestData:
             {
                 'model': {'name': 'fakeModel', 'parameters': {'temperature': 0.1}},
                 'messages': [{'role': 'system', 'content': 'Hello, {{name}}!'}],
-                '_ldMeta': {'enabled': False, 'versionKey': 'abcd'},
+                '_ldMeta': {'enabled': False, 'variationKey': 'abcd'},
             }
         )
         .variation_for_all(0)
diff --git a/ldai/tracker.py b/ldai/tracker.py
index d179674..bf9c7fa 100644
--- a/ldai/tracker.py
+++ b/ldai/tracker.py
@@ -137,18 +137,18 @@ class LDAIConfigTracker:
     """
 
     def __init__(
-        self, ld_client: LDClient, version_key: str, config_key: str, context: Context
+        self, ld_client: LDClient, variation_key: str, config_key: str, context: Context
     ):
         """
         Initialize an AI configuration tracker.
 
         :param ld_client: LaunchDarkly client instance.
-        :param version_key: Version key for tracking.
+        :param variation_key: Variation key for tracking.
         :param config_key: Configuration key for tracking.
         :param context: Context for evaluation.
         """
         self.ld_client = ld_client
-        self.version_key = version_key
+        self.variation_key = variation_key
         self.config_key = config_key
         self.context = context
 
@@ -156,10 +156,10 @@ def __get_track_data(self):
         """
         Get tracking data for events.
 
-        :return: Dictionary containing version and config keys.
+        :return: Dictionary containing variation and config keys.
         """
         return {
-            'versionKey': self.version_key,
+            'variationKey': self.variation_key,
             'configKey': self.config_key,
         }