o i@sddlZddlmZddlmZddlmZmZddlm Z m Z ddl m Z ddl mZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlm Z ddl!m"Z"ddl#m$Z$ddl%m&Z&ddl'm(Z(ddl)m*Z*ddl+m,Z,ddl-m.Z.ddl/m0Z0ddl1m2Z2ddl3m4Z4ddl5m6Z6ddl7m8Z8ddl9m:Z:e;ejGd"d#d#Z?dS)$N)JSONDecodeError)ApiError)AsyncClientWrapperSyncClientWrapper)AsyncHttpResponse HttpResponse)jsonable_encoder)RequestOptions)'convert_and_respect_annotation_metadata)construct_type)BadRequestError)ClientClosedRequestError)ForbiddenError)GatewayTimeoutError)InternalServerError)InvalidTokenError) NotFoundError)NotImplementedError)ServiceUnavailableError)TooManyRequestsError)UnauthorizedError)UnprocessableEntityError)CreateConnectorOAuth)CreateConnectorResponse)CreateConnectorServiceAuth)DeleteConnectorResponse)GetConnectorResponse)ListConnectorsResponse)OAuthAuthorizeResponse)UpdateConnectorResponse.c@eZdZdefddZdddddejedejedejed e e fd d Z e e e e e e dd d e de deje dejeje dejedejedejedejedejed e efddZddde dejed e efddZddde dejed e efddZe e e e e e e ddde d eje deje dejeje dejedejedejedejedejed e efddZddd de d!eje dejed e efd"d#ZdS)$RawConnectorsClientclient_wrappercC ||_dSN_client_wrapperselfr#r*b/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/cohere/connectors/raw_client.py__init__' zRawConnectorsClient.__init__Nlimitoffsetrequest_optionsr/r0r1returnc Cs|jjjdd||d|d}zd|jkrdkr0nntttt|d}t ||dWS|jd krPt t |j ttj tjttj tj|dd |jd krptt |j ttj tjttj tj|dd |jd krtt |j ttj tjttj tj|dd |jd krtt |j ttj tjttj tj|dd |jdkrtt |j ttj tjttj tj|dd |jdkrtt |j ttj tjttj tj|dd |jdkrtt |j ttj tjttj tj|dd |jdkr2tt |j ttj tjttj tj|dd |jdkrStt |j ttj tjttj tj|dd |jdkrttt |j ttj tjttj tj|dd |jdkrtt |j ttj tjttj tj|dd |jdkrtt |j ttj tjttj tj|dd |}Wntyt|jt |j |jdwt|jt |j |d)a Returns a list of connectors ordered by descending creation date (newer first). See ['Managing your Connector'](https://docs.cohere.com/docs/managing-your-connector) for more information. Parameters ---------- limit : typing.Optional[float] Maximum number of connectors to return [0, 100]. offset : typing.Optional[float] Number of connectors to skip before returning results [0, inf]. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[ListConnectorsResponse] OK v1/connectorsGETr/r0methodparamsr1,type_object_responsedataheadersbody status_coderCrD)r' httpx_clientrequestrQtypingcastrr jsonrr dictrCOptionalAnyrrrrrrrrrrrrrtextr)r/r0r1 _response_data_response_jsonr*r*r+list*s6                                      zRawConnectorsClient.list descriptionexcludesoauthactivecontinue_on_failure service_authr1nameurlrarbrcrdrerfc Cs|jjjdd||||t|tdd||t|tdddddi| td} zd | jkr/d krDnnt t t t | d } t | | d WS| jd krdtt| jt tjtjt tjtj| d d| jdkrtt| jt tjtjt tjtj| d d| jdkrtt| jt tjtjt tjtj| d d| jdkrtt| jt tjtjt tjtj| d d| jdkrtt| jt tjtjt tjtj| d d| jdkrtt| jt tjtjt tjtj| d d| jdkr&tt| jt tjtjt tjtj| d d| jdkrGtt| jt tjtjt tjtj| d d| jdkrhtt| jt tjtjt tjtj| d d| jdkrtt| jt tjtjt tjtj| d d| jdkrtt| jt tjtjt tjtj| d d| jdkrtt| jt tjtjt tjtj| d d| } Wntyt| jt| j| j dwt| jt| j| d)a Creates a new connector. The connector is tested during registration and will cancel registration when the test is unsuccessful. See ['Creating and Deploying a Connector'](https://docs.cohere.com/v1/docs/creating-and-deploying-a-connector) for more information. Parameters ---------- name : str A human-readable name for the connector. url : str The URL of the connector that will be used to search for documents. description : typing.Optional[str] A description of the connector. excludes : typing.Optional[typing.Sequence[str]] A list of fields to exclude from the prompt (fields remain in the document). oauth : typing.Optional[CreateConnectorOAuth] The OAuth 2.0 configuration for the connector. Cannot be specified if service_auth is specified. active : typing.Optional[bool] Whether the connector is active or not. continue_on_failure : typing.Optional[bool] Whether a chat request should continue or not if the request to this connector fails. service_auth : typing.Optional[CreateConnectorServiceAuth] The service to service authentication configuration for the connector. Cannot be specified if oauth is specified. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[CreateConnectorResponse] OK r3POSTwriter= annotation directionrgrarhrbrcrdrerf content-typeapplication/jsonr7rVrCr1omitr9r:r;r>rArBrErFrGrHrIrJrKrLrMrNrOrP)!r'rRrSr rrOMITrQrTrUrr rVrr rWrCrXrYrrrrrrrrrrrrrrZ r)rgrhrarbrcrdrerfr1r\r]r^r*r*r+createsP2                                     zRawConnectorsClient.creater1idc C|jjjdt|d|d}zd|jkrdkr1nntttt| d}t ||dWS|jdkrQt t |j ttjtjttjtj| dd |jd krqtt |j ttjtjttjtj| dd |jd krtt |j ttjtjttjtj| dd |jd krtt |j ttjtjttjtj| dd |jd krtt |j ttjtjttjtj| dd |jdkrtt |j ttjtjttjtj| dd |jdkrtt |j ttjtjttjtj| dd |jdkr3tt |j ttjtjttjtj| dd |jdkrTtt |j ttjtjttjtj| dd |jdkrutt |j ttjtjttjtj| dd |jdkrtt |j ttjtjttjtj| dd |jdkrtt |j ttjtjttjtj| dd | }Wntyt|jt |j |jdwt|jt |j |d)a Retrieve a connector by ID. See ['Connectors'](https://docs.cohere.com/docs/connectors) for more information. Parameters ---------- id : str The ID of the connector to retrieve. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[GetConnectorResponse] OK v1/connectors/r4r7r1r9r:r;r>rArBrErFrGrHrIrJrKrLrMrNrOrP)r'rRrSr rQrTrUrr rVrr rWrCrXrYrrrrrrrrrrrrrrZr)rwr1r\r]r^r*r*r+get0                                      zRawConnectorsClient.getc Crx)a Delete a connector by ID. See ['Connectors'](https://docs.cohere.com/docs/connectors) for more information. Parameters ---------- id : str The ID of the connector to delete. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[DeleteConnectorResponse] OK ryDELETErzr9r:r;r>rArBrErFrGrHrIrJrKrLrMrNrOrP)r'rRrSr rQrTrUrr rVrr rWrCrXrYrrrrrrrrrrrrrrZr{r*r*r+deletegr}zRawConnectorsClient.deletergrhrbrcrdrerfr1c Cs|jjjdt|d|||t|tdd||t|tdddddi| td} zd | jkr3d krHnnt t t t | d } t| | d WS| jd krhtt| jt t jt jt t jt j| d d| jdkrtt| jt t jt jt t jt j| d d| jdkrtt| jt t jt jt t jt j| d d| jdkrtt| jt t jt jt t jt j| d d| jdkrtt| jt t jt jt t jt j| d d| jdkr tt| jt t jt jt t jt j| d d| jdkr*tt| jt t jt jt t jt j| d d| jdkrKtt| jt t jt jt t jt j| d d| jdkrltt| jt t jt jt t jt j| d d| jdkrtt| jt t jt jt t jt j| d d| jdkrtt| jt t jt jt t jt j| d d| jdkrtt| jt t jt jt t jt j| d d| } Wntyt | jt| j| j!dwt | jt| j| d)a  Update a connector by ID. Omitted fields will not be updated. See ['Managing your Connector'](https://docs.cohere.com/docs/managing-your-connector) for more information. Parameters ---------- id : str The ID of the connector to update. name : typing.Optional[str] A human-readable name for the connector. url : typing.Optional[str] The URL of the connector that will be used to search for documents. excludes : typing.Optional[typing.Sequence[str]] A list of fields to exclude from the prompt (fields remain in the document). oauth : typing.Optional[CreateConnectorOAuth] The OAuth 2.0 configuration for the connector. Cannot be specified if service_auth is specified. active : typing.Optional[bool] continue_on_failure : typing.Optional[bool] service_auth : typing.Optional[CreateConnectorServiceAuth] The service to service authentication configuration for the connector. Cannot be specified if oauth is specified. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[UpdateConnectorResponse] OK ryPATCHrjrkrgrhrbrcrdrerfrorprqr9r:r;r>rArBrErFrGrHrIrJrKrLrMrNrOrP)"r'rRrSr r rrrsrQrTrUr r rVrr rWrCrXrYrrrrrrrrrrrrrrZ r)rwrgrhrbrcrdrerfr1r\r]r^r*r*r+updatesN0                                      zRawConnectorsClient.updateafter_token_redirectr1rc Cs|jjjdt|ddd|i|d}zd|jkr dkr5nntttt| d}t ||d WS|jd krUt t |j ttjtjttjtj| dd |jd krutt |j ttjtjttjtj| dd |jd krtt |j ttjtjttjtj| dd |jdkrtt |j ttjtjttjtj| dd |jdkrtt |j ttjtjttjtj| dd |jdkrtt |j ttjtjttjtj| dd |jdkrtt |j ttjtjttjtj| dd |jdkr7tt |j ttjtjttjtj| dd |jdkrXtt |j ttjtjttjtj| dd |jdkrytt |j ttjtjttjtj| dd |jdkrtt |j ttjtjttjtj| dd |jdkrtt |j ttjtjttjtj| dd | }Wntyt|jt |j |jdwt|jt |j |d)av Authorize the connector with the given ID for the connector oauth app. See ['Connector Authentication'](https://docs.cohere.com/docs/connector-authentication) for more information. Parameters ---------- id : str The ID of the connector to authorize. after_token_redirect : typing.Optional[str] The URL to redirect to after the connector has been authorized. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[OAuthAuthorizeResponse] OK ry/oauth/authorizerirr6r9r:r;r>rArBrErFrGrHrIrJrKrLrMrNrOrP)r'rRrSr rQrTrUrr rVrr rWrCrXrYrrrrrrrrrrrrrrZr)rwrr1r\r]r^r*r*r+o_auth_authorizes4                                     z$RawConnectorsClient.o_auth_authorize)__name__ __module__ __qualname__rr,rTrXfloatr rrr_rsstrSequencerboolrrrurr|rrr rrrr*r*r*r+r"& <    ^ - 0    ^r"c@r!)$AsyncRawConnectorsClientr#cCr$r%r&r(r*r*r+r,r-z!AsyncRawConnectorsClient.__init__Nr.r/r0r1r2c s|jjjdd||d|dIdH}zd|jkrdkr4nntttt|d}t ||d WS|jd krTt t |j ttj tjttj tj|dd |jd krttt |j ttj tjttj tj|dd |jd krtt |j ttj tjttj tj|dd |jdkrtt |j ttj tjttj tj|dd |jdkrtt |j ttj tjttj tj|dd |jdkrtt |j ttj tjttj tj|dd |jdkrtt |j ttj tjttj tj|dd |jdkr6tt |j ttj tjttj tj|dd |jdkrWtt |j ttj tjttj tj|dd |jdkrxtt |j ttj tjttj tj|dd |jdkrtt |j ttj tjttj tj|dd |jdkrtt |j ttj tjttj tj|dd |}Wntyt|jt |j |jdwt|jt |j |d)a Returns a list of connectors ordered by descending creation date (newer first). See ['Managing your Connector'](https://docs.cohere.com/docs/managing-your-connector) for more information. Parameters ---------- limit : typing.Optional[float] Maximum number of connectors to return [0, 100]. offset : typing.Optional[float] Number of connectors to skip before returning results [0, inf]. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[ListConnectorsResponse] OK r3r4r5r6Nr9r:r;r>rArBrErFrGrHrIrJrKrLrMrNrOrP)r'rRrSrQrTrUrr rVrr rWrCrXrYrrrrrrrrrrrrrrZr[r*r*r+r_s8                                       zAsyncRawConnectorsClient.listr`rgrhrarbrcrdrerfc s|jjjdd||||t|tdd||t|tdddddi| tdId H} zd | jkr3d krHnnt t t t | d } t | | d WS| jdkrhtt| jt tjtjt tjtj| d d| jdkrtt| jt tjtjt tjtj| d d| jdkrtt| jt tjtjt tjtj| d d| jdkrtt| jt tjtjt tjtj| d d| jdkrtt| jt tjtjt tjtj| d d| jdkr tt| jt tjtjt tjtj| d d| jdkr*tt| jt tjtjt tjtj| d d| jdkrKtt| jt tjtjt tjtj| d d| jdkrltt| jt tjtjt tjtj| d d| jdkrtt| jt tjtjt tjtj| d d| jdkrtt| jt tjtjt tjtj| d d| jdkrtt| jt tjtjt tjtj| d d| } Wntyt| jt| j| j dwt| jt| j| d)a Creates a new connector. The connector is tested during registration and will cancel registration when the test is unsuccessful. See ['Creating and Deploying a Connector'](https://docs.cohere.com/v1/docs/creating-and-deploying-a-connector) for more information. Parameters ---------- name : str A human-readable name for the connector. url : str The URL of the connector that will be used to search for documents. description : typing.Optional[str] A description of the connector. excludes : typing.Optional[typing.Sequence[str]] A list of fields to exclude from the prompt (fields remain in the document). oauth : typing.Optional[CreateConnectorOAuth] The OAuth 2.0 configuration for the connector. Cannot be specified if service_auth is specified. active : typing.Optional[bool] Whether the connector is active or not. continue_on_failure : typing.Optional[bool] Whether a chat request should continue or not if the request to this connector fails. service_auth : typing.Optional[CreateConnectorServiceAuth] The service to service authentication configuration for the connector. Cannot be specified if oauth is specified. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[CreateConnectorResponse] OK r3rirjrkrnrorprqNr9r:r;r>rArBrErFrGrHrIrJrKrLrMrNrOrP)!r'rRrSr rrrsrQrTrUrr rVrr rWrCrXrYrrrrrrrrrrrrrrZrtr*r*r+ru[sR2                                      zAsyncRawConnectorsClient.creatervrwc |jjjdt|d|dIdH}zd|jkr dkr5nntttt| d}t ||dWS|jd krUt t |j ttjtjttjtj| dd |jd krutt |j ttjtjttjtj| dd |jd krtt |j ttjtjttjtj| dd |jd krtt |j ttjtjttjtj| dd |jdkrtt |j ttjtjttjtj| dd |jdkrtt |j ttjtjttjtj| dd |jdkrtt |j ttjtjttjtj| dd |jdkr7tt |j ttjtjttjtj| dd |jdkrXtt |j ttjtjttjtj| dd |jdkrytt |j ttjtjttjtj| dd |jdkrtt |j ttjtjttjtj| dd |jdkrtt |j ttjtjttjtj| dd | }Wntyt|jt |j |jdwt|jt |j |d)a Retrieve a connector by ID. See ['Connectors'](https://docs.cohere.com/docs/connectors) for more information. Parameters ---------- id : str The ID of the connector to retrieve. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[GetConnectorResponse] OK ryr4rzNr9r:r;r>rArBrErFrGrHrIrJrKrLrMrNrOrP)r'rRrSr rQrTrUrr rVrr rWrCrXrYrrrrrrrrrrrrrrZr{r*r*r+r|72                                       zAsyncRawConnectorsClient.getc r)a Delete a connector by ID. See ['Connectors'](https://docs.cohere.com/docs/connectors) for more information. Parameters ---------- id : str The ID of the connector to delete. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[DeleteConnectorResponse] OK ryr~rzNr9r:r;r>rArBrErFrGrHrIrJrKrLrMrNrOrP)r'rRrSr rQrTrUrr rVrr rWrCrXrYrrrrrrrrrrrrrrZr{r*r*r+rrzAsyncRawConnectorsClient.deleterc s|jjjdt|d|||t|tdd||t|tdddddi| tdId H} zd | jkr7d krLnnt t t t | d } t| | d WS| jdkrltt| jt t jt jt t jt j| d d| jdkrtt| jt t jt jt t jt j| d d| jdkrtt| jt t jt jt t jt j| d d| jdkrtt| jt t jt jt t jt j| d d| jdkrtt| jt t jt jt t jt j| d d| jdkr tt| jt t jt jt t jt j| d d| jdkr.tt| jt t jt jt t jt j| d d| jdkrOtt| jt t jt jt t jt j| d d| jdkrptt| jt t jt jt t jt j| d d| jdkrtt| jt t jt jt t jt j| d d| jdkrtt| jt t jt jt t jt j| d d| jdkrtt| jt t jt jt t jt j| d d| } Wntyt | jt| j| j!dwt | jt| j| d)a Update a connector by ID. Omitted fields will not be updated. See ['Managing your Connector'](https://docs.cohere.com/docs/managing-your-connector) for more information. Parameters ---------- id : str The ID of the connector to update. name : typing.Optional[str] A human-readable name for the connector. url : typing.Optional[str] The URL of the connector that will be used to search for documents. excludes : typing.Optional[typing.Sequence[str]] A list of fields to exclude from the prompt (fields remain in the document). oauth : typing.Optional[CreateConnectorOAuth] The OAuth 2.0 configuration for the connector. Cannot be specified if service_auth is specified. active : typing.Optional[bool] continue_on_failure : typing.Optional[bool] service_auth : typing.Optional[CreateConnectorServiceAuth] The service to service authentication configuration for the connector. Cannot be specified if oauth is specified. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[UpdateConnectorResponse] OK ryrrjrkrrorprqNr9r:r;r>rArBrErFrGrHrIrJrKrLrMrNrOrP)"r'rRrSr r rrrsrQrTrUr r rVrr rWrCrXrYrrrrrrrrrrrrrrZrr*r*r+rsP0                                       zAsyncRawConnectorsClient.updaterrc s|jjjdt|ddd|i|dIdH}zd|jkr$dkr9nntttt| d }t ||d WS|jd krYt t |j ttjtjttjtj| d d |jd krytt |j ttjtjttjtj| d d |jdkrtt |j ttjtjttjtj| d d |jdkrtt |j ttjtjttjtj| d d |jdkrtt |j ttjtjttjtj| d d |jdkrtt |j ttjtjttjtj| d d |jdkrtt |j ttjtjttjtj| d d |jdkr;tt |j ttjtjttjtj| d d |jdkr\tt |j ttjtjttjtj| d d |jdkr}tt |j ttjtjttjtj| d d |jdkrtt |j ttjtjttjtj| d d |jdkrtt |j ttjtjttjtj| d d | }Wntyt|jt |j |jdwt|jt |j |d)a{ Authorize the connector with the given ID for the connector oauth app. See ['Connector Authentication'](https://docs.cohere.com/docs/connector-authentication) for more information. Parameters ---------- id : str The ID of the connector to authorize. after_token_redirect : typing.Optional[str] The URL to redirect to after the connector has been authorized. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[OAuthAuthorizeResponse] OK ryrrirr6Nr9r:r;r>rArBrErFrGrHrIrJrKrLrMrNrOrP)r'rRrSr rQrTrUrr rVrr rWrCrXrYrrrrrrrrrrrrrrZrr*r*r+rfs6                                      z)AsyncRawConnectorsClient.o_auth_authorize)rrrrr,rTrXrr rrr_rsrrrrrrrurr|rrr rrrr*r*r*r+rrr)@rT json.decoderrcore.api_errorrcore.client_wrapperrrcore.http_responserrcore.jsonable_encoderr core.request_optionsr core.serializationr core.unchecked_base_modelr errors.bad_request_errorr "errors.client_closed_request_errorrerrors.forbidden_errorrerrors.gateway_timeout_errorrerrors.internal_server_errorrerrors.invalid_token_errorrerrors.not_found_errorrerrors.not_implemented_errorr errors.service_unavailable_errorrerrors.too_many_requests_errorrerrors.unauthorized_errorr!errors.unprocessable_entity_errorrtypes.create_connector_o_authrtypes.create_connector_responser#types.create_connector_service_authrtypes.delete_connector_responsertypes.get_connector_responsertypes.list_connectors_responsertypes.o_auth_authorize_responsertypes.update_connector_responser rUrYrsr"rr*r*r*r+sR