c4cef35717
Auth logic isn't shared between the clients anyway, so co-locating them is confusing since you can't use the same clients to call both. This also makes the codegen clients less verbose.
156 lines
3.9 KiB
Python
156 lines
3.9 KiB
Python
from http import HTTPStatus
|
|
from typing import Any, Dict, Optional, Union
|
|
|
|
import httpx
|
|
|
|
from ... import errors
|
|
from ...client import AuthenticatedClient, Client
|
|
from ...models.check_cache_json_body import CheckCacheJsonBody
|
|
from ...models.check_cache_response_200 import CheckCacheResponse200
|
|
from ...types import Response
|
|
|
|
|
|
def _get_kwargs(
|
|
*,
|
|
json_body: CheckCacheJsonBody,
|
|
) -> Dict[str, Any]:
|
|
pass
|
|
|
|
json_json_body = json_body.to_dict()
|
|
|
|
return {
|
|
"method": "post",
|
|
"url": "/check-cache",
|
|
"json": json_json_body,
|
|
}
|
|
|
|
|
|
def _parse_response(
|
|
*, client: Union[AuthenticatedClient, Client], response: httpx.Response
|
|
) -> Optional[CheckCacheResponse200]:
|
|
if response.status_code == HTTPStatus.OK:
|
|
response_200 = CheckCacheResponse200.from_dict(response.json())
|
|
|
|
return response_200
|
|
if client.raise_on_unexpected_status:
|
|
raise errors.UnexpectedStatus(response.status_code, response.content)
|
|
else:
|
|
return None
|
|
|
|
|
|
def _build_response(
|
|
*, client: Union[AuthenticatedClient, Client], response: httpx.Response
|
|
) -> Response[CheckCacheResponse200]:
|
|
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,
|
|
json_body: CheckCacheJsonBody,
|
|
) -> Response[CheckCacheResponse200]:
|
|
"""Check if a prompt is cached
|
|
|
|
Args:
|
|
json_body (CheckCacheJsonBody):
|
|
|
|
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[CheckCacheResponse200]
|
|
"""
|
|
|
|
kwargs = _get_kwargs(
|
|
json_body=json_body,
|
|
)
|
|
|
|
response = client.get_httpx_client().request(
|
|
**kwargs,
|
|
)
|
|
|
|
return _build_response(client=client, response=response)
|
|
|
|
|
|
def sync(
|
|
*,
|
|
client: AuthenticatedClient,
|
|
json_body: CheckCacheJsonBody,
|
|
) -> Optional[CheckCacheResponse200]:
|
|
"""Check if a prompt is cached
|
|
|
|
Args:
|
|
json_body (CheckCacheJsonBody):
|
|
|
|
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:
|
|
CheckCacheResponse200
|
|
"""
|
|
|
|
return sync_detailed(
|
|
client=client,
|
|
json_body=json_body,
|
|
).parsed
|
|
|
|
|
|
async def asyncio_detailed(
|
|
*,
|
|
client: AuthenticatedClient,
|
|
json_body: CheckCacheJsonBody,
|
|
) -> Response[CheckCacheResponse200]:
|
|
"""Check if a prompt is cached
|
|
|
|
Args:
|
|
json_body (CheckCacheJsonBody):
|
|
|
|
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[CheckCacheResponse200]
|
|
"""
|
|
|
|
kwargs = _get_kwargs(
|
|
json_body=json_body,
|
|
)
|
|
|
|
response = await client.get_async_httpx_client().request(**kwargs)
|
|
|
|
return _build_response(client=client, response=response)
|
|
|
|
|
|
async def asyncio(
|
|
*,
|
|
client: AuthenticatedClient,
|
|
json_body: CheckCacheJsonBody,
|
|
) -> Optional[CheckCacheResponse200]:
|
|
"""Check if a prompt is cached
|
|
|
|
Args:
|
|
json_body (CheckCacheJsonBody):
|
|
|
|
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:
|
|
CheckCacheResponse200
|
|
"""
|
|
|
|
return (
|
|
await asyncio_detailed(
|
|
client=client,
|
|
json_body=json_body,
|
|
)
|
|
).parsed
|