orca-sdk 0.0.97__py3-none-any.whl → 0.0.98__py3-none-any.whl

orca_sdk/_generated_api_client/models/list_predictions_request.py

@@ -1,319 +0,0 @@
-"""
-This file is generated by the openapi-python-client tool via the generate_api_client.py script
-
-It is a customized template from the openapi-python-client tool's default template:
-https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/model.py.jinja
-
-The main change is:
-- Fix typing issues
-"""
-
-# flake8: noqa: C901
-
-import datetime
-from enum import Enum
-from typing import Any, List, Type, TypeVar, Union, cast
-
-from attrs import define as _attrs_define
-from attrs import field as _attrs_field
-from dateutil.parser import isoparse
-
-from ..models.prediction_sort_item_item_type_0 import PredictionSortItemItemType0
-from ..models.prediction_sort_item_item_type_1 import PredictionSortItemItemType1
-from ..types import UNSET, Unset
-
-T = TypeVar("T", bound="ListPredictionsRequest")
-
-
-@_attrs_define
-class ListPredictionsRequest:
-    """
-    Attributes:
-        model_id (Union[None, Unset, str]):
-        tag (Union[None, Unset, str]):
-        prediction_ids (Union[List[str], None, Unset]):
-        start_timestamp (Union[None, Unset, datetime.datetime]):
-        end_timestamp (Union[None, Unset, datetime.datetime]):
-        limit (Union[None, Unset, int]):
-        offset (Union[None, Unset, int]):  Default: 0.
-        sort (Union[Unset, List[List[Union[PredictionSortItemItemType0, PredictionSortItemItemType1]]]]):
-        expected_label_match (Union[None, Unset, bool]):
-    """
-
-    model_id: Union[None, Unset, str] = UNSET
-    tag: Union[None, Unset, str] = UNSET
-    prediction_ids: Union[List[str], None, Unset] = UNSET
-    start_timestamp: Union[None, Unset, datetime.datetime] = UNSET
-    end_timestamp: Union[None, Unset, datetime.datetime] = UNSET
-    limit: Union[None, Unset, int] = UNSET
-    offset: Union[None, Unset, int] = 0
-    sort: Union[Unset, List[List[Union[PredictionSortItemItemType0, PredictionSortItemItemType1]]]] = UNSET
-    expected_label_match: Union[None, Unset, bool] = UNSET
-    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> dict[str, Any]:
-        model_id: Union[None, Unset, str]
-        if isinstance(self.model_id, Unset):
-            model_id = UNSET
-        else:
-            model_id = self.model_id
-
-        tag: Union[None, Unset, str]
-        if isinstance(self.tag, Unset):
-            tag = UNSET
-        else:
-            tag = self.tag
-
-        prediction_ids: Union[List[str], None, Unset]
-        if isinstance(self.prediction_ids, Unset):
-            prediction_ids = UNSET
-        elif isinstance(self.prediction_ids, list):
-            prediction_ids = self.prediction_ids
-
-        else:
-            prediction_ids = self.prediction_ids
-
-        start_timestamp: Union[None, Unset, str]
-        if isinstance(self.start_timestamp, Unset):
-            start_timestamp = UNSET
-        elif isinstance(self.start_timestamp, datetime.datetime):
-            start_timestamp = self.start_timestamp.isoformat()
-        else:
-            start_timestamp = self.start_timestamp
-
-        end_timestamp: Union[None, Unset, str]
-        if isinstance(self.end_timestamp, Unset):
-            end_timestamp = UNSET
-        elif isinstance(self.end_timestamp, datetime.datetime):
-            end_timestamp = self.end_timestamp.isoformat()
-        else:
-            end_timestamp = self.end_timestamp
-
-        limit: Union[None, Unset, int]
-        if isinstance(self.limit, Unset):
-            limit = UNSET
-        else:
-            limit = self.limit
-
-        offset: Union[None, Unset, int]
-        if isinstance(self.offset, Unset):
-            offset = UNSET
-        else:
-            offset = self.offset
-
-        sort: Union[Unset, List[List[str]]] = UNSET
-        if not isinstance(self.sort, Unset):
-            sort = []
-            for componentsschemas_prediction_sort_item_data in self.sort:
-                componentsschemas_prediction_sort_item = []
-                for componentsschemas_prediction_sort_item_item_data in componentsschemas_prediction_sort_item_data:
-                    componentsschemas_prediction_sort_item_item: str
-                    if isinstance(componentsschemas_prediction_sort_item_item_data, PredictionSortItemItemType0):
-                        componentsschemas_prediction_sort_item_item = (
-                            componentsschemas_prediction_sort_item_item_data.value
-                            if isinstance(componentsschemas_prediction_sort_item_item_data, Enum)
-                            else componentsschemas_prediction_sort_item_item_data
-                        )
-                    else:
-                        componentsschemas_prediction_sort_item_item = (
-                            componentsschemas_prediction_sort_item_item_data.value
-                            if isinstance(componentsschemas_prediction_sort_item_item_data, Enum)
-                            else componentsschemas_prediction_sort_item_item_data
-                        )
-
-                    componentsschemas_prediction_sort_item.append(componentsschemas_prediction_sort_item_item)
-
-                sort.append(componentsschemas_prediction_sort_item)
-
-        expected_label_match: Union[None, Unset, bool]
-        if isinstance(self.expected_label_match, Unset):
-            expected_label_match = UNSET
-        else:
-            expected_label_match = self.expected_label_match
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update({})
-        if model_id is not UNSET:
-            field_dict["model_id"] = model_id
-        if tag is not UNSET:
-            field_dict["tag"] = tag
-        if prediction_ids is not UNSET:
-            field_dict["prediction_ids"] = prediction_ids
-        if start_timestamp is not UNSET:
-            field_dict["start_timestamp"] = start_timestamp
-        if end_timestamp is not UNSET:
-            field_dict["end_timestamp"] = end_timestamp
-        if limit is not UNSET:
-            field_dict["limit"] = limit
-        if offset is not UNSET:
-            field_dict["offset"] = offset
-        if sort is not UNSET:
-            field_dict["sort"] = sort
-        if expected_label_match is not UNSET:
-            field_dict["expected_label_match"] = expected_label_match
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
-        d = src_dict.copy()
-
-        def _parse_model_id(data: object) -> Union[None, Unset, str]:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            return cast(Union[None, Unset, str], data)
-
-        model_id = _parse_model_id(d.pop("model_id", UNSET))
-
-        def _parse_tag(data: object) -> Union[None, Unset, str]:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            return cast(Union[None, Unset, str], data)
-
-        tag = _parse_tag(d.pop("tag", UNSET))
-
-        def _parse_prediction_ids(data: object) -> Union[List[str], None, Unset]:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            try:
-                if not isinstance(data, list):
-                    raise TypeError()
-                prediction_ids_type_0 = cast(List[str], data)
-
-                return prediction_ids_type_0
-            except:  # noqa: E722
-                pass
-            return cast(Union[List[str], None, Unset], data)
-
-        prediction_ids = _parse_prediction_ids(d.pop("prediction_ids", UNSET))
-
-        def _parse_start_timestamp(data: object) -> Union[None, Unset, datetime.datetime]:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            try:
-                if not isinstance(data, str):
-                    raise TypeError()
-                start_timestamp_type_0 = isoparse(data)
-
-                return start_timestamp_type_0
-            except:  # noqa: E722
-                pass
-            return cast(Union[None, Unset, datetime.datetime], data)
-
-        start_timestamp = _parse_start_timestamp(d.pop("start_timestamp", UNSET))
-
-        def _parse_end_timestamp(data: object) -> Union[None, Unset, datetime.datetime]:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            try:
-                if not isinstance(data, str):
-                    raise TypeError()
-                end_timestamp_type_0 = isoparse(data)
-
-                return end_timestamp_type_0
-            except:  # noqa: E722
-                pass
-            return cast(Union[None, Unset, datetime.datetime], data)
-
-        end_timestamp = _parse_end_timestamp(d.pop("end_timestamp", UNSET))
-
-        def _parse_limit(data: object) -> Union[None, Unset, int]:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            return cast(Union[None, Unset, int], data)
-
-        limit = _parse_limit(d.pop("limit", UNSET))
-
-        def _parse_offset(data: object) -> Union[None, Unset, int]:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            return cast(Union[None, Unset, int], data)
-
-        offset = _parse_offset(d.pop("offset", UNSET))
-
-        sort = []
-        _sort = d.pop("sort", UNSET)
-        for componentsschemas_prediction_sort_item_data in _sort or []:
-            componentsschemas_prediction_sort_item = []
-            _componentsschemas_prediction_sort_item = componentsschemas_prediction_sort_item_data
-            for componentsschemas_prediction_sort_item_item_data in _componentsschemas_prediction_sort_item:
-
-                def _parse_componentsschemas_prediction_sort_item_item(
-                    data: object,
-                ) -> Union[PredictionSortItemItemType0, PredictionSortItemItemType1]:
-                    try:
-                        if not isinstance(data, str):
-                            raise TypeError()
-                        componentsschemas_prediction_sort_item_item_type_0 = PredictionSortItemItemType0(data)
-
-                        return componentsschemas_prediction_sort_item_item_type_0
-                    except:  # noqa: E722
-                        pass
-                    if not isinstance(data, str):
-                        raise TypeError()
-                    componentsschemas_prediction_sort_item_item_type_1 = PredictionSortItemItemType1(data)
-
-                    return componentsschemas_prediction_sort_item_item_type_1
-
-                componentsschemas_prediction_sort_item_item = _parse_componentsschemas_prediction_sort_item_item(
-                    componentsschemas_prediction_sort_item_item_data
-                )
-
-                componentsschemas_prediction_sort_item.append(componentsschemas_prediction_sort_item_item)
-
-            sort.append(componentsschemas_prediction_sort_item)
-
-        def _parse_expected_label_match(data: object) -> Union[None, Unset, bool]:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            return cast(Union[None, Unset, bool], data)
-
-        expected_label_match = _parse_expected_label_match(d.pop("expected_label_match", UNSET))
-
-        list_predictions_request = cls(
-            model_id=model_id,
-            tag=tag,
-            prediction_ids=prediction_ids,
-            start_timestamp=start_timestamp,
-            end_timestamp=end_timestamp,
-            limit=limit,
-            offset=offset,
-            sort=sort,
-            expected_label_match=expected_label_match,
-        )
-
-        list_predictions_request.additional_properties = d
-        return list_predictions_request
-
-    @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

