diff --git a/src/openlayer/lib/integrations/langchain_callback.py b/src/openlayer/lib/integrations/langchain_callback.py index 33d79edc..ed31bc6b 100644 --- a/src/openlayer/lib/integrations/langchain_callback.py +++ b/src/openlayer/lib/integrations/langchain_callback.py @@ -191,6 +191,11 @@ def _process_and_upload_trace(self, trace: traces.Trace) -> None: ) ) + # Add reserved column configurations for user context + if "user_id" in trace_data: + config.update({"user_id_column_name": "user_id"}) + if "session_id" in trace_data: + config.update({"session_id_column_name": "session_id"}) if "groundTruth" in trace_data: config.update({"ground_truth_column_name": "groundTruth"}) if "context" in trace_data: @@ -1225,6 +1230,11 @@ def _process_and_upload_async_trace(self, trace: traces.Trace) -> None: ) ) + # Add reserved column configurations for user context + if "user_id" in trace_data: + config.update({"user_id_column_name": "user_id"}) + if "session_id" in trace_data: + config.update({"session_id_column_name": "session_id"}) if "groundTruth" in trace_data: config.update({"ground_truth_column_name": "groundTruth"}) if "context" in trace_data: