From d082b2ea44b32ff05ddb9e0f0a775442ced2a48f Mon Sep 17 00:00:00 2001 From: Jason Date: Thu, 7 Sep 2023 20:54:49 -0400 Subject: [PATCH] improve imports --- instructor/__init__.py | 4 +++- instructor/dsl/__init__.py | 10 +++++++++- instructor/dsl/validators.py | 2 -- 3 files changed, 12 insertions(+), 4 deletions(-) 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