Retain env info in patch extra (#1162)

langchain-ai · Nov 1, 2024 · 50265fb · 50265fb
1 parent 5135d38
commit 50265fb
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/langsmith/client.py b/python/langsmith/client.py
@@ -1665,6 +1665,8 @@ def update_run(
             data["outputs"] = self._hide_run_outputs(outputs)
         if events is not None:
             data["events"] = events
+        if data["extra"]:
+            self._insert_runtime_env([data])
         if use_multipart and self.tracing_queue is not None:
             # not collecting attachments currently, use empty dict
             serialized_op = serialize_run_dict(operation="patch", payload=data)