[FIX] Update generated rest clients to latest version, enabling pydan…

…tic2 interop

GitOrigin-RevId: b74dde28f5a25bedaa02397961e3f2cbdc94ef30

src/gretel_client/_hybrid/connections_api.py

@@ -9,6 +9,7 @@
     UpdateConnectionRequest,
 )
 from gretel_client.rest_v1.api_response import ApiResponse
+from gretel_client.rest_v1.models.connection import Connection
 
 
 class HybridConnectionsApi(ConnectionsApi):
@@ -43,16 +44,29 @@ def __init__(
         super().__init__(api.api_client)
         self._creds_encryption = creds_encryption
 
-    def create_connection_with_http_info(
-        self, create_connection_request: CreateConnectionRequest, **kwargs
-    ) -> ApiResponse:
+    def _encrypt_creds(
+        self, create_connection_request: CreateConnectionRequest
+    ) -> CreateConnectionRequest:
         if creds := create_connection_request.credentials:
             create_connection_request.credentials = None
             encrypted_creds = self._creds_encryption.apply(
                 creds, project_guid=create_connection_request.project_id
             )
             create_connection_request.encrypted_credentials = encrypted_creds
+        return create_connection_request
 
+    def create_connection(
+        self,
+        create_connection_request: CreateConnectionRequest,
+        **kwargs,
+    ) -> Connection:
+        create_connection_request = self._encrypt_creds(create_connection_request)
+        return super().create_connection(create_connection_request, **kwargs)
+
+    def create_connection_with_http_info(
+        self, create_connection_request: CreateConnectionRequest, **kwargs
+    ) -> ApiResponse:
+        create_connection_request = self._encrypt_creds(create_connection_request)
         return super().create_connection_with_http_info(
             create_connection_request, **kwargs
         )

src/gretel_client/_hybrid/workflows_api.py

@@ -4,6 +4,7 @@
 
 from gretel_client.rest_v1.api.workflows_api import CreateWorkflowRequest, WorkflowsApi
 from gretel_client.rest_v1.api_response import ApiResponse
+from gretel_client.rest_v1.models.workflow import Workflow
 from gretel_client.workflows.runner_mode import RunnerMode
 
 
@@ -25,18 +26,18 @@ class HybridWorkflowsApi(WorkflowsApi):
     def __init__(self, api: WorkflowsApi):
         super().__init__(api.api_client)
 
-    # We only override the ..._with_http_info methods, as these are
-    # the lower-level ones.
+    def create_workflow(
+        self,
+        create_workflow_request: CreateWorkflowRequest,
+        **kwargs,
+    ) -> Workflow:
+        create_workflow_request = _update_runner_mode(create_workflow_request)
+        return super().create_workflow(create_workflow_request, **kwargs)
 
     def create_workflow_with_http_info(
         self, create_workflow_request: CreateWorkflowRequest, **kwargs
     ) -> ApiResponse:
-        if create_workflow_request.runner_mode is None:
-            create_workflow_request.runner_mode = RunnerMode.RUNNER_MODE_HYBRID.value
-        elif create_workflow_request.runner_mode != RunnerMode.RUNNER_MODE_HYBRID.value:
-            raise ValueError(
-                "in hybrid mode, only workflows with runner mode RUNNER_MODE_HYBRID can be created"
-            )
+        create_workflow_request = _update_runner_mode(create_workflow_request)
         return super().create_workflow_with_http_info(create_workflow_request, **kwargs)
 
     def validate_workflow_action_with_http_info(
@@ -54,3 +55,15 @@ def validate_workflow_action_with_http_info(
         return super().validate_workflow_action_with_http_info(
             body, runner_mode, **kwargs
         )
+
+
+def _update_runner_mode(
+    create_workflow_request: CreateWorkflowRequest,
+) -> CreateWorkflowRequest:
+    if create_workflow_request.runner_mode is None:
+        create_workflow_request.runner_mode = RunnerMode.RUNNER_MODE_HYBRID.value
+    elif create_workflow_request.runner_mode != RunnerMode.RUNNER_MODE_HYBRID.value:
+        raise ValueError(
+            "in hybrid mode, only workflows with runner mode RUNNER_MODE_HYBRID can be created"
+        )
+    return create_workflow_request