wip

Author: tymondesigns

src/Agents/Stages/AppendUsage.php

@@ -5,7 +5,6 @@
 namespace Cortex\Agents\Stages;
 
 use Closure;
-use Cortex\LLM\Data\Usage;
 use Cortex\Contracts\Pipeable;
 use Cortex\LLM\Data\ChatResult;
 use Cortex\Pipeline\RuntimeConfig;

src/Agents/Stages/HandleToolCalls.php

@@ -14,7 +14,6 @@
 use Cortex\Support\Traits\CanPipe;
 use Illuminate\Support\Collection;
 use Cortex\LLM\Data\ChatGeneration;
-use Cortex\LLM\Data\ToolCallCollection;
 use Cortex\LLM\Data\ChatGenerationChunk;
 use Cortex\LLM\Data\Messages\ToolMessage;
 
@@ -53,10 +52,7 @@ public function handlePipeable(mixed $payload, RuntimeConfig $config, Closure $n
 
                 // Track the next step before making the LLM call
                 $config->context->addStep(
-                    new Step(
-                        number: $config->context->getCurrentStepNumber() + 1,
-                        toolCalls: new ToolCallCollection(),
-                    ),
+                    new Step($config->context->getCurrentStepNumber() + 1),
                 );
 
                 // Send the tool messages to the execution pipeline to get a new generation.

src/CortexServiceProvider.php

@@ -59,7 +59,7 @@ public function packageBooted(): void
         Cortex::registerAgent(new Agent(
             name: 'quote_of_the_day',
             prompt: 'Generate a quote of the day about {topic}.',
-            llm: 'ollama/phi4',
+            llm: 'ollama/gpt-oss:20b',
             output: [
                 Schema::string('quote')
                     ->description("Don't include the author in the quote. Just a single sentence.")

src/Http/Controllers/AgentsController.php

@@ -27,15 +27,15 @@ public function invoke(string $agent, Request $request): JsonResponse
             //     );
             // });
             $agent->onStepEnd(function (AgentStepEnd $event): void {
-                dump(
-                    sprintf('step end: %d', $event->config?->context?->getCurrentStepNumber()),
-                    $event->config?->toArray(),
-                );
+                // dump(
+                //     sprintf('step end: %d', $event->config?->context?->getCurrentStepNumber()),
+                //     $event->config?->toArray(),
+                // );
             });
             $agent->onStepError(function (AgentStepError $event): void {
-                dump(sprintf('step error: %d', $event->config?->context?->getCurrentStepNumber()));
-                dump($event->exception->getMessage());
-                dump($event->exception->getTraceAsString());
+                // dump(sprintf('step error: %d', $event->config?->context?->getCurrentStepNumber()));
+                // dump($event->exception->getMessage());
+                // dump($event->exception->getTraceAsString());
             });
             $result = $agent->invoke(input: $request->all());
         } catch (Throwable $e) {

src/Pipeline.php

@@ -90,14 +90,14 @@ public function invoke(mixed $payload = null, ?RuntimeConfig $config = null): mi
     /**
      * Ensures that all LLMs in the pipeline are streaming.
      */
-    public function stream(mixed $payload = null): mixed
+    public function stream(mixed $payload = null, ?RuntimeConfig $config = null): mixed
     {
-        return $this->enableStreaming()->invoke($payload);
+        return $this->enableStreaming()->invoke($payload, $config);
     }
 
-    public function __invoke(mixed $payload = null): mixed
+    public function __invoke(mixed $payload = null, ?RuntimeConfig $config = null): mixed
     {
-        return $this->invoke($payload);
+        return $this->invoke($payload, $config);
     }
 
     /**

src/Pipeline/Context.php

@@ -16,8 +16,11 @@
 class Context extends Fluent
 {
     public const string STEPS_KEY = 'steps';
+
     public const string CURRENT_STEP_KEY = 'current_step';
+
     public const string USAGE_SO_FAR_KEY = 'usage_so_far';
+
     public const string MESSAGE_HISTORY_KEY = 'message_history';
 
     /**

src/Pipeline/RuntimeConfig.php

@@ -5,8 +5,6 @@
 namespace Cortex\Pipeline;
 
 use Illuminate\Support\Str;
-use Cortex\Agents\Data\Step;
-use Illuminate\Support\Collection;
 use Illuminate\Contracts\Support\Arrayable;
 
 /**