letta-client 0.1.175__tar.gz → 0.1.176__tar.gz

{letta_client-0.1.175 → letta_client-0.1.176}/PKG-INFO

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

{letta_client-0.1.175 → letta_client-0.1.176}/pyproject.toml

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

{letta_client-0.1.175 → letta_client-0.1.176}/src/letta_client/__init__.py

@@ -315,6 +315,7 @@ from .tools import (
     AddMcpServerResponseItem,
     DeleteMcpServerResponseItem,
     ListMcpServersResponseValue,
+    TestMcpServerRequest,
     UpdateMcpServerRequest,
     UpdateMcpServerResponse,
 )
@@ -562,6 +563,7 @@ __all__ = [
     "TemplatesListResponse",
     "TemplatesListResponseTemplatesItem",
     "TerminalToolRule",
+    "TestMcpServerRequest",
     "TextContent",
     "TextResponseFormat",
     "Tool",

{letta_client-0.1.175 → letta_client-0.1.176}/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.175",
+            "X-Fern-SDK-Version": "0.1.176",
         }
         if self.token is not None:
             headers["Authorization"] = f"Bearer {self.token}"

{letta_client-0.1.175 → letta_client-0.1.176}/src/letta_client/tools/__init__.py

@@ -5,6 +5,7 @@ from .types import (
     AddMcpServerResponseItem,
     DeleteMcpServerResponseItem,
     ListMcpServersResponseValue,
+    TestMcpServerRequest,
     UpdateMcpServerRequest,
     UpdateMcpServerResponse,
 )
@@ -14,6 +15,7 @@ __all__ = [
     "AddMcpServerResponseItem",
     "DeleteMcpServerResponseItem",
     "ListMcpServersResponseValue",
+    "TestMcpServerRequest",
     "UpdateMcpServerRequest",
     "UpdateMcpServerResponse",
 ]

{letta_client-0.1.175 → letta_client-0.1.176}/src/letta_client/tools/client.py

@@ -22,6 +22,7 @@ from ..types.mcp_tool import McpTool
 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 ..core.client_wrapper import AsyncClientWrapper
 
 # this is used as the default value for optional parameters
@@ -1299,6 +1300,73 @@ class ToolsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def test_mcp_server(
+        self, *, request: TestMcpServerRequest, request_options: typing.Optional[RequestOptions] = None
+    ) -> typing.List[McpTool]:
+        """
+        Test connection to an MCP server without adding it.
+        Returns the list of available tools if successful.
+
+        Parameters
+        ----------
+        request : TestMcpServerRequest
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.List[McpTool]
+            Successful Response
+
+        Examples
+        --------
+        from letta_client import Letta, StdioServerConfig
+
+        client = Letta(
+            token="YOUR_TOKEN",
+        )
+        client.tools.test_mcp_server(
+            request=StdioServerConfig(
+                server_name="server_name",
+                command="command",
+                args=["args"],
+            ),
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "v1/tools/mcp/servers/test",
+            method="POST",
+            json=convert_and_respect_annotation_metadata(
+                object_=request, annotation=TestMcpServerRequest, direction="write"
+            ),
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    typing.List[McpTool],
+                    construct_type(
+                        type_=typing.List[McpTool],  # 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 AsyncToolsClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
@@ -2714,3 +2782,78 @@ 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 test_mcp_server(
+        self, *, request: TestMcpServerRequest, request_options: typing.Optional[RequestOptions] = None
+    ) -> typing.List[McpTool]:
+        """
+        Test connection to an MCP server without adding it.
+        Returns the list of available tools if successful.
+
+        Parameters
+        ----------
+        request : TestMcpServerRequest
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.List[McpTool]
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from letta_client import AsyncLetta, StdioServerConfig
+
+        client = AsyncLetta(
+            token="YOUR_TOKEN",
+        )
+
+
+        async def main() -> None:
+            await client.tools.test_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/test",
+            method="POST",
+            json=convert_and_respect_annotation_metadata(
+                object_=request, annotation=TestMcpServerRequest, direction="write"
+            ),
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    typing.List[McpTool],
+                    construct_type(
+                        type_=typing.List[McpTool],  # 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.175 → letta_client-0.1.176}/src/letta_client/tools/types/__init__.py

@@ -4,6 +4,7 @@ from .add_mcp_server_request import AddMcpServerRequest
 from .add_mcp_server_response_item import AddMcpServerResponseItem
 from .delete_mcp_server_response_item import DeleteMcpServerResponseItem
 from .list_mcp_servers_response_value import ListMcpServersResponseValue
+from .test_mcp_server_request import TestMcpServerRequest
 from .update_mcp_server_request import UpdateMcpServerRequest
 from .update_mcp_server_response import UpdateMcpServerResponse
 
@@ -12,6 +13,7 @@ __all__ = [
     "AddMcpServerResponseItem",
     "DeleteMcpServerResponseItem",
     "ListMcpServersResponseValue",
+    "TestMcpServerRequest",
     "UpdateMcpServerRequest",
     "UpdateMcpServerResponse",
 ]

letta_client-0.1.176/src/letta_client/tools/types/test_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
+
+TestMcpServerRequest = typing.Union[StdioServerConfig, SseServerConfig, StreamableHttpServerConfig]