everyrow 0.1.9__py3-none-any.whl → 0.2.0__py3-none-any.whl

everyrow/generated/models/task_metadata.py

@@ -1,323 +0,0 @@
-from __future__ import annotations
-
-from collections.abc import Mapping
-from typing import TYPE_CHECKING, Any, TypeVar, cast
-from uuid import UUID
-
-from attrs import define as _attrs_define
-from attrs import field as _attrs_field
-
-from ..types import UNSET, Unset
-
-if TYPE_CHECKING:
-    from ..models.task_metadata_cols_to_rename_type_0 import TaskMetadataColsToRenameType0
-
-
-T = TypeVar("T", bound="TaskMetadata")
-
-
-@_attrs_define
-class TaskMetadata:
-    """
-    Attributes:
-        llm (None | str | Unset):
-        n_subtask_total (int | None | Unset):
-        n_subtask_successes (int | None | Unset):
-        n_subtask_failures (int | None | Unset):
-        trace_url (None | str | Unset):
-        otel_trace_url (None | str | Unset):
-        langfuse_trace_url (None | str | Unset):
-        duplicate_from_artifact_id (None | Unset | UUID):
-        duplicate_from_task_id (None | Unset | UUID):
-        cols_to_rename (None | TaskMetadataColsToRenameType0 | Unset):
-        cols_to_drop (list[Any] | None | Unset):
-    """
-
-    llm: None | str | Unset = UNSET
-    n_subtask_total: int | None | Unset = UNSET
-    n_subtask_successes: int | None | Unset = UNSET
-    n_subtask_failures: int | None | Unset = UNSET
-    trace_url: None | str | Unset = UNSET
-    otel_trace_url: None | str | Unset = UNSET
-    langfuse_trace_url: None | str | Unset = UNSET
-    duplicate_from_artifact_id: None | Unset | UUID = UNSET
-    duplicate_from_task_id: None | Unset | UUID = UNSET
-    cols_to_rename: None | TaskMetadataColsToRenameType0 | Unset = UNSET
-    cols_to_drop: list[Any] | None | Unset = UNSET
-    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> dict[str, Any]:
-        from ..models.task_metadata_cols_to_rename_type_0 import TaskMetadataColsToRenameType0
-
-        llm: None | str | Unset
-        if isinstance(self.llm, Unset):
-            llm = UNSET
-        else:
-            llm = self.llm
-
-        n_subtask_total: int | None | Unset
-        if isinstance(self.n_subtask_total, Unset):
-            n_subtask_total = UNSET
-        else:
-            n_subtask_total = self.n_subtask_total
-
-        n_subtask_successes: int | None | Unset
-        if isinstance(self.n_subtask_successes, Unset):
-            n_subtask_successes = UNSET
-        else:
-            n_subtask_successes = self.n_subtask_successes
-
-        n_subtask_failures: int | None | Unset
-        if isinstance(self.n_subtask_failures, Unset):
-            n_subtask_failures = UNSET
-        else:
-            n_subtask_failures = self.n_subtask_failures
-
-        trace_url: None | str | Unset
-        if isinstance(self.trace_url, Unset):
-            trace_url = UNSET
-        else:
-            trace_url = self.trace_url
-
-        otel_trace_url: None | str | Unset
-        if isinstance(self.otel_trace_url, Unset):
-            otel_trace_url = UNSET
-        else:
-            otel_trace_url = self.otel_trace_url
-
-        langfuse_trace_url: None | str | Unset
-        if isinstance(self.langfuse_trace_url, Unset):
-            langfuse_trace_url = UNSET
-        else:
-            langfuse_trace_url = self.langfuse_trace_url
-
-        duplicate_from_artifact_id: None | str | Unset
-        if isinstance(self.duplicate_from_artifact_id, Unset):
-            duplicate_from_artifact_id = UNSET
-        elif isinstance(self.duplicate_from_artifact_id, UUID):
-            duplicate_from_artifact_id = str(self.duplicate_from_artifact_id)
-        else:
-            duplicate_from_artifact_id = self.duplicate_from_artifact_id
-
-        duplicate_from_task_id: None | str | Unset
-        if isinstance(self.duplicate_from_task_id, Unset):
-            duplicate_from_task_id = UNSET
-        elif isinstance(self.duplicate_from_task_id, UUID):
-            duplicate_from_task_id = str(self.duplicate_from_task_id)
-        else:
-            duplicate_from_task_id = self.duplicate_from_task_id
-
-        cols_to_rename: dict[str, Any] | None | Unset
-        if isinstance(self.cols_to_rename, Unset):
-            cols_to_rename = UNSET
-        elif isinstance(self.cols_to_rename, TaskMetadataColsToRenameType0):
-            cols_to_rename = self.cols_to_rename.to_dict()
-        else:
-            cols_to_rename = self.cols_to_rename
-
-        cols_to_drop: list[Any] | None | Unset
-        if isinstance(self.cols_to_drop, Unset):
-            cols_to_drop = UNSET
-        elif isinstance(self.cols_to_drop, list):
-            cols_to_drop = self.cols_to_drop
-
-        else:
-            cols_to_drop = self.cols_to_drop
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update({})
-        if llm is not UNSET:
-            field_dict["llm"] = llm
-        if n_subtask_total is not UNSET:
-            field_dict["n_subtask_total"] = n_subtask_total
-        if n_subtask_successes is not UNSET:
-            field_dict["n_subtask_successes"] = n_subtask_successes
-        if n_subtask_failures is not UNSET:
-            field_dict["n_subtask_failures"] = n_subtask_failures
-        if trace_url is not UNSET:
-            field_dict["trace_url"] = trace_url
-        if otel_trace_url is not UNSET:
-            field_dict["otel_trace_url"] = otel_trace_url
-        if langfuse_trace_url is not UNSET:
-            field_dict["langfuse_trace_url"] = langfuse_trace_url
-        if duplicate_from_artifact_id is not UNSET:
-            field_dict["duplicate_from_artifact_id"] = duplicate_from_artifact_id
-        if duplicate_from_task_id is not UNSET:
-            field_dict["duplicate_from_task_id"] = duplicate_from_task_id
-        if cols_to_rename is not UNSET:
-            field_dict["cols_to_rename"] = cols_to_rename
-        if cols_to_drop is not UNSET:
-            field_dict["cols_to_drop"] = cols_to_drop
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
-        from ..models.task_metadata_cols_to_rename_type_0 import TaskMetadataColsToRenameType0
-
-        d = dict(src_dict)
-
-        def _parse_llm(data: object) -> None | str | Unset:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            return cast(None | str | Unset, data)
-
-        llm = _parse_llm(d.pop("llm", UNSET))
-
-        def _parse_n_subtask_total(data: object) -> int | None | Unset:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            return cast(int | None | Unset, data)
-
-        n_subtask_total = _parse_n_subtask_total(d.pop("n_subtask_total", UNSET))
-
-        def _parse_n_subtask_successes(data: object) -> int | None | Unset:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            return cast(int | None | Unset, data)
-
-        n_subtask_successes = _parse_n_subtask_successes(d.pop("n_subtask_successes", UNSET))
-
-        def _parse_n_subtask_failures(data: object) -> int | None | Unset:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            return cast(int | None | Unset, data)
-
-        n_subtask_failures = _parse_n_subtask_failures(d.pop("n_subtask_failures", UNSET))
-
-        def _parse_trace_url(data: object) -> None | str | Unset:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            return cast(None | str | Unset, data)
-
-        trace_url = _parse_trace_url(d.pop("trace_url", UNSET))
-
-        def _parse_otel_trace_url(data: object) -> None | str | Unset:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            return cast(None | str | Unset, data)
-
-        otel_trace_url = _parse_otel_trace_url(d.pop("otel_trace_url", UNSET))
-
-        def _parse_langfuse_trace_url(data: object) -> None | str | Unset:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            return cast(None | str | Unset, data)
-
-        langfuse_trace_url = _parse_langfuse_trace_url(d.pop("langfuse_trace_url", UNSET))
-
-        def _parse_duplicate_from_artifact_id(data: object) -> None | Unset | UUID:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            try:
-                if not isinstance(data, str):
-                    raise TypeError()
-                duplicate_from_artifact_id_type_0 = UUID(data)
-
-                return duplicate_from_artifact_id_type_0
-            except (TypeError, ValueError, AttributeError, KeyError):
-                pass
-            return cast(None | Unset | UUID, data)
-
-        duplicate_from_artifact_id = _parse_duplicate_from_artifact_id(d.pop("duplicate_from_artifact_id", UNSET))
-
-        def _parse_duplicate_from_task_id(data: object) -> None | Unset | UUID:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            try:
-                if not isinstance(data, str):
-                    raise TypeError()
-                duplicate_from_task_id_type_0 = UUID(data)
-
-                return duplicate_from_task_id_type_0
-            except (TypeError, ValueError, AttributeError, KeyError):
-                pass
-            return cast(None | Unset | UUID, data)
-
-        duplicate_from_task_id = _parse_duplicate_from_task_id(d.pop("duplicate_from_task_id", UNSET))
-
-        def _parse_cols_to_rename(data: object) -> None | TaskMetadataColsToRenameType0 | Unset:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            try:
-                if not isinstance(data, dict):
-                    raise TypeError()
-                cols_to_rename_type_0 = TaskMetadataColsToRenameType0.from_dict(data)
-
-                return cols_to_rename_type_0
-            except (TypeError, ValueError, AttributeError, KeyError):
-                pass
-            return cast(None | TaskMetadataColsToRenameType0 | Unset, data)
-
-        cols_to_rename = _parse_cols_to_rename(d.pop("cols_to_rename", UNSET))
-
-        def _parse_cols_to_drop(data: object) -> list[Any] | None | Unset:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            try:
-                if not isinstance(data, list):
-                    raise TypeError()
-                cols_to_drop_type_0 = cast(list[Any], data)
-
-                return cols_to_drop_type_0
-            except (TypeError, ValueError, AttributeError, KeyError):
-                pass
-            return cast(list[Any] | None | Unset, data)
-
-        cols_to_drop = _parse_cols_to_drop(d.pop("cols_to_drop", UNSET))
-
-        task_metadata = cls(
-            llm=llm,
-            n_subtask_total=n_subtask_total,
-            n_subtask_successes=n_subtask_successes,
-            n_subtask_failures=n_subtask_failures,
-            trace_url=trace_url,
-            otel_trace_url=otel_trace_url,
-            langfuse_trace_url=langfuse_trace_url,
-            duplicate_from_artifact_id=duplicate_from_artifact_id,
-            duplicate_from_task_id=duplicate_from_task_id,
-            cols_to_rename=cols_to_rename,
-            cols_to_drop=cols_to_drop,
-        )
-
-        task_metadata.additional_properties = d
-        return task_metadata
-
-    @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

