浏览代码

fix(ops_trace_manager): Adds app_id to TraceTask initialization (#20461)

Signed-off-by: -LAN- <laipz8200@outlook.com>
tags/1.4.2
-LAN- 5 个月前
父节点
当前提交
f2e0d161a1
没有帐户链接到提交者的电子邮件
共有 2 个文件被更改,包括 8 次插入0 次删除
  1. 7
    0
      api/core/ops/langfuse_trace/langfuse_trace.py
  2. 1
    0
      api/core/ops/ops_trace_manager.py

+ 7
- 0
api/core/ops/langfuse_trace/langfuse_trace.py 查看文件

@@ -32,6 +32,7 @@ from core.repositories import SQLAlchemyWorkflowNodeExecutionRepository
from core.workflow.nodes.enums import NodeType
from extensions.ext_database import db
from models import Account, App, EndUser, WorkflowNodeExecutionTriggeredFrom
from models.account import TenantAccountJoin

logger = logging.getLogger(__name__)

@@ -130,6 +131,12 @@ class LangFuseDataTrace(BaseTraceInstance):
service_account = session.query(Account).filter(Account.id == app.created_by).first()
if not service_account:
raise ValueError(f"Creator account with id {app.created_by} not found for app {app_id}")
current_tenant = (
session.query(TenantAccountJoin).filter_by(account_id=service_account.id, current=True).first()
)
if not current_tenant:
raise ValueError(f"Current tenant not found for account {service_account.id}")
service_account.set_tenant_id(current_tenant.tenant_id)

workflow_node_execution_repository = SQLAlchemyWorkflowNodeExecutionRepository(
session_factory=session_factory,

+ 1
- 0
api/core/ops/ops_trace_manager.py 查看文件

@@ -487,6 +487,7 @@ class TraceTask:
"file_list": file_list,
"triggered_from": workflow_run.triggered_from,
"user_id": user_id,
"app_id": workflow_run.app_id,
}

workflow_trace_info = WorkflowTraceInfo(

正在加载...
取消
保存