fix(closes OPEN-8768): langChain callback handler not using sessions and users

gustavocidornelas · gustavocidornelas · commit f4512439b999 · 2026-01-23T16:04:46.000-03:00
diff --git 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:

Original file line number	Diff line number	Diff line change
`@@ -191,6 +191,11 @@ def _process_and_upload_trace(self, trace: traces.Trace) -> None:`
`191`	`191`	`)`
`192`	`192`	`)`
`193`	`193`
	`194`	`+ # Add reserved column configurations for user context`
	`195`	`+ if "user_id" in trace_data:`
	`196`	`+ config.update({"user_id_column_name": "user_id"})`
	`197`	`+ if "session_id" in trace_data:`
	`198`	`+ config.update({"session_id_column_name": "session_id"})`
`194`	`199`	`if "groundTruth" in trace_data:`
`195`	`200`	`config.update({"ground_truth_column_name": "groundTruth"})`
`196`	`201`	`if "context" in trace_data:`
`@@ -1225,6 +1230,11 @@ def _process_and_upload_async_trace(self, trace: traces.Trace) -> None:`
`1225`	`1230`	`)`
`1226`	`1231`	`)`
`1227`	`1232`
	`1233`	`+ # Add reserved column configurations for user context`
	`1234`	`+ if "user_id" in trace_data:`
	`1235`	`+ config.update({"user_id_column_name": "user_id"})`
	`1236`	`+ if "session_id" in trace_data:`
	`1237`	`+ config.update({"session_id_column_name": "session_id"})`
`1228`	`1238`	`if "groundTruth" in trace_data:`
`1229`	`1239`	`config.update({"ground_truth_column_name": "groundTruth"})`
`1230`	`1240`	`if "context" in trace_data:`