everyrow/generated/models/task_response.py

@@ -1,62 +0,0 @@
-from __future__ import annotations
-
-from collections.abc import Mapping
-from typing import Any, TypeVar
-from uuid import UUID
-
-from attrs import define as _attrs_define
-from attrs import field as _attrs_field
-
-T = TypeVar("T", bound="TaskResponse")
-
-
-@_attrs_define
-class TaskResponse:
-    """
-    Attributes:
-        task_id (UUID):
-    """
-
-    task_id: UUID
-    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> dict[str, Any]:
-        task_id = str(self.task_id)
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update(
-            {
-                "task_id": task_id,
-            }
-        )
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
-        d = dict(src_dict)
-        task_id = UUID(d.pop("task_id"))
-
-        task_response = cls(
-            task_id=task_id,
-        )
-
-        task_response.additional_properties = d
-        return task_response
-
-    @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

everyrow/generated/models/task_type.py

@@ -1,31 +0,0 @@
-from enum import Enum
-
-
-class TaskType(str, Enum):
-    AGENT = "agent"
-    CONCATENATE = "concatenate"
-    CREATE = "create"
-    CREATE_GROUP = "create_group"
-    DEDUPE = "dedupe"
-    DEEP_MERGE = "deep_merge"
-    DEEP_RANK = "deep_rank"
-    DEEP_SCREEN = "deep_screen"
-    DERIVE = "derive"
-    DROP_COLUMNS = "drop_columns"
-    FILTER = "filter"
-    FIND_NUMBER = "find_number"
-    FIND_SOURCES = "find_sources"
-    FLATTEN = "flatten"
-    GET_NUMBER = "get_number"
-    GROUP_BY = "group_by"
-    IMPORT_SHEET = "import_sheet"
-    JOIN = "join"
-    MULTIAGENT = "multiagent"
-    POPULATE_REFERENCE_CLASS = "populate_reference_class"
-    REVISE = "revise"
-    UPLOAD_CSV = "upload_csv"
-    VALIDATE_CLAIM = "validate_claim"
-    VALUE_23 = "_provide_workflow_inputs"
-
-    def __str__(self) -> str:
-        return str(self.value)

everyrow/generated/models/text_chat_content_part.py

@@ -1,74 +0,0 @@
-from __future__ import annotations
-
-from collections.abc import Mapping
-from typing import Any, Literal, TypeVar, cast
-
-from attrs import define as _attrs_define
-from attrs import field as _attrs_field
-
-from ..types import UNSET, Unset
-
-T = TypeVar("T", bound="TextChatContentPart")
-
-
-@_attrs_define
-class TextChatContentPart:
-    """
-    Attributes:
-        text (str):
-        type_ (Literal['text'] | Unset):  Default: 'text'.
-    """
-
-    text: str
-    type_: Literal["text"] | Unset = "text"
-    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> dict[str, Any]:
-        text = self.text
-
-        type_ = self.type_
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update(
-            {
-                "text": text,
-            }
-        )
-        if type_ is not UNSET:
-            field_dict["type"] = type_
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
-        d = dict(src_dict)
-        text = d.pop("text")
-
-        type_ = cast(Literal["text"] | Unset, d.pop("type", UNSET))
-        if type_ != "text" and not isinstance(type_, Unset):
-            raise ValueError(f"type must match const 'text', got '{type_}'")
-
-        text_chat_content_part = cls(
-            text=text,
-            type_=type_,
-        )
-
-        text_chat_content_part.additional_properties = d
-        return text_chat_content_part
-
-    @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

