routes.py 1.6 KB

123456789101112131415161718192021222324252627282930313233343536
  1. from core_domain import ChatCompletionRequestContract, ChatCompletionResponseContract, ServiceHealth
  2. from fastapi import APIRouter, Depends, HTTPException
  3. from app.application.services import ModelGatewayApplicationService
  4. from app.bootstrap.settings import ModelGatewayServiceSettings
  5. from app.infrastructure.provider import ModelProviderClient, ModelProviderClientError
  6. router = APIRouter()
  7. def get_model_gateway_settings() -> ModelGatewayServiceSettings:
  8. return ModelGatewayServiceSettings()
  9. def get_model_gateway_application_service(
  10. settings: ModelGatewayServiceSettings = Depends(get_model_gateway_settings)) -> ModelGatewayApplicationService:
  11. return ModelGatewayApplicationService(
  12. provider_client=ModelProviderClient(settings=settings),
  13. settings=settings)
  14. @router.get("/health", response_model=ServiceHealth)
  15. def health_check(
  16. settings: ModelGatewayServiceSettings = Depends(get_model_gateway_settings)) -> ServiceHealth:
  17. provider_status = "configured" if settings.provider_base_url else "missing"
  18. return ServiceHealth(service="model-gateway-service", status="ok", database=provider_status)
  19. @router.post("/chat-completions", response_model=ChatCompletionResponseContract)
  20. def create_chat_completion(
  21. payload: ChatCompletionRequestContract,
  22. service: ModelGatewayApplicationService = Depends(get_model_gateway_application_service)) -> ChatCompletionResponseContract:
  23. try:
  24. return service.create_chat_completion(payload)
  25. except ModelProviderClientError as exc:
  26. raise HTTPException(status_code=502, detail=str(exc)) from exc