Adds agent executor with interceptors

src/Agent/AgentExecutor.php

@@ -21,6 +21,9 @@
 use LLM\Agents\Tool\ToolInterface;
 use LLM\Agents\Tool\ToolRepositoryInterface;
 
+/**
+ * @deprecated
+ */
 final readonly class AgentExecutor
 {
     public function __construct(

src/AgentExecutor/Exception/ExecutorException.php

@@ -0,0 +1,10 @@
+<?php
+
+declare(strict_types=1);
+
+namespace LLM\Agents\AgentExecutor\Exception;
+
+class ExecutorException extends \DomainException
+{
+
+}

src/AgentExecutor/Exception/InvalidPromptException.php

@@ -0,0 +1,10 @@
+<?php
+
+declare(strict_types=1);
+
+namespace LLM\Agents\AgentExecutor\Exception;
+
+final class InvalidPromptException extends ExecutorException
+{
+
+}

src/AgentExecutor/ExecutorInterceptorInterface.php

@@ -0,0 +1,23 @@
+<?php
+
+declare(strict_types=1);
+
+namespace LLM\Agents\AgentExecutor;
+
+use LLM\Agents\Agent\Execution;
+use LLM\Agents\LLM\ContextInterface;
+use LLM\Agents\LLM\OptionsInterface;
+use LLM\Agents\LLM\Prompt\Chat\Prompt;
+use LLM\Agents\LLM\PromptContextInterface;
+
+interface ExecutorInterceptorInterface
+{
+    public function execute(
+        string $agent,
+        string|\Stringable|Prompt $prompt,
+        ContextInterface $context,
+        OptionsInterface $options,
+        PromptContextInterface $promptContext,
+        ExecutorInterface $next,
+    ): Execution;
+}

src/AgentExecutor/ExecutorInterface.php

@@ -0,0 +1,25 @@
+<?php
+
+declare(strict_types=1);
+
+namespace LLM\Agents\AgentExecutor;
+
+use LLM\Agents\Agent\Execution;
+use LLM\Agents\LLM\ContextInterface;
+use LLM\Agents\LLM\OptionsInterface;
+use LLM\Agents\LLM\Prompt\Chat\Prompt;
+use LLM\Agents\LLM\Prompt\Context;
+use LLM\Agents\LLM\PromptContextInterface;
+
+interface ExecutorInterface
+{
+    public function execute(
+        string $agent,
+        string|\Stringable|Prompt $prompt,
+        ?ContextInterface $context = null,
+        ?OptionsInterface $options = null,
+        PromptContextInterface $promptContext = new Context(),
+    ): Execution;
+
+    public function withInterceptor(ExecutorInterceptorInterface ...$interceptor): self;
+}

src/AgentExecutor/ExecutorPipeline.php

@@ -0,0 +1,78 @@
+<?php
+
+declare(strict_types=1);
+
+namespace LLM\Agents\AgentExecutor;
+
+use LLM\Agents\Agent\Execution;
+use LLM\Agents\AgentExecutor\Exception\InvalidPromptException;
+use LLM\Agents\LLM\ContextFactoryInterface;
+use LLM\Agents\LLM\ContextInterface;
+use LLM\Agents\LLM\LLMInterface;
+use LLM\Agents\LLM\OptionsFactoryInterface;
+use LLM\Agents\LLM\OptionsInterface;
+use LLM\Agents\LLM\Prompt\Chat\Prompt;
+use LLM\Agents\LLM\Prompt\Context;
+use LLM\Agents\LLM\Prompt\PromptInterface;
+use LLM\Agents\LLM\PromptContextInterface;
+
+final class ExecutorPipeline implements ExecutorInterface
+{
+    /** @var ExecutorInterceptorInterface[] */
+    private array $interceptors = [];
+    private int $offset = 0;
+
+    public function __construct(
+        private readonly LLMInterface $llm,
+        private readonly OptionsFactoryInterface $optionsFactory,
+        private readonly ContextFactoryInterface $contextFactory,
+    ) {}
+
+    public function execute(
+        string $agent,
+        \Stringable|string|Prompt $prompt,
+        ?ContextInterface $context = null,
+        ?OptionsInterface $options = null,
+        PromptContextInterface $promptContext = new Context(),
+    ): Execution {
+        $context ??= $this->contextFactory->create();
+        $options ??= $this->optionsFactory->create();
+
+        if (!isset($this->interceptors[$this->offset])) {
+            if (!$prompt instanceof PromptInterface) {
+                throw new InvalidPromptException(\sprintf('Prompt must be an instance of %s', PromptInterface::class));
+            }
+
+            $result = $this->llm->generate($context, $prompt, $options);
+
+            return new Execution(
+                result: $result,
+                prompt: $prompt,
+            );
+        }
+
+        return $this->interceptors[$this->offset]->execute(
+            agent: $agent,
+            prompt: $prompt,
+            context: $context,
+            options: $options,
+            promptContext: $promptContext,
+            next: $this->next(),
+        );
+    }
+
+    public function withInterceptor(ExecutorInterceptorInterface ...$interceptor): ExecutorInterface
+    {
+        $pipeline = clone $this;
+        $pipeline->interceptors = \array_merge($this->interceptors, $interceptor);
+
+        return $pipeline;
+    }
+
+    private function next(): self
+    {
+        $pipeline = clone $this;
+        $pipeline->offset++;
+        return $pipeline;
+    }
+}

src/AgentExecutor/Interceptor/GeneratePromptInterceptor.php

@@ -0,0 +1,45 @@
+<?php
+
+declare(strict_types=1);
+
+namespace LLM\Agents\AgentExecutor\Interceptor;
+
+use LLM\Agents\Agent\AgentRepositoryInterface;
+use LLM\Agents\Agent\Execution;
+use LLM\Agents\AgentExecutor\ExecutorInterceptorInterface;
+use LLM\Agents\AgentExecutor\ExecutorInterface;
+use LLM\Agents\LLM\AgentPromptGeneratorInterface;
+use LLM\Agents\LLM\ContextInterface;
+use LLM\Agents\LLM\OptionsInterface;
+use LLM\Agents\LLM\Prompt\Chat\Prompt;
+use LLM\Agents\LLM\PromptContextInterface;
+
+final readonly class GeneratePromptInterceptor implements ExecutorInterceptorInterface
+{
+    public function __construct(
+        private AgentRepositoryInterface $agents,
+        private AgentPromptGeneratorInterface $promptGenerator,
+    ) {}
+
+    public function execute(
+        string $agent,
+        \Stringable|string|Prompt $prompt,
+        ContextInterface $context,
+        OptionsInterface $options,
+        PromptContextInterface $promptContext,
+        ExecutorInterface $next,
+    ): Execution {
+        if (!$prompt instanceof Prompt) {
+            $agent = $this->agents->get($agent);
+            $prompt = $this->promptGenerator->generate($agent, $prompt, $promptContext);
+        }
+
+        return $next->execute(
+            agent: $agent,
+            prompt: $prompt,
+            context: $context,
+            options: $options,
+            promptContext: $promptContext,
+        );
+    }
+}

src/AgentExecutor/Interceptor/InjectModelInterceptor.php

@@ -0,0 +1,41 @@
+<?php
+
+declare(strict_types=1);
+
+namespace LLM\Agents\AgentExecutor\Interceptor;
+
+use LLM\Agents\Agent\AgentRepositoryInterface;
+use LLM\Agents\Agent\Execution;
+use LLM\Agents\AgentExecutor\ExecutorInterceptorInterface;
+use LLM\Agents\AgentExecutor\ExecutorInterface;
+use LLM\Agents\LLM\AgentPromptGeneratorInterface;
+use LLM\Agents\LLM\ContextInterface;
+use LLM\Agents\LLM\OptionsInterface;
+use LLM\Agents\LLM\Prompt\Chat\Prompt;
+use LLM\Agents\LLM\PromptContextInterface;
+
+final readonly class InjectModelInterceptor implements ExecutorInterceptorInterface
+{
+    public function __construct(
+        private AgentRepositoryInterface $agents,
+    ) {}
+
+    public function execute(
+        string $agent,
+        \Stringable|string|Prompt $prompt,
+        ContextInterface $context,
+        OptionsInterface $options,
+        PromptContextInterface $promptContext,
+        ExecutorInterface $next,
+    ): Execution {
+        $agent = $this->agents->get($agent);
+
+        return $next->execute(
+            agent: $agent->getKey(),
+            prompt: $prompt,
+            context: $context,
+            options: $options->with('model', $agent->getModel()->name),
+            promptContext: $promptContext,
+        );
+    }
+}

src/AgentExecutor/Interceptor/InjectOptionsInterceptor.php

@@ -0,0 +1,44 @@
+<?php
+
+declare(strict_types=1);
+
+namespace LLM\Agents\AgentExecutor\Interceptor;
+
+use LLM\Agents\Agent\AgentRepositoryInterface;
+use LLM\Agents\Agent\Execution;
+use LLM\Agents\AgentExecutor\ExecutorInterceptorInterface;
+use LLM\Agents\AgentExecutor\ExecutorInterface;
+use LLM\Agents\LLM\ContextInterface;
+use LLM\Agents\LLM\OptionsInterface;
+use LLM\Agents\LLM\Prompt\Chat\Prompt;
+use LLM\Agents\LLM\PromptContextInterface;
+
+final readonly class InjectOptionsInterceptor implements ExecutorInterceptorInterface
+{
+    public function __construct(
+        private AgentRepositoryInterface $agents,
+    ) {}
+
+    public function execute(
+        string $agent,
+        \Stringable|string|Prompt $prompt,
+        ContextInterface $context,
+        OptionsInterface $options,
+        PromptContextInterface $promptContext,
+        ExecutorInterface $next,
+    ): Execution {
+        $agent = $this->agents->get($agent);
+
+        foreach ($agent->getConfiguration() as $configuration) {
+            $options = $options->with($configuration->key, $configuration->content);
+        }
+
+        return $next->execute(
+            agent: $agent->getKey(),
+            prompt: $prompt,
+            context: $context,
+            options: $options,
+            promptContext: $promptContext,
+        );
+    }
+}

src/AgentExecutor/Interceptor/InjectResponseIntoPromptInterceptor.php

@@ -0,0 +1,47 @@
+<?php
+
+declare(strict_types=1);
+
+namespace LLM\Agents\AgentExecutor\Interceptor;
+
+use LLM\Agents\Agent\Execution;
+use LLM\Agents\AgentExecutor\ExecutorInterceptorInterface;
+use LLM\Agents\AgentExecutor\ExecutorInterface;
+use LLM\Agents\LLM\ContextInterface;
+use LLM\Agents\LLM\OptionsInterface;
+use LLM\Agents\LLM\Prompt\Chat\Prompt;
+use LLM\Agents\LLM\PromptContextInterface;
+use LLM\Agents\LLM\Response\ChatResponse;
+use LLM\Agents\LLM\Response\ToolCalledResponse;
+
+final class InjectResponseIntoPromptInterceptor implements ExecutorInterceptorInterface
+{
+    public function execute(
+        string $agent,
+        \Stringable|string|Prompt $prompt,
+        ContextInterface $context,
+        OptionsInterface $options,
+        PromptContextInterface $promptContext,
+        ExecutorInterface $next,
+    ): Execution {
+        $execution = $next->execute(
+            agent: $agent,
+            prompt: $prompt,
+            context: $context,
+            options: $options,
+            promptContext: $promptContext,
+        );
+
+        if (
+            $execution->result instanceof ChatResponse
+            || $execution->result instanceof ToolCalledResponse
+        ) {
+            $prompt = $execution->prompt->withAddedMessage($execution->result->toMessage());
+        }
+
+        return new Execution(
+            result: $execution->result,
+            prompt: $prompt,
+        );
+    }
+}