fix: python

Author: shortcuts

templates/python/api.mustache

@@ -11,9 +11,10 @@ from algoliasearch.search.models import (
   ReplaceAllObjectsWithTransformationResponse,
   ScopeType,
   SecuredApiKeyRestrictions,
+  WatchResponse,
 )
 
-from algoliasearch.ingestion.models import (WatchResponse)
+from algoliasearch.ingestion.models import WatchResponse as IngestionWatchResponse
 from algoliasearch.ingestion.config import IngestionConfig
 from algoliasearch.ingestion.client import (IngestionClient, IngestionClientSync)
 from algoliasearch.ingestion.models import Action as IngestionAction

templates/python/search_helpers.mustache

@@ -299,7 +299,7 @@
         wait_for_tasks: bool = False,
         batch_size: int = 1000,
         request_options: Optional[Union[dict, RequestOptions]] = None,
-    ) -> List[WatchResponse]:
+    ) -> List[IngestionWatchResponse]:
         """
         Helper: Similar to the `save_objects` method but requires a Push connector (https://www.algolia.com/doc/guides/sending-and-managing-data/send-and-update-your-data/connectors/push/) to be created first, in order to transform records before indexing them to Algolia. The `region` must've been passed to the client's config at instantiation.
         """
@@ -342,7 +342,7 @@
         wait_for_tasks: bool = False,
         batch_size: int = 1000,
         request_options: Optional[Union[dict, RequestOptions]] = None,
-    ) -> List[WatchResponse]:
+    ) -> List[IngestionWatchResponse]:
         """
         Helper: Similar to the `partial_update_objects` method but requires a Push connector (https://www.algolia.com/doc/guides/sending-and-managing-data/send-and-update-your-data/connectors/push/) to be created first, in order to transform records before indexing them to Algolia. The `region` must've been passed to the client instantiation method.
         """
@@ -443,9 +443,11 @@
               index_name=tmp_index_name, task_id=move_operation_response.task_id
           )
 
+          search_watch_responses: List[WatchResponse] = [WatchResponse.model_validate(wr.model_dump()) for wr in watch_responses]
+
           return ReplaceAllObjectsWithTransformationResponse(
               copy_operation_response=copy_operation_response,
-              watch_responses=watch_responses,
+              watch_responses=search_watch_responses,
               move_operation_response=move_operation_response,
           )
         except Exception as e: