example

#!/usr/bin/env php
<?php

use Symfony\Component\Console\Command\Command;
use Symfony\Component\Console\Input\InputInterface;
use Symfony\Component\Console\Output\ConsoleSectionOutput;
use Symfony\Component\Console\Output\OutputInterface;
use Symfony\Component\Console\SingleCommandApplication;
use Symfony\Component\Console\Style\SymfonyStyle;
use Symfony\Component\Finder\Finder;
use Symfony\Component\Finder\SplFileInfo;
use Symfony\Component\Process\Process;

require_once __DIR__.'/vendor/autoload.php';

$app = (new SingleCommandApplication('LLM Chain Example Runner'))
    ->setDescription('Runs all LLM Chain examples in folder examples/')
    ->setCode(function (InputInterface $input, OutputInterface $output) {
        $io = new SymfonyStyle($input, $output);
        $io->title('LLM Chain Examples');

        $examples = (new Finder())
            ->in(__DIR__.'/examples')
            ->name('*.php')
            ->sortByName()
            ->files();

        /** @var array{example: SplFileInfo, section: ConsoleSectionOutput, process: Process, state: 'running'|'finished'} $exampleRuns */
        $exampleRuns = [];
        foreach ($examples as $example) {
            $exampleRuns[] = [
                'example' => $example,
                'section' => $section = $output->section(),
                'process' => $process = new Process(['php', $example->getRealPath()]),
                'state' => 'running',
            ];

            $process->start();
            $section->writeln(sprintf('Example %s: <comment>Running</comment>', $example->getFilename()));
        }

        $examplesRunning = fn () => array_reduce($exampleRuns, fn ($running, $example) => $running || $example['process']->isRunning(), false);
        $examplesSuccessful = fn () => array_reduce($exampleRuns, fn ($successful, $example) => $successful && $example['process']->isSuccessful(), true);

        while ($examplesRunning()) {
            sleep(1);
            foreach ($exampleRuns as $run) {
                if ('running' === $run['state'] && !$run['process']->isRunning()) {
                    $emptyOutput = 0 === strlen(trim($run['process']->getOutput()));
                    $success = $run['process']->isSuccessful() && !$emptyOutput;
                    $result = $success ? '<info>Finished</info>'
                        : (1 === $run['process']->getExitCode() || $emptyOutput ? '<error>Failed</error>' : '<comment>Skipped</comment>');
                    $run['section']->overwrite(sprintf('Example %s: %s', $run['example']->getFilename(), $result));
                    $run['state'] = 'finished';
                    if ($output->isVerbose()) {
                        $exampleOutput = $emptyOutput ? 'Output was empty' : $run['process']->getOutput();
                        $exampleOutput = strlen($exampleOutput) <= 100 ? $exampleOutput : substr($exampleOutput, 0, 100).'...';
                        $run['section']->writeln(
                            sprintf('<%s>%s</>', $success ? 'fg=#999999' : 'fg=red', trim($exampleOutput))
                        );
                    }
                }
            }
        }

        $io->newLine();
        if (!$examplesSuccessful()) {
            $io->error('Some examples failed or were skipped!');

            return Command::FAILURE;
        }

        $io->success('All examples ran successfully!');

        return Command::SUCCESS;
    })
    ->run();