everyrow/generated/models/tool_response_message.py

@@ -1,127 +0,0 @@
-from __future__ import annotations
-
-from collections.abc import Mapping
-from typing import Any, Literal, TypeVar, cast
-
-from attrs import define as _attrs_define
-from attrs import field as _attrs_field
-
-from ..types import UNSET, Unset
-
-T = TypeVar("T", bound="ToolResponseMessage")
-
-
-@_attrs_define
-class ToolResponseMessage:
-    """
-    Attributes:
-        tool_call_id (str):
-        content (str):
-        description (str):
-        tool_name (str):
-        message_type (Literal['tool_response'] | Unset):  Default: 'tool_response'.
-        role (Literal['tool'] | Unset):  Default: 'tool'.
-        cacheable (bool | Unset):  Default: False.
-        explicit_cache_breakpoint (bool | Unset):  Default: False.
-    """
-
-    tool_call_id: str
-    content: str
-    description: str
-    tool_name: str
-    message_type: Literal["tool_response"] | Unset = "tool_response"
-    role: Literal["tool"] | Unset = "tool"
-    cacheable: bool | Unset = False
-    explicit_cache_breakpoint: bool | Unset = False
-    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> dict[str, Any]:
-        tool_call_id = self.tool_call_id
-
-        content = self.content
-
-        description = self.description
-
-        tool_name = self.tool_name
-
-        message_type = self.message_type
-
-        role = self.role
-
-        cacheable = self.cacheable
-
-        explicit_cache_breakpoint = self.explicit_cache_breakpoint
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update(
-            {
-                "tool_call_id": tool_call_id,
-                "content": content,
-                "description": description,
-                "tool_name": tool_name,
-            }
-        )
-        if message_type is not UNSET:
-            field_dict["message_type"] = message_type
-        if role is not UNSET:
-            field_dict["role"] = role
-        if cacheable is not UNSET:
-            field_dict["cacheable"] = cacheable
-        if explicit_cache_breakpoint is not UNSET:
-            field_dict["explicit_cache_breakpoint"] = explicit_cache_breakpoint
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
-        d = dict(src_dict)
-        tool_call_id = d.pop("tool_call_id")
-
-        content = d.pop("content")
-
-        description = d.pop("description")
-
-        tool_name = d.pop("tool_name")
-
-        message_type = cast(Literal["tool_response"] | Unset, d.pop("message_type", UNSET))
-        if message_type != "tool_response" and not isinstance(message_type, Unset):
-            raise ValueError(f"message_type must match const 'tool_response', got '{message_type}'")
-
-        role = cast(Literal["tool"] | Unset, d.pop("role", UNSET))
-        if role != "tool" and not isinstance(role, Unset):
-            raise ValueError(f"role must match const 'tool', got '{role}'")
-
-        cacheable = d.pop("cacheable", UNSET)
-
-        explicit_cache_breakpoint = d.pop("explicit_cache_breakpoint", UNSET)
-
-        tool_response_message = cls(
-            tool_call_id=tool_call_id,
-            content=content,
-            description=description,
-            tool_name=tool_name,
-            message_type=message_type,
-            role=role,
-            cacheable=cacheable,
-            explicit_cache_breakpoint=explicit_cache_breakpoint,
-        )
-
-        tool_response_message.additional_properties = d
-        return tool_response_message
-
-    @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

