From b3a3e7b8e5ef9c8306c7292544919d513e5d3452 Mon Sep 17 00:00:00 2001 From: cristhianzl Date: Fri, 7 Jun 2024 19:53:40 -0300 Subject: [PATCH] =?UTF-8?q?=E2=99=BB=EF=B8=8F=20(schema.py):=20remove=20un?= =?UTF-8?q?necessary=20blank=20line=20to=20improve=20code=20readability=20?= =?UTF-8?q?=E2=99=BB=EF=B8=8F=20(service.py):=20reformat=20SQL=20query=20s?= =?UTF-8?q?tring=20for=20better=20readability=20=E2=99=BB=EF=B8=8F=20(util?= =?UTF-8?q?s.py):=20remove=20unnecessary=20blank=20line=20to=20clean=20up?= =?UTF-8?q?=20code?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/base/langflow/services/monitor/schema.py | 1 + src/backend/base/langflow/services/monitor/service.py | 5 +++-- src/backend/base/langflow/services/monitor/utils.py | 3 +-- 3 files changed, 5 insertions(+), 4 deletions(-) diff --git a/src/backend/base/langflow/services/monitor/schema.py b/src/backend/base/langflow/services/monitor/schema.py index 2c3396c7d..4cb057ccc 100644 --- a/src/backend/base/langflow/services/monitor/schema.py +++ b/src/backend/base/langflow/services/monitor/schema.py @@ -36,6 +36,7 @@ class TransactionModel(BaseModel): return json.dumps(v) return v + class TransactionModelResponse(BaseModel): index: Optional[int] = Field(default=None) timestamp: Optional[datetime] = Field(default_factory=datetime.now, alias="timestamp") diff --git a/src/backend/base/langflow/services/monitor/service.py b/src/backend/base/langflow/services/monitor/service.py index 6f1152fe6..6c37672af 100644 --- a/src/backend/base/langflow/services/monitor/service.py +++ b/src/backend/base/langflow/services/monitor/service.py @@ -168,8 +168,9 @@ class MonitorService(Service): order_by: Optional[str] = "timestamp", flow_id: Optional[str] = None, ): - - query = "SELECT index,flow_id, status, error, timestamp, vertex_id, inputs, outputs, target_id FROM transactions" + query = ( + "SELECT index,flow_id, status, error, timestamp, vertex_id, inputs, outputs, target_id FROM transactions" + ) conditions = [] if source: conditions.append(f"source = '{source}'") diff --git a/src/backend/base/langflow/services/monitor/utils.py b/src/backend/base/langflow/services/monitor/utils.py index 8d62083d2..706d62348 100644 --- a/src/backend/base/langflow/services/monitor/utils.py +++ b/src/backend/base/langflow/services/monitor/utils.py @@ -180,7 +180,6 @@ def build_clean_params(target: "Vertex") -> dict: def log_transaction(flow_id, vertex: "Vertex", status, target: Optional["Vertex"] = None, error=None): try: - monitor_service = get_monitor_service() clean_params = build_clean_params(vertex) data = { @@ -191,7 +190,7 @@ def log_transaction(flow_id, vertex: "Vertex", status, target: Optional["Vertex" "timestamp": monitor_service.get_timestamp(), "status": status, "error": error, - "flow_id": flow_id + "flow_id": flow_id, } monitor_service.add_row(table_name="transactions", data=data) except Exception as e: