fastworkflow 2.17.5__tar.gz → 2.17.6__tar.gz

{fastworkflow-2.17.5 → fastworkflow-2.17.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: fastworkflow
-Version: 2.17.5
+Version: 2.17.6
 Summary: A framework for rapidly building large-scale, deterministic, interactive workflows with a fault-tolerant, conversational UX
 License: Apache-2.0
 Keywords: fastworkflow,ai,workflow,llm,openai

{fastworkflow-2.17.5 → fastworkflow-2.17.6}/fastworkflow/chat_session.py

@@ -467,15 +467,28 @@ class ChatSession:
             self
         )
 
+        # Get available commands for current context and pass to agent.
+        # The CommandsSystemPreludeAdapter will inject these commands into the system 
+        # message, keeping them out of the trajectory to avoid token bloat while still 
+        # providing context-specific command info.
+        from fastworkflow.workflow_agent import _what_can_i_do
+        available_commands = _what_can_i_do(self)
+
         lm = dspy_utils.get_lm("LLM_AGENT", "LITELLM_API_KEY_AGENT")
         from dspy.utils.exceptions import AdapterParseError
+        from fastworkflow.utils.chat_adapter import CommandsSystemPreludeAdapter
+        
+        # Use CommandsSystemPreludeAdapter specifically for workflow agent calls
+        agent_adapter = CommandsSystemPreludeAdapter()
+        
         # Retry logic for AdapterParseError
         max_retries = 2
         for attempt in range(max_retries):
             try:
-                with dspy.context(lm=lm, adapter=dspy.ChatAdapter()):
+                with dspy.context(lm=lm, adapter=agent_adapter):
                     agent_result = self._workflow_tool_agent(
-                        user_query=command_info_and_refined_message_with_todolist
+                        user_query=command_info_and_refined_message_with_todolist,
+                        available_commands=available_commands
                     )
                 break  # Success, exit retry loop
             except AdapterParseError as _:

fastworkflow-2.17.6/fastworkflow/utils/chat_adapter.py

@@ -0,0 +1,99 @@
+"""
+ChatAdapter wrapper for injecting context-specific available commands into system messages.
+
+Design Overview:
+---------------
+This module implements a ChatAdapter wrapper that dynamically injects workflow command information
+into the system message at runtime, avoiding the need to rebuild ReAct agent modules per context.
+
+Key Benefits:
+- Single shared agent: No per-context module caching required
+- Dynamic updates: Commands refresh per call based on current workflow context
+- Token efficiency: Commands appear in system (not repeated in trajectory/history)
+- Zero rebuild cost: Signature and modules remain stable across context changes
+
+Usage:
+------
+The adapter is used specifically for workflow agent calls via dspy.context():
+
+    from fastworkflow.utils.chat_adapter import CommandsSystemPreludeAdapter
+    
+    agent_adapter = CommandsSystemPreludeAdapter()
+    available_commands = _what_can_i_do(chat_session)
+    
+    with dspy.context(lm=lm, adapter=agent_adapter):
+        agent_result = agent(
+            user_query="...",
+            available_commands=available_commands
+        )
+
+The adapter intercepts the format call and prepends commands to the system message,
+keeping them out of the trajectory to prevent token bloat across iterations.
+This scoped approach ensures the adapter only affects workflow agent calls, not other
+DSPy operations in the system.
+"""
+import dspy
+
+
+class CommandsSystemPreludeAdapter(dspy.ChatAdapter):
+    """
+    Wraps a base DSPy ChatAdapter to inject available commands into the system message.
+    
+    This adapter intercepts the render process and prepends a "Available commands" section
+    to the system message when `available_commands` is present in inputs. This ensures
+    commands are visible to the model at each step without being added to the trajectory
+    or conversation history.
+    
+    Args:
+        base: The underlying ChatAdapter to wrap. Defaults to dspy.ChatAdapter() if None.
+        title: The header text for the commands section. Defaults to "Available commands".
+    
+    Example:
+        >>> import dspy
+        >>> from fastworkflow.utils.chat_adapter import CommandsSystemPreludeAdapter
+        >>> dspy.settings.adapter = CommandsSystemPreludeAdapter()
+    """
+    
+    def __init__(self, base: dspy.ChatAdapter | None = None, title: str = "Available commands"):
+        super().__init__()
+        self.base = base or dspy.ChatAdapter()
+        self.title = title
+    
+    def format(self, signature, demos, inputs):
+        """
+        Format the inputs for the model, injecting available_commands into system message.
+        
+        This method wraps the base adapter's format method and modifies the result
+        to include available commands in the system message if present in inputs.
+        
+        Args:
+            signature: The DSPy signature defining the task
+            demos: List of demonstration examples
+            inputs: Dictionary of input values, may include 'available_commands'
+            
+        Returns:
+            Formatted messages with commands injected into system message
+        """
+        # Call the base adapter's format method
+        formatted = self.base.format(signature, demos, inputs)
+        
+        # Check if available_commands is in inputs
+        cmds = inputs.get("available_commands")
+        if not cmds:
+            return formatted
+        
+        # Inject commands into the system message
+        prelude = f"{self.title}:\n{cmds}".strip()
+        
+        # Formatted output is a list of messages, first may be system
+        # Find and modify the system message, or prepend one
+        if formatted and formatted[0].get("role") == "system":
+            # Prepend to existing system message
+            existing_content = formatted[0].get("content", "")
+            formatted[0]["content"] = f"{prelude}\n\n{existing_content}".strip()
+        else:
+            # No system message exists, prepend one
+            formatted.insert(0, {"role": "system", "content": prelude})
+        
+        return formatted
+