everyrow/generated/models/toolkit_constants.py

@@ -1,80 +0,0 @@
-from __future__ import annotations
-
-from collections.abc import Mapping
-from typing import Any, Literal, TypeVar, cast
-
-from attrs import define as _attrs_define
-from attrs import field as _attrs_field
-
-from ..types import UNSET, Unset
-
-T = TypeVar("T", bound="ToolkitConstants")
-
-
-@_attrs_define
-class ToolkitConstants:
-    """
-    Attributes:
-        low_effort_agent_iteration_budget (Literal[5] | Unset):  Default: 5.
-        high_effort_agent_iteration_budget (Literal[10] | Unset):  Default: 10.
-    """
-
-    low_effort_agent_iteration_budget: Literal[5] | Unset = 5
-    high_effort_agent_iteration_budget: Literal[10] | Unset = 10
-    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> dict[str, Any]:
-        low_effort_agent_iteration_budget = self.low_effort_agent_iteration_budget
-
-        high_effort_agent_iteration_budget = self.high_effort_agent_iteration_budget
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update({})
-        if low_effort_agent_iteration_budget is not UNSET:
-            field_dict["low_effort_agent_iteration_budget"] = low_effort_agent_iteration_budget
-        if high_effort_agent_iteration_budget is not UNSET:
-            field_dict["high_effort_agent_iteration_budget"] = high_effort_agent_iteration_budget
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
-        d = dict(src_dict)
-        low_effort_agent_iteration_budget = cast(Literal[5] | Unset, d.pop("low_effort_agent_iteration_budget", UNSET))
-        if low_effort_agent_iteration_budget != 5 and not isinstance(low_effort_agent_iteration_budget, Unset):
-            raise ValueError(
-                f"low_effort_agent_iteration_budget must match const 5, got '{low_effort_agent_iteration_budget}'"
-            )
-
-        high_effort_agent_iteration_budget = cast(
-            Literal[10] | Unset, d.pop("high_effort_agent_iteration_budget", UNSET)
-        )
-        if high_effort_agent_iteration_budget != 10 and not isinstance(high_effort_agent_iteration_budget, Unset):
-            raise ValueError(
-                f"high_effort_agent_iteration_budget must match const 10, got '{high_effort_agent_iteration_budget}'"
-            )
-
-        toolkit_constants = cls(
-            low_effort_agent_iteration_budget=low_effort_agent_iteration_budget,
-            high_effort_agent_iteration_budget=high_effort_agent_iteration_budget,
-        )
-
-        toolkit_constants.additional_properties = d
-        return toolkit_constants
-
-    @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