diff --git a/docs/sections/how_to_guides/advanced/structured_generation.md b/docs/sections/how_to_guides/advanced/structured_generation.md index 17e64430a7..ee7565272c 100644 --- a/docs/sections/how_to_guides/advanced/structured_generation.md +++ b/docs/sections/how_to_guides/advanced/structured_generation.md @@ -139,7 +139,7 @@ When working with model providers behind an API, there's no direct way of access ``` !!! Note - Take a look at [`InstructorStructuredOutputType`][distilabel.steps.tasks.structured_outputs.instructor.InstructorStructuredOutputType] to see the expected format + Take a look at [`InstructorStructuredOutputType`][distilabel.steps.tasks.typing.InstructorStructuredOutputType] to see the expected format of the `structured_output` dict variable. The following is the same example you can see with `outlines`'s `JSON` section for comparison purposes. diff --git a/src/distilabel/steps/tasks/structured_outputs/instructor.py b/src/distilabel/steps/tasks/structured_outputs/instructor.py index 7249457d17..94ab1097e5 100644 --- a/src/distilabel/steps/tasks/structured_outputs/instructor.py +++ b/src/distilabel/steps/tasks/structured_outputs/instructor.py @@ -49,18 +49,6 @@ """Available clients that can be wrapped with `instructor`. """ -# class InstructorStructuredOutputType(TypedDict): -# """TypedDict to represent the structured output configuration from `instructor`.""" - -# schema: Type[BaseModel] -# """The schema to use for the structured output, a `pydantic.BaseModel` class. """ -# mode: Optional["instructor.Mode"] -# """Generation mode. Take a look at `instructor.Mode` for more information, if not informed it will -# be determined automatically. """ -# max_retries: int -# """Number of times to reask the model in case of error, if not set will default to the model's default. """ - - def _client_patcher(framework: InstructorFrameworks) -> Tuple[Callable, str]: """Helper function to return the appropriate instructor client for the given framework.