sdg-hub 0.7.3__py3-none-any.whl → 0.8.0__py3-none-any.whl

sdg_hub/core/connectors/registry.py

@@ -0,0 +1,112 @@
+# SPDX-License-Identifier: Apache-2.0
+"""Registry for connector classes."""
+
+import inspect
+
+from ..utils.logger_config import setup_logger
+from .exceptions import ConnectorError
+
+logger = setup_logger(__name__)
+
+
+class ConnectorRegistry:
+    """Global registry for connector classes.
+
+    Simple registry for registering and retrieving connectors by name.
+
+    Example
+    -------
+    >>> @ConnectorRegistry.register("my_connector")
+    ... class MyConnector(BaseConnector):
+    ...     pass
+    ...
+    >>> connector_class = ConnectorRegistry.get("my_connector")
+    """
+
+    _connectors: dict[str, type] = {}
+
+    @classmethod
+    def register(cls, name: str):
+        """Register a connector class.
+
+        Parameters
+        ----------
+        name : str
+            Name under which to register the connector.
+
+        Returns
+        -------
+        callable
+            Decorator function that registers the class.
+
+        Example
+        -------
+        >>> @ConnectorRegistry.register("langflow")
+        ... class LangflowConnector(BaseAgentConnector):
+        ...     pass
+        """
+
+        def decorator(connector_class: type) -> type:
+            # Validate the class
+            if not inspect.isclass(connector_class):
+                raise ConnectorError(f"Expected a class, got {type(connector_class)}")
+
+            # Check for BaseConnector inheritance
+            from .base import BaseConnector
+
+            if not issubclass(connector_class, BaseConnector):
+                raise ConnectorError(
+                    f"Connector class '{connector_class.__name__}' "
+                    "must inherit from BaseConnector"
+                )
+
+            cls._connectors[name] = connector_class
+            logger.debug(f"Registered connector '{name}' ({connector_class.__name__})")
+
+            return connector_class
+
+        return decorator
+
+    @classmethod
+    def get(cls, name: str) -> type:
+        """Get a connector class by name.
+
+        Parameters
+        ----------
+        name : str
+            Name of the connector to retrieve.
+
+        Returns
+        -------
+        type
+            The connector class.
+
+        Raises
+        ------
+        ConnectorError
+            If the connector is not found.
+        """
+        if name not in cls._connectors:
+            available = sorted(cls._connectors.keys())
+            error_msg = f"Connector '{name}' not found."
+            if available:
+                error_msg += f" Available: {', '.join(available)}"
+            raise ConnectorError(error_msg)
+
+        return cls._connectors[name]
+
+    @classmethod
+    def list_all(cls) -> list[str]:
+        """Get all registered connector names.
+
+        Returns
+        -------
+        list[str]
+            Sorted list of all connector names.
+        """
+        return sorted(cls._connectors.keys())
+
+    @classmethod
+    def clear(cls) -> None:
+        """Clear all registered connectors. Primarily for testing."""
+        cls._connectors.clear()

