fix: bump cerebras version

instructor-ai · Nov 14, 2024 · 802a34a · 802a34a
1 parent ebbf754
commit 802a34a
Show file tree

Hide file tree

Showing 4 changed files with 1,040 additions and 1,014 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,