feat: add YouTube connection

agents/example.json

@@ -31,6 +31,10 @@
       "timeline_read_count": 10,
       "cast_interval": 60
     },
+    {
+      "name": "youtube",
+      "comment_fetch_count": 10
+    },
     {
       "name": "openai",
       "model": "gpt-3.5-turbo"

pyproject.toml

@@ -5,6 +5,9 @@ description = "A launch-pad for AI agents"
 authors = ["Ayoub Ed <[email protected]>", "Efrain <[email protected]>", "Max <[email protected]>"]
 license = "MIT License"
 readme = "README.md"
+packages = [
+    { include = "src" }
+]
 
 [tool.poetry.dependencies]
 python = "^3.10"
@@ -14,7 +17,9 @@ tweepy = "^4.14.0"
 prompt-toolkit = "^3.0.48"
 anthropic = "^0.42.0"
 farcaster = "^0.7.11"
-
+google-api-python-client = "^2.97.0"
+google-auth-oauthlib = "^1.0.0"
+google-auth-httplib2 = "^0.1.0"
 
 [build-system]
 requires = ["poetry-core"]

src/agent.py

@@ -37,8 +37,10 @@ def __init__(
             self.example_accounts = agent_dict["example_accounts"]
             self.loop_delay = agent_dict["loop_delay"]
             self.connection_manager = ConnectionManager(agent_dict["config"])
+            self.connection_manager.parent_agent = self
             self.use_time_based_weights = agent_dict["use_time_based_weights"]
             self.time_based_multipliers = agent_dict["time_based_multipliers"]
+
 
             has_twitter_tasks = any("tweet" in task["name"] for task in agent_dict.get("tasks", []))
 

src/connection_manager.py

@@ -9,12 +9,14 @@
 from src.connections.ollama_connection import OllamaConnection
 from src.connections.echochambers_connection import EchochambersConnection
 from src.connections.hyperbolic_connection import HyperbolicConnection
+from src.connections.youtube_connection import YouTubeConnection
 
 logger = logging.getLogger("connection_manager")
 
 class ConnectionManager:
     def __init__(self, agent_config):
         self.connections : Dict[str, BaseConnection] = {}
+        self.parent_agent = None
         for config in agent_config:
             self._register_connection(config)
 
@@ -36,6 +38,8 @@ def _class_name_to_type(class_name: str) -> Type[BaseConnection]:
             return EchochambersConnection
         elif class_name == "hyperbolic":
             return HyperbolicConnection
+        elif class_name == "youtube":
+            return YouTubeConnection
 
         return None
 
@@ -52,6 +56,7 @@ def _register_connection(self, config_dic: Dict[str, Any]) -> None:
             name = config_dic["name"]
             connection_class = self._class_name_to_type(name)
             connection = connection_class(config_dic)
+            connection.connection_manager = self
             self.connections[name] = connection
         except Exception as e:
             logging.error(f"Failed to initialize connection {name}: {e}")