From 733356a8e4accb82d1d8ae86e0ea219151a50b5b Mon Sep 17 00:00:00 2001 From: Stainless Bot Date: Tue, 19 Dec 2023 12:30:52 +0000 Subject: [PATCH] chore(streaming): update constructor to use direct client names --- src/openai/_streaming.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/openai/_streaming.py b/src/openai/_streaming.py index e48324fc78..e323c59ac0 100644 --- a/src/openai/_streaming.py +++ b/src/openai/_streaming.py @@ -12,7 +12,7 @@ from ._exceptions import APIError if TYPE_CHECKING: - from ._base_client import SyncAPIClient, AsyncAPIClient + from ._client import OpenAI, AsyncOpenAI class Stream(Generic[ResponseT]): @@ -25,7 +25,7 @@ def __init__( *, cast_to: type[ResponseT], response: httpx.Response, - client: SyncAPIClient, + client: OpenAI, ) -> None: self.response = response self._cast_to = cast_to @@ -79,7 +79,7 @@ def __init__( *, cast_to: type[ResponseT], response: httpx.Response, - client: AsyncAPIClient, + client: AsyncOpenAI, ) -> None: self.response = response self._cast_to = cast_to