api/main.py

from fastapi import FastAPI
from api.endpoints import codelama, ollama

app = FastAPI()

# Register routers if you have them
app.include_router(codelama.router, prefix="/run-codelama")
app.include_router(ollama.router, prefix="/ollama-response")