The New Archon (Beta) - The Operating System for AI Coding Assistants!

original_archon/iterations/v3-mcp-support/mcp_server.py

@@ -0,0 +1,95 @@
+import os
+import sys
+import asyncio
+import threading
+from mcp.server.fastmcp import FastMCP
+import requests
+from typing import Dict, List
+import uuid
+from utils.utils import write_to_log
+from graph_service import app
+import uvicorn
+
+
+# Initialize FastMCP server
+mcp = FastMCP("archon")
+
+
+# Store active threads
+active_threads: Dict[str, List[str]] = {}
+
+
+# FastAPI service URL
+GRAPH_SERVICE_URL = "http://127.0.0.1:8100"
+
+
+@mcp.tool()
+async def create_thread() -> str:
+    """Create a new conversation thread for Archon.
+    Always call this tool before invoking Archon for the first time in a conversation.
+    (if you don't already have a thread ID)
+    
+    Returns:
+        str: A unique thread ID for the conversation
+    """
+    thread_id = str(uuid.uuid4())
+    active_threads[thread_id] = []
+    write_to_log(f"Created new thread: {thread_id}")
+    return thread_id
+
+
+def _make_request(thread_id: str, user_input: str, config: dict) -> str:
+    """Make synchronous request to graph service"""
+    response = requests.post(
+        f"{GRAPH_SERVICE_URL}/invoke",
+        json={
+            "message": user_input,
+            "thread_id": thread_id,
+        "is_first_message": not active_threads[thread_id],
+            "config": config
+    }
+    )
+    response.raise_for_status()
+    return response.json()
+
+
+@mcp.tool()
+async def run_agent(thread_id: str, user_input: str) -> str:
+    """Run the Archon agent with user input.
+    Only use this tool after you have called create_thread in this conversation to get a unique thread ID.
+    If you already created a thread ID in this conversation, do not create another one. Reuse the same ID.
+    After you receive the code from Archon, always implement it into the codebase unless asked not to.
+    
+    Args:
+        thread_id: The conversation thread ID
+        user_input: The user's message to process
+    
+    Returns:
+        str: The agent's response which generally includes the code for the agent
+    """
+    if thread_id not in active_threads:
+        write_to_log(f"Error: Thread not found - {thread_id}")
+        raise ValueError("Thread not found")
+
+    write_to_log(f"Processing message for thread {thread_id}: {user_input}")
+
+    config = {
+        "configurable": {
+            "thread_id": thread_id
+        }
+    }
+    
+    try:
+        result = await asyncio.to_thread(_make_request, thread_id, user_input, config)
+        active_threads[thread_id].append(user_input)
+        return result['response']
+        
+    except Exception as e:
+        raise
+
+
+if __name__ == "__main__":
+    write_to_log("Starting MCP server")
+    
+    # Run MCP server
+    mcp.run(transport='stdio')