From d62839448285407c4dfc71607e9a04e5318d49e4 Mon Sep 17 00:00:00 2001 From: tyiuhc Date: Fri, 28 Feb 2025 14:35:05 -0800 Subject: [PATCH] fix: add IP property to user object --- src/amplitude_experiment/assignment/assignment_service.py | 2 +- src/amplitude_experiment/user.py | 3 +++ 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/src/amplitude_experiment/assignment/assignment_service.py b/src/amplitude_experiment/assignment/assignment_service.py index c33d014..0c5389c 100644 --- a/src/amplitude_experiment/assignment/assignment_service.py +++ b/src/amplitude_experiment/assignment/assignment_service.py @@ -17,7 +17,7 @@ def to_event(assignment: Assignment, send_evaluated_props: bool) -> BaseEvent: user_attributes = [ "country", "city", "region", "dma", "language", "platform", "version", "os", "device_manufacturer", - "device_brand", "device_model", "carrier" + "device_brand", "device_model", "carrier", "library", "ip_address" ] for attr in user_attributes: setattr(event, attr, getattr(assignment.user, attr, None)) diff --git a/src/amplitude_experiment/user.py b/src/amplitude_experiment/user.py index 8cbdcea..78ade75 100644 --- a/src/amplitude_experiment/user.py +++ b/src/amplitude_experiment/user.py @@ -26,6 +26,7 @@ def __init__( device_model: str = None, carrier: str = None, library: str = None, + ip_address: str = None, user_properties: Dict[str, Any] = None, groups: Dict[str, List[str]] = None, group_properties: Dict[str, Dict[str, Dict[str, Any]]] = None, @@ -50,6 +51,7 @@ def __init__( device_model (str): Predefined field, must be manually provided carrier (str): Predefined field, must be manually provided library (str): Predefined field, must be manually provided + ip_address (str): Predefined field, must be manually provided user_properties (dict): Custom user properties groups (dict): Groups associated with the user group_properties (dict): Properties for groups @@ -72,6 +74,7 @@ def __init__( self.device_model = device_model self.carrier = carrier self.library = library + self.ip_address = ip_address self.user_properties = user_properties self.groups = groups self.group_properties = group_properties