diff --git a/src/backend/base/langflow/api/v1/schemas.py b/src/backend/base/langflow/api/v1/schemas.py index 123be10ae..4030e8637 100644 --- a/src/backend/base/langflow/api/v1/schemas.py +++ b/src/backend/base/langflow/api/v1/schemas.py @@ -246,7 +246,7 @@ class VerticesOrderResponse(BaseModel): class Log(TypedDict): - message: str + message: Union[dict, str] type: str diff --git a/src/backend/base/langflow/graph/utils.py b/src/backend/base/langflow/graph/utils.py index 795e8f02c..f6a574114 100644 --- a/src/backend/base/langflow/graph/utils.py +++ b/src/backend/base/langflow/graph/utils.py @@ -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