pydantic
diff --git a/‎tests/otel_integrations/test_langchain.py‎
Lines changed: 2 additions & 3 deletions b/‎tests/otel_integrations/test_langchain.py‎
Lines changed: 2 additions & 3 deletions
@@ -107,9 +107,8 @@ def add(a: float, b: float) -> float:
         ]
     )
 
-    spans = [s for s in spans if s['name'] == 'ChatOpenAI']
-    attributes = spans[-1]['attributes']
-    assert attributes['all_messages_events'] == snapshot(
+    [span] = [s for s in spans if s['name'] == 'ChatOpenAI' and len(s['attributes']['all_messages_events']) == 4]
+    assert span['attributes']['all_messages_events'] == snapshot(
         [
             {'content': "what's 123 + 456?", 'role': 'user'},
             {
Original file line number	Diff line number	Diff line change
`@@ -107,9 +107,8 @@ def add(a: float, b: float) -> float:`
`107`	`107`	`]`
`108`	`108`	`)`
`109`	`109`
`110`		`- spans = [s for s in spans if s['name'] == 'ChatOpenAI']`
`111`		`- attributes = spans[-1]['attributes']`
`112`		`- assert attributes['all_messages_events'] == snapshot(`
	`110`	`+ [span] = [s for s in spans if s['name'] == 'ChatOpenAI' and len(s['attributes']['all_messages_events']) == 4]`
	`111`	`+ assert span['attributes']['all_messages_events'] == snapshot(`
`113`	`112`	`[`
`114`	`113`	`{'content': "what's 123 + 456?", 'role': 'user'},`
`115`	`114`	`{`