adding support for async validation

instructor-ai · Jan 25, 2025 · 76a7478 · 76a7478
1 parent f86df78
commit 76a7478
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/instructor/process_response.py b/instructor/process_response.py
@@ -84,6 +84,8 @@ async def process_response_async(
             response,
             mode=mode,
         )
+        if hasattr(response_model, "model_async_validate"):
+            await response_model.model_async_validate(model)
         return model
 
     model = response_model.from_response(
@@ -92,6 +94,8 @@ async def process_response_async(
         strict=strict,
         mode=mode,
     )
+    if hasattr(response_model, "model_async_validate"):
+        await response_model.model_async_validate(model)
 
     # ? This really hints at the fact that we need a better way of
     # ? attaching usage data and the raw response to the model we return.