lightningrod-ai 0.1.6__py3-none-any.whl

lightningrod/_generated/models/estimate_cost_response.py

@@ -0,0 +1,117 @@
+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
+
+if TYPE_CHECKING:
+    from ..models.step_cost_breakdown import StepCostBreakdown
+    from ..models.usage_summary import UsageSummary
+
+
+T = TypeVar("T", bound="EstimateCostResponse")
+
+
+@_attrs_define
+class EstimateCostResponse:
+    """
+    Attributes:
+        total_cost_dollars (float):
+        llm_cost_dollars (float):
+        web_search_cost_dollars (float):
+        url_download_cost_dollars (float):
+        usage (UsageSummary): Flexible usage statistics by event type and LLM model.
+        steps (list[StepCostBreakdown]):
+    """
+
+    total_cost_dollars: float
+    llm_cost_dollars: float
+    web_search_cost_dollars: float
+    url_download_cost_dollars: float
+    usage: UsageSummary
+    steps: list[StepCostBreakdown]
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        total_cost_dollars = self.total_cost_dollars
+
+        llm_cost_dollars = self.llm_cost_dollars
+
+        web_search_cost_dollars = self.web_search_cost_dollars
+
+        url_download_cost_dollars = self.url_download_cost_dollars
+
+        usage = self.usage.to_dict()
+
+        steps = []
+        for steps_item_data in self.steps:
+            steps_item = steps_item_data.to_dict()
+            steps.append(steps_item)
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "total_cost_dollars": total_cost_dollars,
+                "llm_cost_dollars": llm_cost_dollars,
+                "web_search_cost_dollars": web_search_cost_dollars,
+                "url_download_cost_dollars": url_download_cost_dollars,
+                "usage": usage,
+                "steps": steps,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.step_cost_breakdown import StepCostBreakdown
+        from ..models.usage_summary import UsageSummary
+
+        d = dict(src_dict)
+        total_cost_dollars = d.pop("total_cost_dollars")
+
+        llm_cost_dollars = d.pop("llm_cost_dollars")
+
+        web_search_cost_dollars = d.pop("web_search_cost_dollars")
+
+        url_download_cost_dollars = d.pop("url_download_cost_dollars")
+
+        usage = UsageSummary.from_dict(d.pop("usage"))
+
+        steps = []
+        _steps = d.pop("steps")
+        for steps_item_data in _steps:
+            steps_item = StepCostBreakdown.from_dict(steps_item_data)
+
+            steps.append(steps_item)
+
+        estimate_cost_response = cls(
+            total_cost_dollars=total_cost_dollars,
+            llm_cost_dollars=llm_cost_dollars,
+            web_search_cost_dollars=web_search_cost_dollars,
+            url_download_cost_dollars=url_download_cost_dollars,
+            usage=usage,
+            steps=steps,
+        )
+
+        estimate_cost_response.additional_properties = d
+        return estimate_cost_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/event_usage_summary.py

@@ -0,0 +1,80 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="EventUsageSummary")
+
+
+@_attrs_define
+class EventUsageSummary:
+    """Usage for a single event type.
+
+    Attributes:
+        count (int | Unset):  Default: 0.
+        cost (float | Unset):  Default: 0.0.
+        cached_count (int | Unset):  Default: 0.
+    """
+
+    count: int | Unset = 0
+    cost: float | Unset = 0.0
+    cached_count: int | Unset = 0
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        count = self.count
+
+        cost = self.cost
+
+        cached_count = self.cached_count
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if count is not UNSET:
+            field_dict["count"] = count
+        if cost is not UNSET:
+            field_dict["cost"] = cost
+        if cached_count is not UNSET:
+            field_dict["cached_count"] = cached_count
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        count = d.pop("count", UNSET)
+
+        cost = d.pop("cost", UNSET)
+
+        cached_count = d.pop("cached_count", UNSET)
+
+        event_usage_summary = cls(
+            count=count,
+            cost=cost,
+            cached_count=cached_count,
+        )
+
+        event_usage_summary.additional_properties = d
+        return event_usage_summary
+
+    @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/file_set.py

