soprano-sdk 0.2.3__py3-none-any.whl → 0.2.4__py3-none-any.whl

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/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/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/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.3.dist-info → soprano_sdk-0.2.4.dist-info}/METADATA

@@ -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.dist-info → soprano_sdk-0.2.4.dist-info}/RECORD

@@ -6,16 +6,16 @@ soprano_sdk/agents/adaptor.py,sha256=Cm02YKFclrESu-Qq4CTknCgU7KaA7Z_2FspnQDkEVfU
 soprano_sdk/agents/factory.py,sha256=Aucfz4rZVKCXMAQtbGAqp1JR8aYwa66mokRmKkKGhYA,6699
 soprano_sdk/agents/structured_output.py,sha256=7DSVzfMPsZAqBwI3v6XL15qG5Gh4jJ-qddcVPaa3gdc,3326
 soprano_sdk/authenticators/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-soprano_sdk/authenticators/mfa.py,sha256=OHbPYsns2LZEbnqod7H_iW0AE9bdJ_74-EiaAmvOoVc,5644
+soprano_sdk/authenticators/mfa.py,sha256=7aTueLu2jLxl213Q5_BkNQ534RgEwf6lodQa_PaJFWs,5980
 soprano_sdk/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 soprano_sdk/core/constants.py,sha256=F7gL3hkpq5mZgbgiBcEERmzxYJ4zZG2ZFx1YQp5Z580,3130
-soprano_sdk/core/engine.py,sha256=RbLmKSq65I8q-vFTBFC_dqozvhQ13X_eRh_aHuEkRT0,11259
+soprano_sdk/core/engine.py,sha256=UTFJimrgUzfdDZgc4rW5nvkOKevAEGW8C5Zr1D0tgcs,11270
 soprano_sdk/core/rollback_strategies.py,sha256=NjDTtBCZlqyDql5PSwI9SMDLK7_BNlTxbW_cq_5gV0g,7783
 soprano_sdk/core/state.py,sha256=k8ojLfWgjES3p9XWMeGU5s4UK-Xa5T8mS4VtZzTrcDw,2961
 soprano_sdk/nodes/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 soprano_sdk/nodes/base.py,sha256=idFyOGGPnjsASYnrOF_NIh7eFcSuJqw61EoVN_WCTaU,2360
 soprano_sdk/nodes/call_function.py,sha256=PdmSSVXkMT5ptxeUjmKqND-wvqbaXp1i61QzIzFOfgQ,5535
-soprano_sdk/nodes/collect_input.py,sha256=IsHXb8J-4Dpjcza4zVLLtzQ82NMlZDgXhpNxtoJ6FCg,23567
+soprano_sdk/nodes/collect_input.py,sha256=lEltZOU5ALvc57q8I_4SjzzEapVejy9mS0E73Jf7-sk,23759
 soprano_sdk/nodes/factory.py,sha256=l-Gysfgnao-o2dphhnbjjxcH3ojZanZNYN3CBH9dDbA,1624
 soprano_sdk/routing/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 soprano_sdk/routing/router.py,sha256=SrNciTIXXdC9bAbbO5bX7PN9mlRbITjr4RZdNm4jEVA,3450
@@ -26,9 +26,9 @@ soprano_sdk/utils/template.py,sha256=MG_B9TMx1ShpnSGo7s7TO-VfQzuFByuRNhJTvZ668kM
 soprano_sdk/utils/tool.py,sha256=hWN826HIKmLdswLCTURLH8hWlb2WU0MB8nIUErbpB-8,1877
 soprano_sdk/utils/tracing.py,sha256=gSHeBDLe-MbAZ9rkzpCoGFveeMdR9KLaA6tteB0IWjk,1991
 soprano_sdk/validation/__init__.py,sha256=ImChmO86jYHU90xzTttto2-LmOUOmvY_ibOQaLRz5BA,262
-soprano_sdk/validation/schema.py,sha256=R6bm4Pzgx8p08o77FN9FFSKTLTKOCf7WuThrR6rV2FI,14578
+soprano_sdk/validation/schema.py,sha256=7B60_8xIt5bLXTYEGEDwU2DwmQn2f4QFyyIca2FYM9s,15420
 soprano_sdk/validation/validator.py,sha256=GaCvHvjwVe88Z8yatQsueiPnqtq1oo5uN75gogzpQT0,8940
-soprano_sdk-0.2.3.dist-info/METADATA,sha256=gZasJ2fAEfHIcgd7wKjyPsPUJv0PdTeP5B06eimF03U,11297
-soprano_sdk-0.2.3.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-soprano_sdk-0.2.3.dist-info/licenses/LICENSE,sha256=A1aBauSjPNtVehOXJe3WuvdU2xvM9H8XmigFMm6665s,1073
-soprano_sdk-0.2.3.dist-info/RECORD,,
+soprano_sdk-0.2.4.dist-info/METADATA,sha256=W-e-Si4pewUkpuWewKz_pmkWvUeRVR8HPwQ-Z-7tpTQ,11297
+soprano_sdk-0.2.4.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+soprano_sdk-0.2.4.dist-info/licenses/LICENSE,sha256=A1aBauSjPNtVehOXJe3WuvdU2xvM9H8XmigFMm6665s,1073
+soprano_sdk-0.2.4.dist-info/RECORD,,