SDK regeneration

cohere-ai · Sep 13, 2024 · d290fa1 · d290fa1
1 parent 6ceb338
commit d290fa1
Show file tree

Hide file tree

Showing 15 changed files with 390 additions and 265 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "cohere"
-version = "5.9.1"
+version = "5.9.2"
 description = ""
 readme = "README.md"
 authors = []

diff --git a/reference.md b/reference.md
@@ -135,6 +135,14 @@ Text input for the model to respond to.
 Compatible Deployments: Cohere Platform, Azure, AWS Sagemaker/Bedrock, Private Deployments
 
 
+</dd>
+</dl>
+
+<dl>
+<dd>
+
+**accepts:** `typing.Optional[typing.Literal["text/event-stream"]]` — Pass text/event-stream to receive the streamed response as server-sent events. The default is `\n` delimited events.
+
 </dd>
 </dl>
 
@@ -578,14 +586,15 @@ To learn how to use the Chat API with Streaming and RAG follow our [Text Generat
 <dd>
 
 ```python
-from cohere import Client
+from cohere import Client, Message_Tool
 
 client = Client(
     client_name="YOUR_CLIENT_NAME",
     token="YOUR_TOKEN",
 )
 client.chat(
     message="Can you give me a global market overview of solar panels?",
+    chat_history=[Message_Tool(), Message_Tool()],
     prompt_truncation="OFF",
     temperature=0.3,
 )
@@ -611,6 +620,14 @@ Text input for the model to respond to.
 Compatible Deployments: Cohere Platform, Azure, AWS Sagemaker/Bedrock, Private Deployments
 
 
+</dd>
+</dl>
+
+<dl>
+<dd>
+
+**accepts:** `typing.Optional[typing.Literal["text/event-stream"]]` — Pass text/event-stream to receive the streamed response as server-sent events. The default is `\n` delimited events.
+
 </dd>
 </dl>
 
@@ -1592,14 +1609,7 @@ client = Client(
     client_name="YOUR_CLIENT_NAME",
     token="YOUR_TOKEN",
 )
-client.embed(
-    texts=["string"],
-    images=["string"],
-    model="string",
-    input_type="search_document",
-    embedding_types=["float"],
-    truncate="NONE",
-)
+client.embed()
 
 ```
 </dd>
@@ -1615,7 +1625,19 @@ client.embed(
 <dl>
 <dd>
 
-**texts:** `typing.Sequence[str]` — An array of strings for the model to embed. Maximum number of texts per call is `96`. We recommend reducing the length of each text to be under `512` tokens for optimal quality.
+**texts:** `typing.Optional[typing.Sequence[str]]` — An array of strings for the model to embed. Maximum number of texts per call is `96`. We recommend reducing the length of each text to be under `512` tokens for optimal quality.
+
+</dd>
+</dl>
+
+<dl>
+<dd>
+
+**images:** `typing.Optional[typing.Sequence[str]]` 
+
+An array of image data URIs for the model to embed. Maximum number of images per call is `1`.
+
+The image must be a valid [data URI](https://developer.mozilla.org/en-US/docs/Web/URI/Schemes/data). The image must be in either `image/jpeg` or `image/png` format and has a maximum size of 5MB.
 
 </dd>
 </dl>
@@ -2338,6 +2360,7 @@ response = client.v2.chat_stream(
     ],
     citation_mode="FAST",
     response_format=ResponseFormat2_Text(),
+    safety_mode="CONTEXTUAL",
     max_tokens=1,
     stop_sequences=["string"],
     temperature=1.1,
@@ -2408,6 +2431,24 @@ Dictates the approach taken to generating citations as part of the RAG flow by a
 
 **response_format:** `typing.Optional[ResponseFormat2]` 
 
+</dd>
+</dl>
+
+<dl>
+<dd>
+
+**safety_mode:** `typing.Optional[V2ChatStreamRequestSafetyMode]` 
+
+Used to select the [safety instruction](/docs/safety-modes) inserted into the prompt. Defaults to `CONTEXTUAL`.
+When `NONE` is specified, the safety instruction will be omitted.
+
+Safety modes are not yet configurable in combination with `tools`, `tool_results` and `documents` parameters.
+
+**Note**: This parameter is only compatible with models [Command R 08-2024](/docs/command-r#august-2024-release), [Command R+ 08-2024](/docs/command-r-plus#august-2024-release) and newer.
+
+Compatible Deployments: Cohere Platform, Azure, AWS Sagemaker/Bedrock, Private Deployments
+
+
 </dd>
 </dl>
 
@@ -2557,14 +2598,20 @@ Generates a message from the model in response to a provided conversation. To le
 
 ```python
 from cohere import Client
+from cohere.v2 import ChatMessage2_Tool
 
 client = Client(
     client_name="YOUR_CLIENT_NAME",
     token="YOUR_TOKEN",
 )
 client.v2.chat(
     model="model",
-    messages=[],
+    messages=[
+        ChatMessage2_Tool(
+            tool_call_id="messages",
+            tool_content=["messages"],
+        )
+    ],
 )
 
 ```
@@ -2624,6 +2671,24 @@ Dictates the approach taken to generating citations as part of the RAG flow by a
 
 **response_format:** `typing.Optional[ResponseFormat2]` 
 
+</dd>
+</dl>
+
+<dl>
+<dd>
+
+**safety_mode:** `typing.Optional[V2ChatRequestSafetyMode]` 
+
+Used to select the [safety instruction](/docs/safety-modes) inserted into the prompt. Defaults to `CONTEXTUAL`.
+When `NONE` is specified, the safety instruction will be omitted.
+
+Safety modes are not yet configurable in combination with `tools`, `tool_results` and `documents` parameters.
+
+**Note**: This parameter is only compatible with models [Command R 08-2024](/docs/command-r#august-2024-release), [Command R+ 08-2024](/docs/command-r-plus#august-2024-release) and newer.
+
+Compatible Deployments: Cohere Platform, Azure, AWS Sagemaker/Bedrock, Private Deployments
+
+
 </dd>
 </dl>
 

diff --git a/src/cohere/__init__.py b/src/cohere/__init__.py
@@ -163,9 +163,9 @@
 )
 from . import connectors, datasets, embed_jobs, finetuning, models, v2
 from .aws_client import AwsClient
-from .client_v2 import AsyncClientV2, ClientV2
 from .bedrock_client import BedrockClient
 from .client import AsyncClient, Client
+from .client_v2 import AsyncClientV2, ClientV2
 from .datasets import (
     DatasetsCreateResponse,
     DatasetsCreateResponseDatasetPartsItem,
@@ -246,18 +246,17 @@
     Tool2Function,
     ToolCall2,
     ToolCall2Function,
-    ToolContent,
     ToolMessage2,
-    ToolMessage2ToolContentItem,
-    ToolMessage2ToolContentItem_ToolResultObject,
     ToolSource,
     Usage,
     UsageBilledUnits,
     UsageTokens,
     UserMessage,
     UserMessageContent,
     V2ChatRequestCitationMode,
+    V2ChatRequestSafetyMode,
     V2ChatStreamRequestCitationMode,
+    V2ChatStreamRequestSafetyMode,
 )
 from .version import __version__
 
@@ -274,6 +273,7 @@
     "AssistantMessageResponseContentItem",
     "AssistantMessageResponseContentItem_Text",
     "AsyncClient",
+    "AsyncClientV2",
     "AuthTokenType",
     "AwsClient",
     "BadRequestError",
@@ -354,6 +354,7 @@
     "ClientClosedRequestError",
     "ClientClosedRequestErrorBody",
     "ClientEnvironment",
+    "ClientV2",
     "CompatibleEndpoint",
     "Connector",
     "ConnectorAuthStatus",
@@ -493,11 +494,8 @@
     "ToolCall2",
     "ToolCall2Function",
     "ToolCallDelta",
-    "ToolContent",
     "ToolMessage",
     "ToolMessage2",
-    "ToolMessage2ToolContentItem",
-    "ToolMessage2ToolContentItem_ToolResultObject",
     "ToolParameterDefinitionsValue",
     "ToolResult",
     "ToolSource",
@@ -511,7 +509,9 @@
     "UserMessage",
     "UserMessageContent",
     "V2ChatRequestCitationMode",
+    "V2ChatRequestSafetyMode",
     "V2ChatStreamRequestCitationMode",
+    "V2ChatStreamRequestSafetyMode",
     "__version__",
     "connectors",
     "datasets",