{sdg_hub-0.7.3.dist-info → sdg_hub-0.8.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sdg_hub
-Version: 0.7.3
+Version: 0.8.0
 Summary: Synthetic Data Generation
 Author-email: Red Hat AI Innovation <abhandwa@redhat.com>
 License: Apache-2.0

{sdg_hub-0.7.3.dist-info → sdg_hub-0.8.0.dist-info}/RECORD

@@ -1,10 +1,12 @@
 sdg_hub/__init__.py,sha256=TlkZT40-70urdcWLqv3kupaJj8s-SVgd2QyvlSFwb4A,510
-sdg_hub/_version.py,sha256=gWqiHfJwvZSzPQMjJVXuWX-SvdgzetfD6jyorihjahc,704
+sdg_hub/_version.py,sha256=Rttl-BDadtcW1QzGnNffCWA_Wc9mUKDMOBPZp--Mnsc,704
 sdg_hub/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-sdg_hub/core/__init__.py,sha256=e3BoejbqjYhasf9t__L4qE52lkD9EBjx4o--2kqKdro,460
-sdg_hub/core/blocks/__init__.py,sha256=TztBerg_7Ta8AHX9YM6RwzeoiO42nTHx2rekvMZSRf4,1000
+sdg_hub/core/__init__.py,sha256=puXA9S4n-D9MveT9GhpJ84ACKpAzVmwD32TBX8xR8Z8,719
+sdg_hub/core/blocks/__init__.py,sha256=WXlr-9g47rU6_IDpRFxvsPyImfNEBbhvx5JRi3Z3otY,1048
 sdg_hub/core/blocks/base.py,sha256=LgRGvlMqZrxQYUcS_-9HBN_8FZFY1I3O4SBzh0iOCQc,13201
 sdg_hub/core/blocks/registry.py,sha256=FuEN_pnq-nSH1LguY3_oCubT6Kz3SuJjk3TcUpLT-lw,10695
+sdg_hub/core/blocks/agent/__init__.py,sha256=6Vf64AfzPYL8rBbatPkXYrLjgvq-eyhsp5nOatbm02g,162
+sdg_hub/core/blocks/agent/agent_block.py,sha256=3R9tO3YYtEsNO20vQ_2MjdeqcqlCthxlvXxNhWlIw2U,11942
 sdg_hub/core/blocks/filtering/__init__.py,sha256=isxSVSvDqkMjG8dQSl3Q2M4g5c1t9fTjBSA21icf-yA,275
 sdg_hub/core/blocks/filtering/column_value_filter.py,sha256=nT4vBFvi0Q8KoNvh2LkVWRGRp7HkiiLBcPPOGNZF6ig,6528
 sdg_hub/core/blocks/llm/__init__.py,sha256=GTyXZHNYoiyaajx600ouKIrwAg7aS8QjcgkDlThzbag,805
@@ -18,9 +20,18 @@ sdg_hub/core/blocks/transform/duplicate_columns.py,sha256=vK853XsNh62TAcanK8ioVt
 sdg_hub/core/blocks/transform/index_based_mapper.py,sha256=yN0i7kxv9Wn5VGWa1INMry6US3K9FS-jrYrb2GP1-BI,8943
 sdg_hub/core/blocks/transform/json_structure_block.py,sha256=vVnKCdE_Qs6bFoLG5mPp6LFXZf0UhucwGj_6r79nhFM,5104
 sdg_hub/core/blocks/transform/melt_columns.py,sha256=a6UjTKizsBP2CX8Opp0FmuembGXUJ8b0NGi5lZjmPm8,4418
-sdg_hub/core/blocks/transform/rename_columns.py,sha256=QCPD2vK7Xp6WD_tJrupAFtkMcn98xUK7Rt3trPqmTtE,3185
+sdg_hub/core/blocks/transform/rename_columns.py,sha256=tvxW9qMM3fJ7vHyxwkfsbcSeTD2eFP1MQfPhQkspA_E,3590
 sdg_hub/core/blocks/transform/text_concat.py,sha256=E4DoRDeIUgWuF-0a1vE8DpxPpBhQGgXW7wDWAKLSKdg,3124
 sdg_hub/core/blocks/transform/uniform_col_val_setter.py,sha256=mqDG5h0p2qfzWFLlxzOCkDQzYgVUVgKW4PLxKtuon3Q,3274
+sdg_hub/core/connectors/__init__.py,sha256=V66HGS2wGVMYRwjrHHIm5JXxIzTwPTpSimCp2SczrCA,1236
+sdg_hub/core/connectors/base.py,sha256=OLeKrd9QYtvqaAIsc4Bm_f1jUv9lblUkvlixjiyw-EY,2794
+sdg_hub/core/connectors/exceptions.py,sha256=4-L9vQzQfCnymjV6l42fNPtfTCFtxIc4R-6aWJ4bwMk,1089
+sdg_hub/core/connectors/registry.py,sha256=U5OQWi2waiGMeSZAhU_LzADtGvFQFttqZaQk0IjH3RQ,2991
+sdg_hub/core/connectors/agent/__init__.py,sha256=5rUo5fPZ7-8TkyRoWSD5XGdZA2i_X8rElgHvZgzyUn8,221
+sdg_hub/core/connectors/agent/base.py,sha256=GtmRCCof-mo0RzXHJYTFZO7Fc0L9HBugQhe7SacCzdo,6964
+sdg_hub/core/connectors/agent/langflow.py,sha256=DJzIjSnkeDJj_oHSxX9R7IqorARiKV4KUKk8pxCdY5I,4348
+sdg_hub/core/connectors/http/__init__.py,sha256=au0vfsWmNJB7UsHFm0Hsc_JvKoPeDTuYC-SdSQJ2VBM,133
+sdg_hub/core/connectors/http/client.py,sha256=0elw5pOr9akocAF2ZNbLLKxc8dy41BP4kSr2tBd7GR8,4782
 sdg_hub/core/flow/__init__.py,sha256=0_m_htuZfPxk8xQ9IKfp0Pz-JRE4O7lYMUFrKyLNoLA,409
 sdg_hub/core/flow/base.py,sha256=o7OBBH4LkMdJrvYbf3kX6NSAb6U8jQFoRXHKSFHjTEs,58278
 sdg_hub/core/flow/checkpointer.py,sha256=MJay3Q5cfRgJDetk82DaMKJ3ZZUYRHxQabEQTxhGukk,11850
@@ -84,8 +95,8 @@ sdg_hub/flows/text_analysis/structured_insights/extract_entities.yaml,sha256=Q_S
 sdg_hub/flows/text_analysis/structured_insights/extract_keywords.yaml,sha256=_nPPMdHnxag_lYbhYUjGJGo-CvRwWvwdGX7cQhdZ1S0,847
 sdg_hub/flows/text_analysis/structured_insights/flow.yaml,sha256=E9bx5QvmIwm69KITVwZFgwlwe33nhYbVRPRwrMrD8Xw,4978
 sdg_hub/flows/text_analysis/structured_insights/summarize.yaml,sha256=WXwQak1pF8e1OwnOoI1EHu8QB6iUNW89rfkTdi1Oq54,687
-sdg_hub-0.7.3.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-sdg_hub-0.7.3.dist-info/METADATA,sha256=8Pk7CkXl9klRjqpxMGHZp9gt7CuNozv2hV3gGPkRqow,9614
-sdg_hub-0.7.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-sdg_hub-0.7.3.dist-info/top_level.txt,sha256=TqI7d-HE1n6zkXFkU0nF3A1Ct0P0pBaqI675uFokhx4,8
-sdg_hub-0.7.3.dist-info/RECORD,,
+sdg_hub-0.8.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+sdg_hub-0.8.0.dist-info/METADATA,sha256=yC_UNI988aCntR3Kf-YcRMpqhCixp9j0boYFeBMlqOo,9614
+sdg_hub-0.8.0.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+sdg_hub-0.8.0.dist-info/top_level.txt,sha256=TqI7d-HE1n6zkXFkU0nF3A1Ct0P0pBaqI675uFokhx4,8
+sdg_hub-0.8.0.dist-info/RECORD,,

{sdg_hub-0.7.3.dist-info → sdg_hub-0.8.0.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.9.0)
+Generator: setuptools (80.10.2)
 Root-Is-Purelib: true
 Tag: py3-none-any