SDK regeneration

pyproject.toml

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

reference.md

@@ -2298,8 +2298,8 @@ response = client.v2.chat_stream(
             ],
             citations=[
                 Citation(
-                    start="string",
-                    end="string",
+                    start=1,
+                    end=1,
                     text="string",
                     sources=[
                         Source_Tool(

src/cohere/__init__.py

@@ -31,6 +31,10 @@
     ChatToolCallsChunkEvent,
     ChatToolCallsGenerationEvent,
     CheckApiKeyResponse,
+    CitationEndEvent,
+    CitationStartEvent,
+    CitationStartEventDelta,
+    CitationStartEventDeltaMessage,
     ClassifyDataMetrics,
     ClassifyExample,
     ClassifyRequestTruncate,
@@ -153,7 +157,6 @@
 )
 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 .datasets import (
@@ -206,12 +209,15 @@
     Content_Document,
     Content_Text,
     DocumentContent,
+    DocumentContentDocument,
     DocumentSource,
     NonStreamedChatResponse2,
     Source,
     Source_Document,
     Source_Tool,
     StreamedChatResponse2,
+    StreamedChatResponse2_CitationEnd,
+    StreamedChatResponse2_CitationStart,
     StreamedChatResponse2_ContentDelta,
     StreamedChatResponse2_ContentEnd,
     StreamedChatResponse2_ContentStart,
@@ -321,6 +327,10 @@
     "ChatToolPlanDeltaEventDelta",
     "CheckApiKeyResponse",
     "Citation",
+    "CitationEndEvent",
+    "CitationStartEvent",
+    "CitationStartEventDelta",
+    "CitationStartEventDeltaMessage",
     "ClassifyDataMetrics",
     "ClassifyExample",
     "ClassifyRequestTruncate",
@@ -356,6 +366,7 @@
     "DeleteConnectorResponse",
     "DetokenizeResponse",
     "DocumentContent",
+    "DocumentContentDocument",
     "DocumentSource",
     "EmbedByTypeResponse",
     "EmbedByTypeResponseEmbeddings",
@@ -430,6 +441,8 @@
     "Source_Tool",
     "StreamedChatResponse",
     "StreamedChatResponse2",
+    "StreamedChatResponse2_CitationEnd",
+    "StreamedChatResponse2_CitationStart",
     "StreamedChatResponse2_ContentDelta",
     "StreamedChatResponse2_ContentEnd",
     "StreamedChatResponse2_ContentStart",

src/cohere/core/client_wrapper.py

@@ -25,7 +25,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.8.1",
+            "X-Fern-SDK-Version": "5.9.0",
         }
         if self._client_name is not None:
             headers["X-Client-Name"] = self._client_name

src/cohere/types/__init__.py

@@ -30,6 +30,10 @@
 from .chat_tool_calls_chunk_event import ChatToolCallsChunkEvent
 from .chat_tool_calls_generation_event import ChatToolCallsGenerationEvent
 from .check_api_key_response import CheckApiKeyResponse
+from .citation_end_event import CitationEndEvent
+from .citation_start_event import CitationStartEvent
+from .citation_start_event_delta import CitationStartEventDelta
+from .citation_start_event_delta_message import CitationStartEventDeltaMessage
 from .classify_data_metrics import ClassifyDataMetrics
 from .classify_example import ClassifyExample
 from .classify_request_truncate import ClassifyRequestTruncate
@@ -166,6 +170,10 @@
     "ChatToolCallsChunkEvent",
     "ChatToolCallsGenerationEvent",
     "CheckApiKeyResponse",
+    "CitationEndEvent",
+    "CitationStartEvent",
+    "CitationStartEventDelta",
+    "CitationStartEventDeltaMessage",
     "ClassifyDataMetrics",
     "ClassifyExample",
     "ClassifyRequestTruncate",

src/cohere/types/citation_end_event.py

@@ -0,0 +1,25 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from ..v2.types.chat_stream_event_type import ChatStreamEventType
+
+
+class CitationEndEvent(ChatStreamEventType):
+    """
+    A streamed event which signifies a citation has finished streaming.
+    """
+
+    index: typing.Optional[int] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

src/cohere/types/citation_start_event.py

@@ -0,0 +1,27 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from ..v2.types.chat_stream_event_type import ChatStreamEventType
+from .citation_start_event_delta import CitationStartEventDelta
+
+
+class CitationStartEvent(ChatStreamEventType):
+    """
+    A streamed event which signifies a citation has been created.
+    """
+
+    index: typing.Optional[int] = None
+    delta: typing.Optional[CitationStartEventDelta] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

src/cohere/types/citation_start_event_delta.py

@@ -0,0 +1,22 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from ..core.unchecked_base_model import UncheckedBaseModel
+from .citation_start_event_delta_message import CitationStartEventDeltaMessage
+
+
+class CitationStartEventDelta(UncheckedBaseModel):
+    message: typing.Optional[CitationStartEventDeltaMessage] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

src/cohere/types/citation_start_event_delta_message.py

@@ -0,0 +1,22 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from ..core.unchecked_base_model import UncheckedBaseModel
+from ..v2.types.citation import Citation
+
+
+class CitationStartEventDeltaMessage(UncheckedBaseModel):
+    citations: typing.Optional[Citation] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

src/cohere/v2/__init__.py

@@ -40,12 +40,15 @@
     Content_Document,
     Content_Text,
     DocumentContent,
+    DocumentContentDocument,
     DocumentSource,
     NonStreamedChatResponse2,
     Source,
     Source_Document,
     Source_Tool,
     StreamedChatResponse2,
+    StreamedChatResponse2_CitationEnd,
+    StreamedChatResponse2_CitationStart,
     StreamedChatResponse2_ContentDelta,
     StreamedChatResponse2_ContentEnd,
     StreamedChatResponse2_ContentStart,
@@ -124,12 +127,15 @@
     "Content_Document",
     "Content_Text",
     "DocumentContent",
+    "DocumentContentDocument",
     "DocumentSource",
     "NonStreamedChatResponse2",
     "Source",
     "Source_Document",
     "Source_Tool",
     "StreamedChatResponse2",
+    "StreamedChatResponse2_CitationEnd",
+    "StreamedChatResponse2_CitationStart",
     "StreamedChatResponse2_ContentDelta",
     "StreamedChatResponse2_ContentEnd",
     "StreamedChatResponse2_ContentStart",

src/cohere/v2/client.py

@@ -162,8 +162,8 @@ def chat_stream(
                     ],
                     citations=[
                         Citation(
-                            start="string",
-                            end="string",
+                            start=1,
+                            end=1,
                             text="string",
                             sources=[
                                 Source_Tool(
@@ -580,8 +580,8 @@ async def main() -> None:
                         ],
                         citations=[
                             Citation(
-                                start="string",
-                                end="string",
+                                start=1,
+                                end=1,
                                 text="string",
                                 sources=[
                                     Source_Tool(