1a978b79f0
* add websockets lib, bump version Signed-off-by: Jess Frazelle <github@jessfraz.com> * use template Signed-off-by: Jess Frazelle <github@jessfraz.com> * fixes Signed-off-by: Jess Frazelle <github@jessfraz.com> * I have generated the latest API! --------- Signed-off-by: Jess Frazelle <github@jessfraz.com> Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
132 lines
3.5 KiB
Python
132 lines
3.5 KiB
Python
from typing import Any, Dict, Optional, Union
|
|
|
|
import httpx
|
|
|
|
from ...client import Client
|
|
from ...models.error import Error
|
|
from ...models.file_export_format import FileExportFormat
|
|
from ...models.mesh import Mesh
|
|
from ...types import Response
|
|
|
|
|
|
def _get_kwargs(
|
|
output_format: FileExportFormat,
|
|
prompt: str,
|
|
*,
|
|
client: Client,
|
|
) -> Dict[str, Any]:
|
|
url = "{}/ai/text-to-3d/{output_format}".format(
|
|
client.base_url,
|
|
output_format=output_format,
|
|
) # noqa: E501
|
|
|
|
if prompt is not None:
|
|
if "?" in url:
|
|
url = url + "&prompt=" + str(prompt)
|
|
else:
|
|
url = url + "?prompt=" + str(prompt)
|
|
|
|
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(),
|
|
}
|
|
|
|
|
|
def _parse_response(*, response: httpx.Response) -> Optional[Union[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 Error.from_dict(response.json())
|
|
|
|
|
|
def _build_response(
|
|
*, response: httpx.Response
|
|
) -> Response[Optional[Union[Mesh, Error]]]:
|
|
return Response(
|
|
status_code=response.status_code,
|
|
content=response.content,
|
|
headers=response.headers,
|
|
parsed=_parse_response(response=response),
|
|
)
|
|
|
|
|
|
def sync_detailed(
|
|
output_format: FileExportFormat,
|
|
prompt: str,
|
|
*,
|
|
client: Client,
|
|
) -> Response[Optional[Union[Mesh, Error]]]:
|
|
kwargs = _get_kwargs(
|
|
output_format=output_format,
|
|
prompt=prompt,
|
|
client=client,
|
|
)
|
|
|
|
response = httpx.post(
|
|
verify=client.verify_ssl,
|
|
**kwargs,
|
|
)
|
|
|
|
return _build_response(response=response)
|
|
|
|
|
|
def sync(
|
|
output_format: FileExportFormat,
|
|
prompt: str,
|
|
*,
|
|
client: Client,
|
|
) -> Optional[Union[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.""" # noqa: E501
|
|
|
|
return sync_detailed(
|
|
output_format=output_format,
|
|
prompt=prompt,
|
|
client=client,
|
|
).parsed
|
|
|
|
|
|
async def asyncio_detailed(
|
|
output_format: FileExportFormat,
|
|
prompt: str,
|
|
*,
|
|
client: Client,
|
|
) -> Response[Optional[Union[Mesh, Error]]]:
|
|
kwargs = _get_kwargs(
|
|
output_format=output_format,
|
|
prompt=prompt,
|
|
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(
|
|
output_format: FileExportFormat,
|
|
prompt: str,
|
|
*,
|
|
client: Client,
|
|
) -> Optional[Union[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.""" # noqa: E501
|
|
|
|
return (
|
|
await asyncio_detailed(
|
|
output_format=output_format,
|
|
prompt=prompt,
|
|
client=client,
|
|
)
|
|
).parsed
|