49 lines
1.1 KiB
Python
49 lines
1.1 KiB
Python
from fastapi import FastAPI
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from fastapi.responses import StreamingResponse
|
|
from fastapi.encoders import jsonable_encoder
|
|
import uvicorn
|
|
|
|
from . import tools
|
|
|
|
|
|
app = FastAPI(
|
|
title='Clients',
|
|
summary='Provides various clients.'
|
|
)
|
|
|
|
origins = [
|
|
"http://localhost.tiangolo.com",
|
|
"https://localhost.tiangolo.com",
|
|
"http://localhost",
|
|
"http://localhost:8080",
|
|
]
|
|
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=origins,
|
|
allow_credentials=True,
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
|
|
def setup_mcp():
|
|
try:
|
|
from fastapi_mcp import FastApiMCP
|
|
mcp = FastApiMCP(app)
|
|
mcp.mount()
|
|
print('MCP mounted')
|
|
except ImportError:
|
|
print('MCP could not be mounted')
|
|
|
|
def setup_tools():
|
|
for tool in tools.get_tools():
|
|
component, method = tool.__name__.split('.')
|
|
path = f'/{component}/{method}'
|
|
app.get(path, response_model=None,tags=[component])(tool)
|
|
|
|
if __name__ == "__main__":
|
|
setup_tools()
|
|
setup_mcp()
|
|
uvicorn.run(app, host="0.0.0.0", port=8000)
|