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

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

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

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

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

{soprano_sdk-0.2.2 → soprano_sdk-0.2.3}/soprano_sdk/__init__.py

@@ -1,4 +1,5 @@
 from .core.engine import WorkflowEngine, load_workflow
+from .core.constants import MFAConfig
 from .tools import WorkflowTool
 
 __version__ = "0.1.0"
@@ -6,5 +7,6 @@ __version__ = "0.1.0"
 __all__ = [
     "WorkflowEngine",
     "load_workflow",
+    "MFAConfig",
     "WorkflowTool",
 ]

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

@@ -1,6 +1,6 @@
 import requests
-from typing import TypedDict, Literal, NotRequired
-from soprano_sdk.core.constants import MFARestAuthorizerEnv
+from typing import TypedDict, Literal, NotRequired, Optional
+from soprano_sdk.core.constants import MFAConfig
 
 
 class MFAChallenge(TypedDict):
@@ -28,15 +28,21 @@ def build_path(base_url: str, path: str):
     return f"{base_url.rstrip('/')}/{path.lstrip('/')}"
 
 
-def enforce_mfa_if_required(state: dict):
+def enforce_mfa_if_required(state: dict, mfa_config: Optional[MFAConfig] = None):
+    if mfa_config is None:
+        mfa_config = state.get('_mfa_config') or MFAConfig()
+
     _mfa : MFAState = state['_mfa']
     if _mfa['status'] == 'COMPLETED':
         return True
     generate_token_response = requests.post(
         build_path(
-            base_url=MFARestAuthorizerEnv.GENERATE_TOKEN_BASE_URL.get_from_env(),
-            path=MFARestAuthorizerEnv.GENERATE_TOKEN_PATH.get_from_env()
-        ), json=_mfa['post_payload'], timeout=30, headers={"Authorization": f"Bearer {state['bearer_token']}"}
+            base_url=mfa_config.generate_token_base_url,
+            path=mfa_config.generate_token_path
+        ),
+        json=_mfa['post_payload'],
+        timeout=mfa_config.api_timeout,
+        headers={"Authorization": f"Bearer {state['bearer_token']}"}
     )
     _, error = get_response(generate_token_response)
 
@@ -50,7 +56,10 @@ def enforce_mfa_if_required(state: dict):
     return False
 
 
-def mfa_validate_user_input(**state: dict):
+def mfa_validate_user_input(mfa_config: Optional[MFAConfig] = None, **state: dict):
+    if mfa_config is None:
+        mfa_config = state.get('_mfa_config') or MFAConfig()
+
     _mfa : MFAState = state['_mfa']
     input_field_name = state['_active_input_field']
     if not state[input_field_name]:
@@ -61,9 +70,12 @@ def mfa_validate_user_input(**state: dict):
     post_payload.update({challenge_field_name: {"value": state[input_field_name]}})
     validate_token_response = requests.post(
         build_path(
-            base_url=MFARestAuthorizerEnv.VALIDATE_TOKEN_BASE_URL.get_from_env(),
-            path=MFARestAuthorizerEnv.VALIDATE_TOKEN_PATH.get_from_env()
-        ), json=post_payload, timeout=30, headers={"Authorization": f"Bearer {state['bearer_token']}"}
+            base_url=mfa_config.validate_token_base_url,
+            path=mfa_config.validate_token_path
+        ),
+        json=post_payload,
+        timeout=mfa_config.api_timeout,
+        headers={"Authorization": f"Bearer {state['bearer_token']}"}
     )
     _mfa['retry_count'] += 1
     response, error = get_response(validate_token_response)
@@ -78,11 +90,11 @@ def mfa_validate_user_input(**state: dict):
 
         authorize = requests.post(
             build_path(
-                base_url=MFARestAuthorizerEnv.AUTHORIZE_TOKEN_BASE_URL.get_from_env(),
-                path=MFARestAuthorizerEnv.AUTHORIZE_TOKEN_PATH.get_from_env()
+                base_url=mfa_config.authorize_token_base_url,
+                path=mfa_config.authorize_token_path
             ),
             json=post_payload,
-            timeout=30,
+            timeout=mfa_config.api_timeout,
             headers={"Authorization": f"Bearer {state['bearer_token']}"}
         )
         if authorize.status_code == 204:

