src/controllers/__init__.py

from fastapi import APIRouter, Depends
from fastapi.security import APIKeyHeader

from ._auth_controller import AuthController
from ._file_controller import FileController
from ._conversation_controller import ConversationController

api_router = APIRouter()
websocket_router = APIRouter()

auth_router = AuthController().api_router
conversation_api_router = ConversationController().api_router
conversation_websocket_router = ConversationController().websocket_router
file_router = FileController().api_router

API_KEY_HEADER = APIKeyHeader(name="Authorization", auto_error=False)

api_router.include_router(
    auth_router,
    prefix="/v1",
    tags=["Authentication"],
)
api_router.include_router(
    conversation_api_router,
    prefix="/v1",
    tags=["Conversation"],
)
websocket_router.include_router(
    conversation_websocket_router,
    prefix="/v1",
    tags=["Conversation"],
)
api_router.include_router(
    file_router,
    prefix="/v1",
    tags=["Semantic Search"],
)

__all__ = ["api_router", "websocket_router"]
__version__ = "0.1.0"
__author__ = "Ramanjit Singh"