diff --git a/instructor/__init__.py b/instructor/__init__.py index 4a1761f..18a6a26 100644 --- a/instructor/__init__.py +++ b/instructor/__init__.py @@ -1,5 +1,5 @@ from .function_calls import OpenAISchema, openai_function, openai_schema -from .dsl import MultiTask, Maybe +from .dsl import MultiTask, Maybe, Validator, llm_validator from .patch import patch __all__ = [ @@ -9,4 +9,6 @@ __all__ = [ "Maybe", "openai_schema", "patch", + "Validator", + "llm_validator", ] diff --git a/instructor/dsl/__init__.py b/instructor/dsl/__init__.py index c48e738..441128d 100644 --- a/instructor/dsl/__init__.py +++ b/instructor/dsl/__init__.py @@ -2,5 +2,13 @@ from .completion import ChatCompletion from .messages import * from .multitask import MultiTask from .maybe import Maybe +from .validators import Validator, llm_validator -__all__ = ["ChatCompletion", "MultiTask", "messages", "Maybe"] +__all__ = [ + "ChatCompletion", + "MultiTask", + "messages", + "Maybe", + "Validator", + "llm_validator", +] diff --git a/instructor/dsl/validators.py b/instructor/dsl/validators.py index 780125d..10685f7 100644 --- a/instructor/dsl/validators.py +++ b/instructor/dsl/validators.py @@ -58,8 +58,6 @@ def llm_validator( temperature=temperature, ) # type: ignore - print(resp.model_dump_json(indent=2)) - # If the response is not valid, return the reason, this could be used in # the future to generate a better response, via reasking mechanism. assert resp.is_valid, resp.reason