SDK regeneration

Author: fern-api[bot]

poetry.lock

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "cohere"
-version = "5.0.0a1"
+version = "5.0.0a2"
 description = ""
 readme = "README.md"
 authors = []

pyproject.toml

@@ -12,20 +12,25 @@
     ChatMessage,
     ChatMessageRole,
     ChatRequestCitationQuality,
+    ChatRequestPromptOverride,
     ChatRequestPromptTruncation,
+    ChatRequestSearchOptions,
     ChatSearchQueriesGenerationEvent,
     ChatSearchQuery,
     ChatSearchResult,
+    ChatSearchResultConnector,
     ChatSearchResultsEvent,
     ChatStreamEndEvent,
     ChatStreamEndEventFinishReason,
     ChatStreamEndEventResponse,
     ChatStreamEvent,
     ChatStreamRequestCitationQuality,
+    ChatStreamRequestPromptOverride,
     ChatStreamRequestPromptTruncation,
+    ChatStreamRequestSearchOptions,
     ChatStreamStartEvent,
     ChatTextGenerationEvent,
-    ClassifyRequestExamplesItem,
+    ClassifyExample,
     ClassifyRequestTruncate,
     ClassifyResponse,
     ClassifyResponseClassificationsItem,
@@ -51,6 +56,7 @@
     EmbedJob,
     EmbedJobStatus,
     EmbedJobTruncate,
+    EmbedRequestEmbeddingTypesItem,
     EmbedRequestTruncate,
     EmbedResponse,
     EmbedResponse_EmbeddingsByType,
@@ -75,6 +81,7 @@
     ListEmbedJobResponse,
     NonStreamedChatResponse,
     OAuthAuthorizeResponse,
+    ParseInfo,
     RerankRequestDocumentsItem,
     RerankRequestDocumentsItemText,
     RerankResponse,
@@ -124,20 +131,25 @@
     "ChatMessage",
     "ChatMessageRole",
     "ChatRequestCitationQuality",
+    "ChatRequestPromptOverride",
     "ChatRequestPromptTruncation",
+    "ChatRequestSearchOptions",
     "ChatSearchQueriesGenerationEvent",
     "ChatSearchQuery",
     "ChatSearchResult",
+    "ChatSearchResultConnector",
     "ChatSearchResultsEvent",
     "ChatStreamEndEvent",
     "ChatStreamEndEventFinishReason",
     "ChatStreamEndEventResponse",
     "ChatStreamEvent",
     "ChatStreamRequestCitationQuality",
+    "ChatStreamRequestPromptOverride",
     "ChatStreamRequestPromptTruncation",
+    "ChatStreamRequestSearchOptions",
     "ChatStreamStartEvent",
     "ChatTextGenerationEvent",
-    "ClassifyRequestExamplesItem",
+    "ClassifyExample",
     "ClassifyRequestTruncate",
     "ClassifyResponse",
     "ClassifyResponseClassificationsItem",
@@ -169,6 +181,7 @@
     "EmbedJob",
     "EmbedJobStatus",
     "EmbedJobTruncate",
+    "EmbedRequestEmbeddingTypesItem",
     "EmbedRequestTruncate",
     "EmbedResponse",
     "EmbedResponse_EmbeddingsByType",
@@ -196,6 +209,7 @@
     "NonStreamedChatResponse",
     "NotFoundError",
     "OAuthAuthorizeResponse",
+    "ParseInfo",
     "RerankRequestDocumentsItem",
     "RerankRequestDocumentsItemText",
     "RerankResponse",

src/cohere/__init__.py

@@ -21,7 +21,7 @@ def get_headers(self) -> typing.Dict[str, str]:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "cohere",
-            "X-Fern-SDK-Version": "5.0.0a1",
+            "X-Fern-SDK-Version": "5.0.0a2",
         }
         if self._client_name is not None:
             headers["X-Client-Name"] = self._client_name

src/cohere/client.py

@@ -13,7 +13,7 @@
 
 
 class DatasetsGetResponse(pydantic.BaseModel):
-    dataset: typing.Optional[Dataset]
+    dataset: Dataset
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

src/cohere/core/client_wrapper.py

@@ -11,20 +11,25 @@
 from .chat_message import ChatMessage
 from .chat_message_role import ChatMessageRole
 from .chat_request_citation_quality import ChatRequestCitationQuality
+from .chat_request_prompt_override import ChatRequestPromptOverride
 from .chat_request_prompt_truncation import ChatRequestPromptTruncation
+from .chat_request_search_options import ChatRequestSearchOptions
 from .chat_search_queries_generation_event import ChatSearchQueriesGenerationEvent
 from .chat_search_query import ChatSearchQuery
 from .chat_search_result import ChatSearchResult
+from .chat_search_result_connector import ChatSearchResultConnector
 from .chat_search_results_event import ChatSearchResultsEvent
 from .chat_stream_end_event import ChatStreamEndEvent
 from .chat_stream_end_event_finish_reason import ChatStreamEndEventFinishReason
 from .chat_stream_end_event_response import ChatStreamEndEventResponse
 from .chat_stream_event import ChatStreamEvent
 from .chat_stream_request_citation_quality import ChatStreamRequestCitationQuality
