blaxel 0.2.33__py3-none-any.whl → 0.2.35__py3-none-any.whl

blaxel/core/client/api/integrations/get_integration_connection.py

@@ -1,10 +1,11 @@
 from http import HTTPStatus
-from typing import Any
+from typing import Any, Union
 
 import httpx
 
 from ... import errors
 from ...client import Client
+from ...models.error import Error
 from ...models.integration_connection import IntegrationConnection
 from ...types import Response
 
@@ -20,18 +21,38 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(*, client: Client, response: httpx.Response) -> IntegrationConnection | None:
+def _parse_response(
+    *, client: Client, response: httpx.Response
+) -> Union[Error, IntegrationConnection] | None:
     if response.status_code == 200:
         response_200 = IntegrationConnection.from_dict(response.json())
 
         return response_200
+    if response.status_code == 401:
+        response_401 = Error.from_dict(response.json())
+
+        return response_401
+    if response.status_code == 403:
+        response_403 = Error.from_dict(response.json())
+
+        return response_403
+    if response.status_code == 404:
+        response_404 = Error.from_dict(response.json())
+
+        return response_404
+    if response.status_code == 500:
+        response_500 = Error.from_dict(response.json())
+
+        return response_500
     if client.raise_on_unexpected_status:
         raise errors.UnexpectedStatus(response.status_code, response.content)
     else:
         return None
 
 
-def _build_response(*, client: Client, response: httpx.Response) -> Response[IntegrationConnection]:
+def _build_response(
+    *, client: Client, response: httpx.Response
+) -> Response[Union[Error, IntegrationConnection]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -44,10 +65,11 @@ def sync_detailed(
     connection_name: str,
     *,
     client: Client,
-) -> Response[IntegrationConnection]:
-    """Get integration
+) -> Response[Union[Error, IntegrationConnection]]:
+    """Get integration connection
 
-     Returns an integration connection by integration name and connection name.
+     Returns detailed information about an integration connection including its provider type,
+    configuration (secrets are masked), and usage status.
 
     Args:
         connection_name (str):
@@ -57,7 +79,7 @@ def sync_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[IntegrationConnection]
+        Response[Union[Error, IntegrationConnection]]
     """
 
     kwargs = _get_kwargs(
@@ -75,10 +97,11 @@ def sync(
     connection_name: str,
     *,
     client: Client,
-) -> IntegrationConnection | None:
-    """Get integration
+) -> Union[Error, IntegrationConnection] | None:
+    """Get integration connection
 
-     Returns an integration connection by integration name and connection name.
+     Returns detailed information about an integration connection including its provider type,
+    configuration (secrets are masked), and usage status.
 
     Args:
         connection_name (str):
@@ -88,7 +111,7 @@ def sync(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        IntegrationConnection
+        Union[Error, IntegrationConnection]
     """
 
     return sync_detailed(
@@ -101,10 +124,11 @@ async def asyncio_detailed(
     connection_name: str,
     *,
     client: Client,
-) -> Response[IntegrationConnection]:
-    """Get integration
+) -> Response[Union[Error, IntegrationConnection]]:
+    """Get integration connection
 
-     Returns an integration connection by integration name and connection name.
+     Returns detailed information about an integration connection including its provider type,
+    configuration (secrets are masked), and usage status.
 
     Args:
         connection_name (str):
@@ -114,7 +138,7 @@ async def asyncio_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[IntegrationConnection]
+        Response[Union[Error, IntegrationConnection]]
     """
 
     kwargs = _get_kwargs(
@@ -130,10 +154,11 @@ async def asyncio(
     connection_name: str,
     *,
     client: Client,
-) -> IntegrationConnection | None:
-    """Get integration
+) -> Union[Error, IntegrationConnection] | None:
+    """Get integration connection
 
-     Returns an integration connection by integration name and connection name.
+     Returns detailed information about an integration connection including its provider type,
+    configuration (secrets are masked), and usage status.
 
     Args:
         connection_name (str):
@@ -143,7 +168,7 @@ async def asyncio(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        IntegrationConnection
+        Union[Error, IntegrationConnection]
     """
 
     return (

blaxel/core/client/api/integrations/get_integration_connection_model.py

@@ -44,7 +44,7 @@ def sync_detailed(
     *,
     client: Client,
 ) -> Response[Any]:
-    """Get integration model endpoint configurations
+    """Get integration connection model
 
      Returns a model for an integration connection by ID.
 
@@ -78,7 +78,7 @@ async def asyncio_detailed(
     *,
     client: Client,
 ) -> Response[Any]:
-    """Get integration model endpoint configurations
+    """Get integration connection model
 
      Returns a model for an integration connection by ID.
 

blaxel/core/client/api/integrations/list_integration_connections.py

@@ -1,10 +1,11 @@
 from http import HTTPStatus
-from typing import Any
+from typing import Any, Union
 
 import httpx
 
 from ... import errors
 from ...client import Client
+from ...models.error import Error
 from ...models.integration_connection import IntegrationConnection
 from ...types import Response
 
@@ -20,7 +21,7 @@ def _get_kwargs() -> dict[str, Any]:
 
 def _parse_response(
     *, client: Client, response: httpx.Response
-) -> list["IntegrationConnection"] | None:
+) -> Union[Error, list["IntegrationConnection"]] | None:
     if response.status_code == 200:
         response_200 = []
         _response_200 = response.json()