@@ -0,0 +1,128 @@
+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="FileSet")
+
+
+@_attrs_define
+class FileSet:
+    """
+    Attributes:
+        id (str):
+        name (str):
+        description (None | str):
+        file_count (int):
+        indexed_file_count (int):
+        created_at (datetime.datetime):
+        updated_at (datetime.datetime):
+        is_public (bool | Unset):  Default: False.
+    """
+
+    id: str
+    name: str
+    description: None | str
+    file_count: int
+    indexed_file_count: int
+    created_at: datetime.datetime
+    updated_at: datetime.datetime
+    is_public: bool | Unset = False
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        id = self.id
+
+        name = self.name
+
+        description: None | str
+        description = self.description
+
+        file_count = self.file_count
+
+        indexed_file_count = self.indexed_file_count
+
+        created_at = self.created_at.isoformat()
+
+        updated_at = self.updated_at.isoformat()
+
+        is_public = self.is_public
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "id": id,
+                "name": name,
+                "description": description,
+                "file_count": file_count,
+                "indexed_file_count": indexed_file_count,
+                "created_at": created_at,
+                "updated_at": updated_at,
+            }
+        )
+        if is_public is not UNSET:
+            field_dict["is_public"] = is_public
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        id = d.pop("id")
+
+        name = d.pop("name")
+
+        def _parse_description(data: object) -> None | str:
+            if data is None:
+                return data
+            return cast(None | str, data)
+
+        description = _parse_description(d.pop("description"))
+
+        file_count = d.pop("file_count")
+
+        indexed_file_count = d.pop("indexed_file_count")
+
+        created_at = isoparse(d.pop("created_at"))
+
+        updated_at = isoparse(d.pop("updated_at"))
+
+        is_public = d.pop("is_public", UNSET)
+
+        file_set = cls(
+            id=id,
+            name=name,
+            description=description,
+            file_count=file_count,
+            indexed_file_count=indexed_file_count,
+            created_at=created_at,
+            updated_at=updated_at,
+            is_public=is_public,
+        )
+
+        file_set.additional_properties = d
+        return file_set
+
+    @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/file_set_file.py

