letta-client 0.1.232__py3-none-any.whl → 0.1.234__py3-none-any.whl

letta_client/groups/messages/client.py

@@ -1,25 +1,17 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from ...core.client_wrapper import SyncClientWrapper
+
+from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.request_options import RequestOptions
 from ...types.letta_message_union import LettaMessageUnion
-from ...core.jsonable_encoder import jsonable_encoder
-from ...core.unchecked_base_model import construct_type
-from ...errors.unprocessable_entity_error import UnprocessableEntityError
-from ...types.http_validation_error import HttpValidationError
-from json.decoder import JSONDecodeError
-from ...core.api_error import ApiError
+from ...types.letta_response import LettaResponse
 from ...types.message_create import MessageCreate
 from ...types.message_type import MessageType
-from ...types.letta_response import LettaResponse
-from ...core.serialization import convert_and_respect_annotation_metadata
+from .raw_client import AsyncRawMessagesClient, RawMessagesClient
 from .types.letta_streaming_response import LettaStreamingResponse
-import httpx_sse
-import json
 from .types.messages_modify_request import MessagesModifyRequest
 from .types.messages_modify_response import MessagesModifyResponse
-from ...core.client_wrapper import AsyncClientWrapper
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -27,7 +19,18 @@ OMIT = typing.cast(typing.Any, ...)
 
 class MessagesClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
-        self._client_wrapper = client_wrapper
+        self._raw_client = RawMessagesClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> RawMessagesClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        RawMessagesClient
+        """
+        return self._raw_client
 
     def list(
         self,
@@ -86,42 +89,17 @@ class MessagesClient:
             group_id="group_id",
         )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/groups/{jsonable_encoder(group_id)}/messages",
-            method="GET",
-            params={
-                "after": after,
-                "before": before,
-                "limit": limit,
-                "use_assistant_message": use_assistant_message,
-                "assistant_message_tool_name": assistant_message_tool_name,
-                "assistant_message_tool_kwarg": assistant_message_tool_kwarg,
-            },
+        _response = self._raw_client.list(
+            group_id,
+            after=after,
+            before=before,
+            limit=limit,
+            use_assistant_message=use_assistant_message,
+            assistant_message_tool_name=assistant_message_tool_name,
+            assistant_message_tool_kwarg=assistant_message_tool_kwarg,
             request_options=request_options,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.List[LettaMessageUnion],
-                    construct_type(
-                        type_=typing.List[LettaMessageUnion],  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        HttpValidationError,
-                        construct_type(
-                            type_=HttpValidationError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
 
     def create(
         self,
@@ -195,46 +173,18 @@ class MessagesClient:
             ],
         )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/groups/{jsonable_encoder(group_id)}/messages",
-            method="POST",
-            json={
-                "messages": convert_and_respect_annotation_metadata(
-                    object_=messages, annotation=typing.Sequence[MessageCreate], direction="write"
-                ),
-                "max_steps": max_steps,
-                "use_assistant_message": use_assistant_message,
-                "assistant_message_tool_name": assistant_message_tool_name,
-                "assistant_message_tool_kwarg": assistant_message_tool_kwarg,
-                "include_return_message_types": include_return_message_types,
-                "enable_thinking": enable_thinking,
-            },
+        _response = self._raw_client.create(
+            group_id,
+            messages=messages,
+            max_steps=max_steps,
+            use_assistant_message=use_assistant_message,
+            assistant_message_tool_name=assistant_message_tool_name,
+            assistant_message_tool_kwarg=assistant_message_tool_kwarg,
+            include_return_message_types=include_return_message_types,
+            enable_thinking=enable_thinking,
             request_options=request_options,
-            omit=OMIT,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    LettaResponse,
-                    construct_type(
-                        type_=LettaResponse,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        HttpValidationError,
-                        construct_type(
-                            type_=HttpValidationError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
 
     def create_stream(
         self,
@@ -248,6 +198,7 @@ class MessagesClient:
         include_return_message_types: typing.Optional[typing.Sequence[MessageType]] = OMIT,
         enable_thinking: typing.Optional[str] = OMIT,
         stream_tokens: typing.Optional[bool] = OMIT,
+        include_pings: typing.Optional[bool] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.Iterator[LettaStreamingResponse]:
         """
@@ -283,6 +234,9 @@ class MessagesClient:
         stream_tokens : typing.Optional[bool]
             Flag to determine if individual tokens should be streamed. Set to True for token streaming (requires stream_steps = True).
 
