Refactor imports

Author: plutasnyy

uns_mcp/connectors/destination/__init__.py

@@ -1,34 +1,14 @@
 from mcp.server.fastmcp import FastMCP
 
-from connectors.destination.astra import delete_astradb_destination
-from connectors.destination.databricks_vdt import (
-    delete_databricks_delta_table_destination,
-)
-from connectors.destination.databricksvolumes import (
-    delete_databricks_volumes_destination,
-)
-from connectors.destination.destination_tool import (
+from uns_mcp.connectors.destination.destination_tool import (
     create_destination_connector,
     update_destination_connector,
 )
-from connectors.destination.mongo import (
-    delete_mongodb_destination,
-)
-from connectors.destination.neo4j import delete_neo4j_destination
-from connectors.destination.pinecone import delete_pinecone_destination
-from connectors.destination.s3 import delete_s3_destination
-from connectors.destination.weaviate import delete_weaviate_destination
+from uns_mcp.connectors.destination.generic import delete_destination
 
 
 def register_destination_connectors(mcp: FastMCP):
     """Register all destination connector tools with the MCP server."""
     mcp.tool()(create_destination_connector)
     mcp.tool()(update_destination_connector)
-    mcp.tool()(delete_s3_destination)
-    mcp.tool()(delete_weaviate_destination)
-    mcp.tool()(delete_astradb_destination)
-    mcp.tool()(delete_neo4j_destination)
-    mcp.tool()(delete_mongodb_destination)
-    mcp.tool()(delete_databricks_volumes_destination)
-    mcp.tool()(delete_databricks_delta_table_destination)
-    mcp.tool()(delete_pinecone_destination)
+    mcp.tool()(delete_destination)

uns_mcp/connectors/source/__init__.py

@@ -1,39 +1,15 @@
 from mcp.server.fastmcp import FastMCP
 
+from uns_mcp.connectors.source.generic import delete_source
+from uns_mcp.connectors.source.source_tool import (
+    create_source_connector,
+    update_source_connector,
+)
+
 
 def register_source_connectors(mcp: FastMCP):
     """Register all source connector tools with the MCP server."""
-    from .generic import delete_source
 
+    mcp.tool()(create_source_connector)
+    mcp.tool()(update_source_connector)
     mcp.tool()(delete_source)
-
-    from .s3 import create_s3_source, update_s3_source
-
-    # Register S3 source connector tools
-    mcp.tool()(create_s3_source)
-    mcp.tool()(update_s3_source)
-
-    from .azure import create_azure_source, update_azure_source
-
-    mcp.tool()(create_azure_source)
-    mcp.tool()(update_azure_source)
-
-    from .gdrive import create_gdrive_source, update_gdrive_source
-
-    mcp.tool()(create_gdrive_source)
-    mcp.tool()(update_gdrive_source)
-
-    from .onedrive import create_onedrive_source, update_onedrive_source
-
-    mcp.tool()(create_onedrive_source)
-    mcp.tool()(update_onedrive_source)
-
-    from .salesforce import create_salesforce_source, update_salesforce_source
-
-    mcp.tool()(create_salesforce_source)
-    mcp.tool()(update_salesforce_source)
-
-    from .sharepoint import create_sharepoint_source, update_sharepoint_source
-
-    mcp.tool()(create_sharepoint_source)
-    mcp.tool()(update_sharepoint_source)