cohere-python
42 строки · 1.2 Кб
1# This file was auto-generated by Fern from our API Definition.
2
3import datetime as dt
4import typing
5
6from ..core.datetime_utils import serialize_datetime
7from .api_meta import ApiMeta
8from .single_generation import SingleGeneration
9
10try:
11import pydantic.v1 as pydantic # type: ignore
12except ImportError:
13import pydantic # type: ignore
14
15
16class Generation(pydantic.BaseModel):
17id: str
18prompt: typing.Optional[str] = pydantic.Field(default=None)
19"""
20Prompt used for generations.
21"""
22
23generations: typing.List[SingleGeneration] = pydantic.Field()
24"""
25List of generated results
26"""
27
28meta: typing.Optional[ApiMeta] = None
29
30def json(self, **kwargs: typing.Any) -> str:
31kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
32return super().json(**kwargs_with_defaults)
33
34def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
35kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
36return super().dict(**kwargs_with_defaults)
37
38class Config:
39frozen = True
40smart_union = True
41extra = pydantic.Extra.allow
42json_encoders = {dt.datetime: serialize_datetime}
43