@@ -0,0 +1,203 @@
+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
+
+if TYPE_CHECKING:
+    from ..models.file_set_file_metadata_type_0 import FileSetFileMetadataType0
+
+
+T = TypeVar("T", bound="FileSetFile")
+
+
+@_attrs_define
+class FileSetFile:
+    """
+    Attributes:
+        id (str):
+        original_file_name (str):
+        cloud_storage_path (str):
+        mime_type (None | str):
+        size_bytes (int):
+        character_count (int | None):
+        metadata (FileSetFileMetadataType0 | None):
+        gemini_file_id (None | str):
+        file_created_date (datetime.datetime | None):
+        created_at (datetime.datetime):
+        updated_at (datetime.datetime):
+    """
+
+    id: str
+    original_file_name: str
+    cloud_storage_path: str
+    mime_type: None | str
+    size_bytes: int
+    character_count: int | None
+    metadata: FileSetFileMetadataType0 | None
+    gemini_file_id: None | str
+    file_created_date: datetime.datetime | None
+    created_at: datetime.datetime
+    updated_at: datetime.datetime
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        from ..models.file_set_file_metadata_type_0 import FileSetFileMetadataType0
+
+        id = self.id
+
+        original_file_name = self.original_file_name
+
+        cloud_storage_path = self.cloud_storage_path
+
+        mime_type: None | str
+        mime_type = self.mime_type
+
+        size_bytes = self.size_bytes
+
+        character_count: int | None
+        character_count = self.character_count
+
+        metadata: dict[str, Any] | None
+        if isinstance(self.metadata, FileSetFileMetadataType0):
+            metadata = self.metadata.to_dict()
+        else:
+            metadata = self.metadata
+
+        gemini_file_id: None | str
+        gemini_file_id = self.gemini_file_id
+
+        file_created_date: None | str
+        if isinstance(self.file_created_date, datetime.datetime):
+            file_created_date = self.file_created_date.isoformat()
+        else:
+            file_created_date = self.file_created_date
+
+        created_at = self.created_at.isoformat()
+
+        updated_at = self.updated_at.isoformat()
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "id": id,
+                "original_file_name": original_file_name,
+                "cloud_storage_path": cloud_storage_path,
+                "mime_type": mime_type,
+                "size_bytes": size_bytes,
+                "character_count": character_count,
+                "metadata": metadata,
+                "gemini_file_id": gemini_file_id,
+                "file_created_date": file_created_date,
+                "created_at": created_at,
+                "updated_at": updated_at,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.file_set_file_metadata_type_0 import FileSetFileMetadataType0
+
+        d = dict(src_dict)
+        id = d.pop("id")
+
+        original_file_name = d.pop("original_file_name")
+
+        cloud_storage_path = d.pop("cloud_storage_path")
+
+        def _parse_mime_type(data: object) -> None | str:
+            if data is None:
+                return data
+            return cast(None | str, data)
+
+        mime_type = _parse_mime_type(d.pop("mime_type"))
+
+        size_bytes = d.pop("size_bytes")
+
+        def _parse_character_count(data: object) -> int | None:
+            if data is None:
+                return data
+            return cast(int | None, data)
+
+        character_count = _parse_character_count(d.pop("character_count"))
+
+        def _parse_metadata(data: object) -> FileSetFileMetadataType0 | None:
+            if data is None:
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                metadata_type_0 = FileSetFileMetadataType0.from_dict(data)
+
+                return metadata_type_0
+            except (TypeError, ValueError, AttributeError, KeyError):
+                pass
+            return cast(FileSetFileMetadataType0 | None, data)
+
+        metadata = _parse_metadata(d.pop("metadata"))
+
+        def _parse_gemini_file_id(data: object) -> None | str:
+            if data is None:
+                return data
+            return cast(None | str, data)
+
+        gemini_file_id = _parse_gemini_file_id(d.pop("gemini_file_id"))
+
+        def _parse_file_created_date(data: object) -> datetime.datetime | None:
+            if data is None:
+                return data
+            try:
+                if not isinstance(data, str):
+                    raise TypeError()
+                file_created_date_type_0 = isoparse(data)
+
+                return file_created_date_type_0
+            except (TypeError, ValueError, AttributeError, KeyError):
+                pass
+            return cast(datetime.datetime | None, data)
+
+        file_created_date = _parse_file_created_date(d.pop("file_created_date"))
+
+        created_at = isoparse(d.pop("created_at"))
+
+        updated_at = isoparse(d.pop("updated_at"))
+
+        file_set_file = cls(
+            id=id,
+            original_file_name=original_file_name,
+            cloud_storage_path=cloud_storage_path,
+            mime_type=mime_type,
+            size_bytes=size_bytes,
+            character_count=character_count,
+            metadata=metadata,
+            gemini_file_id=gemini_file_id,
+            file_created_date=file_created_date,
+            created_at=created_at,
+            updated_at=updated_at,
+        )
+
+        file_set_file.additional_properties = d
+        return file_set_file
+
+    @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/file_set_file_metadata_type_0.py

@@ -0,0 +1,57 @@
+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="FileSetFileMetadataType0")
+
+
+@_attrs_define
+class FileSetFileMetadataType0:
+    """ """
+
+    additional_properties: dict[str, bool | float | int | str] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        field_dict: dict[str, Any] = {}
+        for prop_name, prop in self.additional_properties.items():
+            field_dict[prop_name] = prop
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        file_set_file_metadata_type_0 = cls()
+
+        additional_properties = {}
+        for prop_name, prop_dict in d.items():
+
+            def _parse_additional_property(data: object) -> bool | float | int | str:
+                return cast(bool | float | int | str, data)
+
+            additional_property = _parse_additional_property(prop_dict)
+
+            additional_properties[prop_name] = additional_property
+
+        file_set_file_metadata_type_0.additional_properties = additional_properties
+        return file_set_file_metadata_type_0
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> bool | float | int | str:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: bool | float | int | str) -> 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