fix fastapi docs (#761)
Co-authored-by: Shuchang Zheng <wintonzheng0325@gmail.com>
This commit is contained in:
@@ -3,7 +3,7 @@ from __future__ import annotations
|
|||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
from enum import StrEnum
|
from enum import StrEnum
|
||||||
|
|
||||||
from pydantic import BaseModel, Field
|
from pydantic import BaseModel, Field, field_serializer
|
||||||
|
|
||||||
|
|
||||||
class ArtifactType(StrEnum):
|
class ArtifactType(StrEnum):
|
||||||
@@ -47,14 +47,17 @@ class Artifact(BaseModel):
|
|||||||
...,
|
...,
|
||||||
description="The creation datetime of the task.",
|
description="The creation datetime of the task.",
|
||||||
examples=["2023-01-01T00:00:00Z"],
|
examples=["2023-01-01T00:00:00Z"],
|
||||||
json_encoders={datetime: lambda v: v.isoformat()},
|
|
||||||
)
|
)
|
||||||
modified_at: datetime = Field(
|
modified_at: datetime = Field(
|
||||||
...,
|
...,
|
||||||
description="The modification datetime of the task.",
|
description="The modification datetime of the task.",
|
||||||
examples=["2023-01-01T00:00:00Z"],
|
examples=["2023-01-01T00:00:00Z"],
|
||||||
json_encoders={datetime: lambda v: v.isoformat()},
|
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@field_serializer("created_at", "modified_at", when_used="json")
|
||||||
|
def serialize_datetime_to_isoformat(self, value: datetime) -> str:
|
||||||
|
return value.isoformat()
|
||||||
|
|
||||||
artifact_id: str = Field(
|
artifact_id: str = Field(
|
||||||
...,
|
...,
|
||||||
description="The ID of the task artifact.",
|
description="The ID of the task artifact.",
|
||||||
|
|||||||
Reference in New Issue
Block a user