routes.py 1.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940
  1. from fastapi import APIRouter, Depends, HTTPException
  2. from core_domain import ChatCompletionRequestContract, ChatCompletionResponseContract, ServiceHealth
  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),
  11. ) -> ModelGatewayApplicationService:
  12. return ModelGatewayApplicationService(
  13. provider_client=ModelProviderClient(settings=settings),
  14. settings=settings,
  15. )
  16. @router.get("/health", response_model=ServiceHealth)
  17. def health_check(
  18. settings: ModelGatewayServiceSettings = Depends(get_model_gateway_settings),
  19. ) -> ServiceHealth:
  20. provider_status = "configured" if settings.provider_base_url else "missing"
  21. return ServiceHealth(service="model-gateway-service", status="ok", database=provider_status)
  22. @router.post("/chat-completions", response_model=ChatCompletionResponseContract)
  23. def create_chat_completion(
  24. payload: ChatCompletionRequestContract,
  25. service: ModelGatewayApplicationService = Depends(get_model_gateway_application_service),
  26. ) -> ChatCompletionResponseContract:
  27. try:
  28. return service.create_chat_completion(payload)
  29. except ModelProviderClientError as exc:
  30. raise HTTPException(status_code=502, detail=str(exc)) from exc