@@ -30,6 +31,18 @@ def _parse_response(
             response_200.append(response_200_item)
 
         return response_200
+    if response.status_code == 401:
+        response_401 = Error.from_dict(response.json())
+
+        return response_401
+    if response.status_code == 403:
+        response_403 = Error.from_dict(response.json())
+
+        return response_403
+    if response.status_code == 500:
+        response_500 = Error.from_dict(response.json())
+
+        return response_500
     if client.raise_on_unexpected_status:
         raise errors.UnexpectedStatus(response.status_code, response.content)
     else:
@@ -38,7 +51,7 @@ def _parse_response(
 
 def _build_response(
     *, client: Client, response: httpx.Response
-) -> Response[list["IntegrationConnection"]]:
+) -> Response[Union[Error, list["IntegrationConnection"]]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -50,17 +63,18 @@ def _build_response(
 def sync_detailed(
     *,
     client: Client,
-) -> Response[list["IntegrationConnection"]]:
-    """List integrations connections
+) -> Response[Union[Error, list["IntegrationConnection"]]]:
+    """List integration connections
 
-     Returns a list of all connections integrations in the workspace.
+     Returns all configured integration connections in the workspace. Each connection stores credentials
+    and settings for an external service (LLM provider, API, database).
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[list['IntegrationConnection']]
+        Response[Union[Error, list['IntegrationConnection']]]
     """
 
     kwargs = _get_kwargs()
@@ -75,17 +89,18 @@ def sync_detailed(
 def sync(
     *,
     client: Client,
-) -> list["IntegrationConnection"] | None:
-    """List integrations connections
+) -> Union[Error, list["IntegrationConnection"]] | None:
+    """List integration connections
 
-     Returns a list of all connections integrations in the workspace.
+     Returns all configured integration connections in the workspace. Each connection stores credentials
+    and settings for an external service (LLM provider, API, database).
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        list['IntegrationConnection']
+        Union[Error, list['IntegrationConnection']]
     """
 
     return sync_detailed(
@@ -96,17 +111,18 @@ def sync(
 async def asyncio_detailed(
     *,
     client: Client,
-) -> Response[list["IntegrationConnection"]]:
-    """List integrations connections
+) -> Response[Union[Error, list["IntegrationConnection"]]]:
+    """List integration connections
 
-     Returns a list of all connections integrations in the workspace.
+     Returns all configured integration connections in the workspace. Each connection stores credentials
+    and settings for an external service (LLM provider, API, database).
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[list['IntegrationConnection']]
+        Response[Union[Error, list['IntegrationConnection']]]
     """
 
     kwargs = _get_kwargs()
@@ -119,17 +135,18 @@ async def asyncio_detailed(
 async def asyncio(
     *,
     client: Client,
-) -> list["IntegrationConnection"] | None:
-    """List integrations connections
+) -> Union[Error, list["IntegrationConnection"]] | None:
+    """List integration connections
 
-     Returns a list of all connections integrations in the workspace.
+     Returns all configured integration connections in the workspace. Each connection stores credentials
+    and settings for an external service (LLM provider, API, database).
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        list['IntegrationConnection']
+        Union[Error, list['IntegrationConnection']]
     """
 
     return (

blaxel/core/client/api/integrations/update_integration_connection.py

@@ -1,10 +1,11 @@
 from http import HTTPStatus
-from typing import Any
+from typing import Any, Union
 
 import httpx
 
 from ... import errors
 from ...client import Client
+from ...models.error import Error
 from ...models.integration_connection import IntegrationConnection
 from ...types import Response
 
@@ -33,18 +34,42 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(*, client: Client, response: httpx.Response) -> IntegrationConnection | None:
+def _parse_response(
+    *, client: Client, response: httpx.Response
+) -> Union[Error, IntegrationConnection] | None:
     if response.status_code == 200:
         response_200 = IntegrationConnection.from_dict(response.json())
 
         return response_200
+    if response.status_code == 400:
+        response_400 = Error.from_dict(response.json())
+
+        return response_400
+    if response.status_code == 401:
+        response_401 = Error.from_dict(response.json())
+
+        return response_401
+    if response.status_code == 403:
+        response_403 = Error.from_dict(response.json())
+
+        return response_403
+    if response.status_code == 404:
+        response_404 = Error.from_dict(response.json())
+
+        return response_404
+    if response.status_code == 500:
+        response_500 = Error.from_dict(response.json())
+
+        return response_500
     if client.raise_on_unexpected_status:
         raise errors.UnexpectedStatus(response.status_code, response.content)
     else:
         return None
 
 
-def _build_response(*, client: Client, response: httpx.Response) -> Response[IntegrationConnection]:
+def _build_response(
+    *, client: Client, response: httpx.Response
+) -> Response[Union[Error, IntegrationConnection]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -58,21 +83,23 @@ def sync_detailed(
     *,
     client: Client,
     body: IntegrationConnection,
-) -> Response[IntegrationConnection]:
+) -> Response[Union[Error, IntegrationConnection]]:
     """Update integration connection
 
-     Update an integration connection by integration name and connection name.
+     Updates an integration connection's configuration or credentials. Changes take effect immediately
+    for all resources using this connection.
 
     Args:
         connection_name (str):
-        body (IntegrationConnection): Integration Connection
+        body (IntegrationConnection): Configured connection to an external service (LLM provider,
+            API, SaaS, database) storing credentials and settings for use by workspace resources.
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[IntegrationConnection]
+        Response[Union[Error, IntegrationConnection]]
     """
 
     kwargs = _get_kwargs(
@@ -92,21 +119,23 @@ def sync(
     *,
     client: Client,
     body: IntegrationConnection,
-) -> IntegrationConnection | None:
+) -> Union[Error, IntegrationConnection] | None:
     """Update integration connection
 
-     Update an integration connection by integration name and connection name.
+     Updates an integration connection's configuration or credentials. Changes take effect immediately
+    for all resources using this connection.
 
     Args:
         connection_name (str):
-        body (IntegrationConnection): Integration Connection
+        body (IntegrationConnection): Configured connection to an external service (LLM provider,
+            API, SaaS, database) storing credentials and settings for use by workspace resources.
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        IntegrationConnection
+        Union[Error, IntegrationConnection]
     """
 
     return sync_detailed(
@@ -121,21 +150,23 @@ async def asyncio_detailed(
     *,
     client: Client,
     body: IntegrationConnection,
-) -> Response[IntegrationConnection]:
+) -> Response[Union[Error, IntegrationConnection]]:
     """Update integration connection
 
-     Update an integration connection by integration name and connection name.
+     Updates an integration connection's configuration or credentials. Changes take effect immediately
+    for all resources using this connection.
 
     Args:
         connection_name (str):
-        body (IntegrationConnection): Integration Connection
+        body (IntegrationConnection): Configured connection to an external service (LLM provider,
+            API, SaaS, database) storing credentials and settings for use by workspace resources.
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[IntegrationConnection]
+        Response[Union[Error, IntegrationConnection]]
     """
 
     kwargs = _get_kwargs(
@@ -153,21 +184,23 @@ async def asyncio(
     *,
     client: Client,
     body: IntegrationConnection,
-) -> IntegrationConnection | None:
+) -> Union[Error, IntegrationConnection] | None:
     """Update integration connection
 
-     Update an integration connection by integration name and connection name.
+     Updates an integration connection's configuration or credentials. Changes take effect immediately
+    for all resources using this connection.
 
     Args:
         connection_name (str):
-        body (IntegrationConnection): Integration Connection
+        body (IntegrationConnection): Configured connection to an external service (LLM provider,
+            API, SaaS, database) storing credentials and settings for use by workspace resources.
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        IntegrationConnection
+        Union[Error, IntegrationConnection]
     """
 
     return (

blaxel/core/client/api/jobs/create_job.py

@@ -57,12 +57,14 @@ def sync_detailed(
     client: Client,
     body: Job,
 ) -> Response[Job]:
-    """Create job
+    """Create batch job
 
-     Creates a job.
+     Creates a new batch job definition for parallel AI task processing. Jobs can be triggered via API or
+    scheduled, and support configurable parallelism, timeouts, and retry logic.
 
     Args:
-        body (Job): Job
+        body (Job): Batch processing job definition for running parallel AI tasks. Jobs can
+            execute multiple tasks concurrently with configurable parallelism, retries, and timeouts.
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -88,12 +90,14 @@ def sync(
     client: Client,
     body: Job,
 ) -> Job | None:
-    """Create job
+    """Create batch job
 
-     Creates a job.
+     Creates a new batch job definition for parallel AI task processing. Jobs can be triggered via API or
+    scheduled, and support configurable parallelism, timeouts, and retry logic.
 
     Args:
-        body (Job): Job
+        body (Job): Batch processing job definition for running parallel AI tasks. Jobs can
+            execute multiple tasks concurrently with configurable parallelism, retries, and timeouts.
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -114,12 +118,14 @@ async def asyncio_detailed(
     client: Client,
     body: Job,
 ) -> Response[Job]:
-    """Create job
+    """Create batch job
 
-     Creates a job.
+     Creates a new batch job definition for parallel AI task processing. Jobs can be triggered via API or
+    scheduled, and support configurable parallelism, timeouts, and retry logic.
 
     Args:
-        body (Job): Job
+        body (Job): Batch processing job definition for running parallel AI tasks. Jobs can
+            execute multiple tasks concurrently with configurable parallelism, retries, and timeouts.
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -143,12 +149,14 @@ async def asyncio(
     client: Client,
     body: Job,
 ) -> Job | None:
-    """Create job
+    """Create batch job
 
-     Creates a job.
+     Creates a new batch job definition for parallel AI task processing. Jobs can be triggered via API or
+    scheduled, and support configurable parallelism, timeouts, and retry logic.
 
     Args:
-        body (Job): Job
+        body (Job): Batch processing job definition for running parallel AI tasks. Jobs can
+            execute multiple tasks concurrently with configurable parallelism, retries, and timeouts.
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.

blaxel/core/client/api/jobs/create_job_execution.py

@@ -6,7 +6,7 @@ import httpx
 from ... import errors
 from ...client import Client
 from ...models.create_job_execution_request import CreateJobExecutionRequest
-from ...models.job_execution import JobExecution
+from ...models.create_job_execution_response import CreateJobExecutionResponse
 from ...types import Response
 
 
@@ -34,9 +34,11 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(*, client: Client, response: httpx.Response) -> Union[Any, JobExecution] | None:
+def _parse_response(
+    *, client: Client, response: httpx.Response
+) -> Union[Any, CreateJobExecutionResponse] | None:
     if response.status_code == 200:
-        response_200 = JobExecution.from_dict(response.json())
+        response_200 = CreateJobExecutionResponse.from_dict(response.json())
 
         return response_200
     if response.status_code == 400:
@@ -53,7 +55,7 @@ def _parse_response(*, client: Client, response: httpx.Response) -> Union[Any, J
 
 def _build_response(
     *, client: Client, response: httpx.Response
-) -> Response[Union[Any, JobExecution]]:
+) -> Response[Union[Any, CreateJobExecutionResponse]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -67,10 +69,11 @@ def sync_detailed(
     *,
     client: Client,
     body: CreateJobExecutionRequest,
-) -> Response[Union[Any, JobExecution]]:
+) -> Response[Union[Any, CreateJobExecutionResponse]]:
     """Create job execution
 
-     Creates a new execution for a job by name.
+     Triggers a new execution of the batch job. Each execution runs multiple tasks in parallel according
+    to the job's configured concurrency. Tasks can be parameterized via the request body.
 
     Args:
         job_id (str):
@@ -81,7 +84,7 @@ def sync_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[Union[Any, JobExecution]]
+        Response[Union[Any, CreateJobExecutionResponse]]
     """
 
     kwargs = _get_kwargs(
@@ -101,10 +104,11 @@ def sync(
     *,
     client: Client,
     body: CreateJobExecutionRequest,
-) -> Union[Any, JobExecution] | None:
+) -> Union[Any, CreateJobExecutionResponse] | None:
     """Create job execution
 
-     Creates a new execution for a job by name.
+     Triggers a new execution of the batch job. Each execution runs multiple tasks in parallel according
+    to the job's configured concurrency. Tasks can be parameterized via the request body.
 
     Args:
         job_id (str):
@@ -115,7 +119,7 @@ def sync(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Union[Any, JobExecution]
+        Union[Any, CreateJobExecutionResponse]
     """
 
     return sync_detailed(
@@ -130,10 +134,11 @@ async def asyncio_detailed(
     *,
     client: Client,
     body: CreateJobExecutionRequest,
-) -> Response[Union[Any, JobExecution]]:
+) -> Response[Union[Any, CreateJobExecutionResponse]]:
     """Create job execution
 
-     Creates a new execution for a job by name.
+     Triggers a new execution of the batch job. Each execution runs multiple tasks in parallel according
+    to the job's configured concurrency. Tasks can be parameterized via the request body.
 
     Args:
         job_id (str):
@@ -144,7 +149,7 @@ async def asyncio_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[Union[Any, JobExecution]]
+        Response[Union[Any, CreateJobExecutionResponse]]
     """
 
     kwargs = _get_kwargs(
@@ -162,10 +167,11 @@ async def asyncio(
     *,
     client: Client,
     body: CreateJobExecutionRequest,
-) -> Union[Any, JobExecution] | None:
+) -> Union[Any, CreateJobExecutionResponse] | None:
     """Create job execution
 
-     Creates a new execution for a job by name.
+     Triggers a new execution of the batch job. Each execution runs multiple tasks in parallel according
+    to the job's configured concurrency. Tasks can be parameterized via the request body.
 
     Args:
         job_id (str):
@@ -176,7 +182,7 @@ async def asyncio(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Union[Any, JobExecution]
+        Union[Any, CreateJobExecutionResponse]
     """
 
     return (