User identifier resource attr

quotientai/_constants.py

@@ -1,3 +1,5 @@
 TRACER_NAME = "quotient.sdk.python"
 DEFAULT_TRACING_ENDPOINT = "https://otel.quotientai.co/v1/traces"
+# DEFAULT_TRACING_ENDPOINT = "http://localhost:4318/v1/traces"
+
 

quotientai/async_client.py

@@ -30,6 +30,7 @@ def __init__(self, api_key: str):
         self.token = None
         self.token_expiry = 0
         self.token_api_key = None
+        self.user_id = None
         self._token_path = (
             token_dir
             / ".quotient"

quotientai/client.py

@@ -32,6 +32,7 @@ def __init__(self, api_key: str):
         self.token = None
         self.token_expiry = 0
         self.token_api_key = None
+        self.user_id = None
         self._token_path = (
             token_dir
             / ".quotient"

quotientai/resources/auth.py

@@ -8,6 +8,11 @@ def authenticate(self):
         A call to GET /auth/profile to initially authenticate the user.
         """
         response = self._client._get("/auth/profile")
+
+        # Set the user_id if successful
+        if response and isinstance(response, dict) and 'user_id' in response:
+            self._client.user_id = response['user_id']
+
         return response
 
 class AsyncAuthResource:
@@ -25,6 +30,11 @@ def authenticate(self):
             task = loop.create_task(self._client._get("/auth/profile"))
             # Run the task to completion
             result = loop.run_until_complete(task)
+
+            # Set the user_id if successful
+            if result and isinstance(result, dict) and 'user_id' in result:
+                self._client.user_id = result['user_id']
+
             return result
         finally:
             loop.close()

quotientai/tracing/core.py

@@ -123,6 +123,17 @@ def _create_otlp_exporter(self, endpoint: str, headers: dict):
         """
         return OTLPSpanExporter(endpoint=endpoint, headers=headers)
 
+
+    def _get_user_id(self):
+        """
+        Get user_id from client.
+        Returns the user_id or None if not found.
+        """
+        if hasattr(self._client, 'user_id'):
+            return self._client.user_id
+        return None
+
+
     @functools.lru_cache()
     def _setup_auto_collector(self, app_name: str, environment: str, instruments: Optional[tuple] = None, detections: Optional[str] = None):
         """
@@ -139,6 +150,7 @@ def _setup_auto_collector(self, app_name: str, environment: str, instruments: Op
                 resource_attributes = {
                     QuotientAttributes.app_name: app_name,
                     QuotientAttributes.environment: environment,
+                    "quotient.user_id": self._get_user_id(),
                 }
 
                 if detections is not None: