DailyRESTHelper now receives an aiohttp client session

examples/simple-chatbot/bot.py

@@ -84,8 +84,9 @@ class TalkingAnimation(FrameProcessor):
 
 
 async def main():
-    (room_url, token) = await configure()
     async with aiohttp.ClientSession() as session:
+        (room_url, token) = await configure(session)
+
         transport = DailyTransport(
             room_url,
             token,

examples/simple-chatbot/runner.py

@@ -4,13 +4,14 @@
 # SPDX-License-Identifier: BSD 2-Clause License
 #
 
+import aiohttp
 import argparse
 import os
 
 from pipecat.transports.services.helpers.daily_rest import DailyRESTHelper
 
 
-async def configure():
+async def configure(aiohttp_session: aiohttp.ClientSession):
     parser = argparse.ArgumentParser(description="Daily AI SDK Bot Sample")
     parser.add_argument(
         "-u",
@@ -38,7 +39,11 @@ async def configure():
     if not key:
         raise Exception("No Daily API key specified. use the -k/--apikey option from the command line, or set DAILY_API_KEY in your environment to specify a Daily API key, available from https://dashboard.daily.co/developers.")
 
-    daily_rest_helper = DailyRESTHelper(key, os.getenv("DAILY_API_URL", "https://api.daily.co/v1"))
+    daily_rest_helper = DailyRESTHelper(
+        daily_api_key=key,
+        daily_api_url=os.getenv("DAILY_API_URL", "https://api.daily.co/v1"),
+        aiohttp_session=aiohttp_session
+    )
 
     # Create a meeting token for the given room with an expiration 1 hour in
     # the future.

examples/simple-chatbot/server.py

@@ -4,10 +4,12 @@
 # SPDX-License-Identifier: BSD 2-Clause License
 #
 
+import aiohttp
 import os
 import argparse
 import subprocess
-import atexit
+
+from contextlib import asynccontextmanager
 
 from fastapi import FastAPI, Request, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
@@ -20,9 +22,7 @@ MAX_BOTS_PER_ROOM = 1
 # Bot sub-process dict for status reporting and concurrency control
 bot_procs = {}
 
-daily_rest_helper = DailyRESTHelper(
-    os.getenv("DAILY_API_KEY", ""),
-    os.getenv("DAILY_API_URL", 'https://api.daily.co/v1'))
+daily_helpers = {}
 
 
 def cleanup():
@@ -33,10 +33,19 @@ def cleanup():
         proc.wait()
 
 
-atexit.register(cleanup)
+@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()
+    cleanup()
 
-
-app = FastAPI()
+app = FastAPI(lifespan=lifespan)
 
 app.add_middleware(
     CORSMiddleware,
@@ -50,7 +59,7 @@ app.add_middleware(
 @app.get("/start")
 async def start_agent(request: Request):
     print(f"!!! Creating room")
-    room = await daily_rest_helper.create_room(DailyRoomParams())
+    room = await daily_helpers["rest"].create_room(DailyRoomParams())
     print(f"!!! Room URL: {room.url}")
     # Ensure the room property is present
     if not room.url:
@@ -66,7 +75,7 @@ async def start_agent(request: Request):
             status_code=500, detail=f"Max bot limited reach for room: {room.url}")
 
     # Get the token for the room
-    token = await daily_rest_helper.get_token(room.url)
+    token = await daily_helpers["rest"].get_token(room.url)
 
     if not token:
         raise HTTPException(