From 0292917c056c19047464ef0cd372a936d589fad4 Mon Sep 17 00:00:00 2001 From: VittorioYan <1209087381@qq.com> Date: Sat, 7 Sep 2024 11:11:52 +0800 Subject: [PATCH] fix fastapi docs (#761) Co-authored-by: Shuchang Zheng --- skyvern/forge/sdk/artifact/models.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/skyvern/forge/sdk/artifact/models.py b/skyvern/forge/sdk/artifact/models.py index cfc475bb..3ea021bf 100644 --- a/skyvern/forge/sdk/artifact/models.py +++ b/skyvern/forge/sdk/artifact/models.py @@ -3,7 +3,7 @@ from __future__ import annotations from datetime import datetime from enum import StrEnum -from pydantic import BaseModel, Field +from pydantic import BaseModel, Field, field_serializer class ArtifactType(StrEnum): @@ -47,14 +47,17 @@ class Artifact(BaseModel): ..., description="The creation datetime of the task.", examples=["2023-01-01T00:00:00Z"], - json_encoders={datetime: lambda v: v.isoformat()}, ) modified_at: datetime = Field( ..., description="The modification datetime of the task.", 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( ..., description="The ID of the task artifact.",