jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Refactor code in utils.py

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-01-30 20:20:11 -03:00
commit 577ede2386
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/backend/langflow/services/monitor/utils.py
View file

@ -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)