server/connectors/base.py

@@ -7,6 +7,9 @@ class BaseConnector(ABC):
     display_name: str      # human-readable: "Reddit", "YouTube"
     required_env: list[str] = []  # env vars needed to activate
 
+    search_enabled: bool
+    categories_enabled: bool
+
     @classmethod
     def is_available(cls) -> bool:
         """Returns True if all required env vars are set."""

server/connectors/boards_api.py

@@ -19,6 +19,9 @@ class BoardsAPI(BaseConnector):
     source_name: str = "boards.ie"
     display_name: str = "Boards.ie"
 
+    categories_enabled: bool = True
+    search_enabled: bool = False
+
     def __init__(self):
         self.url = "https://www.boards.ie"
 

server/connectors/reddit_api.py

@@ -10,8 +10,10 @@ from server.connectors.base import BaseConnector
 logger = logging.getLogger(__name__)
 
 class RedditAPI(BaseConnector):
-    source_name = "reddit"
-    display_name = "Reddit"
+    source_name: str = "reddit"
+    display_name: str = "Reddit"
+    search_enabled: bool = True
+    categories_enabled: bool = True
 
     def __init__(self):
         self.url = "https://www.reddit.com/"

server/connectors/registry.py

@@ -18,8 +18,13 @@ def _discover_connectors() -> list[type[BaseConnector]]:
 def get_available_connectors() -> dict[str, type[BaseConnector]]:
     return {c.source_name: c for c in _discover_connectors() if c.is_available()}
 
-def get_connector_metadata() -> list[dict]:
-    return [
-        {"id": id, "label": obj.display_name}
-        for id, obj in get_available_connectors().items()
-    ]
+def get_connector_metadata() -> dict[str, dict]:
+    res = {}
+    for id, obj in get_available_connectors().items():
+        res[id] = {"id": id, 
+                   "label": obj.display_name, 
+                   "search_enabled": obj.search_enabled, 
+                   "categories_enabled": obj.categories_enabled
+                   }
+
+    return res

server/connectors/youtube_api.py

@@ -13,8 +13,10 @@ load_dotenv()
 API_KEY = os.getenv("YOUTUBE_API_KEY")
 
 class YouTubeAPI(BaseConnector):
-    source_name = "youtube"
-    display_name = "YouTube"
+    source_name: str = "youtube"
+    display_name: str = "YouTube"
+    search_enabled: bool = True
+    categories_enabled: bool = False
 
     def __init__(self):
         self.youtube = build('youtube', 'v3', developerKey=API_KEY)