Update agent.py

agent.py

@@ -398,19 +398,24 @@ def create_documents(data_source: str, data: list) -> list:
     return docs
 
 # Path to your data.json
-file_path = "/home/wendy/my_hf_agent_course_projects/src/data.json"
+file_path = "/home/wendy/Downloads/questions.json"
+
+def load_data(file_path: str) -> List[Dict]:
+    """Safe JSON data loading with error handling"""
+    if not os.path.exists(file_path):
+        raise FileNotFoundError(f"Data file not found: {file_path}")
+        
+    if not file_path.endswith('.json'):
+        raise ValueError("Invalid file format. Only JSON files supported")
+        
+    try:
+        with open(file_path, "r", encoding="utf-8") as f:
+            return json.load(f)
+    except json.JSONDecodeError:
+        raise ValueError("Invalid JSON format in data file")
+    except Exception as e:
+        raise RuntimeError(f"Error loading data: {str(e)}")
 
-# Check if the file exists
-if os.path.exists(file_path):
-    # Load the data from the JSON file
-    with open(file_path, "r", encoding="utf-8") as f:
-        data = json.load(f)
-    
-    # Create documents from the loaded data
-    docs = create_documents("json", data)
-    print(f"Documents created: {len(docs)}")
-else:
-    print(f"Error: File {file_path} not found.")