lightningrod-ai 0.1.6__py3-none-any.whl

lightningrod/_generated/models/seed.py

@@ -0,0 +1,135 @@
+from __future__ import annotations
+
+import datetime
+from collections.abc import Mapping
+from typing import Any, TypeVar, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+from dateutil.parser import isoparse
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="Seed")
+
+
+@_attrs_define
+class Seed:
+    """
+    Attributes:
+        seed_text (str):
+        url (None | str | Unset):
+        seed_creation_date (datetime.datetime | None | Unset):
+        search_query (None | str | Unset):
+    """
+
+    seed_text: str
+    url: None | str | Unset = UNSET
+    seed_creation_date: datetime.datetime | None | Unset = UNSET
+    search_query: None | str | Unset = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        seed_text = self.seed_text
+
+        url: None | str | Unset
+        if isinstance(self.url, Unset):
+            url = UNSET
+        else:
+            url = self.url
+
+        seed_creation_date: None | str | Unset
+        if isinstance(self.seed_creation_date, Unset):
+            seed_creation_date = UNSET
+        elif isinstance(self.seed_creation_date, datetime.datetime):
+            seed_creation_date = self.seed_creation_date.isoformat()
+        else:
+            seed_creation_date = self.seed_creation_date
+
+        search_query: None | str | Unset
+        if isinstance(self.search_query, Unset):
+            search_query = UNSET
+        else:
+            search_query = self.search_query
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "seed_text": seed_text,
+            }
+        )
+        if url is not UNSET:
+            field_dict["url"] = url
+        if seed_creation_date is not UNSET:
+            field_dict["seed_creation_date"] = seed_creation_date
+        if search_query is not UNSET:
+            field_dict["search_query"] = search_query
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        seed_text = d.pop("seed_text")
+
+        def _parse_url(data: object) -> None | str | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(None | str | Unset, data)
+
+        url = _parse_url(d.pop("url", UNSET))
+
+        def _parse_seed_creation_date(data: object) -> datetime.datetime | None | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, str):
+                    raise TypeError()
+                seed_creation_date_type_0 = isoparse(data)
+
+                return seed_creation_date_type_0
+            except (TypeError, ValueError, AttributeError, KeyError):
+                pass
+            return cast(datetime.datetime | None | Unset, data)
+
+        seed_creation_date = _parse_seed_creation_date(d.pop("seed_creation_date", UNSET))
+
+        def _parse_search_query(data: object) -> None | str | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(None | str | Unset, data)
+
+        search_query = _parse_search_query(d.pop("search_query", UNSET))
+
+        seed = cls(
+            seed_text=seed_text,
+            url=url,
+            seed_creation_date=seed_creation_date,
+            search_query=search_query,
+        )
+
+        seed.additional_properties = d
+        return seed
+
+    @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/step_cost_breakdown.py

@@ -0,0 +1,109 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..models.transform_type import TransformType
+
+if TYPE_CHECKING:
+    from ..models.usage_summary import UsageSummary
+
+
+T = TypeVar("T", bound="StepCostBreakdown")
+
+
+@_attrs_define
+class StepCostBreakdown:
+    """
+    Attributes:
+        step_name (str):
+        step_type (TransformType):
+        total_cost_dollars (float):
+        usage (UsageSummary): Flexible usage statistics by event type and LLM model.
+        output_count (int):
+        cost_per_output (float):
+    """
+
+    step_name: str
+    step_type: TransformType
+    total_cost_dollars: float
+    usage: UsageSummary
+    output_count: int
+    cost_per_output: float
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        step_name = self.step_name
+
+        step_type = self.step_type.value
+
+        total_cost_dollars = self.total_cost_dollars
+
+        usage = self.usage.to_dict()
+
+        output_count = self.output_count
+
+        cost_per_output = self.cost_per_output
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "step_name": step_name,
+                "step_type": step_type,
+                "total_cost_dollars": total_cost_dollars,
+                "usage": usage,
+                "output_count": output_count,
+                "cost_per_output": cost_per_output,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.usage_summary import UsageSummary
+
+        d = dict(src_dict)
+        step_name = d.pop("step_name")
+
+        step_type = TransformType(d.pop("step_type"))
+
+        total_cost_dollars = d.pop("total_cost_dollars")
+
+        usage = UsageSummary.from_dict(d.pop("usage"))
+
+        output_count = d.pop("output_count")
+
+        cost_per_output = d.pop("cost_per_output")
+
+        step_cost_breakdown = cls(
+            step_name=step_name,
+            step_type=step_type,
+            total_cost_dollars=total_cost_dollars,
+            usage=usage,
+            output_count=output_count,
+            cost_per_output=cost_per_output,
+        )
+
+        step_cost_breakdown.additional_properties = d
+        return step_cost_breakdown
+
+    @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/transform_job.py

