diff options
Diffstat (limited to 'packages/sdk/python/src/opencode_ai/models/model_limit.py')
| -rw-r--r-- | packages/sdk/python/src/opencode_ai/models/model_limit.py | 67 |
1 files changed, 67 insertions, 0 deletions
diff --git a/packages/sdk/python/src/opencode_ai/models/model_limit.py b/packages/sdk/python/src/opencode_ai/models/model_limit.py new file mode 100644 index 000000000..701f53130 --- /dev/null +++ b/packages/sdk/python/src/opencode_ai/models/model_limit.py @@ -0,0 +1,67 @@ +from collections.abc import Mapping +from typing import Any, TypeVar + +from attrs import define as _attrs_define +from attrs import field as _attrs_field + +T = TypeVar("T", bound="ModelLimit") + + +@_attrs_define +class ModelLimit: + """ + Attributes: + context (float): + output (float): + """ + + context: float + output: float + additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict) + + def to_dict(self) -> dict[str, Any]: + context = self.context + + output = self.output + + field_dict: dict[str, Any] = {} + field_dict.update(self.additional_properties) + field_dict.update( + { + "context": context, + "output": output, + } + ) + + return field_dict + + @classmethod + def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T: + d = dict(src_dict) + context = d.pop("context") + + output = d.pop("output") + + model_limit = cls( + context=context, + output=output, + ) + + model_limit.additional_properties = d + return model_limit + + @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 |
