diff --git a/pyproject.toml b/pyproject.toml index 521c751..9f8570c 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -3,7 +3,7 @@ name = "credal" [tool.poetry] name = "credal" -version = "0.0.21" +version = "0.0.22" description = "" readme = "README.md" authors = [] diff --git a/src/credal/__init__.py b/src/credal/__init__.py index 5a3f257..965acd1 100644 --- a/src/credal/__init__.py +++ b/src/credal/__init__.py @@ -36,6 +36,7 @@ DataFilter, DatetimeFieldSchema, DeleteCopilotResponse, + EndOfMessageChunk, FeedbackEnum, Filter, Filter_Boolean, @@ -60,6 +61,7 @@ StreamingChunk, StreamingChunk_Blocked, StreamingChunk_DataChunk, + StreamingChunk_EndOfMessage, StreamingChunk_FinalChunk, StreamingChunk_Initial, StringFieldSchema, @@ -134,6 +136,7 @@ "DocumentCollectionSearchResult", "DocumentMetadataPatch", "DocumentMetadataPatchRequest", + "EndOfMessageChunk", "ExternalResourceId", "FeedbackEnum", "Filter", @@ -183,6 +186,7 @@ "StreamingChunk", "StreamingChunk_Blocked", "StreamingChunk_DataChunk", + "StreamingChunk_EndOfMessage", "StreamingChunk_FinalChunk", "StreamingChunk_Initial", "StringFieldSchema", diff --git a/src/credal/copilots/__init__.py b/src/credal/copilots/__init__.py index cb15b55..f25d792 100644 --- a/src/credal/copilots/__init__.py +++ b/src/credal/copilots/__init__.py @@ -12,6 +12,7 @@ DataFilter, DatetimeFieldSchema, DeleteCopilotResponse, + EndOfMessageChunk, FeedbackEnum, Filter, Filter_Boolean, @@ -36,6 +37,7 @@ StreamingChunk, StreamingChunk_Blocked, StreamingChunk_DataChunk, + StreamingChunk_EndOfMessage, StreamingChunk_FinalChunk, StreamingChunk_Initial, StringFieldSchema, @@ -54,6 +56,7 @@ "DataFilter", "DatetimeFieldSchema", "DeleteCopilotResponse", + "EndOfMessageChunk", "FeedbackEnum", "Filter", "Filter_Boolean", @@ -78,6 +81,7 @@ "StreamingChunk", "StreamingChunk_Blocked", "StreamingChunk_DataChunk", + "StreamingChunk_EndOfMessage", "StreamingChunk_FinalChunk", "StreamingChunk_Initial", "StringFieldSchema", diff --git a/src/credal/copilots/types/__init__.py b/src/credal/copilots/types/__init__.py index 2ca90bf..9b30138 100644 --- a/src/credal/copilots/types/__init__.py +++ b/src/credal/copilots/types/__init__.py @@ -11,6 +11,7 @@ from .data_filter import DataFilter from .datetime_field_schema import DatetimeFieldSchema from .delete_copilot_response import DeleteCopilotResponse +from .end_of_message_chunk import EndOfMessageChunk from .feedback_enum import FeedbackEnum from .filter import Filter, Filter_Boolean, Filter_Datetime, Filter_Number, Filter_String from .final_chunk import FinalChunk @@ -34,6 +35,7 @@ StreamingChunk, StreamingChunk_Blocked, StreamingChunk_DataChunk, + StreamingChunk_EndOfMessage, StreamingChunk_FinalChunk, StreamingChunk_Initial, ) @@ -52,6 +54,7 @@ "DataFilter", "DatetimeFieldSchema", "DeleteCopilotResponse", + "EndOfMessageChunk", "FeedbackEnum", "Filter", "Filter_Boolean", @@ -76,6 +79,7 @@ "StreamingChunk", "StreamingChunk_Blocked", "StreamingChunk_DataChunk", + "StreamingChunk_EndOfMessage", "StreamingChunk_FinalChunk", "StreamingChunk_Initial", "StringFieldSchema", diff --git a/src/credal/copilots/types/end_of_message_chunk.py b/src/credal/copilots/types/end_of_message_chunk.py new file mode 100644 index 0000000..d3b01d0 --- /dev/null +++ b/src/credal/copilots/types/end_of_message_chunk.py @@ -0,0 +1,17 @@ +# This file was auto-generated by Fern from our API Definition. + +from ...core.pydantic_utilities import UniversalBaseModel +from ...core.pydantic_utilities import IS_PYDANTIC_V2 +import typing +import pydantic + + +class EndOfMessageChunk(UniversalBaseModel): + 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 diff --git a/src/credal/copilots/types/streaming_chunk.py b/src/credal/copilots/types/streaming_chunk.py index 9ca1851..a4a7435 100644 --- a/src/credal/copilots/types/streaming_chunk.py +++ b/src/credal/copilots/types/streaming_chunk.py @@ -47,6 +47,19 @@ class Config: extra = pydantic.Extra.allow +class StreamingChunk_EndOfMessage(UniversalBaseModel): + event: typing.Literal["end_of_message"] = "end_of_message" + + 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 + + class StreamingChunk_FinalChunk(UniversalBaseModel): event: typing.Literal["final_chunk"] = "final_chunk" sources: typing.List[ReferencedSource] @@ -79,5 +92,9 @@ class Config: StreamingChunk = typing.Union[ - StreamingChunk_Initial, StreamingChunk_DataChunk, StreamingChunk_FinalChunk, StreamingChunk_Blocked + StreamingChunk_Initial, + StreamingChunk_DataChunk, + StreamingChunk_EndOfMessage, + StreamingChunk_FinalChunk, + StreamingChunk_Blocked, ] diff --git a/src/credal/core/client_wrapper.py b/src/credal/core/client_wrapper.py index cc141ac..369413b 100644 --- a/src/credal/core/client_wrapper.py +++ b/src/credal/core/client_wrapper.py @@ -22,7 +22,7 @@ def get_headers(self) -> typing.Dict[str, str]: headers: typing.Dict[str, str] = { "X-Fern-Language": "Python", "X-Fern-SDK-Name": "credal", - "X-Fern-SDK-Version": "0.0.21", + "X-Fern-SDK-Version": "0.0.22", } headers["Authorization"] = f"Bearer {self._get_api_key()}" return headers