ruff

instructor-ai · Apr 2, 2024 · 8a28d7e · 8a28d7e
1 parent 6e50514
commit 8a28d7e
Show file tree

Hide file tree

Showing 5 changed files with 8 additions and 9 deletions.
diff --git a/examples/middleware/run.py b/examples/middleware/run.py
@@ -6,7 +6,6 @@
 
 
 class PrintLastUserMessage(instructor.MessageMiddleware):
-
     log: bool = False
 
     def __call__(

diff --git a/instructor/__init__.py b/instructor/__init__.py
@@ -23,6 +23,7 @@
     "Instructor",
     "MessageMiddleware",
     "AsyncMessageMiddleware",
+    "messages_middleware",
     "from_openai",
     "from_anthropic",
     "from_litellm",

diff --git a/instructor/client.py b/instructor/client.py
@@ -390,10 +390,13 @@ def from_anthropic(
     mode: instructor.Mode = instructor.Mode.ANTHROPIC_JSON,
     **kwargs,
 ) -> Instructor | AsyncInstructor:
-    assert mode in {
-        instructor.Mode.ANTHROPIC_JSON,
-        instructor.Mode.ANTHROPIC_TOOLS,
-    }, "Mode be one of {instructor.Mode.ANTHROPIC_JSON, instructor.Mode.ANTHROPIC_TOOLS}"
+    assert (
+        mode
+        in {
+            instructor.Mode.ANTHROPIC_JSON,
+            instructor.Mode.ANTHROPIC_TOOLS,
+        }
+    ), "Mode be one of {instructor.Mode.ANTHROPIC_JSON, instructor.Mode.ANTHROPIC_TOOLS}"
 
     assert isinstance(
         client, (anthropic.Anthropic, anthropic.AsyncAnthropic)

diff --git a/instructor/messages_middleware.py b/instructor/messages_middleware.py
@@ -5,7 +5,6 @@
 
 
 class MessageMiddleware(BaseModel, ABC):
-
     @abstractmethod
     def __call__(
         self, messages: List[ChatCompletionMessageParam]
@@ -14,7 +13,6 @@ def __call__(
 
 
 class AsyncMessageMiddleware(MessageMiddleware):
-
     @abstractmethod
     async def __call__(
         self, messages: List[ChatCompletionMessageParam]
@@ -23,7 +21,6 @@ async def __call__(
 
 
 def messages_middleware(func: Callable) -> MessageMiddleware:
-
     class _Middleware(MessageMiddleware):
         def __call__(
             self, messages: List[ChatCompletionMessageParam]

diff --git a/tests/llm/test_openai/test_validators.py b/tests/llm/test_openai/test_validators.py
@@ -44,7 +44,6 @@ class QuestionAnswerNoEvil(BaseModel):
 
 @pytest.mark.parametrize("model", models)
 def test_runmodel_validator_default_openai_client(model, client):
-
     client = instructor.from_openai(client)
 
     class QuestionAnswerNoEvil(BaseModel):