wehub-resource-sync
164 lines
4.4 KiB
Python
164 lines
4.4 KiB
Python
"""Tests for deerflow.tracing.metadata.build_langfuse_trace_metadata."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from deerflow.trace_context import request_trace_context
|
|
from deerflow.tracing import metadata as tracing_metadata
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def _clear_tracing_env(monkeypatch):
|
|
from deerflow.config.tracing_config import reset_tracing_config
|
|
|
|
for name in (
|
|
"LANGFUSE_TRACING",
|
|
"LANGFUSE_PUBLIC_KEY",
|
|
"LANGFUSE_SECRET_KEY",
|
|
"LANGFUSE_BASE_URL",
|
|
"LANGSMITH_TRACING",
|
|
"LANGCHAIN_TRACING_V2",
|
|
"LANGCHAIN_TRACING",
|
|
"LANGSMITH_API_KEY",
|
|
"LANGCHAIN_API_KEY",
|
|
):
|
|
monkeypatch.delenv(name, raising=False)
|
|
reset_tracing_config()
|
|
yield
|
|
reset_tracing_config()
|
|
|
|
|
|
def _enable_langfuse(monkeypatch):
|
|
monkeypatch.setenv("LANGFUSE_TRACING", "true")
|
|
monkeypatch.setenv("LANGFUSE_PUBLIC_KEY", "pk-lf-test")
|
|
monkeypatch.setenv("LANGFUSE_SECRET_KEY", "sk-lf-test")
|
|
|
|
|
|
def test_returns_empty_when_langfuse_disabled(monkeypatch):
|
|
# No env vars set → langfuse not in enabled providers.
|
|
result = tracing_metadata.build_langfuse_trace_metadata(
|
|
thread_id="t-1",
|
|
user_id="u-1",
|
|
assistant_id="lead-agent",
|
|
model_name="gpt-4o",
|
|
)
|
|
assert result == {}
|
|
|
|
|
|
def test_session_id_maps_to_thread_id(monkeypatch):
|
|
_enable_langfuse(monkeypatch)
|
|
|
|
result = tracing_metadata.build_langfuse_trace_metadata(
|
|
thread_id="thread-abc",
|
|
user_id="user-42",
|
|
)
|
|
|
|
assert result["langfuse_session_id"] == "thread-abc"
|
|
|
|
|
|
def test_user_id_falls_back_to_default(monkeypatch):
|
|
_enable_langfuse(monkeypatch)
|
|
|
|
result = tracing_metadata.build_langfuse_trace_metadata(
|
|
thread_id="thread-abc",
|
|
user_id=None,
|
|
)
|
|
|
|
assert result["langfuse_user_id"] == "default"
|
|
|
|
|
|
def test_user_id_explicit_value_wins(monkeypatch):
|
|
_enable_langfuse(monkeypatch)
|
|
|
|
result = tracing_metadata.build_langfuse_trace_metadata(
|
|
thread_id="thread-abc",
|
|
user_id="alice@example.com",
|
|
)
|
|
|
|
assert result["langfuse_user_id"] == "alice@example.com"
|
|
|
|
|
|
def test_trace_name_uses_assistant_id_when_provided(monkeypatch):
|
|
_enable_langfuse(monkeypatch)
|
|
|
|
result = tracing_metadata.build_langfuse_trace_metadata(
|
|
thread_id="t",
|
|
assistant_id="custom-agent",
|
|
)
|
|
|
|
assert result["langfuse_trace_name"] == "custom-agent"
|
|
|
|
|
|
def test_trace_name_defaults_to_lead_agent(monkeypatch):
|
|
_enable_langfuse(monkeypatch)
|
|
|
|
result = tracing_metadata.build_langfuse_trace_metadata(
|
|
thread_id="t",
|
|
assistant_id=None,
|
|
)
|
|
|
|
assert result["langfuse_trace_name"] == "lead-agent"
|
|
|
|
|
|
def test_tags_include_env_and_model(monkeypatch):
|
|
_enable_langfuse(monkeypatch)
|
|
|
|
result = tracing_metadata.build_langfuse_trace_metadata(
|
|
thread_id="t",
|
|
environment="production",
|
|
model_name="gpt-4o",
|
|
)
|
|
|
|
assert result["langfuse_tags"] == ["env:production", "model:gpt-4o"]
|
|
|
|
|
|
def test_tags_omitted_when_no_tag_inputs(monkeypatch):
|
|
_enable_langfuse(monkeypatch)
|
|
|
|
result = tracing_metadata.build_langfuse_trace_metadata(
|
|
thread_id="t",
|
|
user_id="u",
|
|
)
|
|
|
|
assert "langfuse_tags" not in result
|
|
|
|
|
|
def test_thread_id_none_still_produces_metadata(monkeypatch):
|
|
# Stateless run paths may not have a thread_id — we still want
|
|
# user_id / trace_name to flow through so Users page works.
|
|
_enable_langfuse(monkeypatch)
|
|
|
|
result = tracing_metadata.build_langfuse_trace_metadata(
|
|
thread_id=None,
|
|
user_id="u-1",
|
|
)
|
|
|
|
assert result["langfuse_session_id"] is None
|
|
assert result["langfuse_user_id"] == "u-1"
|
|
|
|
|
|
def test_deerflow_trace_id_comes_from_current_trace_context(monkeypatch):
|
|
_enable_langfuse(monkeypatch)
|
|
|
|
with request_trace_context("gateway-trace-1"):
|
|
result = tracing_metadata.build_langfuse_trace_metadata(
|
|
thread_id="thread-abc",
|
|
user_id="user-42",
|
|
)
|
|
|
|
assert result["deerflow_trace_id"] == "gateway-trace-1"
|
|
|
|
|
|
def test_deerflow_trace_id_explicit_argument_wins(monkeypatch):
|
|
_enable_langfuse(monkeypatch)
|
|
|
|
with request_trace_context("ambient-trace"):
|
|
result = tracing_metadata.build_langfuse_trace_metadata(
|
|
thread_id="thread-abc",
|
|
user_id="user-42",
|
|
deerflow_trace_id="explicit-trace",
|
|
)
|
|
|
|
assert result["deerflow_trace_id"] == "explicit-trace"
|