orca-sdk 0.0.95__py3-none-any.whl → 0.0.97__py3-none-any.whl

orca_sdk/_generated_api_client/api/datasource/create_datasource_from_content_datasource_post.py

@@ -0,0 +1,224 @@
+"""
+This file is generated by the openapi-python-client tool via the generate_api_client.py script
+
+It is a customized template from the openapi-python-client tool's default template:
+https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/endpoint_module.py.jinja
+
+The main changes are:
+- Update the API call responses to either return the successful response type or raise an error by:
+    - Updating the _parse_response function to raise an error if the response status code is not in the 2xx range
+- Inject a client into every method via a context manager
+"""
+
+from http import HTTPStatus
+from typing import Any, Dict
+
+import httpx
+
+from ...client import _client_context
+from ...errors import get_error_for_response
+from ...models.create_datasource_from_content_request import CreateDatasourceFromContentRequest
+from ...models.datasource_metadata import DatasourceMetadata
+from ...types import Response
+
+
+def _get_kwargs(
+    *,
+    body: CreateDatasourceFromContentRequest,
+) -> dict[str, Any]:
+    headers: Dict[str, Any] = {}
+
+    _kwargs: dict[str, Any] = {
+        "method": "post",
+        "url": "/datasource/",
+    }
+
+    _body = body.to_dict()
+
+    _kwargs["json"] = _body
+    headers["Content-Type"] = "application/json"
+
+    _kwargs["headers"] = headers
+    return _kwargs
+
+
+def _parse_response(*, response: httpx.Response) -> DatasourceMetadata:
+    if response.status_code == 201:
+        response_201 = DatasourceMetadata.from_dict(response.json())
+
+        return response_201
+    if response.status_code == 401:
+        raise get_error_for_response(response)
+    if response.status_code == 500:
+        raise get_error_for_response(response)
+    if response.status_code == 503:
+        raise get_error_for_response(response)
+    else:
+        raise get_error_for_response(response)
+
+
+def _build_response(*, response: httpx.Response) -> Response[DatasourceMetadata]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(response=response),
+    )
+
+
+def sync_detailed(
+    *,
+    body: CreateDatasourceFromContentRequest,
+) -> Response[DatasourceMetadata]:
+    r"""Create Datasource From Content
+
+     Create a datasource from JSON content.
+
+    Automatically detects and supports multiple JSON formats:
+    - List of records: [{\"col1\": val1, \"col2\": val2}, {\"col1\": val3, \"col2\": val4}, ...]
+    - Dictionary of columns: {\"col1\": [val1, val3, ...], \"col2\": [val2, val4, ...]}
+
+    Params:
+        auth_context: Authentication context
+        request: Request containing name, description, and JSON content
+
+    Returns:
+        Metadata for the created datasource
+
+    Args:
+        body (CreateDatasourceFromContentRequest): Request model for creating a datasource from
+            JSON content.
+
+    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[DatasourceMetadata]
+    """
+
+    kwargs = _get_kwargs(
+        body=body,
+    )
+
+    with _client_context() as client:
+        response = client.get_httpx_client().request(
+            **kwargs,
+        )
+
+        return _build_response(response=response)
+
+
+def sync(
+    *,
+    body: CreateDatasourceFromContentRequest,
+) -> DatasourceMetadata:
+    r"""Create Datasource From Content
+
+     Create a datasource from JSON content.
+
+    Automatically detects and supports multiple JSON formats:
+    - List of records: [{\"col1\": val1, \"col2\": val2}, {\"col1\": val3, \"col2\": val4}, ...]
+    - Dictionary of columns: {\"col1\": [val1, val3, ...], \"col2\": [val2, val4, ...]}
+
+    Params:
+        auth_context: Authentication context
+        request: Request containing name, description, and JSON content
+
+    Returns:
+        Metadata for the created datasource
+
+    Args:
+        body (CreateDatasourceFromContentRequest): Request model for creating a datasource from
+            JSON content.
+
+    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:
+        DatasourceMetadata
+    """
+
+    return sync_detailed(
+        body=body,
+    ).parsed
+
+
+async def asyncio_detailed(
+    *,
+    body: CreateDatasourceFromContentRequest,
+) -> Response[DatasourceMetadata]:
+    r"""Create Datasource From Content
+
+     Create a datasource from JSON content.
+
+    Automatically detects and supports multiple JSON formats:
+    - List of records: [{\"col1\": val1, \"col2\": val2}, {\"col1\": val3, \"col2\": val4}, ...]
+    - Dictionary of columns: {\"col1\": [val1, val3, ...], \"col2\": [val2, val4, ...]}
+
+    Params:
+        auth_context: Authentication context
+        request: Request containing name, description, and JSON content
+
+    Returns:
+        Metadata for the created datasource
+
+    Args:
+        body (CreateDatasourceFromContentRequest): Request model for creating a datasource from
+            JSON content.
+
+    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[DatasourceMetadata]
+    """
+
+    kwargs = _get_kwargs(
+        body=body,
+    )
+
+    with _client_context() as client:
+        response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(response=response)
+
+
+async def asyncio(
+    *,
+    body: CreateDatasourceFromContentRequest,
+) -> DatasourceMetadata:
+    r"""Create Datasource From Content
+
+     Create a datasource from JSON content.
+
+    Automatically detects and supports multiple JSON formats:
+    - List of records: [{\"col1\": val1, \"col2\": val2}, {\"col1\": val3, \"col2\": val4}, ...]
+    - Dictionary of columns: {\"col1\": [val1, val3, ...], \"col2\": [val2, val4, ...]}
+
+    Params:
+        auth_context: Authentication context
+        request: Request containing name, description, and JSON content
+
+    Returns:
+        Metadata for the created datasource
+
+    Args:
+        body (CreateDatasourceFromContentRequest): Request model for creating a datasource from
+            JSON content.
+
+    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:
+        DatasourceMetadata
+    """
+
+    return (
+        await asyncio_detailed(
+            body=body,
+        )
+    ).parsed

orca_sdk/_generated_api_client/api/datasource/create_datasource_from_files_datasource_upload_post.py

@@ -0,0 +1,229 @@
+"""
+This file is generated by the openapi-python-client tool via the generate_api_client.py script
+
+It is a customized template from the openapi-python-client tool's default template:
+https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/endpoint_module.py.jinja
+
+The main changes are:
+- Update the API call responses to either return the successful response type or raise an error by:
+    - Updating the _parse_response function to raise an error if the response status code is not in the 2xx range
+- Inject a client into every method via a context manager
+"""
+
+from http import HTTPStatus
+from typing import Any, Dict
+
+import httpx
+
+from ...client import _client_context
+from ...errors import get_error_for_response
+from ...models.body_create_datasource_from_files_datasource_upload_post import (
+    BodyCreateDatasourceFromFilesDatasourceUploadPost,
+)
+from ...models.datasource_metadata import DatasourceMetadata
+from ...types import Response
+
+
+def _get_kwargs(
+    *,
+    body: BodyCreateDatasourceFromFilesDatasourceUploadPost,
+) -> dict[str, Any]:
+    headers: Dict[str, Any] = {}
+
+    _kwargs: dict[str, Any] = {
+        "method": "post",
+        "url": "/datasource/upload",
+    }
+
+    _body = body.to_multipart()
+
+    _kwargs["files"] = _body
+
+    _kwargs["headers"] = headers
+    return _kwargs
+
+
+def _parse_response(*, response: httpx.Response) -> DatasourceMetadata:
+    if response.status_code == 201:
+        response_201 = DatasourceMetadata.from_dict(response.json())
+
+        return response_201
+    if response.status_code == 401:
+        raise get_error_for_response(response)
+    if response.status_code == 500:
+        raise get_error_for_response(response)
+    if response.status_code == 503:
+        raise get_error_for_response(response)
+    else:
+        raise get_error_for_response(response)
+
+
+def _build_response(*, response: httpx.Response) -> Response[DatasourceMetadata]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(response=response),
+    )
+
+
+def sync_detailed(
+    *,
+    body: BodyCreateDatasourceFromFilesDatasourceUploadPost,
+) -> Response[DatasourceMetadata]:
+    """Create Datasource From Files
+
+     Create a datasource by uploading files.
+
+    Supports multiple file upload scenarios:
+    - Multiple files: HuggingFace Dataset format (dataset_info.json, state.json, .arrow files, etc.)
+    - Single file: CSV, JSON, JSONL, Parquet, or Pickle files
+
+    Params:
+        auth_context: Authentication context
+        files: List of files to upload
+        name: Name for the datasource
+        description: Optional description for the datasource
+
+    Returns:
+        Metadata for the created datasource
+
+    Args:
+        body (BodyCreateDatasourceFromFilesDatasourceUploadPost):
+
+    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[DatasourceMetadata]
+    """
+
+    kwargs = _get_kwargs(
+        body=body,
+    )
+
+    with _client_context() as client:
+        response = client.get_httpx_client().request(
+            **kwargs,
+        )
+
+        return _build_response(response=response)
+
+
+def sync(
+    *,
+    body: BodyCreateDatasourceFromFilesDatasourceUploadPost,
+) -> DatasourceMetadata:
+    """Create Datasource From Files
+
+     Create a datasource by uploading files.
+
+    Supports multiple file upload scenarios:
+    - Multiple files: HuggingFace Dataset format (dataset_info.json, state.json, .arrow files, etc.)
+    - Single file: CSV, JSON, JSONL, Parquet, or Pickle files
+
+    Params:
+        auth_context: Authentication context
+        files: List of files to upload
+        name: Name for the datasource
+        description: Optional description for the datasource
+
+    Returns:
+        Metadata for the created datasource
+
+    Args:
+        body (BodyCreateDatasourceFromFilesDatasourceUploadPost):
+
+    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:
+        DatasourceMetadata
+    """
+
+    return sync_detailed(
+        body=body,
+    ).parsed
+
+
+async def asyncio_detailed(
+    *,
+    body: BodyCreateDatasourceFromFilesDatasourceUploadPost,
+) -> Response[DatasourceMetadata]:
+    """Create Datasource From Files
+
+     Create a datasource by uploading files.
+
+    Supports multiple file upload scenarios:
+    - Multiple files: HuggingFace Dataset format (dataset_info.json, state.json, .arrow files, etc.)
+    - Single file: CSV, JSON, JSONL, Parquet, or Pickle files
+
+    Params:
+        auth_context: Authentication context
+        files: List of files to upload
+        name: Name for the datasource
+        description: Optional description for the datasource
+
+    Returns:
+        Metadata for the created datasource
+
+    Args:
+        body (BodyCreateDatasourceFromFilesDatasourceUploadPost):
+
+    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[DatasourceMetadata]
+    """
+
+    kwargs = _get_kwargs(
+        body=body,
+    )
+
+    with _client_context() as client:
+        response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(response=response)
+
+
+async def asyncio(
+    *,
+    body: BodyCreateDatasourceFromFilesDatasourceUploadPost,
+) -> DatasourceMetadata:
+    """Create Datasource From Files
+
+     Create a datasource by uploading files.
+
+    Supports multiple file upload scenarios:
+    - Multiple files: HuggingFace Dataset format (dataset_info.json, state.json, .arrow files, etc.)
+    - Single file: CSV, JSON, JSONL, Parquet, or Pickle files
+
+    Params:
+        auth_context: Authentication context
+        files: List of files to upload
+        name: Name for the datasource
+        description: Optional description for the datasource
+
+    Returns:
+        Metadata for the created datasource
+
+    Args:
+        body (BodyCreateDatasourceFromFilesDatasourceUploadPost):
+
+    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:
+        DatasourceMetadata
+    """
+
+    return (
+        await asyncio_detailed(
+            body=body,
+        )
+    ).parsed

