fix(client): accidental blocking sleep in async code (#1415)

openai · May 13, 2024 · 0ac6ecb · 0ac6ecb
1 parent ed73d1d
commit 0ac6ecb
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/openai/resources/beta/threads/runs/runs.py b/src/openai/resources/beta/threads/runs/runs.py
@@ -2,7 +2,6 @@
 
 from __future__ import annotations
 
-import time
 import typing_extensions
 from typing import Union, Iterable, Optional, overload
 from functools import partial
@@ -1105,7 +1104,7 @@ def poll(
                 else:
                     poll_interval_ms = 1000
 
-            time.sleep(poll_interval_ms / 1000)
+            self._sleep(poll_interval_ms / 1000)
 
     @overload
     def stream(
@@ -2639,7 +2638,7 @@ async def poll(
                 else:
                     poll_interval_ms = 1000
 
-            time.sleep(poll_interval_ms / 1000)
+            await self._sleep(poll_interval_ms / 1000)
 
     @overload
     def stream(