soprano-sdk 0.2.3__tar.gz → 0.2.4__tar.gz

{soprano_sdk-0.2.3 → soprano_sdk-0.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: soprano-sdk
-Version: 0.2.3
+Version: 0.2.4
 Summary: YAML-driven workflow engine with AI agent integration for building conversational SOPs
 Author: Arvind Thangamani
 License: MIT

{soprano_sdk-0.2.3 → soprano_sdk-0.2.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "soprano-sdk"
-version = "0.2.3"
+version = "0.2.4"
 description = "YAML-driven workflow engine with AI agent integration for building conversational SOPs"
 readme = "README.md"
 requires-python = ">=3.12"

{soprano_sdk-0.2.3 → soprano_sdk-0.2.4}/soprano_sdk/authenticators/mfa.py

@@ -128,14 +128,18 @@ class MFANodeConfig:
         )
 
     @classmethod
-    def get_validate_user_input(cls, source_node: str, next_node: str, model_name: str):
+    def get_validate_user_input(cls, source_node: str, next_node: str, mfa_config: dict):
+        model_name = mfa_config['model']
+        max_attempts = mfa_config.get('max_attempts', 3)
+        on_max_attempts_reached = mfa_config.get('on_max_attempts_reached')
+
         input_field_name = f"{source_node}_mfa_input"
-        return dict(
+        node_config = dict(
             id=f"{source_node}_mfa_validate",
             action="collect_input_with_agent",
             description="Collect Input for MFA value",
             field=input_field_name,
-            max_attempts=3,
+            max_attempts=max_attempts,
             validator="soprano_sdk.authenticators.mfa.mfa_validate_user_input",
             agent=dict(
                 name="MFA Input Data Collector",
@@ -164,3 +168,8 @@ class MFANodeConfig:
                 )
             ]
         )
+
+        if on_max_attempts_reached:
+            node_config['on_max_attempts_reached'] = on_max_attempts_reached
+
+        return node_config

{soprano_sdk-0.2.3 → soprano_sdk-0.2.4}/soprano_sdk/core/engine.py

@@ -209,8 +209,9 @@ class WorkflowEngine:
 
             if mfa_config := step.get('mfa'):
                 mfa_data_collector = MFANodeConfig.get_validate_user_input(
-                    next_node=step_id, model_name=mfa_config['model'],
-                    source_node=step_id
+                    next_node=step_id,
+                    source_node=step_id,
+                    mfa_config=mfa_config
                 )
                 mfa_start = MFANodeConfig.get_call_function_template(
                     source_node=step_id,

{soprano_sdk-0.2.3 → soprano_sdk-0.2.4}/soprano_sdk/nodes/collect_input.py

@@ -92,6 +92,7 @@ class CollectInputStrategy(ActionStrategy):
         self.field = step_config.get('field')
         self.agent_config = step_config.get('agent', {})
         self.max_attempts = step_config.get('retry_limit') or engine_context.get_config_value("max_retry_limit", DEFAULT_MAX_ATTEMPTS)
+        self.on_max_attempts_reached = step_config.get('on_max_attempts_reached')
         self.transitions = self._get_transitions()
         self.next_step = self.step_config.get("next", None)
         self.is_structured_output = self.agent_config.get("structured_output", {}).get("enabled", False)
@@ -272,7 +273,10 @@ class CollectInputStrategy(ActionStrategy):
     def _handle_max_attempts(self, state: Dict[str, Any]) -> Dict[str, Any]:
         logger.warning(f"Max attempts reached for field '{self.field}'")
         self._set_status(state, 'max_attempts')
-        message = MAX_ATTEMPTS_MESSAGE.format(field=self.field)
+        if self.on_max_attempts_reached:
+            message = self.on_max_attempts_reached
+        else:
+            message = MAX_ATTEMPTS_MESSAGE.format(field=self.field)
         state[WorkflowKeys.MESSAGES] = [message]
         return state
 

{soprano_sdk-0.2.3 → soprano_sdk-0.2.4}/soprano_sdk/validation/schema.py

@@ -90,6 +90,10 @@ WORKFLOW_SCHEMA = {
                         "maximum": 20,
                         "description": "Maximum attempts (for collect_input_with_agent)"
                     },
+                    "on_max_attempts_reached": {
+                        "type": "string",
+                        "description": "Custom error message to display when max attempts are exhausted (for collect_input_with_agent)"
+                    },
                     "agent": {
                         "type": "object",
                         "description": "Agent configuration (for collect_input_with_agent)",
@@ -245,6 +249,16 @@ WORKFLOW_SCHEMA = {
                                 "type": "object",
                                 "description": "MFA payload data that is posted to the RESTAPI, Apart from the properties provided transactionId is sent by the framework in the post payload as an additional property, transactionId is the same throughout the MFA process",
                                 "additionalProperties": True
+                            },
+                            "max_attempts": {
+                                "type": "integer",
+                                "minimum": 1,
+                                "maximum": 20,
+                                "description": "Maximum number of attempts allowed for MFA validation (default: 3)"
+                            },
+                            "on_max_attempts_reached": {
+                                "type": "string",
+                                "description": "Custom error message to display when MFA max attempts are exhausted"
                             }
                         }
                     },

soprano_sdk-0.2.4/tests/test_collect_input_refactor.py

@@ -0,0 +1,66 @@
+import pytest
+from unittest.mock import MagicMock
+from soprano_sdk.nodes.collect_input import CollectInputStrategy, _get_agent_response
+from soprano_sdk.core.constants import WorkflowKeys
+
+class TestCollectInputStrategyRefactor:
+    @pytest.fixture
+    def strategy(self):
+        step_config = {
+            "field": "test_field",
+            "agent": {"name": "test_agent"}
+        }
+        engine_context = MagicMock()
+        engine_context.get_config_value.return_value = "history_based"
+        return CollectInputStrategy(step_config, engine_context)
+
+    def test_get_agent_response_success(self, strategy):
+        agent = MagicMock()
+        conversation = [{"role": "user", "content": "hello"}]
+
+        mock_response = "response content"
+        agent.invoke.return_value = mock_response
+
+        response = _get_agent_response(agent, conversation)
+
+        assert response == "response content"
+        assert conversation[-1]["role"] == "assistant"
+        assert conversation[-1]["content"] == "response content"
+
+    def test_handle_max_attempts_default_message(self):
+        """Test that default error message is used when on_max_attempts_reached is not provided"""
+        step_config = {
+            "field": "customer_name",
+            "agent": {"name": "test_agent"}
+        }
+        engine_context = MagicMock()
+        engine_context.get_config_value.return_value = "history_based"
+
+        strategy = CollectInputStrategy(step_config, engine_context)
+        state = {}
+
+        result = strategy._handle_max_attempts(state)
+
+        assert result[WorkflowKeys.STATUS] == "max_attempts"
+        assert "customer_name" in result[WorkflowKeys.MESSAGES][0]
+        assert "customer service" in result[WorkflowKeys.MESSAGES][0].lower()
+
+    def test_handle_max_attempts_custom_message(self):
+        """Test that custom error message is used when on_max_attempts_reached is provided"""
+        custom_message = "Custom error: Too many attempts. Please call 1-800-SUPPORT."
+        step_config = {
+            "field": "customer_name",
+            "agent": {"name": "test_agent"},
+            "on_max_attempts_reached": custom_message
+        }
+        engine_context = MagicMock()
+        engine_context.get_config_value.return_value = "history_based"
+
+        strategy = CollectInputStrategy(step_config, engine_context)
+        state = {}
+
+        result = strategy._handle_max_attempts(state)
+
+        assert result[WorkflowKeys.STATUS] == "max_attempts"
+        assert result[WorkflowKeys.MESSAGES][0] == custom_message
+        assert "1-800-SUPPORT" in result[WorkflowKeys.MESSAGES][0]

soprano_sdk-0.2.3/tests/test_collect_input_refactor.py

@@ -1,27 +0,0 @@
-import pytest
-from unittest.mock import MagicMock
-from soprano_sdk.nodes.collect_input import CollectInputStrategy, _get_agent_response
-
-class TestCollectInputStrategyRefactor:
-    @pytest.fixture
-    def strategy(self):
-        step_config = {
-            "field": "test_field",
-            "agent": {"name": "test_agent"}
-        }
-        engine_context = MagicMock()
-        engine_context.get_config_value.return_value = "history_based"
-        return CollectInputStrategy(step_config, engine_context)
-
-    def test_get_agent_response_success(self, strategy):
-        agent = MagicMock()
-        conversation = [{"role": "user", "content": "hello"}]
-        
-        mock_response = "response content"
-        agent.invoke.return_value = mock_response
-        
-        response = _get_agent_response(agent, conversation)
-        
-        assert response == "response content"
-        assert conversation[-1]["role"] == "assistant"
-        assert conversation[-1]["content"] == "response content"