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

Refactor ContractEdge log_transaction method

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-23 19:15:16 -03:00
commit a8e27bac6d
1 changed files with 8 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

14
src/backend/langflow/graph/edge/base.py
View file

@ -1,10 +1,11 @@
from typing import TYPE_CHECKING, Any, List, Optional from typing import TYPE_CHECKING, Any, List, Optional
from loguru import logger
from pydantic import BaseModel, Field
from langflow.graph.edge.utils import build_clean_params from langflow.graph.edge.utils import build_clean_params
from langflow.services.deps import get_monitor_service from langflow.services.deps import get_monitor_service
from langflow.services.monitor.utils import log_message from langflow.services.monitor.utils import log_message
from loguru import logger
from pydantic import BaseModel, Field
if TYPE_CHECKING: if TYPE_CHECKING:
from langflow.graph.vertex.base import Vertex from langflow.graph.vertex.base import Vertex
@ -135,11 +136,11 @@ class ContractEdge(Edge):
log_transaction(self, source, target, "success") log_transaction(self, source, target, "success")
# If the target vertex is a power component we log messages # If the target vertex is a power component we log messages
if ( if target.vertex_type == "ChatOutput" and (
target.vertex_type == "ChatOutput" isinstance(target.params.get("message"), str) or isinstance(target.params.get("message"), dict)
and isinstance(target.params.get("message"), str)
or isinstance(target.params.get("message"), dict)
): ):
if target.params.get("message") == "":
return self.result
await log_message( await log_message(
sender=target.params.get("sender", ""), sender=target.params.get("sender", ""),
sender_name=target.params.get("sender_name", ""), sender_name=target.params.get("sender_name", ""),
@ -168,3 +169,4 @@ def log_transaction(edge: ContractEdge, source: "Vertex", target: "Vertex", stat
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:
logger.error(f"Error logging transaction: {e}") logger.error(f"Error logging transaction: {e}")
logger.error(f"Error logging transaction: {e}")