check kwargs

jxnl · Apr 2, 2024 · 4560ae9 · 4560ae9
1 parent 8a28d7e
commit 4560ae9
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/instructor/messages_middleware.py b/instructor/messages_middleware.py
@@ -21,10 +21,15 @@ async def __call__(
 
 
 def messages_middleware(func: Callable) -> MessageMiddleware:
+    import inspect
+
+    if "messages" not in inspect.signature(func).parameters:
+        raise ValueError("`messages` must be a parameter of the middleware function")
+
     class _Middleware(MessageMiddleware):
         def __call__(
             self, messages: List[ChatCompletionMessageParam]
         ) -> List[ChatCompletionMessageParam]:
-            return func(messages)
+            return func(messages=messages)
 
     return _Middleware()