letta-client 0.1.188__tar.gz → 0.1.190__tar.gz

{letta_client-0.1.188 → letta_client-0.1.190}/PKG-INFO

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

{letta_client-0.1.188 → letta_client-0.1.190}/pyproject.toml

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

{letta_client-0.1.188 → letta_client-0.1.190}/src/letta_client/agents/client.py

@@ -1061,6 +1061,75 @@ class AgentsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def cancel_agent_run(
+        self,
+        agent_id: str,
+        *,
+        request: typing.Optional[typing.Sequence[str]] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.Dict[str, typing.Optional[typing.Any]]:
+        """
+        Cancel runs associated with an agent. If run_ids are passed in, cancel those in particular.
+
+        Note to cancel active runs associated with an agent, redis is required.
+
+        Parameters
+        ----------
+        agent_id : str
+
+        request : typing.Optional[typing.Sequence[str]]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.Dict[str, typing.Optional[typing.Any]]
+            Successful Response
+
+        Examples
+        --------
+        from letta_client import Letta
+
+        client = Letta(
+            project="YOUR_PROJECT",
+            token="YOUR_TOKEN",
+        )
+        client.agents.cancel_agent_run(
+            agent_id="agent_id",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"v1/agents/{jsonable_encoder(agent_id)}/messages/cancel",
+            method="POST",
+            json=request,
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    typing.Dict[str, typing.Optional[typing.Any]],
+                    construct_type(
+                        type_=typing.Dict[str, 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 summarize_agent_conversation(
         self, agent_id: str, *, max_message_length: int, request_options: typing.Optional[RequestOptions] = None
     ) -> AgentState:
@@ -2298,6 +2367,83 @@ class AsyncAgentsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    async def cancel_agent_run(
+        self,
+        agent_id: str,
+        *,
+        request: typing.Optional[typing.Sequence[str]] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.Dict[str, typing.Optional[typing.Any]]:
+        """
+        Cancel runs associated with an agent. If run_ids are passed in, cancel those in particular.
+
+        Note to cancel active runs associated with an agent, redis is required.
+
+        Parameters
+        ----------
+        agent_id : str
+
+        request : typing.Optional[typing.Sequence[str]]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.Dict[str, 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.cancel_agent_run(
+                agent_id="agent_id",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"v1/agents/{jsonable_encoder(agent_id)}/messages/cancel",
+            method="POST",
+            json=request,
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    typing.Dict[str, typing.Optional[typing.Any]],
+                    construct_type(
+                        type_=typing.Dict[str, 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 summarize_agent_conversation(
         self, agent_id: str, *, max_message_length: int, request_options: typing.Optional[RequestOptions] = None
     ) -> AgentState:

{letta_client-0.1.188 → letta_client-0.1.190}/src/letta_client/agents/messages/client.py

@@ -454,6 +454,9 @@ class MessagesClient:
         Asynchronously process a user message and return a run object.
         The actual processing happens in the background, and the status can be checked using the run ID.
 
+        This is "asynchronous" in the sense that it's a background job and explicitly must be fetched by the run ID.
+        This is more like `send_message_job`
+
         Parameters
         ----------
         agent_id : str
@@ -1082,6 +1085,9 @@ class AsyncMessagesClient:
         Asynchronously process a user message and return a run object.
         The actual processing happens in the background, and the status can be checked using the run ID.
 
+        This is "asynchronous" in the sense that it's a background job and explicitly must be fetched by the run ID.
+        This is more like `send_message_job`
+
         Parameters
         ----------
         agent_id : str

{letta_client-0.1.188 → letta_client-0.1.190}/src/letta_client/client.py

@@ -8,6 +8,7 @@ from .base_client import AsyncLettaBase, LettaBase
 from .core.request_options import RequestOptions
 from .tools.client import ToolsClient as ToolsClientBase
 from .tools.client import AsyncToolsClient as AsyncToolsClientBase
+from .types.pip_requirement import PipRequirement
 from .types.tool import Tool
 
 # this is used as the default value for optional parameters
@@ -142,6 +143,7 @@ class ToolsClient(ToolsClientBase):
             typing.Dict[str, typing.Optional[typing.Any]]
         ] = OMIT,
         return_char_limit: typing.Optional[int] = OMIT,
+        pip_requirements: typing.Optional[typing.Sequence[PipRequirement]] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> Tool:
         """
@@ -170,6 +172,9 @@ class ToolsClient(ToolsClientBase):
         return_char_limit : typing.Optional[int]
             The maximum number of characters in the response.
 
+        pip_requirements : typing.Optional[typing.Sequence[PipRequirement]]
+            Optional list of pip packages required by this tool.
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -215,6 +220,7 @@ class ToolsClient(ToolsClientBase):
             source_type=source_type,
             json_schema=json_schema,
             return_char_limit=return_char_limit,
+            pip_requirements=pip_requirements,
             request_options=request_options,
         )
 
@@ -231,6 +237,7 @@ class ToolsClient(ToolsClientBase):
             typing.Dict[str, typing.Optional[typing.Any]]
         ] = OMIT,
         return_char_limit: typing.Optional[int] = OMIT,
+        pip_requirements: typing.Optional[typing.Sequence[PipRequirement]] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> Tool:
         """
@@ -259,6 +266,9 @@ class ToolsClient(ToolsClientBase):
         return_char_limit : typing.Optional[int]
             The maximum number of characters in the response.
 
+        pip_requirements : typing.Optional[typing.Sequence[PipRequirement]]
+            Optional list of pip packages required by this tool.
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -304,6 +314,7 @@ class ToolsClient(ToolsClientBase):
             source_type=source_type,
             json_schema=json_schema,
             return_char_limit=return_char_limit,
+            pip_requirements=pip_requirements,
             request_options=request_options,
         )
     
@@ -379,6 +390,7 @@ class ToolsClient(ToolsClientBase):
             source_type=tool.source_type or OMIT,
             json_schema=tool.json_schema or OMIT,
             return_char_limit=tool.return_char_limit or OMIT,
+            pip_requirements=tool.pip_requirements or OMIT,
             request_options=request_options,
         )
     
@@ -397,6 +409,7 @@ class AsyncToolsClient(AsyncToolsClientBase):
             typing.Dict[str, typing.Optional[typing.Any]]
         ] = OMIT,
         return_char_limit: typing.Optional[int] = OMIT,
+        pip_requirements: typing.Optional[typing.Sequence[PipRequirement]] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> Tool:
         """
@@ -425,6 +438,9 @@ class AsyncToolsClient(AsyncToolsClientBase):
         return_char_limit : typing.Optional[int]
             The maximum number of characters in the response.
 
+        pip_requirements : typing.Optional[typing.Sequence[PipRequirement]]
+            Optional list of pip packages required by this tool.
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -470,6 +486,7 @@ class AsyncToolsClient(AsyncToolsClientBase):
             source_type=source_type,
             json_schema=json_schema,
             return_char_limit=return_char_limit,
+            pip_requirements=pip_requirements,
             request_options=request_options,
         )
 
@@ -486,6 +503,7 @@ class AsyncToolsClient(AsyncToolsClientBase):
             typing.Dict[str, typing.Optional[typing.Any]]
         ] = OMIT,
         return_char_limit: typing.Optional[int] = OMIT,
+        pip_requirements: typing.Optional[typing.Sequence[PipRequirement]] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> Tool:
         """
@@ -514,6 +532,9 @@ class AsyncToolsClient(AsyncToolsClientBase):
         return_char_limit : typing.Optional[int]
             The maximum number of characters in the response.
 
+        pip_requirements : typing.Optional[typing.Sequence[PipRequirement]]
+            Optional list of pip packages required by this tool.
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -559,6 +580,7 @@ class AsyncToolsClient(AsyncToolsClientBase):
             source_type=source_type,
             json_schema=json_schema,
             return_char_limit=return_char_limit,
+            pip_requirements=pip_requirements,
             request_options=request_options,
         )
     
@@ -634,5 +656,6 @@ class AsyncToolsClient(AsyncToolsClientBase):
             source_type=tool.source_type or OMIT,
             json_schema=tool.json_schema or OMIT,
             return_char_limit=tool.return_char_limit or OMIT,
+            pip_requirements=tool.pip_requirements or OMIT,
             request_options=request_options,
         )

{letta_client-0.1.188 → letta_client-0.1.190}/src/letta_client/core/client_wrapper.py

@@ -24,7 +24,7 @@ class BaseClientWrapper:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "letta-client",
-            "X-Fern-SDK-Version": "0.1.188",
+            "X-Fern-SDK-Version": "0.1.190",
         }
         if self._project is not None:
             headers["X-Project"] = self._project

{letta_client-0.1.188 → letta_client-0.1.190}/src/letta_client/jobs/client.py

@@ -18,16 +18,36 @@ class JobsClient:
         self._client_wrapper = client_wrapper
 
     def list(
-        self, *, source_id: typing.Optional[str] = None, request_options: typing.Optional[RequestOptions] = None
+        self,
+        *,
+        source_id: typing.Optional[str] = None,
+        before: typing.Optional[str] = None,
+        after: typing.Optional[str] = None,
+        limit: typing.Optional[int] = None,
+        ascending: typing.Optional[bool] = None,
+        request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.List[Job]:
         """
         List all jobs.
+        TODO (cliandy): implementation for pagination
 
         Parameters
         ----------
         source_id : typing.Optional[str]
             Only list jobs associated with the source.
 
+        before : typing.Optional[str]
+            Cursor for pagination
+
+        after : typing.Optional[str]
+            Cursor for pagination
+
+        limit : typing.Optional[int]
+            Limit for pagination
+
+        ascending : typing.Optional[bool]
+            Whether to sort jobs oldest to newest (True, default) or newest to oldest (False)
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -51,6 +71,10 @@ class JobsClient:
             method="GET",
             params={
                 "source_id": source_id,
+                "before": before,
+                "after": after,
+                "limit": limit,
+                "ascending": ascending,
             },
             request_options=request_options,
         )
@@ -79,7 +103,14 @@ class JobsClient:
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
     def list_active(
-        self, *, source_id: typing.Optional[str] = None, request_options: typing.Optional[RequestOptions] = None
+        self,
+        *,
+        source_id: typing.Optional[str] = None,
+        before: typing.Optional[str] = None,
+        after: typing.Optional[str] = None,
+        limit: typing.Optional[int] = None,
+        ascending: typing.Optional[bool] = None,
+        request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.List[Job]:
         """
         List all active jobs.
@@ -89,6 +120,18 @@ class JobsClient:
         source_id : typing.Optional[str]
             Only list jobs associated with the source.
 
+        before : typing.Optional[str]
+            Cursor for pagination
+
+        after : typing.Optional[str]
+            Cursor for pagination
+
+        limit : typing.Optional[int]
+            Limit for pagination
+
+        ascending : typing.Optional[bool]
+            Whether to sort jobs oldest to newest (True, default) or newest to oldest (False)
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -112,6 +155,10 @@ class JobsClient:
             method="GET",
             params={
                 "source_id": source_id,
+                "before": before,
+                "after": after,
+                "limit": limit,
+                "ascending": ascending,
             },
             request_options=request_options,
         )
@@ -253,22 +300,102 @@ class JobsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def cancel_job(self, job_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Job:
+        """
+        Cancel a job by its job_id.
+
+        This endpoint marks a job as cancelled, which will cause any associated
+        agent execution to terminate as soon as possible.
+
+        Parameters
+        ----------
+        job_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        Job
+            Successful Response
+
+        Examples
+        --------
+        from letta_client import Letta
+
+        client = Letta(
+            project="YOUR_PROJECT",
+            token="YOUR_TOKEN",
+        )
+        client.jobs.cancel_job(
+            job_id="job_id",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"v1/jobs/{jsonable_encoder(job_id)}/cancel",
+            method="PATCH",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    Job,
+                    construct_type(
+                        type_=Job,  # 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 AsyncJobsClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
         self._client_wrapper = client_wrapper
 
     async def list(
-        self, *, source_id: typing.Optional[str] = None, request_options: typing.Optional[RequestOptions] = None
+        self,
+        *,
+        source_id: typing.Optional[str] = None,
+        before: typing.Optional[str] = None,
+        after: typing.Optional[str] = None,
+        limit: typing.Optional[int] = None,
+        ascending: typing.Optional[bool] = None,
+        request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.List[Job]:
         """
         List all jobs.
+        TODO (cliandy): implementation for pagination
 
         Parameters
         ----------
         source_id : typing.Optional[str]
             Only list jobs associated with the source.
 
+        before : typing.Optional[str]
+            Cursor for pagination
+
+        after : typing.Optional[str]
+            Cursor for pagination
+
+        limit : typing.Optional[int]
+            Limit for pagination
+
+        ascending : typing.Optional[bool]
+            Whether to sort jobs oldest to newest (True, default) or newest to oldest (False)
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -300,6 +427,10 @@ class AsyncJobsClient:
             method="GET",
             params={
                 "source_id": source_id,
+                "before": before,
+                "after": after,
+                "limit": limit,
+                "ascending": ascending,
             },
             request_options=request_options,
         )
