o i @sddlmZddlZddlZddlmZddlmZddl m Z ddl m Z ddl mZzddlmZWneyEZzed edZ[wwGd d d eeZdS) ) annotationsN) ModelProfile) UserError)cached_async_http_clientcohere_model_profile)Provider) AsyncClientV2uPlease install the `cohere` package to use the Cohere provider, you can use the `cohere` optional group — `pip install "pydantic-ai-slim[cohere]"`c@sXeZdZdZedddZedddZedd d Zdd dZdddddddZ dS)CohereProviderzProvider for Cohere API.returnstrcCsdS)Ncohereselfrrb/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/pydantic_ai/providers/cohere.pynameszCohereProvider.namecCs|jj}t|SN)client_client_wrapperr get_base_url)rclient_wrapperrrrbase_urls zCohereProvider.base_urlr cCs|jSr)_clientrrrrr"szCohereProvider.client model_nameModelProfile | NonecCst|Srr)rrrrr model_profile&szCohereProvider.model_profileN)api_key cohere_client http_clientr str | NonerAsyncClientV2 | Nonerhttpx.AsyncClient | NoneNonecCs|dur|dus Jd|dusJd||_dS|ptd}|s&tdtd}|dur9t|||d|_dStdd }t|||d|_dS) aCreate a new Cohere provider. Args: api_key: The API key to use for authentication, if not provided, the `CO_API_KEY` environment variable will be used if available. cohere_client: An existing [AsyncClientV2](https://github.com/cohere-ai/cohere-python) client to use. If provided, `api_key` and `http_client` must be `None`. http_client: An existing `httpx.AsyncClient` to use for making HTTP requests. Nz5Cannot provide both `cohere_client` and `http_client`z1Cannot provide both `cohere_client` and `api_key` CO_API_KEYzqSet the `CO_API_KEY` environment variable or pass it via `CohereProvider(api_key=...)`to use the Cohere provider. CO_BASE_URL)r httpx_clientrr )provider)rosgetenvrr r)rrrrrrrr__init__)s   zCohereProvider.__init__)r r )r r )rr r r)rr rr!rr"r r#) __name__ __module__ __qualname____doc__propertyrrrrr*rrrrr s    r ) __future__r _annotationsr(httpx pydantic_airpydantic_ai.exceptionsrpydantic_ai.modelsrpydantic_ai.profiles.cohererpydantic_ai.providersrr r ImportError _import_errorr rrrrs&