# 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.agent_workflow_request_model import AgentWorkflowRequestModel from ....types.http_validation_error import HttpValidationError # this is used as the default value for optional parameters OMIT = typing.cast(typing.Any, ...) class RawDraftsClient: def __init__(self, *, client_wrapper: SyncClientWrapper): self._client_wrapper = client_wrapper def create( self, agent_id: str, *, branch_id: str, conversation_config: typing.Dict[str, typing.Any], platform_settings: typing.Dict[str, typing.Any], workflow: AgentWorkflowRequestModel, name: str, tags: typing.Optional[typing.Sequence[str]] = OMIT, request_options: typing.Optional[RequestOptions] = None, ) -> HttpResponse[typing.Any]: """ Create a new draft for an agent Parameters ---------- agent_id : str The id of an agent. This is returned on agent creation. branch_id : str The ID of the agent branch to use conversation_config : typing.Dict[str, typing.Any] Conversation config for the draft platform_settings : typing.Dict[str, typing.Any] Platform settings for the draft workflow : AgentWorkflowRequestModel Workflow for the draft name : str Name for the draft tags : typing.Optional[typing.Sequence[str]] Tags to help classify and filter the agent request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[typing.Any] Successful Response """ _response = self._client_wrapper.httpx_client.request( f"v1/convai/agents/{jsonable_encoder(agent_id)}/drafts", method="POST", params={ "branch_id": branch_id, }, json={ "conversation_config": conversation_config, "platform_settings": platform_settings, "workflow": convert_and_respect_annotation_metadata( object_=workflow, annotation=AgentWorkflowRequestModel, direction="write" ), "name": name, "tags": tags, }, headers={ "content-type": "application/json", }, request_options=request_options, omit=OMIT, ) 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 delete( self, agent_id: str, *, branch_id: str, request_options: typing.Optional[RequestOptions] = None ) -> HttpResponse[typing.Any]: """ Delete a draft for an agent Parameters ---------- agent_id : str The id of an agent. This is returned on agent creation. branch_id : str The ID of the agent branch to use request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[typing.Any] Successful Response """ _response = self._client_wrapper.httpx_client.request( f"v1/convai/agents/{jsonable_encoder(agent_id)}/drafts", method="DELETE", params={ "branch_id": branch_id, }, 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) class AsyncRawDraftsClient: def __init__(self, *, client_wrapper: AsyncClientWrapper): self._client_wrapper = client_wrapper async def create( self, agent_id: str, *, branch_id: str, conversation_config: typing.Dict[str, typing.Any], platform_settings: typing.Dict[str, typing.Any], workflow: AgentWorkflowRequestModel, name: str, tags: typing.Optional[typing.Sequence[str]] = OMIT, request_options: typing.Optional[RequestOptions] = None, ) -> AsyncHttpResponse[typing.Any]: """ Create a new draft for an agent Parameters ---------- agent_id : str The id of an agent. This is returned on agent creation. branch_id : str The ID of the agent branch to use conversation_config : typing.Dict[str, typing.Any] Conversation config for the draft platform_settings : typing.Dict[str, typing.Any] Platform settings for the draft workflow : AgentWorkflowRequestModel Workflow for the draft name : str Name for the draft tags : typing.Optional[typing.Sequence[str]] Tags to help classify and filter the agent 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/agents/{jsonable_encoder(agent_id)}/drafts", method="POST", params={ "branch_id": branch_id, }, json={ "conversation_config": conversation_config, "platform_settings": platform_settings, "workflow": convert_and_respect_annotation_metadata( object_=workflow, annotation=AgentWorkflowRequestModel, direction="write" ), "name": name, "tags": tags, }, headers={ "content-type": "application/json", }, request_options=request_options, omit=OMIT, ) 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 delete( self, agent_id: str, *, branch_id: str, request_options: typing.Optional[RequestOptions] = None ) -> AsyncHttpResponse[typing.Any]: """ Delete a draft for an agent Parameters ---------- agent_id : str The id of an agent. This is returned on agent creation. branch_id : str The ID of the agent branch to use 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/agents/{jsonable_encoder(agent_id)}/drafts", method="DELETE", params={ "branch_id": branch_id, }, 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)