From dc37f279e0b9b3c2b6af2e4d505f4f2b8cf56c81 Mon Sep 17 00:00:00 2001 From: ogabrielluiz Date: Wed, 12 Jun 2024 17:35:30 -0300 Subject: [PATCH] refactor: Update Log message field to allow list type Update the Log class in schema.py to allow the message field to accept a list type in addition to str, dict, and StreamURL. This change enhances the flexibility of the Log class and accommodates scenarios where multiple messages need to be stored. --- src/backend/base/langflow/schema/schema.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/src/backend/base/langflow/schema/schema.py b/src/backend/base/langflow/schema/schema.py index d82ceca83..4939ad82d 100644 --- a/src/backend/base/langflow/schema/schema.py +++ b/src/backend/base/langflow/schema/schema.py @@ -1,5 +1,5 @@ from collections import defaultdict -from typing import Literal +from typing import Any, Literal from typing_extensions import TypedDict @@ -14,7 +14,7 @@ class StreamURL(TypedDict): class Log(TypedDict): - message: str | dict | StreamURL + message: str | dict | StreamURL | list type: str @@ -33,5 +33,8 @@ def build_logs_from_artifacts(artifacts: dict) -> dict: log = Log(message=message, type=message["type"]) logs[key].append(log) - return logs + + +def build_log_from_raw_and_type(raw: Any, log_type: str) -> Log: + return Log(message=raw, type=log_type)