app.pyā¢3.35 kB
"""FastAPI application for Databricks App Template."""
import os
from pathlib import Path
import yaml
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from fastapi.staticfiles import StaticFiles
from fastmcp import FastMCP
from server.prompts import load_prompts
from server.routers import router
from server.routers.agent_chat import router as agent_router
from server.routers.registry import router as registry_router
from server.routers.db_resources import router as db_resources_router
from server.tools import load_tools
# Load environment variables from .env.local if it exists
def load_env_file(filepath: str) -> None:
"""Load environment variables from a file."""
if Path(filepath).exists():
with open(filepath) as f:
for line in f:
line = line.strip()
if line and not line.startswith('#'):
key, _, value = line.partition('=')
if key and value:
os.environ[key] = value
# Load .env files
load_env_file('.env')
load_env_file('.env.local')
# Load configuration from config.yaml
def load_config() -> dict:
"""Load configuration from config.yaml."""
config_path = Path('config.yaml')
if config_path.exists():
with open(config_path, 'r') as f:
return yaml.safe_load(f)
return {}
config = load_config()
servername = config.get('servername', 'databricks-mcp')
# Create MCP server
mcp_server = FastMCP(name=servername)
# Load prompts and tools
load_prompts(mcp_server)
load_tools(mcp_server)
# Create ASGI app from MCP server
# Passing no path automatically hosts this at the /mcp route
mcp_asgi_app = mcp_server.http_app()
# Pass the MCP app's lifespan to FastAPI
app = FastAPI(
title='Databricks App API',
description='Modern FastAPI application template for Databricks Apps with React frontend',
version='0.1.0',
lifespan=mcp_asgi_app.lifespan,
)
app.add_middleware(
CORSMiddleware,
allow_origins=[
'http://localhost:3000',
'http://127.0.0.1:3000',
'http://localhost:5173',
'http://127.0.0.1:5173',
],
allow_credentials=True,
allow_methods=['*'],
allow_headers=['*'],
)
app.include_router(router, prefix='/api', tags=['api'])
app.include_router(agent_router, prefix='/api/agent', tags=['agent'])
app.include_router(registry_router, prefix='/api/registry', tags=['registry'])
app.include_router(db_resources_router, prefix='/api/db', tags=['database'])
# ============================================================================
# SERVE STATIC FILES FROM CLIENT BUILD DIRECTORY (MUST BE LAST!)
# ============================================================================
# This static file mount MUST be the last route registered!
# It catches all unmatched requests and serves the React app.
# Any routes added after this will be unreachable!
if os.path.exists('client/build'):
app.mount('/', StaticFiles(directory='client/build', html=True), name='static')
# Combines both the routes from FastMCP and FastAPI
combined_app = FastAPI(
title="Combined MCP App",
routes=[
*mcp_asgi_app.routes, # MCP routes
*app.routes, # Original API routes
],
lifespan=mcp_asgi_app.lifespan,
)
if __name__ == '__main__':
import uvicorn
port = int(os.environ.get('DATABRICKS_APP_PORT', 8000))
uvicorn.run(combined_app, host='0.0.0.0', port=port)