soprano-sdk 0.2.5__py3-none-any.whl → 0.2.6__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'

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.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: soprano-sdk
-Version: 0.2.5
+Version: 0.2.6
 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.6.dist-info}/RECORD

@@ -6,7 +6,7 @@ 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=Zl1dcwCmuwsAbruFMqguJ4lY0PPnC6v2EZ-xTPULX04,6098
 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
@@ -15,7 +15,7 @@ soprano_sdk/core/state.py,sha256=k8ojLfWgjES3p9XWMeGU5s4UK-Xa5T8mS4VtZzTrcDw,296
 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.6.dist-info/METADATA,sha256=Ykmgc9OD9Q5nBvGSD-z67oPU7Cy4jynpvhjZkwdPxzQ,11297
+soprano_sdk-0.2.6.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+soprano_sdk-0.2.6.dist-info/licenses/LICENSE,sha256=A1aBauSjPNtVehOXJe3WuvdU2xvM9H8XmigFMm6665s,1073
+soprano_sdk-0.2.6.dist-info/RECORD,,