瀏覽代碼

Revert "feat: add langfuse llm node input and output" (#16947)

tags/1.2.0
crazywoola 7 月之前
父節點
當前提交
ea1d459423
沒有連結到貢獻者的電子郵件帳戶。
共有 2 個檔案被更改,包括 2 行新增25 行删除
  1. 1
    24
      api/core/ops/langfuse_trace/langfuse_trace.py
  2. 1
    1
      api/tasks/ops_trace_task.py

+ 1
- 24
api/core/ops/langfuse_trace/langfuse_trace.py 查看文件

) )
from core.ops.utils import filter_none_values from core.ops.utils import filter_none_values
from extensions.ext_database import db from extensions.ext_database import db
from models.model import EndUser, Message
from models.model import EndUser
from models.workflow import WorkflowNodeExecution from models.workflow import WorkflowNodeExecution


logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)


if process_data and process_data.get("model_mode") == "chat": if process_data and process_data.get("model_mode") == "chat":
total_token = metadata.get("total_tokens", 0) total_token = metadata.get("total_tokens", 0)

# through workflow_run_id get message data
message_data = (
db.session.query(
Message.answer_tokens, # input
Message.message_tokens, # output
)
.filter(Message.workflow_run_id == trace_info.workflow_run_id)
.first()
)

if message_data:
# chatflow data
input_tokens = message_data.message_tokens
output_tokens = message_data.answer_tokens
else:
# workflow data
input_tokens = json.loads(node_execution.outputs).get("usage", {}).get("prompt_tokens", 0)
output_tokens = json.loads(node_execution.outputs).get("usage", {}).get("completion_tokens", 0)

# add generation # add generation
generation_usage = GenerationUsage( generation_usage = GenerationUsage(
total=total_token, total=total_token,
input=input_tokens,
output=output_tokens,
unit=UnitEnum.TOKENS,
) )


node_generation_data = LangfuseGeneration( node_generation_data = LangfuseGeneration(

+ 1
- 1
api/tasks/ops_trace_task.py 查看文件

except Exception: except Exception:
failed_key = f"{OPS_TRACE_FAILED_KEY}_{app_id}" failed_key = f"{OPS_TRACE_FAILED_KEY}_{app_id}"
redis_client.incr(failed_key) redis_client.incr(failed_key)
logging.exception(f"Processing trace tasks failed, app_id: {app_id}")
logging.info(f"Processing trace tasks failed, app_id: {app_id}")
finally: finally:
storage.delete(file_path) storage.delete(file_path)

Loading…
取消
儲存