diff --git a/instructor/__init__.py b/instructor/__init__.py index 50f734f..6a659e5 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, Validator, llm_validator, CitationMixin +from .dsl import MultiTask, Maybe, llm_validator, CitationMixin from .patch import patch __all__ = [ @@ -10,6 +10,5 @@ __all__ = [ "Maybe", "openai_schema", "patch", - "Validator", "llm_validator", ] diff --git a/instructor/dsl/citation.py b/instructor/dsl/citation.py index 78c9b7d..ba27827 100644 --- a/instructor/dsl/citation.py +++ b/instructor/dsl/citation.py @@ -64,7 +64,6 @@ class CitationMixin(BaseModel): If the span is not found, remove the substring_phrase from the list. """ if info.context is None: - logger.info("No context found, skipping validation") return self # Get the context from the info diff --git a/instructor/dsl/multitask.py b/instructor/dsl/multitask.py index 15897ca..0a7c344 100644 --- a/instructor/dsl/multitask.py +++ b/instructor/dsl/multitask.py @@ -118,7 +118,7 @@ def MultiTask( new_cls = create_model( name, tasks=list_tasks, - __base__=(OpenAISchema, MultiTaskBase), + __base__=(OpenAISchema, MultiTaskBase), # type: ignore ) # set the class constructor BaseModel new_cls.task_type = subtask_class diff --git a/instructor/patch.py b/instructor/patch.py index 28f46b9..686a0ef 100644 --- a/instructor/patch.py +++ b/instructor/patch.py @@ -103,13 +103,13 @@ def wrap_chatcompletion(func: Callable) -> Callable: @wraps(func) async def new_chatcompletion_async( - response_model=None, valiation_context=None, *args, max_retries=0, **kwargs + response_model=None, validation_context=None, *args, max_retries=0, **kwargs ): response_model, new_kwargs = handle_response_model(response_model, kwargs) # type: ignore response, error = await retry_async( func=func, response_model=response_model, - valiation_context=valiation_context, + validation_context=validation_context, max_retries=max_retries, args=args, kwargs=new_kwargs,