also add span context on proxy runs (#14)

akhatre · web-flow · commit aa7d021ce59d · 2025-08-27T23:08:44.000+05:00
Co-authored-by: akhatre &lt;akhatre@users.noreply.github.com&gt;
diff --git a/overmind/langchain/callbacks.py b/overmind/langchain/callbacks.py
@@ -113,6 +113,8 @@ def on_chain_end(self, outputs, *, run_id, **kwargs):
             self.run_spans[run_id].set_attribute(
                 "policy_results", serialize(outputs.pop("policy_results"))
             )
+
+        if "span_context" in outputs:
             # can't add links to the span that has been started and there is no clean way to pass
             # this span to the backend at the layer run time (since it happening in a downstream node)
             self.run_spans[run_id].set_attribute(
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "overmind"
-version = "0.1.13"
+version = "0.1.14"
 description = "Python client for Overmind API"
 authors = ["Overmind Ltd"]
 readme = "README.md"

Original file line number	Diff line number	Diff line change
`@@ -113,6 +113,8 @@ def on_chain_end(self, outputs, , run_id, *kwargs):`
`113`	`113`	`self.run_spans[run_id].set_attribute(`
`114`	`114`	`"policy_results", serialize(outputs.pop("policy_results"))`
`115`	`115`	`)`
	`116`	`+`
	`117`	`+ if "span_context" in outputs:`
`116`	`118`	`# can't add links to the span that has been started and there is no clean way to pass`
`117`	`119`	`# this span to the backend at the layer run time (since it happening in a downstream node)`
`118`	`120`	`self.run_spans[run_id].set_attribute(`