soprano-sdk 0.2.5__py3-none-any.whl → 0.2.7__py3-none-any.whl

soprano_sdk/authenticators/mfa.py

@@ -10,6 +10,7 @@ class MFAChallenge(TypedDict):
 class MFAState(TypedDict):
     challengeType: Literal['OTP', 'dob']
     post_payload: dict[str, str]
+    post_headers: NotRequired[dict[str, str]]
     otpValue: NotRequired[str]
     status: Literal['IN_PROGRESS', 'COMPLETED', 'ERRORED', 'FAILED'] | None
     message: str
@@ -35,6 +36,10 @@ def enforce_mfa_if_required(state: dict, mfa_config: Optional[MFAConfig] = None)
     _mfa : MFAState = state['_mfa']
     if _mfa['status'] == 'COMPLETED':
         return True
+
+    # Use custom headers if provided, otherwise empty dict
+    headers = _mfa.get('post_headers', {})
+
     generate_token_response = requests.post(
         build_path(
             base_url=mfa_config.generate_token_base_url,
@@ -42,7 +47,7 @@ def enforce_mfa_if_required(state: dict, mfa_config: Optional[MFAConfig] = None)
         ),
         json=_mfa['post_payload'],
         timeout=mfa_config.api_timeout,
-        headers={"Authorization": f"Bearer {state['bearer_token']}"}
+        headers=headers
     )
     _, error = get_response(generate_token_response)
 
@@ -65,6 +70,9 @@ def mfa_validate_user_input(mfa_config: Optional[MFAConfig] = None, **state: dic
     if not state[input_field_name]:
         return False
 
+    # Use custom headers if provided, otherwise empty dict
+    headers = _mfa.get('post_headers', {})
+
     post_payload = _mfa['post_payload']
     challenge_field_name = f"{_mfa['challengeType'].lower()}Challenge"
     post_payload.update({challenge_field_name: {"value": state[input_field_name]}})
@@ -75,7 +83,7 @@ def mfa_validate_user_input(mfa_config: Optional[MFAConfig] = None, **state: dic
         ),
         json=post_payload,
         timeout=mfa_config.api_timeout,
-        headers={"Authorization": f"Bearer {state['bearer_token']}"}
+        headers=headers
     )
     _mfa['retry_count'] += 1
     response, error = get_response(validate_token_response)
@@ -95,7 +103,7 @@ def mfa_validate_user_input(mfa_config: Optional[MFAConfig] = None, **state: dic
             ),
             json=post_payload,
             timeout=mfa_config.api_timeout,
-            headers={"Authorization": f"Bearer {state['bearer_token']}"}
+            headers=headers
         )
         if authorize.status_code == 204:
             _mfa['status'] = 'COMPLETED'
@@ -146,25 +154,47 @@ class MFANodeConfig:
                 model=model_name,
                 initial_message="{{_mfa.message}}",
                 instructions="""
-                    You are an authentication value extractor. Your job is to identify and extract MFA codes from user input.
+                    You are an authentication value extractor. Your job is to identify and extract MFA codes from user input, or detect if the user wants to cancel the authentication flow.
 
                     **Task:**
-                    - Read the user's message
-                    - Extract ONLY the OTP code value
-                    - Output in the exact format shown below
-
-                    Examples:
-                    * User says: "1234" → `MFA_CAPTURED:1223`
-                    * User says: "2345e" → `MFA_CAPTURED:1223e
+                    - Read the user's message carefully
+                    - First, check if the user wants to cancel, stop, or exit the authentication process
+                    - If they want to cancel, output: MFA_CANCELLED:
+                    - Otherwise, extract ONLY the OTP/MFA code value and output in the format shown below
+
+                    **Cancellation Detection:**
+                    If the user expresses any intent to cancel, stop, exit, abort, or quit the authentication process, respond with: MFA_CANCELLED
+
+                    Examples of cancellation phrases:
+                    * "cancel" → MFA_CANCELLED:
+                    * "I want to stop" → MFA_CANCELLED:
+                    * "exit" → MFA_CANCELLED:
+                    * "nevermind" → MFA_CANCELLED:
+                    * "I don't want to continue" → MFA_CANCELLED:
+                    * "stop this" → MFA_CANCELLED:
+                    * "forget it" → MFA_CANCELLED:
+                    * "abort" → MFA_CANCELLED:
+                    * "quit" → MFA_CANCELLED:
+
+                    **OTP Capture Examples:**
+                    * "1234" → MFA_CAPTURED:1234
+                    * "2345e" → MFA_CAPTURED:2345e
+                    * "the code is 567890" → MFA_CAPTURED:567890
+                    * "my otp is 123456" → MFA_CAPTURED:123456
 
                     **Output Format:**
-                    MFA_CAPTURED:<input_field_name>
+                    - For OTP/MFA codes: MFA_CAPTURED:<otp_value>
+                    - For cancellation: MFA_CANCELLED:
 
             """),
             transitions=[
                 dict(
                     pattern="MFA_CAPTURED:",
                     next=next_node
+                ),
+                dict(
+                    pattern="MFA_CANCELLED:",
+                    next="mfa_cancelled"
                 )
             ]
         )

soprano_sdk/core/constants.py

@@ -114,6 +114,10 @@ class MFAConfig(BaseSettings):
         default=30,
         description="API request timeout in seconds"
     )
+    mfa_cancelled_message: str = Field(
+        default="Authentication has been cancelled.",
+        description="Message to display when user cancels MFA authentication"
+    )
 
     model_config = SettingsConfigDict(
         case_sensitive=False,

soprano_sdk/core/engine.py

@@ -39,8 +39,7 @@ class WorkflowEngine:
             self.step_map = {step['id']: step for step in self.steps}
             self.mfa_config = (mfa_config or MFAConfig()) if self.mfa_validator_steps else None
             self.data_fields = self.load_data()
-
-            self.outcomes = self.config['outcomes']
+            self.outcomes = self.load_outcomes()
             self.metadata = self.config.get('metadata', {})
 
             self.StateType = create_state_model(self.data_fields)
@@ -260,6 +259,20 @@ class WorkflowEngine:
             )
         return data
 
