"""
Example REST API Server
Simple REST API endpoints that can be wrapped as MCP tools
"""
from fastapi import FastAPI, HTTPException
from pydantic import BaseModel
from typing import Optional, Dict, Any
import uvicorn
app = FastAPI(title="MCP REST API Tools", version="1.0.0")
class StockPriceRequest(BaseModel):
symbol: str
class StockPriceResponse(BaseModel):
symbol: str
price: float
currency: str
change: float
change_percent: float
@app.get("/health")
async def health_check():
"""Health check endpoint"""
return {"status": "healthy"}
@app.post("/stock/price", response_model=StockPriceResponse)
async def get_stock_price(request: StockPriceRequest):
"""
Get stock price information
This endpoint can be wrapped as an MCP tool
"""
# Mock stock data (in production, call actual stock API)
mock_stocks = {
"AAPL": {"price": 175.50, "change": 2.30, "change_percent": 1.33},
"GOOGL": {"price": 142.20, "change": -1.10, "change_percent": -0.77},
"MSFT": {"price": 378.85, "change": 5.20, "change_percent": 1.39},
}
symbol = request.symbol.upper()
if symbol not in mock_stocks:
raise HTTPException(status_code=404, detail=f"Stock symbol {symbol} not found")
stock_data = mock_stocks[symbol]
return StockPriceResponse(
symbol=symbol,
price=stock_data["price"],
currency="USD",
change=stock_data["change"],
change_percent=stock_data["change_percent"]
)
@app.get("/time/current")
async def get_current_time():
"""
Get current time
This endpoint can be wrapped as an MCP tool
"""
from datetime import datetime, timezone
utc_now = datetime.now(timezone.utc)
return {
"utc_time": utc_now.isoformat(),
"unix_timestamp": int(utc_now.timestamp()),
"timezone": "UTC"
}
@app.post("/text/summarize")
async def summarize_text(text: Dict[str, str]):
"""
Summarize text
This endpoint can be wrapped as an MCP tool
"""
content = text.get("text", "")
if not content:
raise HTTPException(status_code=400, detail="Text field is required")
# Simple mock summarization (in production, use actual NLP model)
words = content.split()
summary = " ".join(words[:20]) + "..." if len(words) > 20 else content
return {
"original_length": len(content),
"summary_length": len(summary),
"summary": summary,
"word_count": len(words)
}
if __name__ == "__main__":
uvicorn.run(app, host="0.0.0.0", port=8000)