+        include_pings : typing.Optional[bool]
+            Whether to include periodic keepalive ping messages in the stream to prevent connection timeouts.
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -315,54 +269,20 @@ class MessagesClient:
         for chunk in response:
             yield chunk
         """
-        with self._client_wrapper.httpx_client.stream(
-            f"v1/groups/{jsonable_encoder(group_id)}/messages/stream",
-            method="POST",
-            json={
-                "messages": convert_and_respect_annotation_metadata(
-                    object_=messages, annotation=typing.Sequence[MessageCreate], direction="write"
-                ),
-                "max_steps": max_steps,
-                "use_assistant_message": use_assistant_message,
-                "assistant_message_tool_name": assistant_message_tool_name,
-                "assistant_message_tool_kwarg": assistant_message_tool_kwarg,
-                "include_return_message_types": include_return_message_types,
-                "enable_thinking": enable_thinking,
-                "stream_tokens": stream_tokens,
-            },
+        with self._raw_client.create_stream(
+            group_id,
+            messages=messages,
+            max_steps=max_steps,
+            use_assistant_message=use_assistant_message,
+            assistant_message_tool_name=assistant_message_tool_name,
+            assistant_message_tool_kwarg=assistant_message_tool_kwarg,
+            include_return_message_types=include_return_message_types,
+            enable_thinking=enable_thinking,
+            stream_tokens=stream_tokens,
+            include_pings=include_pings,
             request_options=request_options,
-            omit=OMIT,
-        ) as _response:
-            try:
-                if 200 <= _response.status_code < 300:
-                    _event_source = httpx_sse.EventSource(_response)
-                    for _sse in _event_source.iter_sse():
-                        try:
-                            yield typing.cast(
-                                LettaStreamingResponse,
-                                construct_type(
-                                    type_=LettaStreamingResponse,  # type: ignore
-                                    object_=json.loads(_sse.data),
-                                ),
-                            )
-                        except:
-                            pass
-                    return
-                _response.read()
-                if _response.status_code == 422:
-                    raise UnprocessableEntityError(
-                        typing.cast(
-                            HttpValidationError,
-                            construct_type(
-                                type_=HttpValidationError,  # type: ignore
-                                object_=_response.json(),
-                            ),
-                        )
-                    )
-                _response_json = _response.json()
-            except JSONDecodeError:
-                raise ApiError(status_code=_response.status_code, body=_response.text)
-            raise ApiError(status_code=_response.status_code, body=_response_json)
+        ) as r:
+            yield from r.data
 
     def modify(
         self,
@@ -407,38 +327,8 @@ class MessagesClient:
             ),
         )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/groups/{jsonable_encoder(group_id)}/messages/{jsonable_encoder(message_id)}",
-            method="PATCH",
-            json=convert_and_respect_annotation_metadata(
-                object_=request, annotation=MessagesModifyRequest, direction="write"
-            ),
-            request_options=request_options,
-            omit=OMIT,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    MessagesModifyResponse,
-                    construct_type(
-                        type_=MessagesModifyResponse,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        HttpValidationError,
-                        construct_type(
-                            type_=HttpValidationError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        _response = self._raw_client.modify(group_id, message_id, request=request, request_options=request_options)
+        return _response.data
 
     def reset(
         self, group_id: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -470,39 +360,24 @@ class MessagesClient:
             group_id="group_id",
         )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/groups/{jsonable_encoder(group_id)}/reset-messages",
-            method="PATCH",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.Optional[typing.Any],
-                    construct_type(
-                        type_=typing.Optional[typing.Any],  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        HttpValidationError,
-                        construct_type(
-                            type_=HttpValidationError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        _response = self._raw_client.reset(group_id, request_options=request_options)
+        return _response.data
 
 
 class AsyncMessagesClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
-        self._client_wrapper = client_wrapper
+        self._raw_client = AsyncRawMessagesClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> AsyncRawMessagesClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        AsyncRawMessagesClient
+        """
+        return self._raw_client
 
     async def list(
         self,
@@ -569,42 +444,17 @@ class AsyncMessagesClient:
 
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/groups/{jsonable_encoder(group_id)}/messages",
-            method="GET",
-            params={
-                "after": after,
-                "before": before,
-                "limit": limit,
-                "use_assistant_message": use_assistant_message,
-                "assistant_message_tool_name": assistant_message_tool_name,
-                "assistant_message_tool_kwarg": assistant_message_tool_kwarg,
-            },
+        _response = await self._raw_client.list(
+            group_id,
+            after=after,
+            before=before,
+            limit=limit,
+            use_assistant_message=use_assistant_message,
+            assistant_message_tool_name=assistant_message_tool_name,
+            assistant_message_tool_kwarg=assistant_message_tool_kwarg,
             request_options=request_options,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.List[LettaMessageUnion],
-                    construct_type(
-                        type_=typing.List[LettaMessageUnion],  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        HttpValidationError,
-                        construct_type(
-                            type_=HttpValidationError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
 
     async def create(
         self,
@@ -686,46 +536,18 @@ class AsyncMessagesClient:
 
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/groups/{jsonable_encoder(group_id)}/messages",
-            method="POST",
-            json={
-                "messages": convert_and_respect_annotation_metadata(
-                    object_=messages, annotation=typing.Sequence[MessageCreate], direction="write"
-                ),
-                "max_steps": max_steps,
-                "use_assistant_message": use_assistant_message,
-                "assistant_message_tool_name": assistant_message_tool_name,
-                "assistant_message_tool_kwarg": assistant_message_tool_kwarg,
-                "include_return_message_types": include_return_message_types,
-                "enable_thinking": enable_thinking,
-            },
+        _response = await self._raw_client.create(
+            group_id,
+            messages=messages,
+            max_steps=max_steps,
+            use_assistant_message=use_assistant_message,
+            assistant_message_tool_name=assistant_message_tool_name,
+            assistant_message_tool_kwarg=assistant_message_tool_kwarg,
+            include_return_message_types=include_return_message_types,
+            enable_thinking=enable_thinking,
             request_options=request_options,
-            omit=OMIT,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    LettaResponse,
-                    construct_type(
-                        type_=LettaResponse,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        HttpValidationError,
-                        construct_type(
-                            type_=HttpValidationError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
 
     async def create_stream(
         self,
@@ -739,6 +561,7 @@ class AsyncMessagesClient:
         include_return_message_types: typing.Optional[typing.Sequence[MessageType]] = OMIT,
         enable_thinking: typing.Optional[str] = OMIT,
         stream_tokens: typing.Optional[bool] = OMIT,
+        include_pings: typing.Optional[bool] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.AsyncIterator[LettaStreamingResponse]:
         """
@@ -774,6 +597,9 @@ class AsyncMessagesClient:
         stream_tokens : typing.Optional[bool]
             Flag to determine if individual tokens should be streamed. Set to True for token streaming (requires stream_steps = True).
 
+        include_pings : typing.Optional[bool]
+            Whether to include periodic keepalive ping messages in the stream to prevent connection timeouts.
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -814,54 +640,21 @@ class AsyncMessagesClient:
 
         asyncio.run(main())
         """
-        async with self._client_wrapper.httpx_client.stream(
-            f"v1/groups/{jsonable_encoder(group_id)}/messages/stream",
-            method="POST",
-            json={
-                "messages": convert_and_respect_annotation_metadata(
-                    object_=messages, annotation=typing.Sequence[MessageCreate], direction="write"
-                ),
-                "max_steps": max_steps,
-                "use_assistant_message": use_assistant_message,
-                "assistant_message_tool_name": assistant_message_tool_name,
-                "assistant_message_tool_kwarg": assistant_message_tool_kwarg,
-                "include_return_message_types": include_return_message_types,
-                "enable_thinking": enable_thinking,
-                "stream_tokens": stream_tokens,
-            },
+        async with self._raw_client.create_stream(
+            group_id,
+            messages=messages,
+            max_steps=max_steps,
+            use_assistant_message=use_assistant_message,
+            assistant_message_tool_name=assistant_message_tool_name,
+            assistant_message_tool_kwarg=assistant_message_tool_kwarg,
+            include_return_message_types=include_return_message_types,
+            enable_thinking=enable_thinking,
+            stream_tokens=stream_tokens,
+            include_pings=include_pings,
             request_options=request_options,
-            omit=OMIT,
-        ) as _response:
-            try:
-                if 200 <= _response.status_code < 300:
-                    _event_source = httpx_sse.EventSource(_response)
-                    async for _sse in _event_source.aiter_sse():
-                        try:
-                            yield typing.cast(
-                                LettaStreamingResponse,
-                                construct_type(
-                                    type_=LettaStreamingResponse,  # type: ignore
-                                    object_=json.loads(_sse.data),
-                                ),
-                            )
-                        except:
-                            pass
-                    return
-                await _response.aread()
-                if _response.status_code == 422:
-                    raise UnprocessableEntityError(
-                        typing.cast(
-                            HttpValidationError,
-                            construct_type(
-                                type_=HttpValidationError,  # type: ignore
-                                object_=_response.json(),
-                            ),
-                        )
-                    )
-                _response_json = _response.json()
-            except JSONDecodeError:
-                raise ApiError(status_code=_response.status_code, body=_response.text)
-            raise ApiError(status_code=_response.status_code, body=_response_json)
+        ) as r:
+            async for _chunk in r.data:
+                yield _chunk
 
     async def modify(
         self,
@@ -914,38 +707,10 @@ class AsyncMessagesClient:
 
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/groups/{jsonable_encoder(group_id)}/messages/{jsonable_encoder(message_id)}",
-            method="PATCH",
-            json=convert_and_respect_annotation_metadata(
-                object_=request, annotation=MessagesModifyRequest, direction="write"
-            ),
-            request_options=request_options,
-            omit=OMIT,
+        _response = await self._raw_client.modify(
+            group_id, message_id, request=request, request_options=request_options
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    MessagesModifyResponse,
-                    construct_type(
-                        type_=MessagesModifyResponse,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        HttpValidationError,
-                        construct_type(
-                            type_=HttpValidationError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
 
     async def reset(
         self, group_id: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -985,31 +750,5 @@ class AsyncMessagesClient:
 
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/groups/{jsonable_encoder(group_id)}/reset-messages",
-            method="PATCH",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.Optional[typing.Any],
-                    construct_type(
-                        type_=typing.Optional[typing.Any],  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        HttpValidationError,
-                        construct_type(
-                            type_=HttpValidationError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        _response = await self._raw_client.reset(group_id, request_options=request_options)
+        return _response.data