@@ -328,7 +459,14 @@ class AsyncJobsClient:
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
     async def list_active(
-        self, *, source_id: typing.Optional[str] = None, request_options: typing.Optional[RequestOptions] = None
+        self,
+        *,
+        source_id: typing.Optional[str] = None,
+        before: typing.Optional[str] = None,
+        after: typing.Optional[str] = None,
+        limit: typing.Optional[int] = None,
+        ascending: typing.Optional[bool] = None,
+        request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.List[Job]:
         """
         List all active jobs.
@@ -338,6 +476,18 @@ class AsyncJobsClient:
         source_id : typing.Optional[str]
             Only list jobs associated with the source.
 
+        before : typing.Optional[str]
+            Cursor for pagination
+
+        after : typing.Optional[str]
+            Cursor for pagination
+
+        limit : typing.Optional[int]
+            Limit for pagination
+
+        ascending : typing.Optional[bool]
+            Whether to sort jobs oldest to newest (True, default) or newest to oldest (False)
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -369,6 +519,10 @@ class AsyncJobsClient:
             method="GET",
             params={
                 "source_id": source_id,
+                "before": before,
+                "after": after,
+                "limit": limit,
+                "ascending": ascending,
             },
             request_options=request_options,
         )
@@ -525,3 +679,71 @@ class AsyncJobsClient:
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    async def cancel_job(self, job_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Job:
+        """
+        Cancel a job by its job_id.
+
+        This endpoint marks a job as cancelled, which will cause any associated
+        agent execution to terminate as soon as possible.
+
+        Parameters
+        ----------
+        job_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        Job
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from letta_client import AsyncLetta
+
+        client = AsyncLetta(
+            project="YOUR_PROJECT",
+            token="YOUR_TOKEN",
+        )
+
+
+        async def main() -> None:
+            await client.jobs.cancel_job(
+                job_id="job_id",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"v1/jobs/{jsonable_encoder(job_id)}/cancel",
+            method="PATCH",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    Job,
+                    construct_type(
+                        type_=Job,  # 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)

{letta_client-0.1.188 → letta_client-0.1.190}/src/letta_client/types/file_metadata.py

@@ -68,6 +68,16 @@ class FileMetadata(UncheckedBaseModel):
     Optional error message if the file failed processing.
     """
 
+    total_chunks: typing.Optional[int] = pydantic.Field(default=None)
+    """
+    Total number of chunks for the file.
+    """
+
+    chunks_embedded: typing.Optional[int] = pydantic.Field(default=None)
+    """
+    Number of chunks that have been embedded.
+    """
+
     created_at: typing.Optional[dt.datetime] = pydantic.Field(default=None)
     """
     The creation date of the file.

{letta_client-0.1.188 → letta_client-0.1.190}/src/letta_client/types/stop_reason_type.py

@@ -3,5 +3,6 @@
 import typing
 
 StopReasonType = typing.Union[
-    typing.Literal["end_turn", "error", "invalid_tool_call", "max_steps", "no_tool_call", "tool_rule"], typing.Any
+    typing.Literal["end_turn", "error", "invalid_tool_call", "max_steps", "no_tool_call", "tool_rule", "cancelled"],
+    typing.Any,
 ]

{letta_client-0.1.188 → letta_client-0.1.190}/src/letta_client/types/update_ssemcp_server.py

@@ -26,6 +26,11 @@ class UpdateSsemcpServer(UncheckedBaseModel):
     The access token or API key for the MCP server (used for SSE authentication)
     """
 
+    custom_headers: typing.Optional[typing.Dict[str, typing.Optional[str]]] = pydantic.Field(default=None)
+    """
+    Custom authentication headers as key-value pairs
+    """
+
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
     else: