Skip to content

Commit 93f4402

Browse files
committed
Update stream close test to match new _closing helper
1 parent f3eb5b3 commit 93f4402

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

tests/test_openai_llm_timeout.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -149,13 +149,9 @@ class MockAsyncStream:
149149
def __init__(self):
150150
self.iteration_count = 0
151151

152-
async def __aenter__(self):
153-
return self
154-
155-
async def __aexit__(self, exc_type, exc_val, exc_tb):
152+
async def close(self):
156153
nonlocal stream_closed
157154
stream_closed = True
158-
return False
159155

160156
def __aiter__(self):
161157
return self

0 commit comments

Comments
 (0)