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

♻️ (schema.py): remove unnecessary blank line to improve code readability

Browse source 
♻️ (service.py): reformat SQL query string for better readability
♻️ (utils.py): remove unnecessary blank line to clean up code
This commit is contained in:
cristhianzl 2024-06-07 19:53:40 -03:00
commit b3a3e7b8e5
3 changed files with 5 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

1
src/backend/base/langflow/services/monitor/schema.py
View file

@ -36,6 +36,7 @@ class TransactionModel(BaseModel):
return json.dumps(v) return json.dumps(v)
return v return v
class TransactionModelResponse(BaseModel): class TransactionModelResponse(BaseModel):
index: Optional[int] = Field(default=None) index: Optional[int] = Field(default=None)
timestamp: Optional[datetime] = Field(default_factory=datetime.now, alias="timestamp") timestamp: Optional[datetime] = Field(default_factory=datetime.now, alias="timestamp")

5
src/backend/base/langflow/services/monitor/service.py
View file

@ -168,8 +168,9 @@ class MonitorService(Service):
order_by: Optional[str] = "timestamp", order_by: Optional[str] = "timestamp",
flow_id: Optional[str] = None, flow_id: Optional[str] = None,
): ):
query = (
query = "SELECT index,flow_id, status, error, timestamp, vertex_id, inputs, outputs, target_id FROM transactions" "SELECT index,flow_id, status, error, timestamp, vertex_id, inputs, outputs, target_id FROM transactions"
)
conditions = [] conditions = []
if source: if source:
conditions.append(f"source = '{source}'") conditions.append(f"source = '{source}'")

3
src/backend/base/langflow/services/monitor/utils.py
View file

@ -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): def log_transaction(flow_id, vertex: "Vertex", status, target: Optional["Vertex"] = None, error=None):
try: try:
monitor_service = get_monitor_service() monitor_service = get_monitor_service()
clean_params = build_clean_params(vertex) clean_params = build_clean_params(vertex)
data = { data = {
@ -191,7 +190,7 @@ def log_transaction(flow_id, vertex: "Vertex", status, target: Optional["Vertex"
"timestamp": monitor_service.get_timestamp(), "timestamp": monitor_service.get_timestamp(),
"status": status, "status": status,
"error": error, "error": error,
"flow_id": flow_id "flow_id": flow_id,
} }
monitor_service.add_row(table_name="transactions", data=data) monitor_service.add_row(table_name="transactions", data=data)
except Exception as e: except Exception as e: