lightningrod-ai 0.1.6__py3-none-any.whl

lightningrod/_generated/models/question_and_label_generator.py

@@ -0,0 +1,217 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Any, Literal, TypeVar, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+if TYPE_CHECKING:
+    from ..models.answer_type import AnswerType
+    from ..models.filter_criteria import FilterCriteria
+
+
+T = TypeVar("T", bound="QuestionAndLabelGenerator")
+
+
+@_attrs_define
+class QuestionAndLabelGenerator:
+    """
+    Attributes:
+        config_type (Literal['QUESTION_AND_LABEL_GENERATOR'] | Unset): Type of transform configuration Default:
+            'QUESTION_AND_LABEL_GENERATOR'.
+        instructions (None | str | Unset): Instructions for question generation. If not provided, uses sensible
+            defaults.
+        examples (list[str] | Unset): Example questions to guide generation
+        bad_examples (list[str] | Unset): Examples of questions to avoid
+        filter_ (FilterCriteria | list[FilterCriteria] | None | Unset): Optional filter criteria to apply after question
+            generation
+        questions_per_seed (int | Unset): Number of questions to generate per seed Default: 1.
+        include_default_filter (bool | Unset): Whether to include the default filter for generated questions Default:
+            False.
+        answer_type (AnswerType | None | Unset): The type of answer expected for generated questions
+    """
+
+    config_type: Literal["QUESTION_AND_LABEL_GENERATOR"] | Unset = "QUESTION_AND_LABEL_GENERATOR"
+    instructions: None | str | Unset = UNSET
+    examples: list[str] | Unset = UNSET
+    bad_examples: list[str] | Unset = UNSET
+    filter_: FilterCriteria | list[FilterCriteria] | None | Unset = UNSET
+    questions_per_seed: int | Unset = 1
+    include_default_filter: bool | Unset = False
+    answer_type: AnswerType | None | Unset = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        from ..models.answer_type import AnswerType
+        from ..models.filter_criteria import FilterCriteria
+
+        config_type = self.config_type
+
+        instructions: None | str | Unset
+        if isinstance(self.instructions, Unset):
+            instructions = UNSET
+        else:
+            instructions = self.instructions
+
+        examples: list[str] | Unset = UNSET
+        if not isinstance(self.examples, Unset):
+            examples = self.examples
+
+        bad_examples: list[str] | Unset = UNSET
+        if not isinstance(self.bad_examples, Unset):
+            bad_examples = self.bad_examples
+
+        filter_: dict[str, Any] | list[dict[str, Any]] | None | Unset
+        if isinstance(self.filter_, Unset):
+            filter_ = UNSET
+        elif isinstance(self.filter_, FilterCriteria):
+            filter_ = self.filter_.to_dict()
+        elif isinstance(self.filter_, list):
+            filter_ = []
+            for filter_type_1_item_data in self.filter_:
+                filter_type_1_item = filter_type_1_item_data.to_dict()
+                filter_.append(filter_type_1_item)
+
+        else:
+            filter_ = self.filter_
+
+        questions_per_seed = self.questions_per_seed
+
+        include_default_filter = self.include_default_filter
+
+        answer_type: dict[str, Any] | None | Unset
+        if isinstance(self.answer_type, Unset):
+            answer_type = UNSET
+        elif isinstance(self.answer_type, AnswerType):
+            answer_type = self.answer_type.to_dict()
+        else:
+            answer_type = self.answer_type
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if config_type is not UNSET:
+            field_dict["config_type"] = config_type
+        if instructions is not UNSET:
+            field_dict["instructions"] = instructions
+        if examples is not UNSET:
+            field_dict["examples"] = examples
+        if bad_examples is not UNSET:
+            field_dict["bad_examples"] = bad_examples
+        if filter_ is not UNSET:
+            field_dict["filter"] = filter_
+        if questions_per_seed is not UNSET:
+            field_dict["questions_per_seed"] = questions_per_seed
+        if include_default_filter is not UNSET:
+            field_dict["include_default_filter"] = include_default_filter
+        if answer_type is not UNSET:
+            field_dict["answer_type"] = answer_type
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.answer_type import AnswerType
+        from ..models.filter_criteria import FilterCriteria
+
+        d = dict(src_dict)
+        config_type = cast(Literal["QUESTION_AND_LABEL_GENERATOR"] | Unset, d.pop("config_type", UNSET))
+        if config_type != "QUESTION_AND_LABEL_GENERATOR" and not isinstance(config_type, Unset):
+            raise ValueError(f"config_type must match const 'QUESTION_AND_LABEL_GENERATOR', got '{config_type}'")
+
+        def _parse_instructions(data: object) -> None | str | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(None | str | Unset, data)
+
+        instructions = _parse_instructions(d.pop("instructions", UNSET))
+
+        examples = cast(list[str], d.pop("examples", UNSET))
+
+        bad_examples = cast(list[str], d.pop("bad_examples", UNSET))
+
+        def _parse_filter_(data: object) -> FilterCriteria | list[FilterCriteria] | None | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                filter_type_0 = FilterCriteria.from_dict(data)
+
+                return filter_type_0
+            except (TypeError, ValueError, AttributeError, KeyError):
+                pass
+            try:
+                if not isinstance(data, list):
+                    raise TypeError()
+                filter_type_1 = []
+                _filter_type_1 = data
+                for filter_type_1_item_data in _filter_type_1:
+                    filter_type_1_item = FilterCriteria.from_dict(filter_type_1_item_data)
+
+                    filter_type_1.append(filter_type_1_item)
+
+                return filter_type_1
+            except (TypeError, ValueError, AttributeError, KeyError):
+                pass
+            return cast(FilterCriteria | list[FilterCriteria] | None | Unset, data)
+
+        filter_ = _parse_filter_(d.pop("filter", UNSET))
+
+        questions_per_seed = d.pop("questions_per_seed", UNSET)
+
+        include_default_filter = d.pop("include_default_filter", UNSET)
+
+        def _parse_answer_type(data: object) -> AnswerType | None | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                answer_type_type_0 = AnswerType.from_dict(data)
+
+                return answer_type_type_0
+            except (TypeError, ValueError, AttributeError, KeyError):
+                pass
+            return cast(AnswerType | None | Unset, data)
+
+        answer_type = _parse_answer_type(d.pop("answer_type", UNSET))
+
+        question_and_label_generator = cls(
+            config_type=config_type,
+            instructions=instructions,
+            examples=examples,
+            bad_examples=bad_examples,
+            filter_=filter_,
+            questions_per_seed=questions_per_seed,
+            include_default_filter=include_default_filter,
+            answer_type=answer_type,
+        )
+
+        question_and_label_generator.additional_properties = d
+        return question_and_label_generator
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

lightningrod/_generated/models/question_generator.py

@@ -0,0 +1,217 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Any, Literal, TypeVar, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+if TYPE_CHECKING:
+    from ..models.answer_type import AnswerType
+    from ..models.filter_criteria import FilterCriteria
+
+
+T = TypeVar("T", bound="QuestionGenerator")
+
+
+@_attrs_define
+class QuestionGenerator:
+    """
+    Attributes:
+        config_type (Literal['QUESTION_GENERATOR'] | Unset): Type of transform configuration Default:
+            'QUESTION_GENERATOR'.
+        instructions (None | str | Unset): Instructions for question generation. If not provided, uses sensible
+            defaults.
+        examples (list[str] | Unset): Example questions to guide generation
+        bad_examples (list[str] | Unset): Examples of questions to avoid
+        filter_ (FilterCriteria | list[FilterCriteria] | None | Unset): Optional filter criteria to apply after question
+            generation
+        questions_per_seed (int | Unset): Number of questions to generate per seed Default: 1.
+        include_default_filter (bool | Unset): Whether to include the default filter for generated questions Default:
+            False.
+        answer_type (AnswerType | None | Unset): The type of answer expected for generated questions
+    """
+
+    config_type: Literal["QUESTION_GENERATOR"] | Unset = "QUESTION_GENERATOR"
+    instructions: None | str | Unset = UNSET
+    examples: list[str] | Unset = UNSET
+    bad_examples: list[str] | Unset = UNSET
+    filter_: FilterCriteria | list[FilterCriteria] | None | Unset = UNSET
+    questions_per_seed: int | Unset = 1
+    include_default_filter: bool | Unset = False
+    answer_type: AnswerType | None | Unset = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        from ..models.answer_type import AnswerType
+        from ..models.filter_criteria import FilterCriteria
+
+        config_type = self.config_type
+
+        instructions: None | str | Unset
+        if isinstance(self.instructions, Unset):
+            instructions = UNSET
+        else:
+            instructions = self.instructions
+
+        examples: list[str] | Unset = UNSET
+        if not isinstance(self.examples, Unset):
+            examples = self.examples
+
+        bad_examples: list[str] | Unset = UNSET
+        if not isinstance(self.bad_examples, Unset):
+            bad_examples = self.bad_examples
+
+        filter_: dict[str, Any] | list[dict[str, Any]] | None | Unset
+        if isinstance(self.filter_, Unset):
+            filter_ = UNSET
+        elif isinstance(self.filter_, FilterCriteria):
+            filter_ = self.filter_.to_dict()
+        elif isinstance(self.filter_, list):
+            filter_ = []
+            for filter_type_1_item_data in self.filter_:
+                filter_type_1_item = filter_type_1_item_data.to_dict()
+                filter_.append(filter_type_1_item)
+
+        else:
+            filter_ = self.filter_
+
+        questions_per_seed = self.questions_per_seed
+
+        include_default_filter = self.include_default_filter
+
+        answer_type: dict[str, Any] | None | Unset
+        if isinstance(self.answer_type, Unset):
+            answer_type = UNSET
+        elif isinstance(self.answer_type, AnswerType):
+            answer_type = self.answer_type.to_dict()
+        else:
+            answer_type = self.answer_type
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if config_type is not UNSET:
+            field_dict["config_type"] = config_type
+        if instructions is not UNSET:
+            field_dict["instructions"] = instructions
+        if examples is not UNSET:
+            field_dict["examples"] = examples
+        if bad_examples is not UNSET:
+            field_dict["bad_examples"] = bad_examples
+        if filter_ is not UNSET:
+            field_dict["filter"] = filter_
+        if questions_per_seed is not UNSET:
+            field_dict["questions_per_seed"] = questions_per_seed
+        if include_default_filter is not UNSET:
+            field_dict["include_default_filter"] = include_default_filter
+        if answer_type is not UNSET:
+            field_dict["answer_type"] = answer_type
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.answer_type import AnswerType
+        from ..models.filter_criteria import FilterCriteria
+
+        d = dict(src_dict)
+        config_type = cast(Literal["QUESTION_GENERATOR"] | Unset, d.pop("config_type", UNSET))
+        if config_type != "QUESTION_GENERATOR" and not isinstance(config_type, Unset):
+            raise ValueError(f"config_type must match const 'QUESTION_GENERATOR', got '{config_type}'")
+
+        def _parse_instructions(data: object) -> None | str | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(None | str | Unset, data)
+
+        instructions = _parse_instructions(d.pop("instructions", UNSET))
+
+        examples = cast(list[str], d.pop("examples", UNSET))
+
+        bad_examples = cast(list[str], d.pop("bad_examples", UNSET))
+
+        def _parse_filter_(data: object) -> FilterCriteria | list[FilterCriteria] | None | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                filter_type_0 = FilterCriteria.from_dict(data)
+
+                return filter_type_0
+            except (TypeError, ValueError, AttributeError, KeyError):
+                pass
+            try:
+                if not isinstance(data, list):
+                    raise TypeError()
+                filter_type_1 = []
+                _filter_type_1 = data
+                for filter_type_1_item_data in _filter_type_1:
+                    filter_type_1_item = FilterCriteria.from_dict(filter_type_1_item_data)
+
+                    filter_type_1.append(filter_type_1_item)
+
+                return filter_type_1
+            except (TypeError, ValueError, AttributeError, KeyError):
+                pass
+            return cast(FilterCriteria | list[FilterCriteria] | None | Unset, data)
+
+        filter_ = _parse_filter_(d.pop("filter", UNSET))
+
+        questions_per_seed = d.pop("questions_per_seed", UNSET)
+
+        include_default_filter = d.pop("include_default_filter", UNSET)
+
+        def _parse_answer_type(data: object) -> AnswerType | None | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                answer_type_type_0 = AnswerType.from_dict(data)
+
+                return answer_type_type_0
+            except (TypeError, ValueError, AttributeError, KeyError):
+                pass
+            return cast(AnswerType | None | Unset, data)
+
+        answer_type = _parse_answer_type(d.pop("answer_type", UNSET))
+
+        question_generator = cls(
+            config_type=config_type,
+            instructions=instructions,
+            examples=examples,
+            bad_examples=bad_examples,
+            filter_=filter_,
+            questions_per_seed=questions_per_seed,
+            include_default_filter=include_default_filter,
+            answer_type=answer_type,
+        )
+
+        question_generator.additional_properties = d
+        return question_generator
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties