blaxel 0.2.29__py3-none-any.whl → 0.2.31__py3-none-any.whl

blaxel/core/sandbox/client/api/filesystem/get_filesystem_search_path.py

@@ -0,0 +1,241 @@
+from http import HTTPStatus
+from typing import Any, Union
+
+import httpx
+
+from ... import errors
+from ...client import Client
+from ...models.error_response import ErrorResponse
+from ...models.fuzzy_search_response import FuzzySearchResponse
+from ...types import UNSET, Response, Unset
+
+
+def _get_kwargs(
+    path: str,
+    *,
+    max_results: Union[Unset, int] = UNSET,
+    patterns: Union[Unset, str] = UNSET,
+    exclude_dirs: Union[Unset, str] = UNSET,
+    exclude_hidden: Union[Unset, bool] = UNSET,
+) -> dict[str, Any]:
+    params: dict[str, Any] = {}
+
+    params["maxResults"] = max_results
+
+    params["patterns"] = patterns
+
+    params["excludeDirs"] = exclude_dirs
+
+    params["excludeHidden"] = exclude_hidden
+
+    params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
+
+    _kwargs: dict[str, Any] = {
+        "method": "get",
+        "url": f"/filesystem-search/{path}",
+        "params": params,
+    }
+
+    return _kwargs
+
+
+def _parse_response(
+    *, client: Client, response: httpx.Response
+) -> Union[ErrorResponse, FuzzySearchResponse] | None:
+    if response.status_code == 200:
+        response_200 = FuzzySearchResponse.from_dict(response.json())
+
+        return response_200
+    if response.status_code == 400:
+        response_400 = ErrorResponse.from_dict(response.json())
+
+        return response_400
+    if response.status_code == 422:
+        response_422 = ErrorResponse.from_dict(response.json())
+
+        return response_422
+    if response.status_code == 500:
+        response_500 = ErrorResponse.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[Union[ErrorResponse, FuzzySearchResponse]]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(client=client, response=response),
+    )
+
+
+def sync_detailed(
+    path: str,
+    *,
+    client: Union[Client],
+    max_results: Union[Unset, int] = UNSET,
+    patterns: Union[Unset, str] = UNSET,
+    exclude_dirs: Union[Unset, str] = UNSET,
+    exclude_hidden: Union[Unset, bool] = UNSET,
+) -> Response[Union[ErrorResponse, FuzzySearchResponse]]:
+    """Fuzzy search for files and directories
+
+     Performs fuzzy search on filesystem paths using fuzzy matching algorithm. Optimized alternative to
+    find and grep commands.
+
+    Args:
+        path (str):
+        max_results (Union[Unset, int]):
+        patterns (Union[Unset, str]):
+        exclude_dirs (Union[Unset, str]):
+        exclude_hidden (Union[Unset, bool]):
+
+    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[Union[ErrorResponse, FuzzySearchResponse]]
+    """
+
+    kwargs = _get_kwargs(
+        path=path,
+        max_results=max_results,
+        patterns=patterns,
+        exclude_dirs=exclude_dirs,
+        exclude_hidden=exclude_hidden,
+    )
+
+    response = client.get_httpx_client().request(
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+def sync(
+    path: str,
+    *,
+    client: Union[Client],
+    max_results: Union[Unset, int] = UNSET,
+    patterns: Union[Unset, str] = UNSET,
+    exclude_dirs: Union[Unset, str] = UNSET,
+    exclude_hidden: Union[Unset, bool] = UNSET,
+) -> Union[ErrorResponse, FuzzySearchResponse] | None:
+    """Fuzzy search for files and directories
+
+     Performs fuzzy search on filesystem paths using fuzzy matching algorithm. Optimized alternative to
+    find and grep commands.
+
+    Args:
+        path (str):
+        max_results (Union[Unset, int]):
+        patterns (Union[Unset, str]):
+        exclude_dirs (Union[Unset, str]):
+        exclude_hidden (Union[Unset, bool]):
+
+    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:
+        Union[ErrorResponse, FuzzySearchResponse]
+    """
+
+    return sync_detailed(
+        path=path,
+        client=client,
+        max_results=max_results,
+        patterns=patterns,
+        exclude_dirs=exclude_dirs,
+        exclude_hidden=exclude_hidden,
+    ).parsed
+
+
+async def asyncio_detailed(
+    path: str,
+    *,
+    client: Union[Client],
+    max_results: Union[Unset, int] = UNSET,
+    patterns: Union[Unset, str] = UNSET,
+    exclude_dirs: Union[Unset, str] = UNSET,
+    exclude_hidden: Union[Unset, bool] = UNSET,
+) -> Response[Union[ErrorResponse, FuzzySearchResponse]]:
+    """Fuzzy search for files and directories
+
+     Performs fuzzy search on filesystem paths using fuzzy matching algorithm. Optimized alternative to
+    find and grep commands.
+
+    Args:
+        path (str):
+        max_results (Union[Unset, int]):
+        patterns (Union[Unset, str]):
+        exclude_dirs (Union[Unset, str]):
+        exclude_hidden (Union[Unset, bool]):
+
+    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[Union[ErrorResponse, FuzzySearchResponse]]
+    """
+
+    kwargs = _get_kwargs(
+        path=path,
+        max_results=max_results,
+        patterns=patterns,
+        exclude_dirs=exclude_dirs,
+        exclude_hidden=exclude_hidden,
+    )
+
+    response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio(
+    path: str,
+    *,
+    client: Union[Client],
+    max_results: Union[Unset, int] = UNSET,
+    patterns: Union[Unset, str] = UNSET,
+    exclude_dirs: Union[Unset, str] = UNSET,
+    exclude_hidden: Union[Unset, bool] = UNSET,
+) -> Union[ErrorResponse, FuzzySearchResponse] | None:
+    """Fuzzy search for files and directories
+
+     Performs fuzzy search on filesystem paths using fuzzy matching algorithm. Optimized alternative to
+    find and grep commands.
+
+    Args:
+        path (str):
+        max_results (Union[Unset, int]):
+        patterns (Union[Unset, str]):
+        exclude_dirs (Union[Unset, str]):
+        exclude_hidden (Union[Unset, bool]):
+
+    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:
+        Union[ErrorResponse, FuzzySearchResponse]
+    """
+
+    return (
+        await asyncio_detailed(
+            path=path,
+            client=client,
+            max_results=max_results,
+            patterns=patterns,
+            exclude_dirs=exclude_dirs,
+            exclude_hidden=exclude_hidden,
+        )
+    ).parsed

blaxel/core/sandbox/client/api/filesystem/get_filesystem_tree_path.py

@@ -0,0 +1,197 @@
+from http import HTTPStatus
+from io import BytesIO
+from typing import Any, Union
+
+import httpx
+
+from ... import errors
+from ...client import Client
+from ...models.directory import Directory
+from ...models.error_response import ErrorResponse
+from ...models.file_with_content import FileWithContent
+from ...types import File, Response
+
+
+def _get_kwargs(
+    path: str,
+) -> dict[str, Any]:
+    _kwargs: dict[str, Any] = {
+        "method": "get",
+        "url": f"/filesystem/tree/{path}",
+    }
+
+    return _kwargs
+
+
+def _parse_response(
+    *, client: Client, response: httpx.Response
+) -> Union[ErrorResponse, Union["Directory", "FileWithContent", File]] | None:
+    if response.status_code == 200:
+
+        def _parse_response_200(data: object) -> Union["Directory", "FileWithContent", File]:
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                response_200_type_0 = Directory.from_dict(data)
+
+                return response_200_type_0
+            except:  # noqa: E722
+                pass
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                response_200_type_1 = FileWithContent.from_dict(data)
+
+                return response_200_type_1
+            except:  # noqa: E722
+                pass
+            if not isinstance(data, bytes):
+                raise TypeError()
+            response_200_type_2 = File(payload=BytesIO(data))
+
+            return response_200_type_2
+
+        response_200 = _parse_response_200(response.json())
+
+        return response_200
+    if response.status_code == 400:
+        response_400 = ErrorResponse.from_dict(response.json())
+
+        return response_400
+    if response.status_code == 422:
+        response_422 = ErrorResponse.from_dict(response.json())
+
+        return response_422
+    if response.status_code == 500:
+        response_500 = ErrorResponse.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[Union[ErrorResponse, Union["Directory", "FileWithContent", File]]]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(client=client, response=response),
+    )
+
+
+def sync_detailed(
+    path: str,
+    *,
+    client: Union[Client],
+) -> Response[Union[ErrorResponse, Union["Directory", "FileWithContent", File]]]:
+    """Get directory tree
+
+     Get a recursive directory tree structure starting from the specified path
+
+    Args:
+        path (str):
+
+    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[Union[ErrorResponse, Union['Directory', 'FileWithContent', File]]]
+    """
+
+    kwargs = _get_kwargs(
+        path=path,
+    )
+
+    response = client.get_httpx_client().request(
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+def sync(
+    path: str,
+    *,
+    client: Union[Client],
+) -> Union[ErrorResponse, Union["Directory", "FileWithContent", File]] | None:
+    """Get directory tree
+
+     Get a recursive directory tree structure starting from the specified path
+
+    Args:
+        path (str):
+
+    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:
+        Union[ErrorResponse, Union['Directory', 'FileWithContent', File]]
+    """
+
+    return sync_detailed(
+        path=path,
+        client=client,
+    ).parsed
+
+
+async def asyncio_detailed(
+    path: str,
+    *,
+    client: Union[Client],
+) -> Response[Union[ErrorResponse, Union["Directory", "FileWithContent", File]]]:
+    """Get directory tree
+
+     Get a recursive directory tree structure starting from the specified path
+
+    Args:
+        path (str):
+
+    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[Union[ErrorResponse, Union['Directory', 'FileWithContent', File]]]
+    """
+
+    kwargs = _get_kwargs(
+        path=path,
+    )
+
+    response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio(
+    path: str,
+    *,
+    client: Union[Client],
+) -> Union[ErrorResponse, Union["Directory", "FileWithContent", File]] | None:
+    """Get directory tree
+
+     Get a recursive directory tree structure starting from the specified path
+
+    Args:
+        path (str):
+
+    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:
+        Union[ErrorResponse, Union['Directory', 'FileWithContent', File]]
+    """
+
+    return (
+        await asyncio_detailed(
+            path=path,
+            client=client,
+        )
+    ).parsed

blaxel/core/sandbox/client/api/filesystem/get_watch_filesystem_path.py

@@ -29,7 +29,9 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(*, client: Client, response: httpx.Response) -> Union[ErrorResponse, str] | None:
+def _parse_response(
+    *, client: Client, response: httpx.Response
+) -> Union[ErrorResponse, str] | None:
     if response.status_code == 200:
         response_200 = response.text
         return response_200
@@ -47,7 +49,9 @@ def _parse_response(*, client: Client, response: httpx.Response) -> Union[ErrorR
         return None
 
 
-def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[ErrorResponse, str]]:
+def _build_response(
+    *, client: Client, response: httpx.Response
+) -> Response[Union[ErrorResponse, str]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

blaxel/core/sandbox/client/api/filesystem/post_filesystem_multipart_upload_id_complete.py

@@ -35,7 +35,9 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(*, client: Client, response: httpx.Response) -> Union[ErrorResponse, SuccessResponse] | None:
+def _parse_response(
+    *, client: Client, response: httpx.Response
+) -> Union[ErrorResponse, SuccessResponse] | None:
     if response.status_code == 200:
         response_200 = SuccessResponse.from_dict(response.json())
 
@@ -58,7 +60,9 @@ def _parse_response(*, client: Client, response: httpx.Response) -> Union[ErrorR
         return None
 
 
-def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[ErrorResponse, SuccessResponse]]:
+def _build_response(
+    *, client: Client, response: httpx.Response
+) -> Response[Union[ErrorResponse, SuccessResponse]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

blaxel/core/sandbox/client/api/filesystem/put_filesystem_path.py

@@ -35,7 +35,9 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(*, client: Client, response: httpx.Response) -> Union[ErrorResponse, SuccessResponse] | None:
+def _parse_response(
+    *, client: Client, response: httpx.Response
+) -> Union[ErrorResponse, SuccessResponse] | None:
     if response.status_code == 200:
         response_200 = SuccessResponse.from_dict(response.json())
 
@@ -58,7 +60,9 @@ def _parse_response(*, client: Client, response: httpx.Response) -> Union[ErrorR
         return None
 
 
-def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[ErrorResponse, SuccessResponse]]:
+def _build_response(
+    *, client: Client, response: httpx.Response
+) -> Response[Union[ErrorResponse, SuccessResponse]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,