diff --git a/swarms/structs/agent.py b/swarms/structs/agent.py index 334cc33a..7a723f9b 100644 --- a/swarms/structs/agent.py +++ b/swarms/structs/agent.py @@ -404,7 +404,7 @@ class Agent: role: agent_roles = "worker", no_print: bool = False, tools_list_dictionary: Optional[List[Dict[str, Any]]] = None, - # mcp_servers: List[MCPServerSseParams] = [], + mcp_servers: Optional[list] = None, # list[MCPServerSseParams] *args, **kwargs, ): @@ -524,7 +524,7 @@ class Agent: self.role = role self.no_print = no_print self.tools_list_dictionary = tools_list_dictionary - # self.mcp_servers = mcp_servers + self.mcp_servers = mcp_servers self._cached_llm = ( None # Add this line to cache the LLM instance diff --git a/swarms/tools/mcp_integration.py b/swarms/tools/mcp_integration.py index 8a1fca1f..dad88d75 100644 --- a/swarms/tools/mcp_integration.py +++ b/swarms/tools/mcp_integration.py @@ -1,7 +1,7 @@ from __future__ import annotations -from typing import Any, Dict, List, Optional, TypedDict, NotRequired -from typing_extensions import TypedDict +from typing import Any, Dict, List, Optional +from typing_extensions import NotRequired, TypedDict from contextlib import AbstractAsyncContextManager from fastmcp import FastClientSession as ClientSession from fastmcp.servers import fast_sse_client as sse_client