soprano-sdk 0.2.12__py3-none-any.whl → 0.2.13__py3-none-any.whl

soprano_sdk/core/engine.py

@@ -56,7 +56,8 @@ class WorkflowEngine:
 
             logger.info(
                 f"Workflow loaded: {self.workflow_name} v{self.workflow_version} "
-                f"({len(self.steps)} steps, {len(self.outcomes)} outcomes)"
+                f"({len(self.steps)} steps, {len(self.outcomes)} outcomes, "
+                f"{len(self.collector_node_field_map)} collector nodes)"
             )
 
         except Exception as e:
@@ -202,6 +203,7 @@ class WorkflowEngine:
     def load_steps(self):
         prepared_steps: list = []
         mfa_redirects: Dict[str, str] = {}
+        self.collector_node_field_map: Dict[str, str] = {}  # Map of node_id -> field
 
         for step in self.config['steps']:
             step_id = step['id']
@@ -228,6 +230,14 @@ class WorkflowEngine:
 
             prepared_steps.append(step)
 
+        # Build collector node -> field map
+        for step in prepared_steps:
+            if step.get('action') == 'collect_input':
+                node_id = step.get('id')
+                field = step.get('field')
+                if node_id and field:
+                    self.collector_node_field_map[node_id] = field
+
         for step in prepared_steps:
             if step['id'] in self.mfa_validator_steps: # MFA Validator
                 continue

soprano_sdk/nodes/collect_input.py

@@ -196,17 +196,6 @@ class CollectInputStrategy(ActionStrategy):
         if context_value is None:
             return
 
-        # Check if this node has already executed - if so, don't overwrite the collected value
-        execution_order = state.get(WorkflowKeys.NODE_EXECUTION_ORDER, [])
-        if self.step_id in execution_order:
-            logger.info(f"Skipping context value for '{self.field}' - node '{self.step_id}' already executed")
-            span.add_event("context.value_skipped", {
-                "field": self.field,
-                "reason": "node_already_executed",
-                "existing_value": str(state.get(self.field))
-            })
-            return
-
         logger.info(f"Using context value for '{self.field}': {context_value}")
         state[self.field] = context_value
         span.add_event("context.value_used", {"field": self.field, "value": str(context_value)})

soprano_sdk/tools.py

@@ -83,26 +83,35 @@ class WorkflowTool:
             callback_handler = CallbackHandler()
             config = {"configurable": {"thread_id": thread_id}, "callbacks": [callback_handler]}
 
-            span.add_event("context.updated", {"fields": list(initial_context.keys())})
-
             state = self.graph.get_state(config)
 
-            # Update engine context on both resume and fresh start
-            # Note: collect_input nodes will check NODE_EXECUTION_ORDER to avoid
-            # overwriting already-collected values
-            self.engine.update_context(initial_context)
-
+            # Intelligently update context based on workflow state
             if state.next:
+                # Workflow is resuming - only update fields that haven't been collected yet
                 span.set_attribute("workflow.resumed", True)
                 logger.info(f"[WorkflowTool] Resuming interrupted workflow {self.name} (thread: {thread_id})")
+
+                filtered_context = self._filter_already_collected_fields(state.values, initial_context)
+                self.engine.update_context(filtered_context)
+
+                span.add_event("context.updated", {
+                    "fields": list(filtered_context.keys()),
+                    "filtered_out": list(set(initial_context.keys()) - set(filtered_context.keys()))
+                })
+
                 result = self.graph.invoke(
-                    Command(resume=user_message or "", update=initial_context),
+                    Command(resume=user_message or "", update=filtered_context),
                     config=config
                 )
 
             else:
+                # Fresh start - update all fields from initial_context
                 span.set_attribute("workflow.resumed", False)
                 logger.info(f"[WorkflowTool] Starting fresh workflow {self.name} (thread: {thread_id})")
+
+                self.engine.update_context(initial_context)
+                span.add_event("context.updated", {"fields": list(initial_context.keys())})
+
                 result = self.graph.invoke(initial_context, config=config)
             
             final_state = self.graph.get_state(config)
@@ -129,6 +138,51 @@ class WorkflowTool:
             span.set_attribute("workflow.status", "completed")
             return self.engine.get_outcome_message(result)
 
