letta-client 0.1.226__py3-none-any.whl → 1.7.2__py3-none-any.whl

letta_client/agents/passages/client.py

@@ -1,793 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import typing
-from ...core.client_wrapper import 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
-
-# this is used as the default value for optional parameters
-OMIT = typing.cast(typing.Any, ...)
-
-
-class PassagesClient:
-    def __init__(self, *, client_wrapper: SyncClientWrapper):
-        self._client_wrapper = client_wrapper
-
-    def list(
-        self,
-        agent_id: str,
-        *,
-        after: typing.Optional[str] = None,
-        before: typing.Optional[str] = None,
-        limit: typing.Optional[int] = None,
-        search: typing.Optional[str] = None,
-        ascending: typing.Optional[bool] = None,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> typing.List[Passage]:
-        """
-        Retrieve the memories in an agent's archival memory store (paginated query).
-
-        Parameters
-        ----------
-        agent_id : str
-
-        after : typing.Optional[str]
-            Unique ID of the memory to start the query range at.
-
-        before : typing.Optional[str]
-            Unique ID of the memory to end the query range at.
-
-        limit : typing.Optional[int]
-            How many results to include in the response.
-
-        search : typing.Optional[str]
-            Search passages by text
-
-        ascending : typing.Optional[bool]
-            Whether to sort passages oldest to newest (True, default) or newest to oldest (False)
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        typing.List[Passage]
-            Successful Response
-
-        Examples
-        --------
-        from letta_client import Letta
-
-        client = Letta(
-            project="YOUR_PROJECT",
-            token="YOUR_TOKEN",
-        )
-        client.agents.passages.list(
-            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,
-            },
-            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)
-
-    def create(
-        self, agent_id: str, *, text: str, request_options: typing.Optional[RequestOptions] = None
-    ) -> typing.List[Passage]:
-        """
-        Insert a memory into an agent's archival memory store.
-
-        Parameters
-        ----------
-        agent_id : str
-
-        text : str
-            Text to write to archival memory.
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        typing.List[Passage]
-            Successful Response
-
-        Examples
-        --------
-        from letta_client import Letta
-
-        client = Letta(
-            project="YOUR_PROJECT",
-            token="YOUR_TOKEN",
-        )
-        client.agents.passages.create(
-            agent_id="agent_id",
-            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)
-
-    def delete(
-        self, agent_id: str, memory_id: str, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> typing.Optional[typing.Any]:
-        """
-        Delete a memory from an agent's archival memory store.
-
-        Parameters
-        ----------
-        agent_id : str
-
-        memory_id : str
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        typing.Optional[typing.Any]
-            Successful Response
-
-        Examples
-        --------
-        from letta_client import Letta
-
-        client = Letta(
-            project="YOUR_PROJECT",
-            token="YOUR_TOKEN",
-        )
-        client.agents.passages.delete(
-            agent_id="agent_id",
-            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)
-
-    def modify(
-        self,
-        agent_id: str,
-        memory_id: str,
-        *,
-        id: str,
-        created_by_id: typing.Optional[str] = OMIT,
-        last_updated_by_id: typing.Optional[str] = OMIT,
-        created_at: typing.Optional[dt.datetime] = OMIT,
-        updated_at: typing.Optional[dt.datetime] = OMIT,
-        is_deleted: typing.Optional[bool] = OMIT,
-        passage_update_agent_id: typing.Optional[str] = OMIT,
-        source_id: typing.Optional[str] = OMIT,
-        file_id: typing.Optional[str] = OMIT,
-        file_name: typing.Optional[str] = OMIT,
-        metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
-        text: typing.Optional[str] = OMIT,
-        embedding: typing.Optional[typing.Sequence[float]] = OMIT,
-        embedding_config: typing.Optional[EmbeddingConfig] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> typing.List[Passage]:
-        """
-        Modify a memory in the agent's archival memory store.
-
-        Parameters
-        ----------
-        agent_id : str
-
-        memory_id : str
-
-        id : str
-            The unique identifier of the passage.
-
-        created_by_id : typing.Optional[str]
-            The id of the user that made this object.
-
-        last_updated_by_id : typing.Optional[str]
-            The id of the user that made this object.
-
-        created_at : typing.Optional[dt.datetime]
-            The timestamp when the object was created.
-
-        updated_at : typing.Optional[dt.datetime]
-            The timestamp when the object was last updated.
-
-        is_deleted : typing.Optional[bool]
-            Whether this passage is deleted or not.
-
-        passage_update_agent_id : typing.Optional[str]
-            The unique identifier of the agent associated with the passage.
-
-        source_id : typing.Optional[str]
-            The data source of the passage.
-
-        file_id : typing.Optional[str]
-            The unique identifier of the file associated with the passage.
-
-        file_name : typing.Optional[str]
-            The name of the file (only for source passages).
-
-        metadata : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
-            The metadata of the passage.
-
-        text : typing.Optional[str]
-            The text of the passage.
-
-        embedding : typing.Optional[typing.Sequence[float]]
-            The embedding of the passage.
-
-        embedding_config : typing.Optional[EmbeddingConfig]
-            The embedding configuration used by the passage.
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        typing.List[Passage]
-            Successful Response
-
-        Examples
-        --------
-        from letta_client import Letta
-
-        client = Letta(
-            project="YOUR_PROJECT",
-            token="YOUR_TOKEN",
-        )
-        client.agents.passages.modify(
-            agent_id="agent_id",
-            memory_id="memory_id",
-            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",
-            },
-            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)
-
-
-class AsyncPassagesClient:
-    def __init__(self, *, client_wrapper: AsyncClientWrapper):
-        self._client_wrapper = client_wrapper
-
-    async def list(
-        self,
-        agent_id: str,
-        *,
-        after: typing.Optional[str] = None,
-        before: typing.Optional[str] = None,
-        limit: typing.Optional[int] = None,
-        search: typing.Optional[str] = None,
-        ascending: typing.Optional[bool] = None,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> typing.List[Passage]:
-        """
-        Retrieve the memories in an agent's archival memory store (paginated query).
-
-        Parameters
-        ----------
-        agent_id : str
-
-        after : typing.Optional[str]
-            Unique ID of the memory to start the query range at.
-
-        before : typing.Optional[str]
-            Unique ID of the memory to end the query range at.
-
-        limit : typing.Optional[int]
-            How many results to include in the response.
-
-        search : typing.Optional[str]
-            Search passages by text
-
-        ascending : typing.Optional[bool]
-            Whether to sort passages oldest to newest (True, default) or newest to oldest (False)
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        typing.List[Passage]
-            Successful Response
-
-        Examples
-        --------
-        import asyncio
-
-        from letta_client import AsyncLetta
-
-        client = AsyncLetta(
-            project="YOUR_PROJECT",
-            token="YOUR_TOKEN",
-        )
-
-
-        async def main() -> None:
-            await client.agents.passages.list(
-                agent_id="agent_id",
-            )
-
-
-        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,
-            },
-            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)
-
-    async def create(
-        self, agent_id: str, *, text: str, request_options: typing.Optional[RequestOptions] = None
-    ) -> typing.List[Passage]:
-        """
-        Insert a memory into an agent's archival memory store.
-
-        Parameters
-        ----------
-        agent_id : str
-
-        text : str
-            Text to write to archival memory.
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        typing.List[Passage]
-            Successful Response
-
-        Examples
-        --------
-        import asyncio
-
-        from letta_client import AsyncLetta
-
-        client = AsyncLetta(
-            project="YOUR_PROJECT",
-            token="YOUR_TOKEN",
-        )
-
-
-        async def main() -> None:
-            await client.agents.passages.create(
-                agent_id="agent_id",
-                text="text",
-            )
-
-
-        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)
-
-    async def delete(
-        self, agent_id: str, memory_id: str, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> typing.Optional[typing.Any]:
-        """
-        Delete a memory from an agent's archival memory store.
-
-        Parameters
-        ----------
-        agent_id : str
-
-        memory_id : str
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        typing.Optional[typing.Any]
-            Successful Response
-
-        Examples
-        --------
-        import asyncio
-
-        from letta_client import AsyncLetta
-
-        client = AsyncLetta(
-            project="YOUR_PROJECT",
-            token="YOUR_TOKEN",
-        )
-
-
-        async def main() -> None:
-            await client.agents.passages.delete(
-                agent_id="agent_id",
-                memory_id="memory_id",
-            )
-
-
-        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)
-
-    async def modify(
-        self,
-        agent_id: str,
-        memory_id: str,
-        *,
-        id: str,
-        created_by_id: typing.Optional[str] = OMIT,
-        last_updated_by_id: typing.Optional[str] = OMIT,
-        created_at: typing.Optional[dt.datetime] = OMIT,
-        updated_at: typing.Optional[dt.datetime] = OMIT,
-        is_deleted: typing.Optional[bool] = OMIT,
-        passage_update_agent_id: typing.Optional[str] = OMIT,
-        source_id: typing.Optional[str] = OMIT,
-        file_id: typing.Optional[str] = OMIT,
-        file_name: typing.Optional[str] = OMIT,
-        metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
-        text: typing.Optional[str] = OMIT,
-        embedding: typing.Optional[typing.Sequence[float]] = OMIT,
-        embedding_config: typing.Optional[EmbeddingConfig] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> typing.List[Passage]:
-        """
-        Modify a memory in the agent's archival memory store.
-
-        Parameters
-        ----------
-        agent_id : str
-
-        memory_id : str
-
-        id : str
-            The unique identifier of the passage.
-
-        created_by_id : typing.Optional[str]
-            The id of the user that made this object.
-
-        last_updated_by_id : typing.Optional[str]
-            The id of the user that made this object.
-
-        created_at : typing.Optional[dt.datetime]
-            The timestamp when the object was created.
-
-        updated_at : typing.Optional[dt.datetime]
-            The timestamp when the object was last updated.
-
-        is_deleted : typing.Optional[bool]
-            Whether this passage is deleted or not.
-
-        passage_update_agent_id : typing.Optional[str]
-            The unique identifier of the agent associated with the passage.
-
-        source_id : typing.Optional[str]
-            The data source of the passage.
-
-        file_id : typing.Optional[str]
-            The unique identifier of the file associated with the passage.
-
-        file_name : typing.Optional[str]
-            The name of the file (only for source passages).
-
-        metadata : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
-            The metadata of the passage.
-
-        text : typing.Optional[str]
-            The text of the passage.
-
-        embedding : typing.Optional[typing.Sequence[float]]
-            The embedding of the passage.
-
-        embedding_config : typing.Optional[EmbeddingConfig]
-            The embedding configuration used by the passage.
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        typing.List[Passage]
-            Successful Response
-
-        Examples
-        --------
-        import asyncio
-
-        from letta_client import AsyncLetta
-
-        client = AsyncLetta(
-            project="YOUR_PROJECT",
-            token="YOUR_TOKEN",
-        )
-
-
-        async def main() -> None:
-            await client.agents.passages.modify(
-                agent_id="agent_id",
-                memory_id="memory_id",
-                id="id",
-            )
-
-
-        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",
-            },
-            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)