conductor-python 1.3.9__tar.gz → 1.3.10__tar.gz

{conductor_python-1.3.9 → conductor_python-1.3.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: conductor-python
-Version: 1.3.9
+Version: 1.3.10
 Summary: Python SDK for working with https://github.com/conductor-oss/conductor
 License: Apache-2.0
 Author: Orkes

{conductor_python-1.3.9 → conductor_python-1.3.10}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "conductor-python"
-version = "1.3.9" # Do not change! Placeholder. Real version injected during build (edited) 
+version = "1.3.10" # Do not change! Placeholder. Real version injected during build (edited) 
 description = "Python SDK for working with https://github.com/conductor-oss/conductor"
 authors = ["Orkes <developers@orkes.io>"]
 license = "Apache-2.0"

{conductor_python-1.3.9 → conductor_python-1.3.10}/src/conductor/client/http/api/workflow_resource_api.py

@@ -3178,4 +3178,94 @@ class WorkflowResourceApi(object):
             _return_http_data_only=params.get('_return_http_data_only'),
             _preload_content=params.get('_preload_content', True),
             _request_timeout=params.get('_request_timeout'),
+            collection_formats=collection_formats)
+
+    def send_workflow_message(self, body, workflow_id, **kwargs):  # noqa: E501
+        """Push a message into a running workflow's message queue (WMQ).  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.send_workflow_message(body, workflow_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param dict(str, object) body: Arbitrary JSON payload (required)
+        :param str workflow_id: (required)
+        :return: str  — the UUID assigned to the pushed message
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        if kwargs.get('async_req'):
+            return self.send_workflow_message_with_http_info(body, workflow_id, **kwargs)  # noqa: E501
+        else:
+            (data) = self.send_workflow_message_with_http_info(body, workflow_id, **kwargs)  # noqa: E501
+            return data
+
+    def send_workflow_message_with_http_info(self, body, workflow_id, **kwargs):  # noqa: E501
+        """Push a message into a running workflow's message queue (WMQ).  # noqa: E501
+
+        :param async_req bool
+        :param dict(str, object) body: Arbitrary JSON payload (required)
+        :param str workflow_id: (required)
+        :return: str  — the UUID assigned to the pushed message
+        """
+
+        all_params = ['body', 'workflow_id']  # noqa: E501
+        all_params.append('async_req')
+        all_params.append('_return_http_data_only')
+        all_params.append('_preload_content')
+        all_params.append('_request_timeout')
+
+        params = locals()
+        for key, val in six.iteritems(params['kwargs']):
+            if key not in all_params:
+                raise TypeError(
+                    "Got an unexpected keyword argument '%s'"
+                    " to method send_workflow_message" % key
+                )
+            params[key] = val
+        del params['kwargs']
+
+        if ('body' not in params or params['body'] is None):
+            raise ValueError("Missing the required parameter `body` when calling `send_workflow_message`")  # noqa: E501
+        if ('workflow_id' not in params or params['workflow_id'] is None):
+            raise ValueError("Missing the required parameter `workflow_id` when calling `send_workflow_message`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'workflow_id' in params:
+            path_params['workflowId'] = params['workflow_id']  # noqa: E501
+
+        query_params = []
+        header_params = {}
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        if 'body' in params:
+            body_params = params['body']
+
+        header_params['Accept'] = self.api_client.select_header_accept(
+            ['text/plain'])  # noqa: E501
+        header_params['Content-Type'] = self.api_client.select_header_content_type(  # noqa: E501
+            ['application/json'])  # noqa: E501
+
+        auth_settings = ['api_key']  # noqa: E501
+
+        return self.api_client.call_api(
+            '/workflow/{workflowId}/messages', 'POST',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='str',  # noqa: E501
+            auth_settings=auth_settings,
+            async_req=params.get('async_req'),
+            _return_http_data_only=params.get('_return_http_data_only'),
+            _preload_content=params.get('_preload_content', True),
+            _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)

{conductor_python-1.3.9 → conductor_python-1.3.10}/src/conductor/client/http/models/__init__.py

@@ -66,4 +66,5 @@ from conductor.client.http.models.service_method import ServiceMethod
 from conductor.client.http.models.circuit_breaker_transition_response import CircuitBreakerTransitionResponse
 from conductor.client.http.models.signal_response import SignalResponse, TaskStatus
 from conductor.client.http.models.authentication_config import AuthenticationConfig
-from conductor.client.http.models.tag_object import TagObject
+from conductor.client.http.models.tag_object import TagObject
+from conductor.client.http.models.workflow_message import WorkflowMessage

conductor_python-1.3.10/src/conductor/client/http/models/workflow_message.py

@@ -0,0 +1,41 @@
+from dataclasses import dataclass, field
+from typing import Dict, Optional
+
+
+@dataclass
+class WorkflowMessage:
+    """Represents a message pushed into a running workflow's queue (WMQ).
+
+    Attributes:
+        id: UUID assigned by the server on push.
+        workflow_id: The workflow instance that owns this message.
+        payload: Arbitrary JSON payload supplied by the caller.
+        received_at: ISO-8601 UTC timestamp set at ingestion time.
+    """
+
+    id: Optional[str] = field(default=None)
+    workflow_id: Optional[str] = field(default=None)
+    payload: Optional[Dict[str, object]] = field(default=None)
+    received_at: Optional[str] = field(default=None)
+
+    swagger_types = {
+        'id': 'str',
+        'workflow_id': 'str',
+        'payload': 'dict(str, object)',
+        'received_at': 'str',
+    }
+
+    attribute_map = {
+        'id': 'id',
+        'workflow_id': 'workflowId',
+        'payload': 'payload',
+        'received_at': 'receivedAt',
+    }
+
+    def to_dict(self) -> dict:
+        result = {}
+        for attr, _ in self.swagger_types.items():
+            value = getattr(self, attr)
+            if value is not None:
+                result[self.attribute_map[attr]] = value
+        return result

{conductor_python-1.3.9 → conductor_python-1.3.10}/src/conductor/client/orkes/orkes_workflow_client.py

@@ -215,3 +215,17 @@ class OrkesWorkflowClient(OrkesBaseClient, WorkflowClient):
             kwargs["wait_for_seconds"] = wait_for_seconds
 
         return self.workflowResourceApi.update_workflow_and_task_state(update_requesst=update_requesst, workflow_id=workflow_id, **kwargs)
+
+    def send_message(self, workflow_id: str, message: Dict[str, object]) -> str:
+        """Push a message into the message queue of a running workflow (WMQ).
+
+        Requires conductor.workflow-message-queue.enabled=true on the server.
+
+        Args:
+            workflow_id: The running workflow instance ID.
+            message: Arbitrary JSON-serialisable dict to deliver to the workflow.
+
+        Returns:
+            The UUID string assigned to the message by the server.
+        """
+        return self.workflowResourceApi.send_workflow_message(message, workflow_id)

{conductor_python-1.3.9 → conductor_python-1.3.10}/src/conductor/client/workflow/executor/workflow_executor.py

@@ -283,6 +283,25 @@ class WorkflowExecutor:
             body=body
         )
 
+    def send_message(self, workflow_id: str, message: Dict[str, Any]) -> str:
+        """Push a message into the message queue of a running workflow (WMQ).
+
+        The workflow must have a PULL_WORKFLOW_MESSAGES task to consume messages.
+        Requires conductor.workflow-message-queue.enabled=true on the server.
+
+        Args:
+            workflow_id: The running workflow instance ID.
+            message: Arbitrary JSON-serialisable dict to deliver to the workflow.
+
+        Returns:
+            The UUID string assigned to the message by the server.
+
+        Raises:
+            ApiException: 404 if the workflow is not found, 409 if not in RUNNING state,
+                          429 if the queue is at capacity.
+        """
+        return self.workflow_client.send_message(workflow_id, message)
+
     def __get_task_result(self, task_id: str, workflow_id: str, task_output: Dict[str, Any], status: str) -> TaskResult:
         return TaskResult(
             workflow_instance_id=workflow_id,

conductor_python-1.3.10/src/conductor/client/workflow/task/pull_workflow_messages_task.py

@@ -0,0 +1,30 @@
+from __future__ import annotations
+
+from typing_extensions import Self
+
+from conductor.client.workflow.task.task import TaskInterface
+from conductor.client.workflow.task.task_type import TaskType
+
+
+class PullWorkflowMessagesTask(TaskInterface):
+    """Consume messages from the workflow's message queue (WMQ).
+
+    When messages are available the task completes with:
+        output.messages  — list of WorkflowMessage objects
+        output.count     — number of messages returned
+
+    When the queue is empty the task stays IN_PROGRESS and is re-evaluated
+    after ~1 second (non-blocking polling behavior).
+
+    Args:
+        task_ref_name: Unique task reference name within the workflow.
+        batch_size: Maximum number of messages to dequeue per execution (default 1,
+                    server cap is typically 100).
+    """
+
+    def __init__(self, task_ref_name: str, batch_size: int = 1) -> Self:
+        super().__init__(
+            task_reference_name=task_ref_name,
+            task_type=TaskType.PULL_WORKFLOW_MESSAGES,
+        )
+        self.input_parameters["batchSize"] = batch_size

{conductor_python-1.3.9 → conductor_python-1.3.10}/src/conductor/client/workflow/task/task_type.py

@@ -40,3 +40,4 @@ class TaskType(str, Enum):
     LLM_SEARCH_EMBEDDINGS = "LLM_SEARCH_EMBEDDINGS"
     LIST_MCP_TOOLS = "LIST_MCP_TOOLS"
     CALL_MCP_TOOL = "CALL_MCP_TOOL"
+    PULL_WORKFLOW_MESSAGES = "PULL_WORKFLOW_MESSAGES"

{conductor_python-1.3.9 → conductor_python-1.3.10}/src/conductor/client/workflow_client.py

@@ -3,7 +3,7 @@ from abc import ABC, abstractmethod
 from typing import Optional, List, Dict
 
 from conductor.client.http.models import WorkflowRun, SkipTaskRequest, WorkflowStatus, \
-    ScrollableSearchResultWorkflowSummary, SignalResponse
+    ScrollableSearchResultWorkflowSummary, SignalResponse, WorkflowMessage
 from conductor.client.http.models.correlation_ids_search_request import CorrelationIdsSearchRequest
 from conductor.client.http.models.rerun_workflow_request import RerunWorkflowRequest
 from conductor.client.http.models.start_workflow_request import StartWorkflowRequest
@@ -120,3 +120,18 @@ class WorkflowClient(ABC):
     def update_state(self, workflow_id: str, update_requesst: WorkflowStateUpdate,
                      wait_until_task_ref_names: Optional[List[str]] = None, wait_for_seconds: Optional[int] = None) -> WorkflowRun:
         pass
+
+    @abstractmethod
+    def send_message(self, workflow_id: str, message: Dict[str, object]) -> str:
+        """Push a message into the message queue of a running workflow (WMQ).
+
+        Requires conductor.workflow-message-queue.enabled=true on the server.
+
+        Args:
+            workflow_id: The running workflow instance ID.
+            message: Arbitrary JSON-serialisable dict to deliver to the workflow.
+
+        Returns:
+            The UUID string assigned to the message by the server.
+        """
+        pass