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

letta_client/agents/templates/client.py

@@ -1,588 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import typing
-from ...core.client_wrapper import SyncClientWrapper
-from .types.templates_create_version_request_return_agent_state import TemplatesCreateVersionRequestReturnAgentState
-from ...core.request_options import RequestOptions
-from ...core.jsonable_encoder import jsonable_encoder
-from ...errors.not_found_error import NotFoundError
-from ...core.unchecked_base_model import construct_type
-from ...errors.internal_server_error import InternalServerError
-from json.decoder import JSONDecodeError
-from ...core.api_error import ApiError
-from .types.templates_migrate_response import TemplatesMigrateResponse
-from ...errors.conflict_error import ConflictError
-from ...types.conflict_error_body import ConflictErrorBody
-from .types.templates_create_response import TemplatesCreateResponse
-from ...core.client_wrapper import AsyncClientWrapper
-
-# this is used as the default value for optional parameters
-OMIT = typing.cast(typing.Any, ...)
-
-
-class TemplatesClient:
-    def __init__(self, *, client_wrapper: SyncClientWrapper):
-        self._client_wrapper = client_wrapper
-
-    def create_version(
-        self,
-        agent_id: str,
-        *,
-        return_agent_state: typing.Optional[TemplatesCreateVersionRequestReturnAgentState] = None,
-        migrate_deployed_agents: typing.Optional[bool] = OMIT,
-        message: typing.Optional[str] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> None:
-        """
-        <Note>This endpoint is only available on Letta Cloud.</Note>
-
-        Creates a new version of the template version of the agent.
-
-        Parameters
-        ----------
-        agent_id : str
-            The agent ID of the agent to migrate, if this agent is not a template, it will create a agent template from the agent provided as well
-
-        return_agent_state : typing.Optional[TemplatesCreateVersionRequestReturnAgentState]
-
-        migrate_deployed_agents : typing.Optional[bool]
-
-        message : typing.Optional[str]
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        None
-
-        Examples
-        --------
-        from letta_client import Letta
-
-        client = Letta(
-            project="YOUR_PROJECT",
-            token="YOUR_TOKEN",
-        )
-        client.agents.templates.create_version(
-            agent_id="agent_id",
-        )
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/version-template",
-            method="POST",
-            params={
-                "returnAgentState": return_agent_state,
-            },
-            json={
-                "migrate_deployed_agents": migrate_deployed_agents,
-                "message": message,
-            },
-            headers={
-                "content-type": "application/json",
-            },
-            request_options=request_options,
-            omit=OMIT,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return
-            if _response.status_code == 404:
-                raise NotFoundError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        construct_type(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            if _response.status_code == 500:
-                raise InternalServerError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        construct_type(
-                            type_=typing.Optional[typing.Any],  # 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 migrate(
-        self,
-        agent_id: str,
-        *,
-        to_template: str,
-        preserve_core_memories: bool,
-        variables: typing.Optional[typing.Dict[str, str]] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> TemplatesMigrateResponse:
-        """
-        <Note>This endpoint is only available on Letta Cloud.</Note>
-
-        Migrate an agent to a new versioned agent template.
-
-        Parameters
-        ----------
-        agent_id : str
-
-        to_template : str
-
-        preserve_core_memories : bool
-
-        variables : typing.Optional[typing.Dict[str, str]]
-            If you chose to not preserve core memories, you should provide the new variables for the core memories
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        TemplatesMigrateResponse
-            200
-
-        Examples
-        --------
-        from letta_client import Letta
-
-        client = Letta(
-            project="YOUR_PROJECT",
-            token="YOUR_TOKEN",
-        )
-        client.agents.templates.migrate(
-            agent_id="agent_id",
-            to_template="to_template",
-            preserve_core_memories=True,
-        )
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/migrate",
-            method="POST",
-            json={
-                "to_template": to_template,
-                "variables": variables,
-                "preserve_core_memories": preserve_core_memories,
-            },
-            headers={
-                "content-type": "application/json",
-            },
-            request_options=request_options,
-            omit=OMIT,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    TemplatesMigrateResponse,
-                    construct_type(
-                        type_=TemplatesMigrateResponse,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 404:
-                raise NotFoundError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        construct_type(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            if _response.status_code == 409:
-                raise ConflictError(
-                    typing.cast(
-                        ConflictErrorBody,
-                        construct_type(
-                            type_=ConflictErrorBody,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            if _response.status_code == 500:
-                raise InternalServerError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        construct_type(
-                            type_=typing.Optional[typing.Any],  # 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,
-        *,
-        project: typing.Optional[str] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> TemplatesCreateResponse:
-        """
-        <Note>This endpoint is only available on Letta Cloud.</Note>
-
-        Creates a template from an agent.
-
-        Parameters
-        ----------
-        agent_id : str
-
-        project : typing.Optional[str]
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        TemplatesCreateResponse
-            201
-
-        Examples
-        --------
-        from letta_client import Letta
-
-        client = Letta(
-            project="YOUR_PROJECT",
-            token="YOUR_TOKEN",
-        )
-        client.agents.templates.create(
-            agent_id="agent_id",
-        )
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/template",
-            method="POST",
-            json={
-                "project": project,
-            },
-            headers={
-                "content-type": "application/json",
-            },
-            request_options=request_options,
-            omit=OMIT,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    TemplatesCreateResponse,
-                    construct_type(
-                        type_=TemplatesCreateResponse,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 500:
-                raise InternalServerError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        construct_type(
-                            type_=typing.Optional[typing.Any],  # 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 AsyncTemplatesClient:
-    def __init__(self, *, client_wrapper: AsyncClientWrapper):
-        self._client_wrapper = client_wrapper
-
-    async def create_version(
-        self,
-        agent_id: str,
-        *,
-        return_agent_state: typing.Optional[TemplatesCreateVersionRequestReturnAgentState] = None,
-        migrate_deployed_agents: typing.Optional[bool] = OMIT,
-        message: typing.Optional[str] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> None:
-        """
-        <Note>This endpoint is only available on Letta Cloud.</Note>
-
-        Creates a new version of the template version of the agent.
-
-        Parameters
-        ----------
-        agent_id : str
-            The agent ID of the agent to migrate, if this agent is not a template, it will create a agent template from the agent provided as well
-
-        return_agent_state : typing.Optional[TemplatesCreateVersionRequestReturnAgentState]
-
-        migrate_deployed_agents : typing.Optional[bool]
-
-        message : typing.Optional[str]
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        None
-
-        Examples
-        --------
-        import asyncio
-
-        from letta_client import AsyncLetta
-
-        client = AsyncLetta(
-            project="YOUR_PROJECT",
-            token="YOUR_TOKEN",
-        )
-
-
-        async def main() -> None:
-            await client.agents.templates.create_version(
-                agent_id="agent_id",
-            )
-
-
-        asyncio.run(main())
-        """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/version-template",
-            method="POST",
-            params={
-                "returnAgentState": return_agent_state,
-            },
-            json={
-                "migrate_deployed_agents": migrate_deployed_agents,
-                "message": message,
-            },
-            headers={
-                "content-type": "application/json",
-            },
-            request_options=request_options,
-            omit=OMIT,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return
-            if _response.status_code == 404:
-                raise NotFoundError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        construct_type(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            if _response.status_code == 500:
-                raise InternalServerError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        construct_type(
-                            type_=typing.Optional[typing.Any],  # 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 migrate(
-        self,
-        agent_id: str,
-        *,
-        to_template: str,
-        preserve_core_memories: bool,
-        variables: typing.Optional[typing.Dict[str, str]] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> TemplatesMigrateResponse:
-        """
-        <Note>This endpoint is only available on Letta Cloud.</Note>
-
-        Migrate an agent to a new versioned agent template.
-
-        Parameters
-        ----------
-        agent_id : str
-
-        to_template : str
-
-        preserve_core_memories : bool
-
-        variables : typing.Optional[typing.Dict[str, str]]
-            If you chose to not preserve core memories, you should provide the new variables for the core memories
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        TemplatesMigrateResponse
-            200
-
-        Examples
-        --------
-        import asyncio
-
-        from letta_client import AsyncLetta
-
-        client = AsyncLetta(
-            project="YOUR_PROJECT",
-            token="YOUR_TOKEN",
-        )
-
-
-        async def main() -> None:
-            await client.agents.templates.migrate(
-                agent_id="agent_id",
-                to_template="to_template",
-                preserve_core_memories=True,
-            )
-
-
-        asyncio.run(main())
-        """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/migrate",
-            method="POST",
-            json={
-                "to_template": to_template,
-                "variables": variables,
-                "preserve_core_memories": preserve_core_memories,
-            },
-            headers={
-                "content-type": "application/json",
-            },
-            request_options=request_options,
-            omit=OMIT,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    TemplatesMigrateResponse,
-                    construct_type(
-                        type_=TemplatesMigrateResponse,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 404:
-                raise NotFoundError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        construct_type(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            if _response.status_code == 409:
-                raise ConflictError(
-                    typing.cast(
-                        ConflictErrorBody,
-                        construct_type(
-                            type_=ConflictErrorBody,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            if _response.status_code == 500:
-                raise InternalServerError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        construct_type(
-                            type_=typing.Optional[typing.Any],  # 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,
-        *,
-        project: typing.Optional[str] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> TemplatesCreateResponse:
-        """
-        <Note>This endpoint is only available on Letta Cloud.</Note>
-
-        Creates a template from an agent.
-
-        Parameters
-        ----------
-        agent_id : str
-
-        project : typing.Optional[str]
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        TemplatesCreateResponse
-            201
-
-        Examples
-        --------
-        import asyncio
-
-        from letta_client import AsyncLetta
-
-        client = AsyncLetta(
-            project="YOUR_PROJECT",
-            token="YOUR_TOKEN",
-        )
-
-
-        async def main() -> None:
-            await client.agents.templates.create(
-                agent_id="agent_id",
-            )
-
-
-        asyncio.run(main())
-        """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/template",
-            method="POST",
-            json={
-                "project": project,
-            },
-            headers={
-                "content-type": "application/json",
-            },
-            request_options=request_options,
-            omit=OMIT,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    TemplatesCreateResponse,
-                    construct_type(
-                        type_=TemplatesCreateResponse,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 500:
-                raise InternalServerError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        construct_type(
-                            type_=typing.Optional[typing.Any],  # 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)

letta_client/agents/templates/types/__init__.py

@@ -1,7 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-from .templates_create_response import TemplatesCreateResponse
-from .templates_create_version_request_return_agent_state import TemplatesCreateVersionRequestReturnAgentState
-from .templates_migrate_response import TemplatesMigrateResponse
-
-__all__ = ["TemplatesCreateResponse", "TemplatesCreateVersionRequestReturnAgentState", "TemplatesMigrateResponse"]

letta_client/agents/templates/types/templates_create_response.py

@@ -1,22 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-from ....core.unchecked_base_model import UncheckedBaseModel
-import typing_extensions
-from ....core.serialization import FieldMetadata
-from ....core.pydantic_utilities import IS_PYDANTIC_V2
-import typing
-import pydantic
-
-
-class TemplatesCreateResponse(UncheckedBaseModel):
-    template_name: typing_extensions.Annotated[str, FieldMetadata(alias="templateName")]
-    template_id: typing_extensions.Annotated[str, FieldMetadata(alias="templateId")]
-
-    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

letta_client/agents/templates/types/templates_create_version_request_return_agent_state.py

@@ -1,5 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import typing
-
-TemplatesCreateVersionRequestReturnAgentState = typing.Union[typing.Literal["true"], typing.Any]

letta_client/agents/templates/types/templates_migrate_response.py

@@ -1,19 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-from ....core.unchecked_base_model import UncheckedBaseModel
-from ....core.pydantic_utilities import IS_PYDANTIC_V2
-import typing
-import pydantic
-
-
-class TemplatesMigrateResponse(UncheckedBaseModel):
-    success: bool
-
-    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

letta_client/agents/tools/__init__.py

@@ -1,2 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-