@@ -0,0 +1,268 @@
+from __future__ import annotations
+
+import datetime
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Any, TypeVar, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+from dateutil.parser import isoparse
+
+from ..models.transform_job_status import TransformJobStatus
+from ..types import UNSET, Unset
+
+if TYPE_CHECKING:
+    from ..models.job_usage import JobUsage
+
+
+T = TypeVar("T", bound="TransformJob")
+
+
+@_attrs_define
+class TransformJob:
+    """
+    Attributes:
+        id (str):
+        organization_id (str):
+        status (TransformJobStatus):
+        modal_function_call_id (str):
+        modal_app_id (str):
+        transform_config (str):
+        input_dataset_id (None | str):
+        output_dataset_id (None | str):
+        created_at (datetime.datetime):
+        updated_at (datetime.datetime):
+        configuration_id (None | str | Unset):
+        error_message (None | str | Unset):
+        warning_message (None | str | Unset):
+        usage (JobUsage | None | Unset):
+        estimated_cost_dollars (float | None | Unset):
+    """
+
+    id: str
+    organization_id: str
+    status: TransformJobStatus
+    modal_function_call_id: str
+    modal_app_id: str
+    transform_config: str
+    input_dataset_id: None | str
+    output_dataset_id: None | str
+    created_at: datetime.datetime
+    updated_at: datetime.datetime
+    configuration_id: None | str | Unset = UNSET
+    error_message: None | str | Unset = UNSET
+    warning_message: None | str | Unset = UNSET
+    usage: JobUsage | None | Unset = UNSET
+    estimated_cost_dollars: float | None | Unset = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        from ..models.job_usage import JobUsage
+
+        id = self.id
+
+        organization_id = self.organization_id
+
+        status = self.status.value
+
+        modal_function_call_id = self.modal_function_call_id
+
+        modal_app_id = self.modal_app_id
+
+        transform_config = self.transform_config
+
+        input_dataset_id: None | str
+        input_dataset_id = self.input_dataset_id
+
+        output_dataset_id: None | str
+        output_dataset_id = self.output_dataset_id
+
+        created_at = self.created_at.isoformat()
+
+        updated_at = self.updated_at.isoformat()
+
+        configuration_id: None | str | Unset
+        if isinstance(self.configuration_id, Unset):
+            configuration_id = UNSET
+        else:
+            configuration_id = self.configuration_id
+
+        error_message: None | str | Unset
+        if isinstance(self.error_message, Unset):
+            error_message = UNSET
+        else:
+            error_message = self.error_message
+
+        warning_message: None | str | Unset
+        if isinstance(self.warning_message, Unset):
+            warning_message = UNSET
+        else:
+            warning_message = self.warning_message
+
+        usage: dict[str, Any] | None | Unset
+        if isinstance(self.usage, Unset):
+            usage = UNSET
+        elif isinstance(self.usage, JobUsage):
+            usage = self.usage.to_dict()
+        else:
+            usage = self.usage
+
+        estimated_cost_dollars: float | None | Unset
+        if isinstance(self.estimated_cost_dollars, Unset):
+            estimated_cost_dollars = UNSET
+        else:
+            estimated_cost_dollars = self.estimated_cost_dollars
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "id": id,
+                "organization_id": organization_id,
+                "status": status,
+                "modal_function_call_id": modal_function_call_id,
+                "modal_app_id": modal_app_id,
+                "transform_config": transform_config,
+                "input_dataset_id": input_dataset_id,
+                "output_dataset_id": output_dataset_id,
+                "created_at": created_at,
+                "updated_at": updated_at,
+            }
+        )
+        if configuration_id is not UNSET:
+            field_dict["configuration_id"] = configuration_id
+        if error_message is not UNSET:
+            field_dict["error_message"] = error_message
+        if warning_message is not UNSET:
+            field_dict["warning_message"] = warning_message
+        if usage is not UNSET:
+            field_dict["usage"] = usage
+        if estimated_cost_dollars is not UNSET:
+            field_dict["estimated_cost_dollars"] = estimated_cost_dollars
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.job_usage import JobUsage
+
+        d = dict(src_dict)
+        id = d.pop("id")
+
+        organization_id = d.pop("organization_id")
+
+        status = TransformJobStatus(d.pop("status"))
+
+        modal_function_call_id = d.pop("modal_function_call_id")
+
+        modal_app_id = d.pop("modal_app_id")
+
+        transform_config = d.pop("transform_config")
+
+        def _parse_input_dataset_id(data: object) -> None | str:
+            if data is None:
+                return data
+            return cast(None | str, data)
+
+        input_dataset_id = _parse_input_dataset_id(d.pop("input_dataset_id"))
+
+        def _parse_output_dataset_id(data: object) -> None | str:
+            if data is None:
+                return data
+            return cast(None | str, data)
+
+        output_dataset_id = _parse_output_dataset_id(d.pop("output_dataset_id"))
+
+        created_at = isoparse(d.pop("created_at"))
+
+        updated_at = isoparse(d.pop("updated_at"))
+
+        def _parse_configuration_id(data: object) -> None | str | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(None | str | Unset, data)
+
+        configuration_id = _parse_configuration_id(d.pop("configuration_id", UNSET))
+
+        def _parse_error_message(data: object) -> None | str | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(None | str | Unset, data)
+
+        error_message = _parse_error_message(d.pop("error_message", UNSET))
+
+        def _parse_warning_message(data: object) -> None | str | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(None | str | Unset, data)
+
+        warning_message = _parse_warning_message(d.pop("warning_message", UNSET))
+
+        def _parse_usage(data: object) -> JobUsage | None | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                usage_type_0 = JobUsage.from_dict(data)
+
+                return usage_type_0
+            except (TypeError, ValueError, AttributeError, KeyError):
+                pass
+            return cast(JobUsage | None | Unset, data)
+
+        usage = _parse_usage(d.pop("usage", UNSET))
+
+        def _parse_estimated_cost_dollars(data: object) -> float | None | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(float | None | Unset, data)
+
+        estimated_cost_dollars = _parse_estimated_cost_dollars(d.pop("estimated_cost_dollars", UNSET))
+
+        transform_job = cls(
+            id=id,
+            organization_id=organization_id,
+            status=status,
+            modal_function_call_id=modal_function_call_id,
+            modal_app_id=modal_app_id,
+            transform_config=transform_config,
+            input_dataset_id=input_dataset_id,
+            output_dataset_id=output_dataset_id,
+            created_at=created_at,
+            updated_at=updated_at,
+            configuration_id=configuration_id,
+            error_message=error_message,
+            warning_message=warning_message,
+            usage=usage,
+            estimated_cost_dollars=estimated_cost_dollars,
+        )
+
+        transform_job.additional_properties = d
+        return transform_job
+
+    @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/transform_job_status.py

@@ -0,0 +1,11 @@
+from enum import Enum
+
+
+class TransformJobStatus(str, Enum):
+    CANCELLED = "CANCELLED"
+    COMPLETED = "COMPLETED"
+    FAILED = "FAILED"
+    RUNNING = "RUNNING"
+
+    def __str__(self) -> str:
+        return str(self.value)

lightningrod/_generated/models/transform_step_metrics_response.py

@@ -0,0 +1,131 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import Any, TypeVar, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="TransformStepMetricsResponse")
+
+
+@_attrs_define
+class TransformStepMetricsResponse:
+    """
+    Attributes:
+        step_index (int):
+        transform_name (str):
+        input_rows (int):
+        output_rows (int):
+        rejected_count (int):
+        error_count (int):
+        duration_seconds (float):
+        progress (float):
+        summary (None | str):
+    """
+
+    step_index: int
+    transform_name: str
+    input_rows: int
+    output_rows: int
+    rejected_count: int
+    error_count: int
+    duration_seconds: float
+    progress: float
+    summary: None | str
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        step_index = self.step_index
+
+        transform_name = self.transform_name
+
+        input_rows = self.input_rows
+
+        output_rows = self.output_rows
+
+        rejected_count = self.rejected_count
+
+        error_count = self.error_count
+
+        duration_seconds = self.duration_seconds
+
+        progress = self.progress
+
+        summary: None | str
+        summary = self.summary
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "step_index": step_index,
+                "transform_name": transform_name,
+                "input_rows": input_rows,
+                "output_rows": output_rows,
+                "rejected_count": rejected_count,
+                "error_count": error_count,
+                "duration_seconds": duration_seconds,
+                "progress": progress,
+                "summary": summary,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        step_index = d.pop("step_index")
+
+        transform_name = d.pop("transform_name")
+
+        input_rows = d.pop("input_rows")
+
+        output_rows = d.pop("output_rows")
+
+        rejected_count = d.pop("rejected_count")
+
+        error_count = d.pop("error_count")
+
+        duration_seconds = d.pop("duration_seconds")
+
+        progress = d.pop("progress")
+
+        def _parse_summary(data: object) -> None | str:
+            if data is None:
+                return data
+            return cast(None | str, data)
+
+        summary = _parse_summary(d.pop("summary"))
+
+        transform_step_metrics_response = cls(
+            step_index=step_index,
+            transform_name=transform_name,
+            input_rows=input_rows,
+            output_rows=output_rows,
+            rejected_count=rejected_count,
+            error_count=error_count,
+            duration_seconds=duration_seconds,
+            progress=progress,
+            summary=summary,
+        )
+
+        transform_step_metrics_response.additional_properties = d
+        return transform_step_metrics_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

lightningrod/_generated/models/transform_type.py

@@ -0,0 +1,25 @@
+from enum import Enum
+
+
+class TransformType(str, Enum):
+    FILESET_QUERY_SEED_GENERATOR = "FILESET_QUERY_SEED_GENERATOR"
+    FILESET_SEED_GENERATOR = "FILESET_SEED_GENERATOR"
+    FILTER_CRITERIA = "FILTER_CRITERIA"
+    FORWARD_LOOKING_QUESTION_GENERATOR = "FORWARD_LOOKING_QUESTION_GENERATOR"
+    GDELT_SEED_GENERATOR = "GDELT_SEED_GENERATOR"
+    MOCK = "MOCK"
+    NEWS_CONTEXT_GENERATOR = "NEWS_CONTEXT_GENERATOR"
+    NEWS_SEED_GENERATOR = "NEWS_SEED_GENERATOR"
+    PIPELINE = "PIPELINE"
+    QUESTION_AND_LABEL_GENERATOR = "QUESTION_AND_LABEL_GENERATOR"
+    QUESTION_FILTER = "QUESTION_FILTER"
+    QUESTION_GENERATOR = "QUESTION_GENERATOR"
+    QUESTION_PIPELINE = "QUESTION_PIPELINE"
+    QUESTION_QUALITY_SCORER = "QUESTION_QUALITY_SCORER"
+    QUESTION_RENDERER = "QUESTION_RENDERER"
+    ROLLOUT_GENERATOR = "ROLLOUT_GENERATOR"
+    SCORE_FILTER = "SCORE_FILTER"
+    WEB_SEARCH_LABELER = "WEB_SEARCH_LABELER"
+
+    def __str__(self) -> str:
+        return str(self.value)