573f4a8bc8
* YOYO NEW API SPEC! * I have generated the latest API! --------- Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
131 lines
3.2 KiB
Python
131 lines
3.2 KiB
Python
from typing import Any, Dict, Optional, Union, cast
|
|
|
|
import httpx
|
|
|
|
from ...client import Client
|
|
from ...models.mesh import Mesh
|
|
from ...models.error import Error
|
|
from ...models.image_type import ImageType
|
|
from ...models.file_export_format import FileExportFormat
|
|
from ...types import Response
|
|
|
|
def _get_kwargs(
|
|
input_format: ImageType,
|
|
output_format: FileExportFormat,
|
|
body: bytes,
|
|
*,
|
|
client: Client,
|
|
) -> Dict[str, Any]:
|
|
url = "{}/ai/image-to-3d/{input_format}/{output_format}".format(client.base_url, input_format=input_format, output_format=output_format)
|
|
|
|
headers: Dict[str, Any] = client.get_headers()
|
|
cookies: Dict[str, Any] = client.get_cookies()
|
|
|
|
return {
|
|
"url": url,
|
|
"headers": headers,
|
|
"cookies": cookies,
|
|
"timeout": client.get_timeout(),
|
|
"content": body,
|
|
}
|
|
|
|
|
|
def _parse_response(*, response: httpx.Response) -> Optional[Union[Any, Mesh, Error]]:
|
|
if response.status_code == 200:
|
|
response_200 = Mesh.from_dict(response.json())
|
|
return response_200
|
|
if response.status_code == 400:
|
|
response_4XX = Error.from_dict(response.json())
|
|
return response_4XX
|
|
if response.status_code == 500:
|
|
response_5XX = Error.from_dict(response.json())
|
|
return response_5XX
|
|
return None
|
|
|
|
|
|
def _build_response(*, response: httpx.Response) -> Response[Union[Any, Mesh, Error]]:
|
|
return Response(
|
|
status_code=response.status_code,
|
|
content=response.content,
|
|
headers=response.headers,
|
|
parsed=_parse_response(response=response),
|
|
)
|
|
|
|
|
|
def sync_detailed(
|
|
input_format: ImageType,
|
|
output_format: FileExportFormat,
|
|
body: bytes,
|
|
*,
|
|
client: Client,
|
|
) -> Response[Union[Any, Mesh, Error]]:
|
|
kwargs = _get_kwargs(
|
|
input_format=input_format,
|
|
output_format=output_format,
|
|
body=body,
|
|
client=client,
|
|
)
|
|
|
|
response = httpx.post(
|
|
verify=client.verify_ssl,
|
|
**kwargs,
|
|
)
|
|
|
|
return _build_response(response=response)
|
|
|
|
|
|
def sync(
|
|
input_format: ImageType,
|
|
output_format: FileExportFormat,
|
|
body: bytes,
|
|
*,
|
|
client: Client,
|
|
) -> Optional[Union[Any, Mesh, Error]]:
|
|
""" This is an alpha endpoint. It will change in the future. The current output is honestly pretty bad. So if you find this endpoint, you get what you pay for, which currently is nothing. But in the future will be made a lot better. """
|
|
|
|
return sync_detailed(
|
|
input_format=input_format,
|
|
output_format=output_format,
|
|
body=body,
|
|
client=client,
|
|
).parsed
|
|
|
|
|
|
async def asyncio_detailed(
|
|
input_format: ImageType,
|
|
output_format: FileExportFormat,
|
|
body: bytes,
|
|
*,
|
|
client: Client,
|
|
) -> Response[Union[Any, Mesh, Error]]:
|
|
kwargs = _get_kwargs(
|
|
input_format=input_format,
|
|
output_format=output_format,
|
|
body=body,
|
|
client=client,
|
|
)
|
|
|
|
async with httpx.AsyncClient(verify=client.verify_ssl) as _client:
|
|
response = await _client.post(**kwargs)
|
|
|
|
return _build_response(response=response)
|
|
|
|
|
|
async def asyncio(
|
|
input_format: ImageType,
|
|
output_format: FileExportFormat,
|
|
body: bytes,
|
|
*,
|
|
client: Client,
|
|
) -> Optional[Union[Any, Mesh, Error]]:
|
|
""" This is an alpha endpoint. It will change in the future. The current output is honestly pretty bad. So if you find this endpoint, you get what you pay for, which currently is nothing. But in the future will be made a lot better. """
|
|
|
|
return (
|
|
await asyncio_detailed(
|
|
input_format=input_format,
|
|
output_format=output_format,
|
|
body=body,
|
|
client=client,
|
|
)
|
|
).parsed
|