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

letta_client/__init__.py

@@ -335,8 +335,10 @@ from .tools import (
     AddMcpServerRequest,
     AddMcpServerResponseItem,
     ConnectMcpServerRequest,
+    ConnectMcpServerResponseEvent,
     DeleteMcpServerResponseItem,
     ListMcpServersResponseValue,
+    StreamingResponse,
     TestMcpServerRequest,
     UpdateMcpServerRequest,
     UpdateMcpServerResponse,
@@ -451,6 +453,7 @@ __all__ = [
     "ConflictError",
     "ConflictErrorBody",
     "ConnectMcpServerRequest",
+    "ConnectMcpServerResponseEvent",
     "ContextWindowOverview",
     "ContinueToolRule",
     "CoreMemoryBlockSchema",
@@ -597,6 +600,7 @@ __all__ = [
     "StepsListRequestFeedback",
     "StopReasonType",
     "StreamableHttpServerConfig",
+    "StreamingResponse",
     "SupervisorManager",
     "SupervisorManagerUpdate",
     "SystemMessage",

letta_client/agents/messages/client.py

@@ -146,6 +146,7 @@ class MessagesClient:
         assistant_message_tool_name: typing.Optional[str] = OMIT,
         assistant_message_tool_kwarg: typing.Optional[str] = OMIT,
         include_return_message_types: typing.Optional[typing.Sequence[MessageType]] = OMIT,
+        enable_thinking: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> LettaResponse:
         """
@@ -174,6 +175,9 @@ class MessagesClient:
         include_return_message_types : typing.Optional[typing.Sequence[MessageType]]
             Only return specified message types in the response. If `None` (default) returns all messages.
 
+        enable_thinking : typing.Optional[str]
+            If set to True, enables reasoning before responses or tool calls from the agent.
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -216,6 +220,7 @@ class MessagesClient:
                 "assistant_message_tool_name": assistant_message_tool_name,
                 "assistant_message_tool_kwarg": assistant_message_tool_kwarg,
                 "include_return_message_types": include_return_message_types,
+                "enable_thinking": enable_thinking,
             },
             request_options=request_options,
             omit=OMIT,
@@ -330,6 +335,7 @@ class MessagesClient:
         assistant_message_tool_name: typing.Optional[str] = OMIT,
         assistant_message_tool_kwarg: typing.Optional[str] = OMIT,
         include_return_message_types: typing.Optional[typing.Sequence[MessageType]] = OMIT,
+        enable_thinking: typing.Optional[str] = OMIT,
         stream_tokens: typing.Optional[bool] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.Iterator[LettaStreamingResponse]:
@@ -360,6 +366,9 @@ class MessagesClient:
         include_return_message_types : typing.Optional[typing.Sequence[MessageType]]
             Only return specified message types in the response. If `None` (default) returns all messages.
 
+        enable_thinking : typing.Optional[str]
+            If set to True, enables reasoning before responses or tool calls from the agent.
+
         stream_tokens : typing.Optional[bool]
             Flag to determine if individual tokens should be streamed. Set to True for token streaming (requires stream_steps = True).
 
@@ -407,6 +416,7 @@ class MessagesClient:
                 "assistant_message_tool_name": assistant_message_tool_name,
                 "assistant_message_tool_kwarg": assistant_message_tool_kwarg,
                 "include_return_message_types": include_return_message_types,
+                "enable_thinking": enable_thinking,
                 "stream_tokens": stream_tokens,
             },
             request_options=request_options,
@@ -522,6 +532,7 @@ class MessagesClient:
         assistant_message_tool_name: typing.Optional[str] = OMIT,
         assistant_message_tool_kwarg: typing.Optional[str] = OMIT,
         include_return_message_types: typing.Optional[typing.Sequence[MessageType]] = OMIT,
+        enable_thinking: typing.Optional[str] = OMIT,
         callback_url: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> Run:
@@ -554,6 +565,9 @@ class MessagesClient:
         include_return_message_types : typing.Optional[typing.Sequence[MessageType]]
             Only return specified message types in the response. If `None` (default) returns all messages.
 
+        enable_thinking : typing.Optional[str]
+            If set to True, enables reasoning before responses or tool calls from the agent.
+
         callback_url : typing.Optional[str]
             Optional callback URL to POST to when the job completes
 
@@ -599,6 +613,7 @@ class MessagesClient:
                 "assistant_message_tool_name": assistant_message_tool_name,
                 "assistant_message_tool_kwarg": assistant_message_tool_kwarg,
                 "include_return_message_types": include_return_message_types,
+                "enable_thinking": enable_thinking,
                 "callback_url": callback_url,
             },
             headers={
@@ -912,6 +927,7 @@ class AsyncMessagesClient:
         assistant_message_tool_name: typing.Optional[str] = OMIT,
         assistant_message_tool_kwarg: typing.Optional[str] = OMIT,
         include_return_message_types: typing.Optional[typing.Sequence[MessageType]] = OMIT,
+        enable_thinking: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> LettaResponse:
         """
@@ -940,6 +956,9 @@ class AsyncMessagesClient:
         include_return_message_types : typing.Optional[typing.Sequence[MessageType]]
             Only return specified message types in the response. If `None` (default) returns all messages.
 
+        enable_thinking : typing.Optional[str]
+            If set to True, enables reasoning before responses or tool calls from the agent.
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -990,6 +1009,7 @@ class AsyncMessagesClient:
                 "assistant_message_tool_name": assistant_message_tool_name,
                 "assistant_message_tool_kwarg": assistant_message_tool_kwarg,
                 "include_return_message_types": include_return_message_types,
+                "enable_thinking": enable_thinking,
             },
             request_options=request_options,
             omit=OMIT,
@@ -1112,6 +1132,7 @@ class AsyncMessagesClient:
         assistant_message_tool_name: typing.Optional[str] = OMIT,
         assistant_message_tool_kwarg: typing.Optional[str] = OMIT,
         include_return_message_types: typing.Optional[typing.Sequence[MessageType]] = OMIT,
+        enable_thinking: typing.Optional[str] = OMIT,
         stream_tokens: typing.Optional[bool] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.AsyncIterator[LettaStreamingResponse]:
@@ -1142,6 +1163,9 @@ class AsyncMessagesClient:
         include_return_message_types : typing.Optional[typing.Sequence[MessageType]]
             Only return specified message types in the response. If `None` (default) returns all messages.
 
+        enable_thinking : typing.Optional[str]
+            If set to True, enables reasoning before responses or tool calls from the agent.
+
         stream_tokens : typing.Optional[bool]
             Flag to determine if individual tokens should be streamed. Set to True for token streaming (requires stream_steps = True).
 
@@ -1197,6 +1221,7 @@ class AsyncMessagesClient:
                 "assistant_message_tool_name": assistant_message_tool_name,
                 "assistant_message_tool_kwarg": assistant_message_tool_kwarg,
                 "include_return_message_types": include_return_message_types,
+                "enable_thinking": enable_thinking,
                 "stream_tokens": stream_tokens,
             },
             request_options=request_options,
@@ -1320,6 +1345,7 @@ class AsyncMessagesClient:
         assistant_message_tool_name: typing.Optional[str] = OMIT,
         assistant_message_tool_kwarg: typing.Optional[str] = OMIT,
         include_return_message_types: typing.Optional[typing.Sequence[MessageType]] = OMIT,
+        enable_thinking: typing.Optional[str] = OMIT,
         callback_url: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> Run:
@@ -1352,6 +1378,9 @@ class AsyncMessagesClient:
         include_return_message_types : typing.Optional[typing.Sequence[MessageType]]
             Only return specified message types in the response. If `None` (default) returns all messages.
 
+        enable_thinking : typing.Optional[str]
+            If set to True, enables reasoning before responses or tool calls from the agent.
+
         callback_url : typing.Optional[str]
             Optional callback URL to POST to when the job completes
 
@@ -1405,6 +1434,7 @@ class AsyncMessagesClient:
                 "assistant_message_tool_name": assistant_message_tool_name,
                 "assistant_message_tool_kwarg": assistant_message_tool_kwarg,
                 "include_return_message_types": include_return_message_types,
+                "enable_thinking": enable_thinking,
                 "callback_url": callback_url,
             },
             headers={

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.224",
+            "X-Fern-SDK-Version": "0.1.226",
         }
         if self._project is not None:
             headers["X-Project"] = self._project

letta_client/groups/messages/client.py

@@ -133,6 +133,7 @@ class MessagesClient:
         assistant_message_tool_name: typing.Optional[str] = OMIT,
         assistant_message_tool_kwarg: typing.Optional[str] = OMIT,
         include_return_message_types: typing.Optional[typing.Sequence[MessageType]] = OMIT,
+        enable_thinking: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> LettaResponse:
         """
@@ -161,6 +162,9 @@ class MessagesClient:
         include_return_message_types : typing.Optional[typing.Sequence[MessageType]]
             Only return specified message types in the response. If `None` (default) returns all messages.
 
+        enable_thinking : typing.Optional[str]
+            If set to True, enables reasoning before responses or tool calls from the agent.
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -203,6 +207,7 @@ class MessagesClient:
                 "assistant_message_tool_name": assistant_message_tool_name,
                 "assistant_message_tool_kwarg": assistant_message_tool_kwarg,
                 "include_return_message_types": include_return_message_types,
+                "enable_thinking": enable_thinking,
             },
             request_options=request_options,
             omit=OMIT,
@@ -241,6 +246,7 @@ class MessagesClient:
         assistant_message_tool_name: typing.Optional[str] = OMIT,
         assistant_message_tool_kwarg: typing.Optional[str] = OMIT,
         include_return_message_types: typing.Optional[typing.Sequence[MessageType]] = OMIT,
+        enable_thinking: typing.Optional[str] = OMIT,
         stream_tokens: typing.Optional[bool] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.Iterator[LettaStreamingResponse]:
@@ -271,6 +277,9 @@ class MessagesClient:
         include_return_message_types : typing.Optional[typing.Sequence[MessageType]]
             Only return specified message types in the response. If `None` (default) returns all messages.
 
+        enable_thinking : typing.Optional[str]
+            If set to True, enables reasoning before responses or tool calls from the agent.
+
         stream_tokens : typing.Optional[bool]
             Flag to determine if individual tokens should be streamed. Set to True for token streaming (requires stream_steps = True).
 
@@ -318,6 +327,7 @@ class MessagesClient:
                 "assistant_message_tool_name": assistant_message_tool_name,
                 "assistant_message_tool_kwarg": assistant_message_tool_kwarg,
                 "include_return_message_types": include_return_message_types,
+                "enable_thinking": enable_thinking,
                 "stream_tokens": stream_tokens,
             },
             request_options=request_options,
@@ -606,6 +616,7 @@ class AsyncMessagesClient:
         assistant_message_tool_name: typing.Optional[str] = OMIT,
         assistant_message_tool_kwarg: typing.Optional[str] = OMIT,
         include_return_message_types: typing.Optional[typing.Sequence[MessageType]] = OMIT,
+        enable_thinking: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> LettaResponse:
         """
@@ -634,6 +645,9 @@ class AsyncMessagesClient:
         include_return_message_types : typing.Optional[typing.Sequence[MessageType]]
             Only return specified message types in the response. If `None` (default) returns all messages.
 
+        enable_thinking : typing.Optional[str]
+            If set to True, enables reasoning before responses or tool calls from the agent.
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -684,6 +698,7 @@ class AsyncMessagesClient:
                 "assistant_message_tool_name": assistant_message_tool_name,
                 "assistant_message_tool_kwarg": assistant_message_tool_kwarg,
                 "include_return_message_types": include_return_message_types,
+                "enable_thinking": enable_thinking,
             },
             request_options=request_options,
             omit=OMIT,
@@ -722,6 +737,7 @@ class AsyncMessagesClient:
         assistant_message_tool_name: typing.Optional[str] = OMIT,
         assistant_message_tool_kwarg: typing.Optional[str] = OMIT,
         include_return_message_types: typing.Optional[typing.Sequence[MessageType]] = OMIT,
+        enable_thinking: typing.Optional[str] = OMIT,
         stream_tokens: typing.Optional[bool] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.AsyncIterator[LettaStreamingResponse]:
@@ -752,6 +768,9 @@ class AsyncMessagesClient:
         include_return_message_types : typing.Optional[typing.Sequence[MessageType]]
             Only return specified message types in the response. If `None` (default) returns all messages.
 
+        enable_thinking : typing.Optional[str]
+            If set to True, enables reasoning before responses or tool calls from the agent.
+
         stream_tokens : typing.Optional[bool]
             Flag to determine if individual tokens should be streamed. Set to True for token streaming (requires stream_steps = True).
 
@@ -807,6 +826,7 @@ class AsyncMessagesClient:
                 "assistant_message_tool_name": assistant_message_tool_name,
                 "assistant_message_tool_kwarg": assistant_message_tool_kwarg,
                 "include_return_message_types": include_return_message_types,
+                "enable_thinking": enable_thinking,
                 "stream_tokens": stream_tokens,
             },
             request_options=request_options,

letta_client/tools/__init__.py

