Skip to content

Commit ad56495

Browse files
committed
fix llmobs traces not submitting for langgraph errors
1 parent f3241d3 commit ad56495

File tree

1 file changed

+8
-0
lines changed
  • ddtrace/contrib/internal/langgraph

1 file changed

+8
-0
lines changed

ddtrace/contrib/internal/langgraph/patch.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -121,6 +121,7 @@ def traced_pregel_stream(langgraph, pin, func, instance, args, kwargs):
121121
result = func(*args, **kwargs)
122122
except Exception:
123123
span.set_exc_info(*sys.exc_info())
124+
integration.llmobs_set_tags(span, args=args, kwargs={**kwargs, "name": name}, response=None, operation="graph")
124125
span.finish()
125126
raise
126127

@@ -139,6 +140,9 @@ def _stream():
139140
break
140141
except Exception:
141142
span.set_exc_info(*sys.exc_info())
143+
integration.llmobs_set_tags(
144+
span, args=args, kwargs={**kwargs, "name": name}, response=None, operation="graph"
145+
)
142146
span.finish()
143147
raise
144148

@@ -160,6 +164,7 @@ def traced_pregel_astream(langgraph, pin, func, instance, args, kwargs):
160164
result = func(*args, **kwargs)
161165
except Exception:
162166
span.set_exc_info(*sys.exc_info())
167+
integration.llmobs_set_tags(span, args=args, kwargs={**kwargs, "name": name}, response=None, operation="graph")
163168
span.finish()
164169
raise
165170

@@ -178,6 +183,9 @@ async def _astream():
178183
break
179184
except Exception:
180185
span.set_exc_info(*sys.exc_info())
186+
integration.llmobs_set_tags(
187+
span, args=args, kwargs={**kwargs, "name": name}, response=None, operation="graph"
188+
)
181189
span.finish()
182190
raise
183191

0 commit comments

Comments
 (0)