fastworkflow 2.17.19__tar.gz → 2.17.21__tar.gz

{fastworkflow-2.17.19 → fastworkflow-2.17.21}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: fastworkflow
-Version: 2.17.19
+Version: 2.17.21
 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.19 → fastworkflow-2.17.21}/fastworkflow/run/__main__.py

@@ -181,12 +181,16 @@ def run_main(args):
 
     StartupProgress.advance("ChatSession ready")
     StartupProgress.end()
-
+   
     with contextlib.suppress(queue.Empty):
-        if command_output := fastworkflow.chat_session.command_output_queue.get(
+        timeout: float | None = None
+        if args.startup_command or args.startup_action:
             timeout=180.0
+        if command_output := fastworkflow.chat_session.command_output_queue.get(
+            timeout=timeout
         ):
             print_command_output(command_output)
+
     while not fastworkflow.chat_session.workflow_is_complete or args.keep_alive:
         with patch_stdout():
             user_command = prompt_session.prompt()

{fastworkflow-2.17.19 → fastworkflow-2.17.21}/fastworkflow/train/__main__.py

@@ -3,6 +3,7 @@ import os
 import json
 import shutil
 from dotenv import dotenv_values
+import importlib.util
 
 from colorama import Fore, Style
 
@@ -16,6 +17,21 @@ from fastworkflow.command_directory import CommandDirectory, get_cached_command_
 from fastworkflow.command_routing import RoutingDefinition, RoutingRegistry
 from fastworkflow.command_context_model import CommandContextModel
 
+# Cache the datasets availability check result
+_DATASETS_AVAILABLE = None
+
+
+def _datasets_available() -> bool:
+    """Check if the datasets package is available in the environment.
+    
+    Returns:
+        bool: True if datasets can be imported, False otherwise.
+    """
+    global _DATASETS_AVAILABLE
+    if _DATASETS_AVAILABLE is None:
+        _DATASETS_AVAILABLE = importlib.util.find_spec("datasets") is not None
+    return _DATASETS_AVAILABLE
+
 
 def train_workflow(workflow_path: str):
     # Ensure context model is parsed so downstream helpers have contexts
@@ -44,6 +60,16 @@ def train_workflow(workflow_path: str):
         logger.info(f"No _commands directory found at {workflow_path}, skipping training")
         return
     
+    # Check if datasets package is available for training
+    # Command directory and routing artifacts are generated above regardless
+    if not _datasets_available():
+        logger.warning(
+            f"datasets package not found in environment. Skipping intent detection training "
+            f"and DSPy few-shot parameter extraction for workflow: {workflow_path}. "
+            f"Other artifacts such as command_directory.json and routing_definition.json have been generated successfully."
+        )
+        return
+    
     # create a workflow and train the main workflow
     workflow = fastworkflow.Workflow.create(
         workflow_path, 

{fastworkflow-2.17.19 → fastworkflow-2.17.21}/fastworkflow/train/generate_synthetic.py

@@ -1,9 +1,16 @@
 from typing import List
-from datasets import load_dataset
 import random
 import fastworkflow
 import litellm
 
+# Conditional import of datasets - only required at runtime during training
+try:
+    from datasets import load_dataset
+    _DATASETS_AVAILABLE = True
+except ImportError:
+    _DATASETS_AVAILABLE = False
+    load_dataset = None
+
 NUMOF_PERSONAS=fastworkflow.get_env_var('SYNTHETIC_UTTERANCE_GEN_NUMOF_PERSONAS', int)
 UTTERANCES_PER_PERSONA=fastworkflow.get_env_var('SYNTHETIC_UTTERANCE_GEN_UTTERANCES_PER_PERSONA', int)
 PERSONAS_PER_BATCH=fastworkflow.get_env_var('SYNTHETIC_UTTERANCE_GEN_PERSONAS_PER_BATCH', int)
@@ -15,6 +22,15 @@ def generate_diverse_utterances(
     utterances_per_persona: int = UTTERANCES_PER_PERSONA,
     personas_per_batch: int = PERSONAS_PER_BATCH
 ) -> list[str]:
+    # If datasets is not available, return only the seed utterances
+    if not _DATASETS_AVAILABLE:
+        from fastworkflow.utils.logging import logger
+        logger.warning(
+            f"datasets package not available. Skipping synthetic utterance generation "
+            f"for command '{command_name}'. Using only seed utterances."
+        )
+        return [command_name] + seed_utterances
+    
     # Initialize LiteLLM with API key
     api_key = fastworkflow.get_env_var("LITELLM_API_KEY_SYNDATA_GEN")
     model=fastworkflow.get_env_var("LLM_SYNDATA_GEN")

{fastworkflow-2.17.19 → fastworkflow-2.17.21}/fastworkflow/workflow_agent.py

@@ -182,7 +182,10 @@ def _execute_workflow_query(command: str, chat_session_obj: fastworkflow.ChatSes
     # Handle parameter extraction errors with abort
     if nlu_stage == fastworkflow.NLUPipelineStage.PARAMETER_EXTRACTION:
         abort_confirmation = _execute_workflow_query('abort', chat_session_obj=chat_session_obj)
-        return f'{response_text}\n{abort_confirmation}'
+        return build_query_with_next_steps(
+            f'{response_text}\n{abort_confirmation}', 
+            chat_session_obj, with_agent_inputs_and_trajectory = True
+        )
 
     # Clean up the context flag after command execution
     workflow = chat_session_obj.get_active_workflow()

{fastworkflow-2.17.19 → fastworkflow-2.17.21}/pyproject.toml

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