mapleflow 0.8.1__py3-none-any.whl

mapleflow/__init__.py

@@ -0,0 +1,8 @@
+"""A client library for accessing MapleFlow"""
+
+from .client import AuthenticatedClient, Client
+
+__all__ = (
+    "AuthenticatedClient",
+    "Client",
+)

mapleflow/api/__init__.py

@@ -0,0 +1 @@
+"""Contains methods for accessing the API"""

mapleflow/api/ai/__init__.py

@@ -0,0 +1 @@
+"""Contains endpoint functions for accessing the API"""

mapleflow/api/ai/post_ai_chat.py

@@ -0,0 +1,195 @@
+from http import HTTPStatus
+from typing import Any
+
+import httpx
+
+from ... import errors
+from ...client import AuthenticatedClient, Client
+from ...models.post_ai_chat_body import PostAiChatBody
+from ...models.post_ai_chat_response_200 import PostAiChatResponse200
+from ...models.post_ai_chat_response_400 import PostAiChatResponse400
+from ...types import UNSET, Response, Unset
+
+
+def _get_kwargs(
+    *,
+    body: PostAiChatBody | Unset = UNSET,
+) -> dict[str, Any]:
+    headers: dict[str, Any] = {}
+
+    _kwargs: dict[str, Any] = {
+        "method": "post",
+        "url": "/ai/chat",
+    }
+
+    if not isinstance(body, Unset):
+        _kwargs["json"] = body.to_dict()
+
+    headers["Content-Type"] = "application/json"
+
+    _kwargs["headers"] = headers
+    return _kwargs
+
+
+def _parse_response(
+    *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> PostAiChatResponse200 | PostAiChatResponse400 | None:
+    if response.status_code == 200:
+        response_200 = PostAiChatResponse200.from_dict(response.json())
+
+        return response_200
+
+    if response.status_code == 400:
+        response_400 = PostAiChatResponse400.from_dict(response.json())
+
+        return response_400
+
+    if client.raise_on_unexpected_status:
+        raise errors.UnexpectedStatus(response.status_code, response.content)
+    else:
+        return None
+
+
+def _build_response(
+    *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Response[PostAiChatResponse200 | PostAiChatResponse400]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(client=client, response=response),
+    )
+
+
+def sync_detailed(
+    *,
+    client: AuthenticatedClient | Client,
+    body: PostAiChatBody | Unset = UNSET,
+) -> Response[PostAiChatResponse200 | PostAiChatResponse400]:
+    """Chat completion
+
+     Generate chat completions.
+
+    | Model | Credits |
+    |---|---|
+    | `llama-3.1-8b` | 2 |
+    | `deepseek-r1-32b` | 5 |
+
+    Args:
+        body (PostAiChatBody | Unset):
+
+    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[PostAiChatResponse200 | PostAiChatResponse400]
+    """
+
+    kwargs = _get_kwargs(
+        body=body,
+    )
+
+    response = client.get_httpx_client().request(
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+def sync(
+    *,
+    client: AuthenticatedClient | Client,
+    body: PostAiChatBody | Unset = UNSET,
+) -> PostAiChatResponse200 | PostAiChatResponse400 | None:
+    """Chat completion
+
+     Generate chat completions.
+
+    | Model | Credits |
+    |---|---|
+    | `llama-3.1-8b` | 2 |
+    | `deepseek-r1-32b` | 5 |
+
+    Args:
+        body (PostAiChatBody | Unset):
+
+    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:
+        PostAiChatResponse200 | PostAiChatResponse400
+    """
+
+    return sync_detailed(
+        client=client,
+        body=body,
+    ).parsed
+
+
+async def asyncio_detailed(
+    *,
+    client: AuthenticatedClient | Client,
+    body: PostAiChatBody | Unset = UNSET,
+) -> Response[PostAiChatResponse200 | PostAiChatResponse400]:
+    """Chat completion
+
+     Generate chat completions.
+
+    | Model | Credits |
+    |---|---|
+    | `llama-3.1-8b` | 2 |
+    | `deepseek-r1-32b` | 5 |
+
+    Args:
+        body (PostAiChatBody | Unset):
+
+    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[PostAiChatResponse200 | PostAiChatResponse400]
+    """
+
+    kwargs = _get_kwargs(
+        body=body,
+    )
+
+    response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio(
+    *,
+    client: AuthenticatedClient | Client,
+    body: PostAiChatBody | Unset = UNSET,
+) -> PostAiChatResponse200 | PostAiChatResponse400 | None:
+    """Chat completion
+
+     Generate chat completions.
+
+    | Model | Credits |
+    |---|---|
+    | `llama-3.1-8b` | 2 |
+    | `deepseek-r1-32b` | 5 |
+
+    Args:
+        body (PostAiChatBody | Unset):
+
+    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:
+        PostAiChatResponse200 | PostAiChatResponse400
+    """
+
+    return (
+        await asyncio_detailed(
+            client=client,
+            body=body,
+        )
+    ).parsed

mapleflow/api/ai/post_ai_embeddings.py

@@ -0,0 +1,191 @@
+from http import HTTPStatus
+from typing import Any
+
+import httpx
+
+from ... import errors
+from ...client import AuthenticatedClient, Client
+from ...models.post_ai_embeddings_body import PostAiEmbeddingsBody
+from ...models.post_ai_embeddings_response_200 import PostAiEmbeddingsResponse200
+from ...models.post_ai_embeddings_response_400 import PostAiEmbeddingsResponse400
+from ...types import UNSET, Response, Unset
+
+
+def _get_kwargs(
+    *,
+    body: PostAiEmbeddingsBody | Unset = UNSET,
+) -> dict[str, Any]:
+    headers: dict[str, Any] = {}
+
+    _kwargs: dict[str, Any] = {
+        "method": "post",
+        "url": "/ai/embeddings",
+    }
+
+    if not isinstance(body, Unset):
+        _kwargs["json"] = body.to_dict()
+
+    headers["Content-Type"] = "application/json"
+
+    _kwargs["headers"] = headers
+    return _kwargs
+
+
+def _parse_response(
+    *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> PostAiEmbeddingsResponse200 | PostAiEmbeddingsResponse400 | None:
+    if response.status_code == 200:
+        response_200 = PostAiEmbeddingsResponse200.from_dict(response.json())
+
+        return response_200
+
+    if response.status_code == 400:
+        response_400 = PostAiEmbeddingsResponse400.from_dict(response.json())
+
+        return response_400
+
+    if client.raise_on_unexpected_status:
+        raise errors.UnexpectedStatus(response.status_code, response.content)
+    else:
+        return None
+
+
+def _build_response(
+    *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Response[PostAiEmbeddingsResponse200 | PostAiEmbeddingsResponse400]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(client=client, response=response),
+    )
+
+
+def sync_detailed(
+    *,
+    client: AuthenticatedClient | Client,
+    body: PostAiEmbeddingsBody | Unset = UNSET,
+) -> Response[PostAiEmbeddingsResponse200 | PostAiEmbeddingsResponse400]:
+    """Text embeddings
+
+     Generate vector embeddings from text.
+
+    | Model | Credits |
+    |---|---|
+    | `bge-base-en` | 1 |
+
+    Args:
+        body (PostAiEmbeddingsBody | Unset):
+
+    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[PostAiEmbeddingsResponse200 | PostAiEmbeddingsResponse400]
+    """
+
+    kwargs = _get_kwargs(
+        body=body,
+    )
+
+    response = client.get_httpx_client().request(
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+def sync(
+    *,
+    client: AuthenticatedClient | Client,
+    body: PostAiEmbeddingsBody | Unset = UNSET,
+) -> PostAiEmbeddingsResponse200 | PostAiEmbeddingsResponse400 | None:
+    """Text embeddings
+
+     Generate vector embeddings from text.
+
+    | Model | Credits |
+    |---|---|
+    | `bge-base-en` | 1 |
+
+    Args:
+        body (PostAiEmbeddingsBody | Unset):
+
+    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:
+        PostAiEmbeddingsResponse200 | PostAiEmbeddingsResponse400
+    """
+
+    return sync_detailed(
+        client=client,
+        body=body,
+    ).parsed
+
+
+async def asyncio_detailed(
+    *,
+    client: AuthenticatedClient | Client,
+    body: PostAiEmbeddingsBody | Unset = UNSET,
+) -> Response[PostAiEmbeddingsResponse200 | PostAiEmbeddingsResponse400]:
+    """Text embeddings
+
+     Generate vector embeddings from text.
+
+    | Model | Credits |
+    |---|---|
+    | `bge-base-en` | 1 |
+
+    Args:
+        body (PostAiEmbeddingsBody | Unset):
+
+    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[PostAiEmbeddingsResponse200 | PostAiEmbeddingsResponse400]
+    """
+
+    kwargs = _get_kwargs(
+        body=body,
+    )
+
+    response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio(
+    *,
+    client: AuthenticatedClient | Client,
+    body: PostAiEmbeddingsBody | Unset = UNSET,
+) -> PostAiEmbeddingsResponse200 | PostAiEmbeddingsResponse400 | None:
+    """Text embeddings
+
+     Generate vector embeddings from text.
+
+    | Model | Credits |
+    |---|---|
+    | `bge-base-en` | 1 |
+
+    Args:
+        body (PostAiEmbeddingsBody | Unset):
+
+    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:
+        PostAiEmbeddingsResponse200 | PostAiEmbeddingsResponse400
+    """
+
+    return (
+        await asyncio_detailed(
+            client=client,
+            body=body,
+        )
+    ).parsed

mapleflow/api/ai/post_ai_images.py

@@ -0,0 +1,195 @@
+from http import HTTPStatus
+from typing import Any
+
+import httpx
+
+from ... import errors
+from ...client import AuthenticatedClient, Client
+from ...models.post_ai_images_body import PostAiImagesBody
+from ...models.post_ai_images_response_200 import PostAiImagesResponse200
+from ...models.post_ai_images_response_400 import PostAiImagesResponse400
+from ...types import UNSET, Response, Unset
+
+
+def _get_kwargs(
+    *,
+    body: PostAiImagesBody | Unset = UNSET,
+) -> dict[str, Any]:
+    headers: dict[str, Any] = {}
+
+    _kwargs: dict[str, Any] = {
+        "method": "post",
+        "url": "/ai/images",
+    }
+
+    if not isinstance(body, Unset):
+        _kwargs["json"] = body.to_dict()
+
+    headers["Content-Type"] = "application/json"
+
+    _kwargs["headers"] = headers
+    return _kwargs
+
+
+def _parse_response(
+    *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> PostAiImagesResponse200 | PostAiImagesResponse400 | None:
+    if response.status_code == 200:
+        response_200 = PostAiImagesResponse200.from_dict(response.json())
+
+        return response_200
+
+    if response.status_code == 400:
+        response_400 = PostAiImagesResponse400.from_dict(response.json())
+
+        return response_400
+
+    if client.raise_on_unexpected_status:
+        raise errors.UnexpectedStatus(response.status_code, response.content)
+    else:
+        return None
+
+
+def _build_response(
+    *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Response[PostAiImagesResponse200 | PostAiImagesResponse400]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(client=client, response=response),
+    )
+
+
+def sync_detailed(
+    *,
+    client: AuthenticatedClient | Client,
+    body: PostAiImagesBody | Unset = UNSET,
+) -> Response[PostAiImagesResponse200 | PostAiImagesResponse400]:
+    """Image generation
+
+     Generate images from text prompts.
+
+    | Model | Credits |
+    |---|---|
+    | `flux-schnell` | 5 |
+    | `flux-dev` | 5 |
+
+    Args:
+        body (PostAiImagesBody | Unset):
+
+    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[PostAiImagesResponse200 | PostAiImagesResponse400]
+    """
+
+    kwargs = _get_kwargs(
+        body=body,
+    )
+
+    response = client.get_httpx_client().request(
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+def sync(
+    *,
+    client: AuthenticatedClient | Client,
+    body: PostAiImagesBody | Unset = UNSET,
+) -> PostAiImagesResponse200 | PostAiImagesResponse400 | None:
+    """Image generation
+
+     Generate images from text prompts.
+
+    | Model | Credits |
+    |---|---|
+    | `flux-schnell` | 5 |
+    | `flux-dev` | 5 |
+
+    Args:
+        body (PostAiImagesBody | Unset):
+
+    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:
+        PostAiImagesResponse200 | PostAiImagesResponse400
+    """
+
+    return sync_detailed(
+        client=client,
+        body=body,
+    ).parsed
+
+
+async def asyncio_detailed(
+    *,
+    client: AuthenticatedClient | Client,
+    body: PostAiImagesBody | Unset = UNSET,
+) -> Response[PostAiImagesResponse200 | PostAiImagesResponse400]:
+    """Image generation
+
+     Generate images from text prompts.
+
+    | Model | Credits |
+    |---|---|
+    | `flux-schnell` | 5 |
+    | `flux-dev` | 5 |
+
+    Args:
+        body (PostAiImagesBody | Unset):
+
+    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[PostAiImagesResponse200 | PostAiImagesResponse400]
+    """
+
+    kwargs = _get_kwargs(
+        body=body,
+    )
+
+    response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio(
+    *,
+    client: AuthenticatedClient | Client,
+    body: PostAiImagesBody | Unset = UNSET,
+) -> PostAiImagesResponse200 | PostAiImagesResponse400 | None:
+    """Image generation
+
+     Generate images from text prompts.
+
+    | Model | Credits |
+    |---|---|
+    | `flux-schnell` | 5 |
+    | `flux-dev` | 5 |
+
+    Args:
+        body (PostAiImagesBody | Unset):
+
+    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:
+        PostAiImagesResponse200 | PostAiImagesResponse400
+    """
+
+    return (
+        await asyncio_detailed(
+            client=client,
+            body=body,
+        )
+    ).parsed