@@ -4,8 +4,10 @@ from .types import (
     AddMcpServerRequest,
     AddMcpServerResponseItem,
     ConnectMcpServerRequest,
+    ConnectMcpServerResponseEvent,
     DeleteMcpServerResponseItem,
     ListMcpServersResponseValue,
+    StreamingResponse,
     TestMcpServerRequest,
     UpdateMcpServerRequest,
     UpdateMcpServerResponse,
@@ -15,8 +17,10 @@ __all__ = [
     "AddMcpServerRequest",
     "AddMcpServerResponseItem",
     "ConnectMcpServerRequest",
+    "ConnectMcpServerResponseEvent",
     "DeleteMcpServerResponseItem",
     "ListMcpServersResponseValue",
+    "StreamingResponse",
     "TestMcpServerRequest",
     "UpdateMcpServerRequest",
     "UpdateMcpServerResponse",

letta_client/tools/client.py

@@ -24,6 +24,9 @@ from .types.update_mcp_server_request import UpdateMcpServerRequest
 from .types.update_mcp_server_response import UpdateMcpServerResponse
 from .types.test_mcp_server_request import TestMcpServerRequest
 from .types.connect_mcp_server_request import ConnectMcpServerRequest
+from .types.streaming_response import StreamingResponse
+import httpx_sse
+import json
 from ..core.client_wrapper import AsyncClientWrapper
 
 # this is used as the default value for optional parameters
@@ -1324,7 +1327,7 @@ class ToolsClient:
     ) -> typing.Optional[typing.Any]:
         """
         Test connection to an MCP server without adding it.
-        Returns the list of available tools if successful, or OAuth information if OAuth is required.
+        Returns the list of available tools if successful.
 
         Parameters
         ----------
@@ -1389,7 +1392,7 @@ class ToolsClient:
 
     def connect_mcp_server(
         self, *, request: ConnectMcpServerRequest, request_options: typing.Optional[RequestOptions] = None
-    ) -> typing.Optional[typing.Any]:
+    ) -> typing.Iterator[StreamingResponse]:
         """
         Connect to an MCP server with support for OAuth via SSE.
         Returns a stream of events handling authorization state and exchange if OAuth is required.
@@ -1401,9 +1404,9 @@ class ToolsClient:
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
-        Returns
-        -------
-        typing.Optional[typing.Any]
+        Yields
+        ------
+        typing.Iterator[StreamingResponse]
             Successful response
 
         Examples
@@ -1414,15 +1417,17 @@ class ToolsClient:
             project="YOUR_PROJECT",
             token="YOUR_TOKEN",
         )
