letta-client 0.1.126__tar.gz → 0.1.128__tar.gz

{letta_client-0.1.126 → letta_client-0.1.128}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: letta-client
-Version: 0.1.126
+Version: 0.1.128
 Summary: 
 Requires-Python: >=3.8,<4.0
 Classifier: Intended Audience :: Developers

{letta_client-0.1.126 → letta_client-0.1.128}/pyproject.toml

@@ -3,7 +3,7 @@ name = "letta-client"
 
 [tool.poetry]
 name = "letta-client"
-version = "0.1.126"
+version = "0.1.128"
 description = ""
 readme = "README.md"
 authors = []

{letta_client-0.1.126 → letta_client-0.1.128}/src/letta_client/__init__.py

@@ -213,6 +213,8 @@ from .types import (
     UpdateUserMessage,
     UpdateUserMessageContent,
     UsageStatistics,
+    UsageStatisticsCompletionTokenDetails,
+    UsageStatisticsPromptTokenDetails,
     User,
     UserCreate,
     UserMessage,
@@ -531,6 +533,8 @@ __all__ = [
     "UpdateUserMessage",
     "UpdateUserMessageContent",
     "UsageStatistics",
+    "UsageStatisticsCompletionTokenDetails",
+    "UsageStatisticsPromptTokenDetails",
     "User",
     "UserCreate",
     "UserMessage",

{letta_client-0.1.126 → letta_client-0.1.128}/src/letta_client/agents/client.py

@@ -457,6 +457,58 @@ class AgentsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def count(self, *, request_options: typing.Optional[RequestOptions] = None) -> int:
+        """
+        Get the count of all agents associated with a given user.
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        int
+            Successful Response
+
+        Examples
+        --------
+        from letta_client import Letta
+
+        client = Letta(
+            token="YOUR_TOKEN",
+        )
+        client.agents.count()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "v1/agents/count",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    int,
+                    construct_type(
+                        type_=int,  # 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 export_file(self, agent_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> str:
         """
         Export the serialized JSON representation of an agent, formatted with indentation.
@@ -1400,6 +1452,66 @@ class AsyncAgentsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    async def count(self, *, request_options: typing.Optional[RequestOptions] = None) -> int:
+        """
+        Get the count of all agents associated with a given user.
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        int
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from letta_client import AsyncLetta
+
+        client = AsyncLetta(
+            token="YOUR_TOKEN",
+        )
+
+
+        async def main() -> None:
+            await client.agents.count()
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "v1/agents/count",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    int,
+                    construct_type(
+                        type_=int,  # 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 export_file(self, agent_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> str:
         """
         Export the serialized JSON representation of an agent, formatted with indentation.

{letta_client-0.1.126 → letta_client-0.1.128}/src/letta_client/core/client_wrapper.py

@@ -16,7 +16,7 @@ class BaseClientWrapper:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "letta-client",
-            "X-Fern-SDK-Version": "0.1.126",
+            "X-Fern-SDK-Version": "0.1.128",
         }
         if self.token is not None:
             headers["Authorization"] = f"Bearer {self.token}"

{letta_client-0.1.126 → letta_client-0.1.128}/src/letta_client/identities/client.py

@@ -322,6 +322,58 @@ class IdentitiesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def count(self, *, request_options: typing.Optional[RequestOptions] = None) -> int:
+        """
+        Get count of all identities for a user
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        int
+            Successful Response
+
+        Examples
+        --------
+        from letta_client import Letta
+
+        client = Letta(
+            token="YOUR_TOKEN",
+        )
+        client.identities.count()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "v1/identities/count",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    int,
+                    construct_type(
+                        type_=int,  # 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 retrieve(self, identity_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Identity:
         """
         Parameters
@@ -857,6 +909,66 @@ class AsyncIdentitiesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    async def count(self, *, request_options: typing.Optional[RequestOptions] = None) -> int:
+        """
+        Get count of all identities for a user
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        int
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from letta_client import AsyncLetta
+
+        client = AsyncLetta(
+            token="YOUR_TOKEN",
+        )
+
+
+        async def main() -> None:
+            await client.identities.count()
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "v1/identities/count",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    int,
+                    construct_type(
+                        type_=int,  # 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 retrieve(self, identity_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Identity:
         """
         Parameters

{letta_client-0.1.126 → letta_client-0.1.128}/src/letta_client/sources/client.py

@@ -5,13 +5,13 @@ from ..core.client_wrapper import SyncClientWrapper
 from .files.client import FilesClient
 from .passages.client import PassagesClient
 from ..core.request_options import RequestOptions
-from ..types.source import Source
-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.source import Source
+from ..core.jsonable_encoder import jsonable_encoder
 from ..types.embedding_config import EmbeddingConfig
 from ..core.serialization import convert_and_respect_annotation_metadata
 from ..core.client_wrapper import AsyncClientWrapper
@@ -28,6 +28,58 @@ class SourcesClient:
         self.files = FilesClient(client_wrapper=self._client_wrapper)
         self.passages = PassagesClient(client_wrapper=self._client_wrapper)
 
+    def count(self, *, request_options: typing.Optional[RequestOptions] = None) -> int:
+        """
+        Count all data sources created by a user.
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        int
+            Successful Response
+
+        Examples
+        --------
+        from letta_client import Letta
+
+        client = Letta(
+            token="YOUR_TOKEN",
+        )
+        client.sources.count()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "v1/sources/count",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    int,
+                    construct_type(
+                        type_=int,  # 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 retrieve(self, source_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Source:
         """
         Get all sources
@@ -442,6 +494,66 @@ class AsyncSourcesClient:
         self.files = AsyncFilesClient(client_wrapper=self._client_wrapper)
         self.passages = AsyncPassagesClient(client_wrapper=self._client_wrapper)
 
+    async def count(self, *, request_options: typing.Optional[RequestOptions] = None) -> int:
+        """
+        Count all data sources created by a user.
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        int
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from letta_client import AsyncLetta
+
+        client = AsyncLetta(
+            token="YOUR_TOKEN",
+        )
+
+
+        async def main() -> None:
+            await client.sources.count()
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "v1/sources/count",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    int,
+                    construct_type(
+                        type_=int,  # 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 retrieve(self, source_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Source:
         """
         Get all sources

{letta_client-0.1.126 → letta_client-0.1.128}/src/letta_client/tools/client.py

@@ -245,6 +245,69 @@ class ToolsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def count(
+        self,
+        *,
+        include_base_tools: typing.Optional[bool] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> int:
+        """
+        Get a count of all tools available to agents belonging to the org of the user.
+
+        Parameters
+        ----------
+        include_base_tools : typing.Optional[bool]
+            Include built-in Letta tools in the count
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        int
+            Successful Response
+
+        Examples
+        --------
+        from letta_client import Letta
+
+        client = Letta(
+            token="YOUR_TOKEN",
+        )
+        client.tools.count()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "v1/tools/count",
+            method="GET",
+            params={
+                "include_base_tools": include_base_tools,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    int,
+                    construct_type(
+                        type_=int,  # 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 list(
         self,
         *,
@@ -1381,6 +1444,77 @@ class AsyncToolsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    async def count(
+        self,
+        *,
+        include_base_tools: typing.Optional[bool] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> int:
+        """
+        Get a count of all tools available to agents belonging to the org of the user.
+
+        Parameters
+        ----------
+        include_base_tools : typing.Optional[bool]
+            Include built-in Letta tools in the count
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        int
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from letta_client import AsyncLetta
+
+        client = AsyncLetta(
+            token="YOUR_TOKEN",
+        )
+
+
+        async def main() -> None:
+            await client.tools.count()
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "v1/tools/count",
+            method="GET",
+            params={
+                "include_base_tools": include_base_tools,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    int,
+                    construct_type(
+                        type_=int,  # 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 list(
         self,
         *,

{letta_client-0.1.126 → letta_client-0.1.128}/src/letta_client/types/__init__.py

@@ -216,6 +216,8 @@ from .update_system_message import UpdateSystemMessage
 from .update_user_message import UpdateUserMessage
 from .update_user_message_content import UpdateUserMessageContent
 from .usage_statistics import UsageStatistics
+from .usage_statistics_completion_token_details import UsageStatisticsCompletionTokenDetails
+from .usage_statistics_prompt_token_details import UsageStatisticsPromptTokenDetails
 from .user import User
 from .user_create import UserCreate
 from .user_message import UserMessage
@@ -441,6 +443,8 @@ __all__ = [
     "UpdateUserMessage",
     "UpdateUserMessageContent",
     "UsageStatistics",
+    "UsageStatisticsCompletionTokenDetails",
+    "UsageStatisticsPromptTokenDetails",
     "User",
     "UserCreate",
     "UserMessage",

{letta_client-0.1.126 → letta_client-0.1.128}/src/letta_client/types/message_create.py

@@ -38,6 +38,11 @@ class MessageCreate(UncheckedBaseModel):
     The id of the sender of the message, can be an identity id or agent id
     """
 
+    group_id: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    The multi-agent group that the message was sent in
+    """
+
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
     else:

{letta_client-0.1.126 → letta_client-0.1.128}/src/letta_client/types/omitted_reasoning_content.py

@@ -2,16 +2,12 @@
 
 from ..core.unchecked_base_model import UncheckedBaseModel
 import typing
-import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
 
 
 class OmittedReasoningContent(UncheckedBaseModel):
     type: typing.Literal["omitted_reasoning"] = "omitted_reasoning"
-    tokens: int = pydantic.Field()
-    """
-    The reasoning token count for intermediate reasoning content.
-    """
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

{letta_client-0.1.126 → letta_client-0.1.128}/src/letta_client/types/usage_statistics.py

@@ -2,6 +2,8 @@
 
 from ..core.unchecked_base_model import UncheckedBaseModel
 import typing
+from .usage_statistics_prompt_token_details import UsageStatisticsPromptTokenDetails
+from .usage_statistics_completion_token_details import UsageStatisticsCompletionTokenDetails
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
 
@@ -10,6 +12,8 @@ class UsageStatistics(UncheckedBaseModel):
     completion_tokens: typing.Optional[int] = None
     prompt_tokens: typing.Optional[int] = None
     total_tokens: typing.Optional[int] = None
+    prompt_tokens_details: typing.Optional[UsageStatisticsPromptTokenDetails] = None
+    completion_tokens_details: typing.Optional[UsageStatisticsCompletionTokenDetails] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

letta_client-0.1.128/src/letta_client/types/usage_statistics_completion_token_details.py

@@ -0,0 +1,19 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.unchecked_base_model import UncheckedBaseModel
+import typing
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class UsageStatisticsCompletionTokenDetails(UncheckedBaseModel):
+    reasoning_tokens: typing.Optional[int] = None
+
+    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