youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-04-27 22:18:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat(telemetry): add enterprise OTEL telemetry with gateway, traces, metrics, and logs

Browse Source
This commit is contained in:
GareArc
2026-02-06 01:02:19 -08:00
parent 576eca2113
commit 4e3112bd7f
4 changed files with 45 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
api/core/llm_generator/llm_generator.py
View File

@ -25,6 +25,7 @@ from core.model_runtime.entities.llm_entities import LLMResult
from core.model_runtime.entities.message_entities import PromptMessage, SystemPromptMessage, UserPromptMessage
from core.model_runtime.entities.model_entities import ModelType
from core.model_runtime.errors.invoke import InvokeAuthorizationError, InvokeError
from core.ops.entities.trace_entity import OperationType
from core.ops.utils import measure_time
from core.prompt.utils.prompt_template_parser import PromptTemplateParser
from core.telemetry import TelemetryContext, TelemetryEvent, TraceTaskName
@ -215,7 +216,7 @@ class LLMGenerator:
tenant_id=tenant_id,
user_id=user_id,
app_id=app_id,
operation_type="rule_generate",
operation_type=OperationType.RULE_GENERATE,
instruction=instruction,
generated_output=generated_output,
llm_result=llm_result,
@ -272,7 +273,7 @@ class LLMGenerator:
tenant_id=tenant_id,
user_id=user_id,
app_id=app_id,
operation_type="rule_generate",
operation_type=OperationType.RULE_GENERATE,
instruction=instruction,
generated_output="",
llm_result=llm_result,
@ -338,7 +339,7 @@ class LLMGenerator:
tenant_id=tenant_id,
user_id=user_id,
app_id=app_id,
operation_type="rule_generate",
operation_type=OperationType.RULE_GENERATE,
instruction=instruction,
generated_output=str(generated_output) if generated_output else "",
llm_result=llm_result,
@ -409,7 +410,7 @@ class LLMGenerator:
tenant_id=tenant_id,
user_id=user_id,
app_id=app_id,
operation_type="code_generate",
operation_type=OperationType.CODE_GENERATE,
instruction=instruction,
generated_output=result.get("code", ""),
llm_result=llm_result,
@ -504,7 +505,7 @@ class LLMGenerator:
tenant_id=tenant_id,
user_id=user_id,
app_id=app_id,
operation_type="structured_output",
operation_type=OperationType.STRUCTURED_OUTPUT,
instruction=instruction,
generated_output=result.get("output", ""),
llm_result=llm_result,
@ -736,7 +737,7 @@ class LLMGenerator:
tenant_id=tenant_id,
user_id=user_id,
app_id=app_id,
operation_type="instruction_modify",
operation_type=OperationType.INSTRUCTION_MODIFY,
instruction=instruction,
generated_output=generated_output,
llm_result=llm_result,
@ -753,7 +754,7 @@ class LLMGenerator:
tenant_id: str,
user_id: str,
app_id: str | None,
operation_type: str,
operation_type: OperationType,
instruction: str,
generated_output: str,
llm_result: LLMResult | None,

16
api/core/ops/entities/trace_entity.py
View File

@ -211,6 +211,22 @@ trace_info_info_map = {
}
class OperationType(StrEnum):
"""Operation type for token metric labels.
Used as a metric attribute on ``dify.tokens.input`` / ``dify.tokens.output``
counters so consumers can break down token usage by operation.
"""
WORKFLOW = "workflow"
NODE_EXECUTION = "node_execution"
MESSAGE = "message"
RULE_GENERATE = "rule_generate"
CODE_GENERATE = "code_generate"
STRUCTURED_OUTPUT = "structured_output"
INSTRUCTION_MODIFY = "instruction_modify"
class TraceTaskName(StrEnum):
CONVERSATION_TRACE = "conversation"
WORKFLOW_TRACE = "workflow"