Class IFlowCodeRunner

internagent/experiments_utils_iflow.py:78–142

iFlow Code Runner class to handle interactions with iFlow CLI

class IFlowCodeRunner:
    """iFlow Code Runner class to handle interactions with iFlow CLI"""

    def __init__(self, proxy_settings=None, model='iflow-default'):
        """
        Initialize the iFlow Code Runner

        Args:
            proxy_settings: Optional dictionary with HTTP_PROXY and HTTPS_PROXY settings
            model: Model name to use (default: iflow-default)
        """
        self.proxy_settings = proxy_settings or {
            'HTTP_PROXY': 'http://127.0.0.1:7890',
            'HTTPS_PROXY': 'http://127.0.0.1:7890'
        }
        self.model = model
        
    def run(self, prompt, cwd=None):
        """
        Run iFlow Code with the given prompt
        
        Args:
            prompt: The prompt to send to iFlow
            cwd: The working directory for iFlow to operate in
            
        Returns:
            The stdout output from iFlow
        """
        # Set proxy environment variables
        env = os.environ.copy()
        for key, value in self.proxy_settings.items():
            env[key] = value
        
        # Enhanced logging - Log start time and command
        timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
        log_message = f"[{timestamp}] Running iFlow CLI with prompt: {prompt}..."
        print(log_message)
            
        # Run iFlow with appropriate parameters
        # Capture both stdout and stderr so we can log them properly
        result = subprocess.run(
            ['iflow',  '--prompt', prompt], # '--model', self.model,
            cwd=cwd,
            capture_output=True,
            text=True,
            env=env
        )

        # Enhanced logging - Log completion and output
        timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
        log_message = f"[{timestamp}] iFlow command completed with return code: {result.returncode}"
        print(log_message)

        # Log output summary
        if result.stdout:
            output_summary = f"iFlow output : {result.stdout}..."
            print(output_summary)