letta-client 0.1.221__py3-none-any.whl → 0.1.222__py3-none-any.whl

letta_client/__init__.py

@@ -334,6 +334,7 @@ from .templates import TemplatesListResponse, TemplatesListResponseTemplatesItem
 from .tools import (
     AddMcpServerRequest,
     AddMcpServerResponseItem,
+    ConnectMcpServerRequest,
     DeleteMcpServerResponseItem,
     ListMcpServersResponseValue,
     TestMcpServerRequest,
@@ -449,6 +450,7 @@ __all__ = [
     "ConditionalToolRuleSchema",
     "ConflictError",
     "ConflictErrorBody",
+    "ConnectMcpServerRequest",
     "ContextWindowOverview",
     "ContinueToolRule",
     "CoreMemoryBlockSchema",

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

letta_client/tools/__init__.py

@@ -3,6 +3,7 @@
 from .types import (
     AddMcpServerRequest,
     AddMcpServerResponseItem,
+    ConnectMcpServerRequest,
     DeleteMcpServerResponseItem,
     ListMcpServersResponseValue,
     TestMcpServerRequest,
@@ -13,6 +14,7 @@ from .types import (
 __all__ = [
     "AddMcpServerRequest",
     "AddMcpServerResponseItem",
+    "ConnectMcpServerRequest",
     "DeleteMcpServerResponseItem",
     "ListMcpServersResponseValue",
     "TestMcpServerRequest",

letta_client/tools/client.py

@@ -23,6 +23,7 @@ from .types.delete_mcp_server_response_item import DeleteMcpServerResponseItem
 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 ..core.client_wrapper import AsyncClientWrapper
 
 # this is used as the default value for optional parameters
@@ -1320,10 +1321,10 @@ class ToolsClient:
 
     def test_mcp_server(
         self, *, request: TestMcpServerRequest, request_options: typing.Optional[RequestOptions] = None
-    ) -> typing.List[McpTool]:
+    ) -> typing.Optional[typing.Any]:
         """
         Test connection to an MCP server without adding it.
-        Returns the list of available tools if successful.
+        Returns the list of available tools if successful, or OAuth information if OAuth is required.
 
         Parameters
         ----------
@@ -1334,7 +1335,7 @@ class ToolsClient:
 
         Returns
         -------
-        typing.List[McpTool]
+        typing.Optional[typing.Any]
             Successful Response
 
         Examples
@@ -1365,9 +1366,77 @@ class ToolsClient:
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    typing.List[McpTool],
+                    typing.Optional[typing.Any],
                     construct_type(
-                        type_=typing.List[McpTool],  # type: ignore
+                        type_=typing.Optional[typing.Any],  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+            if _response.status_code == 422:
+                raise UnprocessableEntityError(
+                    typing.cast(
+                        HttpValidationError,
+                        construct_type(
+                            type_=HttpValidationError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    )
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    def connect_mcp_server(
+        self, *, request: ConnectMcpServerRequest, request_options: typing.Optional[RequestOptions] = None
+    ) -> typing.Optional[typing.Any]:
+        """
+        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.
+
+        Parameters
+        ----------
+        request : ConnectMcpServerRequest
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.Optional[typing.Any]
+            Successful response
+
+        Examples
+        --------
+        from letta_client import Letta, StdioServerConfig
+
+        client = Letta(
+            project="YOUR_PROJECT",
+            token="YOUR_TOKEN",
+        )
+        client.tools.connect_mcp_server(
+            request=StdioServerConfig(
+                server_name="server_name",
+                command="command",
+                args=["args"],
+            ),
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "v1/tools/mcp/servers/connect",
+            method="POST",
+            json=convert_and_respect_annotation_metadata(
+                object_=request, annotation=ConnectMcpServerRequest, direction="write"
+            ),
+            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(),
                     ),
                 )
@@ -1386,6 +1455,83 @@ class ToolsClient:
             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)
+
 
 class AsyncToolsClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
@@ -2822,10 +2968,10 @@ class AsyncToolsClient:
 
     async def test_mcp_server(
         self, *, request: TestMcpServerRequest, request_options: typing.Optional[RequestOptions] = None
-    ) -> typing.List[McpTool]:
+    ) -> typing.Optional[typing.Any]:
         """
         Test connection to an MCP server without adding it.
-        Returns the list of available tools if successful.
+        Returns the list of available tools if successful, or OAuth information if OAuth is required.
 
         Parameters
         ----------
@@ -2836,7 +2982,7 @@ class AsyncToolsClient:
 
         Returns
         -------
-        typing.List[McpTool]
+        typing.Optional[typing.Any]
             Successful Response
 
         Examples
@@ -2875,9 +3021,9 @@ class AsyncToolsClient:
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    typing.List[McpTool],
+                    typing.Optional[typing.Any],
                     construct_type(
-                        type_=typing.List[McpTool],  # type: ignore
+                        type_=typing.Optional[typing.Any],  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -2895,3 +3041,164 @@ class AsyncToolsClient:
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    async def connect_mcp_server(
+        self, *, request: ConnectMcpServerRequest, request_options: typing.Optional[RequestOptions] = None
+    ) -> typing.Optional[typing.Any]:
+        """
+        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.
+
+        Parameters
+        ----------
+        request : ConnectMcpServerRequest
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.Optional[typing.Any]
+            Successful response
+
+        Examples
+        --------
+        import asyncio
+
+        from letta_client import AsyncLetta, StdioServerConfig
+
+        client = AsyncLetta(
+            project="YOUR_PROJECT",
+            token="YOUR_TOKEN",
+        )
+
+
+        async def main() -> None:
+            await client.tools.connect_mcp_server(
+                request=StdioServerConfig(
+                    server_name="server_name",
+                    command="command",
+                    args=["args"],
+                ),
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "v1/tools/mcp/servers/connect",
+            method="POST",
+            json=convert_and_respect_annotation_metadata(
+                object_=request, annotation=ConnectMcpServerRequest, direction="write"
+            ),
+            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(),
+                        ),
+                    )
+                )
+            _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)

