Skip to content

Commit 14e379f

Browse files
fix pydantic-ai test
1 parent dcc3fbe commit 14e379f

1 file changed

Lines changed: 6 additions & 3 deletions

File tree

tests/integrations/pydantic_ai/test_pydantic_ai.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1366,9 +1366,6 @@ async def test_message_history(
13661366
_experiments={"stream_gen_ai_spans": stream_gen_ai_spans},
13671367
)
13681368

1369-
# First message
1370-
await agent.run("Hello, I'm Alice")
1371-
13721369
# Second message with history
13731370
from pydantic_ai import messages
13741371

@@ -1385,6 +1382,9 @@ async def test_message_history(
13851382
if stream_gen_ai_spans:
13861383
items = capture_items("transaction", "span")
13871384

1385+
# First message
1386+
await agent.run("Hello, I'm Alice")
1387+
13881388
await agent.run("What is my name?", message_history=history)
13891389

13901390
# We should have 2 transactions
@@ -1407,6 +1407,9 @@ async def test_message_history(
14071407
else:
14081408
events = capture_events()
14091409

1410+
# First message
1411+
await agent.run("Hello, I'm Alice")
1412+
14101413
await agent.run("What is my name?", message_history=history)
14111414

14121415
# We should have 2 transactions

0 commit comments

Comments
 (0)