James Abrahams
--- a/tests/src/Unit/OpenAiChatMessageIteratorTest.php 0 → 100644
+++ b/tests/src/Unit/OpenAiChatMessageIteratorTest.php 0 → 100644
+<?php
+
+namespace Drupal\Tests\ai_provider_openai\Unit;
+
+use Drupal\ai\OperationType\Chat\Tools\ChatFunction;
+use Drupal\ai\OperationType\Chat\Tools\ChatTools;
+use Drupal\ai_provider_openai\OpenAiChatMessageIterator;
--- a/src/OpenAiChatMessageIterator.php
+++ b/src/OpenAiChatMessageIterator.php
+   * @var \Drupal\ai\OperationType\Chat\Tools\ChatTools|null
+   */
+  protected $chatTools = NULL;
+
+  /**
+   * Constructor.
+   *
+   * @param \IteratorAggregate $iterator
+   *   The iterator to wrap.
+   * @param \Drupal\ai\OperationType\Chat\Tools\ChatTools|null $chatTools
+   *   Optional chat tools from the input.
+   */
+  public function __construct(\IteratorAggregate $iterator, $chatTools = NULL) {
+    parent::__construct($iterator);
+    $this->chatTools = $chatTools;
+  }
-  }
+  public function __construct(\IteratorAggregate $iterator, $chatTools) {
+    parent::__construct($iterator);
+    $this->chatTools = $chatTools;
+  }
+
+ /**
+   * Constructor.
+   *
+   * @param \IteratorAggregate $iterator
+   *   The iterator to wrap.
+   */
+  public function __construct(\IteratorAggregate $iterator) {
+    parent::__construct($iterator);
+  }