soprano-sdk 0.2.6__tar.gz → 0.2.7__tar.gz

{soprano_sdk-0.2.6 → soprano_sdk-0.2.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: soprano-sdk
-Version: 0.2.6
+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.6 → soprano_sdk-0.2.7}/pyproject.toml

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

{soprano_sdk-0.2.6 → soprano_sdk-0.2.7}/soprano_sdk/authenticators/mfa.py

@@ -154,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-0.2.6 → soprano_sdk-0.2.7}/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-0.2.6 → soprano_sdk-0.2.7}/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-0.2.6 → soprano_sdk-0.2.7}/tests/test_mfa_scenarios.py

@@ -447,7 +447,7 @@ def test_mfa_default_max_attempts():
 
         assert max_attempts == 3, \
             f"Expected default max_attempts=3, got {max_attempts}"
-        print(f"  ✅ Correctly uses default max_attempts=3")
+        print("  ✅ Correctly uses default max_attempts=3")
 
     print("\n✅ TEST PASSED: Default max_attempts value verified")
     print("=" * 80)
@@ -466,7 +466,6 @@ def test_mfa_custom_max_attempts():
 
     # Create a temporary YAML with custom max_attempts
     import tempfile
-    import shutil
 
     yaml_path = os.path.join(get_examples_dir(), "concert_ticket_booking.yaml")
 
@@ -511,7 +510,7 @@ def test_mfa_custom_max_attempts():
 
         assert max_attempts == 5, \
             f"Expected custom max_attempts=5, got {max_attempts}"
-        print(f"  ✅ Correctly uses custom max_attempts=5")
+        print("  ✅ Correctly uses custom max_attempts=5")
 
         print("\n✅ TEST PASSED: Custom max_attempts value applied correctly")
         print("=" * 80)
@@ -579,7 +578,7 @@ def test_mfa_custom_error_message():
 
         assert error_message == custom_error, \
             f"Expected custom error message, got {error_message}"
-        print(f"  ✅ Correctly uses custom error message")
+        print("  ✅ Correctly uses custom error message")
 
         print("\n✅ TEST PASSED: Custom error message applied correctly")
         print("=" * 80)
@@ -656,7 +655,7 @@ def test_mfa_custom_max_attempts_and_error():
 
         assert error_message == custom_error, \
             f"Expected custom error message, got {error_message}"
-        print(f"  ✅ Correctly uses custom error message")
+        print("  ✅ Correctly uses custom error message")
 
         print("\n✅ TEST PASSED: Both custom values applied correctly")
         print("=" * 80)
@@ -742,8 +741,6 @@ def test_mfa_custom_headers_with_jinja():
         assert process_payment_step in engine.list_steps(), \
             f"Expected {process_payment_step} step to exist"
 
-        step_info = engine.get_step_info(process_payment_step)
-
         # Check the MFA start node - this is where the MFA config is stored
         process_payment_mfa_start = 'process_payment_mfa_start'
         assert process_payment_mfa_start in engine.list_steps(), \
@@ -756,7 +753,7 @@ def test_mfa_custom_headers_with_jinja():
         assert 'headers' in mfa_start_info['mfa'], "MFA config should have headers"
 
         headers_config = mfa_start_info['mfa']['headers']
-        print(f"\nHeaders defined in MFA config:")
+        print("\nHeaders defined in MFA config:")
         for key, value in headers_config.items():
             print(f"  {key}: {value}")