From 577ede23868babd5931789c50524423f88fb35b6 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 30 Jan 2024 20:20:11 -0300 Subject: [PATCH] Refactor code in utils.py --- src/backend/langflow/services/monitor/utils.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/backend/langflow/services/monitor/utils.py b/src/backend/langflow/services/monitor/utils.py index c8fc8f581..e4ea16d09 100644 --- a/src/backend/langflow/services/monitor/utils.py +++ b/src/backend/langflow/services/monitor/utils.py @@ -1,11 +1,11 @@ from typing import Any, Dict, Optional, Type -from langflow.services.deps import get_monitor_service -from loguru import logger - import duckdb +from loguru import logger from pydantic import BaseModel +from langflow.services.deps import get_monitor_service + def get_table_schema_as_dict(conn: duckdb.DuckDBPyConnection, table_name: str) -> dict: result = conn.execute(f"PRAGMA table_info('{table_name}')").fetchall() @@ -68,7 +68,7 @@ def add_row_to_table( validated_data = model(**monitor_data) # Extract data for the insert statement - validated_dict = validated_data.dict(exclude_unset=True) + validated_dict = validated_data.model_dump(exclude_unset=True) keys = [key for key in validated_dict.keys() if key != "id"] columns = ", ".join(keys)