From 8bee638af2f23c5c4a1fd7fac1171899e830d252 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Wed, 21 Feb 2024 16:01:12 -0300
Subject: [PATCH] Add Union type for monitor_data parameter in add_row_to_table
 function

---
 src/backend/langflow/services/monitor/utils.py | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/src/backend/langflow/services/monitor/utils.py b/src/backend/langflow/services/monitor/utils.py
index 6ebba74eb..9dabdc7f2 100644
--- a/src/backend/langflow/services/monitor/utils.py
+++ b/src/backend/langflow/services/monitor/utils.py
@@ -1,4 +1,4 @@
-from typing import TYPE_CHECKING, Any, Dict, Optional, Type
+from typing import TYPE_CHECKING, Any, Dict, Optional, Type, Union
 
 import duckdb
 from langflow.services.deps import get_monitor_service
@@ -79,10 +79,13 @@ def add_row_to_table(
     conn: duckdb.DuckDBPyConnection,
     table_name: str,
     model: Type[BaseModel],
-    monitor_data: Dict[str, Any],
+    monitor_data: Union[Dict[str, Any], BaseModel],
 ):
     # Validate the data with the Pydantic model
-    validated_data = model(**monitor_data)
+    if isinstance(monitor_data, model):
+        validated_data = monitor_data
+    else:
+        validated_data = model(**monitor_data)
 
     # Extract data for the insert statement
     validated_dict = validated_data.model_dump(exclude_unset=True)
@@ -107,7 +110,7 @@ def add_row_to_table(
 
 
 async def log_message(
-    sender_type: str,
+    sender: str,
     sender_name: str,
     message: str,
     session_id: str,
@@ -121,7 +124,7 @@ async def log_message(
 
         monitor_service = get_monitor_service()
         row = {
-            "sender_type": sender_type,
+            "sender": sender,
             "sender_name": sender_name,
             "message": message,
             "artifacts": artifacts or {},