Skip to content

Commit 0132cfb

Browse files
update openapi
1 parent 9ad1480 commit 0132cfb

File tree

1 file changed

+20
-21
lines changed

1 file changed

+20
-21
lines changed

mostlyai/sdk/domain.py

Lines changed: 20 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -13,22 +13,21 @@
1313
# limitations under the License.
1414

1515
# generated by datamodel-codegen:
16-
# timestamp: 2025-08-18T08:11:01+00:00
16+
# timestamp: 2025-08-25T09:31:42+00:00
1717

1818
from __future__ import annotations
1919

2020
import inspect
2121
import sys
2222
import uuid
2323
import zipfile
24-
from datetime import datetime
2524
from enum import Enum
2625
from pathlib import Path
2726
from typing import Annotated, Any, ClassVar, Literal
2827

2928
import pandas as pd
3029
import rich
31-
from pydantic import Field, RootModel, field_validator, model_validator
30+
from pydantic import AwareDatetime, Field, RootModel, field_validator, model_validator
3231

3332
from mostlyai.sdk.client._base_utils import convert_to_base64, read_table_from_path
3433
from mostlyai.sdk.client.base import CustomBaseModel
@@ -97,12 +96,12 @@ class UsageCreditStats(CustomBaseModel):
9796
None,
9897
description="The credit limit for the user. If empty, then there is no limit.",
9998
)
100-
period_start: datetime | None = Field(
99+
period_start: AwareDatetime | None = Field(
101100
None,
102101
alias="periodStart",
103102
description="The UTC date and time when the current time period started.",
104103
)
105-
period_end: datetime | None = Field(
104+
period_end: AwareDatetime | None = Field(
106105
None,
107106
alias="periodEnd",
108107
description="The UTC date and time when the current time period ends.",
@@ -1294,7 +1293,7 @@ class OrganizationInvite(CustomBaseModel):
12941293
description="The generated token, encrypting organization, expiration timestamp, and role (VIEW).",
12951294
)
12961295
link: str | None = Field(None, description="The generated invite link.")
1297-
expiration_date: datetime | None = Field(
1296+
expiration_date: AwareDatetime | None = Field(
12981297
None,
12991298
alias="expirationDate",
13001299
description="The expiration date of the invite link. 72 hours after creation.",
@@ -1370,12 +1369,12 @@ class UsageReportPeriod(CustomBaseModel):
13701369
The usage report for a specific month
13711370
"""
13721371

1373-
period_start: datetime | None = Field(
1372+
period_start: AwareDatetime | None = Field(
13741373
None,
13751374
alias="periodStart",
13761375
description="The UTC date and time when the reported time period started",
13771376
)
1378-
period_end: datetime | None = Field(
1377+
period_end: AwareDatetime | None = Field(
13791378
None,
13801379
alias="periodEnd",
13811380
description="The UTC date and time when the reported time period started",
@@ -1845,7 +1844,7 @@ class Notification(CustomBaseModel):
18451844
type: NotificationType
18461845
message: str = Field(..., description="The message of the notification.")
18471846
status: NotificationStatus
1848-
created_at: datetime = Field(
1847+
created_at: AwareDatetime = Field(
18491848
...,
18501849
alias="createdAt",
18511850
description="The UTC date and time when the notification has been created.",
@@ -1869,7 +1868,7 @@ class Metadata(CustomBaseModel):
18691868
alias="creatorName",
18701869
description="The name of a user.\nContains only alphanumeric characters, hyphens, and underscores. Must start or end with alphanumeric.\nIt must be globally case-insensitive unique considering organizations and users.\n",
18711870
)
1872-
created_at: datetime | None = Field(
1871+
created_at: AwareDatetime | None = Field(
18731872
None,
18741873
alias="createdAt",
18751874
description="The UTC date and time when the resource has been created.",
@@ -2314,7 +2313,7 @@ class GeneratorListItem(CustomBaseModel):
23142313
name: str | None = Field(None, description="The name of a generator.")
23152314
description: str | None = Field(None, description="The description of a generator.")
23162315
training_status: ProgressStatus = Field(..., alias="trainingStatus")
2317-
training_time: datetime | None = Field(
2316+
training_time: AwareDatetime | None = Field(
23182317
None,
23192318
alias="trainingTime",
23202319
description="The UTC date and time when the training has finished.",
@@ -2531,13 +2530,13 @@ class ProgressStep(CustomBaseModel):
25312530
restarts: int | None = Field(0, description="The number of previous restarts for the corresponding task.")
25322531
task_type: TaskType | None = Field(None, alias="taskType")
25332532
step_code: StepCode | None = Field(None, alias="stepCode")
2534-
start_date: datetime | None = Field(
2533+
start_date: AwareDatetime | None = Field(
25352534
None,
25362535
alias="startDate",
25372536
description="The UTC date and time when the job has started.\nIf the job has not started yet, then this is None.\n",
25382537
examples=["2024-01-25T12:34:56Z"],
25392538
)
2540-
end_date: datetime | None = Field(
2539+
end_date: AwareDatetime | None = Field(
25412540
None,
25422541
alias="endDate",
25432542
description="The UTC date and time when the job has ended.\nIf the job is still, then this is None.\n",
@@ -2571,7 +2570,7 @@ class SyntheticDatasetListItem(CustomBaseModel):
25712570
name: str | None = Field(None, description="The name of a synthetic dataset.")
25722571
description: str | None = Field(None, description="The description of a synthetic dataset.")
25732572
generation_status: ProgressStatus = Field(..., alias="generationStatus")
2574-
generation_time: datetime | None = Field(
2573+
generation_time: AwareDatetime | None = Field(
25752574
None,
25762575
alias="generationTime",
25772576
description="The UTC date and time when the generation has finished.",
@@ -2607,17 +2606,17 @@ class SyntheticTableConfiguration(CustomBaseModel):
26072606
sample_seed_connector_id: str | None = Field(
26082607
None,
26092608
alias="sampleSeedConnectorId",
2610-
description="The connector id of the seed data for conditional simulation.\nOnly applicable for subject tables.\n",
2609+
description="The connector id of the seed data for conditional simulation",
26112610
)
26122611
sample_seed_dict: str | None = Field(
26132612
None,
26142613
alias="sampleSeedDict",
2615-
description="The base64-encoded string derived from a json line file containing the specified sample seed data.\nThis allows conditional live probing via non-python clients.\nOnly applicable for subject tables.\n",
2614+
description="The base64-encoded string derived from a json line file containing the specified sample seed data.\nThis allows conditional live probing via non-python clients.\n",
26162615
)
26172616
sample_seed_data: str | None = Field(
26182617
None,
26192618
alias="sampleSeedData",
2620-
description="The base64-encoded string derived from a Parquet file containing the specified sample seed data.\nThis allows conditional simulation as well as live probing via python clients.\nOnly applicable for subject tables.\n",
2619+
description="The base64-encoded string derived from a Parquet file containing the specified sample seed data.\nThis allows conditional simulation as well as live probing via python clients.\n",
26212620
)
26222621
sampling_temperature: float | None = Field(
26232622
1.0,
@@ -3117,13 +3116,13 @@ class JobProgress(CustomBaseModel):
31173116
"""
31183117

31193118
id: str | None = None
3120-
start_date: datetime | None = Field(
3119+
start_date: AwareDatetime | None = Field(
31213120
None,
31223121
alias="startDate",
31233122
description="The UTC date and time when the job has started.\nIf the job has not started yet, then this is None.\n",
31243123
examples=["2024-01-25T12:34:56Z"],
31253124
)
3126-
end_date: datetime | None = Field(
3125+
end_date: AwareDatetime | None = Field(
31273126
None,
31283127
alias="endDate",
31293128
description="The UTC date and time when the job has ended.\nIf the job is still, then this is None.\n",
@@ -3313,7 +3312,7 @@ class Generator(CustomBaseModel):
33133312
name: str | None = Field(None, description="The name of a generator.")
33143313
description: str | None = Field(None, description="The description of a generator.")
33153314
training_status: ProgressStatus = Field(..., alias="trainingStatus")
3316-
training_time: datetime | None = Field(
3315+
training_time: AwareDatetime | None = Field(
33173316
None,
33183317
alias="trainingTime",
33193318
description="The UTC date and time when the training has finished.",
@@ -3622,7 +3621,7 @@ class SyntheticDataset(CustomBaseModel):
36223621
name: str | None = Field(None, description="The name of a synthetic dataset.")
36233622
description: str | None = Field(None, description="The description of a synthetic dataset.")
36243623
generation_status: ProgressStatus = Field(..., alias="generationStatus")
3625-
generation_time: datetime | None = Field(
3624+
generation_time: AwareDatetime | None = Field(
36263625
None,
36273626
alias="generationTime",
36283627
description="The UTC date and time when the generation has finished.",

0 commit comments

Comments
 (0)