+from .chat_stream_request_prompt_override import ChatStreamRequestPromptOverride
 from .chat_stream_request_prompt_truncation import ChatStreamRequestPromptTruncation
+from .chat_stream_request_search_options import ChatStreamRequestSearchOptions
 from .chat_stream_start_event import ChatStreamStartEvent
 from .chat_text_generation_event import ChatTextGenerationEvent
-from .classify_request_examples_item import ClassifyRequestExamplesItem
+from .classify_example import ClassifyExample
 from .classify_request_truncate import ClassifyRequestTruncate
 from .classify_response import ClassifyResponse
 from .classify_response_classifications_item import ClassifyResponseClassificationsItem
@@ -52,6 +57,7 @@
 from .embed_job import EmbedJob
 from .embed_job_status import EmbedJobStatus
 from .embed_job_truncate import EmbedJobTruncate
+from .embed_request_embedding_types_item import EmbedRequestEmbeddingTypesItem
 from .embed_request_truncate import EmbedRequestTruncate
 from .embed_response import EmbedResponse, EmbedResponse_EmbeddingsByType, EmbedResponse_EmbeddingsFloats
 from .finish_reason import FinishReason
@@ -76,6 +82,7 @@
 from .list_embed_job_response import ListEmbedJobResponse
 from .non_streamed_chat_response import NonStreamedChatResponse
 from .o_auth_authorize_response import OAuthAuthorizeResponse
+from .parse_info import ParseInfo
 from .rerank_request_documents_item import RerankRequestDocumentsItem
 from .rerank_request_documents_item_text import RerankRequestDocumentsItemText
 from .rerank_response import RerankResponse
@@ -113,20 +120,25 @@
     "ChatMessage",
     "ChatMessageRole",
     "ChatRequestCitationQuality",
+    "ChatRequestPromptOverride",
     "ChatRequestPromptTruncation",
+    "ChatRequestSearchOptions",
     "ChatSearchQueriesGenerationEvent",
     "ChatSearchQuery",
     "ChatSearchResult",
+    "ChatSearchResultConnector",
     "ChatSearchResultsEvent",
     "ChatStreamEndEvent",
     "ChatStreamEndEventFinishReason",
     "ChatStreamEndEventResponse",
     "ChatStreamEvent",
     "ChatStreamRequestCitationQuality",
+    "ChatStreamRequestPromptOverride",
     "ChatStreamRequestPromptTruncation",
+    "ChatStreamRequestSearchOptions",
     "ChatStreamStartEvent",
     "ChatTextGenerationEvent",
-    "ClassifyRequestExamplesItem",
+    "ClassifyExample",
     "ClassifyRequestTruncate",
     "ClassifyResponse",
     "ClassifyResponseClassificationsItem",
@@ -152,6 +164,7 @@
     "EmbedJob",
     "EmbedJobStatus",
     "EmbedJobTruncate",
+    "EmbedRequestEmbeddingTypesItem",
     "EmbedRequestTruncate",
     "EmbedResponse",
     "EmbedResponse_EmbeddingsByType",
@@ -176,6 +189,7 @@
     "ListEmbedJobResponse",
     "NonStreamedChatResponse",
     "OAuthAuthorizeResponse",
+    "ParseInfo",
     "RerankRequestDocumentsItem",
     "RerankRequestDocumentsItemText",
     "RerankResponse",

src/cohere/resources/datasets/types/datasets_get_response.py

@@ -16,7 +16,7 @@ class ChatConnector(pydantic.BaseModel):
     The connector used for fetching documents.
     """
 
-    id: str = pydantic.Field(description="The identifier of the connector. Currently only 'web-search' is supported.")
+    id: str = pydantic.Field(description="The identifier of the connector.")
     user_access_token: typing.Optional[str] = pydantic.Field(
         description="An optional override to set the token that Cohere passes to the connector in the Authorization header."
     )

src/cohere/types/__init__.py

@@ -0,0 +1,34 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class ChatRequestPromptOverride(pydantic.BaseModel):
+    """
+    (internal) Overrides specified parts of the default Chat or RAG preamble. It is recommended that these options only be used in specific scenarios where the defaults are not adequate.
+    """
+
+    preamble: typing.Optional[typing.Any]
+    task_description: typing.Optional[typing.Any]
+    style_guide: typing.Optional[typing.Any]
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        json_encoders = {dt.datetime: serialize_datetime}

src/cohere/types/chat_connector.py

@@ -0,0 +1,35 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class ChatRequestSearchOptions(pydantic.BaseModel):
+    """
+    (internal) Sets inference and model options for RAG search query and tool use generations. Defaults are used when options are not specified here, meaning that other parameters outside of search_options are ignored (such as model= or temperature=).
+    """
+
+    model: typing.Optional[typing.Any]
+    temperature: typing.Optional[typing.Any]
+    max_tokens: typing.Optional[typing.Any]
+    preamble: typing.Optional[typing.Any]
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        json_encoders = {dt.datetime: serialize_datetime}