fix: model retorna corretamente

O model estava retornando None no caso em que a classe precisava fazer
chamada para a OpenAI

openiziai/fine_tuning.py

@@ -127,6 +127,8 @@ def model(self) -> Optional[GPTModel]:
             name=model_name, base_model=self.base_model, task=self.task
         )
 
+        return self._model
+
     def __repr__(self) -> str:
         return (
             'FineTuning('

tests/conftest.py

@@ -49,7 +49,11 @@ def openai_fine_tuning():
     fine_tuning_mock = MagicMock(
         jobs=MagicMock(
             create=MagicMock(return_value=MagicMock(id='job-id')),
-            retrieve=MagicMock(return_value=MagicMock(status='succeeded')),
+            retrieve=MagicMock(
+                return_value=MagicMock(
+                    status='succeeded', fine_tuned_model='fine-tuned'
+                )
+            ),
         )
     )
 

tests/test_fine_tuning.py

@@ -72,3 +72,12 @@ def test_repr_class(fine_tuning, valid_task, openai_fine_tuning):
         f'model=None'
         ')'
     )
+
+
+def test_retrieve_fine_tuned_model(fine_tuning, valid_task):
+    with patch('builtins.open', mock_open(read_data='data')):
+        fine_tuning.upload_file_to_openai().start()
+        assert fine_tuning.model.name == 'fine-tuned'
+        assert fine_tuning.model.task == valid_task
+        assert fine_tuning.model.base_model == 'gpt-3.5-turbo'
+        assert fine_tuning.model.created_at