add new type to Log

This commit is contained in:
italojohnny 2024-05-29 16:12:53 -03:00
commit d05e7c12f4
2 changed files with 7 additions and 6 deletions

View file

@ -246,7 +246,7 @@ class VerticesOrderResponse(BaseModel):
class Log(TypedDict):
message: str
message: Union[dict, str]
type: str

View file

@ -19,6 +19,7 @@ class UnbuiltResult:
class ArtifactType(str, Enum):
TEXT = "text"
RECORD = "record"
OBJECT = "object"
UNKNOWN = "unknown"
@ -60,16 +61,16 @@ def serialize_field(value):
return value
def get_artifact_type(build_result: Any) -> str:
result = None
match build_result:
def get_artifact_type(value: Any) -> str:
result = ArtifactType.UNKNOWN
match value:
case Record():
result = ArtifactType.RECORD
case str():
result = ArtifactType.TEXT
case _:
result = ArtifactType.UNKNOWN
case dict():
result = ArtifactType.OBJECT
return result.value