Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 4 additions & 0 deletions src/cohere/client.py
Original file line number Diff line number Diff line change
Expand Up @@ -139,6 +139,7 @@ def __init__(
environment: ClientEnvironment = ClientEnvironment.PRODUCTION,
client_name: typing.Optional[str] = None,
timeout: typing.Optional[float] = None,
max_retries: typing.Optional[int] = None,
httpx_client: typing.Optional[httpx.Client] = None,
thread_pool_executor: ThreadPoolExecutor = ThreadPoolExecutor(64),
log_warning_experimental_features: bool = True,
Expand All @@ -157,6 +158,7 @@ def __init__(
client_name=client_name,
token=api_key,
timeout=timeout,
max_retries=max_retries,
httpx_client=httpx_client,
)

Expand Down Expand Up @@ -386,6 +388,7 @@ def __init__(
environment: ClientEnvironment = ClientEnvironment.PRODUCTION,
client_name: typing.Optional[str] = None,
timeout: typing.Optional[float] = None,
max_retries: typing.Optional[int] = None,
httpx_client: typing.Optional[httpx.AsyncClient] = None,
thread_pool_executor: ThreadPoolExecutor = ThreadPoolExecutor(64),
log_warning_experimental_features: bool = True,
Expand All @@ -404,6 +407,7 @@ def __init__(
client_name=client_name,
token=api_key,
timeout=timeout,
max_retries=max_retries,
httpx_client=httpx_client,
)

Expand Down
4 changes: 4 additions & 0 deletions src/cohere/client_v2.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,6 +38,7 @@ def __init__(
environment: ClientEnvironment = ClientEnvironment.PRODUCTION,
client_name: typing.Optional[str] = None,
timeout: typing.Optional[float] = None,
max_retries: typing.Optional[int] = None,
httpx_client: typing.Optional[httpx.Client] = None,
thread_pool_executor: ThreadPoolExecutor = ThreadPoolExecutor(64),
log_warning_experimental_features: bool = True,
Expand All @@ -49,6 +50,7 @@ def __init__(
environment=environment,
client_name=client_name,
timeout=timeout,
max_retries=max_retries,
httpx_client=httpx_client,
thread_pool_executor=thread_pool_executor,
log_warning_experimental_features=log_warning_experimental_features,
Expand All @@ -71,6 +73,7 @@ def __init__(
environment: ClientEnvironment = ClientEnvironment.PRODUCTION,
client_name: typing.Optional[str] = None,
timeout: typing.Optional[float] = None,
max_retries: typing.Optional[int] = None,
httpx_client: typing.Optional[httpx.AsyncClient] = None,
thread_pool_executor: ThreadPoolExecutor = ThreadPoolExecutor(64),
log_warning_experimental_features: bool = True,
Expand All @@ -82,6 +85,7 @@ def __init__(
environment=environment,
client_name=client_name,
timeout=timeout,
max_retries=max_retries,
httpx_client=httpx_client,
thread_pool_executor=thread_pool_executor,
log_warning_experimental_features=log_warning_experimental_features,
Expand Down
Loading