Add file system tools to Aura

Author: richard-gyiko

pyproject.toml

@@ -6,6 +6,7 @@ readme = "README.md"
 requires-python = ">=3.12"
 dependencies = [
     "autogen-agentchat==0.4.0.dev13",
+    "autogen-ext-mcp>=0.2.0",
     "autogen-ext[langchain,openai,azure]==0.4.0.dev13",
     "langchain-google-community[gmail]>=2.0.3",
     "python-dateutil>=2.9.0.post0",

src/aura/agents/aura.py

@@ -3,6 +3,7 @@
 from autogen_agentchat.agents import AssistantAgent
 from autogen_ext.models.openai import OpenAIChatCompletionClient
 from tools.tool_factory import (
+    get_file_system_tools,
     get_gmail_tools,
     get_google_calendar_tools,
     get_utility_tools,
@@ -34,11 +35,12 @@ def _get_timezone() -> ZoneInfo:
     return ZoneInfo(str(get_localzone()))
 
 
-def aura() -> AssistantAgent:
+async def aura() -> AssistantAgent:
     tools = (
         get_gmail_tools(SCOPES)
         + get_google_calendar_tools(SCOPES)
         + get_utility_tools()
+        + await get_file_system_tools()
     )
 
     assistant = AssistantAgent(

src/aura/main.py

@@ -3,13 +3,15 @@
 from autogen_agentchat.messages import TextMessage
 from autogen_core import CancellationToken
 from dotenv import load_dotenv
-
 from agents.aura import aura
+
 from utils.console import RichConsole
 
+load_dotenv()
+
 
 async def main():
-    agent = aura()
+    agent = await aura()
 
     while True:
         try:
@@ -31,5 +33,4 @@ async def main():
 
 
 if __name__ == "__main__":
-    load_dotenv()
     asyncio.run(main())

src/aura/tools/tool_factory.py

@@ -1,8 +1,11 @@
+from pathlib import Path
 from autogen_ext.tools.langchain import LangChainToolAdapter
+from autogen_ext_mcp.tools import get_tools_from_mcp_server
 from langchain_google_community import GmailToolkit
 from langchain_google_community.gmail.utils import (
     build_resource_service as build_gmail_resource_service,
 )
+from mcp import StdioServerParameters
 
 from .gmail.toolkit import GmailToolkitExt
 from .google_calendar.tookit import GoogleCalendarToolkit
@@ -12,6 +15,20 @@
 from .utilities.get_current_time import GetCurrentTime
 
 
+file_system_server = StdioServerParameters(
+    command="npx.cmd",
+    args=[
+        "-y",
+        "@modelcontextprotocol/server-filesystem",
+        str(Path.home() / "Desktop" / "aura"),
+    ],
+)
+
+
+async def get_file_system_tools():
+    return await get_tools_from_mcp_server(file_system_server)
+
+
 def get_gmail_tools(scopes: list[str]):
     api_resource = build_gmail_resource_service(scopes=scopes)
 

src/aura/utils/mcp.py

@@ -0,0 +1,28 @@
+from typing import List
+from autogen_ext_mcp.tools import MCPToolAdapter
+from rich.console import Console
+
+
+def print_tools(tools: List[MCPToolAdapter]) -> None:
+    console = Console()
+    console.print("\n[bold blue]📦 Loaded MCP Tools:[/bold blue]")
+
+    for tool in tools:
+        console.print(f"\n[bold green]🔧 {tool.schema['name']}[/bold green]")
+        if "description" in tool.schema:
+            console.print(f"[italic]{tool.schema['description']}[/italic]")
+
+        if "parameters" in tool.schema:
+            console.print("\n[yellow]Parameters:[/yellow]")
+            params = tool.schema["parameters"]
+            if "properties" in params:
+                for prop_name, prop_details in params["properties"].items():
+                    required = prop_name in params.get("required", [])
+                    required_mark = "[red]*[/red]" if required else ""
+                    console.print(
+                        f"  • [cyan]{prop_name}{required_mark}[/cyan]: {prop_details.get('type', 'any')}"
+                    )
+                    if "description" in prop_details:
+                        console.print(f"    [dim]{prop_details['description']}[/dim]")
+
+        console.print("─" * 50)