diff --git a/ops_bot/main.py b/ops_bot/main.py index 90e0f64..c3999b0 100644 --- a/ops_bot/main.py +++ b/ops_bot/main.py @@ -1,7 +1,7 @@ import asyncio import json import logging -from typing import Any, Dict, Literal, Optional, cast +from typing import Any, Dict, Literal, Optional, Tuple, cast import uvicorn from fastapi import Depends, FastAPI, HTTPException, Request, status @@ -77,7 +77,7 @@ def get_destination(bot_settings: BotSettings, routing_key: str) -> Optional[str return bot_settings.routing_keys.get(routing_key, None) -async def receive_helper(request: Request): +async def receive_helper(request: Request) -> Tuple[str, Any]: payload: Any = await request.json() routing_key = request.path_params["routing_key"] room_id = get_destination(request.app.state.bot_settings, routing_key) @@ -88,7 +88,7 @@ async def receive_helper(request: Request): ) payload_str = json.dumps(payload, sort_keys=True, indent=2) logging.info(f"received payload: \n {payload_str}") - return payload + return room_id, payload @app.post("/hook/pagerduty/{routing_key}")