DailyRESTHelper now receives an aiohttp client session

examples/deployment/flyio-example/bot_runner.py

@@ -9,6 +9,8 @@ import argparse
 import subprocess
 import os
 
+from contextlib import asynccontextmanager
+
 from fastapi import FastAPI, Request, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import JSONResponse
@@ -39,14 +41,23 @@ FLY_HEADERS = {
     'Content-Type': 'application/json'
 }
 
-daily_rest_helper = DailyRESTHelper(
-    os.getenv("DAILY_API_KEY", ""),
-    os.getenv("DAILY_API_URL", 'https://api.daily.co/v1'))
+daily_helpers = {}
 
 
 # ----------------- API ----------------- #
 
-app = FastAPI()
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    aiohttp_session = aiohttp.ClientSession()
+    daily_helpers["rest"] = DailyRESTHelper(
+        daily_api_key=os.getenv("DAILY_API_KEY", ""),
+        daily_api_url=os.getenv("DAILY_API_URL", 'https://api.daily.co/v1'),
+        aiohttp_session=aiohttp_session
+    )
+    yield
+    await aiohttp_session.close()
+
+app = FastAPI(lifespan=lifespan)
 
 app.add_middleware(
     CORSMiddleware,
@@ -127,7 +138,7 @@ async def start_bot(request: Request) -> JSONResponse:
             properties=DailyRoomProperties()
         )
         try:
-            room: DailyRoomObject = await daily_rest_helper.create_room(params=params)
+            room: DailyRoomObject = await daily_helpers["rest"].create_room(params=params)
         except Exception as e:
             raise HTTPException(
                 status_code=500,
@@ -135,13 +146,13 @@ async def start_bot(request: Request) -> JSONResponse:
     else:
         # Check passed room URL exists, we should assume that it already has a sip set up
         try:
-            room: DailyRoomObject = await daily_rest_helper.get_room_from_url(room_url)
+            room: DailyRoomObject = await daily_helpers["rest"].get_room_from_url(room_url)
         except Exception:
             raise HTTPException(
                 status_code=500, detail=f"Room not found: {room_url}")
 
     # Give the agent a token to join the session
-    token = await daily_rest_helper.get_token(room.url, MAX_SESSION_TIME)
+    token = await daily_helpers["rest"].get_token(room.url, MAX_SESSION_TIME)
 
     if not room or not token:
         raise HTTPException(
@@ -168,7 +179,7 @@ async def start_bot(request: Request) -> JSONResponse:
                 status_code=500, detail=f"Failed to spawn VM: {e}")
 
     # Grab a token for the user to join with
-    user_token = await daily_rest_helper.get_token(room.url, MAX_SESSION_TIME)
+    user_token = await daily_helpers["rest"].get_token(room.url, MAX_SESSION_TIME)
 
     return JSONResponse({
         "room_url": room.url,
@@ -200,6 +211,5 @@ if __name__ == "__main__":
             port=config.port,
             reload=config.reload
         )
-
     except KeyboardInterrupt:
         print("Pipecat runner shutting down...")