letta_client/tools/types/__init__.py

@@ -2,6 +2,7 @@
 
 from .add_mcp_server_request import AddMcpServerRequest
 from .add_mcp_server_response_item import AddMcpServerResponseItem
+from .connect_mcp_server_request import ConnectMcpServerRequest
 from .delete_mcp_server_response_item import DeleteMcpServerResponseItem
 from .list_mcp_servers_response_value import ListMcpServersResponseValue
 from .test_mcp_server_request import TestMcpServerRequest
@@ -11,6 +12,7 @@ from .update_mcp_server_response import UpdateMcpServerResponse
 __all__ = [
     "AddMcpServerRequest",
     "AddMcpServerResponseItem",
+    "ConnectMcpServerRequest",
     "DeleteMcpServerResponseItem",
     "ListMcpServersResponseValue",
     "TestMcpServerRequest",

letta_client/tools/types/connect_mcp_server_request.py

@@ -0,0 +1,8 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+from ...types.stdio_server_config import StdioServerConfig
+from ...types.sse_server_config import SseServerConfig
+from ...types.streamable_http_server_config import StreamableHttpServerConfig
+
+ConnectMcpServerRequest = typing.Union[StdioServerConfig, SseServerConfig, StreamableHttpServerConfig]

{letta_client-0.1.221.dist-info → letta_client-0.1.222.dist-info}/METADATA

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

{letta_client-0.1.221.dist-info → letta_client-0.1.222.dist-info}/RECORD

@@ -1,4 +1,4 @@
-letta_client/__init__.py,sha256=aiCSJoHJGKQtTMjXahT6WjiMyw1IlwC6owUrb1tIu2s,19424
+letta_client/__init__.py,sha256=SHGQJZEiJUN0sO7ZVdwwTenizJTldEs9YqrB-ux8vCk,19484
 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=4UGPYxfGwNN3ZW-SkIdfVZK6cvCcumVAw0_AM8OmoBY,25046
@@ -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=igW7nKm6krTfc7tX6GMLo5a0-zx49SPQFt-ITwtC9s4,2336
+letta_client/core/client_wrapper.py,sha256=b0zYRfgXmdOyNiBfVM4TwkcAkFdNDPqaceDjjnGsHUc,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
@@ -168,11 +168,12 @@ 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=kSIxsMdzxX6TlriU43BYesswM2yq_iPoF-6-p0LbJKI,525
-letta_client/tools/client.py,sha256=hedvLbLtOgS5evYipbQdp8RBQjUja9Mk9nWfuUrNtxQ,97003
-letta_client/tools/types/__init__.py,sha256=-U2DGiEFPko6D5VRQQRoiJeHDJqb2iZnrbtHSxhkL5w,751
+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/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/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/test_mcp_server_request.py,sha256=sLlOEZdmLfkHqHCkUjntGbr8_MkBhsqpMQ-HwdNOnq0,372
@@ -444,6 +445,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.221.dist-info/METADATA,sha256=nMGSzoL0jIZOL35N-cSn_pQdxiK6y5lw34y0jfHt7yQ,5177
-letta_client-0.1.221.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-letta_client-0.1.221.dist-info/RECORD,,
+letta_client-0.1.222.dist-info/METADATA,sha256=kyaM48OWqtv4Wy5AAwhRJ2zWNk-ZIC5919-n3HPXzP8,5177
+letta_client-0.1.222.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+letta_client-0.1.222.dist-info/RECORD,,