Update agents

agent.py

@@ -39,7 +39,7 @@ def fetch_file(task_id: str, file_name: str):
     try:
         response = requests.get(f"{DEFAULT_API_URL}/files/{task_id}")
         response.raise_for_status()
-        with open(f"/content/Final_Assignment_Template/data/question_files/{file_name}", "wb") as f:
+        with open(f"data/question_files/{file_name}", "wb") as f:
             f.write(response.content)
         file_content = response.content
         return file_content
@@ -90,7 +90,7 @@ def read_python_file(file_name: str) -> str:
     Returns:
         The content of the file.
     """
-    base_path = "/content/Final_Assignment_Template/data/question_files"
+    base_path = "data/question_files"
     with open(os.path.join(base_path, file_name), "r") as f:
         return f.read()
 
@@ -102,7 +102,7 @@ def read_excel_file(file_name: str) -> str:
     Returns:
         The content of the file.
     """
-    base_path = "/content/Final_Assignment_Template/data/question_files"
+    base_path = "data/question_files"
     df = pd.read_excel(os.path.join(base_path, file_name))
     return df.to_string()
 
@@ -145,7 +145,7 @@ print(API_KEY, "HEELLLOOoooooo", os.getenv("OPENAI_API_KEY_AG"))
 model = OpenAIServerModel(model_id="gpt-4.1-nano", api_key=API_KEY)
 MODEL_ID = "openai/gpt-4.1-nano"
 
-with open("/content/Final_Assignment_Template/system_prompt.yaml", 'r') as stream:
+with open("system_prompt.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
 
 agent = CodeAgent(
@@ -222,7 +222,7 @@ for index, question in enumerate(questions):
         }
     )
 import json
-with open(f"/content/Final_Assignment_Template/data/answers_with_prompt_{MODEL_ID.split('/')[-1]}_with_file_content_handling.json", "w") as f:
+with open(f"data/answers_with_prompt_{MODEL_ID.split('/')[-1]}_with_file_content_handling.json", "w") as f:
     json.dump(answers, f, indent=2)
 print("Submitting answers...")
 submit_answers(answers)