orca_sdk/_generated_api_client/models/lookup_request.py

@@ -1,101 +0,0 @@
-"""
-This file is generated by the openapi-python-client tool via the generate_api_client.py script
-
-It is a customized template from the openapi-python-client tool's default template:
-https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/model.py.jinja
-
-The main change is:
-- Fix typing issues
-"""
-
-# flake8: noqa: C901
-
-from typing import Any, List, Type, TypeVar, Union, cast
-
-from attrs import define as _attrs_define
-from attrs import field as _attrs_field
-
-from ..types import UNSET, Unset
-
-T = TypeVar("T", bound="LookupRequest")
-
-
-@_attrs_define
-class LookupRequest:
-    """
-    Attributes:
-        query (List[str]):
-        count (Union[Unset, int]):  Default: 1.
-        prompt (Union[None, Unset, str]):
-    """
-
-    query: List[str]
-    count: Union[Unset, int] = 1
-    prompt: Union[None, Unset, str] = UNSET
-    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> dict[str, Any]:
-        query = self.query
-
-        count = self.count
-
-        prompt: Union[None, Unset, str]
-        if isinstance(self.prompt, Unset):
-            prompt = UNSET
-        else:
-            prompt = self.prompt
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update(
-            {
-                "query": query,
-            }
-        )
-        if count is not UNSET:
-            field_dict["count"] = count
-        if prompt is not UNSET:
-            field_dict["prompt"] = prompt
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
-        d = src_dict.copy()
-        query = cast(List[str], d.pop("query"))
-
-        count = d.pop("count", UNSET)
-
-        def _parse_prompt(data: object) -> Union[None, Unset, str]:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            return cast(Union[None, Unset, str], data)
-
-        prompt = _parse_prompt(d.pop("prompt", UNSET))
-
-        lookup_request = cls(
-            query=query,
-            count=count,
-            prompt=prompt,
-        )
-
-        lookup_request.additional_properties = d
-        return lookup_request
-
-    @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

