Compare commits

...

1 Commits

Author SHA1 Message Date
Joe
25033a9180 fix: workflow slow db and logical operator error 2024-07-04 10:36:07 +08:00
3 changed files with 30 additions and 4 deletions

View File

@@ -24,10 +24,10 @@ class LangfuseConfig(BaseTracingConfig):
host: str = 'https://api.langfuse.com'
@field_validator("host")
def set_value(cls, v, info: ValidationInfo):
def set_value(cls, v):
if v is None or v == "":
v = 'https://api.langfuse.com'
if not v.startswith('https://') or not v.startswith('http://'):
if not v.startswith('https://') and not v.startswith('http://'):
raise ValueError('host must start with https:// or http://')
return v

View File

@@ -107,7 +107,20 @@ class LangFuseDataTrace(BaseTraceInstance):
# through workflow_run_id get all_nodes_execution
workflow_nodes_executions = (
db.session.query(WorkflowNodeExecution)
db.session.query(
WorkflowNodeExecution.id,
WorkflowNodeExecution.tenant_id,
WorkflowNodeExecution.app_id,
WorkflowNodeExecution.title,
WorkflowNodeExecution.node_type,
WorkflowNodeExecution.status,
WorkflowNodeExecution.inputs,
WorkflowNodeExecution.outputs,
WorkflowNodeExecution.created_at,
WorkflowNodeExecution.elapsed_time,
WorkflowNodeExecution.execution_metadata,
WorkflowNodeExecution.process_data,
)
.filter(WorkflowNodeExecution.workflow_run_id == trace_info.workflow_run_id)
.all()
)

View File

@@ -100,7 +100,20 @@ class LangSmithDataTrace(BaseTraceInstance):
# through workflow_run_id get all_nodes_execution
workflow_nodes_executions = (
db.session.query(WorkflowNodeExecution)
db.session.query(
WorkflowNodeExecution.id,
WorkflowNodeExecution.tenant_id,
WorkflowNodeExecution.app_id,
WorkflowNodeExecution.title,
WorkflowNodeExecution.node_type,
WorkflowNodeExecution.status,
WorkflowNodeExecution.inputs,
WorkflowNodeExecution.outputs,
WorkflowNodeExecution.created_at,
WorkflowNodeExecution.elapsed_time,
WorkflowNodeExecution.execution_metadata,
WorkflowNodeExecution.process_data,
)
.filter(WorkflowNodeExecution.workflow_run_id == trace_info.workflow_run_id)
.all()
)