fix: removed use_async flag in cerebras client

instructor-ai · Nov 14, 2024 · 1b747af · 1b747af
1 parent 78a1926
commit 1b747af
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/instructor/client_cerebras.py b/instructor/client_cerebras.py
@@ -13,7 +13,6 @@
 def from_cerebras(
     client: Cerebras,
     mode: instructor.Mode = instructor.Mode.CEREBRAS_TOOLS,
-    use_async: Literal[False] = False,
     **kwargs: Any,
 ) -> Instructor: ...
 
@@ -22,15 +21,13 @@ def from_cerebras(
 def from_cerebras(
     client: AsyncCerebras,
     mode: instructor.Mode = instructor.Mode.CEREBRAS_TOOLS,
-    use_async: Literal[True] = True,
     **kwargs: Any,
 ) -> AsyncInstructor: ...
 
 
 def from_cerebras(
     client: Cerebras | AsyncCerebras,
     mode: instructor.Mode = instructor.Mode.CEREBRAS_TOOLS,
-    use_async: bool = False,
     **kwargs: Any,
 ) -> Instructor | AsyncInstructor:
     assert (
@@ -45,7 +42,7 @@ def from_cerebras(
         client, (Cerebras, AsyncCerebras)
     ), "Client must be an instance of Cerebras or AsyncCerebras"
 
-    if use_async:
+    if isinstance(client, AsyncCerebras):
         create = client.chat.completions.create
         return AsyncInstructor(
             client=client,