Upload 8 files

Dockerfile

@@ -0,0 +1,36 @@
+# Use Python 3.13.2 slim image for efficiency
+FROM python:3.13.2-slim
+
+# Install system dependencies and uv
+RUN apt-get update && apt-get install -y \
+    git \
+    curl \
+    && rm -rf /var/lib/apt/lists/*
+
+# Install uv
+COPY --from=ghcr.io/astral-sh/uv:latest /uv /bin/uv
+
+# Set working directory
+WORKDIR /app
+
+# Set environment variables
+ENV PYTHONUNBUFFERED=1
+ENV PYTHONDONTWRITEBYTECODE=1
+
+# Copy project configuration files first to leverage Docker layer caching
+COPY pyproject.toml uv.lock ./
+
+# Install Python dependencies using uv
+RUN uv sync --frozen --no-dev
+
+# Copy application files
+COPY src/ ./src/
+
+# Create directories that might be needed
+RUN mkdir -p notebooks
+
+# Expose the port that Gradio will run on
+EXPOSE 7860
+
+# Use uv to run the application
+CMD ["uv", "run", "python", "src/gradio_app/app.py"] 

pyproject.toml

@@ -0,0 +1,65 @@
+[project]
+name = "mcp-agent"
+version = "0.1.0"
+description = "MCP Agent for AI-powered task automation"
+# hide for hugging face
+# readme = "README.md"
+requires-python = ">=3.13.2"
+license = {text = "MIT"}
+authors = [
+    {name = "Tristan Padiou", email = "Padioutristan@gmail.com"},
+]
+classifiers = [
+    "Development Status :: 3 - Alpha",
+    "Intended Audience :: Developers",
+    "License :: OSI Approved :: MIT License",
+    "Operating System :: OS Independent",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.13",
+]
+dependencies = [
+    "pydantic-ai==0.3.5",
+    "gradio==5.35.0",
+]
+
+[project.optional-dependencies]
+dev = [
+    "pytest>=7.0.0",
+    "black>=23.0.0",
+    "flake8>=5.0.0",
+    "mypy>=1.0.0",
+    "pre-commit>=2.20.0",
+]
+
+[project.urls]
+Homepage = "https://github.com/tristanpadiou/MCP-Agent"
+Repository = "https://github.com/tristanpadiou/MCP-Agent"
+Issues = "https://github.com/tristanpadiou/MCP-Agentissues"
+
+[tool.uv]
+dev-dependencies = [
+    "pytest>=7.0.0",
+    "black>=23.0.0",
+    "flake8>=5.0.0",
+    "mypy>=1.0.0",
+    "pre-commit>=2.20.0",
+]
+
+
+
+[tool.black]
+line-length = 100
+target-version = ['py313']
+include = '\.pyi?$'
+
+[tool.mypy]
+python_version = "3.13"
+warn_return_any = true
+warn_unused_configs = true
+disallow_untyped_defs = true
+
+[tool.pytest.ini_options]
+testpaths = ["tests"]
+python_files = ["test_*.py", "*_test.py"]
+python_classes = ["Test*"]
+python_functions = ["test_*"] 

src/__init__.py

@@ -0,0 +1 @@
+#empty file to make the directory a package

src/gradio_app/__init__.py

@@ -0,0 +1 @@
+#empty file to make the directory a package

src/gradio_app/app.py

@@ -0,0 +1,454 @@
+import gradio as gr
+import asyncio
+import json
+import atexit
+import signal
+import sys
+from src.mcp_agent.agent import MCP_Agent
+
+class GradioMCPApp:
+    def __init__(self):
+        self.agent = None
+        self.chat_history = []
+        self._loop = None
+        self.server_count = 1
+        
+    def get_or_create_loop(self):
+        """Get existing event loop or create a new one"""
+        if self._loop is None or self._loop.is_closed():
+            self._loop = asyncio.new_event_loop()
+            asyncio.set_event_loop(self._loop)
+        return self._loop
+        
+    async def initialize_agent(self, openai_api_key, *server_configs):
+        """Initialize the MCP Agent with provided configuration"""
+        try:
+            # Clean up existing agent first
+            if self.agent:
+                await self.disconnect_agent()
+            
+            # Build MCP servers configuration from form fields
+            # server_configs comes as a flat list: [url1, name1, type1, token1, url2, name2, type2, token2, ...]
+            mcp_servers = []
+            for i in range(0, len(server_configs), 4):
+                if i + 3 < len(server_configs):
+                    server_url = server_configs[i]
+                    server_name = server_configs[i + 1]
+                    server_type = server_configs[i + 2]
+                    headers = server_configs[i + 3]
+                    
+                    if server_url and server_url.strip():
+                        server_config = {
+                            'url': server_url.strip(),
+                            'name': server_name.strip() if server_name and server_name.strip() else f'server_{i//4 + 1}',
+                            'type': server_type if server_type else 'http',
+                            'headers': headers.strip() if headers and headers.strip() else None
+                        }
+                        mcp_servers.append(server_config)
+            
+            # Initialize agent
+            api_keys = {'openai_api_key': openai_api_key}
+            self.agent = MCP_Agent(api_keys=api_keys, mpc_server_urls=mcp_servers)
+            
+            # Connect to MCP servers
+            await self.agent.connect()
+            
+            server_count = len(mcp_servers)
+            if server_count == 0:
+                return True, "Agent initialized successfully (no MCP servers configured)!"
+            else:
+                return True, f"Agent initialized successfully with {server_count} MCP server(s)!"
+            
+        except Exception as e:
+            return False, f"Error initializing agent: {str(e)}"
+    
+    async def chat_with_agent(self, message):
+        """Handle text chat with the agent"""
+        if not self.agent:
+            return self.chat_history, "Please initialize the agent first by providing your OpenAI API key and clicking 'Initialize Agent'."
+        
+        if not message or not message.strip():
+            return self.chat_history, "Please provide a message."
+        
+        try:
+            # Get response from agent
+            response = await self.agent.chat(message.strip())
+            
+            # Update chat history
+            self.chat_history.append([message.strip(), str(response)])
+            
+            return self.chat_history, ""
+            
+        except Exception as e:
+            error_msg = f"Error during chat: {str(e)}"
+            self.chat_history.append([message, error_msg])
+            return self.chat_history, error_msg
+    
+    async def reset_agent(self):
+        """Reset the agent's conversation history"""
+        if self.agent:
+            self.agent.reset()
+            self.chat_history = []
+            return [], "Agent conversation history reset successfully!"
+        else:
+            return [], "No agent to reset. Please initialize the agent first."
+    
+    async def disconnect_agent(self):
+        """Disconnect from MCP servers"""
+        if self.agent:
+            try:
+                await self.agent.disconnect()
+            except Exception as e:
+                print(f"Error during disconnect: {e}")
+            finally:
+                self.agent = None
+                self.chat_history = []
+        return [], "Agent disconnected successfully!"
+    
+    async def cleanup(self):
+        """Clean up resources"""
+        if self.agent:
+            await self.disconnect_agent()
+        if self._loop and not self._loop.is_closed():
+            self._loop.close()
+
+# Create the app instance
+app_instance = GradioMCPApp()
+
+def run_async_safely(coro, *args):
+    """Safely run async function with proper error handling"""
+    loop = app_instance.get_or_create_loop()
+    try:
+        return loop.run_until_complete(coro(*args))
+    except Exception as e:
+        print(f"Error in async operation: {e}")
+        return None, f"Error: {str(e)}"
+
+# Define async wrapper functions for Gradio
+def initialize_agent_wrapper(openai_api_key, *server_configs):
+    success, message = run_async_safely(
+        app_instance.initialize_agent, 
+        openai_api_key, *server_configs
+    )
+    if success is None:
+        return gr.update(visible=False), gr.update(visible=True), message
+    return gr.update(visible=success), gr.update(visible=not success), message
+
+def chat_wrapper(message):
+    chat_history, error_msg = run_async_safely(app_instance.chat_with_agent, message)
+    if chat_history is None:
+        return [], error_msg, ""
+    return chat_history, error_msg, ""  # Clear input
+
+def reset_wrapper():
+    chat_history, message = run_async_safely(app_instance.reset_agent)
+    if chat_history is None:
+        return [], message
+    return chat_history, message
+
+def disconnect_wrapper():
+    chat_history, message = run_async_safely(app_instance.disconnect_agent)
+    if chat_history is None:
+        return [], gr.update(visible=False), gr.update(visible=True), message
+    return chat_history, gr.update(visible=False), gr.update(visible=True), message
+
+# Cleanup function for graceful shutdown
+def cleanup_on_exit():
+    """Cleanup function to run on exit"""
+    try:
+        loop = app_instance.get_or_create_loop()
+        if not loop.is_closed():
+            loop.run_until_complete(app_instance.cleanup())
+    except Exception as e:
+        print(f"Error during cleanup: {e}")
+
+# Register cleanup function
+atexit.register(cleanup_on_exit)
+
+# Handle SIGINT (Ctrl+C) gracefully
+def signal_handler(signum, frame):
+    print("\nReceived interrupt signal. Cleaning up...")
+    cleanup_on_exit()
+    sys.exit(0)
+
+signal.signal(signal.SIGINT, signal_handler)
+
+# Server management functions
+def add_server(current_count):
+    """Show the next server configuration"""
+    new_count = min(current_count + 1, 3)  # Max 3 servers
+    return (
+        new_count,
+        gr.update(visible=new_count >= 2),  # server2_group
+        gr.update(visible=new_count >= 3),  # server3_group
+        gr.update(interactive=new_count < 3),  # add_server_btn
+        gr.update(interactive=new_count > 1)   # remove_server_btn
+    )
+
+def remove_server(current_count):
+    """Hide the last server configuration"""
+    new_count = max(current_count - 1, 1)  # Min 1 server
+    return (
+        new_count,
+        gr.update(visible=new_count >= 2),  # server2_group
+        gr.update(visible=new_count >= 3),  # server3_group
+        gr.update(interactive=new_count < 3),  # add_server_btn
+        gr.update(interactive=new_count > 1)   # remove_server_btn
+    )
+
+
+
+# Create the Gradio interface
+with gr.Blocks(title="MCP Agent Chat", theme=gr.themes.Soft()) as demo:
+    with gr.Row():
+        
+        gr.Markdown("# MCP Agent Chat Interface")
+        gr.HTML("")  # Balance spacing
+    
+  
+    
+    with gr.Sidebar():
+        # Sidebar for configuration
+        sidebar_column = gr.Column(scale=1, min_width=350)
+        with sidebar_column:
+            gr.Markdown("## 🔧 Configuration")
+            
+            openai_key = gr.Textbox(
+                label="OpenAI API Key",
+                type="password",
+                placeholder="sk-...",
+                info="Your OpenAI API key for the language model"
+            )
+            
+            gr.Markdown("### MCP Servers Setup")
+            gr.Markdown("Configure your MCP server connections (leave all URLs empty to run without MCP servers)")
+            
+            # Container for dynamic server configurations
+            servers_container = gr.Column()
+            
+            # Initial server configuration
+            with servers_container:
+                # Server 1 (always present)
+                with gr.Tab("Url based servers"):
+                    with gr.Group():
+                        gr.Markdown("#### Server 1")
+                        server1_url = gr.Textbox(
+                            label="Server URL",
+                            placeholder="http://localhost:8000",
+                            info="The URL of your MCP server"
+                        )
+                        server1_name = gr.Textbox(
+                            label="Server Name",
+                            placeholder="server_1",
+                            info="A friendly name for your MCP server"
+                        )
+                        server1_type = gr.Dropdown(
+                            label="Server Type",
+                            choices=["http", "SSE"],
+                            value="http",
+                            info="The type of MCP server connection"
+                        )
+                        server1_headers = gr.Textbox(
+                            label="Headers (Optional)",
+                            type="password",
+                            placeholder="Leave empty if not required",
+                            info="Headers for the MCP server (if required), example: {'Authorization': 'Bearer 1234567890'}"
+                        )
+                    
+                    # Server 2 (optional)
+                    server2_group = gr.Group(visible=False)
+                    with server2_group:
+                        gr.Markdown("#### Server 2")
+                        server2_url = gr.Textbox(
+                            label="Server URL",
+                            placeholder="http://localhost:8001",
+                            info="The URL of your MCP server"
+                        )
+                        server2_name = gr.Textbox(
+                            label="Server Name",
+                            placeholder="server_2",
+                            info="A friendly name for your MCP server"
+                        )
+                        server2_type = gr.Dropdown(
+                            label="Server Type",
+                            choices=["http", "SSE"],
+                            value="http",
+                            info="The type of MCP server connection"
+                        )
+                        server2_headers = gr.Textbox(
+                            label="Headers (Optional)",
+                            type="password",
+                            placeholder="Leave empty if not required",
+                            info="Headers for the MCP server (if required), example: {'Authorization': 'Bearer 1234567890'}"
+                        )
+                    
+                    # Server 3 (optional)
+                    server3_group = gr.Group(visible=False)
+                    with server3_group:
+                        gr.Markdown("#### Server 3")
+                        server3_url = gr.Textbox(
+                            label="Server URL",
+                            placeholder="http://localhost:8002",
+                            info="The URL of your MCP server"
+                        )
+                        server3_name = gr.Textbox(
+                            label="Server Name",
+                            placeholder="server_3",
+                            info="A friendly name for your MCP server"
+                        )
+                        server3_type = gr.Dropdown(
+                            label="Server Type",
+                            choices=["http", "SSE"],
+                            value="http",
+                            info="The type of MCP server connection"
+                        )
+                        server3_headers = gr.Textbox(
+                            label="Headers (Optional)",
+                            type="password",
+                            placeholder="Leave empty if not required",
+                            info="Headers for the MCP server (if required), example: {'Authorization': 'Bearer 1234567890'}"
+                        )
+                with gr.Tab("Stdio based servers"):
+                    with gr.Group():
+                        gr.Markdown("#### Server 1")
+                        server1_command = gr.Textbox(
+                            label="Command",
+                            placeholder="npx",
+                            info="The command to use to run the MCP server, docker, npm, python, etc."
+                        )
+                        server1_args = gr.Textbox(
+                            label="Arguments",
+                            placeholder="['-y', '@modelcontextprotocol/server-memory']",
+                            info="The arguments to use to run the MCP server"
+                        )
+                        server1_name = gr.Textbox(
+                            label="Server Name",
+                            placeholder="server_1",
+                            info="A friendly name for your MCP server"
+                        )
+                # Server management buttons
+                with gr.Row():
+                    add_server_btn = gr.Button("+ Add Server", variant="secondary", size="sm")
+                    remove_server_btn = gr.Button("- Remove Server", variant="secondary", size="sm", interactive=False)
+            
+            # Track current server count
+            server_count_state = gr.State(1)
+            
+            init_btn = gr.Button("Initialize Agent", variant="primary", size="lg")
+            init_status = gr.Textbox(label="Status", interactive=False, max_lines=3)
+    
+    # Main chat area
+    chat_column = gr.Column(scale=2)
+    with chat_column:
+        with gr.Row():
+            gr.Markdown("## 💬 Chat with your MCP Agent")
+            config_status = gr.Markdown("", visible=False)  # Status when sidebar is collapsed
+        
+        chat_interface = gr.Column(visible=False)
+        with chat_interface:
+            chatbot = gr.Chatbot(
+                label="Conversation",
+                height=500,
+                show_copy_button=True,
+                avatar_images=("👤", "🤖")
+            )
+            
+            with gr.Row():
+                msg = gr.Textbox(
+                    label="Message",
+                    placeholder="Type your message here...",
+                    scale=4,
+                    lines=2
+                )
+                send_btn = gr.Button("Send", variant="primary", scale=1)
+            
+            error_display = gr.Textbox(
+                label="Error Messages",
+                visible=False,
+                interactive=False
+            )
+            
+            with gr.Row():
+                reset_btn = gr.Button("Reset Conversation", variant="secondary")
+                disconnect_btn = gr.Button("Disconnect Agent", variant="secondary")
+        
+        # Placeholder when agent is not initialized
+        placeholder = gr.Markdown(
+            "### 👋 Welcome!\n\nPlease configure and initialize your MCP Agent using the sidebar to start chatting.",
+            visible=True
+        )
+
+    
+    # Server management event handlers
+    add_server_btn.click(
+        fn=add_server,
+        inputs=[server_count_state],
+        outputs=[server_count_state, server2_group, server3_group, add_server_btn, remove_server_btn]
+    )
+    
+    remove_server_btn.click(
+        fn=remove_server,
+        inputs=[server_count_state],
+        outputs=[server_count_state, server2_group, server3_group, add_server_btn, remove_server_btn]
+    )
+    
+    # Event handlers
+    init_btn.click(
+        fn=initialize_agent_wrapper,
+        inputs=[
+            openai_key,
+            server1_url, server1_name, server1_type, server1_headers,
+            server2_url, server2_name, server2_type, server2_headers,
+            server3_url, server3_name, server3_type, server3_headers,
+            server1_command, server1_args, server1_name
+        ],
+        outputs=[chat_interface, placeholder, init_status]
+    )
+    
+    # Chat functionality
+    def handle_chat(message):
+        if not message or not message.strip():
+            return app_instance.chat_history, "Please provide a message.", ""
+        return chat_wrapper(message)
+    
+    send_btn.click(
+        fn=handle_chat,
+        inputs=[msg],
+        outputs=[chatbot, error_display, msg]
+    ).then(
+        lambda error: gr.update(visible=bool(error)),
+        inputs=[error_display],
+        outputs=[error_display]
+    )
+    
+    msg.submit(
+        fn=handle_chat,
+        inputs=[msg],
+        outputs=[chatbot, error_display, msg]
+    ).then(
+        lambda error: gr.update(visible=bool(error)),
+        inputs=[error_display],
+        outputs=[error_display]
+    )
+    
+    reset_btn.click(
+        fn=reset_wrapper,
+        outputs=[chatbot, error_display]
+    ).then(
+        lambda error: gr.update(visible=bool(error)),
+        inputs=[error_display],
+        outputs=[error_display]
+    )
+    
+    disconnect_btn.click(
+        fn=disconnect_wrapper,
+        outputs=[chatbot, chat_interface, placeholder, init_status]
+    )
+
+if __name__ == "__main__":
+    demo.launch(
+        server_name="127.0.0.1",
+        server_port=7860,
+        share=False,
+        show_error=True
+    )

src/mcp_agent/__init__.py

@@ -0,0 +1 @@
+#empty file to make the directory a package

src/mcp_agent/agent.py

@@ -0,0 +1,193 @@
+from __future__ import annotations
+
+from pydantic_ai import Agent, RunContext, format_as_xml
+
+from pydantic_ai.models.openai import OpenAIModel
+from pydantic_ai.providers.openai import OpenAIProvider
+from pydantic_ai.mcp import MCPServerStreamableHTTP, MCPServerSSE, MCPServerStdio
+from dataclasses import dataclass
+from datetime import datetime
+from pydantic import Field
+import json
+from pydantic_ai.messages import (
+    ModelMessage,
+    FinalResultEvent,
+    FunctionToolCallEvent,
+    FunctionToolResultEvent,
+    PartDeltaEvent,
+    PartStartEvent,
+    TextPartDelta,
+    ToolCallPartDelta,
+)
+
+
+
+
+@dataclass
+class Api_keys:
+    api_keys: dict
+
+
+@dataclass
+class Message_state:
+    messages: list[ModelMessage]
+
+
+    
+    
+class MCP_Agent:
+    def __init__(self, api_keys:dict, mpc_server_urls:list = [], mpc_stdio_commands:list = []):
+        """
+        Args:
+            
+            api_keys (dict): The API keys to use as a dictionary
+            mpc_server_urls (list): The list of dicts containing the url and the name
+              of the mpc server and the type of connection, and the bearer token if necessary
+              example:
+              [
+                {
+                  'url': 'http://localhost:8000',
+                  'name': 'mcp_server_1',
+                  'type': 'http','SSE'
+                  'headers': {'Authorization': 'Bearer', '1234567890'} #optional or None
+                }
+              ]
+            mpc_stdio_commands (list): The list of commands to use with the stdio mpc server
+              example:
+              [
+                {
+                  'name': 'memory',
+                  'command': 'npx', 'docker', 'npm', 'python'
+                  'args': ['-y', '@modelcontextprotocol/server-memory'] 
+                }
+              ]
+
+            
+        """
+        
+        self.api_keys=Api_keys(api_keys=api_keys)
+        
+        self.mpc_server_urls = mpc_server_urls
+        self.mpc_stdio_commands = mpc_stdio_commands
+        # tools
+        self.llms={'mcp_llm':OpenAIModel('gpt-4.1-mini',provider=OpenAIProvider(api_key=self.api_keys.api_keys['openai_api_key']))}
+        
+        
+        #mpc servers
+        self.mpc_servers=[]
+        for mpc_server_url in self.mpc_server_urls:
+            if mpc_server_url['type'] == 'http':
+                if mpc_server_url['headers'] is not None:
+                    self.mpc_servers.append(MCPServerStreamableHTTP(url=mpc_server_url['url'], headers=mpc_server_url['headers']))
+                else:
+                    self.mpc_servers.append(MCPServerStreamableHTTP(mpc_server_url['url']))
+            elif mpc_server_url['type'] == 'SSE':
+                if mpc_server_url['headers'] is not None:
+                    self.mpc_servers.append(MCPServerSSE(url=mpc_server_url['url'], headers=mpc_server_url['headers']))
+                else:
+                    self.mpc_servers.append(MCPServerSSE(mpc_server_url['url']))
+        for mpc_stdio_command in self.mpc_stdio_commands:
+            self.mpc_servers.append(MCPServerStdio(command=mpc_stdio_command['command'], args=mpc_stdio_command['args']))
+
+        self._mcp_context_manager = None
+        self._is_connected = False
+        #agent
+
+        self.agent=Agent(self.llms['mcp_llm'],tools=[], mcp_servers=self.mpc_servers, instructions="you are a helpful assistant that can help with a wide range of tasks,\
+                          you have the current time and the user query, you can use the tools provided to you if necessary to help the user with their queries, ask how you can help the user, sometimes the user will ask you not to use the tools, in this case you should not use the tools")
+        self.memory=Message_state(messages=[])
+        
+    
+    async def connect(self):
+        """Establish persistent connection to MCP server"""
+        if not self._is_connected:
+            self._mcp_context_manager = self.agent.run_mcp_servers()
+            await self._mcp_context_manager.__aenter__()
+            self._is_connected = True
+            return "Connected to MCP server"
+
+    async def disconnect(self):
+        """Close the MCP server connection"""
+        if self._is_connected and self._mcp_context_manager:
+            await self._mcp_context_manager.__aexit__(None, None, None)
+            self._is_connected = False
+            self._mcp_context_manager = None
+            return "Disconnected from MCP server"
+    async def chat(self, query:any):
+        """
+        # Chat Function Documentation
+
+        This function enables interaction with the user through various types of input.
+
+        ## Parameters
+
+        - `query`: The input to process. Can be one of the following types:
+        - String: Direct text input passed to the agent
+        - Binary content: Special format for media files (see below)
+
+        ## Binary Content Types
+
+        The function supports different types of media through `BinaryContent` objects:
+
+        ### Audio
+        ```python
+        agent.chat([
+            'optional string message',
+            BinaryContent(data=audio, media_type='audio/wav')
+        ])
+        ```
+
+        ### PDF Files
+        ```python
+        agent.chat([
+            'optional string message',
+            BinaryContent(data=pdf_path.read_bytes(), media_type='application/pdf')
+        ])
+        ```
+
+        ### Images
+        ```python
+        agent.chat([
+            'optional string message',
+            BinaryContent(data=image_response.content, media_type='image/png')
+        ])
+        ```
+
+        ## Returns
+
+        - `Agent_output`: as a pydantic object, the ui_version and voice_version are the two fields of the object
+
+        ## Extra Notes
+        The message_history of Agent can be accessed using the following code:
+        ```python
+        
+        agent.memory.messages
+        ```
+        """
+        if not self._is_connected:
+            await self.connect()
+            
+        result=await self.agent.run(query, message_history=self.memory.messages)
+        self.memory.messages=result.all_messages()
+        return result.output
+    
+    
+    
+    def reset(self):
+        """
+        Resets the Agent to its initial state.
+
+        Returns:
+            str: A confirmation message indicating that the agent has been reset.
+        """
+        self.memory.messages=[]
+        return f'Agent has been reset'
+    
+    async def __aenter__(self):
+        """Async context manager entry"""
+        await self.connect()
+        return self
+
+    async def __aexit__(self, exc_type, exc_val, exc_tb):
+        """Async context manager exit"""
+        await self.disconnect()