+    def _filter_already_collected_fields(
+        self,
+        current_state: Dict[str, Any],
+        initial_context: Optional[Dict[str, Any]]
+    ) -> Dict[str, Any]:
+        """
+        Filter initial_context to exclude fields that have already been collected.
+
+        Args:
+            current_state: Current workflow state
+            initial_context: Context to filter
+
+        Returns:
+            Filtered context with only uncollected fields
+        """
+        if not initial_context:
+            return {}
+
+        from .core.constants import WorkflowKeys
+
+        execution_order = current_state.get(WorkflowKeys.NODE_EXECUTION_ORDER, [])
+
+        node_to_field_map = self.engine.collector_node_field_map
+
+        # Determine which fields have already been collected
+        collected_fields = set()
+        for executed_node_id in execution_order:
+            if executed_node_id in node_to_field_map:
+                collected_fields.add(node_to_field_map[executed_node_id])
+
+        # Filter initial_context to exclude already-collected fields
+        filtered_context = {
+            field: value
+            for field, value in initial_context.items()
+            if field not in collected_fields
+        }
+
+        if collected_fields:
+            logger.info(
+                f"[WorkflowTool] Filtered out already-collected fields: {collected_fields}. "
+                f"Updating context with: {list(filtered_context.keys())}"
+            )
+
+        return filtered_context
+
     def resume(
         self,
         thread_id: str,

{soprano_sdk-0.2.12.dist-info → soprano_sdk-0.2.13.dist-info}/METADATA

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

{soprano_sdk-0.2.12.dist-info → soprano_sdk-0.2.13.dist-info}/RECORD

@@ -1,6 +1,6 @@
 soprano_sdk/__init__.py,sha256=YZVl_SwQ0C-E_5_f1AwUe_hPcbgCt8k7k4_WAHM8vjE,243
 soprano_sdk/engine.py,sha256=EFK91iTHjp72otLN6Kg-yeLye2J3CAKN0QH4FI2taL8,14838
-soprano_sdk/tools.py,sha256=0zIMCQ0voPGXFHgQCN4jKaMPBXCbnia4Ly5hssPI3oU,9033
+soprano_sdk/tools.py,sha256=dmJ0OZ7Bj3rvjBQvLzgWlYRFVtNJOyMO2jLqaS13cAc,10971
 soprano_sdk/agents/__init__.py,sha256=Yzbtv6iP_ABRgZo0IUjy9vDofEvLFbOjuABw758176A,636
 soprano_sdk/agents/adaptor.py,sha256=dRk4pU1UgUhsKRotOzILjeXk0Zzoj7GzB-SKyBgKbNs,3242
 soprano_sdk/agents/factory.py,sha256=Aucfz4rZVKCXMAQtbGAqp1JR8aYwa66mokRmKkKGhYA,6699
@@ -9,14 +9,14 @@ soprano_sdk/authenticators/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJ
 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=UPXlRbF7gsOUNOV0Lm0jvgFfgZX7JrsV6n9I5csMfns,3508
-soprano_sdk/core/engine.py,sha256=vM2-nAvDc6Oam-q_BegabBT4uKVMV8DVV10SyJJPgVw,11762
+soprano_sdk/core/engine.py,sha256=HKYoqwDm541pWSWwEKHxLlL3PX90Ux_5l_-HqihgL-g,12245
 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=afYBmj5Aditbkvb_7gD3CsXBEEUohcsC1_cdHfcOunE,5847
-soprano_sdk/nodes/collect_input.py,sha256=IHDzhfh97-AxtaWfRbSFXgaaxnw4BLgumfPt7rbqA0s,25125
+soprano_sdk/nodes/collect_input.py,sha256=0M_-orYlOFvEBAiW1dDnPef1i77n41CjdV8ZKpWqLBE,24562
 soprano_sdk/nodes/factory.py,sha256=IbBzT4FKBnYw5PuSo7uDONV3HSFtoyqjBQQtXtUY2IY,1756
 soprano_sdk/routing/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 soprano_sdk/routing/router.py,sha256=Z218r4BMbmlL9282ombutAoKsIs1WHZ2d5YHnbCeet8,3698
@@ -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.12.dist-info/METADATA,sha256=t7Kq8Jrr6wYLa17TuLFyCBS6ACpX6NL7CXqPf2-QAH8,11343
-soprano_sdk-0.2.12.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-soprano_sdk-0.2.12.dist-info/licenses/LICENSE,sha256=A1aBauSjPNtVehOXJe3WuvdU2xvM9H8XmigFMm6665s,1073
-soprano_sdk-0.2.12.dist-info/RECORD,,
+soprano_sdk-0.2.13.dist-info/METADATA,sha256=UcRp0Owv8akIqBA-y5fQmQBgIFlFkelf9CfQ0iXIV4g,11343
+soprano_sdk-0.2.13.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+soprano_sdk-0.2.13.dist-info/licenses/LICENSE,sha256=A1aBauSjPNtVehOXJe3WuvdU2xvM9H8XmigFMm6665s,1073
+soprano_sdk-0.2.13.dist-info/RECORD,,