# 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.get_tool_dependent_agents_response_model import GetToolDependentAgentsResponseModel from ...types.http_validation_error import HttpValidationError from ...types.tool_request_model import ToolRequestModel from ...types.tool_response_model import ToolResponseModel from ...types.tools_response_model import ToolsResponseModel # this is used as the default value for optional parameters OMIT = typing.cast(typing.Any, ...) class RawToolsClient: def __init__(self, *, client_wrapper: SyncClientWrapper): self._client_wrapper = client_wrapper def list(self, *, request_options: typing.Optional[RequestOptions] = None) -> HttpResponse[ToolsResponseModel]: """ Get all available tools in the workspace. Parameters ---------- request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[ToolsResponseModel] Successful Response """ _response = self._client_wrapper.httpx_client.request( "v1/convai/tools", method="GET", request_options=request_options, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( ToolsResponseModel, construct_type( type_=ToolsResponseModel, # 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 create( self, *, request: ToolRequestModel, request_options: typing.Optional[RequestOptions] = None ) -> HttpResponse[ToolResponseModel]: """ Add a new tool to the available tools in the workspace. Parameters ---------- request : ToolRequestModel request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[ToolResponseModel] Successful Response """ _response = self._client_wrapper.httpx_client.request( "v1/convai/tools", method="POST", json=convert_and_respect_annotation_metadata( object_=request, annotation=ToolRequestModel, direction="write" ), headers={ "content-type": "application/json", }, request_options=request_options, omit=OMIT, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( ToolResponseModel, construct_type( type_=ToolResponseModel, # 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, tool_id: str, *, request_options: typing.Optional[RequestOptions] = None ) -> HttpResponse[ToolResponseModel]: """ Get tool that is available in the workspace. Parameters ---------- tool_id : str ID of the requested tool. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[ToolResponseModel] Successful Response """ _response = self._client_wrapper.httpx_client.request( f"v1/convai/tools/{jsonable_encoder(tool_id)}", method="GET", request_options=request_options, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( ToolResponseModel, construct_type( type_=ToolResponseModel, # 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, tool_id: str, *, request_options: typing.Optional[RequestOptions] = None ) -> HttpResponse[typing.Any]: """ Delete tool from the workspace. Parameters ---------- tool_id : str ID of the requested tool. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[typing.Any] Successful Response """ _response = self._client_wrapper.httpx_client.request( f"v1/convai/tools/{jsonable_encoder(tool_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 update( self, tool_id: str, *, request: ToolRequestModel, request_options: typing.Optional[RequestOptions] = None ) -> HttpResponse[ToolResponseModel]: """ Update tool that is available in the workspace. Parameters ---------- tool_id : str ID of the requested tool. request : ToolRequestModel request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[ToolResponseModel] Successful Response """ _response = self._client_wrapper.httpx_client.request( f"v1/convai/tools/{jsonable_encoder(tool_id)}", method="PATCH", json=convert_and_respect_annotation_metadata( object_=request, annotation=ToolRequestModel, direction="write" ), headers={ "content-type": "application/json", }, request_options=request_options, omit=OMIT, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( ToolResponseModel, construct_type( type_=ToolResponseModel, # 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_dependent_agents( self, tool_id: str, *, cursor: typing.Optional[str] = None, page_size: typing.Optional[int] = None, request_options: typing.Optional[RequestOptions] = None, ) -> HttpResponse[GetToolDependentAgentsResponseModel]: """ Get a list of agents depending on this tool Parameters ---------- tool_id : str ID of the requested tool. cursor : typing.Optional[str] Used for fetching next page. Cursor is returned in the response. page_size : typing.Optional[int] How many documents to return at maximum. Can not exceed 100, defaults to 30. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[GetToolDependentAgentsResponseModel] Successful Response """ _response = self._client_wrapper.httpx_client.request( f"v1/convai/tools/{jsonable_encoder(tool_id)}/dependent-agents", method="GET", params={ "cursor": cursor, "page_size": page_size, }, request_options=request_options, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( GetToolDependentAgentsResponseModel, construct_type( type_=GetToolDependentAgentsResponseModel, # 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 AsyncRawToolsClient: def __init__(self, *, client_wrapper: AsyncClientWrapper): self._client_wrapper = client_wrapper async def list( self, *, request_options: typing.Optional[RequestOptions] = None ) -> AsyncHttpResponse[ToolsResponseModel]: """ Get all available tools in the workspace. Parameters ---------- request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[ToolsResponseModel] Successful Response """ _response = await self._client_wrapper.httpx_client.request( "v1/convai/tools", method="GET", request_options=request_options, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( ToolsResponseModel, construct_type( type_=ToolsResponseModel, # 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 create( self, *, request: ToolRequestModel, request_options: typing.Optional[RequestOptions] = None ) -> AsyncHttpResponse[ToolResponseModel]: """ Add a new tool to the available tools in the workspace. Parameters ---------- request : ToolRequestModel request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[ToolResponseModel] Successful Response """ _response = await self._client_wrapper.httpx_client.request( "v1/convai/tools", method="POST", json=convert_and_respect_annotation_metadata( object_=request, annotation=ToolRequestModel, direction="write" ), headers={ "content-type": "application/json", }, request_options=request_options, omit=OMIT, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( ToolResponseModel, construct_type( type_=ToolResponseModel, # 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, tool_id: str, *, request_options: typing.Optional[RequestOptions] = None ) -> AsyncHttpResponse[ToolResponseModel]: """ Get tool that is available in the workspace. Parameters ---------- tool_id : str ID of the requested tool. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[ToolResponseModel] Successful Response """ _response = await self._client_wrapper.httpx_client.request( f"v1/convai/tools/{jsonable_encoder(tool_id)}", method="GET", request_options=request_options, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( ToolResponseModel, construct_type( type_=ToolResponseModel, # 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, tool_id: str, *, request_options: typing.Optional[RequestOptions] = None ) -> AsyncHttpResponse[typing.Any]: """ Delete tool from the workspace. Parameters ---------- tool_id : str ID of the requested tool. 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/tools/{jsonable_encoder(tool_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 update( self, tool_id: str, *, request: ToolRequestModel, request_options: typing.Optional[RequestOptions] = None ) -> AsyncHttpResponse[ToolResponseModel]: """ Update tool that is available in the workspace. Parameters ---------- tool_id : str ID of the requested tool. request : ToolRequestModel request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[ToolResponseModel] Successful Response """ _response = await self._client_wrapper.httpx_client.request( f"v1/convai/tools/{jsonable_encoder(tool_id)}", method="PATCH", json=convert_and_respect_annotation_metadata( object_=request, annotation=ToolRequestModel, direction="write" ), headers={ "content-type": "application/json", }, request_options=request_options, omit=OMIT, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( ToolResponseModel, construct_type( type_=ToolResponseModel, # 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_dependent_agents( self, tool_id: str, *, cursor: typing.Optional[str] = None, page_size: typing.Optional[int] = None, request_options: typing.Optional[RequestOptions] = None, ) -> AsyncHttpResponse[GetToolDependentAgentsResponseModel]: """ Get a list of agents depending on this tool Parameters ---------- tool_id : str ID of the requested tool. cursor : typing.Optional[str] Used for fetching next page. Cursor is returned in the response. page_size : typing.Optional[int] How many documents to return at maximum. Can not exceed 100, defaults to 30. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[GetToolDependentAgentsResponseModel] Successful Response """ _response = await self._client_wrapper.httpx_client.request( f"v1/convai/tools/{jsonable_encoder(tool_id)}/dependent-agents", method="GET", params={ "cursor": cursor, "page_size": page_size, }, request_options=request_options, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( GetToolDependentAgentsResponseModel, construct_type( type_=GetToolDependentAgentsResponseModel, # 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)