-        client.tools.connect_mcp_server(
+        response = client.tools.connect_mcp_server(
             request=StdioServerConfig(
                 server_name="server_name",
                 command="command",
                 args=["args"],
             ),
         )
+        for chunk in response:
+            yield chunk
         """
-        _response = self._client_wrapper.httpx_client.request(
+        with self._client_wrapper.httpx_client.stream(
             "v1/tools/mcp/servers/connect",
             method="POST",
             json=convert_and_respect_annotation_metadata(
@@ -1430,107 +1435,37 @@ class ToolsClient:
             ),
             request_options=request_options,
             omit=OMIT,
-        )
-        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(),
-                        ),
+        ) as _response:
+            try:
+                if 200 <= _response.status_code < 300:
+                    _event_source = httpx_sse.EventSource(_response)
+                    for _sse in _event_source.iter_sse():
+                        try:
+                            yield typing.cast(
+                                StreamingResponse,
+                                construct_type(
+                                    type_=StreamingResponse,  # type: ignore
+                                    object_=json.loads(_sse.data),
+                                ),
+                            )
+                        except:
+                            pass
+                    return
+                _response.read()
+                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 mcp_oauth_callback(
-        self,
-        session_id: str,
-        *,
-        code: typing.Optional[str] = None,
-        state: typing.Optional[str] = None,
-        error: typing.Optional[str] = None,
-        error_description: typing.Optional[str] = None,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> None:
-        """
-        Handle OAuth callback for MCP server authentication.
-
-        Parameters
-        ----------
-        session_id : str
-
-        code : typing.Optional[str]
-            OAuth authorization code
-
-        state : typing.Optional[str]
-            OAuth state parameter
-
-        error : typing.Optional[str]
-            OAuth error
-
-        error_description : typing.Optional[str]
-            OAuth error description
-
-        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.tools.mcp_oauth_callback(
-            session_id="session_id",
-        )
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/tools/mcp/oauth/callback/{jsonable_encoder(session_id)}",
-            method="GET",
-            params={
-                "code": code,
-                "state": state,
-                "error": error,
-                "error_description": error_description,
-            },
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        HttpValidationError,
-                        construct_type(
-                            type_=HttpValidationError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+                _response_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 AsyncToolsClient:
@@ -2971,7 +2906,7 @@ class AsyncToolsClient:
     ) -> typing.Optional[typing.Any]:
         """
         Test connection to an MCP server without adding it.
-        Returns the list of available tools if successful, or OAuth information if OAuth is required.
+        Returns the list of available tools if successful.
 
         Parameters
         ----------
@@ -3044,7 +2979,7 @@ class AsyncToolsClient:
 
     async def connect_mcp_server(
         self, *, request: ConnectMcpServerRequest, request_options: typing.Optional[RequestOptions] = None
-    ) -> typing.Optional[typing.Any]:
+    ) -> typing.AsyncIterator[StreamingResponse]:
         """
         Connect to an MCP server with support for OAuth via SSE.
         Returns a stream of events handling authorization state and exchange if OAuth is required.
@@ -3056,9 +2991,9 @@ class AsyncToolsClient:
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
-        Returns
-        -------
-        typing.Optional[typing.Any]
+        Yields
+        ------
+        typing.AsyncIterator[StreamingResponse]
             Successful response
 
         Examples
@@ -3074,18 +3009,20 @@ class AsyncToolsClient:
 
 
         async def main() -> None:
-            await client.tools.connect_mcp_server(
+            response = await client.tools.connect_mcp_server(
                 request=StdioServerConfig(
                     server_name="server_name",
                     command="command",
                     args=["args"],
                 ),
             )
+            async for chunk in response:
+                yield chunk
 
 
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
+        async with self._client_wrapper.httpx_client.stream(
             "v1/tools/mcp/servers/connect",
             method="POST",
             json=convert_and_respect_annotation_metadata(
@@ -3093,112 +3030,34 @@ class AsyncToolsClient:
             ),
             request_options=request_options,
             omit=OMIT,
-        )
-        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(),
-                        ),
+        ) as _response:
+            try:
+                if 200 <= _response.status_code < 300:
+                    _event_source = httpx_sse.EventSource(_response)
+                    async for _sse in _event_source.aiter_sse():
+                        try:
+                            yield typing.cast(
+                                StreamingResponse,
+                                construct_type(
+                                    type_=StreamingResponse,  # type: ignore
+                                    object_=json.loads(_sse.data),
+                                ),
+                            )
+                        except:
+                            pass
+                    return
+                await _response.aread()
+                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 mcp_oauth_callback(
-        self,
-        session_id: str,
-        *,
-        code: typing.Optional[str] = None,
-        state: typing.Optional[str] = None,
-        error: typing.Optional[str] = None,
-        error_description: typing.Optional[str] = None,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> None:
-        """
-        Handle OAuth callback for MCP server authentication.
-
-        Parameters
-        ----------
-        session_id : str
-
-        code : typing.Optional[str]
-            OAuth authorization code
-
-        state : typing.Optional[str]
-            OAuth state parameter
-
-        error : typing.Optional[str]
-            OAuth error
-
-        error_description : typing.Optional[str]
-            OAuth error description
-
-        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.tools.mcp_oauth_callback(
-                session_id="session_id",
-            )
-
-
-        asyncio.run(main())
-        """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/tools/mcp/oauth/callback/{jsonable_encoder(session_id)}",
-            method="GET",
-            params={
-                "code": code,
-                "state": state,
-                "error": error,
-                "error_description": error_description,
-            },
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        HttpValidationError,
-                        construct_type(
-                            type_=HttpValidationError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+                _response_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/tools/types/__init__.py

@@ -3,8 +3,10 @@
 from .add_mcp_server_request import AddMcpServerRequest
 from .add_mcp_server_response_item import AddMcpServerResponseItem
 from .connect_mcp_server_request import ConnectMcpServerRequest
+from .connect_mcp_server_response_event import ConnectMcpServerResponseEvent
 from .delete_mcp_server_response_item import DeleteMcpServerResponseItem
 from .list_mcp_servers_response_value import ListMcpServersResponseValue
+from .streaming_response import StreamingResponse
 from .test_mcp_server_request import TestMcpServerRequest
 from .update_mcp_server_request import UpdateMcpServerRequest
 from .update_mcp_server_response import UpdateMcpServerResponse
@@ -13,8 +15,10 @@ __all__ = [
     "AddMcpServerRequest",
     "AddMcpServerResponseItem",
     "ConnectMcpServerRequest",
+    "ConnectMcpServerResponseEvent",
     "DeleteMcpServerResponseItem",
     "ListMcpServersResponseValue",
+    "StreamingResponse",
     "TestMcpServerRequest",
     "UpdateMcpServerRequest",
     "UpdateMcpServerResponse",

letta_client/tools/types/connect_mcp_server_response_event.py

@@ -0,0 +1,8 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+ConnectMcpServerResponseEvent = typing.Union[
+    typing.Literal["connection_attempt", "success", "error", "oauth_required", "authorization_url", "waiting_for_auth"],
+    typing.Any,
+]

letta_client/tools/types/streaming_response.py

@@ -0,0 +1,23 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ...core.unchecked_base_model import UncheckedBaseModel
+from .connect_mcp_server_response_event import ConnectMcpServerResponseEvent
+import typing
+from ...types.mcp_tool import McpTool
+from ...core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class StreamingResponse(UncheckedBaseModel):
+    event: ConnectMcpServerResponseEvent
+    message: typing.Optional[str] = None
+    tools: typing.Optional[McpTool] = 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

letta_client/types/letta_batch_request.py

@@ -39,6 +39,11 @@ class LettaBatchRequest(UncheckedBaseModel):
     Only return specified message types in the response. If `None` (default) returns all messages.
     """
 
+    enable_thinking: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    If set to True, enables reasoning before responses or tool calls from the agent.
+    """
+
     agent_id: str = pydantic.Field()
     """
     The ID of the agent to send this batch request for

letta_client/types/letta_request.py

@@ -39,6 +39,11 @@ class LettaRequest(UncheckedBaseModel):
     Only return specified message types in the response. If `None` (default) returns all messages.
     """
 
+    enable_thinking: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    If set to True, enables reasoning before responses or tool calls from the agent.
+    """
+
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
     else:

letta_client/types/letta_streaming_request.py

@@ -39,6 +39,11 @@ class LettaStreamingRequest(UncheckedBaseModel):
     Only return specified message types in the response. If `None` (default) returns all messages.
     """
 
+    enable_thinking: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    If set to True, enables reasoning before responses or tool calls from the agent.
+    """
+
     stream_tokens: typing.Optional[bool] = pydantic.Field(default=None)
     """
     Flag to determine if individual tokens should be streamed. Set to True for token streaming (requires stream_steps = True).

{letta_client-0.1.224.dist-info → letta_client-0.1.226.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: letta-client
-Version: 0.1.224
+Version: 0.1.226
 Summary: 
 Requires-Python: >=3.8,<4.0
 Classifier: Intended Audience :: Developers
@@ -102,24 +102,18 @@ except ApiError as e:
 The SDK supports streaming responses, as well, the response will be a generator that you can loop over.
 
 ```python
-from letta_client import Letta, MessageCreate, TextContent
+from letta_client import Letta, StdioServerConfig
 
 client = Letta(
     project="YOUR_PROJECT",
     token="YOUR_TOKEN",
 )
-response = client.agents.messages.create_stream(
-    agent_id="agent_id",
-    messages=[
-        MessageCreate(
-            role="user",
-            content=[
-                TextContent(
-                    text="text",
-                )
-            ],
-        )
-    ],
+response = client.tools.connect_mcp_server(
+    request=StdioServerConfig(
+        server_name="server_name",
+        command="command",
+        args=["args"],
+    ),
 )
 for chunk in response:
     yield chunk

{letta_client-0.1.224.dist-info → letta_client-0.1.226.dist-info}/RECORD

@@ -1,4 +1,4 @@
-letta_client/__init__.py,sha256=SHGQJZEiJUN0sO7ZVdwwTenizJTldEs9YqrB-ux8vCk,19484
+letta_client/__init__.py,sha256=kQxkWRS8b6iG1Tb3TvpqswqflQ_k9nexhmX_MxZXRiE,19604
 letta_client/agents/__init__.py,sha256=9wEJMighDL1OFg_7Qh-D50bubPbV4BWo1ZKYxdDJGIQ,2146
 letta_client/agents/blocks/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
 letta_client/agents/blocks/client.py,sha256=kCAcU1ACOKRCrjEYPyCQslWBjtsq7m4QboQP0wwLQy8,25392
@@ -18,7 +18,7 @@ letta_client/agents/memory_variables/client.py,sha256=DGJvV5k5H-BRE-FWMLNrCqKkHJ
 letta_client/agents/memory_variables/types/__init__.py,sha256=EoznK0WvhCyFYd4KDdU-cGDQWpSXmq79BSkqVHN-j7A,180
 letta_client/agents/memory_variables/types/memory_variables_list_response.py,sha256=bsF__n_B4ZXEHzg--OVD6tHHXt_aM-FjHm2x1ZXPnL0,599
 letta_client/agents/messages/__init__.py,sha256=RSmlezGH90RirElX0LHusjD03EN6UqGsrS-A0Bue2hA,353
-letta_client/agents/messages/client.py,sha256=J3VPSf9oSFb9qdyBfOcbRTkKtvnyT903PR6ADsofH5M,58066
+letta_client/agents/messages/client.py,sha256=yh3gXasaH-rwTRyz64CJjR_AbpX05Ncb4LvilqnaN0Y,59548
 letta_client/agents/messages/types/__init__.py,sha256=uDn0e1-EJ7g2U0EHD8SkP38liALoVCTJ7bJhUAOutwE,473
 letta_client/agents/messages/types/letta_streaming_response.py,sha256=8VR2F32xjoPFXL4YBvBbAZclaJG4ENPTjk7BrlZkmtw,742
 letta_client/agents/messages/types/messages_modify_request.py,sha256=7C2X3BKye-YDSXOkdEmxxt34seI4jkLK0-govtc4nhg,475
@@ -75,7 +75,7 @@ letta_client/client_side_access_tokens/types/client_side_access_tokens_list_clie
 letta_client/client_side_access_tokens/types/client_side_access_tokens_list_client_side_access_tokens_response_tokens_item_policy_data_item_access_item.py,sha256=kNHfEWFl7u71Pu8NPqutod0a2NXfvq8il05Hqm0iBB4,284
 letta_client/core/__init__.py,sha256=OKbX2aCZXgHCDUsCouqv-OiX32xA6eFFCKIUH9M5Vzk,1591
 letta_client/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
-letta_client/core/client_wrapper.py,sha256=xSoUSJswIYwHXAHKAT7yDaEoUmNqKq707H1hsvj6AKw,2336
+letta_client/core/client_wrapper.py,sha256=IaGc6llVLIjSdRed41AC3ZZDUFKMyrENDHxGP4yyrcs,2336
 letta_client/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
 letta_client/core/file.py,sha256=d4NNbX8XvXP32z8KpK2Xovv33nFfruIrpz0QWxlgpZk,2663
 letta_client/core/http_client.py,sha256=Z77OIxIbL4OAB2IDqjRq_sYa5yNYAWfmdhdCSSvh6Y4,19552
@@ -105,7 +105,7 @@ letta_client/folders/passages/client.py,sha256=ejEUmwrwfUBG4UihNTrIk7b1-a1v6QziR
 letta_client/groups/__init__.py,sha256=WzkNp5Q_5zQj_NHv4hJCOKvW6ftM9EuNxw8hkPRRbko,434
 letta_client/groups/client.py,sha256=H6PW_E2QUX_hfoa5aunu90KEU1sF5d4xpTL1hxZFifU,30369
 letta_client/groups/messages/__init__.py,sha256=M7Ar6Rmb8we4dfYE6jj3FCL9UvVFy1bNQIPflUXMWHA,243
-letta_client/groups/messages/client.py,sha256=lK8P1J3ETHxdsS8zbdNnxagVSef03Z_tLiqQiEXq6t0,36386
+letta_client/groups/messages/client.py,sha256=POUMzgG6YgvLzfKeWsDPYL706VbNL6lWOOnqW8w9Mrc,37374
 letta_client/groups/messages/types/__init__.py,sha256=Oc2j0oGOs96IEFf9xsJIkjBjoq3OMtse64YwWv3F9Io,335
 letta_client/groups/messages/types/letta_streaming_response.py,sha256=MdE2PxQ1x1AviakHXsWVcFv97a3RchzzzIiD77w4EC8,665
 letta_client/groups/messages/types/messages_modify_request.py,sha256=7C2X3BKye-YDSXOkdEmxxt34seI4jkLK0-govtc4nhg,475
@@ -168,14 +168,16 @@ letta_client/templates/client.py,sha256=wcidkaF0eRgKEYRrDTgKAS_A57MO7H2S_muzznEV
 letta_client/templates/types/__init__.py,sha256=dAr_dEh0BdwUxAcV1sJ9RM07Z8nCv4dCK6fmTltqQ6c,286
 letta_client/templates/types/templates_list_response.py,sha256=HYloMVzk086c6fFGRYZz-Ozc_Yylozp2aPpweHS5uXI,866
 letta_client/templates/types/templates_list_response_templates_item.py,sha256=yyJq8wEOb2XIg99uhRMKoy2qD2CbuvI_5FAspwYWnfI,593
-letta_client/tools/__init__.py,sha256=qPV3aBl81CCoPAvuGdX-GiMyYkz0i49KerSRO-7BWXA,585
-letta_client/tools/client.py,sha256=ljRP0WZUhkgMFGXwM327QJReaW_dt4WRYkcDWJ1SIzY,107013
-letta_client/tools/types/__init__.py,sha256=U6Z2ZrKY4czFQyksSjoeoQHPbNQ0gJBM7QOGPJltqAA,846
+letta_client/tools/__init__.py,sha256=S7wUWB3P-uxAsKEFnzvVBpsDloETL1gks3l6ecdPpBY,705
+letta_client/tools/client.py,sha256=26C9XDrSucsPXxgUYzIYdmKloZPGRevXwFLhoZxyVO8,103238
+letta_client/tools/types/__init__.py,sha256=IodcTbnUxVsTQrNyZnuSfre7Cl2XESyiDkdHmFYfbHY,1035
 letta_client/tools/types/add_mcp_server_request.py,sha256=m3QdTmY2ZHQUWbxMTNsOhPnseWHVipsOTdSXuC7KHQI,371
 letta_client/tools/types/add_mcp_server_response_item.py,sha256=DNrB3LwstJzKrw_GRJ8tb3XCEJWfD16WzBoGrGY_ZQI,376
 letta_client/tools/types/connect_mcp_server_request.py,sha256=KDHYKXnRRf9g8bHUUWvWdyIiJk-enBZ5EO7N80r-i80,375
+letta_client/tools/types/connect_mcp_server_response_event.py,sha256=P_IXBx-DeJ2FdsIzIbRhekUKnU8ugiTgBqOzpRwEK6U,265
 letta_client/tools/types/delete_mcp_server_response_item.py,sha256=YLIBE7OD535NJAncGpzMDGaQRe1831DNKcj2UzS9e0c,379
 letta_client/tools/types/list_mcp_servers_response_value.py,sha256=Eyji5qB7FhowiogsAbpcU_aMyH9zClv9lUMmHOmNPYk,379
+letta_client/tools/types/streaming_response.py,sha256=E7W0OqxHFmhU2slJdzVZBxQA508AF5Im3pAM4UmuLs4,790
 letta_client/tools/types/test_mcp_server_request.py,sha256=sLlOEZdmLfkHqHCkUjntGbr8_MkBhsqpMQ-HwdNOnq0,372
 letta_client/tools/types/update_mcp_server_request.py,sha256=nCpx9-OvpH0l5iJxEi8kgSok1F1r7liEAZm-kaqBtEo,402
 letta_client/tools/types/update_mcp_server_response.py,sha256=muwHagaQBMwQI0of9EBCBtG9lD-jELFAevgTB2MjpFQ,375
@@ -310,15 +312,15 @@ letta_client/types/json_object_response_format.py,sha256=kz1wkWKO2H9Ad9GgLzLHgnY
 letta_client/types/json_schema.py,sha256=EHcLKBSGRsSzCKTpujKFHylcLJG6ODQIBrjQkU4lWDQ,870
 letta_client/types/json_schema_response_format.py,sha256=vTBC5qyuUm9u1uf1IZmNyEH-wSXm8c_7cOwd7ua_aJw,816
 letta_client/types/letta_batch_messages.py,sha256=kMefbiarujv7hCw3FyU-eVY2RgDV0ZXLOpkOooWNw6g,613
-letta_client/types/letta_batch_request.py,sha256=s0lLeWYKxS4ATaElzdZJyTxn7nVnaPTXeLYa01Ok-rA,1773
+letta_client/types/letta_batch_request.py,sha256=cNxnDcz7ob626oCJb4a6tpD3cQqk5f9ZcIyQXV5gJIQ,1948
 letta_client/types/letta_image.py,sha256=HA__0Gq0BpKHeUUxIT2TmEfyY18F2DrauiGlijWv-HA,1138
 letta_client/types/letta_message_content_union.py,sha256=ypGyeR8PGqPGaAWtLWjDkWwG9oa4AWcL44y-RfLJVvw,540
 letta_client/types/letta_message_union.py,sha256=TTQwlur2CZNdZ466Nb_2TFcSFXrgoMliaNzD33t7Ktw,603
-letta_client/types/letta_request.py,sha256=vhS3qQYxNN2dlE_kXWPW3BJekBDMTSRp1MXmj4-ztfQ,1659
+letta_client/types/letta_request.py,sha256=icbaI29DCFiCE5Y19WrdXz0VdIu3kq34hlGTlD9uTZE,1834
 letta_client/types/letta_request_config.py,sha256=OW5TNyTfbBMfRcWp5Aje_-7vjS0zVzYCmQGKLUtJezU,1346
 letta_client/types/letta_response.py,sha256=5brJ39B8PxTBpm5_clL9XGP1SQ1pox-G0gxMDkitMlw,1395
 letta_client/types/letta_stop_reason.py,sha256=5uqJibhaT6LFTj6Sf6m0VJKS1FJzIIgymjZTvu2a0Zk,931
-letta_client/types/letta_streaming_request.py,sha256=YJ_6ge5xrzQjOs2UTY3FxwhTv6QoCi9XWv5Rod01W1E,1884
+letta_client/types/letta_streaming_request.py,sha256=-mWkurh8AywX0YjIt5AYbxlE3m6iDYCUCkgpkzUE3G4,2059
 letta_client/types/letta_usage_statistics.py,sha256=k6V72J2TEPd-RQBuUQxF3oylrAMcuSKBskd2nnZmGOw,1886
 letta_client/types/letta_user_message_content_union.py,sha256=3Gbs3mRk-tJj2z0Mf-BNDomWHEytQd3OTUN4xnEVsuE,229
 letta_client/types/llm_config.py,sha256=m8IlVCS4uSXVYsuIadilgO32UpuD4_ij0a1hcRQiMLo,4306
@@ -445,6 +447,6 @@ letta_client/types/web_search_options_user_location_approximate.py,sha256=Ywk01J
 letta_client/version.py,sha256=bttKLbIhO3UonCYQlqs600zzbQgfhCCMjeXR9WRzid4,79
 letta_client/voice/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
 letta_client/voice/client.py,sha256=47iQYCuW_qpKI4hM3pYVxn3hw7kgQj3emU1_oRpkRMA,5811
-letta_client-0.1.224.dist-info/METADATA,sha256=vWY5btWhUxfgAemsZCMFxzK2SDhykSCQkkJAHABYCT8,5177
-letta_client-0.1.224.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-letta_client-0.1.224.dist-info/RECORD,,
+letta_client-0.1.226.dist-info/METADATA,sha256=M0_FnvWA9H9CyQ4oeGVYPOJhy66xl2ODW5bdV09-hj8,5064
+letta_client-0.1.226.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+letta_client-0.1.226.dist-info/RECORD,,