Minor fix

Author: bryanvimigo

src/Criteria/OnlyTrashedCriteria.php

@@ -21,7 +21,9 @@ class OnlyTrashedCriteria implements CriteriaInterface
      */
     public function apply(Model|Builder $model, RepositoryInterface $repository): mixed
     {
-        if (!method_exists($model, 'onlyTrashed'))
+        $model = $model instanceof Builder ? $model->getModel() : $model;
+
+        if (!in_array(SoftDeletes::class, class_uses($model)))
             throw new RepositoryException('Model must implement SoftDeletes Trait to use this criteria.');
 
         return $model->onlyTrashed();

src/Criteria/WithTrashedCriteria.php

@@ -21,7 +21,9 @@ class WithTrashedCriteria implements CriteriaInterface
      */
     public function apply(Model|Builder $model, RepositoryInterface $repository): mixed
     {
-        if (!method_exists($model, 'withTrashed'))
+        $model = $model instanceof Builder ? $model->getModel() : $model;
+
+        if (!in_array(SoftDeletes::class, class_uses($model)))
             throw new RepositoryException('Model must implement SoftDeletes Trait to use this criteria.');
 
         return $model->withTrashed();