From 67a686cf9885c0d24d37eb31bc563f04bff6e571 Mon Sep 17 00:00:00 2001 From: Ritoban Dutta <124308320+ritoban23@users.noreply.github.com> Date: Mon, 15 Sep 2025 07:15:35 +0530 Subject: [PATCH] [Chore/Refactor] use __all__ to specify export member. (#25681) --- api/controllers/console/__init__.py | 187 ++++++++++++------ api/controllers/files/__init__.py | 11 +- api/controllers/inner_api/__init__.py | 15 +- api/controllers/mcp/__init__.py | 9 +- api/controllers/web/__init__.py | 48 +++-- .../unstructured_doc_extractor.py | 2 +- 6 files changed, 192 insertions(+), 80 deletions(-) diff --git a/api/controllers/console/__init__.py b/api/controllers/console/__init__.py index a6df78fa56..e13edf6a37 100644 --- a/api/controllers/console/__init__.py +++ b/api/controllers/console/__init__.py @@ -54,90 +54,90 @@ api.add_resource(AppImportCheckDependenciesApi, "/apps/imports//c # Import other controllers from . import ( - admin, # pyright: ignore[reportUnusedImport] - apikey, # pyright: ignore[reportUnusedImport] - extension, # pyright: ignore[reportUnusedImport] - feature, # pyright: ignore[reportUnusedImport] - init_validate, # pyright: ignore[reportUnusedImport] - ping, # pyright: ignore[reportUnusedImport] - setup, # pyright: ignore[reportUnusedImport] - version, # pyright: ignore[reportUnusedImport] + admin, + apikey, + extension, + feature, + init_validate, + ping, + setup, + version, ) # Import app controllers from .app import ( - advanced_prompt_template, # pyright: ignore[reportUnusedImport] - agent, # pyright: ignore[reportUnusedImport] - annotation, # pyright: ignore[reportUnusedImport] - app, # pyright: ignore[reportUnusedImport] - audio, # pyright: ignore[reportUnusedImport] - completion, # pyright: ignore[reportUnusedImport] - conversation, # pyright: ignore[reportUnusedImport] - conversation_variables, # pyright: ignore[reportUnusedImport] - generator, # pyright: ignore[reportUnusedImport] - mcp_server, # pyright: ignore[reportUnusedImport] - message, # pyright: ignore[reportUnusedImport] - model_config, # pyright: ignore[reportUnusedImport] - ops_trace, # pyright: ignore[reportUnusedImport] - site, # pyright: ignore[reportUnusedImport] - statistic, # pyright: ignore[reportUnusedImport] - workflow, # pyright: ignore[reportUnusedImport] - workflow_app_log, # pyright: ignore[reportUnusedImport] - workflow_draft_variable, # pyright: ignore[reportUnusedImport] - workflow_run, # pyright: ignore[reportUnusedImport] - workflow_statistic, # pyright: ignore[reportUnusedImport] + advanced_prompt_template, + agent, + annotation, + app, + audio, + completion, + conversation, + conversation_variables, + generator, + mcp_server, + message, + model_config, + ops_trace, + site, + statistic, + workflow, + workflow_app_log, + workflow_draft_variable, + workflow_run, + workflow_statistic, ) # Import auth controllers from .auth import ( - activate, # pyright: ignore[reportUnusedImport] - data_source_bearer_auth, # pyright: ignore[reportUnusedImport] - data_source_oauth, # pyright: ignore[reportUnusedImport] - email_register, # pyright: ignore[reportUnusedImport] - forgot_password, # pyright: ignore[reportUnusedImport] - login, # pyright: ignore[reportUnusedImport] - oauth, # pyright: ignore[reportUnusedImport] - oauth_server, # pyright: ignore[reportUnusedImport] + activate, + data_source_bearer_auth, + data_source_oauth, + email_register, + forgot_password, + login, + oauth, + oauth_server, ) # Import billing controllers -from .billing import billing, compliance # pyright: ignore[reportUnusedImport] +from .billing import billing, compliance # Import datasets controllers from .datasets import ( - data_source, # pyright: ignore[reportUnusedImport] - datasets, # pyright: ignore[reportUnusedImport] - datasets_document, # pyright: ignore[reportUnusedImport] - datasets_segments, # pyright: ignore[reportUnusedImport] - external, # pyright: ignore[reportUnusedImport] - hit_testing, # pyright: ignore[reportUnusedImport] - metadata, # pyright: ignore[reportUnusedImport] - website, # pyright: ignore[reportUnusedImport] + data_source, + datasets, + datasets_document, + datasets_segments, + external, + hit_testing, + metadata, + website, ) # Import explore controllers from .explore import ( - installed_app, # pyright: ignore[reportUnusedImport] - parameter, # pyright: ignore[reportUnusedImport] - recommended_app, # pyright: ignore[reportUnusedImport] - saved_message, # pyright: ignore[reportUnusedImport] + installed_app, + parameter, + recommended_app, + saved_message, ) # Import tag controllers -from .tag import tags # pyright: ignore[reportUnusedImport] +from .tag import tags # Import workspace controllers from .workspace import ( - account, # pyright: ignore[reportUnusedImport] - agent_providers, # pyright: ignore[reportUnusedImport] - endpoint, # pyright: ignore[reportUnusedImport] - load_balancing_config, # pyright: ignore[reportUnusedImport] - members, # pyright: ignore[reportUnusedImport] - model_providers, # pyright: ignore[reportUnusedImport] - models, # pyright: ignore[reportUnusedImport] - plugin, # pyright: ignore[reportUnusedImport] - tool_providers, # pyright: ignore[reportUnusedImport] - workspace, # pyright: ignore[reportUnusedImport] + account, + agent_providers, + endpoint, + load_balancing_config, + members, + model_providers, + models, + plugin, + tool_providers, + workspace, ) # Explore Audio @@ -212,3 +212,70 @@ api.add_resource( ) api.add_namespace(console_ns) + +__all__ = [ + "account", + "activate", + "admin", + "advanced_prompt_template", + "agent", + "agent_providers", + "annotation", + "api", + "apikey", + "app", + "audio", + "billing", + "bp", + "completion", + "compliance", + "console_ns", + "conversation", + "conversation_variables", + "data_source", + "data_source_bearer_auth", + "data_source_oauth", + "datasets", + "datasets_document", + "datasets_segments", + "email_register", + "endpoint", + "extension", + "external", + "feature", + "forgot_password", + "generator", + "hit_testing", + "init_validate", + "installed_app", + "load_balancing_config", + "login", + "mcp_server", + "members", + "message", + "metadata", + "model_config", + "model_providers", + "models", + "oauth", + "oauth_server", + "ops_trace", + "parameter", + "ping", + "plugin", + "recommended_app", + "saved_message", + "setup", + "site", + "statistic", + "tags", + "tool_providers", + "version", + "website", + "workflow", + "workflow_app_log", + "workflow_draft_variable", + "workflow_run", + "workflow_statistic", + "workspace", +] diff --git a/api/controllers/files/__init__.py b/api/controllers/files/__init__.py index 26fbf7097e..f8976b86b9 100644 --- a/api/controllers/files/__init__.py +++ b/api/controllers/files/__init__.py @@ -14,6 +14,15 @@ api = ExternalApi( files_ns = Namespace("files", description="File operations", path="/") -from . import image_preview, tool_files, upload # pyright: ignore[reportUnusedImport] +from . import image_preview, tool_files, upload api.add_namespace(files_ns) + +__all__ = [ + "api", + "bp", + "files_ns", + "image_preview", + "tool_files", + "upload", +] diff --git a/api/controllers/inner_api/__init__.py b/api/controllers/inner_api/__init__.py index f29f624ba5..74005217ef 100644 --- a/api/controllers/inner_api/__init__.py +++ b/api/controllers/inner_api/__init__.py @@ -15,8 +15,17 @@ api = ExternalApi( # Create namespace inner_api_ns = Namespace("inner_api", description="Internal API operations", path="/") -from . import mail as _mail # pyright: ignore[reportUnusedImport] -from .plugin import plugin as _plugin # pyright: ignore[reportUnusedImport] -from .workspace import workspace as _workspace # pyright: ignore[reportUnusedImport] +from . import mail as _mail +from .plugin import plugin as _plugin +from .workspace import workspace as _workspace api.add_namespace(inner_api_ns) + +__all__ = [ + "_mail", + "_plugin", + "_workspace", + "api", + "bp", + "inner_api_ns", +] diff --git a/api/controllers/mcp/__init__.py b/api/controllers/mcp/__init__.py index 336a7801bb..d6fb2981e4 100644 --- a/api/controllers/mcp/__init__.py +++ b/api/controllers/mcp/__init__.py @@ -14,6 +14,13 @@ api = ExternalApi( mcp_ns = Namespace("mcp", description="MCP operations", path="/") -from . import mcp # pyright: ignore[reportUnusedImport] +from . import mcp api.add_namespace(mcp_ns) + +__all__ = [ + "api", + "bp", + "mcp", + "mcp_ns", +] diff --git a/api/controllers/web/__init__.py b/api/controllers/web/__init__.py index 97bcd3d53c..1d22954308 100644 --- a/api/controllers/web/__init__.py +++ b/api/controllers/web/__init__.py @@ -16,20 +16,40 @@ api = ExternalApi( web_ns = Namespace("web", description="Web application API operations", path="/") from . import ( - app, # pyright: ignore[reportUnusedImport] - audio, # pyright: ignore[reportUnusedImport] - completion, # pyright: ignore[reportUnusedImport] - conversation, # pyright: ignore[reportUnusedImport] - feature, # pyright: ignore[reportUnusedImport] - files, # pyright: ignore[reportUnusedImport] - forgot_password, # pyright: ignore[reportUnusedImport] - login, # pyright: ignore[reportUnusedImport] - message, # pyright: ignore[reportUnusedImport] - passport, # pyright: ignore[reportUnusedImport] - remote_files, # pyright: ignore[reportUnusedImport] - saved_message, # pyright: ignore[reportUnusedImport] - site, # pyright: ignore[reportUnusedImport] - workflow, # pyright: ignore[reportUnusedImport] + app, + audio, + completion, + conversation, + feature, + files, + forgot_password, + login, + message, + passport, + remote_files, + saved_message, + site, + workflow, ) api.add_namespace(web_ns) + +__all__ = [ + "api", + "app", + "audio", + "bp", + "completion", + "conversation", + "feature", + "files", + "forgot_password", + "login", + "message", + "passport", + "remote_files", + "saved_message", + "site", + "web_ns", + "workflow", +] diff --git a/api/core/rag/extractor/unstructured/unstructured_doc_extractor.py b/api/core/rag/extractor/unstructured/unstructured_doc_extractor.py index 4ed8dfbbd8..5199208f70 100644 --- a/api/core/rag/extractor/unstructured/unstructured_doc_extractor.py +++ b/api/core/rag/extractor/unstructured/unstructured_doc_extractor.py @@ -23,7 +23,7 @@ class UnstructuredWordExtractor(BaseExtractor): unstructured_version = tuple(int(x) for x in __unstructured_version__.split(".")) # check the file extension try: - import magic # noqa: F401 # pyright: ignore[reportUnusedImport] + import magic # noqa: F401 is_doc = detect_filetype(self._file_path) == FileType.DOC except ImportError: