diff --git a/src/mcp_server_cli/server.py b/src/mcp_server_cli/server.py index 91264c0..07d306e 100644 --- a/src/mcp_server_cli/server.py +++ b/src/mcp_server_cli/server.py @@ -1,34 +1,29 @@ """MCP Protocol Server implementation using FastAPI.""" -import asyncio -import json import logging from contextlib import asynccontextmanager -from typing import Any, Dict, List, Optional, Callable, Awaitable from enum import Enum +from pathlib import Path +from typing import Dict, List, Optional -from fastapi import FastAPI, HTTPException, Request, Depends +from fastapi import FastAPI, HTTPException, Request from fastapi.middleware.cors import CORSMiddleware -from fastapi.responses import JSONResponse, StreamingResponse -from pydantic import BaseModel -import sse_starlette.sse as sse +from mcp_server_cli.config import AppConfig, ConfigManager from mcp_server_cli.models import ( - MCPRequest, - MCPResponse, - MCPNotification, - MCPMethod, - ToolDefinition, - ToolCallParams, - ToolCallResult, InitializeParams, InitializeResult, - ServerInfo, + MCPMethod, + MCPRequest, + MCPResponse, ServerCapabilities, + ServerInfo, + ToolCallParams, + ToolCallResult, + ToolDefinition, ToolsListResult, ) -from mcp_server_cli.config import AppConfig, ConfigManager -from mcp_server_cli.tools import ToolBase, ToolResult +from mcp_server_cli.tools import ToolBase logger = logging.getLogger(__name__)