Skip to content

Commit ac5eeae

Browse files
committed
Fixed tests
1 parent c43f5b4 commit ac5eeae

File tree

2 files changed

+6
-4
lines changed

2 files changed

+6
-4
lines changed

sentry_sdk/integrations/openai.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -577,12 +577,12 @@ async def _execute_async(f, *args, **kwargs):
577577
return e.value
578578

579579
@wraps(f)
580-
async def _sentry_patched_create_async(*args, **kwargs):
580+
async def _sentry_patched_responses_async(*args, **kwargs):
581581
# type: (Any, Any) -> Any
582582
integration = sentry_sdk.get_client().get_integration(OpenAIIntegration)
583583
if integration is None:
584584
return await f(*args, **kwargs)
585585

586586
return await _execute_async(f, *args, **kwargs)
587587

588-
return _sentry_patched_create_async
588+
return _sentry_patched_responses_async

tests/integrations/openai/test_openai.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1076,6 +1076,7 @@ def test_error_in_responses_api(sentry_init, capture_events):
10761076
)
10771077

10781078

1079+
@pytest.mark.asyncio
10791080
@pytest.mark.skipif(SKIP_RESPONSES_TESTS, reason="Responses API not available")
10801081
async def test_ai_client_span_responses_async_api(sentry_init, capture_events):
10811082
sentry_init(
@@ -1086,7 +1087,7 @@ async def test_ai_client_span_responses_async_api(sentry_init, capture_events):
10861087
events = capture_events()
10871088

10881089
client = AsyncOpenAI(api_key="z")
1089-
client.responses._post = mock.Mock(return_value=EXAMPLE_RESPONSE)
1090+
client.responses._post = AsyncMock(return_value=EXAMPLE_RESPONSE)
10901091

10911092
with start_transaction(name="openai tx"):
10921093
await client.responses.create(
@@ -1115,6 +1116,7 @@ async def test_ai_client_span_responses_async_api(sentry_init, capture_events):
11151116
}
11161117

11171118

1119+
@pytest.mark.asyncio
11181120
@pytest.mark.skipif(SKIP_RESPONSES_TESTS, reason="Responses API not available")
11191121
async def test_error_in_responses_async_api(sentry_init, capture_events):
11201122
sentry_init(
@@ -1125,7 +1127,7 @@ async def test_error_in_responses_async_api(sentry_init, capture_events):
11251127
events = capture_events()
11261128

11271129
client = AsyncOpenAI(api_key="z")
1128-
client.responses._post = mock.Mock(
1130+
client.responses._post = AsyncMock(
11291131
side_effect=OpenAIError("API rate limit reached")
11301132
)
11311133

0 commit comments

Comments
 (0)