Fix for mypy 1.16 (#101)

Author: vhaldemar

src/yandex_cloud_ml_sdk/_models/completions/model.py

@@ -259,8 +259,8 @@ class AsyncGPTModel(
     ],
     AsyncModelBatchMixin,
 ):
-    _operation_type = AsyncOperation
-    _tune_operation_type = AsyncTuningTask
+    _operation_type = AsyncOperation[GPTModelResult[AsyncToolCall]]
+    _tune_operation_type = AsyncTuningTask['AsyncGPTModel']
     _result_type = GPTModelResult[AsyncToolCall]
 
     async def run(
@@ -396,8 +396,8 @@ class GPTModel(
     ],
     ModelBatchMixin,
 ):
-    _operation_type = Operation
-    _tune_operation_type = TuningTask
+    _operation_type = Operation[GPTModelResult[ToolCall]]
+    _tune_operation_type = TuningTask['GPTModel']
     _result_type = GPTModelResult[ToolCall]
     __run = run_sync(BaseGPTModel._run)
     __run_stream = run_sync_generator(BaseGPTModel._run_stream)

src/yandex_cloud_ml_sdk/_models/text_classifiers/model.py

@@ -135,7 +135,7 @@ async def _run_few_shot(
 
 
 class AsyncTextClassifiersModel(BaseTextClassifiersModel[AsyncTuningTask['AsyncTextClassifiersModel']]):
-    _tune_operation_type = AsyncTuningTask
+    _tune_operation_type = AsyncTuningTask['AsyncTextClassifiersModel']
 
     async def run(
         self,
@@ -234,7 +234,7 @@ async def attach_tune_deferred(
 
 
 class TextClassifiersModel(BaseTextClassifiersModel[TuningTask['TextClassifiersModel']]):
-    _tune_operation_type = TuningTask
+    _tune_operation_type = TuningTask['TextClassifiersModel']
     __run = run_sync(BaseTextClassifiersModel._run)
     __tune_deferred = run_sync(BaseTextClassifiersModel._tune_deferred)
     __tune = run_sync(BaseTextClassifiersModel._tune)

src/yandex_cloud_ml_sdk/_search_indexes/domain.py

@@ -131,7 +131,7 @@ async def _list(
 
 class AsyncSearchIndexes(BaseSearchIndexes[AsyncSearchIndex, AsyncOperation[AsyncSearchIndex]]):
     _impl = AsyncSearchIndex
-    _operation_type = AsyncOperation
+    _operation_type = AsyncOperation[AsyncSearchIndex]
 
     async def create_deferred(
         self,
@@ -182,7 +182,7 @@ async def list(
 
 class SearchIndexes(BaseSearchIndexes[SearchIndex, Operation[SearchIndex]]):
     _impl = SearchIndex
-    _operation_type = Operation
+    _operation_type = Operation[SearchIndex]
 
     __get = run_sync(BaseSearchIndexes._get)
     __create_deferred = run_sync(BaseSearchIndexes._create_deferred)