orca_sdk/_generated_api_client/models/lookup_score_metrics.py

@@ -1,94 +0,0 @@
-"""
-This file is generated by the openapi-python-client tool via the generate_api_client.py script
-
-It is a customized template from the openapi-python-client tool's default template:
-https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/model.py.jinja
-
-The main change is:
-- Fix typing issues
-"""
-
-# flake8: noqa: C901
-
-from typing import Any, List, Type, TypeVar, cast
-
-from attrs import define as _attrs_define
-from attrs import field as _attrs_field
-
-T = TypeVar("T", bound="LookupScoreMetrics")
-
-
-@_attrs_define
-class LookupScoreMetrics:
-    """
-    Attributes:
-        median (float):
-        std (float):
-        quantiles (List[float]):
-        quantile_values (List[float]):
-    """
-
-    median: float
-    std: float
-    quantiles: List[float]
-    quantile_values: List[float]
-    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> dict[str, Any]:
-        median = self.median
-
-        std = self.std
-
-        quantiles = self.quantiles
-
-        quantile_values = self.quantile_values
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update(
-            {
-                "median": median,
-                "std": std,
-                "quantiles": quantiles,
-                "quantile_values": quantile_values,
-            }
-        )
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
-        d = src_dict.copy()
-        median = d.pop("median")
-
-        std = d.pop("std")
-
-        quantiles = cast(List[float], d.pop("quantiles"))
-
-        quantile_values = cast(List[float], d.pop("quantile_values"))
-
-        lookup_score_metrics = cls(
-            median=median,
-            std=std,
-            quantiles=quantiles,
-            quantile_values=quantile_values,
-        )
-
-        lookup_score_metrics.additional_properties = d
-        return lookup_score_metrics
-
-    @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