Update applyAfterQuery return type to include Collection and MorphToMany

Author: luanfreitasdev

src/PowerGridComponent.php

@@ -9,6 +9,7 @@
 use Illuminate\Foundation\Application;
 use Illuminate\Pagination\AbstractPaginator;
 use Illuminate\Pagination\{LengthAwarePaginator, Paginator};
+use Illuminate\Support\Collection;
 use Illuminate\Support\{Collection as BaseCollection, Facades\Cache, Facades\DB};
 use Livewire\{Attributes\Computed, Component, WithPagination};
 use PowerComponents\LivewirePowerGrid\DataSource\ProcessDataSource;
@@ -202,7 +203,7 @@ private function getRecordsDataSource(): Paginator|MorphToMany|\Illuminate\Contr
         return $this->applyAfterQuery($processResult['results']);
     }
 
-    private function applyAfterQuery(mixed $results): mixed
+    private function applyAfterQuery(mixed $results): Paginator|MorphToMany|LengthAwarePaginator|\Illuminate\Contracts\Pagination\LengthAwarePaginator|BaseCollection
     {
         if ($results instanceof AbstractPaginator || $results instanceof \Illuminate\Contracts\Pagination\Paginator) {
             /** @var Paginator|LengthAwarePaginator|\Illuminate\Contracts\Pagination\LengthAwarePaginator $results */
@@ -215,6 +216,7 @@ private function applyAfterQuery(mixed $results): mixed
             return $this->transformRows($results);
         }
 
+        /** @var MorphToMany|Collection $results */
         return $results;
     }