Update agent.py

agent.py

@@ -1,6 +1,6 @@
 import requests
 from typing import Dict
-from retriever import retrieve_context
+from retriever import retrieve_relevant_context
 from tools import clean_answer_with_prompt, build_prompt
 from transformers import pipeline
 
@@ -19,7 +19,7 @@ def run_agent_on_question(task: Dict) -> str:
     file_name = task.get('file_name', '')
 
     # Step 1: Retrieve relevant context (as list of strings)
-    context_list = retrieve_context(task_id, question)
+    context_list = retrieve_relevant_context(task_id, question)
     context = "\n".join(context_list)
 
     # Step 2: Build a prompt using system instruction + context + question