Rename BanditData fields from model_ -> bandit_model_ prefix

eppo_client/client.py

@@ -339,7 +339,7 @@ def get_bandit_action_detail(
             subject_key,
             subject_attributes,
             action_contexts,
-            bandit_data.model_data,
+            bandit_data.bandit_model_data,
         )
 
         # log bandit action
@@ -350,7 +350,7 @@ def get_bandit_action_detail(
             "action": evaluation.action_key if evaluation else None,
             "actionProbability": evaluation.action_weight if evaluation else None,
             "optimalityGap": evaluation.optimality_gap if evaluation else None,
-            "modelVersion": bandit_data.model_version if evaluation else None,
+            "modelVersion": bandit_data.bandit_model_version if evaluation else None,
             "timestamp": datetime.datetime.utcnow().isoformat(),
             "subjectNumericAttributes": (
                 subject_attributes.numeric_attributes

eppo_client/models.py

@@ -1,5 +1,6 @@
 from datetime import datetime
 from enum import Enum
+from pydantic import Field
 from typing import Dict, List, Optional
 
 from eppo_client.base_model import SdkBaseModel
@@ -96,7 +97,7 @@ class BanditModelData(SdkBaseModel):
 
 class BanditData(SdkBaseModel):
     bandit_key: str
-    model_name: str
+    bandit_model_name: str = Field(alias="modelName")
+    bandit_model_version: str = Field(alias="modelVersion")
+    bandit_model_data: BanditModelData = Field(alias="modelData")
     updated_at: datetime
-    model_version: str
-    model_data: BanditModelData

eppo_client/version.py

@@ -1 +1 @@
-__version__ = "3.3.0"
+__version__ = "3.4.0"

test/client_bandit_test.py

@@ -175,9 +175,9 @@ def test_bandit_generic_test_cases(test_case):
             flag,
             subject["subjectKey"],
             ContextAttributes(
-                numeric_attributes=subject["subjectAttributes"]["numeric_attributes"],
+                numeric_attributes=subject["subjectAttributes"]["numericAttributes"],
                 categorical_attributes=subject["subjectAttributes"][
-                    "categorical_attributes"
+                    "categoricalAttributes"
                 ],
             ),
             {