chore(streaming): update constructor to use direct client names (#285)

anthropics · Dec 19, 2023 · 0c55c84 · 0c55c84
1 parent b9e38b2
commit 0c55c84
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/anthropic/_streaming.py b/src/anthropic/_streaming.py
@@ -10,7 +10,7 @@
 from ._types import ResponseT
 
 if TYPE_CHECKING:
-    from ._base_client import SyncAPIClient, AsyncAPIClient
+    from ._client import Anthropic, AsyncAnthropic
 
 
 class Stream(Generic[ResponseT]):
@@ -23,7 +23,7 @@ def __init__(
         *,
         cast_to: type[ResponseT],
         response: httpx.Response,
-        client: SyncAPIClient,
+        client: Anthropic,
     ) -> None:
         self.response = response
         self._cast_to = cast_to
@@ -84,7 +84,7 @@ def __init__(
         *,
         cast_to: type[ResponseT],
         response: httpx.Response,
-        client: AsyncAPIClient,
+        client: AsyncAnthropic,
     ) -> None:
         self.response = response
         self._cast_to = cast_to