Patch 184dc24004ca7c09767440e2217769b5c13846c2

app.py

@@ -5,7 +5,7 @@ from CustomAgent import CustomAgent
 from utils import get_agents, run_and_submit_all
 
 # Get all available agents
-AGENTS = get_agents()
+AGENTS = get_agents(globals(), BasicAgent)
 
 # Demo app
 with gr.Blocks() as demo:

utils.py

@@ -1,6 +1,5 @@
 import inspect
 import os
-import sys
 from typing import Any
 
 import gradio as gr
@@ -13,14 +12,14 @@ from BasicAgent import BasicAgent
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 
-def get_agents() -> dict[str, type]:
+def get_agents(namespace: dict[str, type], parent_cls: object) -> dict[str, type]:
     """
     Get all agents that inherit from BasicAgent, including imported ones.
     """
     return {
         name: obj
-        for name, obj in globals().items()
-        if inspect.isclass(obj) and issubclass(obj, BasicAgent)
+        for name, obj in namespace.items()
+        if inspect.isclass(obj) and issubclass(obj, parent_cls)
     }