feat: adds toFiles() method

Author: JasonTheAdams

src/Results/DTO/GenerativeAiResult.php

@@ -274,26 +274,26 @@ public function toTexts(): array
     }
 
     /**
-     * Converts all candidates to image files.
+     * Converts all candidates to files.
      *
      * @since n.e.x.t
      *
-     * @return FileInterface[] Array of image files.
+     * @return FileInterface[] Array of files.
      */
-    public function toImageFiles(): array
+    public function toFiles(): array
     {
         $files = [];
         foreach ($this->candidates as $candidate) {
             $message = $candidate->getMessage();
             foreach ($message->getParts() as $part) {
                 $inlineFile = $part->getInlineFile();
-                if ($inlineFile !== null && $inlineFile->getMimeType()->isImage()) {
+                if ($inlineFile !== null) {
                     $files[] = $inlineFile;
                     break;
                 }
 
                 $remoteFile = $part->getRemoteFile();
-                if ($remoteFile !== null && $remoteFile->getMimeType()->isImage()) {
+                if ($remoteFile !== null) {
                     $files[] = $remoteFile;
                     break;
                 }
@@ -302,6 +302,21 @@ public function toImageFiles(): array
         return $files;
     }
 
+    /**
+     * Converts all candidates to image files.
+     *
+     * @since n.e.x.t
+     *
+     * @return FileInterface[] Array of image files.
+     */
+    public function toImageFiles(): array
+    {
+        return array_filter(
+            $this->toFiles(),
+            fn(FileInterface $file) => $file->getMimeType()->isImage()
+        );
+    }
+
     /**
      * Converts all candidates to audio files.
      *
@@ -311,24 +326,10 @@ public function toImageFiles(): array
      */
     public function toAudioFiles(): array
     {
-        $files = [];
-        foreach ($this->candidates as $candidate) {
-            $message = $candidate->getMessage();
-            foreach ($message->getParts() as $part) {
-                $inlineFile = $part->getInlineFile();
-                if ($inlineFile !== null && $inlineFile->getMimeType()->isAudio()) {
-                    $files[] = $inlineFile;
-                    break;
-                }
-
-                $remoteFile = $part->getRemoteFile();
-                if ($remoteFile !== null && $remoteFile->getMimeType()->isAudio()) {
-                    $files[] = $remoteFile;
-                    break;
-                }
-            }
-        }
-        return $files;
+        return array_filter(
+            $this->toFiles(),
+            fn(FileInterface $file) => $file->getMimeType()->isAudio()
+        );
     }
 
     /**
@@ -340,24 +341,10 @@ public function toAudioFiles(): array
      */
     public function toVideoFiles(): array
     {
-        $files = [];
-        foreach ($this->candidates as $candidate) {
-            $message = $candidate->getMessage();
-            foreach ($message->getParts() as $part) {
-                $inlineFile = $part->getInlineFile();
-                if ($inlineFile !== null && $inlineFile->getMimeType()->isVideo()) {
-                    $files[] = $inlineFile;
-                    break;
-                }
-
-                $remoteFile = $part->getRemoteFile();
-                if ($remoteFile !== null && $remoteFile->getMimeType()->isVideo()) {
-                    $files[] = $remoteFile;
-                    break;
-                }
-            }
-        }
-        return $files;
+        return array_filter(
+            $this->toFiles(),
+            fn(FileInterface $file) => $file->getMimeType()->isVideo()
+        );
     }
 
     /**