File tree Expand file tree Collapse file tree
tests/integrations/pydantic_ai Expand file tree Collapse file tree Original file line number Diff line number Diff line change 1919from sentry_sdk .integrations .pydantic_ai .spans .utils import _set_usage_data
2020from sentry_sdk .utils import package_version
2121
22- PYDANTIC_VERSION = package_version ("pydantic-ai" )
22+ PYDANTIC_AI_VERSION = package_version ("pydantic-ai" )
2323
2424
2525@pytest .fixture
@@ -520,7 +520,7 @@ async def test_agent_run_stream_events(
520520 if stream_gen_ai_spans :
521521 items = capture_items ("transaction" , "span" )
522522
523- if PYDANTIC_VERSION > (2 ,):
523+ if PYDANTIC_AI_VERSION > (2 ,):
524524 async with test_agent .run_stream_events (
525525 ["Message demonstrating the absence of truncation." , "Test input" ]
526526 ) as stream_events :
@@ -551,7 +551,7 @@ async def test_agent_run_stream_events(
551551 else :
552552 events = capture_events ()
553553
554- if PYDANTIC_VERSION > (2 ,):
554+ if PYDANTIC_AI_VERSION > (2 ,):
555555 async with test_agent .run_stream_events ("Test input" ) as stream_events :
556556 async for _ in stream_events :
557557 pass
You can’t perform that action at this time.
0 commit comments