From c1e5160c114e940cf1fc798968e06da0f17b3916 Mon Sep 17 00:00:00 2001
From: Nuno Campos <nuno@langchain.dev>
Date: Thu, 10 Oct 2024 15:24:38 -0700
Subject: [PATCH] py: Add session_id and session_name to patch parts in
 batch/multipart

---
 python/langsmith/client.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/python/langsmith/client.py b/python/langsmith/client.py
index 6fe4d33c0..3ea781a36 100644
--- a/python/langsmith/client.py
+++ b/python/langsmith/client.py
@@ -1774,6 +1774,8 @@ def update_run(
             "dotted_order": kwargs.pop("dotted_order", None),
             "tags": tags,
             "extra": extra,
+            "session_id": kwargs.pop("session_id", None),
+            "session_name": kwargs.pop("session_name", None),
         }
         if not self._filter_for_sampling([data], patch=True):
             return