Merge pull request #336 from conductor-oss/feature_tags_field_for_event_handler

Tags field for event handler

Author: EugeneKisel-sm

src/conductor/asyncio_client/configuration/configuration.py

@@ -203,7 +203,7 @@ def __init__(
             "ca_cert_data": ca_cert_data or os.getenv("CONDUCTOR_SSL_CA_CERT_DATA"),
             "debug": debug,
         }
-        
+
         # Add SSL parameters if they exist in HttpConfiguration
         if cert_file or os.getenv("CONDUCTOR_CERT_FILE"):
             http_config_kwargs["cert_file"] = cert_file or os.getenv("CONDUCTOR_CERT_FILE")
@@ -229,6 +229,12 @@ def __init__(
         if self.proxy_headers:
             self._http_config.proxy_headers = self.proxy_headers
 
+        # Set proxy configuration on the HTTP config
+        if self.proxy:
+            self._http_config.proxy = self.proxy
+        if self.proxy_headers:
+            self._http_config.proxy_headers = self.proxy_headers
+
         # Debug switch and logging setup
         self.__debug = debug
         if self.__debug:

src/conductor/asyncio_client/event/event_client.py

@@ -1,6 +1,11 @@
+from __future__ import annotations
+
+from typing import List
+
 from conductor.asyncio_client.adapters.api.event_resource_api import (
     EventResourceApiAdapter,
 )
+from conductor.asyncio_client.adapters.models.tag_adapter import TagAdapter
 from conductor.asyncio_client.adapters import ApiClient
 from conductor.shared.event.configuration import QueueConfiguration
 
@@ -34,3 +39,91 @@ async def put_queue_configuration(self, queue_configuration: QueueConfiguration)
             queue_name=queue_configuration.queue_name,
             queue_type=queue_configuration.queue_type,
         )
+
+    async def get_event_handler_tags(self, name: str) -> List[TagAdapter]:
+        """Get tags for an event handler.
+
+        Retrieves all tags associated with a specific event handler.
+        Tags are used for organizing and categorizing event handlers.
+
+        Parameters:
+        -----------
+        name : str
+            The name of the event handler
+
+        Returns:
+        --------
+        List[TagAdapter]
+            List of tags associated with the event handler
+
+        Example:
+        --------
+        ```python
+        # Get tags for an event handler
+        tags = await event_client.get_event_handler_tags("workflow_trigger")
+        for tag in tags:
+            print(f"Tag: {tag.key} = {tag.value}")
+        ```
+        """
+        return await self.client.get_tags_for_event_handler(name=name)
+
+    async def add_event_handler_tag(self, name: str, tags: List[TagAdapter]) -> None:
+        """Add tags to an event handler.
+
+        Associates one or more tags with an event handler for organization and categorization.
+
+        Parameters:
+        -----------
+        name : str
+            The name of the event handler
+        tags : List[TagAdapter]
+            List of tags to add to the event handler
+
+        Example:
+        --------
+        ```python
+        from conductor.asyncio_client.adapters.models.tag_adapter import TagAdapter
+
+        # Add tags to an event handler
+        tags = [
+            TagAdapter(key="environment", value="production"),
+            TagAdapter(key="team", value="platform"),
+            TagAdapter(key="priority", value="high")
+        ]
+
+        await event_client.add_event_handler_tag("workflow_trigger", tags)
+        ```
+        """
+        # Note: Async API uses (name=name, tag=tags) keyword args to match the server signature.
+        # Sync API uses (tags, name) positional args due to swagger-codegen parameter ordering.
+        return await self.client.put_tag_for_event_handler(name=name, tag=tags)
+
+    async def remove_event_handler_tag(self, name: str, tags: List[TagAdapter]) -> None:
+        """Remove tags from an event handler.
+
+        Removes one or more tags from an event handler.
+
+        Parameters:
+        -----------
+        name : str
+            The name of the event handler
+        tags : List[TagAdapter]
+            List of tags to remove from the event handler
+
+        Example:
+        --------
+        ```python
+        from conductor.asyncio_client.adapters.models.tag_adapter import TagAdapter
+
+        # Remove specific tags from an event handler
+        tags_to_remove = [
+            TagAdapter(key="environment", value="production"),
+            TagAdapter(key="priority", value="high")
+        ]
+
+        await event_client.remove_event_handler_tag("workflow_trigger", tags_to_remove)
+        ```
+        """
+        # Note: Async API uses (name=name, tag=tags) keyword args to match the server signature.
+        # Sync API uses (tags, name) positional args due to swagger-codegen parameter ordering.
+        return await self.client.delete_tag_for_event_handler(name=name, tag=tags)

src/conductor/asyncio_client/orkes/orkes_base_client.py

@@ -4,6 +4,10 @@
     ApplicationResourceApiAdapter
 from conductor.asyncio_client.adapters.api.authorization_resource_api import \
     AuthorizationResourceApiAdapter
+from conductor.asyncio_client.adapters.api.event_execution_resource_api import \
+    EventExecutionResourceApiAdapter
+from conductor.asyncio_client.adapters.api.event_resource_api import \
+    EventResourceApiAdapter
 from conductor.asyncio_client.adapters.api.group_resource_api import \
     GroupResourceApiAdapter
 from conductor.asyncio_client.adapters.api.integration_resource_api import \
@@ -67,3 +71,5 @@ def __init__(self, configuration: Configuration, api_client: ApiClient):
         self.integration_api = IntegrationResourceApiAdapter(self.api_client)
         self.prompt_api = PromptResourceApiAdapter(self.api_client)
         self.schema_api = SchemaResourceApiAdapter(self.api_client)
+        self.event_api = EventResourceApiAdapter(self.api_client)
+        self.event_execution_api = EventExecutionResourceApiAdapter(self.api_client)

src/conductor/asyncio_client/orkes/orkes_clients.py

@@ -6,6 +6,7 @@
 from conductor.asyncio_client.adapters import ApiClient
 from conductor.asyncio_client.orkes.orkes_authorization_client import \
     OrkesAuthorizationClient
+from conductor.asyncio_client.orkes.orkes_event_client import OrkesEventClient
 from conductor.asyncio_client.orkes.orkes_integration_client import \
     OrkesIntegrationClient
 from conductor.asyncio_client.orkes.orkes_metadata_client import \
@@ -273,6 +274,26 @@ def get_schema_client(self) -> OrkesSchemaClient:
         """
         return OrkesSchemaClient(self.configuration, self.api_client)
 
+    def get_event_client(self) -> OrkesEventClient:
+        """
+        Create and return an event management client.
+
+        The event client manages event handlers and event processing within the
+        Conductor platform, allowing you to create, configure, and monitor
+        event-driven workflows and integrations.
+
+        Returns:
+        --------
+        OrkesEventClient
+            Client for event operations including:
+            - Creating and managing event handlers
+            - Configuring event processing rules
+            - Monitoring event executions
+            - Managing event handler tags and metadata
+            - Configuring queue settings for event processing
+        """
+        return OrkesEventClient(self.configuration, self.api_client)
+
     def get_workflow_executor(self) -> AsyncWorkflowExecutor:
         """
         Create and return an asynchronous workflow executor.