soprano-sdk 0.1.96__py3-none-any.whl → 0.1.97__py3-none-any.whl

soprano_sdk/agents/structured_output.py

@@ -20,7 +20,7 @@ def create_structured_output_model(
     if not fields:
         raise ValueError("At least one field definition is required")
     
-    field_definitions = {"bot_response": (Optional[str], Field(None, description="bot response for the user query"))}
+    field_definitions = {"bot_response": (Optional[str], Field(None, description="bot response for the user query, only use this for clarification or asking for more information"))}
 
     if needs_intent_change:
         field_definitions["intent_change"] = (Optional[str], Field(None, description="node name for handling new intent"))

soprano_sdk/nodes/collect_input.py

@@ -61,6 +61,12 @@ IF the user's query continues with the SAME intent OR does not match any intent
 - Proceed with your normal response
 - Do NOT mention intent detection
 - Answer the user's question as configured
+
+BOT RESPONSE RULES:
+- If the user is asking a question or needs information, provide a helpful and concise response
+- If the user input is unclear or does not provide enough information, ask for clarification or more details
+- { "populate bot_response field to respond back to the user" if with_structured_output else ""}
+- Do not respond or use bot_response if the user provides a valid input
 """
 
 def _create_rollback_strategy(strategy_name: str) -> RollbackStrategy:
@@ -178,7 +184,8 @@ class CollectInputStrategy(ActionStrategy):
         return template_loader.from_string(template_str).render(state)
 
     def _apply_context_value(self, state: Dict[str, Any], span) -> None:
-        if not (context_value := self.engine_context.get_context_value(self.field)):
+        context_value = self.engine_context.get_context_value(self.field)
+        if context_value is None:
             return
         logger.info(f"Using context value for '{self.field}': {context_value}")
         state[self.field] = context_value
@@ -231,7 +238,7 @@ class CollectInputStrategy(ActionStrategy):
     def _handle_pre_populated_field(self, state: Dict[str, Any], conversation: List) -> Dict[str, Any]:
         logger.info(f"Field '{self.field}' is populated, skipping collection")
 
-        is_valid_input, _ = self._validate_collected_input(state)
+        is_valid_input, message = self._validate_collected_input(state)
         if not is_valid_input:
             self._set_status(state, "collecting")
             return self._handle_validation_failure(state, conversation, message=f"{state[self.field]}", role="user")
@@ -391,6 +398,11 @@ class CollectInputStrategy(ActionStrategy):
             workflow_steps=workflow_steps
         )
 
+        for key, value in restored_state.items():
+            context_value = self.engine_context.get_context_value(key)
+            if context_value is not None:
+                restored_state[key] = context_value
+
         if not restored_state:
             logger.warning(f"Rollback strategy returned empty state for node '{target_node}'")
             return {}
@@ -472,25 +484,17 @@ class CollectInputStrategy(ActionStrategy):
         return state
 
     def _find_matching_transition(self, agent_response: Any) -> Optional[str]:
-        is_structured_output = isinstance(agent_response, dict)
-        
         for transition in self.transitions:
-            if is_structured_output:
-                next_node = transition.get("next")
-                match_value = transition.get("match")
-                ref_field = transition.get("ref")
+            next_node = transition.get("next")
+            match_value = transition.get("match")
+            ref_field = transition.get("ref")
 
-                if not all([next_node, match_value, ref_field]):
-                    raise RuntimeError(f"Transition in step '{self.step_id}' missing required properties for structured output routing")
+            if not next_node or not ref_field or match_value is None:
+                raise RuntimeError(f"Transition in step '{self.step_id}' missing required properties for structured output routing")
 
-                if field_value := agent_response.get(ref_field):
-                    if field_value == match_value:
-                        return next_node
-            else:
-                next_node = transition.get("next")
-                pattern = transition.get("pattern")
-                if pattern in agent_response:
-                    return next_node
+            field_value = agent_response.get(ref_field)
+            if field_value == match_value:
+                return next_node
 
         return None
 

soprano_sdk/tools.py

@@ -78,35 +78,23 @@ class WorkflowTool:
         ) as span:
             callback_handler = CallbackHandler()
             config = {"configurable": {"thread_id": thread_id}, "callbacks": [callback_handler]}
-            
-            update_context = {}
-            engine_context_data = {}
-            for key, value in initial_context.items():
-                if key in self.engine.collect_input_fields:
-                    engine_context_data[key] = value
-                    continue
-                if  value:
-                    update_context[key] = value
-
-            if engine_context_data:
-                self.engine.update_context(engine_context_data)
-                span.add_event("context.updated", {"fields": list(engine_context_data.keys())})
+
+            self.engine.update_context(initial_context)
+            span.add_event("context.updated", {"fields": list(initial_context.keys())})
 
             state = self.graph.get_state(config)
 
             if state.next:
-                # Workflow is interrupted and waiting for input
                 span.set_attribute("workflow.resumed", True)
                 logger.info(f"[WorkflowTool] Resuming interrupted workflow {self.name} (thread: {thread_id})")
                 result = self.graph.invoke(
-                    Command(resume=user_message or "", update=update_context),
+                    Command(resume=user_message or "", update=initial_context),
                     config=config
                 )
             else:
-                # Workflow is fresh or completed, start/restart
                 span.set_attribute("workflow.resumed", False)
                 logger.info(f"[WorkflowTool] Starting fresh workflow {self.name} (thread: {thread_id})")
-                result = self.graph.invoke(update_context, config=config)
+                result = self.graph.invoke(initial_context, config=config)
             
             final_state = self.graph.get_state(config)
             if not final_state.next and self.checkpointer:

{soprano_sdk-0.1.96.dist-info → soprano_sdk-0.1.97.dist-info}/METADATA

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

{soprano_sdk-0.1.96.dist-info → soprano_sdk-0.1.97.dist-info}/RECORD

@@ -1,10 +1,10 @@
 soprano_sdk/__init__.py,sha256=y3c4i7Q7SAPS2Tee7V0TzWdhgMxBWfDJJ98eqD1HxGI,188
 soprano_sdk/engine.py,sha256=EFK91iTHjp72otLN6Kg-yeLye2J3CAKN0QH4FI2taL8,14838
-soprano_sdk/tools.py,sha256=A0qFEwn208GCYc-_7ZbZKR7N3HG5-6TA3Ma1RPo6YnM,8029
+soprano_sdk/tools.py,sha256=_lSZJoVwjy2RjCs4yXDupMlMSSApYB-XX57qYuy7mmE,7497
 soprano_sdk/agents/__init__.py,sha256=Yzbtv6iP_ABRgZo0IUjy9vDofEvLFbOjuABw758176A,636
 soprano_sdk/agents/adaptor.py,sha256=IMMgo9_KLI82i1eenOaojw7UE0jjx9vjm8mjfsodKSM,3226
 soprano_sdk/agents/factory.py,sha256=Aucfz4rZVKCXMAQtbGAqp1JR8aYwa66mokRmKkKGhYA,6699
-soprano_sdk/agents/structured_output.py,sha256=LDBWCMJFclOvcFB3OJpu37tO0Ct_M-L2PIH5MCYSjLI,3262
+soprano_sdk/agents/structured_output.py,sha256=7DSVzfMPsZAqBwI3v6XL15qG5Gh4jJ-qddcVPaa3gdc,3326
 soprano_sdk/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 soprano_sdk/core/constants.py,sha256=pEwW_NeHhxs7aG457uiBCs65czAapozY6r9JAegc01Y,1451
 soprano_sdk/core/engine.py,sha256=WEyqGaBasGnSqlBAkFSQIZXJb7T6OEsN-DWznIBAzNs,8341
@@ -13,7 +13,7 @@ soprano_sdk/core/state.py,sha256=h0Uo4uCwBAGTWrmzpDbcTwH6lI97-fXU9ek0tc3p2bM,261
 soprano_sdk/nodes/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 soprano_sdk/nodes/base.py,sha256=H6wvvN3kUeXluZAt5Hf3iocO9NjTElAH4fTb2-1JKr0,2233
 soprano_sdk/nodes/call_function.py,sha256=UM-JpaG6DutuaFhOrXxEcT_fnK85aH6sFCVkfeYIfwc,4350
-soprano_sdk/nodes/collect_input.py,sha256=LhDsqtlBKK-uIz90QIODDoR5PZbnv4rWyj-hvTR64do,21987
+soprano_sdk/nodes/collect_input.py,sha256=lAt1lonezXab0pVdzRGXPa1POAuIOmkMcAj0Kxwfi3I,22277
 soprano_sdk/nodes/factory.py,sha256=l-Gysfgnao-o2dphhnbjjxcH3ojZanZNYN3CBH9dDbA,1624
 soprano_sdk/routing/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 soprano_sdk/routing/router.py,sha256=SrNciTIXXdC9bAbbO5bX7PN9mlRbITjr4RZdNm4jEVA,3450
@@ -26,7 +26,7 @@ soprano_sdk/utils/tracing.py,sha256=iSJlTAaiGzgBvZhLISCGAd9_7F2HRzhcIUNHuaFv_Zc,
 soprano_sdk/validation/__init__.py,sha256=ImChmO86jYHU90xzTttto2-LmOUOmvY_ibOQaLRz5BA,262
 soprano_sdk/validation/schema.py,sha256=uJJZRDgwzWT2W8amd_W8mUAULvDnHJhiMEl-5so1ZK0,13559
 soprano_sdk/validation/validator.py,sha256=l2P24wiCWBNTZ9-dRbgWwK48BGaR1xIdnBxzSCu0RPM,6498
-soprano_sdk-0.1.96.dist-info/METADATA,sha256=eLSUnBX8Gg_37Okmm3g82Jm4UZp6j4pA64Y3cSd0YbQ,11269
-soprano_sdk-0.1.96.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-soprano_sdk-0.1.96.dist-info/licenses/LICENSE,sha256=A1aBauSjPNtVehOXJe3WuvdU2xvM9H8XmigFMm6665s,1073
-soprano_sdk-0.1.96.dist-info/RECORD,,
+soprano_sdk-0.1.97.dist-info/METADATA,sha256=opdDgT9ecUMXJJP7Dr93lnrPgwXsFqMfeJ1jRnCeh38,11269
+soprano_sdk-0.1.97.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+soprano_sdk-0.1.97.dist-info/licenses/LICENSE,sha256=A1aBauSjPNtVehOXJe3WuvdU2xvM9H8XmigFMm6665s,1073
+soprano_sdk-0.1.97.dist-info/RECORD,,