SYMBOL INDEX (1368 symbols across 285 files) FILE: src/cohere/__init__.py function __getattr__ (line 635) | def __getattr__(attr_name: str) -> typing.Any: function __dir__ (line 651) | def __dir__(): FILE: src/cohere/_default_clients.py class DefaultAioHttpClient (line 13) | class DefaultAioHttpClient(httpx.AsyncClient): # type: ignore method __init__ (line 14) | def __init__(self, **kwargs: typing.Any) -> None: method __init__ (line 20) | def __init__(self, **kwargs: typing.Any) -> None: class DefaultAioHttpClient (line 19) | class DefaultAioHttpClient(httpx_aiohttp.HttpxAiohttpClient): # type: i... method __init__ (line 14) | def __init__(self, **kwargs: typing.Any) -> None: method __init__ (line 20) | def __init__(self, **kwargs: typing.Any) -> None: class DefaultAsyncHttpxClient (line 26) | class DefaultAsyncHttpxClient(httpx.AsyncClient): method __init__ (line 27) | def __init__(self, **kwargs: typing.Any) -> None: FILE: src/cohere/audio/__init__.py function __getattr__ (line 17) | def __getattr__(attr_name: str) -> typing.Any: function __dir__ (line 33) | def __dir__(): FILE: src/cohere/audio/client.py class AudioClient (line 14) | class AudioClient: method __init__ (line 15) | def __init__(self, *, client_wrapper: SyncClientWrapper): method with_raw_response (line 21) | def with_raw_response(self) -> RawAudioClient: method transcriptions (line 32) | def transcriptions(self): class AsyncAudioClient (line 40) | class AsyncAudioClient: method __init__ (line 41) | def __init__(self, *, client_wrapper: AsyncClientWrapper): method with_raw_response (line 47) | def with_raw_response(self) -> AsyncRawAudioClient: method transcriptions (line 58) | def transcriptions(self): FILE: src/cohere/audio/raw_client.py class RawAudioClient (line 6) | class RawAudioClient: method __init__ (line 7) | def __init__(self, *, client_wrapper: SyncClientWrapper): class AsyncRawAudioClient (line 11) | class AsyncRawAudioClient: method __init__ (line 12) | def __init__(self, *, client_wrapper: AsyncClientWrapper): FILE: src/cohere/audio/transcriptions/__init__.py function __getattr__ (line 13) | def __getattr__(attr_name: str) -> typing.Any: function __dir__ (line 29) | def __dir__(): FILE: src/cohere/audio/transcriptions/client.py class TranscriptionsClient (line 15) | class TranscriptionsClient: method __init__ (line 16) | def __init__(self, *, client_wrapper: SyncClientWrapper): method with_raw_response (line 20) | def with_raw_response(self) -> RawTranscriptionsClient: method create (line 30) | def create( class AsyncTranscriptionsClient (line 83) | class AsyncTranscriptionsClient: method __init__ (line 84) | def __init__(self, *, client_wrapper: AsyncClientWrapper): method with_raw_response (line 88) | def with_raw_response(self) -> AsyncRawTranscriptionsClient: method create (line 98) | async def create( FILE: src/cohere/audio/transcriptions/raw_client.py class RawTranscriptionsClient (line 32) | class RawTranscriptionsClient: method __init__ (line 33) | def __init__(self, *, client_wrapper: SyncClientWrapper): method create (line 36) | def create( class AsyncRawTranscriptionsClient (line 237) | class AsyncRawTranscriptionsClient: method __init__ (line 238) | def __init__(self, *, client_wrapper: AsyncClientWrapper): method create (line 241) | async def create( FILE: src/cohere/audio/transcriptions/types/__init__.py function __getattr__ (line 13) | def __getattr__(attr_name: str) -> typing.Any: function __dir__ (line 29) | def __dir__(): FILE: src/cohere/audio/transcriptions/types/audio_transcriptions_create_response.py class AudioTranscriptionsCreateResponse (line 10) | class AudioTranscriptionsCreateResponse(UncheckedBaseModel): class Config (line 20) | class Config: FILE: src/cohere/aws_client.py class AwsClient (line 17) | class AwsClient(Client): method __init__ (line 18) | def __init__( class AwsClientV2 (line 48) | class AwsClientV2(ClientV2): method __init__ (line 49) | def __init__( function get_event_hooks (line 82) | def get_event_hooks( class Streamer (line 115) | class Streamer(SyncByteStream): method __init__ (line 118) | def __init__(self, lines: typing.Iterator[bytes]): method __iter__ (line 121) | def __iter__(self) -> typing.Iterator[bytes]: function stream_generator (line 138) | def stream_generator(response: httpx.Response, endpoint: str) -> typing.... function map_token_counts (line 155) | def map_token_counts(response: httpx.Response) -> ApiMeta: function map_response_from_bedrock (line 164) | def map_response_from_bedrock(): function get_boto3_session (line 199) | def get_boto3_session( function map_request_to_bedrock (line 207) | def map_request_to_bedrock( function get_url (line 272) | def get_url( function get_api_version (line 288) | def get_api_version(*, version: str): FILE: src/cohere/base_client.py class BaseCohere (line 65) | class BaseCohere: method __init__ (line 110) | def __init__( method with_raw_response (line 152) | def with_raw_response(self) -> RawBaseCohere: method chat_stream (line 162) | def chat_stream( method chat (line 463) | def chat( method generate_stream (line 761) | def generate_stream( method generate (line 909) | def generate( method embed (line 1055) | def embed( method rerank (line 1139) | def rerank( method classify (line 1234) | def classify( method summarize (line 1343) | def summarize( method tokenize (line 1416) | def tokenize( method detokenize (line 1454) | def detokenize( method check_api_key (line 1492) | def check_api_key(self, *, request_options: typing.Optional[RequestOpt... method v2 (line 1520) | def v2(self): method batches (line 1528) | def batches(self): method embed_jobs (line 1536) | def embed_jobs(self): method datasets (line 1544) | def datasets(self): method connectors (line 1552) | def connectors(self): method models (line 1560) | def models(self): method finetuning (line 1568) | def finetuning(self): method audio (line 1576) | def audio(self): function _make_default_async_client (line 1584) | def _make_default_async_client( class AsyncBaseCohere (line 1602) | class AsyncBaseCohere: method __init__ (line 1650) | def __init__( method with_raw_response (line 1692) | def with_raw_response(self) -> AsyncRawBaseCohere: method chat_stream (line 1702) | async def chat_stream( method chat (line 2012) | async def chat( method generate_stream (line 2318) | async def generate_stream( method generate (line 2475) | async def generate( method embed (line 2629) | async def embed( method rerank (line 2721) | async def rerank( method classify (line 2824) | async def classify( method summarize (line 2941) | async def summarize( method tokenize (line 3022) | async def tokenize( method detokenize (line 3068) | async def detokenize( method check_api_key (line 3114) | async def check_api_key(self, *, request_options: typing.Optional[Requ... method v2 (line 3150) | def v2(self): method batches (line 3158) | def batches(self): method embed_jobs (line 3166) | def embed_jobs(self): method datasets (line 3174) | def datasets(self): method connectors (line 3182) | def connectors(self): method models (line 3190) | def models(self): method finetuning (line 3198) | def finetuning(self): method audio (line 3206) | def audio(self): function _get_base_url (line 3214) | def _get_base_url(*, base_url: typing.Optional[str] = None, environment:... FILE: src/cohere/batches/__init__.py function __getattr__ (line 27) | def __getattr__(attr_name: str) -> typing.Any: function __dir__ (line 43) | def __dir__(): FILE: src/cohere/batches/client.py class BatchesClient (line 18) | class BatchesClient: method __init__ (line 19) | def __init__(self, *, client_wrapper: SyncClientWrapper): method with_raw_response (line 23) | def with_raw_response(self) -> RawBatchesClient: method list (line 33) | def list( method create (line 87) | def create(self, *, request: Batch, request_options: typing.Optional[R... method retrieve (line 123) | def retrieve(self, id: str, *, request_options: typing.Optional[Reques... method cancel (line 155) | def cancel(self, id: str, *, request_options: typing.Optional[RequestO... class AsyncBatchesClient (line 188) | class AsyncBatchesClient: method __init__ (line 189) | def __init__(self, *, client_wrapper: AsyncClientWrapper): method with_raw_response (line 193) | def with_raw_response(self) -> AsyncRawBatchesClient: method list (line 203) | async def list( method create (line 265) | async def create( method retrieve (line 311) | async def retrieve(self, id: str, *, request_options: typing.Optional[... method cancel (line 351) | async def cancel(self, id: str, *, request_options: typing.Optional[Re... FILE: src/cohere/batches/raw_client.py class RawBatchesClient (line 31) | class RawBatchesClient: method __init__ (line 32) | def __init__(self, *, client_wrapper: SyncClientWrapper): method list (line 35) | def list( method create (line 165) | def create( method retrieve (line 278) | def retrieve( method cancel (line 387) | def cancel( class AsyncRawBatchesClient (line 497) | class AsyncRawBatchesClient: method __init__ (line 498) | def __init__(self, *, client_wrapper: AsyncClientWrapper): method list (line 501) | async def list( method create (line 631) | async def create( method retrieve (line 744) | async def retrieve( method cancel (line 853) | async def cancel( FILE: src/cohere/batches/types/__init__.py function __getattr__ (line 25) | def __getattr__(attr_name: str) -> typing.Any: function __dir__ (line 41) | def __dir__(): FILE: src/cohere/batches/types/batch.py class Batch (line 12) | class Batch(UncheckedBaseModel): class Config (line 97) | class Config: FILE: src/cohere/batches/types/create_batch_response.py class CreateBatchResponse (line 11) | class CreateBatchResponse(UncheckedBaseModel): class Config (line 25) | class Config: FILE: src/cohere/batches/types/get_batch_response.py class GetBatchResponse (line 11) | class GetBatchResponse(UncheckedBaseModel): class Config (line 25) | class Config: FILE: src/cohere/batches/types/list_batches_response.py class ListBatchesResponse (line 11) | class ListBatchesResponse(UncheckedBaseModel): class Config (line 31) | class Config: FILE: src/cohere/bedrock_client.py class BedrockClient (line 8) | class BedrockClient(AwsClient): method __init__ (line 9) | def __init__( method rerank (line 28) | def rerank(self, *, query, documents, model = ..., top_n = ..., rank_f... class BedrockClientV2 (line 31) | class BedrockClientV2(AwsClientV2): method __init__ (line 32) | def __init__( FILE: src/cohere/client.py function validate_args (line 30) | def validate_args(obj: typing.Any, method_name: str, check_fn: typing.Ca... function throw_if_stream_is_true (line 51) | def throw_if_stream_is_true(*args, **kwargs) -> None: function moved_function (line 58) | def moved_function(fn_name: str, new_fn_name: str) -> typing.Any: function deprecated_function (line 72) | def deprecated_function(fn_name: str) -> typing.Any: function experimental_kwarg_decorator (line 88) | def experimental_kwarg_decorator(func, deprecated_kwarg): function fix_base_url (line 123) | def fix_base_url(base_url: typing.Optional[str]) -> typing.Optional[str]: class Client (line 131) | class Client(BaseCohere, CacheMixin): method __init__ (line 134) | def __init__( method __enter__ (line 172) | def __enter__(self): method __exit__ (line 175) | def __exit__(self, exc_type, exc_value, traceback): method embed (line 180) | def embed( method embed_stream (line 226) | def embed_stream( method tokenize (line 326) | def tokenize( method detokenize (line 348) | def detokenize( method fetch_tokenizer (line 371) | def fetch_tokenizer(self, *, model: str) -> Tokenizer: class AsyncClient (line 378) | class AsyncClient(AsyncBaseCohere, CacheMixin): method __init__ (line 381) | def __init__( method __aenter__ (line 419) | async def __aenter__(self): method __aexit__ (line 422) | async def __aexit__(self, exc_type, exc_value, traceback): method embed (line 427) | async def embed( method tokenize (line 520) | async def tokenize( method detokenize (line 541) | async def detokenize( method fetch_tokenizer (line 562) | async def fetch_tokenizer(self, *, model: str) -> Tokenizer: function _get_api_key_from_environment (line 569) | def _get_api_key_from_environment() -> typing.Optional[str]: FILE: src/cohere/client_v2.py class _CombinedRawClient (line 11) | class _CombinedRawClient: method __init__ (line 20) | def __init__(self, v1_raw_client: typing.Any, v2_raw_client: typing.Any): method __getattr__ (line 24) | def __getattr__(self, name: str) -> typing.Any: class ClientV2 (line 31) | class ClientV2(V2Client, Client): # type: ignore method __init__ (line 32) | def __init__( class AsyncClientV2 (line 64) | class AsyncClientV2(AsyncV2Client, AsyncClient): # type: ignore method __init__ (line 65) | def __init__( FILE: src/cohere/connectors/client.py class ConnectorsClient (line 21) | class ConnectorsClient: method __init__ (line 22) | def __init__(self, *, client_wrapper: SyncClientWrapper): method with_raw_response (line 26) | def with_raw_response(self) -> RawConnectorsClient: method list (line 36) | def list( method create (line 78) | def create( method get (line 154) | def get(self, id: str, *, request_options: typing.Optional[RequestOpti... method delete (line 186) | def delete(self, id: str, *, request_options: typing.Optional[RequestO... method update (line 218) | def update( method o_auth_authorize (line 291) | def o_auth_authorize( class AsyncConnectorsClient (line 336) | class AsyncConnectorsClient: method __init__ (line 337) | def __init__(self, *, client_wrapper: AsyncClientWrapper): method with_raw_response (line 341) | def with_raw_response(self) -> AsyncRawConnectorsClient: method list (line 351) | async def list( method create (line 401) | async def create( method get (line 485) | async def get(self, id: str, *, request_options: typing.Optional[Reque... method delete (line 525) | async def delete( method update (line 567) | async def update( method o_auth_authorize (line 648) | async def o_auth_authorize( FILE: src/cohere/connectors/raw_client.py class RawConnectorsClient (line 40) | class RawConnectorsClient: method __init__ (line 41) | def __init__(self, *, client_wrapper: SyncClientWrapper): method list (line 44) | def list( method create (line 230) | def create( method get (line 454) | def get( method delete (line 629) | def delete( method update (line 804) | def update( method o_auth_authorize (line 1025) | def o_auth_authorize( class AsyncRawConnectorsClient (line 1211) | class AsyncRawConnectorsClient: method __init__ (line 1212) | def __init__(self, *, client_wrapper: AsyncClientWrapper): method list (line 1215) | async def list( method create (line 1401) | async def create( method get (line 1625) | async def get( method delete (line 1800) | async def delete( method update (line 1975) | async def update( method o_auth_authorize (line 2196) | async def o_auth_authorize( FILE: src/cohere/core/__init__.py function __getattr__ (line 73) | def __getattr__(attr_name: str) -> typing.Any: function __dir__ (line 89) | def __dir__(): FILE: src/cohere/core/api_error.py class ApiError (line 6) | class ApiError(Exception): method __init__ (line 11) | def __init__( method __str__ (line 22) | def __str__(self) -> str: FILE: src/cohere/core/client_wrapper.py class BaseClientWrapper (line 10) | class BaseClientWrapper: method __init__ (line 11) | def __init__( method get_headers (line 28) | def get_headers(self) -> typing.Dict[str, str]: method _get_token (line 45) | def _get_token(self) -> str: method get_custom_headers (line 51) | def get_custom_headers(self) -> typing.Optional[typing.Dict[str, str]]: method get_base_url (line 54) | def get_base_url(self) -> str: method get_timeout (line 57) | def get_timeout(self) -> typing.Optional[float]: class SyncClientWrapper (line 61) | class SyncClientWrapper(BaseClientWrapper): method __init__ (line 62) | def __init__( class AsyncClientWrapper (line 85) | class AsyncClientWrapper(BaseClientWrapper): method __init__ (line 86) | def __init__( method async_get_headers (line 111) | async def async_get_headers(self) -> typing.Dict[str, str]: FILE: src/cohere/core/datetime_utils.py function parse_rfc2822_datetime (line 12) | def parse_rfc2822_datetime(v: Any) -> dt.datetime: class Rfc2822DateTime (line 30) | class Rfc2822DateTime(dt.datetime): method __get_validators__ (line 38) | def __get_validators__(cls): # type: ignore[no-untyped-def] method __get_pydantic_core_schema__ (line 42) | def __get_pydantic_core_schema__(cls, _source_type: Any, _handler: Any... function serialize_datetime (line 48) | def serialize_datetime(v: dt.datetime) -> str: FILE: src/cohere/core/file.py function convert_file_dict_to_httpx_tuples (line 25) | def convert_file_dict_to_httpx_tuples( function with_content_type (line 46) | def with_content_type(*, file: File, default_content_type: str) -> File: FILE: src/cohere/core/force_multipart.py class ForceMultipartDict (line 6) | class ForceMultipartDict(Dict[str, Any]): method __bool__ (line 14) | def __bool__(self) -> bool: FILE: src/cohere/core/http_client.py function _parse_retry_after (line 26) | def _parse_retry_after(response_headers: httpx.Headers) -> typing.Option... function _add_positive_jitter (line 67) | def _add_positive_jitter(delay: float) -> float: function _add_symmetric_jitter (line 73) | def _add_symmetric_jitter(delay: float) -> float: function _parse_x_ratelimit_reset (line 79) | def _parse_x_ratelimit_reset(response_headers: httpx.Headers) -> typing.... function _retry_timeout (line 99) | def _retry_timeout(response: httpx.Response, retries: int) -> float: function _retry_timeout_from_retries (line 121) | def _retry_timeout_from_retries(retries: int) -> float: function _should_retry (line 127) | def _should_retry(response: httpx.Response) -> bool: function _redact_headers (line 154) | def _redact_headers(headers: typing.Dict[str, str]) -> typing.Dict[str, ... function _build_url (line 158) | def _build_url(base_url: str, path: typing.Optional[str]) -> str: function _maybe_filter_none_from_multipart_data (line 182) | def _maybe_filter_none_from_multipart_data( function remove_omit_from_dict (line 197) | def remove_omit_from_dict( function maybe_filter_request_body (line 210) | def maybe_filter_request_body( function get_request_body (line 236) | def get_request_body( class HttpClient (line 266) | class HttpClient: method __init__ (line 267) | def __init__( method get_base_url (line 284) | def get_base_url(self, maybe_base_url: typing.Optional[str]) -> str: method request (line 293) | def request( method stream (line 453) | def stream( class AsyncHttpClient (line 548) | class AsyncHttpClient: method __init__ (line 549) | def __init__( method _get_headers (line 568) | async def _get_headers(self) -> typing.Dict[str, str]: method get_base_url (line 573) | def get_base_url(self, maybe_base_url: typing.Optional[str]) -> str: method request (line 582) | async def request( method stream (line 745) | async def stream( FILE: src/cohere/core/http_response.py class BaseHttpResponse (line 11) | class BaseHttpResponse: method __init__ (line 16) | def __init__(self, response: httpx.Response): method headers (line 20) | def headers(self) -> Dict[str, str]: method status_code (line 24) | def status_code(self) -> int: class HttpResponse (line 28) | class HttpResponse(Generic[T], BaseHttpResponse): method __init__ (line 33) | def __init__(self, response: httpx.Response, data: T): method data (line 38) | def data(self) -> T: method close (line 41) | def close(self) -> None: class AsyncHttpResponse (line 45) | class AsyncHttpResponse(Generic[T], BaseHttpResponse): method __init__ (line 50) | def __init__(self, response: httpx.Response, data: T): method data (line 55) | def data(self) -> T: method close (line 58) | async def close(self) -> None: FILE: src/cohere/core/http_sse/__init__.py function __getattr__ (line 21) | def __getattr__(attr_name: str) -> typing.Any: function __dir__ (line 37) | def __dir__(): FILE: src/cohere/core/http_sse/_api.py class EventSource (line 13) | class EventSource: method __init__ (line 14) | def __init__(self, response: httpx.Response) -> None: method _check_content_type (line 17) | def _check_content_type(self) -> None: method _get_charset (line 24) | def _get_charset(self) -> str: method response (line 45) | def response(self) -> httpx.Response: method iter_sse (line 48) | def iter_sse(self) -> Iterator[ServerSentEvent]: method aiter_sse (line 76) | async def aiter_sse(self) -> AsyncGenerator[ServerSentEvent, None]: function connect_sse (line 91) | def connect_sse(client: httpx.Client, method: str, url: str, **kwargs: A... function aconnect_sse (line 101) | async def aconnect_sse( FILE: src/cohere/core/http_sse/_decoders.py class SSEDecoder (line 8) | class SSEDecoder: method __init__ (line 9) | def __init__(self) -> None: method decode (line 15) | def decode(self, line: str) -> Optional[ServerSentEvent]: FILE: src/cohere/core/http_sse/_exceptions.py class SSEError (line 6) | class SSEError(httpx.TransportError): FILE: src/cohere/core/http_sse/_models.py class ServerSentEvent (line 9) | class ServerSentEvent: method json (line 15) | def json(self) -> Any: FILE: src/cohere/core/jsonable_encoder.py function jsonable_encoder (line 31) | def jsonable_encoder(obj: Any, custom_encoder: Optional[Dict[Any, Callab... FILE: src/cohere/core/logging.py class ILogger (line 16) | class ILogger(typing.Protocol): method debug (line 17) | def debug(self, message: str, **kwargs: typing.Any) -> None: ... method info (line 18) | def info(self, message: str, **kwargs: typing.Any) -> None: ... method warn (line 19) | def warn(self, message: str, **kwargs: typing.Any) -> None: ... method error (line 20) | def error(self, message: str, **kwargs: typing.Any) -> None: ... class ConsoleLogger (line 23) | class ConsoleLogger: method __init__ (line 26) | def __init__(self) -> None: method debug (line 34) | def debug(self, message: str, **kwargs: typing.Any) -> None: method info (line 37) | def info(self, message: str, **kwargs: typing.Any) -> None: method warn (line 40) | def warn(self, message: str, **kwargs: typing.Any) -> None: method error (line 43) | def error(self, message: str, **kwargs: typing.Any) -> None: class LogConfig (line 47) | class LogConfig(typing.TypedDict, total=False): class Logger (line 53) | class Logger: method __init__ (line 58) | def __init__(self, *, level: LogLevel, logger: ILogger, silent: bool) ... method _should_log (line 63) | def _should_log(self, level: LogLevel) -> bool: method is_debug (line 66) | def is_debug(self) -> bool: method is_info (line 69) | def is_info(self) -> bool: method is_warn (line 72) | def is_warn(self) -> bool: method is_error (line 75) | def is_error(self) -> bool: method debug (line 78) | def debug(self, message: str, **kwargs: typing.Any) -> None: method info (line 82) | def info(self, message: str, **kwargs: typing.Any) -> None: method warn (line 86) | def warn(self, message: str, **kwargs: typing.Any) -> None: method error (line 90) | def error(self, message: str, **kwargs: typing.Any) -> None: function create_logger (line 98) | def create_logger(config: typing.Optional[typing.Union[LogConfig, Logger... FILE: src/cohere/core/parse_error.py class ParsingError (line 6) | class ParsingError(Exception): method __init__ (line 18) | def __init__( method __str__ (line 34) | def __str__(self) -> str: FILE: src/cohere/core/pydantic_utilities.py function parse_datetime (line 42) | def parse_datetime(value: Any) -> dt.datetime: # type: ignore[misc] function parse_date (line 47) | def parse_date(value: Any) -> dt.date: # type: ignore[misc] function is_literal_type (line 58) | def is_literal_type(tp: Optional[Type[Any]]) -> bool: # type: ignore[misc] function is_union (line 61) | def is_union(tp: Optional[Type[Any]]) -> bool: # type: ignore[misc] function _decimal_encoder (line 93) | def _decimal_encoder(dec_value: Any) -> Any: function _get_discriminator_and_variants (line 138) | def _get_discriminator_and_variants(type_: Type[Any]) -> Tuple[Optional[... function _get_field_annotation (line 165) | def _get_field_annotation(model: Type[Any], field_name: str) -> Optional... function _find_variant_by_discriminator (line 180) | def _find_variant_by_discriminator( function _is_string_type (line 198) | def _is_string_type(type_: Type[Any]) -> bool: function parse_sse_obj (line 214) | def parse_sse_obj(sse: "ServerSentEvent", type_: Type[T]) -> T: function parse_obj_as (line 313) | def parse_obj_as(type_: Type[T], object_: Any) -> T: function to_jsonable_with_fallback (line 350) | def to_jsonable_with_fallback(obj: Any, fallback_serializer: Callable[[A... class UniversalBaseModel (line 358) | class UniversalBaseModel(pydantic.BaseModel): method _coerce_field_names_to_aliases (line 367) | def _coerce_field_names_to_aliases(cls, data: Any) -> Any: method serialize_model (line 403) | def serialize_model(self) -> Any: # type: ignore[name-defined] class Config (line 410) | class Config: method _coerce_field_names_to_aliases (line 415) | def _coerce_field_names_to_aliases(cls, values: Any) -> Any: method model_construct (line 449) | def model_construct(cls: Type["Model"], _fields_set: Optional[Set[str]... method construct (line 454) | def construct(cls: Type["Model"], _fields_set: Optional[Set[str]] = No... method json (line 460) | def json(self, **kwargs: Any) -> str: method dict (line 470) | def dict(self, **kwargs: Any) -> Dict[str, Any]: function _union_list_of_pydantic_dicts (line 530) | def _union_list_of_pydantic_dicts(source: List[Any], destination: List[A... function deep_union_pydantic_dicts (line 543) | def deep_union_pydantic_dicts(source: Dict[str, Any], destination: Dict[... class V2RootModel (line 561) | class V2RootModel(UniversalBaseModel, pydantic.RootModel): # type: igno... function encode_by_type (line 569) | def encode_by_type(o: Any) -> Any: function update_forward_refs (line 581) | def update_forward_refs(model: Type["Model"], **localns: Any) -> None: function universal_root_validator (line 592) | def universal_root_validator( function universal_field_validator (line 605) | def universal_field_validator(field_name: str, pre: bool = False) -> Cal... function _get_model_fields (line 617) | def _get_model_fields(model: Type["Model"]) -> Mapping[str, PydanticField]: function _get_field_default (line 623) | def _get_field_default(field: PydanticField) -> Any: FILE: src/cohere/core/query_encoder.py function traverse_query_dict (line 9) | def traverse_query_dict(dict_flat: Dict[str, Any], key_prefix: Optional[... function single_query_encoder (line 26) | def single_query_encoder(query_key: str, query_value: Any) -> List[Tuple... function encode_query (line 51) | def encode_query(query: Optional[Dict[str, Any]]) -> Optional[List[Tuple... FILE: src/cohere/core/remove_none_from_dict.py function remove_none_from_dict (line 6) | def remove_none_from_dict(original: Mapping[str, Optional[Any]]) -> Dict... FILE: src/cohere/core/request_options.py class RequestOptions (line 11) | class RequestOptions(typing.TypedDict, total=False): FILE: src/cohere/core/serialization.py class FieldMetadata (line 11) | class FieldMetadata: method __init__ (line 25) | def __init__(self, *, alias: str) -> None: function convert_and_respect_annotation_metadata (line 29) | def convert_and_respect_annotation_metadata( function _convert_mapping (line 157) | def _convert_mapping( function _get_annotation (line 194) | def _get_annotation(type_: typing.Any) -> typing.Optional[typing.Any]: function _remove_annotations (line 209) | def _remove_annotations(type_: typing.Any) -> typing.Any: function get_alias_to_field_mapping (line 223) | def get_alias_to_field_mapping(type_: typing.Any) -> typing.Dict[str, str]: function get_field_to_alias_mapping (line 228) | def get_field_to_alias_mapping(type_: typing.Any) -> typing.Dict[str, str]: function _get_alias_to_field_name (line 233) | def _get_alias_to_field_name( function _get_field_to_alias_name (line 244) | def _get_field_to_alias_name( function _get_alias_from_type (line 255) | def _get_alias_from_type(type_: typing.Any) -> typing.Optional[str]: function _alias_key (line 268) | def _alias_key( FILE: src/cohere/core/unchecked_base_model.py class UnionMetadata (line 28) | class UnionMetadata: method __init__ (line 31) | def __init__(self, *, discriminant: str) -> None: function _maybe_resolve_forward_ref (line 38) | def _maybe_resolve_forward_ref( class UncheckedBaseModel (line 60) | class UncheckedBaseModel(UniversalBaseModel): class Config (line 65) | class Config: method model_construct (line 69) | def model_construct( method construct (line 80) | def construct( function _validate_collection_items_compatible (line 151) | def _validate_collection_items_compatible(collection: typing.Any, target... function _get_literal_field_value (line 177) | def _get_literal_field_value( function _literal_fields_match_strict (line 192) | def _literal_fields_match_strict(inner_type: typing.Type[typing.Any], ob... function _convert_undiscriminated_union_type (line 216) | def _convert_undiscriminated_union_type( function _convert_union_type (line 309) | def _convert_union_type( function construct_type (line 336) | def construct_type( function _get_is_populate_by_name (line 452) | def _get_is_populate_by_name(model: typing.Type["Model"]) -> bool: function _get_model_fields (line 465) | def _get_model_fields( function _get_field_default (line 474) | def _get_field_default(field: PydanticField) -> typing.Any: FILE: src/cohere/datasets/__init__.py function __getattr__ (line 18) | def __getattr__(attr_name: str) -> typing.Any: function __dir__ (line 34) | def __dir__(): FILE: src/cohere/datasets/client.py class DatasetsClient (line 21) | class DatasetsClient: method __init__ (line 22) | def __init__(self, *, client_wrapper: SyncClientWrapper): method with_raw_response (line 26) | def with_raw_response(self) -> RawDatasetsClient: method list (line 36) | def list( method create (line 112) | def create( method get_usage (line 202) | def get_usage(self, *, request_options: typing.Optional[RequestOptions... method get (line 229) | def get(self, id: str, *, request_options: typing.Optional[RequestOpti... method delete (line 260) | def delete( class AsyncDatasetsClient (line 294) | class AsyncDatasetsClient: method __init__ (line 295) | def __init__(self, *, client_wrapper: AsyncClientWrapper): method with_raw_response (line 299) | def with_raw_response(self) -> AsyncRawDatasetsClient: method list (line 309) | async def list( method create (line 392) | async def create( method get_usage (line 490) | async def get_usage(self, *, request_options: typing.Optional[RequestO... method get (line 525) | async def get(self, id: str, *, request_options: typing.Optional[Reque... method delete (line 564) | async def delete( FILE: src/cohere/datasets/raw_client.py class RawDatasetsClient (line 40) | class RawDatasetsClient: method __init__ (line 41) | def __init__(self, *, client_wrapper: SyncClientWrapper): method list (line 44) | def list( method create (line 250) | def create( method get_usage (line 481) | def get_usage( method get (line 653) | def get( method delete (line 827) | def delete( class AsyncRawDatasetsClient (line 1002) | class AsyncRawDatasetsClient: method __init__ (line 1003) | def __init__(self, *, client_wrapper: AsyncClientWrapper): method list (line 1006) | async def list( method create (line 1212) | async def create( method get_usage (line 1443) | async def get_usage( method get (line 1615) | async def get( method delete (line 1789) | async def delete( FILE: src/cohere/datasets/types/__init__.py function __getattr__ (line 21) | def __getattr__(attr_name: str) -> typing.Any: function __dir__ (line 37) | def __dir__(): FILE: src/cohere/datasets/types/datasets_create_response.py class DatasetsCreateResponse (line 10) | class DatasetsCreateResponse(UncheckedBaseModel): class Config (line 20) | class Config: FILE: src/cohere/datasets/types/datasets_get_response.py class DatasetsGetResponse (line 11) | class DatasetsGetResponse(UncheckedBaseModel): class Config (line 18) | class Config: FILE: src/cohere/datasets/types/datasets_get_usage_response.py class DatasetsGetUsageResponse (line 10) | class DatasetsGetUsageResponse(UncheckedBaseModel): class Config (line 20) | class Config: FILE: src/cohere/datasets/types/datasets_list_response.py class DatasetsListResponse (line 11) | class DatasetsListResponse(UncheckedBaseModel): class Config (line 18) | class Config: FILE: src/cohere/embed_jobs/__init__.py function __getattr__ (line 13) | def __getattr__(attr_name: str) -> typing.Any: function __dir__ (line 29) | def __dir__(): FILE: src/cohere/embed_jobs/client.py class EmbedJobsClient (line 19) | class EmbedJobsClient: method __init__ (line 20) | def __init__(self, *, client_wrapper: SyncClientWrapper): method with_raw_response (line 24) | def with_raw_response(self) -> RawEmbedJobsClient: method list (line 34) | def list(self, *, request_options: typing.Optional[RequestOptions] = N... method create (line 61) | def create( method get (line 142) | def get(self, id: str, *, request_options: typing.Optional[RequestOpti... method cancel (line 174) | def cancel(self, id: str, *, request_options: typing.Optional[RequestO... class AsyncEmbedJobsClient (line 206) | class AsyncEmbedJobsClient: method __init__ (line 207) | def __init__(self, *, client_wrapper: AsyncClientWrapper): method with_raw_response (line 211) | def with_raw_response(self) -> AsyncRawEmbedJobsClient: method list (line 221) | async def list(self, *, request_options: typing.Optional[RequestOption... method create (line 256) | async def create( method get (line 345) | async def get(self, id: str, *, request_options: typing.Optional[Reque... method cancel (line 385) | async def cancel(self, id: str, *, request_options: typing.Optional[Re... FILE: src/cohere/embed_jobs/raw_client.py class RawEmbedJobsClient (line 37) | class RawEmbedJobsClient: method __init__ (line 38) | def __init__(self, *, client_wrapper: SyncClientWrapper): method list (line 41) | def list(self, *, request_options: typing.Optional[RequestOptions] = N... method create (line 211) | def create( method get (line 435) | def get(self, id: str, *, request_options: typing.Optional[RequestOpti... method cancel (line 608) | def cancel(self, id: str, *, request_options: typing.Optional[RequestO... class AsyncRawEmbedJobsClient (line 774) | class AsyncRawEmbedJobsClient: method __init__ (line 775) | def __init__(self, *, client_wrapper: AsyncClientWrapper): method list (line 778) | async def list( method create (line 950) | async def create( method get (line 1174) | async def get( method cancel (line 1349) | async def cancel( FILE: src/cohere/embed_jobs/types/__init__.py function __getattr__ (line 13) | def __getattr__(attr_name: str) -> typing.Any: function __dir__ (line 29) | def __dir__(): FILE: src/cohere/environment.py class ClientEnvironment (line 6) | class ClientEnvironment(enum.Enum): FILE: src/cohere/errors/__init__.py function __getattr__ (line 37) | def __getattr__(attr_name: str) -> typing.Any: function __dir__ (line 53) | def __dir__(): FILE: src/cohere/errors/bad_request_error.py class BadRequestError (line 8) | class BadRequestError(ApiError): method __init__ (line 9) | def __init__(self, body: typing.Any, headers: typing.Optional[typing.D... FILE: src/cohere/errors/client_closed_request_error.py class ClientClosedRequestError (line 8) | class ClientClosedRequestError(ApiError): method __init__ (line 9) | def __init__(self, body: typing.Any, headers: typing.Optional[typing.D... FILE: src/cohere/errors/forbidden_error.py class ForbiddenError (line 8) | class ForbiddenError(ApiError): method __init__ (line 9) | def __init__(self, body: typing.Any, headers: typing.Optional[typing.D... FILE: src/cohere/errors/gateway_timeout_error.py class GatewayTimeoutError (line 8) | class GatewayTimeoutError(ApiError): method __init__ (line 9) | def __init__(self, body: typing.Any, headers: typing.Optional[typing.D... FILE: src/cohere/errors/internal_server_error.py class InternalServerError (line 8) | class InternalServerError(ApiError): method __init__ (line 9) | def __init__(self, body: typing.Any, headers: typing.Optional[typing.D... FILE: src/cohere/errors/invalid_token_error.py class InvalidTokenError (line 8) | class InvalidTokenError(ApiError): method __init__ (line 9) | def __init__(self, body: typing.Any, headers: typing.Optional[typing.D... FILE: src/cohere/errors/not_found_error.py class NotFoundError (line 8) | class NotFoundError(ApiError): method __init__ (line 9) | def __init__(self, body: typing.Any, headers: typing.Optional[typing.D... FILE: src/cohere/errors/not_implemented_error.py class NotImplementedError (line 8) | class NotImplementedError(ApiError): method __init__ (line 9) | def __init__(self, body: typing.Any, headers: typing.Optional[typing.D... FILE: src/cohere/errors/service_unavailable_error.py class ServiceUnavailableError (line 8) | class ServiceUnavailableError(ApiError): method __init__ (line 9) | def __init__(self, body: typing.Any, headers: typing.Optional[typing.D... FILE: src/cohere/errors/too_many_requests_error.py class TooManyRequestsError (line 8) | class TooManyRequestsError(ApiError): method __init__ (line 9) | def __init__(self, body: typing.Any, headers: typing.Optional[typing.D... FILE: src/cohere/errors/unauthorized_error.py class UnauthorizedError (line 8) | class UnauthorizedError(ApiError): method __init__ (line 9) | def __init__(self, body: typing.Any, headers: typing.Optional[typing.D... FILE: src/cohere/errors/unprocessable_entity_error.py class UnprocessableEntityError (line 8) | class UnprocessableEntityError(ApiError): method __init__ (line 9) | def __init__(self, body: typing.Any, headers: typing.Optional[typing.D... FILE: src/cohere/finetuning/__init__.py function __getattr__ (line 53) | def __getattr__(attr_name: str) -> typing.Any: function __dir__ (line 69) | def __dir__(): FILE: src/cohere/finetuning/client.py class FinetuningClient (line 22) | class FinetuningClient: method __init__ (line 23) | def __init__(self, *, client_wrapper: SyncClientWrapper): method with_raw_response (line 27) | def with_raw_response(self) -> RawFinetuningClient: method list_finetuned_models (line 37) | def list_finetuned_models( method create_finetuned_model (line 92) | def create_finetuned_model( method get_finetuned_model (line 134) | def get_finetuned_model( method delete_finetuned_model (line 168) | def delete_finetuned_model( method update_finetuned_model (line 203) | def update_finetuned_model( method list_events (line 253) | def list_events( method list_training_step_metrics (line 319) | def list_training_step_metrics( class AsyncFinetuningClient (line 372) | class AsyncFinetuningClient: method __init__ (line 373) | def __init__(self, *, client_wrapper: AsyncClientWrapper): method with_raw_response (line 377) | def with_raw_response(self) -> AsyncRawFinetuningClient: method list_finetuned_models (line 387) | async def list_finetuned_models( method create_finetuned_model (line 450) | async def create_finetuned_model( method get_finetuned_model (line 500) | async def get_finetuned_model( method delete_finetuned_model (line 542) | async def delete_finetuned_model( method update_finetuned_model (line 585) | async def update_finetuned_model( method list_events (line 643) | async def list_events( method list_training_step_metrics (line 717) | async def list_training_step_metrics( FILE: src/cohere/finetuning/finetuning/__init__.py function __getattr__ (line 51) | def __getattr__(attr_name: str) -> typing.Any: function __dir__ (line 67) | def __dir__(): FILE: src/cohere/finetuning/finetuning/types/__init__.py function __getattr__ (line 49) | def __getattr__(attr_name: str) -> typing.Any: function __dir__ (line 65) | def __dir__(): FILE: src/cohere/finetuning/finetuning/types/base_model.py class BaseModel (line 12) | class BaseModel(UncheckedBaseModel): class Config (line 41) | class Config: FILE: src/cohere/finetuning/finetuning/types/create_finetuned_model_response.py class CreateFinetunedModelResponse (line 11) | class CreateFinetunedModelResponse(UncheckedBaseModel): class Config (line 25) | class Config: FILE: src/cohere/finetuning/finetuning/types/event.py class Event (line 12) | class Event(UncheckedBaseModel): class Config (line 36) | class Config: FILE: src/cohere/finetuning/finetuning/types/finetuned_model.py class FinetunedModel (line 13) | class FinetunedModel(UncheckedBaseModel): class Config (line 72) | class Config: FILE: src/cohere/finetuning/finetuning/types/get_finetuned_model_response.py class GetFinetunedModelResponse (line 11) | class GetFinetunedModelResponse(UncheckedBaseModel): class Config (line 25) | class Config: FILE: src/cohere/finetuning/finetuning/types/hyperparameters.py class Hyperparameters (line 11) | class Hyperparameters(UncheckedBaseModel): class Config (line 64) | class Config: FILE: src/cohere/finetuning/finetuning/types/list_events_response.py class ListEventsResponse (line 11) | class ListEventsResponse(UncheckedBaseModel): class Config (line 36) | class Config: FILE: src/cohere/finetuning/finetuning/types/list_finetuned_models_response.py class ListFinetunedModelsResponse (line 11) | class ListFinetunedModelsResponse(UncheckedBaseModel): class Config (line 36) | class Config: FILE: src/cohere/finetuning/finetuning/types/list_training_step_metrics_response.py class ListTrainingStepMetricsResponse (line 11) | class ListTrainingStepMetricsResponse(UncheckedBaseModel): class Config (line 31) | class Config: FILE: src/cohere/finetuning/finetuning/types/settings.py class Settings (line 13) | class Settings(UncheckedBaseModel): class Config (line 47) | class Config: FILE: src/cohere/finetuning/finetuning/types/training_step_metrics.py class TrainingStepMetrics (line 11) | class TrainingStepMetrics(UncheckedBaseModel): class Config (line 35) | class Config: FILE: src/cohere/finetuning/finetuning/types/update_finetuned_model_response.py class UpdateFinetunedModelResponse (line 11) | class UpdateFinetunedModelResponse(UncheckedBaseModel): class Config (line 25) | class Config: FILE: src/cohere/finetuning/finetuning/types/wandb_config.py class WandbConfig (line 10) | class WandbConfig(UncheckedBaseModel): class Config (line 34) | class Config: FILE: src/cohere/finetuning/raw_client.py class RawFinetuningClient (line 35) | class RawFinetuningClient: method __init__ (line 36) | def __init__(self, *, client_wrapper: SyncClientWrapper): method list_finetuned_models (line 39) | def list_finetuned_models( method create_finetuned_model (line 170) | def create_finetuned_model( method get_finetuned_model (line 283) | def get_finetuned_model( method delete_finetuned_model (line 392) | def delete_finetuned_model( method update_finetuned_model (line 502) | def update_finetuned_model( method list_events (line 627) | def list_events( method list_training_step_metrics (line 764) | def list_training_step_metrics( class AsyncRawFinetuningClient (line 892) | class AsyncRawFinetuningClient: method __init__ (line 893) | def __init__(self, *, client_wrapper: AsyncClientWrapper): method list_finetuned_models (line 896) | async def list_finetuned_models( method create_finetuned_model (line 1027) | async def create_finetuned_model( method get_finetuned_model (line 1140) | async def get_finetuned_model( method delete_finetuned_model (line 1249) | async def delete_finetuned_model( method update_finetuned_model (line 1359) | async def update_finetuned_model( method list_events (line 1484) | async def list_events( method list_training_step_metrics (line 1621) | async def list_training_step_metrics( FILE: src/cohere/manually_maintained/cache.py class CacheMixin (line 5) | class CacheMixin: method _cache_get (line 9) | def _cache_get(self, key: str) -> typing.Any: method _cache_set (line 19) | def _cache_set(self, key: str, value: typing.Any, ttl: int = 60 * 60) ... FILE: src/cohere/manually_maintained/cohere_aws/chat.py class ToolParameterDefinitionsValue (line 10) | class ToolParameterDefinitionsValue(CohereObject, dict): method __init__ (line 11) | def __init__( class Tool (line 26) | class Tool(CohereObject, dict): method __init__ (line 27) | def __init__( class ToolCall (line 42) | class ToolCall(CohereObject, dict): method __init__ (line 43) | def __init__( method from_dict (line 57) | def from_dict(cls, tool_call_res: Dict[str, Any]) -> "ToolCall": method from_list (line 65) | def from_list(cls, tool_calls_res: Optional[List[Dict[str, Any]]]) -> ... class Chat (line 73) | class Chat(CohereObject): method __init__ (line 74) | def __init__( method from_dict (line 105) | def from_dict(cls, response: Dict[str, Any]) -> "Chat": class StreamEvent (line 126) | class StreamEvent(str, Enum): class StreamResponse (line 135) | class StreamResponse(CohereObject): method __init__ (line 136) | def __init__( class StreamStart (line 149) | class StreamStart(StreamResponse): method __init__ (line 150) | def __init__( class StreamTextGeneration (line 161) | class StreamTextGeneration(StreamResponse): method __init__ (line 162) | def __init__( class StreamCitationGeneration (line 171) | class StreamCitationGeneration(StreamResponse): method __init__ (line 172) | def __init__( class StreamQueryGeneration (line 181) | class StreamQueryGeneration(StreamResponse): method __init__ (line 182) | def __init__( class StreamSearchResults (line 191) | class StreamSearchResults(StreamResponse): method __init__ (line 192) | def __init__( class StreamEnd (line 203) | class StreamEnd(StreamResponse): method __init__ (line 204) | def __init__( class ChatToolCallsGenerationEvent (line 213) | class ChatToolCallsGenerationEvent(StreamResponse): method __init__ (line 214) | def __init__( class StreamingChat (line 222) | class StreamingChat(CohereObject): method __init__ (line 223) | def __init__(self, stream_response, mode): method _make_response_item (line 248) | def _make_response_item(self, index, streaming_item) -> Any: method __iter__ (line 312) | def __iter__(self) -> Generator[StreamResponse, None, None]: FILE: src/cohere/manually_maintained/cohere_aws/classification.py class Classification (line 8) | class Classification(CohereObject): method __init__ (line 9) | def __init__(self, classification: Union[Prediction, ClassificationDic... method is_multilabel (line 15) | def is_multilabel(self) -> bool: method prediction (line 23) | def prediction(self) -> Prediction: method confidence (line 29) | def confidence(self) -> List[float]: method text (line 37) | def text(self) -> str: class Classifications (line 45) | class Classifications(CohereObject): method __init__ (line 46) | def __init__(self, classifications: List[Classification]) -> None: method __iter__ (line 53) | def __iter__(self) -> Iterator: method __len__ (line 56) | def __len__(self) -> int: method is_multilabel (line 59) | def is_multilabel(self) -> bool: FILE: src/cohere/manually_maintained/cohere_aws/client.py class Client (line 19) | class Client: method __init__ (line 20) | def __init__( method _require_sagemaker (line 43) | def _require_sagemaker(self) -> None: method _does_endpoint_exist (line 47) | def _does_endpoint_exist(self, endpoint_name: str) -> bool: method connect_to_endpoint (line 54) | def connect_to_endpoint(self, endpoint_name: str) -> None: method _s3_models_dir_to_tarfile (line 68) | def _s3_models_dir_to_tarfile(self, s3_models_dir: str) -> str: method create_endpoint (line 127) | def create_endpoint( method chat (line 262) | def chat( method _sagemaker_chat (line 401) | def _sagemaker_chat(self, json_params: Dict[str, Any], variant: str) : method _bedrock_chat (line 426) | def _bedrock_chat(self, json_params: Dict[str, Any], model_id: str) : method generate (line 458) | def generate( method _sagemaker_generations (line 507) | def _sagemaker_generations(self, json_params: Dict[str, Any], variant:... method _bedrock_generations (line 533) | def _bedrock_generations(self, json_params: Dict[str, Any], model_id: ... method embed (line 558) | def embed( method _sagemaker_embed (line 586) | def _sagemaker_embed(self, json_params: Dict[str, Any], variant: str): method _bedrock_embed (line 615) | def _bedrock_embed(self, json_params: Dict[str, Any], model_id: str): method rerank (line 640) | def rerank(self, method classify (line 703) | def classify(self, input: List[str], name: str) -> Classifications: method create_finetune (line 730) | def create_finetune( method export_finetune (line 808) | def export_finetune( method wait_for_finetune_job (line 875) | def wait_for_finetune_job(self, job_id: str, timeout: int = 2*60*60) -... method provision_throughput (line 895) | def provision_throughput( method _bedrock_create_finetune (line 923) | def _bedrock_create_finetune( method summarize (line 962) | def summarize( method delete_endpoint (line 1015) | def delete_endpoint(self) -> None: method close (line 1029) | def close(self) -> None: FILE: src/cohere/manually_maintained/cohere_aws/embeddings.py class Embedding (line 5) | class Embedding(CohereObject): method __init__ (line 7) | def __init__(self, embedding: List[float]) -> None: method __iter__ (line 10) | def __iter__(self) -> Iterator: method __len__ (line 13) | def __len__(self) -> int: class Embeddings (line 17) | class Embeddings(CohereObject): method __init__ (line 19) | def __init__(self, embeddings: List[Embedding]) -> None: method __iter__ (line 22) | def __iter__(self) -> Iterator: method __len__ (line 25) | def __len__(self) -> int: FILE: src/cohere/manually_maintained/cohere_aws/error.py class CohereError (line 1) | class CohereError(Exception): method __init__ (line 2) | def __init__( method __str__ (line 14) | def __str__(self) -> str: method __repr__ (line 18) | def __repr__(self) -> str: FILE: src/cohere/manually_maintained/cohere_aws/generation.py class TokenLikelihood (line 7) | class TokenLikelihood(CohereObject): method __init__ (line 8) | def __init__(self, token: str, likelihood: float) -> None: class Generation (line 13) | class Generation(CohereObject): method __init__ (line 14) | def __init__(self, class Generations (line 21) | class Generations(CohereObject): method __init__ (line 22) | def __init__(self, method from_dict (line 28) | def from_dict(cls, response: Dict[str, Any]) -> List[Generation]: method __iter__ (line 45) | def __iter__(self) -> iter: method __next__ (line 48) | def __next__(self) -> next: class StreamingGenerations (line 58) | class StreamingGenerations(CohereObject): method __init__ (line 59) | def __init__(self, stream, mode): method _make_response_item (line 75) | def _make_response_item(self, streaming_item) -> Optional[StreamingText]: method __iter__ (line 96) | def __iter__(self) -> Generator[StreamingText, None, None]: FILE: src/cohere/manually_maintained/cohere_aws/mode.py class Mode (line 4) | class Mode(Enum): FILE: src/cohere/manually_maintained/cohere_aws/rerank.py class RerankResult (line 12) | class RerankResult(CohereObject): method __init__ (line 14) | def __init__(self, method __repr__ (line 24) | def __repr__(self) -> str: class Reranking (line 36) | class Reranking(CohereObject): method __init__ (line 38) | def __init__(self, method _results (line 45) | def _results(self, response: Dict[str, Any]) -> List[RerankResult]: method __str__ (line 56) | def __str__(self) -> str: method __repr__ (line 59) | def __repr__(self) -> str: method __iter__ (line 62) | def __iter__(self) -> Iterator: method __getitem__ (line 65) | def __getitem__(self, index) -> RerankResult: FILE: src/cohere/manually_maintained/cohere_aws/response.py class CohereObject (line 1) | class CohereObject(): method __repr__ (line 2) | def __repr__(self) -> str: FILE: src/cohere/manually_maintained/cohere_aws/summary.py class Summary (line 6) | class Summary(CohereObject): method __init__ (line 7) | def __init__(self, method __str__ (line 15) | def __str__(self) -> str: FILE: src/cohere/manually_maintained/lazy_aws_deps.py function lazy_sagemaker (line 4) | def lazy_sagemaker(): function lazy_boto3 (line 11) | def lazy_boto3(): function lazy_botocore (line 18) | def lazy_botocore(): FILE: src/cohere/manually_maintained/lazy_oci_deps.py function lazy_oci (line 16) | def lazy_oci() -> Any: FILE: src/cohere/manually_maintained/streaming_embed.py class StreamedEmbedding (line 10) | class StreamedEmbedding: function extract_embeddings_from_response (line 18) | def extract_embeddings_from_response( FILE: src/cohere/manually_maintained/tokenizers.py function tokenizer_cache_key (line 15) | def tokenizer_cache_key(model: str) -> str: function get_hf_tokenizer (line 19) | def get_hf_tokenizer(co: "Client", model: str) -> Tokenizer: function local_tokenize (line 43) | def local_tokenize(co: "Client", model: str, text: str) -> typing.List[i... function local_detokenize (line 49) | def local_detokenize(co: "Client", model: str, tokens: typing.Sequence[i... function async_get_hf_tokenizer (line 55) | async def async_get_hf_tokenizer(co: "AsyncClient", model: str) -> Token... function async_local_tokenize (line 80) | async def async_local_tokenize(co: "AsyncClient", model: str, text: str)... function async_local_detokenize (line 86) | async def async_local_detokenize(co: "AsyncClient", model: str, tokens: ... function _get_tokenizer_config_size (line 92) | def _get_tokenizer_config_size(tokenizer_url: str) -> float: FILE: src/cohere/models/client.py class ModelsClient (line 13) | class ModelsClient: method __init__ (line 14) | def __init__(self, *, client_wrapper: SyncClientWrapper): method with_raw_response (line 18) | def with_raw_response(self) -> RawModelsClient: method get (line 28) | def get(self, model: str, *, request_options: typing.Optional[RequestO... method list (line 59) | def list( class AsyncModelsClient (line 119) | class AsyncModelsClient: method __init__ (line 120) | def __init__(self, *, client_wrapper: AsyncClientWrapper): method with_raw_response (line 124) | def with_raw_response(self) -> AsyncRawModelsClient: method get (line 134) | async def get(self, model: str, *, request_options: typing.Optional[Re... method list (line 173) | async def list( FILE: src/cohere/models/raw_client.py class RawModelsClient (line 31) | class RawModelsClient: method __init__ (line 32) | def __init__(self, *, client_wrapper: SyncClientWrapper): method get (line 35) | def get( method list (line 209) | def list( class AsyncRawModelsClient (line 407) | class AsyncRawModelsClient: method __init__ (line 408) | def __init__(self, *, client_wrapper: AsyncClientWrapper): method get (line 411) | async def get( method list (line 585) | async def list( FILE: src/cohere/oci_client.py class OciClient (line 19) | class OciClient(Client): method __init__ (line 55) | def __init__( class OciClientV2 (line 105) | class OciClientV2(ClientV2): method __init__ (line 169) | def __init__( function _load_oci_config (line 230) | def _load_oci_config( function _remove_inherited_session_auth (line 291) | def _remove_inherited_session_auth( function _usage_from_oci (line 329) | def _usage_from_oci(usage_data: typing.Optional[typing.Dict[str, typing.... function get_event_hooks (line 346) | def get_event_hooks( function map_request_to_oci (line 377) | def map_request_to_oci( function map_response_from_oci (line 511) | def map_response_from_oci() -> EventHook: function get_oci_url (line 559) | def get_oci_url( function normalize_model_for_oci (line 590) | def normalize_model_for_oci(model: str) -> str: function transform_request_to_oci (line 625) | def transform_request_to_oci( function transform_oci_response_to_cohere (line 867) | def transform_oci_response_to_cohere( function transform_oci_stream_wrapper (line 981) | def transform_oci_stream_wrapper( function transform_stream_event (line 1164) | def transform_stream_event( FILE: src/cohere/overrides.py function get_fields (line 10) | def get_fields(obj) -> typing.List[str]: function get_aliases_or_field (line 14) | def get_aliases_or_field(obj) -> typing.List[str]: function get_aliases_and_fields (line 22) | def get_aliases_and_fields(obj): function allow_access_to_aliases (line 27) | def allow_access_to_aliases(self: typing.Type["Model"], name): function make_tool_call_v2_id_optional (line 37) | def make_tool_call_v2_id_optional(cls): function run_overrides (line 62) | def run_overrides(): FILE: src/cohere/raw_base_client.py class RawBaseCohere (line 69) | class RawBaseCohere: method __init__ (line 70) | def __init__(self, *, client_wrapper: SyncClientWrapper): method chat_stream (line 74) | def chat_stream( method chat (line 548) | def chat( method generate_stream (line 1002) | def generate_stream( method generate (line 1314) | def generate( method embed (line 1606) | def embed( method rerank (line 1833) | def rerank( method classify (line 2056) | def classify( method summarize (line 2269) | def summarize( method tokenize (line 2487) | def tokenize( method detokenize (line 2673) | def detokenize( method check_api_key (line 2859) | def check_api_key( class AsyncRawBaseCohere (line 3032) | class AsyncRawBaseCohere: method __init__ (line 3033) | def __init__(self, *, client_wrapper: AsyncClientWrapper): method chat_stream (line 3037) | async def chat_stream( method chat (line 3511) | async def chat( method generate_stream (line 3965) | async def generate_stream( method generate (line 4277) | async def generate( method embed (line 4569) | async def embed( method rerank (line 4796) | async def rerank( method classify (line 5019) | async def classify( method summarize (line 5232) | async def summarize( method tokenize (line 5450) | async def tokenize( method detokenize (line 5636) | async def detokenize( method check_api_key (line 5822) | async def check_api_key( FILE: src/cohere/sagemaker_client.py class SagemakerClient (line 8) | class SagemakerClient(AwsClient): method __init__ (line 11) | def __init__( class SagemakerClientV2 (line 35) | class SagemakerClientV2(AwsClientV2): method __init__ (line 38) | def __init__( FILE: src/cohere/types/__init__.py function __getattr__ (line 478) | def __getattr__(attr_name: str) -> typing.Any: function __dir__ (line 494) | def __dir__(): FILE: src/cohere/types/api_meta.py class ApiMeta (line 13) | class ApiMeta(UncheckedBaseModel): class Config (line 28) | class Config: FILE: src/cohere/types/api_meta_api_version.py class ApiMetaApiVersion (line 10) | class ApiMetaApiVersion(UncheckedBaseModel): class Config (line 19) | class Config: FILE: src/cohere/types/api_meta_billed_units.py class ApiMetaBilledUnits (line 10) | class ApiMetaBilledUnits(UncheckedBaseModel): class Config (line 45) | class Config: FILE: src/cohere/types/api_meta_tokens.py class ApiMetaTokens (line 10) | class ApiMetaTokens(UncheckedBaseModel): class Config (line 25) | class Config: FILE: src/cohere/types/assistant_message.py class AssistantMessage (line 13) | class AssistantMessage(UncheckedBaseModel): class Config (line 31) | class Config: FILE: src/cohere/types/assistant_message_response.py class AssistantMessageResponse (line 13) | class AssistantMessageResponse(UncheckedBaseModel): class Config (line 32) | class Config: FILE: src/cohere/types/assistant_message_response_content_item.py class TextAssistantMessageResponseContentItem (line 13) | class TextAssistantMessageResponseContentItem(UncheckedBaseModel): class Config (line 21) | class Config: class ThinkingAssistantMessageResponseContentItem (line 26) | class ThinkingAssistantMessageResponseContentItem(UncheckedBaseModel): class Config (line 34) | class Config: FILE: src/cohere/types/assistant_message_v2content_one_item.py class TextAssistantMessageV2ContentOneItem (line 13) | class TextAssistantMessageV2ContentOneItem(UncheckedBaseModel): class Config (line 21) | class Config: class ThinkingAssistantMessageV2ContentOneItem (line 26) | class ThinkingAssistantMessageV2ContentOneItem(UncheckedBaseModel): class Config (line 34) | class Config: FILE: src/cohere/types/chat_citation.py class ChatCitation (line 11) | class ChatCitation(UncheckedBaseModel): class Config (line 45) | class Config: FILE: src/cohere/types/chat_citation_generation_event.py class ChatCitationGenerationEvent (line 11) | class ChatCitationGenerationEvent(ChatStreamEvent): class Config (line 21) | class Config: FILE: src/cohere/types/chat_connector.py class ChatConnector (line 10) | class ChatConnector(UncheckedBaseModel): class Config (line 43) | class Config: FILE: src/cohere/types/chat_content_delta_event.py class ChatContentDeltaEvent (line 12) | class ChatContentDeltaEvent(ChatStreamEventType): class Config (line 25) | class Config: FILE: src/cohere/types/chat_content_delta_event_delta.py class ChatContentDeltaEventDelta (line 11) | class ChatContentDeltaEventDelta(UncheckedBaseModel): class Config (line 18) | class Config: FILE: src/cohere/types/chat_content_delta_event_delta_message.py class ChatContentDeltaEventDeltaMessage (line 11) | class ChatContentDeltaEventDeltaMessage(UncheckedBaseModel): class Config (line 18) | class Config: FILE: src/cohere/types/chat_content_delta_event_delta_message_content.py class ChatContentDeltaEventDeltaMessageContent (line 10) | class ChatContentDeltaEventDeltaMessageContent(UncheckedBaseModel): class Config (line 18) | class Config: FILE: src/cohere/types/chat_content_end_event.py class ChatContentEndEvent (line 10) | class ChatContentEndEvent(ChatStreamEventType): class Config (line 21) | class Config: FILE: src/cohere/types/chat_content_start_event.py class ChatContentStartEvent (line 11) | class ChatContentStartEvent(ChatStreamEventType): class Config (line 23) | class Config: FILE: src/cohere/types/chat_content_start_event_delta.py class ChatContentStartEventDelta (line 11) | class ChatContentStartEventDelta(UncheckedBaseModel): class Config (line 18) | class Config: FILE: src/cohere/types/chat_content_start_event_delta_message.py class ChatContentStartEventDeltaMessage (line 11) | class ChatContentStartEventDeltaMessage(UncheckedBaseModel): class Config (line 18) | class Config: FILE: src/cohere/types/chat_content_start_event_delta_message_content.py class ChatContentStartEventDeltaMessageContent (line 11) | class ChatContentStartEventDeltaMessageContent(UncheckedBaseModel): class Config (line 20) | class Config: FILE: src/cohere/types/chat_data_metrics.py class ChatDataMetrics (line 10) | class ChatDataMetrics(UncheckedBaseModel): class Config (line 30) | class Config: FILE: src/cohere/types/chat_debug_event.py class ChatDebugEvent (line 10) | class ChatDebugEvent(ChatStreamEvent): class Config (line 17) | class Config: FILE: src/cohere/types/chat_document_source.py class ChatDocumentSource (line 10) | class ChatDocumentSource(UncheckedBaseModel): class Config (line 26) | class Config: FILE: src/cohere/types/chat_message.py class ChatMessage (line 11) | class ChatMessage(UncheckedBaseModel): class Config (line 29) | class Config: FILE: src/cohere/types/chat_message_end_event.py class ChatMessageEndEvent (line 11) | class ChatMessageEndEvent(ChatStreamEventType): class Config (line 23) | class Config: FILE: src/cohere/types/chat_message_end_event_delta.py class ChatMessageEndEventDelta (line 12) | class ChatMessageEndEventDelta(UncheckedBaseModel): class Config (line 25) | class Config: FILE: src/cohere/types/chat_message_start_event.py class ChatMessageStartEvent (line 11) | class ChatMessageStartEvent(ChatStreamEventType): class Config (line 27) | class Config: FILE: src/cohere/types/chat_message_start_event_delta.py class ChatMessageStartEventDelta (line 11) | class ChatMessageStartEventDelta(UncheckedBaseModel): class Config (line 18) | class Config: FILE: src/cohere/types/chat_message_start_event_delta_message.py class ChatMessageStartEventDeltaMessage (line 10) | class ChatMessageStartEventDeltaMessage(UncheckedBaseModel): class Config (line 20) | class Config: FILE: src/cohere/types/chat_message_v2.py class UserChatMessageV2 (line 19) | class UserChatMessageV2(UncheckedBaseModel): class Config (line 31) | class Config: class AssistantChatMessageV2 (line 36) | class AssistantChatMessageV2(UncheckedBaseModel): class Config (line 51) | class Config: class SystemChatMessageV2 (line 56) | class SystemChatMessageV2(UncheckedBaseModel): class Config (line 68) | class Config: class ToolChatMessageV2 (line 73) | class ToolChatMessageV2(UncheckedBaseModel): class Config (line 86) | class Config: FILE: src/cohere/types/chat_search_queries_generation_event.py class ChatSearchQueriesGenerationEvent (line 11) | class ChatSearchQueriesGenerationEvent(ChatStreamEvent): class Config (line 21) | class Config: FILE: src/cohere/types/chat_search_query.py class ChatSearchQuery (line 10) | class ChatSearchQuery(UncheckedBaseModel): class Config (line 29) | class Config: FILE: src/cohere/types/chat_search_result.py class ChatSearchResult (line 12) | class ChatSearchResult(UncheckedBaseModel): class Config (line 38) | class Config: FILE: src/cohere/types/chat_search_result_connector.py class ChatSearchResultConnector (line 10) | class ChatSearchResultConnector(UncheckedBaseModel): class Config (line 24) | class Config: FILE: src/cohere/types/chat_search_results_event.py class ChatSearchResultsEvent (line 12) | class ChatSearchResultsEvent(ChatStreamEvent): class Config (line 27) | class Config: FILE: src/cohere/types/chat_stream_end_event.py class ChatStreamEndEvent (line 12) | class ChatStreamEndEvent(ChatStreamEvent): class Config (line 31) | class Config: FILE: src/cohere/types/chat_stream_event.py class ChatStreamEvent (line 10) | class ChatStreamEvent(UncheckedBaseModel): class Config (line 15) | class Config: FILE: src/cohere/types/chat_stream_event_type.py class ChatStreamEventType (line 10) | class ChatStreamEventType(UncheckedBaseModel): class Config (line 19) | class Config: FILE: src/cohere/types/chat_stream_start_event.py class ChatStreamStartEvent (line 10) | class ChatStreamStartEvent(ChatStreamEvent): class Config (line 20) | class Config: FILE: src/cohere/types/chat_text_content.py class ChatTextContent (line 10) | class ChatTextContent(UncheckedBaseModel): class Config (line 21) | class Config: FILE: src/cohere/types/chat_text_generation_event.py class ChatTextGenerationEvent (line 10) | class ChatTextGenerationEvent(ChatStreamEvent): class Config (line 20) | class Config: FILE: src/cohere/types/chat_text_response_format.py class ChatTextResponseFormat (line 10) | class ChatTextResponseFormat(UncheckedBaseModel): class Config (line 15) | class Config: FILE: src/cohere/types/chat_text_response_format_v2.py class ChatTextResponseFormatV2 (line 10) | class ChatTextResponseFormatV2(UncheckedBaseModel): class Config (line 15) | class Config: FILE: src/cohere/types/chat_thinking_content.py class ChatThinkingContent (line 10) | class ChatThinkingContent(UncheckedBaseModel): class Config (line 21) | class Config: FILE: src/cohere/types/chat_tool_call_delta_event.py class ChatToolCallDeltaEvent (line 11) | class ChatToolCallDeltaEvent(ChatStreamEventType): class Config (line 23) | class Config: FILE: src/cohere/types/chat_tool_call_delta_event_delta.py class ChatToolCallDeltaEventDelta (line 11) | class ChatToolCallDeltaEventDelta(UncheckedBaseModel): class Config (line 18) | class Config: FILE: src/cohere/types/chat_tool_call_delta_event_delta_message.py class ChatToolCallDeltaEventDeltaMessage (line 11) | class ChatToolCallDeltaEventDeltaMessage(UncheckedBaseModel): class Config (line 18) | class Config: FILE: src/cohere/types/chat_tool_call_delta_event_delta_message_tool_calls.py class ChatToolCallDeltaEventDeltaMessageToolCalls (line 13) | class ChatToolCallDeltaEventDeltaMessageToolCalls(UncheckedBaseModel): class Config (line 20) | class Config: FILE: src/cohere/types/chat_tool_call_delta_event_delta_message_tool_calls_function.py class ChatToolCallDeltaEventDeltaMessageToolCallsFunction (line 10) | class ChatToolCallDeltaEventDeltaMessageToolCallsFunction(UncheckedBaseM... class Config (line 17) | class Config: FILE: src/cohere/types/chat_tool_call_end_event.py class ChatToolCallEndEvent (line 10) | class ChatToolCallEndEvent(ChatStreamEventType): class Config (line 21) | class Config: FILE: src/cohere/types/chat_tool_call_start_event.py class ChatToolCallStartEvent (line 11) | class ChatToolCallStartEvent(ChatStreamEventType): class Config (line 23) | class Config: FILE: src/cohere/types/chat_tool_call_start_event_delta.py class ChatToolCallStartEventDelta (line 11) | class ChatToolCallStartEventDelta(UncheckedBaseModel): class Config (line 18) | class Config: FILE: src/cohere/types/chat_tool_call_start_event_delta_message.py class ChatToolCallStartEventDeltaMessage (line 11) | class ChatToolCallStartEventDeltaMessage(UncheckedBaseModel): class Config (line 18) | class Config: FILE: src/cohere/types/chat_tool_calls_chunk_event.py class ChatToolCallsChunkEvent (line 11) | class ChatToolCallsChunkEvent(ChatStreamEvent): class Config (line 19) | class Config: FILE: src/cohere/types/chat_tool_calls_generation_event.py class ChatToolCallsGenerationEvent (line 11) | class ChatToolCallsGenerationEvent(ChatStreamEvent): class Config (line 23) | class Config: FILE: src/cohere/types/chat_tool_message.py class ChatToolMessage (line 11) | class ChatToolMessage(UncheckedBaseModel): class Config (line 22) | class Config: FILE: src/cohere/types/chat_tool_plan_delta_event.py class ChatToolPlanDeltaEvent (line 11) | class ChatToolPlanDeltaEvent(ChatStreamEventType): class Config (line 22) | class Config: FILE: src/cohere/types/chat_tool_plan_delta_event_delta.py class ChatToolPlanDeltaEventDelta (line 11) | class ChatToolPlanDeltaEventDelta(UncheckedBaseModel): class Config (line 18) | class Config: FILE: src/cohere/types/chat_tool_plan_delta_event_delta_message.py class ChatToolPlanDeltaEventDeltaMessage (line 10) | class ChatToolPlanDeltaEventDeltaMessage(UncheckedBaseModel): class Config (line 17) | class Config: FILE: src/cohere/types/chat_tool_source.py class ChatToolSource (line 10) | class ChatToolSource(UncheckedBaseModel): class Config (line 22) | class Config: FILE: src/cohere/types/check_api_key_response.py class CheckApiKeyResponse (line 10) | class CheckApiKeyResponse(UncheckedBaseModel): class Config (line 19) | class Config: FILE: src/cohere/types/citation.py class Citation (line 12) | class Citation(UncheckedBaseModel): class Config (line 44) | class Config: FILE: src/cohere/types/citation_end_event.py class CitationEndEvent (line 10) | class CitationEndEvent(ChatStreamEventType): class Config (line 21) | class Config: FILE: src/cohere/types/citation_options.py class CitationOptions (line 11) | class CitationOptions(UncheckedBaseModel): class Config (line 26) | class Config: FILE: src/cohere/types/citation_start_event.py class CitationStartEvent (line 11) | class CitationStartEvent(ChatStreamEventType): class Config (line 23) | class Config: FILE: src/cohere/types/citation_start_event_delta.py class CitationStartEventDelta (line 11) | class CitationStartEventDelta(UncheckedBaseModel): class Config (line 18) | class Config: FILE: src/cohere/types/citation_start_event_delta_message.py class CitationStartEventDeltaMessage (line 11) | class CitationStartEventDeltaMessage(UncheckedBaseModel): class Config (line 18) | class Config: FILE: src/cohere/types/classify_data_metrics.py class ClassifyDataMetrics (line 11) | class ClassifyDataMetrics(UncheckedBaseModel): class Config (line 18) | class Config: FILE: src/cohere/types/classify_example.py class ClassifyExample (line 10) | class ClassifyExample(UncheckedBaseModel): class Config (line 18) | class Config: FILE: src/cohere/types/classify_response.py class ClassifyResponse (line 12) | class ClassifyResponse(UncheckedBaseModel): class Config (line 21) | class Config: FILE: src/cohere/types/classify_response_classifications_item.py class ClassifyResponseClassificationsItem (line 14) | class ClassifyResponseClassificationsItem(UncheckedBaseModel): class Config (line 55) | class Config: FILE: src/cohere/types/classify_response_classifications_item_labels_value.py class ClassifyResponseClassificationsItemLabelsValue (line 10) | class ClassifyResponseClassificationsItemLabelsValue(UncheckedBaseModel): class Config (line 17) | class Config: FILE: src/cohere/types/connector.py class Connector (line 13) | class Connector(UncheckedBaseModel): class Config (line 90) | class Config: FILE: src/cohere/types/connector_o_auth.py class ConnectorOAuth (line 10) | class ConnectorOAuth(UncheckedBaseModel): class Config (line 40) | class Config: FILE: src/cohere/types/content.py class TextContent (line 14) | class TextContent(UncheckedBaseModel): class Config (line 26) | class Config: class ImageUrlContent (line 31) | class ImageUrlContent(UncheckedBaseModel): class Config (line 43) | class Config: FILE: src/cohere/types/create_connector_o_auth.py class CreateConnectorOAuth (line 10) | class CreateConnectorOAuth(UncheckedBaseModel): class Config (line 40) | class Config: FILE: src/cohere/types/create_connector_response.py class CreateConnectorResponse (line 11) | class CreateConnectorResponse(UncheckedBaseModel): class Config (line 18) | class Config: FILE: src/cohere/types/create_connector_service_auth.py class CreateConnectorServiceAuth (line 11) | class CreateConnectorServiceAuth(UncheckedBaseModel): class Config (line 22) | class Config: FILE: src/cohere/types/create_embed_job_response.py class CreateEmbedJobResponse (line 11) | class CreateEmbedJobResponse(UncheckedBaseModel): class Config (line 23) | class Config: FILE: src/cohere/types/dataset.py class Dataset (line 16) | class Dataset(UncheckedBaseModel): class Config (line 65) | class Config: FILE: src/cohere/types/dataset_part.py class DatasetPart (line 10) | class DatasetPart(UncheckedBaseModel): class Config (line 55) | class Config: FILE: src/cohere/types/detokenize_response.py class DetokenizeResponse (line 11) | class DetokenizeResponse(UncheckedBaseModel): class Config (line 23) | class Config: FILE: src/cohere/types/document.py class Document (line 10) | class Document(UncheckedBaseModel): class Config (line 31) | class Config: FILE: src/cohere/types/document_content.py class DocumentContent (line 11) | class DocumentContent(UncheckedBaseModel): class Config (line 22) | class Config: FILE: src/cohere/types/embed_by_type_response.py class EmbedByTypeResponse (line 14) | class EmbedByTypeResponse(UncheckedBaseModel): class Config (line 38) | class Config: FILE: src/cohere/types/embed_by_type_response_embeddings.py class EmbedByTypeResponseEmbeddings (line 12) | class EmbedByTypeResponseEmbeddings(UncheckedBaseModel): class Config (line 51) | class Config: FILE: src/cohere/types/embed_content.py class ImageUrlEmbedContent (line 14) | class ImageUrlEmbedContent(UncheckedBaseModel): class Config (line 22) | class Config: class TextEmbedContent (line 27) | class TextEmbedContent(UncheckedBaseModel): class Config (line 35) | class Config: FILE: src/cohere/types/embed_floats_response.py class EmbedFloatsResponse (line 12) | class EmbedFloatsResponse(UncheckedBaseModel): class Config (line 35) | class Config: FILE: src/cohere/types/embed_image.py class EmbedImage (line 11) | class EmbedImage(UncheckedBaseModel): class Config (line 22) | class Config: FILE: src/cohere/types/embed_image_url.py class EmbedImageUrl (line 10) | class EmbedImageUrl(UncheckedBaseModel): class Config (line 21) | class Config: FILE: src/cohere/types/embed_input.py class EmbedInput (line 11) | class EmbedInput(UncheckedBaseModel): class Config (line 21) | class Config: FILE: src/cohere/types/embed_job.py class EmbedJob (line 14) | class EmbedJob(UncheckedBaseModel): class Config (line 61) | class Config: FILE: src/cohere/types/embed_response.py class EmbeddingsFloatsEmbedResponse (line 16) | class EmbeddingsFloatsEmbedResponse(UncheckedBaseModel): class Config (line 28) | class Config: class EmbeddingsByTypeEmbedResponse (line 33) | class EmbeddingsByTypeEmbedResponse(UncheckedBaseModel): class Config (line 45) | class Config: FILE: src/cohere/types/embed_text.py class EmbedText (line 10) | class EmbedText(UncheckedBaseModel): class Config (line 21) | class Config: FILE: src/cohere/types/finetune_dataset_metrics.py class FinetuneDatasetMetrics (line 10) | class FinetuneDatasetMetrics(UncheckedBaseModel): class Config (line 45) | class Config: FILE: src/cohere/types/generate_stream_end.py class GenerateStreamEnd (line 12) | class GenerateStreamEnd(GenerateStreamEvent): class Config (line 21) | class Config: FILE: src/cohere/types/generate_stream_end_response.py class GenerateStreamEndResponse (line 11) | class GenerateStreamEndResponse(UncheckedBaseModel): class Config (line 20) | class Config: FILE: src/cohere/types/generate_stream_error.py class GenerateStreamError (line 11) | class GenerateStreamError(GenerateStreamEvent): class Config (line 28) | class Config: FILE: src/cohere/types/generate_stream_event.py class GenerateStreamEvent (line 10) | class GenerateStreamEvent(UncheckedBaseModel): class Config (line 15) | class Config: FILE: src/cohere/types/generate_stream_text.py class GenerateStreamText (line 10) | class GenerateStreamText(GenerateStreamEvent): class Config (line 27) | class Config: FILE: src/cohere/types/generate_streamed_response.py class TextGenerationGenerateStreamedResponse (line 15) | class TextGenerationGenerateStreamedResponse(UncheckedBaseModel): class Config (line 29) | class Config: class StreamEndGenerateStreamedResponse (line 34) | class StreamEndGenerateStreamedResponse(UncheckedBaseModel): class Config (line 48) | class Config: class StreamErrorGenerateStreamedResponse (line 53) | class StreamErrorGenerateStreamedResponse(UncheckedBaseModel): class Config (line 68) | class Config: FILE: src/cohere/types/generation.py class Generation (line 12) | class Generation(UncheckedBaseModel): class Config (line 30) | class Config: FILE: src/cohere/types/get_connector_response.py class GetConnectorResponse (line 11) | class GetConnectorResponse(UncheckedBaseModel): class Config (line 18) | class Config: FILE: src/cohere/types/get_model_response.py class GetModelResponse (line 12) | class GetModelResponse(UncheckedBaseModel): class Config (line 66) | class Config: FILE: src/cohere/types/get_model_response_sampling_defaults.py class GetModelResponseSamplingDefaults (line 10) | class GetModelResponseSamplingDefaults(UncheckedBaseModel): class Config (line 26) | class Config: FILE: src/cohere/types/image.py class Image (line 10) | class Image(UncheckedBaseModel): class Config (line 35) | class Config: FILE: src/cohere/types/image_content.py class ImageContent (line 11) | class ImageContent(UncheckedBaseModel): class Config (line 22) | class Config: FILE: src/cohere/types/image_url.py class ImageUrl (line 11) | class ImageUrl(UncheckedBaseModel): class Config (line 26) | class Config: FILE: src/cohere/types/json_response_format.py class JsonResponseFormat (line 12) | class JsonResponseFormat(UncheckedBaseModel): class Config (line 26) | class Config: FILE: src/cohere/types/json_response_format_v2.py class JsonResponseFormatV2 (line 10) | class JsonResponseFormatV2(UncheckedBaseModel): class Config (line 33) | class Config: FILE: src/cohere/types/label_metric.py class LabelMetric (line 10) | class LabelMetric(UncheckedBaseModel): class Config (line 30) | class Config: FILE: src/cohere/types/list_connectors_response.py class ListConnectorsResponse (line 11) | class ListConnectorsResponse(UncheckedBaseModel): class Config (line 22) | class Config: FILE: src/cohere/types/list_embed_job_response.py class ListEmbedJobResponse (line 11) | class ListEmbedJobResponse(UncheckedBaseModel): class Config (line 18) | class Config: FILE: src/cohere/types/list_models_response.py class ListModelsResponse (line 11) | class ListModelsResponse(UncheckedBaseModel): class Config (line 22) | class Config: FILE: src/cohere/types/logprob_item.py class LogprobItem (line 10) | class LogprobItem(UncheckedBaseModel): class Config (line 30) | class Config: FILE: src/cohere/types/message.py class ChatbotMessage (line 15) | class ChatbotMessage(UncheckedBaseModel): class Config (line 24) | class Config: class SystemMessage (line 29) | class SystemMessage(UncheckedBaseModel): class Config (line 38) | class Config: class UserMessage (line 43) | class UserMessage(UncheckedBaseModel): class Config (line 52) | class Config: class ToolMessage (line 57) | class ToolMessage(UncheckedBaseModel): class Config (line 65) | class Config: FILE: src/cohere/types/metrics.py class Metrics (line 11) | class Metrics(UncheckedBaseModel): class Config (line 18) | class Config: FILE: src/cohere/types/non_streamed_chat_response.py class NonStreamedChatResponse (line 18) | class NonStreamedChatResponse(UncheckedBaseModel): class Config (line 72) | class Config: FILE: src/cohere/types/o_auth_authorize_response.py class OAuthAuthorizeResponse (line 10) | class OAuthAuthorizeResponse(UncheckedBaseModel): class Config (line 20) | class Config: FILE: src/cohere/types/parse_info.py class ParseInfo (line 10) | class ParseInfo(UncheckedBaseModel): class Config (line 18) | class Config: FILE: src/cohere/types/rerank_response.py class RerankResponse (line 12) | class RerankResponse(UncheckedBaseModel): class Config (line 25) | class Config: FILE: src/cohere/types/rerank_response_results_item.py class RerankResponseResultsItem (line 11) | class RerankResponseResultsItem(UncheckedBaseModel): class Config (line 31) | class Config: FILE: src/cohere/types/rerank_response_results_item_document.py class RerankResponseResultsItemDocument (line 10) | class RerankResponseResultsItemDocument(UncheckedBaseModel): class Config (line 24) | class Config: FILE: src/cohere/types/reranker_data_metrics.py class RerankerDataMetrics (line 10) | class RerankerDataMetrics(UncheckedBaseModel): class Config (line 45) | class Config: FILE: src/cohere/types/response_format.py class TextResponseFormat (line 14) | class TextResponseFormat(UncheckedBaseModel): class Config (line 32) | class Config: class JsonObjectResponseFormat (line 37) | class JsonObjectResponseFormat(UncheckedBaseModel): class Config (line 58) | class Config: FILE: src/cohere/types/response_format_v2.py class TextResponseFormatV2 (line 13) | class TextResponseFormatV2(UncheckedBaseModel): class Config (line 34) | class Config: class JsonObjectResponseFormatV2 (line 39) | class JsonObjectResponseFormatV2(UncheckedBaseModel): class Config (line 61) | class Config: FILE: src/cohere/types/single_generation.py class SingleGeneration (line 11) | class SingleGeneration(UncheckedBaseModel): class Config (line 29) | class Config: FILE: src/cohere/types/single_generation_in_stream.py class SingleGenerationInStream (line 11) | class SingleGenerationInStream(UncheckedBaseModel): class Config (line 29) | class Config: FILE: src/cohere/types/single_generation_token_likelihoods_item.py class SingleGenerationTokenLikelihoodsItem (line 10) | class SingleGenerationTokenLikelihoodsItem(UncheckedBaseModel): class Config (line 18) | class Config: FILE: src/cohere/types/source.py class ToolSource (line 13) | class ToolSource(UncheckedBaseModel): class Config (line 26) | class Config: class DocumentSource (line 31) | class DocumentSource(UncheckedBaseModel): class Config (line 44) | class Config: FILE: src/cohere/types/streamed_chat_response.py class StreamStartStreamedChatResponse (line 21) | class StreamStartStreamedChatResponse(UncheckedBaseModel): class Config (line 33) | class Config: class SearchQueriesGenerationStreamedChatResponse (line 38) | class SearchQueriesGenerationStreamedChatResponse(UncheckedBaseModel): class Config (line 50) | class Config: class SearchResultsStreamedChatResponse (line 55) | class SearchResultsStreamedChatResponse(UncheckedBaseModel): class Config (line 68) | class Config: class TextGenerationStreamedChatResponse (line 73) | class TextGenerationStreamedChatResponse(UncheckedBaseModel): class Config (line 85) | class Config: class CitationGenerationStreamedChatResponse (line 90) | class CitationGenerationStreamedChatResponse(UncheckedBaseModel): class Config (line 102) | class Config: class ToolCallsGenerationStreamedChatResponse (line 107) | class ToolCallsGenerationStreamedChatResponse(UncheckedBaseModel): class Config (line 120) | class Config: class StreamEndStreamedChatResponse (line 125) | class StreamEndStreamedChatResponse(UncheckedBaseModel): class Config (line 138) | class Config: class ToolCallsChunkStreamedChatResponse (line 143) | class ToolCallsChunkStreamedChatResponse(UncheckedBaseModel): class Config (line 156) | class Config: class DebugStreamedChatResponse (line 161) | class DebugStreamedChatResponse(UncheckedBaseModel): class Config (line 173) | class Config: FILE: src/cohere/types/summarize_response.py class SummarizeResponse (line 11) | class SummarizeResponse(UncheckedBaseModel): class Config (line 28) | class Config: FILE: src/cohere/types/system_message_v2.py class SystemMessageV2 (line 11) | class SystemMessageV2(UncheckedBaseModel): class Config (line 22) | class Config: FILE: src/cohere/types/system_message_v2content_one_item.py class TextSystemMessageV2ContentOneItem (line 12) | class TextSystemMessageV2ContentOneItem(UncheckedBaseModel): class Config (line 20) | class Config: FILE: src/cohere/types/thinking.py class Thinking (line 11) | class Thinking(UncheckedBaseModel): class Config (line 31) | class Config: FILE: src/cohere/types/tokenize_response.py class TokenizeResponse (line 11) | class TokenizeResponse(UncheckedBaseModel): class Config (line 24) | class Config: FILE: src/cohere/types/tool.py class Tool (line 11) | class Tool(UncheckedBaseModel): class Config (line 42) | class Config: FILE: src/cohere/types/tool_call.py class ToolCall (line 10) | class ToolCall(UncheckedBaseModel): class Config (line 29) | class Config: FILE: src/cohere/types/tool_call_delta.py class ToolCallDelta (line 10) | class ToolCallDelta(UncheckedBaseModel): class Config (line 39) | class Config: FILE: src/cohere/types/tool_call_v2.py class ToolCallV2 (line 11) | class ToolCallV2(UncheckedBaseModel): class Config (line 24) | class Config: FILE: src/cohere/types/tool_call_v2function.py class ToolCallV2Function (line 10) | class ToolCallV2Function(UncheckedBaseModel): class Config (line 18) | class Config: FILE: src/cohere/types/tool_content.py class TextToolContent (line 14) | class TextToolContent(UncheckedBaseModel): class Config (line 26) | class Config: class DocumentToolContent (line 31) | class DocumentToolContent(UncheckedBaseModel): class Config (line 43) | class Config: FILE: src/cohere/types/tool_message_v2.py class ToolMessageV2 (line 11) | class ToolMessageV2(UncheckedBaseModel): class Config (line 30) | class Config: FILE: src/cohere/types/tool_parameter_definitions_value.py class ToolParameterDefinitionsValue (line 10) | class ToolParameterDefinitionsValue(UncheckedBaseModel): class Config (line 30) | class Config: FILE: src/cohere/types/tool_result.py class ToolResult (line 11) | class ToolResult(UncheckedBaseModel): class Config (line 19) | class Config: FILE: src/cohere/types/tool_v2.py class ToolV2 (line 11) | class ToolV2(UncheckedBaseModel): class Config (line 22) | class Config: FILE: src/cohere/types/tool_v2function.py class ToolV2Function (line 10) | class ToolV2Function(UncheckedBaseModel): class Config (line 34) | class Config: FILE: src/cohere/types/update_connector_response.py class UpdateConnectorResponse (line 11) | class UpdateConnectorResponse(UncheckedBaseModel): class Config (line 18) | class Config: FILE: src/cohere/types/usage.py class Usage (line 12) | class Usage(UncheckedBaseModel): class Config (line 24) | class Config: FILE: src/cohere/types/usage_billed_units.py class UsageBilledUnits (line 10) | class UsageBilledUnits(UncheckedBaseModel): class Config (line 35) | class Config: FILE: src/cohere/types/usage_tokens.py class UsageTokens (line 10) | class UsageTokens(UncheckedBaseModel): class Config (line 25) | class Config: FILE: src/cohere/types/user_message_v2.py class UserMessageV2 (line 11) | class UserMessageV2(UncheckedBaseModel): class Config (line 26) | class Config: FILE: src/cohere/utils.py function get_terminal_states (line 20) | def get_terminal_states(): function get_success_states (line 24) | def get_success_states(): function get_failed_states (line 28) | def get_failed_states(): function get_id (line 32) | def get_id( function get_validation_status (line 38) | def get_validation_status(awaitable: typing.Union[EmbedJob, DatasetsGetR... function get_job (line 42) | def get_job(cohere: typing.Any, function async_get_job (line 54) | async def async_get_job(cohere: typing.Any, awaitable: typing.Union[Crea... function get_failure_reason (line 65) | def get_failure_reason(job: typing.Union[EmbedJob, DatasetsGetResponse])... function wait (line 74) | def wait( function wait (line 84) | def wait( function wait (line 93) | def wait( function async_wait (line 120) | async def async_wait( function async_wait (line 130) | async def async_wait( function async_wait (line 139) | async def async_wait( function sum_fields_if_not_none (line 165) | def sum_fields_if_not_none(obj: typing.Any, field: str) -> Optional[int]: function merge_meta_field (line 170) | def merge_meta_field(metas: typing.List[ApiMeta]) -> ApiMeta: function merge_embed_responses (line 190) | def merge_embed_responses(responses: typing.List[EmbedResponse]) -> Embe... function save_avro (line 250) | def save_avro(dataset: Dataset, filepath: str): function save_jsonl (line 258) | def save_jsonl(dataset: Dataset, filepath: str): function save_csv (line 265) | def save_csv(dataset: Dataset, filepath: str): function dataset_generator (line 274) | def dataset_generator(dataset: Dataset): class SdkUtils (line 285) | class SdkUtils: method save_dataset (line 288) | def save_dataset(dataset: Dataset, filepath: str, format: typing.Liter... class SyncSdkUtils (line 298) | class SyncSdkUtils(SdkUtils): class AsyncSdkUtils (line 302) | class AsyncSdkUtils(SdkUtils): FILE: src/cohere/v2/__init__.py function __getattr__ (line 61) | def __getattr__(attr_name: str) -> typing.Any: function __dir__ (line 77) | def __dir__(): FILE: src/cohere/v2/client.py class V2Client (line 32) | class V2Client: method __init__ (line 33) | def __init__(self, *, client_wrapper: SyncClientWrapper): method with_raw_response (line 37) | def with_raw_response(self) -> RawV2Client: method chat_stream (line 47) | def chat_stream( method chat (line 217) | def chat( method embed (line 385) | def embed( method rerank (line 492) | def rerank( class AsyncV2Client (line 572) | class AsyncV2Client: method __init__ (line 573) | def __init__(self, *, client_wrapper: AsyncClientWrapper): method with_raw_response (line 577) | def with_raw_response(self) -> AsyncRawV2Client: method chat_stream (line 587) | async def chat_stream( method chat (line 766) | async def chat( method embed (line 942) | async def embed( method rerank (line 1057) | async def rerank( FILE: src/cohere/v2/raw_client.py class RawV2Client (line 54) | class RawV2Client: method __init__ (line 55) | def __init__(self, *, client_wrapper: SyncClientWrapper): method chat_stream (line 59) | def chat_stream( method chat (line 409) | def chat( method embed (line 730) | def embed( method rerank (line 981) | def rerank( class AsyncRawV2Client (line 1197) | class AsyncRawV2Client: method __init__ (line 1198) | def __init__(self, *, client_wrapper: AsyncClientWrapper): method chat_stream (line 1202) | async def chat_stream( method chat (line 1552) | async def chat( method embed (line 1873) | async def embed( method rerank (line 2124) | async def rerank( FILE: src/cohere/v2/types/__init__.py function __getattr__ (line 61) | def __getattr__(attr_name: str) -> typing.Any: function __dir__ (line 77) | def __dir__(): FILE: src/cohere/v2/types/v2chat_response.py class V2ChatResponse (line 14) | class V2ChatResponse(UncheckedBaseModel): class Config (line 29) | class Config: FILE: src/cohere/v2/types/v2chat_stream_response.py class MessageStartV2ChatStreamResponse (line 22) | class MessageStartV2ChatStreamResponse(UncheckedBaseModel): class Config (line 35) | class Config: class ContentStartV2ChatStreamResponse (line 40) | class ContentStartV2ChatStreamResponse(UncheckedBaseModel): class Config (line 53) | class Config: class ContentDeltaV2ChatStreamResponse (line 58) | class ContentDeltaV2ChatStreamResponse(UncheckedBaseModel): class Config (line 72) | class Config: class ContentEndV2ChatStreamResponse (line 77) | class ContentEndV2ChatStreamResponse(UncheckedBaseModel): class Config (line 89) | class Config: class ToolPlanDeltaV2ChatStreamResponse (line 94) | class ToolPlanDeltaV2ChatStreamResponse(UncheckedBaseModel): class Config (line 106) | class Config: class ToolCallStartV2ChatStreamResponse (line 111) | class ToolCallStartV2ChatStreamResponse(UncheckedBaseModel): class Config (line 124) | class Config: class ToolCallDeltaV2ChatStreamResponse (line 129) | class ToolCallDeltaV2ChatStreamResponse(UncheckedBaseModel): class Config (line 142) | class Config: class ToolCallEndV2ChatStreamResponse (line 147) | class ToolCallEndV2ChatStreamResponse(UncheckedBaseModel): class Config (line 159) | class Config: class CitationStartV2ChatStreamResponse (line 164) | class CitationStartV2ChatStreamResponse(UncheckedBaseModel): class Config (line 177) | class Config: class CitationEndV2ChatStreamResponse (line 182) | class CitationEndV2ChatStreamResponse(UncheckedBaseModel): class Config (line 194) | class Config: class MessageEndV2ChatStreamResponse (line 199) | class MessageEndV2ChatStreamResponse(UncheckedBaseModel): class Config (line 212) | class Config: class DebugV2ChatStreamResponse (line 217) | class DebugV2ChatStreamResponse(UncheckedBaseModel): class Config (line 229) | class Config: FILE: src/cohere/v2/types/v2rerank_response.py class V2RerankResponse (line 12) | class V2RerankResponse(UncheckedBaseModel): class Config (line 25) | class Config: FILE: src/cohere/v2/types/v2rerank_response_results_item.py class V2RerankResponseResultsItem (line 10) | class V2RerankResponseResultsItem(UncheckedBaseModel): class Config (line 25) | class Config: FILE: tests/test_async_client.py class TestClient (line 12) | class TestClient(unittest.IsolatedAsyncioTestCase): method setUp (line 15) | def setUp(self) -> None: method test_token_falls_back_on_env_variable (line 18) | async def test_token_falls_back_on_env_variable(self) -> None: method test_context_manager (line 22) | async def test_context_manager(self) -> None: method test_chat (line 26) | async def test_chat(self) -> None: method test_chat_stream (line 40) | async def test_chat_stream(self) -> None: method test_stream_equals_true (line 63) | async def test_stream_equals_true(self) -> None: method test_deprecated_fn (line 70) | async def test_deprecated_fn(self) -> None: method test_moved_fn (line 74) | async def test_moved_fn(self) -> None: method test_embed (line 80) | async def test_embed(self) -> None: method test_embed_batch_types (line 88) | async def test_embed_batch_types(self) -> None: method test_embed_batch_v1 (line 109) | async def test_embed_batch_v1(self) -> None: method test_embed_job_crud (line 125) | async def test_embed_job_crud(self) -> None: method test_rerank (line 157) | async def test_rerank(self) -> None: method test_datasets_crud (line 174) | async def test_datasets_crud(self) -> None: method test_save_load (line 194) | async def test_save_load(self) -> None: method test_tokenize (line 214) | async def test_tokenize(self) -> None: method test_detokenize (line 222) | async def test_detokenize(self) -> None: method test_tool_use (line 231) | async def test_tool_use(self) -> None: method test_local_tokenize (line 303) | async def test_local_tokenize(self) -> None: method test_local_detokenize (line 310) | async def test_local_detokenize(self) -> None: method test_tokenize_async_context_with_sync_client (line 317) | async def test_tokenize_async_context_with_sync_client(self) -> None: FILE: tests/test_aws_client_unit.py class TestSigV4HostHeader (line 21) | class TestSigV4HostHeader(unittest.TestCase): method test_sigv4_signs_with_correct_host (line 25) | def test_sigv4_signs_with_correct_host(self) -> None: class TestModeConditionalInit (line 77) | class TestModeConditionalInit(unittest.TestCase): method test_sagemaker_mode_creates_sagemaker_clients (line 81) | def test_sagemaker_mode_creates_sagemaker_clients(self) -> None: method test_bedrock_mode_creates_bedrock_clients (line 103) | def test_bedrock_mode_creates_bedrock_clients(self) -> None: method test_default_mode_is_sagemaker (line 125) | def test_default_mode_is_sagemaker(self) -> None: class TestEmbedV4Params (line 132) | class TestEmbedV4Params(unittest.TestCase): method _make_bedrock_client (line 137) | def _make_bedrock_client(): # type: ignore method test_embed_accepts_new_params (line 159) | def test_embed_accepts_new_params(self) -> None: method test_embed_passes_params_to_bedrock (line 168) | def test_embed_passes_params_to_bedrock(self) -> None: method test_embed_omits_none_params (line 190) | def test_embed_omits_none_params(self) -> None: method test_embed_with_embedding_types_returns_dict (line 210) | def test_embed_with_embedding_types_returns_dict(self) -> None: FILE: tests/test_bedrock_client.py function _setup_boto3_env (line 14) | def _setup_boto3_env(): class TestClient (line 25) | class TestClient(unittest.TestCase): method setUp (line 33) | def setUp(self) -> None: method test_rerank (line 41) | def test_rerank(self) -> None: method test_embed (line 60) | def test_embed(self) -> None: method test_generate (line 68) | def test_generate(self) -> None: method test_generate_stream (line 75) | def test_generate_stream(self) -> None: method test_chat (line 85) | def test_chat(self) -> None: method test_chat_stream (line 108) | def test_chat_stream(self) -> None: class TestBedrockClientV2 (line 128) | class TestBedrockClientV2(unittest.TestCase): method setUp (line 135) | def setUp(self) -> None: method test_embed (line 143) | def test_embed(self) -> None: method test_embed_with_output_dimension (line 152) | def test_embed_with_output_dimension(self) -> None: class TestCohereAwsBedrockClient (line 164) | class TestCohereAwsBedrockClient(unittest.TestCase): method setUpClass (line 174) | def setUpClass(cls) -> None: method test_client_is_bedrock_mode (line 180) | def test_client_is_bedrock_mode(self) -> None: method test_embed (line 184) | def test_embed(self) -> None: method test_embed_with_embedding_types (line 194) | def test_embed_with_embedding_types(self) -> None: method test_embed_with_output_dimension (line 206) | def test_embed_with_output_dimension(self) -> None: method test_embed_without_new_params (line 219) | def test_embed_without_new_params(self) -> None: FILE: tests/test_client.py class TestClient (line 15) | class TestClient(unittest.TestCase): method test_token_falls_back_on_env_variable (line 17) | def test_token_falls_back_on_env_variable(self) -> None: method test_context_manager (line 21) | def test_context_manager(self) -> None: method test_chat (line 25) | def test_chat(self) -> None: method test_chat_stream (line 38) | def test_chat_stream(self) -> None: method test_stream_equals_true (line 60) | def test_stream_equals_true(self) -> None: method test_deprecated_fn (line 67) | def test_deprecated_fn(self) -> None: method test_moved_fn (line 71) | def test_moved_fn(self) -> None: method test_embed (line 75) | def test_embed(self) -> None: method test_image_embed (line 100) | def test_image_embed(self) -> None: method test_embed_batch_types (line 125) | def test_embed_batch_types(self) -> None: method test_embed_batch_v1 (line 146) | def test_embed_batch_v1(self) -> None: method test_embed_job_crud (line 162) | def test_embed_job_crud(self) -> None: method test_rerank (line 194) | def test_rerank(self) -> None: method test_datasets_crud (line 211) | def test_datasets_crud(self) -> None: method test_save_load (line 231) | def test_save_load(self) -> None: method test_tokenize (line 251) | def test_tokenize(self) -> None: method test_detokenize (line 259) | def test_detokenize(self) -> None: method test_tool_use (line 269) | def test_tool_use(self) -> None: method test_local_tokenize (line 341) | def test_local_tokenize(self) -> None: method test_local_detokenize (line 348) | def test_local_detokenize(self) -> None: FILE: tests/test_client_init.py class TestClientInit (line 11) | class TestClientInit(unittest.TestCase): method test_aws_inits (line 13) | def test_aws_inits(self) -> None: method test_inits (line 19) | def test_inits(self) -> None: FILE: tests/test_client_v2.py class TestClientV2 (line 14) | class TestClientV2(unittest.TestCase): method test_chat (line 16) | def test_chat(self) -> None: method test_chat_stream (line 22) | def test_chat_stream(self) -> None: method test_legacy_methods_available (line 40) | def test_legacy_methods_available(self) -> None: method test_chat_documents (line 47) | def test_chat_documents(self) -> None: method test_chat_tools (line 65) | def test_chat_tools(self) -> None: FILE: tests/test_embed_streaming.py class TestStreamedEmbedding (line 17) | class TestStreamedEmbedding(unittest.TestCase): method test_creation (line 20) | def test_creation(self): method test_text_optional (line 27) | def test_text_optional(self): class TestExtractEmbeddings (line 32) | class TestExtractEmbeddings(unittest.TestCase): method test_v1_embeddings_floats (line 35) | def test_v1_embeddings_floats(self): method test_v1_embeddings_by_type (line 51) | def test_v1_embeddings_by_type(self): method test_v2_response_format (line 69) | def test_v2_response_format(self): method test_global_offset (line 82) | def test_global_offset(self): method test_empty_embeddings (line 93) | def test_empty_embeddings(self): method test_texts_shorter_than_embeddings (line 99) | def test_texts_shorter_than_embeddings(self): class TestBatchSizeConstant (line 112) | class TestBatchSizeConstant(unittest.TestCase): method test_default_batch_size_matches_api_limit (line 115) | def test_default_batch_size_matches_api_limit(self): FILE: tests/test_embed_utils.py class TestClient (line 130) | class TestClient(unittest.TestCase): method test_merge_embeddings_by_type (line 132) | def test_merge_embeddings_by_type(self) -> None: method test_merge_embeddings_floats (line 165) | def test_merge_embeddings_floats(self) -> None: method test_merge_partial_embeddings_floats (line 192) | def test_merge_partial_embeddings_floats(self) -> None: FILE: tests/test_oci_client.py class TestOciClient (line 75) | class TestOciClient(unittest.TestCase): method setUp (line 78) | def setUp(self): method test_embed (line 92) | def test_embed(self): method test_chat (line 105) | def test_chat(self): method test_chat_stream (line 115) | def test_chat_stream(self): class TestOciClientV2 (line 135) | class TestOciClientV2(unittest.TestCase): method setUp (line 138) | def setUp(self): method test_embed_v2 (line 153) | def test_embed_v2(self): method test_embed_with_model_prefix_v2 (line 170) | def test_embed_with_model_prefix_v2(self): method test_chat_v2 (line 183) | def test_chat_v2(self): method test_chat_vision_v2 (line 193) | def test_chat_vision_v2(self): method test_chat_tool_use_v2 (line 224) | def test_chat_tool_use_v2(self): method test_chat_tool_use_response_type_lowered (line 254) | def test_chat_tool_use_response_type_lowered(self): method test_chat_multi_turn_tool_use_v2 (line 281) | def test_chat_multi_turn_tool_use_v2(self): method test_chat_safety_mode_v2 (line 343) | def test_chat_safety_mode_v2(self): method test_chat_stream_v2 (line 353) | def test_chat_stream_v2(self): class TestOciClientAuthentication (line 390) | class TestOciClientAuthentication(unittest.TestCase): method test_config_file_auth (line 393) | def test_config_file_auth(self): method test_custom_profile_auth (line 416) | def test_custom_profile_auth(self): class TestOciClientErrors (line 440) | class TestOciClientErrors(unittest.TestCase): method test_missing_compartment_id (line 443) | def test_missing_compartment_id(self): method test_invalid_model (line 451) | def test_invalid_model(self): class TestOciClientModels (line 474) | class TestOciClientModels(unittest.TestCase): method setUp (line 477) | def setUp(self): method test_embed_english_v3 (line 492) | def test_embed_english_v3(self): method test_embed_multilingual_v3 (line 503) | def test_embed_multilingual_v3(self): method test_command_a_chat (line 514) | def test_command_a_chat(self): method test_embed_english_light_v3 (line 522) | def test_embed_english_light_v3(self): method test_embed_multilingual_light_v3 (line 532) | def test_embed_multilingual_light_v3(self): method test_embed_search_query_input_type (line 542) | def test_embed_search_query_input_type(self): method test_embed_with_embedding_types (line 552) | def test_embed_with_embedding_types(self): method test_embed_with_truncate (line 563) | def test_embed_with_truncate(self): method test_command_r_plus_chat (line 576) | def test_command_r_plus_chat(self): method test_v2_multi_turn_chat (line 590) | def test_v2_multi_turn_chat(self): method test_v2_system_message (line 604) | def test_v2_system_message(self): class TestOciClientTransformations (line 617) | class TestOciClientTransformations(unittest.TestCase): method test_thinking_parameter_transformation (line 620) | def test_thinking_parameter_transformation(self): method test_thinking_parameter_disabled (line 641) | def test_thinking_parameter_disabled(self): method test_thinking_response_transformation (line 660) | def test_thinking_response_transformation(self): method test_stream_event_thinking_transformation (line 685) | def test_stream_event_thinking_transformation(self): method test_stream_event_text_transformation (line 702) | def test_stream_event_text_transformation(self): method test_thinking_parameter_none (line 719) | def test_thinking_parameter_none(self): method test_v2_response_role_lowercased (line 736) | def test_v2_response_role_lowercased(self): method test_v2_response_finish_reason_uppercase (line 757) | def test_v2_response_finish_reason_uppercase(self): method test_v2_response_tool_calls_conversion (line 778) | def test_v2_response_tool_calls_conversion(self): method test_normalize_model_for_oci (line 809) | def test_normalize_model_for_oci(self): method test_transform_embed_request (line 823) | def test_transform_embed_request(self): method test_transform_embed_request_with_optional_params (line 843) | def test_transform_embed_request_with_optional_params(self): method test_transform_embed_request_rejects_images (line 862) | def test_transform_embed_request_rejects_images(self): method test_transform_chat_request_optional_params (line 879) | def test_transform_chat_request_optional_params(self): method test_v2_client_rejects_v1_request (line 909) | def test_v2_client_rejects_v1_request(self): method test_v1_client_rejects_v2_request (line 922) | def test_v1_client_rejects_v2_request(self): method test_unsupported_endpoint_raises (line 935) | def test_unsupported_endpoint_raises(self): method test_v1_chat_request_optional_params (line 944) | def test_v1_chat_request_optional_params(self): method test_v1_stream_wrapper_preserves_finish_reason (line 980) | def test_v1_stream_wrapper_preserves_finish_reason(self): method test_transform_chat_request_tool_message_fields (line 1004) | def test_transform_chat_request_tool_message_fields(self): method test_get_oci_url_known_endpoints (line 1031) | def test_get_oci_url_known_endpoints(self): method test_get_oci_url_unknown_endpoint_raises (line 1042) | def test_get_oci_url_unknown_endpoint_raises(self): method test_load_oci_config_missing_private_key_raises (line 1050) | def test_load_oci_config_missing_private_key_raises(self): method test_load_oci_config_ignores_inherited_session_auth (line 1067) | def test_load_oci_config_ignores_inherited_session_auth(self): method test_session_auth_prefers_security_token_signer (line 1109) | def test_session_auth_prefers_security_token_signer(self): method test_session_token_refreshed_on_subsequent_requests (line 1150) | def test_session_token_refreshed_on_subsequent_requests(self): method test_embed_response_lowercases_embedding_keys (line 1206) | def test_embed_response_lowercases_embedding_keys(self): method test_embed_response_includes_response_type_v1 (line 1225) | def test_embed_response_includes_response_type_v1(self): method test_embed_response_includes_response_type_v2 (line 1241) | def test_embed_response_includes_response_type_v2(self): method test_normalize_model_for_oci_rejects_empty_model (line 1257) | def test_normalize_model_for_oci_rejects_empty_model(self): method test_stream_wrapper_emits_full_event_lifecycle (line 1265) | def test_stream_wrapper_emits_full_event_lifecycle(self): method test_stream_wrapper_emits_new_content_block_on_thinking_transition (line 1299) | def test_stream_wrapper_emits_new_content_block_on_thinking_transition... method test_stream_wrapper_no_spurious_block_on_finish_only_event (line 1327) | def test_stream_wrapper_no_spurious_block_on_finish_only_event(self): method test_stream_wrapper_skips_malformed_json_with_warning (line 1353) | def test_stream_wrapper_skips_malformed_json_with_warning(self): method test_stream_wrapper_skips_message_end_for_empty_stream (line 1366) | def test_stream_wrapper_skips_message_end_for_empty_stream(self): method test_stream_wrapper_done_uses_current_content_index_after_transition (line 1374) | def test_stream_wrapper_done_uses_current_content_index_after_transiti... method test_stream_wrapper_raises_on_transform_error (line 1394) | def test_stream_wrapper_raises_on_transform_error(self): method test_stream_event_finish_reason_keeps_final_text (line 1407) | def test_stream_event_finish_reason_keeps_final_text(self): FILE: tests/test_oci_mypy.py function _run_mypy (line 35) | def _run_mypy(files: list[str], extra_env: dict[str, str] | None = None)... class TestOciMypy (line 49) | class TestOciMypy(unittest.TestCase): method test_oci_source_types (line 52) | def test_oci_source_types(self): method test_oci_test_types (line 57) | def test_oci_test_types(self): FILE: tests/test_overrides.py class TestClient (line 10) | class TestClient(unittest.TestCase): method test_float_alias (line 12) | def test_float_alias(self) -> None: