From 8764e401ea7b9706099ff90d036ebd3a3ee5e692 Mon Sep 17 00:00:00 2001 From: GareArc Date: Thu, 29 Jan 2026 17:09:48 -0800 Subject: [PATCH] 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/ --- api/enterprise/__init__.py | 0 api/enterprise/telemetry/__init__.py | 0 api/enterprise/telemetry/event_handlers.py | 8 ++++++++ api/extensions/ext_enterprise_telemetry.py | 2 +- api/tasks/ops_trace_task.py | 2 +- api/tests/integration_tests/enterprise/__init__.py | 0 api/tests/unit_tests/enterprise/__init__.py | 0 7 files changed, 10 insertions(+), 2 deletions(-) create mode 100644 api/enterprise/__init__.py create mode 100644 api/enterprise/telemetry/__init__.py create mode 100644 api/tests/integration_tests/enterprise/__init__.py create mode 100644 api/tests/unit_tests/enterprise/__init__.py diff --git a/api/enterprise/__init__.py b/api/enterprise/__init__.py new file mode 100644 index 0000000000..e69de29bb2 diff --git a/api/enterprise/telemetry/__init__.py b/api/enterprise/telemetry/__init__.py new file mode 100644 index 0000000000..e69de29bb2 diff --git a/api/enterprise/telemetry/event_handlers.py b/api/enterprise/telemetry/event_handlers.py index f9312b72d4..f8a6dd5909 100644 --- a/api/enterprise/telemetry/event_handlers.py +++ b/api/enterprise/telemetry/event_handlers.py @@ -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: diff --git a/api/extensions/ext_enterprise_telemetry.py b/api/extensions/ext_enterprise_telemetry.py index 5d01a5371a..f2c68c479c 100644 --- a/api/extensions/ext_enterprise_telemetry.py +++ b/api/extensions/ext_enterprise_telemetry.py @@ -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") diff --git a/api/tasks/ops_trace_task.py b/api/tasks/ops_trace_task.py index 925a63d713..f7ae4fbb1f 100644 --- a/api/tasks/ops_trace_task.py +++ b/api/tasks/ops_trace_task.py @@ -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 diff --git a/api/tests/integration_tests/enterprise/__init__.py b/api/tests/integration_tests/enterprise/__init__.py new file mode 100644 index 0000000000..e69de29bb2 diff --git a/api/tests/unit_tests/enterprise/__init__.py b/api/tests/unit_tests/enterprise/__init__.py new file mode 100644 index 0000000000..e69de29bb2