{fastworkflow-2.17.5 → fastworkflow-2.17.6}/fastworkflow/utils/react.py

@@ -67,13 +67,17 @@ class fastWorkflowReAct(Module):
             args={},
         )
 
-        for idx, tool in enumerate(tools.values()):
-            instr.append(f"({idx + 1}) {tool}")
+        instr.extend(f"({idx + 1}) {tool}" for idx, tool in enumerate(tools.values()))
         instr.append("When providing `next_tool_args`, the value inside the field must be in JSON format")
 
+        # Build the ReAct signature with trajectory and available_commands inputs.
+        # available_commands is injected into system message by CommandsSystemPreludeAdapter
+        # (see fastworkflow/utils/chat_adapter.py) and is NOT included in the trajectory
+        # formatting to avoid token bloat across iterations.
         react_signature = (
             dspy.Signature({**signature.input_fields}, "\n".join(instr))
             .append("trajectory", dspy.InputField(), type_=str)
+            .append("available_commands", dspy.InputField(), type_=str)
             .append("next_thought", dspy.OutputField(), type_=str)
             .append("next_tool_name", dspy.OutputField(), type_=Literal[tuple(tools.keys())])
             .append("next_tool_args", dspy.OutputField(), type_=dict[str, Any])
@@ -82,7 +86,7 @@ class fastWorkflowReAct(Module):
         fallback_signature = dspy.Signature(
             {**signature.input_fields, **signature.output_fields},
             signature.instructions,
-        ).append("trajectory", dspy.InputField(), type_=str)
+        ).append("trajectory", dspy.InputField(), type_=str).append("available_commands", dspy.InputField(), type_=str)
 
         self.tools = tools
         self.react = dspy.Predict(react_signature)

{fastworkflow-2.17.5 → fastworkflow-2.17.6}/pyproject.toml

@@ -9,7 +9,7 @@ repository = "https://github.com/radiantlogicinc/fastworkflow"
 
 [tool.poetry]
 name = "fastworkflow"
-version = "2.17.5"
+version = "2.17.6"
 description = "A framework for rapidly building large-scale, deterministic, interactive workflows with a fault-tolerant, conversational UX"
 authors = ["Dhar Rawal <drawal@radiantlogic.com>"]
 license = "Apache-2.0"