Update bot_definitions.py

bot_definitions.py

@@ -1,55 +1,49 @@
-# bot_definitions.py
-"""Definitions of different bot types for the bot registry."""
-
-from bot_registry import BotRegistry, BotType
-from bot_runner_helpers import (
-    create_call_transfer_settings,
-    create_simple_dialin_settings,
-    create_simple_dialout_settings,
-)
+from dataclasses import dataclass
+from typing import Dict, Optional, Any
 
-# Create and configure the bot registry
-bot_registry = BotRegistry()
-
-# Register bot types
-bot_registry.register(
-    BotType(
-        name="call_transfer",
-        settings_creator=create_call_transfer_settings,
-        required_settings=["dialin_settings"],
-        incompatible_with=["simple_dialin", "simple_dialout", "voicemail_detection"],
-        auto_add_settings={"dialin_settings": {}},
-    )
-)
 
-bot_registry.register(
-    BotType(
-        name="simple_dialin",
-        settings_creator=create_simple_dialin_settings,
-        required_settings=["dialin_settings"],
-        incompatible_with=["call_transfer", "simple_dialout", "voicemail_detection"],
-        auto_add_settings={"dialin_settings": {}},
-    )
+@dataclass
+class BotType:
+    name: str
+    module: str
+    config_key: str
+    environment_variables: list[str]
+    optional_environment_variables: list[str]
+
+    def has_test_mode(self, body: Dict[str, Any]) -> bool:
+        if self.config_key in body:
+            return bool(body[self.config_key].get("testInPrebuilt"))
+        return False
+
+
+call_transfer = BotType(
+    name="call_transfer",
+    module="call_transfer",
+    config_key="call_transfer",
+    environment_variables=["DAILY_API_KEY", "OPENAI_API_KEY", "CARTESIA_API_KEY"],
+    optional_environment_variables=["DIAL_IN_FROM_NUMBER", "DIAL_OUT_TO_NUMBER", "OPERATOR_NUMBER"],
 )
 
-bot_registry.register(
-    BotType(
-        name="simple_dialout",
-        settings_creator=create_simple_dialout_settings,
-        required_settings=["dialout_settings"],
-        incompatible_with=["call_transfer", "simple_dialin", "voicemail_detection"],
-        auto_add_settings={"dialout_settings": [{}]},
-    )
+simple_dialin = BotType(
+    name="simple_dialin",
+    module="simple_dialin",
+    config_key="simple_dialin",
+    environment_variables=["DAILY_API_KEY", "OPENAI_API_KEY", "CARTESIA_API_KEY"],
+    optional_environment_variables=["DIAL_IN_FROM_NUMBER"],
 )
 
-bot_registry.register(
-    BotType(
-        name="voicemail_detection",
-        settings_creator=lambda body: body.get(
-            "voicemail_detection", {}
-        ),  # No creator function in original code
-        required_settings=["dialout_settings"],
-        incompatible_with=["call_transfer", "simple_dialin", "simple_dialout"],
-        auto_add_settings={"dialout_settings": [{}]},
-    )
+simple_dialout = BotType(
+    name="simple_dialout",
+    module="simple_dialout",
+    config_key="simple_dialout",
+    environment_variables=["DAILY_API_KEY", "OPENAI_API_KEY", "CARTESIA_API_KEY"],
+    optional_environment_variables=["DIAL_OUT_TO_NUMBER"],
 )
+
+voicemail_detection = BotType(
+    name="voicemail_detection",
+    module="voicemail_detection",
+    config_key="voicemail_detection",
+    environment_variables=["DAILY_API_KEY", "OPENAI_API_KEY", "DEEPGRAM_API_KEY", "CARTESIA_API_KEY"],
+    optional_environment_variables=["DIAL_OUT_TO_NUMBER"],
+)