orca_sdk/_generated_api_client/api/task/list_tasks_task_get.py

@@ -19,7 +19,7 @@ import httpx
 
 from ...client import _client_context
 from ...errors import get_error_for_response
-from ...models.task import Task
+from ...models.paginated_task import PaginatedTask
 from ...models.task_status import TaskStatus
 from ...types import UNSET, Response, Unset
 
@@ -28,7 +28,7 @@ def _get_kwargs(
     *,
     status: Union[List[TaskStatus], None, TaskStatus, Unset] = UNSET,
     type: Union[List[str], None, Unset, str] = UNSET,
-    limit: Union[None, Unset, int] = UNSET,
+    limit: Union[None, Unset, int] = 50,
     offset: Union[Unset, int] = 0,
     start_timestamp: Union[None, Unset, datetime.datetime] = UNSET,
     end_timestamp: Union[None, Unset, datetime.datetime] = UNSET,
@@ -100,14 +100,9 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(*, response: httpx.Response) -> List["Task"]:
+def _parse_response(*, response: httpx.Response) -> PaginatedTask:
     if response.status_code == 200:
-        response_200 = []
-        _response_200 = response.json()
-        for response_200_item_data in _response_200:
-            response_200_item = Task.from_dict(response_200_item_data)
-
-            response_200.append(response_200_item)
+        response_200 = PaginatedTask.from_dict(response.json())
 
         return response_200
     if response.status_code == 401:
@@ -122,7 +117,7 @@ def _parse_response(*, response: httpx.Response) -> List["Task"]:
         raise get_error_for_response(response)
 
 
-def _build_response(*, response: httpx.Response) -> Response[List["Task"]]:
+def _build_response(*, response: httpx.Response) -> Response[PaginatedTask]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -135,17 +130,17 @@ def sync_detailed(
     *,
     status: Union[List[TaskStatus], None, TaskStatus, Unset] = UNSET,
     type: Union[List[str], None, Unset, str] = UNSET,
-    limit: Union[None, Unset, int] = UNSET,
+    limit: Union[None, Unset, int] = 50,
     offset: Union[Unset, int] = 0,
     start_timestamp: Union[None, Unset, datetime.datetime] = UNSET,
     end_timestamp: Union[None, Unset, datetime.datetime] = UNSET,
-) -> Response[List["Task"]]:
+) -> Response[PaginatedTask]:
     """List Tasks
 
     Args:
         status (Union[List[TaskStatus], None, TaskStatus, Unset]):
         type (Union[List[str], None, Unset, str]):
-        limit (Union[None, Unset, int]):
+        limit (Union[None, Unset, int]):  Default: 50.
         offset (Union[Unset, int]):  Default: 0.
         start_timestamp (Union[None, Unset, datetime.datetime]):
         end_timestamp (Union[None, Unset, datetime.datetime]):
@@ -155,7 +150,7 @@ def sync_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[List['Task']]
+        Response[PaginatedTask]
     """
 
     kwargs = _get_kwargs(
@@ -179,17 +174,17 @@ def sync(
     *,
     status: Union[List[TaskStatus], None, TaskStatus, Unset] = UNSET,
     type: Union[List[str], None, Unset, str] = UNSET,
-    limit: Union[None, Unset, int] = UNSET,
+    limit: Union[None, Unset, int] = 50,
     offset: Union[Unset, int] = 0,
     start_timestamp: Union[None, Unset, datetime.datetime] = UNSET,
     end_timestamp: Union[None, Unset, datetime.datetime] = UNSET,
-) -> List["Task"]:
+) -> PaginatedTask:
     """List Tasks
 
     Args:
         status (Union[List[TaskStatus], None, TaskStatus, Unset]):
         type (Union[List[str], None, Unset, str]):
-        limit (Union[None, Unset, int]):
+        limit (Union[None, Unset, int]):  Default: 50.
         offset (Union[Unset, int]):  Default: 0.
         start_timestamp (Union[None, Unset, datetime.datetime]):
         end_timestamp (Union[None, Unset, datetime.datetime]):
@@ -199,7 +194,7 @@ def sync(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        List['Task']
+        PaginatedTask
     """
 
     return sync_detailed(
@@ -216,17 +211,17 @@ async def asyncio_detailed(
     *,
     status: Union[List[TaskStatus], None, TaskStatus, Unset] = UNSET,
     type: Union[List[str], None, Unset, str] = UNSET,
-    limit: Union[None, Unset, int] = UNSET,
+    limit: Union[None, Unset, int] = 50,
     offset: Union[Unset, int] = 0,
     start_timestamp: Union[None, Unset, datetime.datetime] = UNSET,
     end_timestamp: Union[None, Unset, datetime.datetime] = UNSET,
-) -> Response[List["Task"]]:
+) -> Response[PaginatedTask]:
     """List Tasks
 
     Args:
         status (Union[List[TaskStatus], None, TaskStatus, Unset]):
         type (Union[List[str], None, Unset, str]):
-        limit (Union[None, Unset, int]):
+        limit (Union[None, Unset, int]):  Default: 50.
         offset (Union[Unset, int]):  Default: 0.
         start_timestamp (Union[None, Unset, datetime.datetime]):
         end_timestamp (Union[None, Unset, datetime.datetime]):
@@ -236,7 +231,7 @@ async def asyncio_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[List['Task']]
+        Response[PaginatedTask]
     """
 
     kwargs = _get_kwargs(
@@ -258,17 +253,17 @@ async def asyncio(
     *,
     status: Union[List[TaskStatus], None, TaskStatus, Unset] = UNSET,
     type: Union[List[str], None, Unset, str] = UNSET,
-    limit: Union[None, Unset, int] = UNSET,
+    limit: Union[None, Unset, int] = 50,
     offset: Union[Unset, int] = 0,
     start_timestamp: Union[None, Unset, datetime.datetime] = UNSET,
     end_timestamp: Union[None, Unset, datetime.datetime] = UNSET,
-) -> List["Task"]:
+) -> PaginatedTask:
     """List Tasks
 
     Args:
         status (Union[List[TaskStatus], None, TaskStatus, Unset]):
         type (Union[List[str], None, Unset, str]):
-        limit (Union[None, Unset, int]):
+        limit (Union[None, Unset, int]):  Default: 50.
         offset (Union[Unset, int]):  Default: 0.
         start_timestamp (Union[None, Unset, datetime.datetime]):
         end_timestamp (Union[None, Unset, datetime.datetime]):
@@ -278,7 +273,7 @@ async def asyncio(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        List['Task']
+        PaginatedTask
     """
 
     return (