soprano_sdk-0.2.3/soprano_sdk/core/constants.py

@@ -0,0 +1,114 @@
+from enum import Enum
+from typing import Optional
+from pydantic import Field
+from pydantic_settings import BaseSettings, SettingsConfigDict
+
+
+class WorkflowKeys:
+    STEP_ID = '_step_id'
+    STATUS = '_status'
+    OUTCOME_ID = '_outcome_id'
+    MESSAGES = '_messages'
+    CONVERSATIONS = '_conversations'
+    STATE_HISTORY = '_state_history'
+    COLLECTOR_NODES = '_collector_nodes'
+    ATTEMPT_COUNTS = '_attempt_counts'
+    NODE_EXECUTION_ORDER = '_node_execution_order'
+    NODE_FIELD_MAP = '_node_field_map'
+    COMPUTED_FIELDS = '_computed_fields'
+    ERROR = 'error'
+
+
+class ActionType(Enum):
+    COLLECT_INPUT_WITH_AGENT = 'collect_input_with_agent'
+    CALL_FUNCTION = 'call_function'
+
+
+class DataType(Enum):
+    TEXT = 'text'
+    NUMBER = 'number'
+    DOUBLE = 'double'
+    BOOLEAN = 'boolean'
+    LIST = 'list'
+    DICT = 'dict'
+    ANY = "any"
+
+
+class OutcomeType(Enum):
+    SUCCESS = 'success'
+    FAILURE = 'failure'
+
+
+class StatusPattern:
+    COLLECTING = '{step_id}_collecting'
+    MAX_ATTEMPTS = '{step_id}_max_attempts'
+    NEXT_STEP = '{step_id}_{next_step}'
+    SUCCESS = '{step_id}_success'
+    FAILED = '{step_id}_failed'
+    INTENT_CHANGE = '{step_id}_{target_node}'
+
+
+class TransitionPattern:
+    CAPTURED = '{field}_CAPTURED:'
+    FAILED = '{field}_FAILED:'
+    INTENT_CHANGE = 'INTENT_CHANGE:'
+
+
+DEFAULT_MAX_ATTEMPTS = 3
+DEFAULT_MODEL = 'gpt-4o-mini'
+DEFAULT_TIMEOUT = 300
+
+MAX_ATTEMPTS_MESSAGE = "I'm having trouble understanding your {field}. Please contact customer service for assistance."
+WORKFLOW_COMPLETE_MESSAGE = "Workflow completed."
+
+
+class MFAConfig(BaseSettings):
+    """
+    Configuration for MFA REST API endpoints.
+
+    Values can be provided during initialization or will be automatically
+    loaded from environment variables with the same name (uppercase).
+
+    Example:
+        # Load from environment variables
+        config = MFAConfig()
+
+        # Or provide specific values
+        config = MFAConfig(
+            generate_token_base_url="https://api.example.com",
+            generate_token_path="/v1/mfa/generate"
+        )
+    """
+    generate_token_base_url: Optional[str] = Field(
+        default=None,
+        description="Base URL for the generate token endpoint"
+    )
+    generate_token_path: Optional[str] = Field(
+        default=None,
+        description="Path for the generate token endpoint"
+    )
+    validate_token_base_url: Optional[str] = Field(
+        default=None,
+        description="Base URL for the validate token endpoint"
+    )
+    validate_token_path: Optional[str] = Field(
+        default=None,
+        description="Path for the validate token endpoint"
+    )
+    authorize_token_base_url: Optional[str] = Field(
+        default=None,
+        description="Base URL for the authorize token endpoint"
+    )
+    authorize_token_path: Optional[str] = Field(
+        default=None,
+        description="Path for the authorize token endpoint"
+    )
+    api_timeout: int = Field(
+        default=30,
+        description="API request timeout in seconds"
+    )
+
+    model_config = SettingsConfigDict(
+        case_sensitive=False,
+        extra='ignore'
+    )

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

@@ -7,7 +7,7 @@ from langgraph.constants import START
 from langgraph.graph import StateGraph
 from langgraph.graph.state import CompiledStateGraph
 
-from .constants import WorkflowKeys
+from .constants import WorkflowKeys, MFAConfig
 from .state import create_state_model
 from ..nodes.factory import NodeFactory
 from ..routing.router import WorkflowRouter
@@ -19,7 +19,7 @@ from soprano_sdk.authenticators.mfa import MFANodeConfig
 
 class WorkflowEngine:
 
-    def __init__(self, yaml_path: str, configs: dict):
+    def __init__(self, yaml_path: str, configs: dict, mfa_config: Optional[MFAConfig] = None):
         self.yaml_path = yaml_path
         self.configs = configs or {}
         logger.info(f"Loading workflow from: {yaml_path}")
@@ -29,7 +29,7 @@ class WorkflowEngine:
                 self.config = yaml.safe_load(f)
 
             logger.info("Validating workflow configuration")
-            validate_workflow(self.config)
+            validate_workflow(self.config, mfa_config=mfa_config or MFAConfig())
 
             self.workflow_name = self.config['name']
             self.workflow_description = self.config['description']
@@ -37,6 +37,7 @@ class WorkflowEngine:
             self.mfa_validator_steps: set[str] = set()
             self.steps: list = self.load_steps()
             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']
@@ -259,7 +260,7 @@ class WorkflowEngine:
         return data
 
 
-def load_workflow(yaml_path: str, checkpointer=None, config=None) -> Tuple[CompiledStateGraph, WorkflowEngine]:
+def load_workflow(yaml_path: str, checkpointer=None, config=None, mfa_config: Optional[MFAConfig] = None) -> Tuple[CompiledStateGraph, WorkflowEngine]:
     """
     Load a workflow from YAML configuration.
 
@@ -270,6 +271,8 @@ def load_workflow(yaml_path: str, checkpointer=None, config=None) -> Tuple[Compi
         checkpointer: Optional checkpointer for state persistence.
                      Defaults to InMemorySaver() if not provided.
                      Example: MongoDBSaver for production persistence.
+        config: Optional configuration dictionary
+        mfa_config: Optional MFA configuration. If not provided, will load from environment variables.
 
     Returns:
         Tuple of (compiled_graph, engine) where:
@@ -278,11 +281,21 @@ def load_workflow(yaml_path: str, checkpointer=None, config=None) -> Tuple[Compi
 
     Example:
         ```python
+        # Load with environment variables
         graph, engine = load_workflow("workflow.yaml")
+
+        # Or provide MFA configuration explicitly
+        from soprano_sdk.core.constants import MFAConfig
+        mfa_config = MFAConfig(
+            generate_token_base_url="https://api.example.com",
+            generate_token_path="/v1/mfa/generate"
+        )
+        graph, engine = load_workflow("workflow.yaml", mfa_config=mfa_config)
+
         result = graph.invoke({}, config={"configurable": {"thread_id": "123"}})
         message = engine.get_outcome_message(result)
         ```
     """
-    engine = WorkflowEngine(yaml_path, configs=config)
+    engine = WorkflowEngine(yaml_path, configs=config, mfa_config=mfa_config)
     graph = engine.build_graph(checkpointer=checkpointer)
     return graph, engine

{soprano_sdk-0.2.2 → soprano_sdk-0.2.3}/soprano_sdk/core/state.py

@@ -48,6 +48,7 @@ def create_state_model(data_fields: List[dict]):
     fields['_computed_fields'] = Annotated[List[str], replace]
     fields['error'] = Annotated[Optional[Dict[str, str]], replace]
     fields['_mfa'] = Annotated[Optional[Dict[str, str]], replace]
+    fields['_mfa_config'] = Annotated[Optional[Any], replace]
     fields['mfa_input'] = Annotated[Optional[Dict[str, str]], replace]
 
     return types.new_class('WorkflowState', (TypedDict,), {}, lambda ns: ns.update({'__annotations__': fields}))

{soprano_sdk-0.2.2 → soprano_sdk-0.2.3}/soprano_sdk/nodes/call_function.py

