167 lines
4.8 KiB
Python
167 lines
4.8 KiB
Python
from http import HTTPStatus
|
|
from typing import Any, Dict, Optional, Union, cast
|
|
|
|
import httpx
|
|
|
|
from ... import errors
|
|
from ...client import Client
|
|
from ...models.completion_request import CompletionRequest
|
|
from ...models.completion_response import CompletionResponse
|
|
from ...types import Response
|
|
|
|
|
|
def _get_kwargs(
|
|
*,
|
|
client: Client,
|
|
json_body: CompletionRequest,
|
|
) -> Dict[str, Any]:
|
|
url = "{}/v1/completions".format(client.base_url)
|
|
|
|
headers: Dict[str, str] = client.get_headers()
|
|
cookies: Dict[str, Any] = client.get_cookies()
|
|
|
|
json_json_body = json_body.to_dict()
|
|
|
|
return {
|
|
"method": "post",
|
|
"url": url,
|
|
"headers": headers,
|
|
"cookies": cookies,
|
|
"timeout": client.get_timeout(),
|
|
"follow_redirects": client.follow_redirects,
|
|
"json": json_json_body,
|
|
}
|
|
|
|
|
|
def _parse_response(*, client: Client, response: httpx.Response) -> Optional[Union[Any, CompletionResponse]]:
|
|
if response.status_code == HTTPStatus.OK:
|
|
response_200 = CompletionResponse.from_dict(response.json())
|
|
|
|
return response_200
|
|
if response.status_code == HTTPStatus.BAD_REQUEST:
|
|
response_400 = cast(Any, None)
|
|
return response_400
|
|
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[Any, CompletionResponse]]:
|
|
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: Client,
|
|
json_body: CompletionRequest,
|
|
) -> Response[Union[Any, CompletionResponse]]:
|
|
r"""
|
|
Args:
|
|
json_body (CompletionRequest): Example: {'language': 'python', 'segments': {'prefix':
|
|
'def fib(n):\n ', 'suffix': '\n return fib(n - 1) + fib(n - 2)'}}.
|
|
|
|
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[Any, CompletionResponse]]
|
|
"""
|
|
|
|
kwargs = _get_kwargs(
|
|
client=client,
|
|
json_body=json_body,
|
|
)
|
|
|
|
response = httpx.request(
|
|
verify=client.verify_ssl,
|
|
**kwargs,
|
|
)
|
|
|
|
return _build_response(client=client, response=response)
|
|
|
|
|
|
def sync(
|
|
*,
|
|
client: Client,
|
|
json_body: CompletionRequest,
|
|
) -> Optional[Union[Any, CompletionResponse]]:
|
|
r"""
|
|
Args:
|
|
json_body (CompletionRequest): Example: {'language': 'python', 'segments': {'prefix':
|
|
'def fib(n):\n ', 'suffix': '\n return fib(n - 1) + fib(n - 2)'}}.
|
|
|
|
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[Any, CompletionResponse]
|
|
"""
|
|
|
|
return sync_detailed(
|
|
client=client,
|
|
json_body=json_body,
|
|
).parsed
|
|
|
|
|
|
async def asyncio_detailed(
|
|
*,
|
|
client: Client,
|
|
json_body: CompletionRequest,
|
|
) -> Response[Union[Any, CompletionResponse]]:
|
|
r"""
|
|
Args:
|
|
json_body (CompletionRequest): Example: {'language': 'python', 'segments': {'prefix':
|
|
'def fib(n):\n ', 'suffix': '\n return fib(n - 1) + fib(n - 2)'}}.
|
|
|
|
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[Any, CompletionResponse]]
|
|
"""
|
|
|
|
kwargs = _get_kwargs(
|
|
client=client,
|
|
json_body=json_body,
|
|
)
|
|
|
|
async with httpx.AsyncClient(verify=client.verify_ssl) as _client:
|
|
response = await _client.request(**kwargs)
|
|
|
|
return _build_response(client=client, response=response)
|
|
|
|
|
|
async def asyncio(
|
|
*,
|
|
client: Client,
|
|
json_body: CompletionRequest,
|
|
) -> Optional[Union[Any, CompletionResponse]]:
|
|
r"""
|
|
Args:
|
|
json_body (CompletionRequest): Example: {'language': 'python', 'segments': {'prefix':
|
|
'def fib(n):\n ', 'suffix': '\n return fib(n - 1) + fib(n - 2)'}}.
|
|
|
|
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[Any, CompletionResponse]
|
|
"""
|
|
|
|
return (
|
|
await asyncio_detailed(
|
|
client=client,
|
|
json_body=json_body,
|
|
)
|
|
).parsed
|