8fb75cf6a7
Rename Save changes Batches of 50 No sort Only write dialout false Add rate limit handling Save Improvements Save progress with recording Fixed stop Add backoff Save logs Make table nicer Move to new folder
94 lines
3.1 KiB
Python
94 lines
3.1 KiB
Python
import asyncio
|
|
import csv
|
|
import os
|
|
import subprocess
|
|
import time
|
|
from datetime import datetime
|
|
|
|
import aiohttp
|
|
|
|
from pipecat.transports.services.helpers.daily_rest import (
|
|
DailyRESTHelper,
|
|
DailyRoomParams,
|
|
DailyRoomProperties,
|
|
)
|
|
|
|
BOT_RUN_TIME = 300
|
|
|
|
|
|
async def run_bot(id: int, run_number: int, csv_writer):
|
|
async with aiohttp.ClientSession() as aiohttp_session:
|
|
print(f"Starting bot number: {id}")
|
|
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,
|
|
)
|
|
|
|
# Create daily.co room with dialin and dialout enabled
|
|
exp = time.time() + BOT_RUN_TIME
|
|
room_params = DailyRoomParams(
|
|
properties=DailyRoomProperties(
|
|
exp=exp,
|
|
enable_dialout=True,
|
|
eject_at_room_exp=True,
|
|
enable_recording="cloud",
|
|
)
|
|
)
|
|
|
|
try:
|
|
# Create the room with the specified parameters
|
|
room = await rest.create_room(room_params)
|
|
# Create token with owner permissions
|
|
token = await rest.get_token(
|
|
room_url=room.url,
|
|
expiry_time=60 * 60,
|
|
owner=True, # Ensure the token has owner permissions
|
|
)
|
|
# print(f"{id}: Room Token: {token}")
|
|
room_info = await rest.get_room_from_url(room.url)
|
|
# print(f"{id}: Room Info: {room_info}")
|
|
current_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S.%f")[:-3]
|
|
# Other party joined or not and start dialout joined
|
|
csv_writer.writerow([id, room_info.config.enable_dialout, current_time])
|
|
|
|
except Exception as e:
|
|
print(f"Error creating room for bot {id}: {e}")
|
|
print("Sleeping for 10 seconds")
|
|
await asyncio.sleep(10)
|
|
csv_writer.writerow(
|
|
[id, "Rate Limit Error", datetime.now().strftime("%Y-%m-%d %H:%M:%S.%f")[:-3]]
|
|
)
|
|
|
|
bot_proc = f"python3 -m batch_dialout_bot -u {room.url} -t {token} -i {id} -r {run_number}"
|
|
|
|
try:
|
|
subprocess.Popen(
|
|
[bot_proc], shell=True, bufsize=1, cwd=os.path.dirname(os.path.abspath(__file__))
|
|
)
|
|
except Exception as e:
|
|
print(f"Failed to start subprocess: {e}")
|
|
|
|
|
|
async def main():
|
|
# Open the CSV file in append mode
|
|
with open("output.csv", mode="w", newline="") as file:
|
|
csv_writer = csv.writer(file)
|
|
# Write the header row
|
|
csv_writer.writerow(["bot_id", "enable_dialout", "timestamp"])
|
|
|
|
for run_number in range(1):
|
|
print(f"-- Starting batch run number: {run_number}")
|
|
bots = [run_bot(i, run_number, csv_writer) for i in range(1)]
|
|
await asyncio.gather(*bots)
|
|
print("-- Batch finished, waiting 60 seconds...")
|
|
await asyncio.sleep(BOT_RUN_TIME + 60)
|
|
print("-- Finished waiting 60 seconds...")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
try:
|
|
asyncio.run(main())
|
|
except KeyboardInterrupt:
|
|
print("Parent process interrupted")
|