mirror of https://github.com/langgenius/dify.git
61 lines
1.8 KiB
Python
61 lines
1.8 KiB
Python
"""Community telemetry helpers.
|
|
|
|
Provides ``emit()`` which enqueues trace events into the CE trace pipeline
|
|
(``TraceQueueManager`` → ``ops_trace`` Celery queue → Langfuse / LangSmith / etc.).
|
|
|
|
Enterprise-only traces (node execution, draft node execution, prompt generation)
|
|
are silently dropped when enterprise telemetry is disabled.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
from core.ops.entities.trace_entity import TraceTaskName
|
|
from core.telemetry.events import TelemetryContext, TelemetryEvent
|
|
|
|
if TYPE_CHECKING:
|
|
from core.ops.ops_trace_manager import TraceQueueManager
|
|
|
|
_ENTERPRISE_ONLY_TRACES: frozenset[TraceTaskName] = frozenset(
|
|
{
|
|
TraceTaskName.DRAFT_NODE_EXECUTION_TRACE,
|
|
TraceTaskName.NODE_EXECUTION_TRACE,
|
|
TraceTaskName.PROMPT_GENERATION_TRACE,
|
|
}
|
|
)
|
|
|
|
|
|
def _is_enterprise_telemetry_enabled() -> bool:
|
|
try:
|
|
from enterprise.telemetry.exporter import is_enterprise_telemetry_enabled
|
|
|
|
return is_enterprise_telemetry_enabled()
|
|
except Exception:
|
|
return False
|
|
|
|
|
|
def emit(event: TelemetryEvent, trace_manager: TraceQueueManager | None = None) -> None:
|
|
from core.ops.ops_trace_manager import TraceQueueManager as LocalTraceQueueManager
|
|
from core.ops.ops_trace_manager import TraceTask
|
|
|
|
if event.name in _ENTERPRISE_ONLY_TRACES and not _is_enterprise_telemetry_enabled():
|
|
return
|
|
|
|
queue_manager = trace_manager or LocalTraceQueueManager(
|
|
app_id=event.context.app_id,
|
|
user_id=event.context.user_id,
|
|
)
|
|
queue_manager.add_trace_task(TraceTask(event.name, **event.payload))
|
|
|
|
|
|
is_enterprise_telemetry_enabled = _is_enterprise_telemetry_enabled
|
|
|
|
__all__ = [
|
|
"TelemetryContext",
|
|
"TelemetryEvent",
|
|
"TraceTaskName",
|
|
"emit",
|
|
"is_enterprise_telemetry_enabled",
|
|
]
|