From 9f85d7a75bb2b131b3df7c025f71ef8b887ebe2b Mon Sep 17 00:00:00 2001
From: Arjay Angeles <aqangeles@gmail.com>
Date: Tue, 18 Mar 2025 14:37:33 +0800
Subject: [PATCH] fix: stubs with generics

---
 src/Generators/stubs/builder.stub    | 4 +++-
 src/Generators/stubs/datatables.stub | 4 +++-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/Generators/stubs/builder.stub b/src/Generators/stubs/builder.stub
index 2ed2af7..511be79 100644
--- a/src/Generators/stubs/builder.stub
+++ b/src/Generators/stubs/builder.stub
@@ -13,7 +13,7 @@ class DummyClass extends DataTable
     /**
      * Build the DataTable class.
      *
-     * @param QueryBuilder $query Results from query() method.
+     * @param QueryBuilder<ModelName> $query Results from query() method.
      */
     public function dataTable(QueryBuilder $query): EloquentDataTable
     {
@@ -24,6 +24,8 @@ class DummyClass extends DataTable
 
     /**
      * Get the query source of dataTable.
+     *
+     * @return QueryBuilder<ModelName>
      */
     public function query(ModelName $model): QueryBuilder
     {
diff --git a/src/Generators/stubs/datatables.stub b/src/Generators/stubs/datatables.stub
index bccfede..1a574c1 100644
--- a/src/Generators/stubs/datatables.stub
+++ b/src/Generators/stubs/datatables.stub
@@ -17,7 +17,7 @@ class DummyClass extends DataTable
     /**
      * Build the DataTable class.
      *
-     * @param QueryBuilder $query Results from query() method.
+     * @param QueryBuilder<ModelName> $query Results from query() method.
      */
     public function dataTable(QueryBuilder $query): EloquentDataTable
     {
@@ -28,6 +28,8 @@ class DummyClass extends DataTable
 
     /**
      * Get the query source of dataTable.
+     *
+     * @return QueryBuilder<ModelName>
      */
     public function query(ModelName $model): QueryBuilder
     {