add beta support for anthropic

instructor-ai · Nov 25, 2024 · 047a5bb · 047a5bb
1 parent 58eef74
commit 047a5bb
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/instructor/client_anthropic.py b/instructor/client_anthropic.py
@@ -41,6 +41,7 @@ def from_anthropic(
     ),
     mode: instructor.Mode = instructor.Mode.ANTHROPIC_TOOLS,
     enable_prompt_caching: bool = False,
+    beta:bool = False,
     **kwargs: Any,
 ) -> instructor.Instructor | instructor.AsyncInstructor:
     assert (
@@ -70,6 +71,8 @@ def from_anthropic(
             raise TypeError(
                 "Client must be an instance of {anthropic.Anthropic, anthropic.AsyncAnthropic} to enable prompt caching"
             )
+    elif beta:
+        create = client.beta.messages.create
     else:
         create = client.messages.create