Source code for kittycad.api.ml.create_kcl_code_completions

from typing import Any, Dict, Optional, Union

import httpx

from ...client import Client
from ...models.error import Error
from ...models.kcl_code_completion_request import KclCodeCompletionRequest
from ...models.kcl_code_completion_response import KclCodeCompletionResponse
from ...types import Response


[docs] def _get_kwargs( body: KclCodeCompletionRequest, *, client: Client, ) -> Dict[str, Any]: url = "{}/ai/kcl/completions".format( client.base_url, ) # noqa: E501 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.model_dump_json(), }
[docs] def _parse_response( *, response: httpx.Response ) -> Optional[Union[KclCodeCompletionResponse, Error]]: if response.status_code == 201: response_201 = KclCodeCompletionResponse(**response.json()) return response_201 if response.status_code == 400: response_4XX = Error(**response.json()) return response_4XX if response.status_code == 500: response_5XX = Error(**response.json()) return response_5XX return Error(**response.json())
[docs] def _build_response( *, response: httpx.Response ) -> Response[Optional[Union[KclCodeCompletionResponse, Error]]]: return Response( status_code=response.status_code, content=response.content, headers=response.headers, parsed=_parse_response(response=response), )
[docs] def sync_detailed( body: KclCodeCompletionRequest, *, client: Client, ) -> Response[Optional[Union[KclCodeCompletionResponse, Error]]]: kwargs = _get_kwargs( body=body, client=client, ) response = httpx.post( verify=client.verify_ssl, **kwargs, ) return _build_response(response=response)
[docs] def sync( body: KclCodeCompletionRequest, *, client: Client, ) -> Optional[Union[KclCodeCompletionResponse, Error]]: return sync_detailed( body=body, client=client, ).parsed
[docs] async def asyncio_detailed( body: KclCodeCompletionRequest, *, client: Client, ) -> Response[Optional[Union[KclCodeCompletionResponse, Error]]]: kwargs = _get_kwargs( body=body, client=client, ) async with httpx.AsyncClient(verify=client.verify_ssl) as _client: response = await _client.post(**kwargs) return _build_response(response=response)
[docs] async def asyncio( body: KclCodeCompletionRequest, *, client: Client, ) -> Optional[Union[KclCodeCompletionResponse, Error]]: return ( await asyncio_detailed( body=body, client=client, ) ).parsed