files with questions data

.gitignore

@@ -1,6 +1,13 @@
 
 # Omit HF Questions Data
 data/*.json
+data/1f975693-876d-457b-a649-393859e79bf3.mp3
+data/7bd855d8-463d-4ed5-93ca-5fe35145f733.xlsx
+data/99c9cc74-fdc8-46c6-8f8d-3ce2d3bfeea3.mp3
+data/cca530fc-4052-43b2-b130-b30968d8aa44.png
+data/f918266a-b3e0-4914-865d-4faa564f1aef.py
+
+
 
 # omit IDE configs
 .vscode/settings.json

agents/basic_agent.py

@@ -1,14 +1,19 @@
 """Base Agent"""
 
+from agents.manager_agent import manager_agent_factory
+
 
 class BasicAgent:  # pylint: disable=too-few-public-methods
     """Base Agent for Evaluation"""
 
     def __init__(self):
+        self.manager_agent = manager_agent_factory()
         print("BasicAgent initialized.")
 
     def __call__(self, question: str) -> str:
-        print(f"Agent received question (first 50 chars): {question[:50]}...")
-        fixed_answer = "This is a default answer."
-        print(f"Agent returning fixed answer: {fixed_answer}")
-        return fixed_answer
+        if not self.manager_agent:
+            print(f"Agent received question (first 50 chars): {question[:50]}...")
+            fixed_answer = "This is a default answer."
+            print(f"Agent returning fixed answer: {fixed_answer}")
+            return fixed_answer
+        return self.manager_agent.run(question)

agents/manager_agent.py

@@ -0,0 +1,23 @@
+""" "Manger Agent"""
+
+import os
+
+from smolagents import CodeAgent, OpenAIServerModel, DuckDuckGoSearchTool
+
+
+MODEL_ID = os.getenv("MANAGER_MODEL_ID")
+BASE_URL = os.getenv("MANGER_BASE_URL")
+API_KEY = os.getenv("OPENAI_API_KEY")
+API_BASE = "/".join([BASE_URL, "v1"])
+
+model = OpenAIServerModel(model_id=MODEL_ID, api_base=API_BASE, api_key=API_KEY)
+
+
+def manager_agent_factory() -> CodeAgent:
+    """Manager Agent Instance"""
+    return CodeAgent(
+        model=model,
+        tools=[DuckDuckGoSearchTool(verify=False)],
+        planning_interval=3,
+        max_steps=20,
+    )

app.py

@@ -12,8 +12,7 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 API_URL = DEFAULT_API_URL
 QUESTIONS_URL = f"{API_URL}/questions"
 SUBMIT_URL = f"{API_URL}/submit"
-
-SAFETY_OFF = os.getenv("SAFETY_OFF")
+FILES_URL =  f"{API_URL}/files"
 
 SPACE_ID = os.getenv("SPACE_ID")
 if SPACE_ID:
@@ -25,6 +24,13 @@ def _check_agent_dependent_environment_variables() -> str:
     openai_api_key = os.getenv("OPENAI_API_KEY")
     if not openai_api_key:
         return "OPENAI_API_KEY environment variable must be set"
+    manger_model_id = os.getenv("MANAGER_MODEL_ID")
+    if not manger_model_id:
+        return "MANAGER_MODEL_ID environment variable must be set"
+    manager_base_url = os.getenv("MANGER_BASE_URL")
+    if not manager_base_url:
+        return "MANGER_BASE_URL environment variable must be set"
+
     return None
 
 
@@ -46,7 +52,7 @@ def run_and_submit_all(
     if error_message:
         return error_message, None
 
-    questions_data = get_questions_data(questions_url=QUESTIONS_URL)
+    questions_data = get_questions_data(questions_url=QUESTIONS_URL, files_url=FILES_URL)
     if not questions_data:
         print("Questions list is empty.")
         return "Questions list is empty or invalid format.", None
@@ -67,13 +73,9 @@ def run_and_submit_all(
     print(status_update)
     print(f"Submitting {len(answers_payload)} answers to: {SUBMIT_URL}")
 
-    if SAFETY_OFF:
-        final_status, error_message = submit_answers(
-            submit_url=SUBMIT_URL, submission_data=submission_data
-        )
-    else:
-        final_status = "\n\tSAFETY_OFF bypass active\n"
-        error_message = None
+    final_status, error_message = submit_answers(
+        submit_url=SUBMIT_URL, submission_data=submission_data
+    )
 
     if final_status:
         return final_status, results_df

questions.py

@@ -7,6 +7,20 @@ import requests
 QUESTIONS_DATA_FILE = "data/questions_data.json"
 
 
+def _get_task_files(files_url: str, questions_data: dict):
+    for item in questions_data:
+        task_id = item.get("task_id")
+        file_name = item.get("file_name")
+        local_file = "/".join(["data", file_name])
+        if task_id and file_name:
+            file_url = "/".join([files_url,  task_id])
+            response = requests.get(file_url, stream=True)
+            response.raise_for_status()
+            with open(local_file, "wb") as f:
+                f.write(response.content)
+                print(f"File '{file_name}' downloaded successfully.")
+
+
 def _is_exists_questions_data_file() -> bool:
     if os.path.exists(QUESTIONS_DATA_FILE):
         print(f"The file '{QUESTIONS_DATA_FILE}' exists.")
@@ -52,7 +66,7 @@ def _json_question_data_from_res(res: requests.Response) -> dict:
     return None
 
 
-def get_questions_data(questions_url: str) -> dict:
+def get_questions_data(questions_url: str, files_url: str) -> dict:
     """Get Questions Data"""
     if _is_exists_questions_data_file():
         questions_data = _read_question_data_from_file()
@@ -62,5 +76,6 @@ def get_questions_data(questions_url: str) -> dict:
         questions_data = _json_question_data_from_res(res)
         if questions_data:
             _write_question_data_to_file(questions_data=questions_data)
+            _get_task_files(files_url=files_url, questions_data=questions_data)
             return questions_data
     return None

requirements.txt

@@ -1,3 +1,7 @@
+duckduckgo_search
+gradio
 gradio[oauth]
 requests
+smolagents
+smolagents[openai]