o iUT@sddlZddlmZmZddlmZddlmZddlm Z ddl m Z ddl m Z dd lmZdd lmZdd lmZdd lmZd dlmZmZeejdZGdddZGdddZdS)N)AsyncClientWrapperSyncClientWrapper)RequestOptions)CreateConnectorOAuth)CreateConnectorResponse)CreateConnectorServiceAuth)DeleteConnectorResponse)GetConnectorResponse)ListConnectorsResponse)OAuthAuthorizeResponse)UpdateConnectorResponse)AsyncRawConnectorsClientRawConnectorsClient.c@eZdZdefddZedefddZddddd ej e d ej e d ej e de fd d Z eeeeeedddededej edej ejedej edej edej edej ed ej e defddZddded ej e defddZddded ej e defddZeeeeeeedddedej edej edej ejedej edej edej edej ed ej e defd d!Zddd"ded#ej ed ej e defd$d%ZdS)&ConnectorsClientclient_wrappercCt|d|_dSN)r)r _raw_clientselfrr^/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/cohere/connectors/client.py__init__zConnectorsClient.__init__returncC|jS)z Retrieves a raw implementation of this client that returns raw responses. Returns ------- RawConnectorsClient rrrrrwith_raw_response z"ConnectorsClient.with_raw_responseNlimitoffsetrequest_optionsr$r%r&cC|jj|||d}|jS)aY 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 ------- ListConnectorsResponse OK Examples -------- from cohere import Client client = Client( client_name="YOUR_CLIENT_NAME", token="YOUR_TOKEN", ) client.connectors.list() r#rlistdatarr$r%r& _responserrrr)$s$zConnectorsClient.list descriptionexcludesoauthactivecontinue_on_failure service_authr&nameurlr.r/r0r1r2r3c C$|jj||||||||| d } | jS)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 ------- CreateConnectorResponse OK Examples -------- from cohere import Client client = Client( client_name="YOUR_CLIENT_NAME", token="YOUR_TOKEN", ) client.connectors.create( name="name", url="url", ) r4r5r.r/r0r1r2r3r&rcreater* rr4r5r.r/r0r1r2r3r&r,rrrr9Ks? zConnectorsClient.creater&idcC|jj||d}|jS)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 ------- GetConnectorResponse OK Examples -------- from cohere import Client client = Client( client_name="YOUR_CLIENT_NAME", token="YOUR_TOKEN", ) client.connectors.get( id="id", ) r;rgetr*rr<r&r,rrrr?zConnectorsClient.getcCr=)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 ------- DeleteConnectorResponse OK Examples -------- from cohere import Client client = Client( client_name="YOUR_CLIENT_NAME", token="YOUR_TOKEN", ) client.connectors.delete( id="id", ) r;rdeleter*r@rrrrCrAzConnectorsClient.deleter4r5r/r0r1r2r3r&c Cr6)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 ------- UpdateConnectorResponse OK Examples -------- from cohere import Client client = Client( client_name="YOUR_CLIENT_NAME", token="YOUR_TOKEN", ) client.connectors.update( id="id", ) rDrupdater* rr<r4r5r/r0r1r2r3r&r,rrrrFs< zConnectorsClient.updateafter_token_redirectr&rIcCr')ah 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 ------- OAuthAuthorizeResponse OK Examples -------- from cohere import Client client = Client( client_name="YOUR_CLIENT_NAME", token="YOUR_TOKEN", ) client.connectors.o_auth_authorize( id="id", ) rHro_auth_authorizer*rr<rIr&r,rrrrK s&z!ConnectorsClient.o_auth_authorize)__name__ __module__ __qualname__rrpropertyrr!typingOptionalfloatrr r)OMITstrSequencerboolrrr9r r?r rCr rFr rKrrrrrs  ,    "L" $    Mrc@r)&AsyncConnectorsClientrcCrr)rrrrrrrMrzAsyncConnectorsClient.__init__rcCr)z Retrieves a raw implementation of this client that returns raw responses. Returns ------- AsyncRawConnectorsClient rr rrrr!Pr"z'AsyncConnectorsClient.with_raw_responseNr#r$r%r&c |jj|||dIdH}|jS)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 ------- ListConnectorsResponse OK Examples -------- import asyncio from cohere import AsyncClient client = AsyncClient( client_name="YOUR_CLIENT_NAME", token="YOUR_TOKEN", ) async def main() -> None: await client.connectors.list() asyncio.run(main()) r#Nr(r+rrrr)[s,zAsyncConnectorsClient.listr-r4r5r.r/r0r1r2r3c ,|jj||||||||| d IdH} | jS)ao 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 ------- CreateConnectorResponse OK Examples -------- import asyncio from cohere import AsyncClient client = AsyncClient( client_name="YOUR_CLIENT_NAME", token="YOUR_TOKEN", ) async def main() -> None: await client.connectors.create( name="name", url="url", ) asyncio.run(main()) r7Nr8r:rrrr9sG  zAsyncConnectorsClient.creater;r<c|jj||dIdH}|jS)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 ------- GetConnectorResponse OK Examples -------- import asyncio from cohere import AsyncClient client = AsyncClient( client_name="YOUR_CLIENT_NAME", token="YOUR_TOKEN", ) async def main() -> None: await client.connectors.get( id="id", ) asyncio.run(main()) r;Nr>r@rrrr?s%zAsyncConnectorsClient.getcr[)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 ------- DeleteConnectorResponse OK Examples -------- import asyncio from cohere import AsyncClient client = AsyncClient( client_name="YOUR_CLIENT_NAME", token="YOUR_TOKEN", ) async def main() -> None: await client.connectors.delete( id="id", ) asyncio.run(main()) r;NrBr@rrrrCs'zAsyncConnectorsClient.deleterDc rZ)ai 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 ------- UpdateConnectorResponse OK Examples -------- import asyncio from cohere import AsyncClient client = AsyncClient( client_name="YOUR_CLIENT_NAME", token="YOUR_TOKEN", ) async def main() -> None: await client.connectors.update( id="id", ) asyncio.run(main()) rDNrErGrrrrF0sD  zAsyncConnectorsClient.updaterHrIcrY)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 ------- OAuthAuthorizeResponse OK Examples -------- import asyncio from cohere import AsyncClient client = AsyncClient( client_name="YOUR_CLIENT_NAME", token="YOUR_TOKEN", ) async def main() -> None: await client.connectors.o_auth_authorize( id="id", ) asyncio.run(main()) rHNrJrLrrrrKs . z&AsyncConnectorsClient.o_auth_authorize)rMrNrOrrrPrr!rQrRrSrr r)rTrUrVrrWrrr9r r?r rCr rFr rKrrrrrXLs  4    "T) .    UrX)rQcore.client_wrapperrrcore.request_optionsrtypes.create_connector_o_authrtypes.create_connector_responser#types.create_connector_service_authrtypes.delete_connector_responser types.get_connector_responser types.list_connectors_responser types.o_auth_authorize_responser types.update_connector_responser raw_clientrrcastAnyrTrrXrrrrs"         9