improve imports

This commit is contained in:
Jason
2023-09-07 20:54:49 -04:00
parent a10e8c81da
commit d082b2ea44
3 changed files with 12 additions and 4 deletions
+3 -1
View File
@@ -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",
]
+9 -1
View File
@@ -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",
]
-2
View File
@@ -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