fastworkflow 2.15.1__tar.gz → 2.15.2__tar.gz

{fastworkflow-2.15.1 → fastworkflow-2.15.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: fastworkflow
-Version: 2.15.1
+Version: 2.15.2
 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.1 → fastworkflow-2.15.2}/fastworkflow/chat_session.py

@@ -121,6 +121,8 @@ class ChatSession:
         self._command_output_queue = Queue()
         self._status = SessionStatus.STOPPED
         self._chat_worker = None
+
+        self._conversation_history = dspy.History(messages=[])
         
         # Import here to avoid circular imports
         from fastworkflow.command_executor import CommandExecutor
@@ -297,6 +299,11 @@ class ChatSession:
     def workflow_is_complete(self, value: bool) -> None:
         if workflow := ChatSession.get_active_workflow():
             workflow.is_complete = value
+    
+    @property
+    def conversation_history(self) -> dspy.History:
+        """Return the conversation history."""
+        return self._conversation_history
        
 
     def _run_workflow_loop(self) -> Optional[fastworkflow.CommandOutput]:
@@ -421,7 +428,10 @@ class ChatSession:
 
         lm = dspy_utils.get_lm("LLM_AGENT", "LITELLM_API_KEY_AGENT")
         with dspy.context(lm=lm):
-            agent_result = self._workflow_tool_agent(user_query=refined_message)
+            agent_result = self._workflow_tool_agent(
+                user_query=refined_message,
+                conversation_history=self.conversation_history
+            )
             # dspy.inspect_history(n=1)
 
         # Extract the final result from the agent
@@ -431,6 +441,11 @@ class ChatSession:
             else str(agent_result)
         )
 
+        self.conversation_history.messages.append(
+            {"user_query": message, 
+             "agent_response": result_text}
+        )
+
         # Create CommandOutput with the agent's response
         command_response = fastworkflow.CommandResponse(response=result_text)
         # Attach actions captured during agent execution as artifacts if available

{fastworkflow-2.15.1 → fastworkflow-2.15.2}/fastworkflow/workflow_agent.py

@@ -18,6 +18,7 @@ class WorkflowAgentSignature(dspy.Signature):
     All the tasks in the todo list must be completed before returning the final answer.
     """
     user_query = dspy.InputField(desc="The natural language user query.")
+    conversation_history: dspy.History = dspy.InputField()
     final_answer = dspy.OutputField(desc="Comprehensive final answer with supporting evidence to demonstrate that all the tasks in the todo list have been completed.")
 
 def _what_can_i_do(chat_session_obj: fastworkflow.ChatSession) -> str:
@@ -119,7 +120,8 @@ def _ask_user_tool(clarification_request: str, chat_session_obj: fastworkflow.Ch
         command_responses=[fastworkflow.CommandResponse(response=clarification_request)]
     )
     chat_session_obj.command_output_queue.put(command_output)
-    return chat_session_obj.user_message_queue.get()
+    user_query = chat_session_obj.user_message_queue.get()
+    return _think_and_plan(user_query, chat_session_obj)
 
 def initialize_workflow_tool_agent(mcp_server: FastWorkflowMCPServer, max_iters: int = 25):
     """
@@ -133,11 +135,6 @@ def initialize_workflow_tool_agent(mcp_server: FastWorkflowMCPServer, max_iters:
     Returns:
         DSPy ReAct agent configured with workflow tools
     """
-    # available_tools = mcp_server.list_tools()
-
-    # if not available_tools:
-    #     return None
-
     chat_session_obj = mcp_server.chat_session
     if not chat_session_obj:
         return None
@@ -225,3 +222,33 @@ def initialize_workflow_tool_agent(mcp_server: FastWorkflowMCPServer, max_iters:
         tools=tools,
         max_iters=max_iters,
     )
+
+def _think_and_plan(user_query: str, chat_session_obj: fastworkflow.ChatSession) -> str:
+    """
+    Returns a refined plan by breaking down a user_query into simpler tasks based only on available commands and returns a todo list.
+    """
+    class TaskPlannerSignature(dspy.Signature):
+        """
+        Break down a user_query into simpler tasks based only on available commands and return a todo list.
+        If user_query is simple, return a single todo that is the user_query as-is
+        """
+        user_query: str = dspy.InputField()
+        available_commands: list[str] = dspy.InputField()
+        todo_list: list[str] = dspy.OutputField(desc="task descriptions as short sentences")
+
+    current_workflow = chat_session_obj.get_active_workflow()
+    available_commands = CommandMetadataAPI.get_command_display_text(
+        subject_workflow_path=current_workflow.folderpath,
+        cme_workflow_path=fastworkflow.get_internal_workflow_path("command_metadata_extraction"),
+        active_context_name=current_workflow.current_command_context_name,
+    )
+
+    planner_lm = dspy_utils.get_lm("LLM_PLANNER", "LITELLM_API_KEY_PLANNER")
+    with dspy.context(lm=planner_lm):
+        task_planner_func = dspy.ChainOfThought(TaskPlannerSignature)
+        prediction = task_planner_func(user_query=user_query, available_commands=available_commands)
+
+        if not prediction.todo_list or (len(prediction.todo_list) == 1 and prediction.todo_list[0] == user_query):
+            return user_query
+
+        return f"{user_query}\nNext steps:\n{'\n'.join([f'{i + 1}. {task}' for i, task in enumerate(prediction.todo_list)])}"

{fastworkflow-2.15.1 → fastworkflow-2.15.2}/pyproject.toml

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