BUG: Fix general test

Author: miroslav

tests/Unit/AIModels/GeneralTest.php

@@ -3,6 +3,7 @@
 namespace Unit\AIModels;
 
 use Illuminate\Contracts\Container\BindingResolutionException;
+use Illuminate\Support\Collection;
 use Itsimiro\OpenAI\Services\AIModels\General;
 use Itsimiro\OpenAI\Services\API\ApiService;
 use Itsimiro\OpenAI\Services\API\Results\ModelResult;
@@ -24,9 +25,8 @@ public function testGetModels(): void
 
         $expectedModel = $this->getMockModel();
 
-        $dataCollection = $this->getMockBuilder(DataCollection::class)
+        $dataCollection = $this->getMockBuilder(Collection::class)
             ->setConstructorArgs([
-                'dataClass' => ModelResult::class,
                 'items' => []
             ])->getMock();