@@ -121,6 +121,7 @@ def traced_pregel_stream(langgraph, pin, func, instance, args, kwargs):
121
121
result = func (* args , ** kwargs )
122
122
except Exception :
123
123
span .set_exc_info (* sys .exc_info ())
124
+ integration .llmobs_set_tags (span , args = args , kwargs = {** kwargs , "name" : name }, response = None , operation = "graph" )
124
125
span .finish ()
125
126
raise
126
127
@@ -139,6 +140,9 @@ def _stream():
139
140
break
140
141
except Exception :
141
142
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
+ )
142
146
span .finish ()
143
147
raise
144
148
@@ -160,6 +164,7 @@ def traced_pregel_astream(langgraph, pin, func, instance, args, kwargs):
160
164
result = func (* args , ** kwargs )
161
165
except Exception :
162
166
span .set_exc_info (* sys .exc_info ())
167
+ integration .llmobs_set_tags (span , args = args , kwargs = {** kwargs , "name" : name }, response = None , operation = "graph" )
163
168
span .finish ()
164
169
raise
165
170
@@ -178,6 +183,9 @@ async def _astream():
178
183
break
179
184
except Exception :
180
185
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
+ )
181
189
span .finish ()
182
190
raise
183
191
0 commit comments