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

letta_client/agents/messages/types/__init__.py

@@ -1,5 +1,7 @@
 # This file was auto-generated by Fern from our API Definition.
 
+# isort: skip_file
+
 from .letta_streaming_response import LettaStreamingResponse
 from .messages_modify_request import MessagesModifyRequest
 from .messages_modify_response import MessagesModifyResponse

letta_client/agents/messages/types/letta_streaming_response.py

@@ -1,15 +1,16 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from ....types.system_message import SystemMessage
-from ....types.user_message import UserMessage
-from ....types.reasoning_message import ReasoningMessage
-from ....types.tool_call_message import ToolCallMessage
-from ....types.tool_return_message import ToolReturnMessage
+
 from ....types.assistant_message import AssistantMessage
 from ....types.letta_ping import LettaPing
 from ....types.letta_stop_reason import LettaStopReason
 from ....types.letta_usage_statistics import LettaUsageStatistics
+from ....types.reasoning_message import ReasoningMessage
+from ....types.system_message import SystemMessage
+from ....types.tool_call_message import ToolCallMessage
+from ....types.tool_return_message import ToolReturnMessage
+from ....types.user_message import UserMessage
 
 LettaStreamingResponse = typing.Union[
     SystemMessage,

letta_client/agents/messages/types/messages_modify_request.py

@@ -1,10 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
+
+from ....types.update_assistant_message import UpdateAssistantMessage
+from ....types.update_reasoning_message import UpdateReasoningMessage
 from ....types.update_system_message import UpdateSystemMessage
 from ....types.update_user_message import UpdateUserMessage
-from ....types.update_reasoning_message import UpdateReasoningMessage
-from ....types.update_assistant_message import UpdateAssistantMessage
 
 MessagesModifyRequest = typing.Union[
     UpdateSystemMessage, UpdateUserMessage, UpdateReasoningMessage, UpdateAssistantMessage

letta_client/agents/messages/types/messages_modify_response.py

@@ -1,13 +1,14 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from ....types.system_message import SystemMessage
-from ....types.user_message import UserMessage
-from ....types.reasoning_message import ReasoningMessage
+
+from ....types.assistant_message import AssistantMessage
 from ....types.hidden_reasoning_message import HiddenReasoningMessage
+from ....types.reasoning_message import ReasoningMessage
+from ....types.system_message import SystemMessage
 from ....types.tool_call_message import ToolCallMessage
 from ....types.tool_return_message import ToolReturnMessage
-from ....types.assistant_message import AssistantMessage
+from ....types.user_message import UserMessage
 
 MessagesModifyResponse = typing.Union[
     SystemMessage,

letta_client/agents/messages/types/messages_preview_raw_payload_request.py

@@ -1,6 +1,7 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
+
 from ....types.letta_request import LettaRequest
 from ....types.letta_streaming_request import LettaStreamingRequest
 

letta_client/agents/passages/__init__.py

@@ -1,2 +1,4 @@
 # This file was auto-generated by Fern from our API Definition.
 
+# isort: skip_file
+

letta_client/agents/passages/client.py

@@ -1,19 +1,13 @@
 # This file was auto-generated by Fern from our API Definition.
 
+import datetime as dt
 import typing
-from ...core.client_wrapper import SyncClientWrapper
+
+from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.request_options import RequestOptions
-from ...types.passage import Passage
-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
-import datetime as dt
 from ...types.embedding_config import EmbeddingConfig
-from ...core.serialization import convert_and_respect_annotation_metadata
-from ...core.client_wrapper import AsyncClientWrapper
+from ...types.passage import Passage
+from .raw_client import AsyncRawPassagesClient, RawPassagesClient
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -21,7 +15,18 @@ OMIT = typing.cast(typing.Any, ...)
 
 class PassagesClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
-        self._client_wrapper = client_wrapper
+        self._raw_client = RawPassagesClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> RawPassagesClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        RawPassagesClient
+        """
+        return self._raw_client
 
     def list(
         self,
@@ -76,41 +81,16 @@ class PassagesClient:
             agent_id="agent_id",
         )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/archival-memory",
-            method="GET",
-            params={
-                "after": after,
-                "before": before,
-                "limit": limit,
-                "search": search,
-                "ascending": ascending,
-            },
+        _response = self._raw_client.list(
+            agent_id,
+            after=after,
+            before=before,
+            limit=limit,
+            search=search,
+            ascending=ascending,
             request_options=request_options,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.List[Passage],
-                    construct_type(
-                        type_=typing.List[Passage],  # 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, agent_id: str, *, text: str, request_options: typing.Optional[RequestOptions] = None
@@ -146,41 +126,8 @@ class PassagesClient:
             text="text",
         )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/archival-memory",
-            method="POST",
-            json={
-                "text": text,
-            },
-            headers={
-                "content-type": "application/json",
-            },
-            request_options=request_options,
-            omit=OMIT,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.List[Passage],
-                    construct_type(
-                        type_=typing.List[Passage],  # 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.create(agent_id, text=text, request_options=request_options)
+        return _response.data
 
     def delete(
         self, agent_id: str, memory_id: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -215,34 +162,8 @@ class PassagesClient:
             memory_id="memory_id",
         )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/archival-memory/{jsonable_encoder(memory_id)}",
-            method="DELETE",
-            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.delete(agent_id, memory_id, request_options=request_options)
+        return _response.data
 
     def modify(
         self,
@@ -338,61 +259,42 @@ class PassagesClient:
             id="id",
         )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/archival-memory/{jsonable_encoder(memory_id)}",
-            method="PATCH",
-            json={
-                "created_by_id": created_by_id,
-                "last_updated_by_id": last_updated_by_id,
-                "created_at": created_at,
-                "updated_at": updated_at,
-                "is_deleted": is_deleted,
-                "agent_id": passage_update_agent_id,
-                "source_id": source_id,
-                "file_id": file_id,
-                "file_name": file_name,
-                "metadata_": metadata,
-                "text": text,
-                "embedding": embedding,
-                "embedding_config": convert_and_respect_annotation_metadata(
-                    object_=embedding_config, annotation=EmbeddingConfig, direction="write"
-                ),
-                "id": id,
-            },
-            headers={
-                "content-type": "application/json",
-            },
+        _response = self._raw_client.modify(
+            agent_id,
+            memory_id,
+            id=id,
+            created_by_id=created_by_id,
+            last_updated_by_id=last_updated_by_id,
+            created_at=created_at,
+            updated_at=updated_at,
+            is_deleted=is_deleted,
+            passage_update_agent_id=passage_update_agent_id,
+            source_id=source_id,
+            file_id=file_id,
+            file_name=file_name,
+            metadata=metadata,
+            text=text,
+            embedding=embedding,
+            embedding_config=embedding_config,
             request_options=request_options,
-            omit=OMIT,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.List[Passage],
-                    construct_type(
-                        type_=typing.List[Passage],  # 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
 
 
 class AsyncPassagesClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
-        self._client_wrapper = client_wrapper
+        self._raw_client = AsyncRawPassagesClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> AsyncRawPassagesClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        AsyncRawPassagesClient
+        """
+        return self._raw_client
 
     async def list(
         self,
@@ -455,41 +357,16 @@ class AsyncPassagesClient:
 
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/archival-memory",
-            method="GET",
-            params={
-                "after": after,
-                "before": before,
-                "limit": limit,
-                "search": search,
-                "ascending": ascending,
-            },
+        _response = await self._raw_client.list(
+            agent_id,
+            after=after,
+            before=before,
+            limit=limit,
+            search=search,
+            ascending=ascending,
             request_options=request_options,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.List[Passage],
-                    construct_type(
-                        type_=typing.List[Passage],  # 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, agent_id: str, *, text: str, request_options: typing.Optional[RequestOptions] = None
@@ -533,41 +410,8 @@ class AsyncPassagesClient:
 
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/archival-memory",
-            method="POST",
-            json={
-                "text": text,
-            },
-            headers={
-                "content-type": "application/json",
-            },
-            request_options=request_options,
-            omit=OMIT,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.List[Passage],
-                    construct_type(
-                        type_=typing.List[Passage],  # 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.create(agent_id, text=text, request_options=request_options)
+        return _response.data
 
     async def delete(
         self, agent_id: str, memory_id: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -610,34 +454,8 @@ class AsyncPassagesClient:
 
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/archival-memory/{jsonable_encoder(memory_id)}",
-            method="DELETE",
-            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.delete(agent_id, memory_id, request_options=request_options)
+        return _response.data
 
     async def modify(
         self,
@@ -741,53 +559,23 @@ class AsyncPassagesClient:
 
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/archival-memory/{jsonable_encoder(memory_id)}",
-            method="PATCH",
-            json={
-                "created_by_id": created_by_id,
-                "last_updated_by_id": last_updated_by_id,
-                "created_at": created_at,
-                "updated_at": updated_at,
-                "is_deleted": is_deleted,
-                "agent_id": passage_update_agent_id,
-                "source_id": source_id,
-                "file_id": file_id,
-                "file_name": file_name,
-                "metadata_": metadata,
-                "text": text,
-                "embedding": embedding,
-                "embedding_config": convert_and_respect_annotation_metadata(
-                    object_=embedding_config, annotation=EmbeddingConfig, direction="write"
-                ),
-                "id": id,
-            },
-            headers={
-                "content-type": "application/json",
-            },
+        _response = await self._raw_client.modify(
+            agent_id,
+            memory_id,
+            id=id,
+            created_by_id=created_by_id,
+            last_updated_by_id=last_updated_by_id,
+            created_at=created_at,
+            updated_at=updated_at,
+            is_deleted=is_deleted,
+            passage_update_agent_id=passage_update_agent_id,
+            source_id=source_id,
+            file_id=file_id,
+            file_name=file_name,
+            metadata=metadata,
+            text=text,
+            embedding=embedding,
+            embedding_config=embedding_config,
             request_options=request_options,
-            omit=OMIT,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.List[Passage],
-                    construct_type(
-                        type_=typing.List[Passage],  # 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