# This file was auto-generated by Fern from our API Definition. import typing from json.decoder import JSONDecodeError from ...core.api_error import ApiError from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper from ...core.http_response import AsyncHttpResponse, HttpResponse from ...core.jsonable_encoder import jsonable_encoder from ...core.request_options import RequestOptions from ...core.serialization import convert_and_respect_annotation_metadata from ...core.unchecked_base_model import construct_type from ...errors.unprocessable_entity_error import UnprocessableEntityError from ...types.create_agent_test_response_model import CreateAgentTestResponseModel from ...types.get_tests_page_response_model import GetTestsPageResponseModel from ...types.get_tests_summaries_by_ids_response_model import GetTestsSummariesByIdsResponseModel from ...types.http_validation_error import HttpValidationError from .types.tests_create_request_body import TestsCreateRequestBody from .types.tests_get_response import TestsGetResponse from .types.tests_update_request_body import TestsUpdateRequestBody from .types.tests_update_response import TestsUpdateResponse # this is used as the default value for optional parameters OMIT = typing.cast(typing.Any, ...) class RawTestsClient: def __init__(self, *, client_wrapper: SyncClientWrapper): self._client_wrapper = client_wrapper def create( self, *, request: TestsCreateRequestBody, request_options: typing.Optional[RequestOptions] = None ) -> HttpResponse[CreateAgentTestResponseModel]: """ Creates a new agent response test. Parameters ---------- request : TestsCreateRequestBody request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[CreateAgentTestResponseModel] Successful Response """ _response = self._client_wrapper.httpx_client.request( "v1/convai/agent-testing/create", method="POST", json=convert_and_respect_annotation_metadata( object_=request, annotation=TestsCreateRequestBody, direction="write" ), headers={ "content-type": "application/json", }, request_options=request_options, omit=OMIT, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( CreateAgentTestResponseModel, construct_type( type_=CreateAgentTestResponseModel, # type: ignore object_=_response.json(), ), ) return HttpResponse(response=_response, data=_data) if _response.status_code == 422: raise UnprocessableEntityError( headers=dict(_response.headers), body=typing.cast( HttpValidationError, construct_type( type_=HttpValidationError, # type: ignore object_=_response.json(), ), ), ) _response_json = _response.json() except JSONDecodeError: raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json) def get( self, test_id: str, *, request_options: typing.Optional[RequestOptions] = None ) -> HttpResponse[TestsGetResponse]: """ Gets an agent response test by ID. Parameters ---------- test_id : str The id of a chat response test. This is returned on test creation. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[TestsGetResponse] Successful Response """ _response = self._client_wrapper.httpx_client.request( f"v1/convai/agent-testing/{jsonable_encoder(test_id)}", method="GET", request_options=request_options, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( TestsGetResponse, construct_type( type_=TestsGetResponse, # type: ignore object_=_response.json(), ), ) return HttpResponse(response=_response, data=_data) if _response.status_code == 422: raise UnprocessableEntityError( headers=dict(_response.headers), body=typing.cast( HttpValidationError, construct_type( type_=HttpValidationError, # type: ignore object_=_response.json(), ), ), ) _response_json = _response.json() except JSONDecodeError: raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json) def update( self, test_id: str, *, request: TestsUpdateRequestBody, request_options: typing.Optional[RequestOptions] = None ) -> HttpResponse[TestsUpdateResponse]: """ Updates an agent response test by ID. Parameters ---------- test_id : str The id of a chat response test. This is returned on test creation. request : TestsUpdateRequestBody request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[TestsUpdateResponse] Successful Response """ _response = self._client_wrapper.httpx_client.request( f"v1/convai/agent-testing/{jsonable_encoder(test_id)}", method="PUT", json=convert_and_respect_annotation_metadata( object_=request, annotation=TestsUpdateRequestBody, direction="write" ), headers={ "content-type": "application/json", }, request_options=request_options, omit=OMIT, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( TestsUpdateResponse, construct_type( type_=TestsUpdateResponse, # type: ignore object_=_response.json(), ), ) return HttpResponse(response=_response, data=_data) if _response.status_code == 422: raise UnprocessableEntityError( headers=dict(_response.headers), body=typing.cast( HttpValidationError, construct_type( type_=HttpValidationError, # type: ignore object_=_response.json(), ), ), ) _response_json = _response.json() except JSONDecodeError: raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json) def delete( self, test_id: str, *, request_options: typing.Optional[RequestOptions] = None ) -> HttpResponse[typing.Any]: """ Deletes an agent response test by ID. Parameters ---------- test_id : str The id of a chat response test. This is returned on test creation. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[typing.Any] Successful Response """ _response = self._client_wrapper.httpx_client.request( f"v1/convai/agent-testing/{jsonable_encoder(test_id)}", method="DELETE", request_options=request_options, ) try: if _response is None or not _response.text.strip(): return HttpResponse(response=_response, data=None) if 200 <= _response.status_code < 300: _data = typing.cast( typing.Any, construct_type( type_=typing.Any, # type: ignore object_=_response.json(), ), ) return HttpResponse(response=_response, data=_data) if _response.status_code == 422: raise UnprocessableEntityError( headers=dict(_response.headers), body=typing.cast( HttpValidationError, construct_type( type_=HttpValidationError, # type: ignore object_=_response.json(), ), ), ) _response_json = _response.json() except JSONDecodeError: raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json) def summaries( self, *, test_ids: typing.Sequence[str], request_options: typing.Optional[RequestOptions] = None ) -> HttpResponse[GetTestsSummariesByIdsResponseModel]: """ Gets multiple agent response tests by their IDs. Returns a dictionary mapping test IDs to test summaries. Parameters ---------- test_ids : typing.Sequence[str] List of test IDs to fetch. No duplicates allowed. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[GetTestsSummariesByIdsResponseModel] Successful Response """ _response = self._client_wrapper.httpx_client.request( "v1/convai/agent-testing/summaries", method="POST", json={ "test_ids": test_ids, }, headers={ "content-type": "application/json", }, request_options=request_options, omit=OMIT, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( GetTestsSummariesByIdsResponseModel, construct_type( type_=GetTestsSummariesByIdsResponseModel, # type: ignore object_=_response.json(), ), ) return HttpResponse(response=_response, data=_data) if _response.status_code == 422: raise UnprocessableEntityError( headers=dict(_response.headers), body=typing.cast( HttpValidationError, construct_type( type_=HttpValidationError, # type: ignore object_=_response.json(), ), ), ) _response_json = _response.json() except JSONDecodeError: raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json) def list( self, *, cursor: typing.Optional[str] = None, page_size: typing.Optional[int] = None, search: typing.Optional[str] = None, request_options: typing.Optional[RequestOptions] = None, ) -> HttpResponse[GetTestsPageResponseModel]: """ Lists all agent response tests with pagination support and optional search filtering. Parameters ---------- cursor : typing.Optional[str] Used for fetching next page. Cursor is returned in the response. page_size : typing.Optional[int] How many Tests to return at maximum. Can not exceed 100, defaults to 30. search : typing.Optional[str] Search query to filter tests by name. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[GetTestsPageResponseModel] Successful Response """ _response = self._client_wrapper.httpx_client.request( "v1/convai/agent-testing", method="GET", params={ "cursor": cursor, "page_size": page_size, "search": search, }, request_options=request_options, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( GetTestsPageResponseModel, construct_type( type_=GetTestsPageResponseModel, # type: ignore object_=_response.json(), ), ) return HttpResponse(response=_response, data=_data) if _response.status_code == 422: raise UnprocessableEntityError( headers=dict(_response.headers), body=typing.cast( HttpValidationError, construct_type( type_=HttpValidationError, # type: ignore object_=_response.json(), ), ), ) _response_json = _response.json() except JSONDecodeError: raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json) class AsyncRawTestsClient: def __init__(self, *, client_wrapper: AsyncClientWrapper): self._client_wrapper = client_wrapper async def create( self, *, request: TestsCreateRequestBody, request_options: typing.Optional[RequestOptions] = None ) -> AsyncHttpResponse[CreateAgentTestResponseModel]: """ Creates a new agent response test. Parameters ---------- request : TestsCreateRequestBody request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[CreateAgentTestResponseModel] Successful Response """ _response = await self._client_wrapper.httpx_client.request( "v1/convai/agent-testing/create", method="POST", json=convert_and_respect_annotation_metadata( object_=request, annotation=TestsCreateRequestBody, direction="write" ), headers={ "content-type": "application/json", }, request_options=request_options, omit=OMIT, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( CreateAgentTestResponseModel, construct_type( type_=CreateAgentTestResponseModel, # type: ignore object_=_response.json(), ), ) return AsyncHttpResponse(response=_response, data=_data) if _response.status_code == 422: raise UnprocessableEntityError( headers=dict(_response.headers), body=typing.cast( HttpValidationError, construct_type( type_=HttpValidationError, # type: ignore object_=_response.json(), ), ), ) _response_json = _response.json() except JSONDecodeError: raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json) async def get( self, test_id: str, *, request_options: typing.Optional[RequestOptions] = None ) -> AsyncHttpResponse[TestsGetResponse]: """ Gets an agent response test by ID. Parameters ---------- test_id : str The id of a chat response test. This is returned on test creation. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[TestsGetResponse] Successful Response """ _response = await self._client_wrapper.httpx_client.request( f"v1/convai/agent-testing/{jsonable_encoder(test_id)}", method="GET", request_options=request_options, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( TestsGetResponse, construct_type( type_=TestsGetResponse, # type: ignore object_=_response.json(), ), ) return AsyncHttpResponse(response=_response, data=_data) if _response.status_code == 422: raise UnprocessableEntityError( headers=dict(_response.headers), body=typing.cast( HttpValidationError, construct_type( type_=HttpValidationError, # type: ignore object_=_response.json(), ), ), ) _response_json = _response.json() except JSONDecodeError: raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json) async def update( self, test_id: str, *, request: TestsUpdateRequestBody, request_options: typing.Optional[RequestOptions] = None ) -> AsyncHttpResponse[TestsUpdateResponse]: """ Updates an agent response test by ID. Parameters ---------- test_id : str The id of a chat response test. This is returned on test creation. request : TestsUpdateRequestBody request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[TestsUpdateResponse] Successful Response """ _response = await self._client_wrapper.httpx_client.request( f"v1/convai/agent-testing/{jsonable_encoder(test_id)}", method="PUT", json=convert_and_respect_annotation_metadata( object_=request, annotation=TestsUpdateRequestBody, direction="write" ), headers={ "content-type": "application/json", }, request_options=request_options, omit=OMIT, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( TestsUpdateResponse, construct_type( type_=TestsUpdateResponse, # type: ignore object_=_response.json(), ), ) return AsyncHttpResponse(response=_response, data=_data) if _response.status_code == 422: raise UnprocessableEntityError( headers=dict(_response.headers), body=typing.cast( HttpValidationError, construct_type( type_=HttpValidationError, # type: ignore object_=_response.json(), ), ), ) _response_json = _response.json() except JSONDecodeError: raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json) async def delete( self, test_id: str, *, request_options: typing.Optional[RequestOptions] = None ) -> AsyncHttpResponse[typing.Any]: """ Deletes an agent response test by ID. Parameters ---------- test_id : str The id of a chat response test. This is returned on test creation. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[typing.Any] Successful Response """ _response = await self._client_wrapper.httpx_client.request( f"v1/convai/agent-testing/{jsonable_encoder(test_id)}", method="DELETE", request_options=request_options, ) try: if _response is None or not _response.text.strip(): return AsyncHttpResponse(response=_response, data=None) if 200 <= _response.status_code < 300: _data = typing.cast( typing.Any, construct_type( type_=typing.Any, # type: ignore object_=_response.json(), ), ) return AsyncHttpResponse(response=_response, data=_data) if _response.status_code == 422: raise UnprocessableEntityError( headers=dict(_response.headers), body=typing.cast( HttpValidationError, construct_type( type_=HttpValidationError, # type: ignore object_=_response.json(), ), ), ) _response_json = _response.json() except JSONDecodeError: raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json) async def summaries( self, *, test_ids: typing.Sequence[str], request_options: typing.Optional[RequestOptions] = None ) -> AsyncHttpResponse[GetTestsSummariesByIdsResponseModel]: """ Gets multiple agent response tests by their IDs. Returns a dictionary mapping test IDs to test summaries. Parameters ---------- test_ids : typing.Sequence[str] List of test IDs to fetch. No duplicates allowed. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[GetTestsSummariesByIdsResponseModel] Successful Response """ _response = await self._client_wrapper.httpx_client.request( "v1/convai/agent-testing/summaries", method="POST", json={ "test_ids": test_ids, }, headers={ "content-type": "application/json", }, request_options=request_options, omit=OMIT, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( GetTestsSummariesByIdsResponseModel, construct_type( type_=GetTestsSummariesByIdsResponseModel, # type: ignore object_=_response.json(), ), ) return AsyncHttpResponse(response=_response, data=_data) if _response.status_code == 422: raise UnprocessableEntityError( headers=dict(_response.headers), body=typing.cast( HttpValidationError, construct_type( type_=HttpValidationError, # type: ignore object_=_response.json(), ), ), ) _response_json = _response.json() except JSONDecodeError: raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json) async def list( self, *, cursor: typing.Optional[str] = None, page_size: typing.Optional[int] = None, search: typing.Optional[str] = None, request_options: typing.Optional[RequestOptions] = None, ) -> AsyncHttpResponse[GetTestsPageResponseModel]: """ Lists all agent response tests with pagination support and optional search filtering. Parameters ---------- cursor : typing.Optional[str] Used for fetching next page. Cursor is returned in the response. page_size : typing.Optional[int] How many Tests to return at maximum. Can not exceed 100, defaults to 30. search : typing.Optional[str] Search query to filter tests by name. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[GetTestsPageResponseModel] Successful Response """ _response = await self._client_wrapper.httpx_client.request( "v1/convai/agent-testing", method="GET", params={ "cursor": cursor, "page_size": page_size, "search": search, }, request_options=request_options, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( GetTestsPageResponseModel, construct_type( type_=GetTestsPageResponseModel, # type: ignore object_=_response.json(), ), ) return AsyncHttpResponse(response=_response, data=_data) if _response.status_code == 422: raise UnprocessableEntityError( headers=dict(_response.headers), body=typing.cast( HttpValidationError, construct_type( type_=HttpValidationError, # type: ignore object_=_response.json(), ), ), ) _response_json = _response.json() except JSONDecodeError: raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)