+    def load_outcomes(self):
+        outcomes: list = self.config['outcomes']
+
+        if self.mfa_config:
+            mfa_cancelled_outcome = {
+                'id': 'mfa_cancelled',
+                'type': 'failure',
+                'message': self.mfa_config.mfa_cancelled_message
+            }
+            outcomes.append(mfa_cancelled_outcome)
+            logger.info(f"Auto-generated 'mfa_cancelled' outcome with message: {self.mfa_config.mfa_cancelled_message}")
+
+        return outcomes
+
 
 def load_workflow(yaml_path: str, checkpointer=None, config=None, mfa_config: Optional[MFAConfig] = None) -> Tuple[CompiledStateGraph, WorkflowEngine]:
     """

soprano_sdk/nodes/call_function.py

@@ -39,11 +39,17 @@ class CallFunctionStrategy(ActionStrategy):
         if 'mfa' in self.step_config:
             state['_mfa'] = state.get('_mfa', {})
             state['_mfa']['post_payload'] = dict(transactionId=str(uuid.uuid4()))
+            state['_mfa']['post_headers'] = {}
             state['_mfa_config'] = self.engine_context.mfa_config
             template_loader = self.engine_context.get_config_value("template_loader", Environment())
             for k, v in self.step_config['mfa']['payload'].items():
                 state['_mfa']['post_payload'][k] = compile_values(template_loader, state, v)
 
+            # Process headers if provided
+            if 'headers' in self.step_config['mfa']:
+                for k, v in self.step_config['mfa']['headers'].items():
+                    state['_mfa']['post_headers'][k] = compile_values(template_loader, state, v)
+
     def execute(self, state: Dict[str, Any]) -> Dict[str, Any]:
         from ..utils.tracing import trace_node_execution
         

{soprano_sdk-0.2.5.dist-info → soprano_sdk-0.2.7.dist-info}/METADATA

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

{soprano_sdk-0.2.5.dist-info → soprano_sdk-0.2.7.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=7aTueLu2jLxl213Q5_BkNQ534RgEwf6lodQa_PaJFWs,5980
+soprano_sdk/authenticators/mfa.py,sha256=Vew9Nb8pIRTw9hKbEZTH3YScY-fZ_TLq4ZuCzc-wbr8,7387
 soprano_sdk/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-soprano_sdk/core/constants.py,sha256=21ful8skHgDeDJJMdB_oJDw7Xq3lkbO6KWOhHuuLIa0,3330
-soprano_sdk/core/engine.py,sha256=UTFJimrgUzfdDZgc4rW5nvkOKevAEGW8C5Zr1D0tgcs,11270
+soprano_sdk/core/constants.py,sha256=UPXlRbF7gsOUNOV0Lm0jvgFfgZX7JrsV6n9I5csMfns,3508
+soprano_sdk/core/engine.py,sha256=vM2-nAvDc6Oam-q_BegabBT4uKVMV8DVV10SyJJPgVw,11762
 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/async_function.py,sha256=v6WujLKm8NXX2iAkJ7Gz_QIVCtWFrpC6nnPyyfuCxXs,9354
 soprano_sdk/nodes/base.py,sha256=idFyOGGPnjsASYnrOF_NIh7eFcSuJqw61EoVN_WCTaU,2360
-soprano_sdk/nodes/call_function.py,sha256=PdmSSVXkMT5ptxeUjmKqND-wvqbaXp1i61QzIzFOfgQ,5535
+soprano_sdk/nodes/call_function.py,sha256=afYBmj5Aditbkvb_7gD3CsXBEEUohcsC1_cdHfcOunE,5847
 soprano_sdk/nodes/collect_input.py,sha256=lEltZOU5ALvc57q8I_4SjzzEapVejy9mS0E73Jf7-sk,23759
 soprano_sdk/nodes/factory.py,sha256=IbBzT4FKBnYw5PuSo7uDONV3HSFtoyqjBQQtXtUY2IY,1756
 soprano_sdk/routing/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -29,7 +29,7 @@ soprano_sdk/utils/tracing.py,sha256=gSHeBDLe-MbAZ9rkzpCoGFveeMdR9KLaA6tteB0IWjk,
 soprano_sdk/validation/__init__.py,sha256=ImChmO86jYHU90xzTttto2-LmOUOmvY_ibOQaLRz5BA,262
 soprano_sdk/validation/schema.py,sha256=SlC4sq-ueEg0p_8Uox_cgPj9S-0AEEiOOlA1Vsu0DsE,15443
 soprano_sdk/validation/validator.py,sha256=GaCvHvjwVe88Z8yatQsueiPnqtq1oo5uN75gogzpQT0,8940
-soprano_sdk-0.2.5.dist-info/METADATA,sha256=KYdstjCuIfKrBuMAGOTVZDDYLESuWEcBFQY7oEhNdNg,11297
-soprano_sdk-0.2.5.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-soprano_sdk-0.2.5.dist-info/licenses/LICENSE,sha256=A1aBauSjPNtVehOXJe3WuvdU2xvM9H8XmigFMm6665s,1073
-soprano_sdk-0.2.5.dist-info/RECORD,,
+soprano_sdk-0.2.7.dist-info/METADATA,sha256=L9wKEDqgskIRsFIIhtYJq8gfUZ2TJuwXUqrtb-4qNEI,11297
+soprano_sdk-0.2.7.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+soprano_sdk-0.2.7.dist-info/licenses/LICENSE,sha256=A1aBauSjPNtVehOXJe3WuvdU2xvM9H8XmigFMm6665s,1073
+soprano_sdk-0.2.7.dist-info/RECORD,,