fastworkflow 2.15.4__tar.gz → 2.15.5__tar.gz

{fastworkflow-2.15.4 → fastworkflow-2.15.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: fastworkflow
-Version: 2.15.4
+Version: 2.15.5
 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.15.4 → fastworkflow-2.15.5}/fastworkflow/__init__.py

@@ -1,6 +1,8 @@
 import contextlib
+from dataclasses import dataclass
 from enum import Enum
 import os
+import time
 from typing import Any, Optional, Union
 
 from pydantic import BaseModel
@@ -47,6 +49,20 @@ class MCPToolResult(BaseModel):
     content: list[MCPContent]
     isError: bool = False
 
+class CommandTraceEventDirection(str, Enum):
+    AGENT_TO_WORKFLOW = "agent_to_workflow"
+    WORKFLOW_TO_AGENT = "workflow_to_agent"
+
+@dataclass
+class CommandTraceEvent:
+    direction: CommandTraceEventDirection
+    raw_command: str | None               # for AGENT_TO_WORKFLOW
+    command_name: str | None              # for WORKFLOW_TO_AGENT
+    parameters: dict | str | None
+    response_text: str | None
+    success: bool | None
+    timestamp_ms: int
+
 class CommandOutput(BaseModel):
     command_responses: list[CommandResponse]
     workflow_name: str = ""

{fastworkflow-2.15.4 → fastworkflow-2.15.5}/fastworkflow/chat_session.py

@@ -119,6 +119,7 @@ class ChatSession:
         # Create queues for user messages and command outputs
         self._user_message_queue = Queue()
         self._command_output_queue = Queue()
+        self._command_trace_queue = Queue()
         self._status = SessionStatus.STOPPED
         self._chat_worker = None
 
@@ -290,6 +291,10 @@ class ChatSession:
     def command_output_queue(self) -> Queue:
         return self._command_output_queue
 
+    @property
+    def command_trace_queue(self) -> Queue:
+        return self._command_trace_queue
+
     @property
     def workflow_is_complete(self) -> bool:
         workflow = ChatSession.get_active_workflow()

{fastworkflow-2.15.4 → fastworkflow-2.15.5}/fastworkflow/command_executor.py

@@ -45,18 +45,7 @@ class CommandExecutor(CommandExecutorInterface):
                 command = command)
         )
 
-        import json
-
         if command_output.command_handled or not command_output.success:           
-            # Append executed action to action.json for external consumers
-            record = {
-                "command" if command_output.success else "failing command": command,
-                "command_name": command_output.command_name, 
-                "parameters": command_output.command_parameters.model_dump() if command_output.command_parameters else None,
-                "response": command_output.command_responses[0].response if command_output.success else ''}
-            with open("action.json", "a", encoding="utf-8") as f:
-                f.write(json.dumps(record, ensure_ascii=False) + "\n")
-
             return command_output
 
         command_name = command_output.command_responses[0].artifacts["command_name"]
@@ -93,16 +82,7 @@ class CommandExecutor(CommandExecutorInterface):
         command_output.workflow_name = workflow_name
         command_output.context = context
         command_output.command_name = command_name
-        command_output.command_parameters = str(input_obj) if input_obj else ''
-
-        # Append executed action to action.json for external consumers
-        record = {
-            "command": command, 
-            "command_name": command_name, 
-            "parameters": input_obj.model_dump() if input_obj else None,
-            "response": command_output.command_responses[0].response}
-        with open("action.json", "a", encoding="utf-8") as f:
-            f.write(json.dumps(record, ensure_ascii=False) + "\n")
+        command_output.command_parameters = input_obj or None
 
         return command_output
 

{fastworkflow-2.15.4 → fastworkflow-2.15.5}/fastworkflow/run/__main__.py

@@ -219,6 +219,29 @@ def run_main(args):
         with console.status("[bold cyan]Processing command...[/bold cyan]", spinner="dots") as status:
             counter = 0
             while wait_thread.is_alive():
+                # Check for agent traces if in agent mode
+                if args.run_as_agent:
+                    while True:
+                        try:
+                            evt = fastworkflow.chat_session.command_trace_queue.get_nowait()
+                        except queue.Empty:
+                            break
+                        
+                        # Choose styles based on success
+                        info_style = "dim orange3" if (evt.success is False) else "dim yellow"
+                        resp_style = "dim orange3" if (evt.success is False) else "dim green"
+
+                        if evt.direction == fastworkflow.CommandTraceEventDirection.AGENT_TO_WORKFLOW:
+                            console.print(Text("Agent -> Workflow: ", style=info_style), end="")
+                            console.print(Text(str(evt.raw_command or ""), style=info_style))
+                        else:
+                            # command info (dim yellow or dim orange3)
+                            info = f"{evt.command_name or ''}, {evt.parameters}: "
+                            console.print(Text("Workflow -> Agent: ", style=info_style), end="")
+                            console.print(Text(info, style=info_style), end="")
+                            # response (dim green or dim orange3)
+                            console.print(Text(str(evt.response_text or ""), style=resp_style))
+
                 time.sleep(0.5)
                 counter += 1
                 if counter % 2 == 0:  # Update message every second

{fastworkflow-2.15.4 → fastworkflow-2.15.5}/fastworkflow/workflow_agent.py

@@ -3,6 +3,8 @@ Agent integration module for fastWorkflow.
 Provides workflow tool agent functionality for intelligent tool selection.
 """
 
+import json
+import time
 import dspy
 
 import fastworkflow
@@ -78,18 +80,67 @@ def _execute_workflow_query(command: str, chat_session_obj: fastworkflow.ChatSes
     Commands must be formatted using plain text for command name followed by XML tags enclosing parameter values (if any) as follows: command_name <param1_name>param1_value</param1_name> <param2_name>param2_value</param2_name> ...
     Don't use this tool to respond to a clarification requests in PARAMETER EXTRACTION ERROR state
     """
+    # Emit trace event before execution
+    chat_session_obj.command_trace_queue.put(fastworkflow.CommandTraceEvent(
+        direction=fastworkflow.CommandTraceEventDirection.AGENT_TO_WORKFLOW,
+        raw_command=command,
+        command_name=None,
+        parameters=None,
+        response_text=None,
+        success=None,
+        timestamp_ms=int(time.time() * 1000),
+    ))
+
     # Directly invoke the command without going through queues
     # This allows the agent to synchronously call workflow tools
     from fastworkflow.command_executor import CommandExecutor
     command_output = CommandExecutor.invoke_command(chat_session_obj, command)
 
+    # Emit trace event after execution
+    # Extract command name and parameters from command_output
+    name = command_output.command_name
+    params = command_output.command_parameters
+    
+    # Handle parameter serialization
+    params_dict = params.model_dump() if params else None
+    
+    # Extract response text
+    resp_text = ""
+    if command_output.command_responses:
+        response_parts = [
+            cmd_response.response
+            for cmd_response in command_output.command_responses
+            if cmd_response.response
+        ]
+        resp_text = "\n".join(response_parts)
+    
+    chat_session_obj.command_trace_queue.put(fastworkflow.CommandTraceEvent(
+        direction=fastworkflow.CommandTraceEventDirection.WORKFLOW_TO_AGENT,
+        raw_command=None,
+        command_name=name,
+        parameters=params_dict,
+        response_text=resp_text or "",
+        success=bool(command_output.success),
+        timestamp_ms=int(time.time() * 1000),
+    ))
+
+    # Append executed action to action.json for external consumers (agent mode only)
+    record = {
+        "command" if command_output.success else "failing command": command,
+        "command_name": name,
+        "parameters": params_dict,
+        "response": resp_text if command_output.success else ""
+    }
+    with open("action.json", "a", encoding="utf-8") as f:
+        f.write(json.dumps(record, ensure_ascii=False) + "\n")
+
     # Format output - extract text from command response
-    if hasattr(command_output, 'command_responses') and command_output.command_responses:
+    if command_output.command_responses:
         response_parts = []
         response_parts.extend(
             cmd_response.response
             for cmd_response in command_output.command_responses
-            if hasattr(cmd_response, 'response') and cmd_response.response
+            if cmd_response.response
         )
         return "\n".join(response_parts) if response_parts else "Command executed successfully."
 

{fastworkflow-2.15.4 → fastworkflow-2.15.5}/pyproject.toml

@@ -9,7 +9,7 @@ repository = "https://github.com/radiantlogicinc/fastworkflow"
 
 [tool.poetry]
 name = "fastworkflow"
-version = "2.15.4"
+version = "2.15.5"
 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"