DailyRESTHelper now receives an aiohttp client session

examples/translation-chatbot/bot.py

@@ -4,6 +4,7 @@
 # SPDX-License-Identifier: BSD 2-Clause License
 #
 
+import aiohttp
 import asyncio
 import os
 import sys
@@ -89,57 +90,58 @@ class TranslationSubtitles(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,
-        "Translator",
-        DailyParams(
-            audio_out_enabled=True,
-            transcription_enabled=True,
-            transcription_settings=DailyTranscriptionSettings(extra={
-                "interim_results": False
-            })
+        transport = DailyTransport(
+            room_url,
+            token,
+            "Translator",
+            DailyParams(
+                audio_out_enabled=True,
+                transcription_enabled=True,
+                transcription_settings=DailyTranscriptionSettings(extra={
+                    "interim_results": False
+                })
+            )
         )
-    )
 
-    tts = AzureTTSService(
-        api_key=os.getenv("AZURE_SPEECH_API_KEY"),
-        region=os.getenv("AZURE_SPEECH_REGION"),
-        voice="es-ES-AlvaroNeural",
-    )
+        tts = AzureTTSService(
+            api_key=os.getenv("AZURE_SPEECH_API_KEY"),
+            region=os.getenv("AZURE_SPEECH_REGION"),
+            voice="es-ES-AlvaroNeural",
+        )
 
-    llm = OpenAILLMService(
-        api_key=os.getenv("OPENAI_API_KEY"),
-        model="gpt-4o"
-    )
+        llm = OpenAILLMService(
+            api_key=os.getenv("OPENAI_API_KEY"),
+            model="gpt-4o"
+        )
 
-    sa = SentenceAggregator()
-    tp = TranslationProcessor("Spanish")
-    lfra = LLMFullResponseAggregator()
-    ts = TranslationSubtitles("spanish")
+        sa = SentenceAggregator()
+        tp = TranslationProcessor("Spanish")
+        lfra = LLMFullResponseAggregator()
+        ts = TranslationSubtitles("spanish")
 
-    pipeline = Pipeline([
-        transport.input(),
-        sa,
-        tp,
-        llm,
-        lfra,
-        ts,
-        tts,
-        transport.output()
-    ])
+        pipeline = Pipeline([
+            transport.input(),
+            sa,
+            tp,
+            llm,
+            lfra,
+            ts,
+            tts,
+            transport.output()
+        ])
 
-    task = PipelineTask(pipeline)
+        task = PipelineTask(pipeline)
 
-    @transport.event_handler("on_first_participant_joined")
-    async def on_first_participant_joined(transport, participant):
-        transport.capture_participant_transcription(participant["id"])
+        @transport.event_handler("on_first_participant_joined")
+        async def on_first_participant_joined(transport, participant):
+            transport.capture_participant_transcription(participant["id"])
 
-    runner = PipelineRunner()
+        runner = PipelineRunner()
 
-    await runner.run(task)
+        await runner.run(task)
 
 
 if __name__ == "__main__":

examples/translation-chatbot/runner.py

@@ -7,10 +7,12 @@
 import argparse
 import os
 
+import aiohttp
+
 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 +40,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/translation-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(