@@ -39,6 +39,7 @@ 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_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)

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

@@ -116,6 +116,9 @@ class CollectInputStrategy(ActionStrategy):
 
     def pre_execute(self, state: Dict[str, Any]) -> Dict[str, Any]:
         state['_active_input_field'] = self.step_config.get('field')
+        # Inject MFA config for MFA validator nodes
+        if self.step_id in self.engine_context.mfa_validator_steps:
+            state['_mfa_config'] = self.engine_context.mfa_config
 
     @property
     def _formatted_field_name(self) -> str:

{soprano_sdk-0.2.2 → soprano_sdk-0.2.3}/soprano_sdk/tools.py

@@ -1,7 +1,7 @@
 """
 Workflow Tools - Wraps workflows as callable tools for agent frameworks
 """
-
+from __future__ import annotations
 import uuid
 from typing import Optional, Dict, Any
 from .utils.logger import logger
@@ -9,6 +9,7 @@ from .utils.logger import logger
 from langfuse.langchain import CallbackHandler
 
 from .core.engine import load_workflow
+from .core.constants import MFAConfig
 
 
 class WorkflowTool:
@@ -25,7 +26,8 @@ class WorkflowTool:
         name: str,
         description: str,
         checkpointer=None,
-        config: Optional[Dict]=None
+        config: Optional[Dict]=None,
+        mfa_config: Optional[MFAConfig] = None
     ):
         """Initialize workflow tool
 
@@ -39,9 +41,13 @@ class WorkflowTool:
         self.name = name
         self.description = description
         self.checkpointer = checkpointer
+        self.mfa_config = mfa_config
 
         # Load workflow
-        self.graph, self.engine = load_workflow(yaml_path, checkpointer=checkpointer, config=config)
+        self.graph, self.engine = load_workflow(
+            yaml_path, checkpointer=checkpointer,
+            config=config, mfa_config=mfa_config
+        )
 
     def execute(
         self,

{soprano_sdk-0.2.2 → soprano_sdk-0.2.3}/soprano_sdk/validation/validator.py

@@ -1,9 +1,11 @@
-from typing import List, Set
+from typing import List, Set, Optional, TYPE_CHECKING
 
 import jsonschema
 
 from .schema import WORKFLOW_SCHEMA
-from soprano_sdk.core.constants import MFARestAuthorizerEnv
+
+if TYPE_CHECKING:
+    from soprano_sdk.core.constants import MFAConfig
 
 
 class ValidationResult:
@@ -21,8 +23,9 @@ class ValidationResult:
 
 
 class WorkflowValidator:
-    def __init__(self, config: dict):
+    def __init__(self, config: dict, mfa_config: Optional['MFAConfig'] = None):
         self.config = config
+        self.mfa_config = mfa_config
         self.errors: List[str] = []
 
     def validate(self) -> ValidationResult:
@@ -110,20 +113,10 @@ class WorkflowValidator:
                     )
 
     def _validate_authorizer(self, step):
-
-        def _validate_rest_fields():
-            if mfa_authorizer['type'] == 'REST':
-                for field in MFARestAuthorizerEnv:
-                    is_present = field.get_from_env()
-                    if not is_present:
-                        self.errors.append(f"`{field.value}` needs to be set as Environment for REST MFA")
-            else:
-                self.errors.append(f"step({step['id']}) -> mfa -> type is unsupported")
-
         mfa_authorizer = step.get("mfa", None)
         if mfa_authorizer is None:
             return
-        
+
         if mfa_authorizer and step['action'] != 'call_function':
             self.errors.append(
                 f"MFA is enabled in step({step['id']}). MFA is supported only for `call_function` nodes"
@@ -133,7 +126,31 @@ class WorkflowValidator:
         if not model:
             self.errors.append(f"step({step['id']}) -> mfa -> model is missing")
 
-        _validate_rest_fields()
+        mfa_type = mfa_authorizer.get('type')
+        if mfa_type and mfa_type != 'REST':
+            self.errors.append(f"step({step['id']}) -> mfa -> type '{mfa_type}' is unsupported. Only 'REST' is supported.")
+
+        # Validate mfa_config if provided
+        if self.mfa_config is not None:
+            missing_fields = []
+            if not self.mfa_config.generate_token_base_url:
+                missing_fields.append('generate_token_base_url')
+            if not self.mfa_config.generate_token_path:
+                missing_fields.append('generate_token_path')
+            if not self.mfa_config.validate_token_base_url:
+                missing_fields.append('validate_token_base_url')
+            if not self.mfa_config.validate_token_path:
+                missing_fields.append('validate_token_path')
+            if not self.mfa_config.authorize_token_base_url:
+                missing_fields.append('authorize_token_base_url')
+            if not self.mfa_config.authorize_token_path:
+                missing_fields.append('authorize_token_path')
+
+            if missing_fields:
+                self.errors.append(
+                    f"MFA configuration is missing required fields: {', '.join(missing_fields)}. "
+                    f"Either provide them via mfa_config parameter or set corresponding environment variables."
+                )
 
 
     def _validate_data_fields(self):
@@ -200,8 +217,8 @@ class WorkflowValidator:
                     )
 
 
-def validate_workflow(config: dict) -> ValidationResult:
-    validator = WorkflowValidator(config)
+def validate_workflow(config: dict, mfa_config: Optional['MFAConfig'] = None) -> ValidationResult:
+    validator = WorkflowValidator(config, mfa_config=mfa_config)
     result = validator.validate()
 
     if not result.is_valid:

{soprano_sdk-0.2.2 → soprano_sdk-0.2.3}/tests/test_mfa_scenarios.py

@@ -1,19 +1,10 @@
-"""
-Comprehensive test cases for MFA (Multi-Factor Authentication) functionality
-Tests cover all scenarios including forward flow, loop backs, and multiple MFA steps
-
-Test Scenario: Concert Ticket Booking System
-"""
 import os
-import sys
-from typing import Dict, Any
 import pytest
+from soprano_sdk import load_workflow
 
-# Add examples directory to path
-examples_dir = os.path.join(os.path.dirname(__file__), "..", "examples", "concert_booking")
-sys.path.insert(0, os.path.abspath(examples_dir))
 
-from soprano_sdk import load_workflow
+def get_examples_dir():
+    return os.path.join(os.path.dirname(__file__), "..", "examples", "concert_booking")
 
 
 @pytest.fixture(scope="module", autouse=True)
@@ -57,14 +48,14 @@ def test_mfa_nodes_created():
     print("TEST 1: MFA Node Creation")
     print("=" * 80)
 
-    yaml_path = os.path.join(examples_dir, "concert_ticket_booking.yaml")
+    yaml_path = os.path.join(get_examples_dir(), "concert_ticket_booking.yaml")
     graph, engine = load_workflow(yaml_path)
 
     # Count MFA-protected steps in original config
     mfa_protected_steps = ['process_payment', 'send_confirmation']
 
     print(f"\nOriginal MFA-protected steps: {mfa_protected_steps}")
-    print(f"Expected MFA nodes per protected step: 2 (start + collector)")
+    print("Expected MFA nodes per protected step: 2 (start + collector)")
     print(f"Total expected MFA nodes: {len(mfa_protected_steps) * 2}")
 
     # Verify MFA nodes were created
@@ -104,7 +95,7 @@ def test_mfa_previous_step_redirection():
     print("TEST 2: Previous Step Redirection (Forward Flow)")
     print("=" * 80)
 
-    yaml_path = os.path.join(examples_dir, "concert_ticket_booking.yaml")
+    yaml_path = os.path.join(get_examples_dir(), "concert_ticket_booking.yaml")
     graph, engine = load_workflow(yaml_path)
 
     # Test Case 2a: check_seat_availability should point to process_payment_mfa_start
@@ -153,7 +144,7 @@ def test_mfa_loop_back_redirection():
     print("TEST 3: Loop Back Redirection (CRITICAL BUG FIX TEST)")
     print("=" * 80)
 
-    yaml_path = os.path.join(examples_dir, "concert_ticket_booking.yaml")
+    yaml_path = os.path.join(get_examples_dir(), "concert_ticket_booking.yaml")
     graph, engine = load_workflow(yaml_path)
 
     print("\nScenario: User completes booking, then requests modification")
@@ -162,7 +153,7 @@ def test_mfa_loop_back_redirection():
 
     # Verify ask_modification_needed loops back to collect_seat_preference
     modification_step = engine.get_step_info('ask_modification_needed')
-    print(f"\nStep: ask_modification_needed")
+    print("\nStep: ask_modification_needed")
 
     if transitions := modification_step.get('transitions'):
         loop_back = next((t for t in transitions if 'collect_seat_preference' in t.get('next', '')), None)
@@ -203,7 +194,7 @@ def test_mfa_loop_back_redirection():
 
             # Check if we've reached payment MFA
             if 'process_payment' in next_step and '_mfa_start' in next_step:
-                print(f"\n✅ CRITICAL: Loop back correctly goes through MFA!")
+                print("\n✅ CRITICAL: Loop back correctly goes through MFA!")
                 print(f"   Full path: {' -> '.join(path)}")
                 break
         else:
@@ -231,7 +222,7 @@ def test_mfa_multiple_steps_sequence():
     print("TEST 4: Multiple MFA Steps in Sequence")
     print("=" * 80)
 
-    yaml_path = os.path.join(examples_dir, "concert_ticket_booking.yaml")
+    yaml_path = os.path.join(get_examples_dir(), "concert_ticket_booking.yaml")
     graph, engine = load_workflow(yaml_path)
 
     print("\nScenario: Two consecutive MFA-protected steps")
@@ -239,7 +230,7 @@ def test_mfa_multiple_steps_sequence():
 
     # Get process_payment step
     payment_step = engine.get_step_info('process_payment')
-    print(f"\nStep: process_payment")
+    print("\nStep: process_payment")
 
     # Find transition to send_confirmation
     next_after_payment = None
@@ -286,7 +277,7 @@ def test_mfa_data_fields_created():
     print("TEST 5: MFA Data Fields Creation")
     print("=" * 80)
 
-    yaml_path = os.path.join(examples_dir, "concert_ticket_booking.yaml")
+    yaml_path = os.path.join(get_examples_dir(), "concert_ticket_booking.yaml")
     graph, engine = load_workflow(yaml_path)
 
     print("\nChecking if MFA-specific data fields were created...")
@@ -308,7 +299,7 @@ def test_mfa_data_fields_created():
 
             assert field_name in data_fields, \
                 f"MFA field {field_name} should be in data fields"
-            print(f"  ✅ Field exists in data schema")
+            print("  ✅ Field exists in data schema")
 
     print("\n✅ TEST PASSED: MFA data fields created correctly")
     print("=" * 80)
@@ -327,7 +318,7 @@ def test_mfa_alternative_paths():
     print("TEST 6: Alternative Paths to MFA Steps")
     print("=" * 80)
 
-    yaml_path = os.path.join(examples_dir, "concert_ticket_booking.yaml")
+    yaml_path = os.path.join(get_examples_dir(), "concert_ticket_booking.yaml")
     graph, engine = load_workflow(yaml_path)
 
     print("\nScenario: User selects unavailable seats, tries alternatives")
@@ -335,7 +326,7 @@ def test_mfa_alternative_paths():
 
     # Check offer_alternative_seats loops back correctly
     alternative_step = engine.get_step_info('offer_alternative_seats')
-    print(f"\nStep: offer_alternative_seats")
+    print("\nStep: offer_alternative_seats")
 
     if transitions := alternative_step.get('transitions'):
         for t in transitions:
@@ -371,7 +362,7 @@ def test_mfa_comprehensive_flow():
     print("TEST 7: Comprehensive MFA Flow Validation")
     print("=" * 80)
 
-    yaml_path = os.path.join(examples_dir, "concert_ticket_booking.yaml")
+    yaml_path = os.path.join(get_examples_dir(), "concert_ticket_booking.yaml")
     graph, engine = load_workflow(yaml_path)
 
     mfa_protected_original = ['process_payment', 'send_confirmation']