summaryrefslogtreecommitdiffhomepage
path: root/packages/sdk/python/src/opencode_ai/models/file_part_input.py
blob: 0bebb3cb5dbcc6afae8fb8102625e460abc4442d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
from collections.abc import Mapping
from typing import TYPE_CHECKING, Any, Literal, TypeVar, Union, cast

from attrs import define as _attrs_define
from attrs import field as _attrs_field

from ..types import UNSET, Unset

if TYPE_CHECKING:
    from ..models.file_source import FileSource
    from ..models.symbol_source import SymbolSource


T = TypeVar("T", bound="FilePartInput")


@_attrs_define
class FilePartInput:
    """
    Attributes:
        type_ (Literal['file']):
        mime (str):
        url (str):
        id (Union[Unset, str]):
        filename (Union[Unset, str]):
        source (Union['FileSource', 'SymbolSource', Unset]):
    """

    type_: Literal["file"]
    mime: str
    url: str
    id: Union[Unset, str] = UNSET
    filename: Union[Unset, str] = UNSET
    source: Union["FileSource", "SymbolSource", Unset] = UNSET
    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)

    def to_dict(self) -> dict[str, Any]:
        from ..models.file_source import FileSource

        type_ = self.type_

        mime = self.mime

        url = self.url

        id = self.id

        filename = self.filename

        source: Union[Unset, dict[str, Any]]
        if isinstance(self.source, Unset):
            source = UNSET
        elif isinstance(self.source, FileSource):
            source = self.source.to_dict()
        else:
            source = self.source.to_dict()

        field_dict: dict[str, Any] = {}
        field_dict.update(self.additional_properties)
        field_dict.update(
            {
                "type": type_,
                "mime": mime,
                "url": url,
            }
        )
        if id is not UNSET:
            field_dict["id"] = id
        if filename is not UNSET:
            field_dict["filename"] = filename
        if source is not UNSET:
            field_dict["source"] = source

        return field_dict

    @classmethod
    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
        from ..models.file_source import FileSource
        from ..models.symbol_source import SymbolSource

        d = dict(src_dict)
        type_ = cast(Literal["file"], d.pop("type"))
        if type_ != "file":
            raise ValueError(f"type must match const 'file', got '{type_}'")

        mime = d.pop("mime")

        url = d.pop("url")

        id = d.pop("id", UNSET)

        filename = d.pop("filename", UNSET)

        def _parse_source(data: object) -> Union["FileSource", "SymbolSource", Unset]:
            if isinstance(data, Unset):
                return data
            try:
                if not isinstance(data, dict):
                    raise TypeError()
                componentsschemas_file_part_source_type_0 = FileSource.from_dict(data)

                return componentsschemas_file_part_source_type_0
            except:  # noqa: E722
                pass
            if not isinstance(data, dict):
                raise TypeError()
            componentsschemas_file_part_source_type_1 = SymbolSource.from_dict(data)

            return componentsschemas_file_part_source_type_1

        source = _parse_source(d.pop("source", UNSET))

        file_part_input = cls(
            type_=type_,
            mime=mime,
            url=url,
            id=id,
            filename=filename,
            source=source,
        )

        file_part_input.additional_properties = d
        return file_part_input

    @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