From f4512439b99906f68e79f908ec78c656d2e8761b Mon Sep 17 00:00:00 2001 From: Gustavo Cid Date: Fri, 23 Jan 2026 16:04:46 -0300 Subject: [PATCH] fix(closes OPEN-8768): langChain callback handler not using sessions and users --- src/openlayer/lib/integrations/langchain_callback.py | 10 ++++++++++ 1 file changed, 10 insertions(+) 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: