chore: Add __init__.py files for enterprise Python packages

Add package initialization files for:
- enterprise/
- enterprise/telemetry/
- tests/integration_tests/enterprise/
- tests/unit_tests/enterprise/
This commit is contained in:
GareArc
2026-01-29 17:09:48 -08:00
parent 775b28d3c9
commit 8764e401ea
7 changed files with 10 additions and 2 deletions

View File

View File

View File

@ -14,6 +14,14 @@ from events.feedback_event import feedback_was_created
logger = logging.getLogger(__name__)
# Export handlers to mark them as intentionally public (accessed via Blinker decorators)
__all__ = [
"_handle_app_created",
"_handle_app_deleted",
"_handle_app_updated",
"_handle_feedback_created",
]
@app_was_created.connect
def _handle_app_created(sender: object, **kwargs: object) -> None:

View File

@ -39,7 +39,7 @@ def init_app(app: DifyApp) -> None:
atexit.register(_exporter.shutdown)
# Import to trigger @signal.connect decorator registration
import enterprise.telemetry.event_handlers # noqa: F401
import enterprise.telemetry.event_handlers # noqa: F401 # type: ignore[reportUnusedImport]
logger.info("Enterprise telemetry initialized")

View File

@ -43,9 +43,9 @@ def process_trace_tasks(file_info):
if trace_type:
trace_info = trace_type(**trace_info)
# process enterprise trace separately
from enterprise.telemetry.exporter import is_enterprise_telemetry_enabled
if is_enterprise_telemetry_enabled():
from enterprise.telemetry.enterprise_trace import EnterpriseDataTrace